The-Homebrew-Cloud/Cemu
2
0
Fork 0
mirror of https://github.com/cemu-project/Cemu.git synced 2024-11-25 02:26:54 +01:00
Code Issues Packages Projects Releases Wiki Activity
main
Cemu/dependencies/ih264d/decoder/ih264d_parse_islice.c

1498 lines
58 KiB
C
Raw Permalink Normal View History

Add all the files
2022-08-22 22:21:23 +02:00
/******************************************************************************
*
* 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_parse_islice.c
*
* \brief
* Contains routines that decode a I slice type
*
* Detailed_description
*
* \date
* 07/07/2003
*
* \author NS
**************************************************************************
*/
#include <string.h>
#include "ih264_defs.h"
#include "ih264d_error_handler.h"
#include "ih264d_debug.h"
#include "ih264d_bitstrm.h"
#include "ih264d_defs.h"
#include "ih264d_debug.h"
#include "ih264d_tables.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_cabac.h"
#include "ih264d_parse_cabac.h"
#include "ih264d_parse_mb_header.h"
#include "ih264d_parse_slice.h"
#include "ih264d_process_pslice.h"
#include "ih264d_process_intra_mb.h"
#include "ih264d_parse_islice.h"
#include "ih264d_error_handler.h"
#include "ih264d_mvpred.h"
#include "ih264d_defs.h"
#include "ih264d_thread_parse_decode.h"
#include "ithread.h"
#include "ih264d_parse_mb_header.h"
#include "assert.h"
#include "ih264d_utils.h"
#include "ih264d_format_conv.h"
void ih264d_init_cabac_contexts(UWORD8 u1_slice_type, dec_struct_t * ps_dec);
void ih264d_itrans_recon_luma_dc(dec_struct_t *ps_dec,
WORD16* pi2_src,
WORD16* pi2_coeff_block,
const UWORD16 *pu2_weigh_mat);
/*!
**************************************************************************
* \if Function name : ParseIMb \endif
*
* \brief
* This function parses CAVLC syntax of a I MB. If 16x16 Luma DC transform
* is also done here. Transformed Luma DC values are copied in their
* 0th pixel location of corrosponding CoeffBlock.
*
* \return
* 0 on Success and Error code otherwise
**************************************************************************
*/
WORD32 ih264d_parse_imb_cavlc(dec_struct_t * ps_dec,
dec_mb_info_t * ps_cur_mb_info,
UWORD8 u1_mb_num,
UWORD8 u1_mb_type)
{
WORD32 i4_delta_qp;
UWORD32 u4_temp;
UWORD32 ui_is_top_mb_available;
UWORD32 ui_is_left_mb_available;
UWORD32 u4_cbp;
UWORD32 u4_offset;
UWORD32 *pu4_bitstrm_buf;
WORD32 ret;
dec_bit_stream_t * const ps_bitstrm = ps_dec->ps_bitstrm;
UWORD32 *pu4_bitstrm_ofst = &ps_bitstrm->u4_ofst;
UNUSED(u1_mb_num);
ps_cur_mb_info->u1_tran_form8x8 = 0;
ps_cur_mb_info->ps_curmb->u1_tran_form8x8 = 0;
ps_cur_mb_info->u1_yuv_dc_block_flag = 0;
u4_temp = ps_dec->u1_mb_ngbr_availablity;
ui_is_top_mb_available = BOOLEAN(u4_temp & TOP_MB_AVAILABLE_MASK);
ui_is_left_mb_available = BOOLEAN(u4_temp & LEFT_MB_AVAILABLE_MASK);
pu4_bitstrm_buf = ps_bitstrm->pu4_buffer;
if(u1_mb_type == I_4x4_MB)
{
ps_cur_mb_info->ps_curmb->u1_mb_type = I_4x4_MB;
u4_offset = 0;
/*--------------------------------------------------------------------*/
/* Read transform_size_8x8_flag if present */
/*--------------------------------------------------------------------*/
if(ps_dec->s_high_profile.u1_transform8x8_present)
{
ps_cur_mb_info->u1_tran_form8x8 = ih264d_get_bit_h264(ps_bitstrm);
COPYTHECONTEXT("transform_size_8x8_flag", ps_cur_mb_info->u1_tran_form8x8);
ps_cur_mb_info->ps_curmb->u1_tran_form8x8 = ps_cur_mb_info->u1_tran_form8x8;
}
/*--------------------------------------------------------------------*/
/* Read the IntraPrediction modes for LUMA */
/*--------------------------------------------------------------------*/
if (!ps_cur_mb_info->u1_tran_form8x8)
{
UWORD8 *pu1_temp;
ih264d_read_intra_pred_modes(ps_dec,
((UWORD8 *)ps_dec->pv_parse_tu_coeff_data),
((UWORD8 *)ps_dec->pv_parse_tu_coeff_data+16),
ps_cur_mb_info->u1_tran_form8x8);
pu1_temp = (UWORD8 *)ps_dec->pv_parse_tu_coeff_data;
pu1_temp += 32;
ps_dec->pv_parse_tu_coeff_data = (void *)pu1_temp;
}
else
{
UWORD8 *pu1_temp;
ih264d_read_intra_pred_modes(ps_dec,
((UWORD8 *)ps_dec->pv_parse_tu_coeff_data),
((UWORD8 *)ps_dec->pv_parse_tu_coeff_data+4),
ps_cur_mb_info->u1_tran_form8x8);
pu1_temp = (UWORD8 *)ps_dec->pv_parse_tu_coeff_data;
pu1_temp += 8;
ps_dec->pv_parse_tu_coeff_data = (void *)pu1_temp;
}
/*--------------------------------------------------------------------*/
/* Read the IntraPrediction mode for CHROMA */
/*--------------------------------------------------------------------*/
//Inlined ih264d_uev
{
UWORD32 u4_bitstream_offset = *pu4_bitstrm_ofst;
UWORD32 u4_word, u4_ldz, u4_temp;
/***************************************************************/
/* Find leading zeros in next 32 bits */
/***************************************************************/
NEXTBITS_32(u4_word, u4_bitstream_offset, pu4_bitstrm_buf);
u4_ldz = CLZ(u4_word);
/* Flush the ps_bitstrm */
u4_bitstream_offset += (u4_ldz + 1);
/* Read the suffix from the ps_bitstrm */
u4_word = 0;
if(u4_ldz)
{
GETBITS(u4_word, u4_bitstream_offset, pu4_bitstrm_buf,
u4_ldz);
}
*pu4_bitstrm_ofst = u4_bitstream_offset;
u4_temp = ((1 << u4_ldz) + u4_word - 1);
if(u4_temp > 3)
{
return ERROR_CHROMA_PRED_MODE;
}
ps_cur_mb_info->u1_chroma_pred_mode = u4_temp;
COPYTHECONTEXT("intra_chroma_pred_mode", ps_cur_mb_info->u1_chroma_pred_mode);
}
/*--------------------------------------------------------------------*/
/* Read the Coded block pattern */
/*--------------------------------------------------------------------*/
{
UWORD32 u4_bitstream_offset = *pu4_bitstrm_ofst;
UWORD32 u4_word, u4_ldz;
/***************************************************************/
/* Find leading zeros in next 32 bits */
/***************************************************************/
NEXTBITS_32(u4_word, u4_bitstream_offset, pu4_bitstrm_buf);
u4_ldz = CLZ(u4_word);
/* Flush the ps_bitstrm */
u4_bitstream_offset += (u4_ldz + 1);
/* Read the suffix from the ps_bitstrm */
u4_word = 0;
if(u4_ldz)
{
GETBITS(u4_word, u4_bitstream_offset, pu4_bitstrm_buf,
u4_ldz);
}
*pu4_bitstrm_ofst = u4_bitstream_offset;
u4_cbp = ((1 << u4_ldz) + u4_word - 1);
}
if(u4_cbp > 47)
{
return ERROR_CBP;
}
u4_cbp = gau1_ih264d_cbp_table[u4_cbp][0];
COPYTHECONTEXT("coded_block_pattern", u1_cbp);
ps_cur_mb_info->u1_cbp = u4_cbp;
/*--------------------------------------------------------------------*/
/* Read mb_qp_delta */
/*--------------------------------------------------------------------*/
if(ps_cur_mb_info->u1_cbp)
{
UWORD32 u4_bitstream_offset = *pu4_bitstrm_ofst;
UWORD32 u4_word, u4_ldz, u4_abs_val;
/***************************************************************/
/* Find leading zeros in next 32 bits */
/***************************************************************/
NEXTBITS_32(u4_word, u4_bitstream_offset, pu4_bitstrm_buf);
u4_ldz = CLZ(u4_word);
/* Flush the ps_bitstrm */
u4_bitstream_offset += (u4_ldz + 1);
/* Read the suffix from the ps_bitstrm */
u4_word = 0;
if(u4_ldz)
{
GETBITS(u4_word, u4_bitstream_offset, pu4_bitstrm_buf,
u4_ldz);
}
*pu4_bitstrm_ofst = u4_bitstream_offset;
u4_abs_val = ((1 << u4_ldz) + u4_word) >> 1;
if(u4_word & 0x1)
{
i4_delta_qp = (-(WORD32)u4_abs_val);
}
else
{
i4_delta_qp = (u4_abs_val);
}
if((i4_delta_qp < -26) || (i4_delta_qp > 25))
{
return ERROR_INV_RANGE_QP_T;
}
COPYTHECONTEXT("mb_qp_delta", i1_delta_qp);
if(i4_delta_qp != 0)
{
ret = ih264d_update_qp(ps_dec, (WORD8)i4_delta_qp);
if(ret != OK)
return ret;
}
}
}
else
{
u4_offset = 1;
ps_cur_mb_info->ps_curmb->u1_mb_type = I_16x16_MB;
/*-------------------------------------------------------------------*/
/* Read the IntraPrediction mode for CHROMA */
/*-------------------------------------------------------------------*/
{
UWORD32 u4_bitstream_offset = *pu4_bitstrm_ofst;
UWORD32 u4_word, u4_ldz;
/***************************************************************/
/* Find leading zeros in next 32 bits */
/***************************************************************/
NEXTBITS_32(u4_word, u4_bitstream_offset, pu4_bitstrm_buf);
u4_ldz = CLZ(u4_word);
/* Flush the ps_bitstrm */
u4_bitstream_offset += (u4_ldz + 1);
/* Read the suffix from the ps_bitstrm */
u4_word = 0;
if(u4_ldz)
{
GETBITS(u4_word, u4_bitstream_offset, pu4_bitstrm_buf,
u4_ldz);
}
*pu4_bitstrm_ofst = u4_bitstream_offset;
u4_temp = ((1 << u4_ldz) + u4_word - 1);
//Inlined ih264d_uev
if(u4_temp > 3)
{
return ERROR_CHROMA_PRED_MODE;
}
ps_cur_mb_info->u1_chroma_pred_mode = u4_temp;
COPYTHECONTEXT("intra_chroma_pred_mode", ps_cur_mb_info->u1_chroma_pred_mode);
}
/*-------------------------------------------------------------------*/
/* Read the Coded block pattern */
/*-------------------------------------------------------------------*/
u4_cbp = gau1_ih264d_cbp_tab[(u1_mb_type - 1) >> 2];
ps_cur_mb_info->u1_cbp = u4_cbp;
/*-------------------------------------------------------------------*/
/* Read mb_qp_delta */
/*-------------------------------------------------------------------*/
{
UWORD32 u4_bitstream_offset = *pu4_bitstrm_ofst;
UWORD32 u4_word, u4_ldz, u4_abs_val;
/***************************************************************/
/* Find leading zeros in next 32 bits */
/***************************************************************/
NEXTBITS_32(u4_word, u4_bitstream_offset, pu4_bitstrm_buf);
u4_ldz = CLZ(u4_word);
/* Flush the ps_bitstrm */
u4_bitstream_offset += (u4_ldz + 1);
/* Read the suffix from the ps_bitstrm */
u4_word = 0;
if(u4_ldz)
GETBITS(u4_word, u4_bitstream_offset, pu4_bitstrm_buf,
u4_ldz);
*pu4_bitstrm_ofst = u4_bitstream_offset;
u4_abs_val = ((1 << u4_ldz) + u4_word) >> 1;
if(u4_word & 0x1)
i4_delta_qp = (-(WORD32)u4_abs_val);
else
i4_delta_qp = (u4_abs_val);
if((i4_delta_qp < -26) || (i4_delta_qp > 25))
return ERROR_INV_RANGE_QP_T;
}
//inlinined ih264d_sev
COPYTHECONTEXT("Delta quant", i1_delta_qp);
if(i4_delta_qp != 0)
{
ret = ih264d_update_qp(ps_dec, (WORD8)i4_delta_qp);
if(ret != OK)
return ret;
}
{
WORD16 i_scaleFactor;
UWORD32 ui_N = 0;
WORD16 *pi2_scale_matrix_ptr;
/*******************************************************************/
/* for luma DC coefficients the scaling is done during the parsing */
/* to preserve the precision */
/*******************************************************************/
if(ps_dec->s_high_profile.u1_scaling_present)
{
pi2_scale_matrix_ptr =
ps_dec->s_high_profile.i2_scalinglist4x4[0];
}
else
{
i_scaleFactor = 16;
pi2_scale_matrix_ptr = &i_scaleFactor;
}
/*---------------------------------------------------------------*/
/* Decode DC coefficients */
/*---------------------------------------------------------------*/
/*---------------------------------------------------------------*/
/* Calculation of N */
/*---------------------------------------------------------------*/
if(ui_is_left_mb_available)
{
if(ui_is_top_mb_available)
{
ui_N = ((ps_cur_mb_info->ps_top_mb->pu1_nnz_y[0]
+ ps_dec->pu1_left_nnz_y[0] + 1) >> 1);
}
else
{
ui_N = ps_dec->pu1_left_nnz_y[0];
}
}
else if(ui_is_top_mb_available)
{
ui_N = ps_cur_mb_info->ps_top_mb->pu1_nnz_y[0];
}
{
WORD16 pi2_dc_coef[16];
WORD32 pi4_tmp[16];
tu_sblk4x4_coeff_data_t *ps_tu_4x4 =
(tu_sblk4x4_coeff_data_t *)ps_dec->pv_parse_tu_coeff_data;
WORD16 *pi2_coeff_block =
(WORD16 *)ps_dec->pv_parse_tu_coeff_data;
UWORD32 u4_num_coeff;
ps_tu_4x4->u2_sig_coeff_map = 0;
ret = ps_dec->pf_cavlc_parse4x4coeff[(ui_N > 7)](pi2_dc_coef, 0, ui_N,
ps_dec, &u4_num_coeff);
if(ret != OK)
return ret;
if(EXCEED_OFFSET(ps_bitstrm))
return ERROR_EOB_TERMINATE_T;
if(ps_tu_4x4->u2_sig_coeff_map)
{
memset(pi2_dc_coef,0,sizeof(pi2_dc_coef));
ih264d_unpack_coeff4x4_dc_4x4blk(ps_tu_4x4,
pi2_dc_coef,
ps_dec->pu1_inv_scan);
PROFILE_DISABLE_IQ_IT_RECON()
ps_dec->pf_ihadamard_scaling_4x4(pi2_dc_coef,
pi2_coeff_block,
ps_dec->pu2_quant_scale_y,
(UWORD16 *)pi2_scale_matrix_ptr,
ps_dec->u1_qp_y_div6,
pi4_tmp);
pi2_coeff_block += 16;
ps_dec->pv_parse_tu_coeff_data = (void *)pi2_coeff_block;
SET_BIT(ps_cur_mb_info->u1_yuv_dc_block_flag,0);
}
}
}
}
if(u4_cbp)
{
ret = ih264d_parse_residual4x4_cavlc(ps_dec, ps_cur_mb_info,
(UWORD8)u4_offset);
if(ret != OK)
return ret;
if(EXCEED_OFFSET(ps_bitstrm))
return ERROR_EOB_TERMINATE_T;
/* Store Left Mb NNZ and TOP chroma NNZ */
}
else
{
ps_cur_mb_info->u1_qp_div6 = ps_dec->u1_qp_y_div6;
ps_cur_mb_info->u1_qpc_div6 = ps_dec->u1_qp_u_div6;
ps_cur_mb_info->u1_qpcr_div6 = ps_dec->u1_qp_v_div6;
ps_cur_mb_info->u1_qp_rem6 = ps_dec->u1_qp_y_rem6;
ps_cur_mb_info->u1_qpc_rem6 = ps_dec->u1_qp_u_rem6;
ps_cur_mb_info->u1_qpcr_rem6 = ps_dec->u1_qp_v_rem6;
ih264d_update_nnz_for_skipmb(ps_dec, ps_cur_mb_info, CAVLC);
}
return OK;
}
/*!
**************************************************************************
* \if Function name : ParseIMbCab \endif
*
* \brief
* This function parses CABAC syntax of a I MB. If 16x16 Luma DC transform
* is also done here. Transformed Luma DC values are copied in their
* 0th pixel location of corrosponding CoeffBlock.
*
* \return
* 0 on Success and Error code otherwise
**************************************************************************
*/
WORD32 ih264d_parse_imb_cabac(dec_struct_t * ps_dec,
dec_mb_info_t * ps_cur_mb_info,
UWORD8 u1_mb_type)
{
WORD8 i1_delta_qp;
UWORD8 u1_cbp;
UWORD8 u1_offset;
/* Variables for handling Cabac contexts */
ctxt_inc_mb_info_t *p_curr_ctxt = ps_dec->ps_curr_ctxt_mb_info;
ctxt_inc_mb_info_t *ps_left_ctxt = ps_dec->p_left_ctxt_mb_info;
dec_bit_stream_t * const ps_bitstrm = ps_dec->ps_bitstrm;
bin_ctxt_model_t *p_bin_ctxt;
UWORD8 u1_intra_chrom_pred_mode;
UWORD8 u1_dc_block_flag = 0;
WORD32 ret;
ps_cur_mb_info->u1_yuv_dc_block_flag = 0;
if(ps_left_ctxt == ps_dec->ps_def_ctxt_mb_info)
{
ps_dec->pu1_left_yuv_dc_csbp[0] = 0xf;
}
if(ps_dec->ps_cur_slice->u1_slice_type != I_SLICE)
{
WORD32 *pi4_buf;
WORD8 *pi1_buf;
MEMSET_16BYTES(&ps_dec->pu1_left_mv_ctxt_inc[0][0], 0);
*((UWORD32 *)ps_dec->pi1_left_ref_idx_ctxt_inc) = 0;
MEMSET_16BYTES(p_curr_ctxt->u1_mv, 0);
memset(p_curr_ctxt->i1_ref_idx, 0, 4);
}
if(u1_mb_type == I_4x4_MB)
{
ps_cur_mb_info->ps_curmb->u1_mb_type = I_4x4_MB;
p_curr_ctxt->u1_mb_type = CAB_I4x4;
u1_offset = 0;
ps_cur_mb_info->u1_tran_form8x8 = 0;
ps_cur_mb_info->ps_curmb->u1_tran_form8x8 = 0;
/*--------------------------------------------------------------------*/
/* Read transform_size_8x8_flag if present */
/*--------------------------------------------------------------------*/
if(ps_dec->s_high_profile.u1_transform8x8_present)
{
ps_cur_mb_info->u1_tran_form8x8 = ih264d_parse_transform8x8flag_cabac(
ps_dec, ps_cur_mb_info);
COPYTHECONTEXT("transform_size_8x8_flag", ps_cur_mb_info->u1_tran_form8x8);
p_curr_ctxt->u1_transform8x8_ctxt = ps_cur_mb_info->u1_tran_form8x8;
ps_cur_mb_info->ps_curmb->u1_tran_form8x8 = ps_cur_mb_info->u1_tran_form8x8;
}
else
{
p_curr_ctxt->u1_transform8x8_ctxt = 0;
}
/*--------------------------------------------------------------------*/
/* Read the IntraPrediction modes for LUMA */
/*--------------------------------------------------------------------*/
if (!ps_cur_mb_info->u1_tran_form8x8)
{
UWORD8 *pu1_temp;
ih264d_read_intra_pred_modes_cabac(
ps_dec,
((UWORD8 *)ps_dec->pv_parse_tu_coeff_data),
((UWORD8 *)ps_dec->pv_parse_tu_coeff_data+16),
ps_cur_mb_info->u1_tran_form8x8);
pu1_temp = (UWORD8 *)ps_dec->pv_parse_tu_coeff_data;
pu1_temp += 32;
ps_dec->pv_parse_tu_coeff_data = (void *)pu1_temp;
}
else
{
UWORD8 *pu1_temp;
ih264d_read_intra_pred_modes_cabac(
ps_dec,
((UWORD8 *)ps_dec->pv_parse_tu_coeff_data),
((UWORD8 *)ps_dec->pv_parse_tu_coeff_data+4),
ps_cur_mb_info->u1_tran_form8x8);
pu1_temp = (UWORD8 *)ps_dec->pv_parse_tu_coeff_data;
pu1_temp += 8;
ps_dec->pv_parse_tu_coeff_data = (void *)pu1_temp;
}
/*--------------------------------------------------------------------*/
/* Read the IntraPrediction mode for CHROMA */
/*--------------------------------------------------------------------*/
u1_intra_chrom_pred_mode = ih264d_parse_chroma_pred_mode_cabac(ps_dec);
COPYTHECONTEXT("intra_chroma_pred_mode", u1_intra_chrom_pred_mode);
p_curr_ctxt->u1_intra_chroma_pred_mode = ps_cur_mb_info->u1_chroma_pred_mode =
u1_intra_chrom_pred_mode;
/*--------------------------------------------------------------------*/
/* Read the Coded block pattern */
/*--------------------------------------------------------------------*/
u1_cbp = ih264d_parse_ctx_cbp_cabac(ps_dec);
COPYTHECONTEXT("coded_block_pattern", u1_cbp);
ps_cur_mb_info->u1_cbp = u1_cbp;
p_curr_ctxt->u1_cbp = u1_cbp;
/*--------------------------------------------------------------------*/
/* Read mb_qp_delta */
/*--------------------------------------------------------------------*/
if(ps_cur_mb_info->u1_cbp)
{
ret = ih264d_parse_mb_qp_delta_cabac(ps_dec, &i1_delta_qp);
if(ret != OK)
return ret;
COPYTHECONTEXT("mb_qp_delta", i1_delta_qp);
if(i1_delta_qp != 0)
{
ret = ih264d_update_qp(ps_dec, i1_delta_qp);
if(ret != OK)
return ret;
}
}
else
ps_dec->i1_prev_mb_qp_delta = 0;
p_curr_ctxt->u1_yuv_dc_csbp &= 0xFE;
}
else
{
u1_offset = 1;
ps_cur_mb_info->ps_curmb->u1_mb_type = I_16x16_MB;
p_curr_ctxt->u1_mb_type = CAB_I16x16;
ps_cur_mb_info->u1_tran_form8x8 = 0;
p_curr_ctxt->u1_transform8x8_ctxt = 0;
ps_cur_mb_info->ps_curmb->u1_tran_form8x8 = 0;
/*--------------------------------------------------------------------*/
/* Read the IntraPrediction mode for CHROMA */
/*--------------------------------------------------------------------*/
u1_intra_chrom_pred_mode = ih264d_parse_chroma_pred_mode_cabac(ps_dec);
if(u1_intra_chrom_pred_mode > 3)
return ERROR_CHROMA_PRED_MODE;
COPYTHECONTEXT("Chroma intra_chroma_pred_mode pred mode", u1_intra_chrom_pred_mode);
p_curr_ctxt->u1_intra_chroma_pred_mode = ps_cur_mb_info->u1_chroma_pred_mode =
u1_intra_chrom_pred_mode;
/*--------------------------------------------------------------------*/
/* Read the Coded block pattern */
/*--------------------------------------------------------------------*/
u1_cbp = gau1_ih264d_cbp_tab[(u1_mb_type - 1) >> 2];
ps_cur_mb_info->u1_cbp = u1_cbp;
p_curr_ctxt->u1_cbp = u1_cbp;
/*--------------------------------------------------------------------*/
/* Read mb_qp_delta */
/*--------------------------------------------------------------------*/
ret = ih264d_parse_mb_qp_delta_cabac(ps_dec, &i1_delta_qp);
if(ret != OK)
return ret;
COPYTHECONTEXT("mb_qp_delta", i1_delta_qp);
if(i1_delta_qp != 0)
{
ret = ih264d_update_qp(ps_dec, i1_delta_qp);
if(ret != OK)
return ret;
}
{
WORD16 i_scaleFactor;
WORD16* pi2_scale_matrix_ptr;
/*******************************************************************/
/* for luma DC coefficients the scaling is done during the parsing */
/* to preserve the precision */
/*******************************************************************/
if(ps_dec->s_high_profile.u1_scaling_present)
{
pi2_scale_matrix_ptr =
ps_dec->s_high_profile.i2_scalinglist4x4[0];
}
else
{
i_scaleFactor = 16;
pi2_scale_matrix_ptr = &i_scaleFactor;
}
{
ctxt_inc_mb_info_t *ps_top_ctxt = ps_dec->p_top_ctxt_mb_info;
UWORD8 uc_a, uc_b;
UWORD32 u4_ctx_inc;
INC_SYM_COUNT(&(ps_dec->s_cab_dec_env));
/* if MbAddrN not available then CondTermN = 1 */
uc_b = ((ps_top_ctxt->u1_yuv_dc_csbp) & 0x01);
/* if MbAddrN not available then CondTermN = 1 */
uc_a = ((ps_dec->pu1_left_yuv_dc_csbp[0]) & 0x01);
u4_ctx_inc = (uc_a + (uc_b << 1));
{
WORD16 pi2_dc_coef[16];
tu_sblk4x4_coeff_data_t *ps_tu_4x4 =
(tu_sblk4x4_coeff_data_t *)ps_dec->pv_parse_tu_coeff_data;
WORD16 *pi2_coeff_block =
(WORD16 *)ps_dec->pv_parse_tu_coeff_data;
p_bin_ctxt = (ps_dec->p_cbf_t[LUMA_DC_CTXCAT]) + u4_ctx_inc;
u1_dc_block_flag =
ih264d_read_coeff4x4_cabac(ps_bitstrm,
LUMA_DC_CTXCAT,
ps_dec->p_significant_coeff_flag_t[LUMA_DC_CTXCAT],
ps_dec, p_bin_ctxt);
/* Store coded_block_flag */
p_curr_ctxt->u1_yuv_dc_csbp &= 0xFE;
p_curr_ctxt->u1_yuv_dc_csbp |= u1_dc_block_flag;
if(u1_dc_block_flag)
{
WORD32 pi4_tmp[16];
memset(pi2_dc_coef,0,sizeof(pi2_dc_coef));
ih264d_unpack_coeff4x4_dc_4x4blk(ps_tu_4x4,
pi2_dc_coef,
ps_dec->pu1_inv_scan);
PROFILE_DISABLE_IQ_IT_RECON()
ps_dec->pf_ihadamard_scaling_4x4(pi2_dc_coef,
pi2_coeff_block,
ps_dec->pu2_quant_scale_y,
(UWORD16 *)pi2_scale_matrix_ptr,
ps_dec->u1_qp_y_div6,
pi4_tmp);
pi2_coeff_block += 16;
ps_dec->pv_parse_tu_coeff_data = (void *)pi2_coeff_block;
SET_BIT(ps_cur_mb_info->u1_yuv_dc_block_flag,0);
}
}
}
}
}
ps_dec->pu1_left_yuv_dc_csbp[0] &= 0x6;
ps_dec->pu1_left_yuv_dc_csbp[0] |= u1_dc_block_flag;
ih264d_parse_residual4x4_cabac(ps_dec, ps_cur_mb_info, u1_offset);
if(EXCEED_OFFSET(ps_bitstrm))
return ERROR_EOB_TERMINATE_T;
return OK;
}
/*****************************************************************************/
/* */
/* Function Name : ih264d_parse_islice_data_cavlc */
/* */
/* Description : This function parses cabac syntax of a inter slice on */
/* N MB basis. */
/* */
/* Inputs : ps_dec */
/* sliceparams */
/* firstMbInSlice */
/* */
/* Processing : 1. After parsing syntax for N MBs those N MBs are */
/* decoded till the end of slice. */
/* */
/* Returns : 0 */
/* */
/* Issues : <List any issues or problems with this function> */
/* */
/* Revision History: */
/* */
/* DD MM YYYY Author(s) Changes (Describe the changes made) */
/* 24 06 2005 ARNY Draft */
/* */
/*****************************************************************************/
WORD32 ih264d_parse_islice_data_cavlc(dec_struct_t * ps_dec,
dec_slice_params_t * ps_slice,
UWORD16 u2_first_mb_in_slice)
{
UWORD8 uc_more_data_flag;
UWORD8 u1_num_mbs, u1_mb_idx;
dec_mb_info_t *ps_cur_mb_info;
deblk_mb_t *ps_cur_deblk_mb;
dec_bit_stream_t * const ps_bitstrm = ps_dec->ps_bitstrm;
UWORD32 *pu4_bitstrm_ofst = &ps_bitstrm->u4_ofst;
UWORD32 *pu4_bitstrm_buf = ps_bitstrm->pu4_buffer;
UWORD16 i2_pic_wdin_mbs = ps_dec->u2_frm_wd_in_mbs;
WORD16 i2_cur_mb_addr;
UWORD8 u1_mbaff;
UWORD8 u1_num_mbs_next, u1_end_of_row, u1_tfr_n_mb;
WORD32 ret = OK;
ps_dec->u1_qp = ps_slice->u1_slice_qp;
ih264d_update_qp(ps_dec, 0);
u1_mbaff = ps_slice->u1_mbaff_frame_flag;
/* initializations */
u1_mb_idx = ps_dec->u1_mb_idx;
u1_num_mbs = u1_mb_idx;
uc_more_data_flag = 1;
i2_cur_mb_addr = u2_first_mb_in_slice << u1_mbaff;
do
{
UWORD8 u1_mb_type;
ps_dec->pv_prev_mb_parse_tu_coeff_data = ps_dec->pv_parse_tu_coeff_data;
if(i2_cur_mb_addr > ps_dec->ps_cur_sps->u2_max_mb_addr)
{
break;
}
ps_cur_mb_info = ps_dec->ps_nmb_info + u1_num_mbs;
ps_dec->u4_num_mbs_cur_nmb = u1_num_mbs;
ps_dec->u4_num_pmbair = (u1_num_mbs >> u1_mbaff);
ps_cur_mb_info->u1_end_of_slice = 0;
/***************************************************************/
/* Get the required information for decoding of MB */
/* mb_x, mb_y , neighbour availablity, */
/***************************************************************/
ps_dec->pf_get_mb_info(ps_dec, i2_cur_mb_addr, ps_cur_mb_info, 0);
/***************************************************************/
/* Set the deblocking parameters for this MB */
/***************************************************************/
ps_cur_deblk_mb = ps_dec->ps_deblk_mbn + u1_num_mbs;
if(ps_dec->u4_app_disable_deblk_frm == 0)
ih264d_set_deblocking_parameters(ps_cur_deblk_mb, ps_slice,
ps_dec->u1_mb_ngbr_availablity,
ps_dec->u1_cur_mb_fld_dec_flag);
ps_cur_deblk_mb->u1_mb_type = ps_cur_deblk_mb->u1_mb_type | D_INTRA_MB;
/**************************************************************/
/* Macroblock Layer Begins, Decode the u1_mb_type */
/**************************************************************/
//Inlined ih264d_uev
{
UWORD32 u4_bitstream_offset = *pu4_bitstrm_ofst;
UWORD32 u4_word, u4_ldz, u4_temp;
/***************************************************************/
/* Find leading zeros in next 32 bits */
/***************************************************************/
NEXTBITS_32(u4_word, u4_bitstream_offset, pu4_bitstrm_buf);
u4_ldz = CLZ(u4_word);
/* Flush the ps_bitstrm */
u4_bitstream_offset += (u4_ldz + 1);
/* Read the suffix from the ps_bitstrm */
u4_word = 0;
if(u4_ldz)
GETBITS(u4_word, u4_bitstream_offset, pu4_bitstrm_buf,
u4_ldz);
*pu4_bitstrm_ofst = u4_bitstream_offset;
u4_temp = ((1 << u4_ldz) + u4_word - 1);
if(u4_temp > 25)
return ERROR_MB_TYPE;
u1_mb_type = u4_temp;
}
//Inlined ih264d_uev
ps_cur_mb_info->u1_mb_type = u1_mb_type;
COPYTHECONTEXT("u1_mb_type", u1_mb_type);
/**************************************************************/
/* Parse Macroblock data */
/**************************************************************/
if(25 == u1_mb_type)
{
/* I_PCM_MB */
ps_cur_mb_info->ps_curmb->u1_mb_type = I_PCM_MB;
ret = ih264d_parse_ipcm_mb(ps_dec, ps_cur_mb_info, u1_num_mbs);
if(ret != OK)
return ret;
ps_cur_deblk_mb->u1_mb_qp = 0;
}
else
{
ret = ih264d_parse_imb_cavlc(ps_dec, ps_cur_mb_info, u1_num_mbs, u1_mb_type);
if(ret != OK)
return ret;
ps_cur_deblk_mb->u1_mb_qp = ps_dec->u1_qp;
}
uc_more_data_flag = MORE_RBSP_DATA(ps_bitstrm);
if(u1_mbaff)
{
ih264d_update_mbaff_left_nnz(ps_dec, ps_cur_mb_info);
if(!uc_more_data_flag && (0 == (i2_cur_mb_addr & 1)))
{
return ERROR_EOB_FLUSHBITS_T;
}
}
/**************************************************************/
/* Get next Macroblock address */
/**************************************************************/
i2_cur_mb_addr++;
/* Store the colocated information */
{
mv_pred_t *ps_mv_nmb_start = ps_dec->ps_mv_cur + (u1_num_mbs << 4);
mv_pred_t s_mvPred =
{
{ 0, 0, 0, 0 },
{ -1, -1 }, 0, 0};
ih264d_rep_mv_colz(ps_dec, &s_mvPred, ps_mv_nmb_start, 0,
(UWORD8)(ps_dec->u1_cur_mb_fld_dec_flag << 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)(u1_num_mbs >> u1_mbaff));
}
u1_num_mbs++;
/****************************************************************/
/* Check for End Of Row */
/****************************************************************/
u1_num_mbs_next = i2_pic_wdin_mbs - ps_dec->u2_mbx - 1;
u1_end_of_row = (!u1_num_mbs_next) && (!(u1_mbaff && (u1_num_mbs & 0x01)));
u1_tfr_n_mb = (u1_num_mbs == ps_dec->u1_recon_mb_grp) || u1_end_of_row
|| (!uc_more_data_flag);
ps_cur_mb_info->u1_end_of_slice = (!uc_more_data_flag);
/*H264_DEC_DEBUG_PRINT("Pic: %d Mb_X=%d Mb_Y=%d",
ps_slice->i4_poc >> ps_slice->u1_field_pic_flag,
ps_dec->u2_mbx,ps_dec->u2_mby + (1 - ps_cur_mb_info->u1_topmb));
H264_DEC_DEBUG_PRINT("u1_tfr_n_mb || (!uc_more_data_flag): %d", u1_tfr_n_mb || (!uc_more_data_flag));*/
if(u1_tfr_n_mb || (!uc_more_data_flag))
{
if(ps_dec->u1_separate_parse)
{
ih264d_parse_tfr_nmb(ps_dec, u1_mb_idx, u1_num_mbs,
u1_num_mbs_next, u1_tfr_n_mb, u1_end_of_row);
ps_dec->ps_nmb_info += u1_num_mbs;
}
else
{
ih264d_decode_recon_tfr_nmb(ps_dec, u1_mb_idx, u1_num_mbs,
u1_num_mbs_next, u1_tfr_n_mb,
u1_end_of_row);
}
ps_dec->u2_total_mbs_coded += u1_num_mbs;
if(u1_tfr_n_mb)
u1_num_mbs = 0;
u1_mb_idx = u1_num_mbs;
ps_dec->u1_mb_idx = u1_num_mbs;
}
}
while(uc_more_data_flag);
ps_dec->u4_num_mbs_cur_nmb = 0;
ps_dec->ps_cur_slice->u4_mbs_in_slice = i2_cur_mb_addr
- (u2_first_mb_in_slice << u1_mbaff);
return ret;
}
/*****************************************************************************/
/* */
/* Function Name : ih264d_parse_islice_data_cabac */
/* */
/* Description : This function parses cabac syntax of a inter slice on */
/* N MB basis. */
/* */
/* Inputs : ps_dec */
/* sliceparams */
/* firstMbInSlice */
/* */
/* Processing : 1. After parsing syntax for N MBs those N MBs are */
/* decoded till the end of slice. */
/* */
/* Returns : 0 */
/* */
/* Issues : <List any issues or problems with this function> */
/* */
/* Revision History: */
/* */
/* DD MM YYYY Author(s) Changes (Describe the changes made) */
/* 24 06 2005 ARNY Draft */
/* */
/*****************************************************************************/
WORD32 ih264d_parse_islice_data_cabac(dec_struct_t * ps_dec,
dec_slice_params_t * ps_slice,
UWORD16 u2_first_mb_in_slice)
{
UWORD8 uc_more_data_flag;
UWORD8 u1_num_mbs, u1_mb_idx;
dec_mb_info_t *ps_cur_mb_info;
deblk_mb_t *ps_cur_deblk_mb;
dec_bit_stream_t * const ps_bitstrm = ps_dec->ps_bitstrm;
UWORD16 i2_pic_wdin_mbs = ps_dec->u2_frm_wd_in_mbs;
WORD16 i2_cur_mb_addr;
UWORD8 u1_mbaff;
UWORD8 u1_num_mbs_next, u1_end_of_row, u1_tfr_n_mb;
WORD32 ret = OK;
ps_dec->u1_qp = ps_slice->u1_slice_qp;
ih264d_update_qp(ps_dec, 0);
u1_mbaff = ps_slice->u1_mbaff_frame_flag;
if(ps_bitstrm->u4_ofst & 0x07)
{
ps_bitstrm->u4_ofst += 8;
ps_bitstrm->u4_ofst &= 0xFFFFFFF8;
}
ret = ih264d_init_cabac_dec_envirnoment(&(ps_dec->s_cab_dec_env), ps_bitstrm);
if(ret != OK)
return ret;
ih264d_init_cabac_contexts(I_SLICE, ps_dec);
ps_dec->i1_prev_mb_qp_delta = 0;
/* initializations */
u1_mb_idx = ps_dec->u1_mb_idx;
u1_num_mbs = u1_mb_idx;
uc_more_data_flag = 1;
i2_cur_mb_addr = u2_first_mb_in_slice << u1_mbaff;
do
{
UWORD16 u2_mbx;
ps_dec->pv_prev_mb_parse_tu_coeff_data = ps_dec->pv_parse_tu_coeff_data;
if(i2_cur_mb_addr > ps_dec->ps_cur_sps->u2_max_mb_addr)
{
break;
}
{
UWORD8 u1_mb_type;
ps_cur_mb_info = ps_dec->ps_nmb_info + u1_num_mbs;
ps_dec->u4_num_mbs_cur_nmb = u1_num_mbs;
ps_dec->u4_num_pmbair = (u1_num_mbs >> u1_mbaff);
ps_cur_mb_info->u1_end_of_slice = 0;
/***************************************************************/
/* Get the required information for decoding of MB */
/* mb_x, mb_y , neighbour availablity, */
/***************************************************************/
ps_dec->pf_get_mb_info(ps_dec, i2_cur_mb_addr, ps_cur_mb_info, 0);
u2_mbx = ps_dec->u2_mbx;
/*********************************************************************/
/* initialize u1_tran_form8x8 to zero to aviod uninitialized accesses */
/*********************************************************************/
ps_cur_mb_info->u1_tran_form8x8 = 0;
ps_cur_mb_info->ps_curmb->u1_tran_form8x8 = 0;
/***************************************************************/
/* Set the deblocking parameters for this MB */
/***************************************************************/
ps_cur_deblk_mb = ps_dec->ps_deblk_mbn + u1_num_mbs;
if(ps_dec->u4_app_disable_deblk_frm == 0)
ih264d_set_deblocking_parameters(
ps_cur_deblk_mb, ps_slice,
ps_dec->u1_mb_ngbr_availablity,
ps_dec->u1_cur_mb_fld_dec_flag);
ps_cur_deblk_mb->u1_mb_type = ps_cur_deblk_mb->u1_mb_type
| D_INTRA_MB;
/* Macroblock Layer Begins */
/* Decode the u1_mb_type */
u1_mb_type = ih264d_parse_mb_type_intra_cabac(0, ps_dec);
if(u1_mb_type > 25)
return ERROR_MB_TYPE;
ps_cur_mb_info->u1_mb_type = u1_mb_type;
COPYTHECONTEXT("u1_mb_type", u1_mb_type);
/* Parse Macroblock Data */
if(25 == u1_mb_type)
{
/* I_PCM_MB */
ps_cur_mb_info->ps_curmb->u1_mb_type = I_PCM_MB;
ret = ih264d_parse_ipcm_mb(ps_dec, ps_cur_mb_info, u1_num_mbs);
if(ret != OK)
return ret;
ps_cur_deblk_mb->u1_mb_qp = 0;
}
else
{
ret = ih264d_parse_imb_cabac(ps_dec, ps_cur_mb_info, u1_mb_type);
if(ret != OK)
return ret;
ps_cur_deblk_mb->u1_mb_qp = ps_dec->u1_qp;
}
if(u1_mbaff)
{
ih264d_update_mbaff_left_nnz(ps_dec, ps_cur_mb_info);
}
if(ps_cur_mb_info->u1_topmb && u1_mbaff)
uc_more_data_flag = 1;
else
{
uc_more_data_flag = ih264d_decode_terminate(&ps_dec->s_cab_dec_env,
ps_bitstrm);
uc_more_data_flag = !uc_more_data_flag;
COPYTHECONTEXT("Decode Sliceterm",!uc_more_data_flag);
}
if(u1_mbaff)
{
if(!uc_more_data_flag && (0 == (i2_cur_mb_addr & 1)))
{
return ERROR_EOB_FLUSHBITS_T;
}
}
/* Next macroblock information */
i2_cur_mb_addr++;
/* Store the colocated information */
{
mv_pred_t *ps_mv_nmb_start = ps_dec->ps_mv_cur + (u1_num_mbs << 4);
mv_pred_t s_mvPred =
{
{ 0, 0, 0, 0 },
{ -1, -1 }, 0, 0};
ih264d_rep_mv_colz(
ps_dec, &s_mvPred, ps_mv_nmb_start, 0,
(UWORD8)(ps_dec->u1_cur_mb_fld_dec_flag << 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)(u1_num_mbs >> u1_mbaff));
}
u1_num_mbs++;
}
/****************************************************************/
/* Check for End Of Row */
/****************************************************************/
u1_num_mbs_next = i2_pic_wdin_mbs - u2_mbx - 1;
u1_end_of_row = (!u1_num_mbs_next) && (!(u1_mbaff && (u1_num_mbs & 0x01)));
u1_tfr_n_mb = (u1_num_mbs == ps_dec->u1_recon_mb_grp) || u1_end_of_row
|| (!uc_more_data_flag);
ps_cur_mb_info->u1_end_of_slice = (!uc_more_data_flag);
if(u1_tfr_n_mb || (!uc_more_data_flag))
{
if(ps_dec->u1_separate_parse)
{
ih264d_parse_tfr_nmb(ps_dec, u1_mb_idx, u1_num_mbs,
u1_num_mbs_next, u1_tfr_n_mb, u1_end_of_row);
ps_dec->ps_nmb_info += u1_num_mbs;
}
else
{
ih264d_decode_recon_tfr_nmb(ps_dec, u1_mb_idx, u1_num_mbs,
u1_num_mbs_next, u1_tfr_n_mb,
u1_end_of_row);
}
ps_dec->u2_total_mbs_coded += u1_num_mbs;
if(u1_tfr_n_mb)
u1_num_mbs = 0;
u1_mb_idx = u1_num_mbs;
ps_dec->u1_mb_idx = u1_num_mbs;
}
}
while(uc_more_data_flag);
ps_dec->u4_num_mbs_cur_nmb = 0;
ps_dec->ps_cur_slice->u4_mbs_in_slice = i2_cur_mb_addr
- (u2_first_mb_in_slice << u1_mbaff);
return ret;
}
/*****************************************************************************/
/* */
/* Function Name : ih264d_parse_ipcm_mb */
/* */
/* Description : This function decodes the pixel values of I_PCM Mb. */
/* */
/* Inputs : ps_dec, ps_cur_mb_info and mb number */
/* */
/* Description : This function reads the luma and chroma pixels directly */
/* from the bitstream when the mbtype is I_PCM and stores */
/* them in recon buffer. If the entropy coding mode is */
/* cabac, decoding engine is re-initialized. The nnzs and */
/* cabac contexts are appropriately modified. */
/* Returns : void */
/* */
/* Revision History: */
/* DD MM YYYY Author(s) Changes (Describe the changes made) */
/* 13 07 2002 Jay */
/* */
/*****************************************************************************/
WORD32 ih264d_parse_ipcm_mb(dec_struct_t * ps_dec,
dec_mb_info_t *ps_cur_mb_info,
UWORD8 u1_mbNum)
{
dec_bit_stream_t * const ps_bitstrm = ps_dec->ps_bitstrm;
UWORD8 u1_mbaff = ps_dec->ps_cur_slice->u1_mbaff_frame_flag;
UWORD8 *pu1_y, *pu1_u, *pu1_v;
WORD32 ret;
UWORD32 u4_rec_width_y, u4_rec_width_uv;
UWORD32 u1_num_mb_pair;
UWORD8 u1_x, u1_y;
/* CHANGED CODE */
tfr_ctxt_t *ps_frame_buf;
UWORD8 u1_mb_field_decoding_flag;
UWORD32 *pu4_buf;
UWORD8 *pu1_buf;
/* CHANGED CODE */
if(ps_dec->u1_separate_parse)
{
ps_frame_buf = &ps_dec->s_tran_addrecon_parse;
}
else
{
ps_frame_buf = &ps_dec->s_tran_addrecon;
}
/* align bistream to byte boundary. */
/* pcm_alignment_zero_bit discarded */
/* For XX GotoByteBoundary */
if(ps_bitstrm->u4_ofst & 0x07)
{
ps_bitstrm->u4_ofst += 8;
ps_bitstrm->u4_ofst &= 0xFFFFFFF8;
}
/* Store left Nnz as 16 for each 4x4 blk */
pu1_buf = ps_dec->pu1_left_nnz_y;
pu4_buf = (UWORD32 *)pu1_buf;
*pu4_buf = 0x10101010;
pu1_buf = ps_cur_mb_info->ps_curmb->pu1_nnz_y;
pu4_buf = (UWORD32 *)pu1_buf;
*pu4_buf = 0x10101010;
pu1_buf = ps_cur_mb_info->ps_curmb->pu1_nnz_uv;
pu4_buf = (UWORD32 *)pu1_buf;
*pu4_buf = 0x10101010;
pu1_buf = ps_dec->pu1_left_nnz_uv;
pu4_buf = (UWORD32 *)pu1_buf;
*pu4_buf = 0x10101010;
ps_cur_mb_info->u1_cbp = 0xff;
ps_dec->i1_prev_mb_qp_delta = 0;
/* Get neighbour MB's */
u1_num_mb_pair = (u1_mbNum >> u1_mbaff);
/*****************************************************************************/
/* calculate the RECON buffer YUV pointers for the PCM data */
/*****************************************************************************/
/* CHANGED CODE */
u1_mb_field_decoding_flag = ps_cur_mb_info->u1_mb_field_decodingflag;
pu1_y = ps_frame_buf->pu1_dest_y + (u1_num_mb_pair << 4);
pu1_u = ps_frame_buf->pu1_dest_u + (u1_num_mb_pair << 4);
pu1_v = pu1_u + 1;
u4_rec_width_y = ps_dec->u2_frm_wd_y << u1_mb_field_decoding_flag;
u4_rec_width_uv = ps_dec->u2_frm_wd_uv << u1_mb_field_decoding_flag;
/* CHANGED CODE */
if(u1_mbaff)
{
UWORD8 u1_top_mb;
u1_top_mb = ps_cur_mb_info->u1_topmb;
if(u1_top_mb == 0)
{
pu1_y += (u1_mb_field_decoding_flag ?
(u4_rec_width_y >> 1) : (u4_rec_width_y << 4));
pu1_u += (u1_mb_field_decoding_flag ?
(u4_rec_width_uv) : (u4_rec_width_uv << 4));
pu1_v = pu1_u + 1;
}
}
/* Read Luma samples */
for(u1_y = 0; u1_y < 16; u1_y++)
{
for(u1_x = 0; u1_x < 16; u1_x++)
pu1_y[u1_x] = ih264d_get_bits_h264(ps_bitstrm, 8);
pu1_y += u4_rec_width_y;
}
/* Read Chroma samples */
for(u1_y = 0; u1_y < 8; u1_y++)
{
for(u1_x = 0; u1_x < 8; u1_x++)
pu1_u[u1_x * YUV420SP_FACTOR] = ih264d_get_bits_h264(ps_bitstrm, 8);
pu1_u += u4_rec_width_uv;
}
for(u1_y = 0; u1_y < 8; u1_y++)
{
for(u1_x = 0; u1_x < 8; u1_x++)
pu1_v[u1_x * YUV420SP_FACTOR] = ih264d_get_bits_h264(ps_bitstrm, 8);
pu1_v += u4_rec_width_uv;
}
if(CABAC == ps_dec->ps_cur_pps->u1_entropy_coding_mode)
{
UWORD32 *pu4_buf;
UWORD8 *pu1_buf;
ctxt_inc_mb_info_t *p_curr_ctxt = ps_dec->ps_curr_ctxt_mb_info;
/* Re-initialize the cabac decoding engine. */
ret = ih264d_init_cabac_dec_envirnoment(&(ps_dec->s_cab_dec_env), ps_bitstrm);
if(ret != OK)
return ret;
/* update the cabac contetxs */
p_curr_ctxt->u1_mb_type = CAB_I_PCM;
p_curr_ctxt->u1_cbp = 47;
p_curr_ctxt->u1_intra_chroma_pred_mode = 0;
p_curr_ctxt->u1_transform8x8_ctxt = 0;
ps_cur_mb_info->ps_curmb->u1_tran_form8x8 = 0;
pu1_buf = ps_dec->pu1_left_nnz_y;
pu4_buf = (UWORD32 *)pu1_buf;
*pu4_buf = 0x01010101;
pu1_buf = ps_cur_mb_info->ps_curmb->pu1_nnz_y;
pu4_buf = (UWORD32 *)pu1_buf;
*pu4_buf = 0x01010101;
pu1_buf = ps_cur_mb_info->ps_curmb->pu1_nnz_uv;
pu4_buf = (UWORD32 *)pu1_buf;
*pu4_buf = 0x01010101;
pu1_buf = ps_dec->pu1_left_nnz_uv;
pu4_buf = (UWORD32 *)pu1_buf;
*pu4_buf = 0x01010101;
p_curr_ctxt->u1_yuv_dc_csbp = 0x7;
ps_dec->pu1_left_yuv_dc_csbp[0] = 0x7;
if(ps_dec->ps_cur_slice->u1_slice_type != I_SLICE)
{
MEMSET_16BYTES(&ps_dec->pu1_left_mv_ctxt_inc[0][0], 0);
memset(ps_dec->pi1_left_ref_idx_ctxt_inc, 0, 4);
MEMSET_16BYTES(p_curr_ctxt->u1_mv, 0);
memset(p_curr_ctxt->i1_ref_idx, 0, 4);
}
}
return OK;
}
/*!
**************************************************************************
* \if Function name : ih264d_decode_islice \endif
*
* \brief
* Decodes an I Slice
*
*
* \return
* 0 on Success and Error code otherwise
**************************************************************************
*/
WORD32 ih264d_parse_islice(dec_struct_t *ps_dec,
UWORD16 u2_first_mb_in_slice)
{
dec_pic_params_t * ps_pps = ps_dec->ps_cur_pps;
dec_slice_params_t * ps_slice = ps_dec->ps_cur_slice;
UWORD32 *pu4_bitstrm_buf = ps_dec->ps_bitstrm->pu4_buffer;
UWORD32 *pu4_bitstrm_ofst = &ps_dec->ps_bitstrm->u4_ofst;
UWORD32 u4_temp;
WORD32 i_temp;
WORD32 ret;
/*--------------------------------------------------------------------*/
/* Read remaining contents of the slice header */
/*--------------------------------------------------------------------*/
/* dec_ref_pic_marking function */
/* G050 */
if(ps_slice->u1_nal_ref_idc != 0)
{
if(!ps_dec->ps_dpb_cmds->u1_dpb_commands_read)
{
i_temp = ih264d_read_mmco_commands(ps_dec);
if (i_temp < 0)
{
return ERROR_DBP_MANAGER_T;
}
ps_dec->u4_bitoffset = i_temp;
}
else
ps_dec->ps_bitstrm->u4_ofst += ps_dec->u4_bitoffset;
}
/* G050 */
/* Read slice_qp_delta */
WORD64 i8_temp = (WORD64)ps_pps->u1_pic_init_qp
+ ih264d_sev(pu4_bitstrm_ofst, pu4_bitstrm_buf);
if((i8_temp < MIN_H264_QP) || (i8_temp > MAX_H264_QP))
return ERROR_INV_RANGE_QP_T;
ps_slice->u1_slice_qp = i8_temp;
COPYTHECONTEXT("SH: slice_qp_delta",
ps_slice->u1_slice_qp - ps_pps->u1_pic_init_qp);
if(ps_pps->u1_deblocking_filter_parameters_present_flag == 1)
{
u4_temp = ih264d_uev(pu4_bitstrm_ofst, pu4_bitstrm_buf);
COPYTHECONTEXT("SH: disable_deblocking_filter_idc", u4_temp);
if(u4_temp > SLICE_BOUNDARY_DBLK_DISABLED)
{
return ERROR_INV_SLICE_HDR_T;
}
ps_slice->u1_disable_dblk_filter_idc = u4_temp;
if(u4_temp != 1)
{
i_temp = ih264d_sev(pu4_bitstrm_ofst, pu4_bitstrm_buf)
<< 1;
if((MIN_DBLK_FIL_OFF > i_temp) || (i_temp > MAX_DBLK_FIL_OFF))
{
return ERROR_INV_SLICE_HDR_T;
}
ps_slice->i1_slice_alpha_c0_offset = i_temp;
COPYTHECONTEXT("SH: slice_alpha_c0_offset_div2",
ps_slice->i1_slice_alpha_c0_offset >> 1);
i_temp = ih264d_sev(pu4_bitstrm_ofst, pu4_bitstrm_buf)
<< 1;
if((MIN_DBLK_FIL_OFF > i_temp) || (i_temp > MAX_DBLK_FIL_OFF))
{
return ERROR_INV_SLICE_HDR_T;
}
ps_slice->i1_slice_beta_offset = i_temp;
COPYTHECONTEXT("SH: slice_beta_offset_div2",
ps_slice->i1_slice_beta_offset >> 1);
}
else
{
ps_slice->i1_slice_alpha_c0_offset = 0;
ps_slice->i1_slice_beta_offset = 0;
}
}
else
{
ps_slice->u1_disable_dblk_filter_idc = 0;
ps_slice->i1_slice_alpha_c0_offset = 0;
ps_slice->i1_slice_beta_offset = 0;
}
/* Initialization to check if number of motion vector per 2 Mbs */
/* are exceeding the range or not */
ps_dec->u2_mv_2mb[0] = 0;
ps_dec->u2_mv_2mb[1] = 0;
/*set slice header cone to 2 ,to indicate correct header*/
ps_dec->u1_slice_header_done = 2;
if(ps_pps->u1_entropy_coding_mode)
{
SWITCHOFFTRACE; SWITCHONTRACECABAC;
if(ps_dec->ps_cur_slice->u1_mbaff_frame_flag)
{
ps_dec->pf_get_mb_info = ih264d_get_mb_info_cabac_mbaff;
}
else
ps_dec->pf_get_mb_info = ih264d_get_mb_info_cabac_nonmbaff;
ret = ih264d_parse_islice_data_cabac(ps_dec, ps_slice,
u2_first_mb_in_slice);
if(ret != OK)
return ret;
SWITCHONTRACE; SWITCHOFFTRACECABAC;
}
else
{
if(ps_dec->ps_cur_slice->u1_mbaff_frame_flag)
{
ps_dec->pf_get_mb_info = ih264d_get_mb_info_cavlc_mbaff;
}
else
ps_dec->pf_get_mb_info = ih264d_get_mb_info_cavlc_nonmbaff;
ret = ih264d_parse_islice_data_cavlc(ps_dec, ps_slice,
u2_first_mb_in_slice);
if(ret != OK)
return ret;
}
return OK;
}
Reference in New Issue Copy Permalink