The-Homebrew-Cloud/Cemu
2
0
Fork 0
mirror of https://github.com/cemu-project/Cemu.git synced 2024-11-22 09:09:18 +01:00
Code Issues Packages Projects Releases Wiki Activity
454b587e36
Cemu/dependencies/ih264d/decoder/ih264d_mb_utils.c
Exzap d60742f52b Add all the files
2022-08-22 22:21:23 +02:00

1471 lines
64 KiB
C
Raw Blame History

/******************************************************************************
*
* 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_mb_utils.c
*
* \brief
* Contains utitlity functions needed for Macroblock decoding
*
* \date
* 18/12/2002
*
* \author AI
**************************************************************************
*/
#include <string.h>
#include <stdlib.h>
#include "ih264d_bitstrm.h"
#include "ih264d_defs.h"
#include "ih264d_debug.h"
#include "ih264d_structs.h"
#include "ih264d_defs.h"
#include "ih264d_mb_utils.h"
#include "ih264d_parse_slice.h"
#include "ih264d_error_handler.h"
#include "ih264d_parse_mb_header.h"
#include "ih264d_cabac.h"
#include "ih264d_defs.h"
#include "ih264d_tables.h"
/*****************************************************************************/
/* */
/* Function Name : get_mb_info_cavlc */
/* */
/* Description : This function sets the following information of cur MB */
/* (a) mb_x and mb_y */
/* (b) Neighbour availablity */
/* (c) Macroblock location in the frame buffer */
/* (e) For mbaff predicts field/frame u4_flag for topMb */
/* and sets the field/frame for botMb. This is */
/* written in ps_dec->u1_cur_mb_fld_dec_flag */
/* */
/* Inputs : pointer to decstruct */
/* pointer to current mb info */
/* currentMbaddress */
/* */
/* Processing : leftMb and TopMb params are used by DecMbskip and */
/* DecCtxMbfield modules so that these modules do not */
/* check for neigbour availability and then find the */
/* neigbours for context increments */
/* */
/* Returns : OK */
/* */
/* Issues : <List any issues or problems with this function> */
/* */
/* Revision History: */
/* */
/* DD MM YYYY Author(s) Changes (Describe the changes made) */
/* 13 07 2002 Jay Draft */
/* */
/*****************************************************************************/
UWORD32 ih264d_get_mb_info_cavlc_nonmbaff(dec_struct_t *ps_dec,
const UWORD16 u2_cur_mb_address,
dec_mb_info_t * ps_cur_mb_info,
UWORD32 u4_mbskip_run)
{
WORD32 mb_x;
WORD32 mb_y;
UWORD8 u1_mb_ngbr_avail = 0;
UWORD16 u2_frm_width_in_mb = ps_dec->u2_frm_wd_in_mbs;
WORD16 i2_prev_slice_mbx = ps_dec->i2_prev_slice_mbx;
UWORD16 u2_top_right_mask = TOP_RIGHT_DEFAULT_AVAILABLE;
UWORD16 u2_top_left_mask = TOP_LEFT_DEFAULT_AVAILABLE;
UNUSED(u4_mbskip_run);
/*--------------------------------------------------------------------*/
/* Calculate values of mb_x and mb_y */
/*--------------------------------------------------------------------*/
mb_x = (WORD16)ps_dec->u2_mbx;
mb_y = (WORD16)ps_dec->u2_mby;
ps_dec->u2_cur_mb_addr = u2_cur_mb_address;
mb_x++;
if(mb_x == u2_frm_width_in_mb)
{
mb_x = 0;
mb_y++;
}
if(mb_y > ps_dec->i2_prev_slice_mby)
{
/* if not in the immemdiate row of prev slice end then top
will be available */
if(mb_y > (ps_dec->i2_prev_slice_mby + 1))
i2_prev_slice_mbx = -1;
if(mb_x > i2_prev_slice_mbx)
{
u1_mb_ngbr_avail |= TOP_MB_AVAILABLE_MASK;
u2_top_right_mask |= TOP_RIGHT_TOP_AVAILABLE;
u2_top_left_mask |= TOP_LEFT_TOP_AVAILABLE;
}
if((mb_x > (i2_prev_slice_mbx - 1))
&& (mb_x != (u2_frm_width_in_mb - 1)))
{
u1_mb_ngbr_avail |= TOP_RIGHT_MB_AVAILABLE_MASK;
u2_top_right_mask |= TOP_RIGHT_TOPR_AVAILABLE;
}
if(mb_x > (i2_prev_slice_mbx + 1))
{
u1_mb_ngbr_avail |= TOP_LEFT_MB_AVAILABLE_MASK;
u2_top_left_mask |= TOP_LEFT_TOPL_AVAILABLE;
}
/* Next row Left will be available*/
i2_prev_slice_mbx = -1;
}
/* Same row */
if(mb_x > (i2_prev_slice_mbx + 1))
{
u1_mb_ngbr_avail |= LEFT_MB_AVAILABLE_MASK;
u2_top_left_mask |= TOP_LEFT_LEFT_AVAILABLE;
}
{
mb_neigbour_params_t *ps_cur_mb_row = ps_dec->ps_cur_mb_row;
mb_neigbour_params_t *ps_top_mb_row = ps_dec->ps_top_mb_row;
/* copy the parameters of topleft Mb */
ps_cur_mb_info->u1_topleft_mbtype = ps_dec->u1_topleft_mbtype;
/* Neighbour pointer assignments*/
ps_cur_mb_info->ps_curmb = ps_cur_mb_row + mb_x;
ps_cur_mb_info->ps_left_mb = ps_cur_mb_row + mb_x - 1;
ps_cur_mb_info->ps_top_mb = ps_top_mb_row + mb_x;
ps_cur_mb_info->ps_top_right_mb = ps_top_mb_row + mb_x + 1;
/* Update the parameters of topleftmb*/
ps_dec->u1_topleft_mbtype = ps_cur_mb_info->ps_top_mb->u1_mb_type;
}
ps_dec->u2_mby = mb_y;
ps_dec->u2_mbx = mb_x;
ps_cur_mb_info->u2_mbx = mb_x;
ps_cur_mb_info->u2_mby = mb_y;
ps_cur_mb_info->u1_topmb = 1;
ps_dec->i4_submb_ofst += SUB_BLK_SIZE;
ps_dec->u1_mb_ngbr_availablity = u1_mb_ngbr_avail;
ps_cur_mb_info->u1_mb_ngbr_availablity = u1_mb_ngbr_avail;
ps_cur_mb_info->ps_curmb->u1_mb_fld = ps_dec->u1_cur_mb_fld_dec_flag;
ps_cur_mb_info->u1_mb_field_decodingflag = ps_dec->u1_cur_mb_fld_dec_flag;
ps_cur_mb_info->u2_top_left_avail_mask = u2_top_left_mask;
ps_cur_mb_info->u2_top_right_avail_mask = u2_top_right_mask;
return (OK);
}
/*****************************************************************************/
/* */
/* Function Name : get_mb_info_cavlc */
/* */
/* Description : This function sets the following information of cur MB */
/* (a) mb_x and mb_y */
/* (b) Neighbour availablity */
/* (c) Macroblock location in the frame buffer */
/* (e) For mbaff predicts field/frame u4_flag for topMb */
/* and sets the field/frame for botMb. This is */
/* written in ps_dec->u1_cur_mb_fld_dec_flag */
/* */
/* Inputs : pointer to decstruct */
/* pointer to current mb info */
/* currentMbaddress */
/* */
/* Processing : leftMb and TopMb params are used by DecMbskip and */
/* DecCtxMbfield modules so that these modules do not */
/* check for neigbour availability and then find the */
/* neigbours for context increments */
/* */
/* Returns : OK */
/* */
/* Issues : <List any issues or problems with this function> */
/* */
/* Revision History: */
/* */
/* DD MM YYYY Author(s) Changes (Describe the changes made) */
/* 13 07 2002 Jay Draft */
/* */
/*****************************************************************************/
UWORD32 ih264d_get_mb_info_cavlc_mbaff(dec_struct_t *ps_dec,
const UWORD16 u2_cur_mb_address,
dec_mb_info_t * ps_cur_mb_info,
UWORD32 u4_mbskip_run)
{
UWORD16 u2_mb_x;
UWORD16 u2_mb_y;
UWORD8 u1_mb_ngbr_avail = 0;
UWORD16 u2_frm_width_in_mb = ps_dec->u2_frm_wd_in_mbs;
UWORD8 u1_top_mb = 1 - (u2_cur_mb_address & 0x01);
WORD16 i2_prev_slice_mbx = ps_dec->i2_prev_slice_mbx;
UWORD8 u1_cur_mb_field = 0;
UWORD16 u2_top_right_mask = TOP_RIGHT_DEFAULT_AVAILABLE;
UWORD16 u2_top_left_mask = TOP_LEFT_DEFAULT_AVAILABLE;
/*--------------------------------------------------------------------*/
/* Calculate values of mb_x and mb_y */
/*--------------------------------------------------------------------*/
u2_mb_x = ps_dec->u2_mbx;
u2_mb_y = ps_dec->u2_mby;
ps_dec->u2_cur_mb_addr = u2_cur_mb_address;
if(u1_top_mb)
{
u2_mb_x++;
if(u2_mb_x == u2_frm_width_in_mb)
{
u2_mb_x = 0;
u2_mb_y += 2;
}
if(u2_mb_y > ps_dec->i2_prev_slice_mby)
{
/* if not in the immemdiate row of prev slice end then top
will be available */
if(u2_mb_y > (ps_dec->i2_prev_slice_mby + 2))
i2_prev_slice_mbx = -1;
if(u2_mb_x > i2_prev_slice_mbx)
{
u1_mb_ngbr_avail |= TOP_MB_AVAILABLE_MASK;
u1_cur_mb_field = ps_dec->ps_top_mb_row[u2_mb_x << 1].u1_mb_fld;
u2_top_right_mask |= TOP_RIGHT_TOP_AVAILABLE;
u2_top_left_mask |= TOP_LEFT_TOP_AVAILABLE;
}
if((u2_mb_x > (i2_prev_slice_mbx - 1))
&& (u2_mb_x != (u2_frm_width_in_mb - 1)))
{
u1_mb_ngbr_avail |= TOP_RIGHT_MB_AVAILABLE_MASK;
u2_top_right_mask |= TOP_RIGHT_TOPR_AVAILABLE;
}
if(u2_mb_x > (i2_prev_slice_mbx + 1))
{
u1_mb_ngbr_avail |= TOP_LEFT_MB_AVAILABLE_MASK;
u2_top_left_mask |= TOP_LEFT_TOPL_AVAILABLE;
}
i2_prev_slice_mbx = -1;
}
/* Same row */
if(u2_mb_x > (i2_prev_slice_mbx + 1))
{
u1_mb_ngbr_avail |= LEFT_MB_AVAILABLE_MASK;
u1_cur_mb_field =
ps_dec->ps_cur_mb_row[(u2_mb_x << 1) - 1].u1_mb_fld;
u2_top_left_mask |= TOP_LEFT_LEFT_AVAILABLE;
}
/* Read u1_cur_mb_field from the bitstream if u4_mbskip_run <= 1*/
if(u4_mbskip_run <= 1)
u1_cur_mb_field = (UWORD8)ih264d_get_bit_h264(ps_dec->ps_bitstrm);
ps_dec->u1_cur_mb_fld_dec_flag = u1_cur_mb_field;
ps_dec->u2_top_left_mask = u2_top_left_mask;
ps_dec->u2_top_right_mask = u2_top_right_mask;
}
else
{
u1_mb_ngbr_avail = ps_dec->u1_mb_ngbr_availablity;
u1_cur_mb_field = ps_dec->u1_cur_mb_fld_dec_flag;
u2_top_left_mask = ps_dec->u2_top_left_mask;
u2_top_right_mask = ps_dec->u2_top_right_mask;
if(!u1_cur_mb_field)
{
/* Top is available */
u1_mb_ngbr_avail |= TOP_MB_AVAILABLE_MASK;
u2_top_right_mask |= TOP_RIGHT_TOP_AVAILABLE;
u2_top_left_mask |= TOP_LEFT_TOP_AVAILABLE;
/* Top Right not available */
u1_mb_ngbr_avail &= TOP_RT_SUBBLOCK_MASK_MOD;
u2_top_right_mask &= (~TOP_RIGHT_TOPR_AVAILABLE);
if(u1_mb_ngbr_avail & LEFT_MB_AVAILABLE_MASK)
{
u1_mb_ngbr_avail |= TOP_LEFT_MB_AVAILABLE_MASK;
u2_top_left_mask |= TOP_LEFT_LEFT_AVAILABLE;
u2_top_left_mask |= TOP_LEFT_TOPL_AVAILABLE;
}
}
}
ps_dec->u2_mby = u2_mb_y;
ps_dec->u2_mbx = u2_mb_x;
ps_cur_mb_info->u2_mbx = u2_mb_x;
ps_cur_mb_info->u2_mby = u2_mb_y;
ps_cur_mb_info->u1_topmb = u1_top_mb;
ps_dec->i4_submb_ofst += SUB_BLK_SIZE;
ps_dec->u1_mb_ngbr_availablity = u1_mb_ngbr_avail;
ps_cur_mb_info->u1_mb_ngbr_availablity = u1_mb_ngbr_avail;
ps_cur_mb_info->u1_mb_field_decodingflag = u1_cur_mb_field;
ps_cur_mb_info->u2_top_left_avail_mask = u2_top_left_mask;
ps_cur_mb_info->u2_top_right_avail_mask = u2_top_right_mask;
ih264d_get_mbaff_neighbours(ps_dec, ps_cur_mb_info, u1_cur_mb_field);
return (OK);
}
/*****************************************************************************/
/* */
/* Function Name : get_mb_info_cabac */
/* */
/* Description : This function sets the following information of cur MB */
/* (a) mb_x and mb_y */
/* (b) Neighbour availablity */
/* (c) Macroblock location in the frame buffer */
/* (e) leftMb parama and TopMb params of curMB */
/* (f) For Mbaff case leftMb params and TopMb params of */
/* bottomMb are also set if curMB is top */
/* (g) For mbaff predicts field/frame u4_flag for topMb */
/* and sets the field/frame for botMb. This is */
/* written in ps_dec->u1_cur_mb_fld_dec_flag */
/* */
/* Inputs : pointer to decstruct */
/* pointer to current mb info */
/* currentMbaddress */
/* */
/* Processing : leftMb and TopMb params are used by DecMbskip and */
/* DecCtxMbfield modules so that these modules do not */
/* check for neigbour availability and then find the */
/* neigbours for context increments */
/* */
/* Returns : OK */
/* */
/* Issues : <List any issues or problems with this function> */
/* */
/* Revision History: */
/* */
/* DD MM YYYY Author(s) Changes (Describe the changes made) */
/* 13 07 2002 Jay Draft */
/* */
/*****************************************************************************/
UWORD32 ih264d_get_mb_info_cabac_nonmbaff(dec_struct_t *ps_dec,
const UWORD16 u2_cur_mb_address,
dec_mb_info_t * ps_cur_mb_info,
UWORD32 u4_mbskip)
{
WORD32 mb_x;
WORD32 mb_y;
UWORD32 u1_mb_ngbr_avail = 0;
UWORD32 u2_frm_width_in_mb = ps_dec->u2_frm_wd_in_mbs;
UWORD32 u1_top_mb = 1;
WORD32 i2_prev_slice_mbx = ps_dec->i2_prev_slice_mbx;
UWORD32 u2_top_right_mask = TOP_RIGHT_DEFAULT_AVAILABLE;
UWORD32 u2_top_left_mask = TOP_LEFT_DEFAULT_AVAILABLE;
ctxt_inc_mb_info_t * const p_ctx_inc_mb_map = ps_dec->p_ctxt_inc_mb_map;
/*--------------------------------------------------------------------*/
/* Calculate values of mb_x and mb_y */
/*--------------------------------------------------------------------*/
mb_x = (WORD16)ps_dec->u2_mbx;
mb_y = (WORD16)ps_dec->u2_mby;
ps_dec->u2_cur_mb_addr = u2_cur_mb_address;
mb_x++;
if((UWORD32)mb_x == u2_frm_width_in_mb)
{
mb_x = 0;
mb_y++;
}
/*********************************************************************/
/* Cabac Context Initialisations */
/*********************************************************************/
ps_dec->ps_curr_ctxt_mb_info = p_ctx_inc_mb_map + mb_x;
ps_dec->p_left_ctxt_mb_info = p_ctx_inc_mb_map - 1;
ps_dec->p_top_ctxt_mb_info = p_ctx_inc_mb_map - 1;
/********************************************************************/
/* neighbour availablility */
/********************************************************************/
if(mb_y > ps_dec->i2_prev_slice_mby)
{
/* if not in the immemdiate row of prev slice end then top
will be available */
if(mb_y > (ps_dec->i2_prev_slice_mby + 1))
i2_prev_slice_mbx = -1;
if(mb_x > i2_prev_slice_mbx)
{
u1_mb_ngbr_avail |= TOP_MB_AVAILABLE_MASK;
u2_top_right_mask |= TOP_RIGHT_TOP_AVAILABLE;
u2_top_left_mask |= TOP_LEFT_TOP_AVAILABLE;
ps_dec->p_top_ctxt_mb_info = ps_dec->ps_curr_ctxt_mb_info;
}
if((mb_x > (i2_prev_slice_mbx - 1))
&& ((UWORD32)mb_x != (u2_frm_width_in_mb - 1)))
{
u1_mb_ngbr_avail |= TOP_RIGHT_MB_AVAILABLE_MASK;
u2_top_right_mask |= TOP_RIGHT_TOPR_AVAILABLE;
}
if(mb_x > (i2_prev_slice_mbx + 1))
{
u1_mb_ngbr_avail |= TOP_LEFT_MB_AVAILABLE_MASK;
u2_top_left_mask |= TOP_LEFT_TOPL_AVAILABLE;
}
/* Next row */
i2_prev_slice_mbx = -1;
}
/* Same row */
if(mb_x > (i2_prev_slice_mbx + 1))
{
u1_mb_ngbr_avail |= LEFT_MB_AVAILABLE_MASK;
u2_top_left_mask |= TOP_LEFT_LEFT_AVAILABLE;
ps_dec->p_left_ctxt_mb_info = ps_dec->ps_curr_ctxt_mb_info - 1;
}
{
mb_neigbour_params_t *ps_cur_mb_row = ps_dec->ps_cur_mb_row;
mb_neigbour_params_t *ps_top_mb_row = ps_dec->ps_top_mb_row;
/* copy the parameters of topleft Mb */
ps_cur_mb_info->u1_topleft_mbtype = ps_dec->u1_topleft_mbtype;
/* Neighbour pointer assignments*/
ps_cur_mb_info->ps_curmb = ps_cur_mb_row + mb_x;
ps_cur_mb_info->ps_left_mb = ps_cur_mb_row + mb_x - 1;
ps_cur_mb_info->ps_top_mb = ps_top_mb_row + mb_x;
ps_cur_mb_info->ps_top_right_mb = ps_top_mb_row + mb_x + 1;
/* Update the parameters of topleftmb*/
ps_dec->u1_topleft_mbtype = ps_cur_mb_info->ps_top_mb->u1_mb_type;
}
ps_dec->u2_mby = mb_y;
ps_dec->u2_mbx = mb_x;
ps_cur_mb_info->u2_mbx = mb_x;
ps_cur_mb_info->u2_mby = mb_y;
ps_cur_mb_info->u1_topmb = u1_top_mb;
ps_dec->i4_submb_ofst += SUB_BLK_SIZE;
ps_dec->u1_mb_ngbr_availablity = u1_mb_ngbr_avail;
ps_cur_mb_info->u1_mb_ngbr_availablity = u1_mb_ngbr_avail;
ps_cur_mb_info->ps_curmb->u1_mb_fld = ps_dec->u1_cur_mb_fld_dec_flag;
ps_cur_mb_info->u1_mb_field_decodingflag = ps_dec->u1_cur_mb_fld_dec_flag;
ps_cur_mb_info->u2_top_left_avail_mask = u2_top_left_mask;
ps_cur_mb_info->u2_top_right_avail_mask = u2_top_right_mask;
/*********************************************************************/
/* Assign the neigbours */
/*********************************************************************/
if(u4_mbskip)
{
UWORD32 u4_ctx_inc =
2
- ((!!(ps_dec->p_top_ctxt_mb_info->u1_mb_type
& CAB_SKIP_MASK))
+ (!!(ps_dec->p_left_ctxt_mb_info->u1_mb_type
& CAB_SKIP_MASK)));
u4_mbskip = ih264d_decode_bin(u4_ctx_inc, ps_dec->p_mb_skip_flag_t,
ps_dec->ps_bitstrm, &ps_dec->s_cab_dec_env);
if(!u4_mbskip)
{
if(!(u1_mb_ngbr_avail & LEFT_MB_AVAILABLE_MASK))
{
UWORD32 *pu4_buf;
UWORD8 *pu1_buf;
pu1_buf = ps_dec->pu1_left_nnz_y;
pu4_buf = (UWORD32 *)pu1_buf;
*pu4_buf = 0;
pu1_buf = ps_dec->pu1_left_nnz_uv;
pu4_buf = (UWORD32 *)pu1_buf;
*pu4_buf = 0;
*(ps_dec->pu1_left_yuv_dc_csbp) = 0;
MEMSET_16BYTES(&ps_dec->pu1_left_mv_ctxt_inc[0][0], 0);
*(UWORD32 *)ps_dec->pi1_left_ref_idx_ctxt_inc = 0;
}
if(!(u1_mb_ngbr_avail & TOP_MB_AVAILABLE_MASK))
{
MEMSET_16BYTES(ps_dec->ps_curr_ctxt_mb_info->u1_mv, 0);
memset(ps_dec->ps_curr_ctxt_mb_info->i1_ref_idx, 0, 4);
}
}
}
return (u4_mbskip);
}
/*****************************************************************************/
/* */
/* Function Name : get_mb_info_cabac */
/* */
/* Description : This function sets the following information of cur MB */
/* (a) mb_x and mb_y */
/* (b) Neighbour availablity */
/* (c) Macroblock location in the frame buffer */
/* (e) leftMb parama and TopMb params of curMB */
/* (f) For Mbaff case leftMb params and TopMb params of */
/* bottomMb are also set if curMB is top */
/* (g) For mbaff predicts field/frame u4_flag for topMb */
/* and sets the field/frame for botMb. This is */
/* written in ps_dec->u1_cur_mb_fld_dec_flag */
/* */
/* Inputs : pointer to decstruct */
/* pointer to current mb info */
/* currentMbaddress */
/* */
/* Processing : leftMb and TopMb params are used by DecMbskip and */
/* DecCtxMbfield modules so that these modules do not */
/* check for neigbour availability and then find the */
/* neigbours for context increments */
/* */
/* Returns : OK */
/* */
/* Issues : <List any issues or problems with this function> */
/* */
/* Revision History: */
/* */
/* DD MM YYYY Author(s) Changes (Describe the changes made) */
/* 13 07 2002 Jay Draft */
/* */
/*****************************************************************************/
UWORD32 ih264d_get_mb_info_cabac_mbaff(dec_struct_t *ps_dec,
const UWORD16 u2_cur_mb_address,
dec_mb_info_t * ps_cur_mb_info,
UWORD32 u4_mbskip)
{
WORD32 mb_x;
WORD32 mb_y;
UWORD8 u1_mb_ngbr_avail = 0;
UWORD16 u2_frm_width_in_mb = ps_dec->u2_frm_wd_in_mbs;
ctxt_inc_mb_info_t * const p_ctx_inc_mb_map = ps_dec->p_ctxt_inc_mb_map;
ctxt_inc_mb_info_t *ps_curr_ctxt, *ps_top_ctxt, *ps_left_ctxt;
mb_neigbour_params_t *ps_cur_mb_row = ps_dec->ps_cur_mb_row;
mb_neigbour_params_t *ps_top_mb_row = ps_dec->ps_top_mb_row;
UWORD32 u4_left_mb_pair_fld = 0;
UWORD32 u4_top_mb_pair_fld = 0;
UWORD8 u1_cur_mb_field = 0;
UWORD8 u1_top_mb = 1 - (u2_cur_mb_address & 0x01);
WORD16 i2_prev_slice_mbx = ps_dec->i2_prev_slice_mbx;
UWORD16 u2_top_right_mask = TOP_RIGHT_DEFAULT_AVAILABLE;
UWORD16 u2_top_left_mask = TOP_LEFT_DEFAULT_AVAILABLE;
/*--------------------------------------------------------------------*/
/* Calculate values of mb_x and mb_y */
/*--------------------------------------------------------------------*/
mb_x = (WORD16)ps_dec->u2_mbx;
mb_y = (WORD16)ps_dec->u2_mby;
ps_dec->u2_cur_mb_addr = u2_cur_mb_address;
ps_top_ctxt = ps_left_ctxt = p_ctx_inc_mb_map - 1;
if(u1_top_mb)
{
ctxt_inc_mb_info_t *ps_left_mb_of_bot = ps_left_ctxt;
ctxt_inc_mb_info_t *ps_top_mb_of_bot = ps_top_ctxt;
mb_x++;
if(mb_x == u2_frm_width_in_mb)
{
mb_x = 0;
mb_y += 2;
}
ps_curr_ctxt = p_ctx_inc_mb_map + (mb_x << 1);
if(mb_y > ps_dec->i2_prev_slice_mby)
{
UWORD8 u1_cur_mb_fld_flag_known = 0;
/* Next row */
if(mb_x > 0)
{
/***********************************************************************/
/* Left Mb is avialable */
/***********************************************************************/
u1_mb_ngbr_avail |= LEFT_MB_AVAILABLE_MASK;
ps_left_ctxt = ps_curr_ctxt - 2;
ps_left_mb_of_bot = ps_curr_ctxt - 1;
u1_cur_mb_field = u4_left_mb_pair_fld = ps_cur_mb_row[(mb_x
<< 1) - 1].u1_mb_fld;
u1_cur_mb_fld_flag_known = 1;
u2_top_left_mask |= TOP_LEFT_LEFT_AVAILABLE;
}
/* if not in the immemdiate row of prev slice end then top
will be available */
if(mb_y > (ps_dec->i2_prev_slice_mby + 2))
i2_prev_slice_mbx = -1;
if(mb_x > i2_prev_slice_mbx)
{
/*********************************************************************/
/* Top Mb is avialable */
/*********************************************************************/
u1_mb_ngbr_avail |= TOP_MB_AVAILABLE_MASK;
u2_top_right_mask |= TOP_RIGHT_TOP_AVAILABLE;
u2_top_left_mask |= TOP_LEFT_TOP_AVAILABLE;
/* point to MbAddrB + 1 */
ps_top_ctxt = ps_curr_ctxt + 1;
u4_top_mb_pair_fld = ps_top_mb_row[(mb_x << 1)].u1_mb_fld;
u1_cur_mb_field =
u1_cur_mb_fld_flag_known ?
u1_cur_mb_field :
u4_top_mb_pair_fld;
ps_top_mb_of_bot = u1_cur_mb_field ? ps_top_ctxt : ps_curr_ctxt;
/* MbAddrB */
ps_top_ctxt -= (u1_cur_mb_field && u4_top_mb_pair_fld);
}
if((mb_x > (i2_prev_slice_mbx - 1))
&& (mb_x != (u2_frm_width_in_mb - 1)))
{
u1_mb_ngbr_avail |= TOP_RIGHT_MB_AVAILABLE_MASK;
u2_top_right_mask |= TOP_RIGHT_TOPR_AVAILABLE;
}
if(mb_x > (i2_prev_slice_mbx + 1))
{
u1_mb_ngbr_avail |= TOP_LEFT_MB_AVAILABLE_MASK;
u2_top_left_mask |= TOP_LEFT_TOPL_AVAILABLE;
}
}
else
{
/* Same row */
if(mb_x > (i2_prev_slice_mbx + 1))
{
/***************************************************************/
/* Left Mb is avialable */
/***************************************************************/
u1_mb_ngbr_avail |= LEFT_MB_AVAILABLE_MASK;
u1_cur_mb_field = u4_left_mb_pair_fld = ps_cur_mb_row[(mb_x
<< 1) - 1].u1_mb_fld;
ps_left_ctxt = ps_curr_ctxt - 2;
ps_left_mb_of_bot = ps_curr_ctxt - 1;
u2_top_left_mask |= TOP_LEFT_LEFT_AVAILABLE;
}
}
/*********************************************************/
/* Check whether the call is from I slice or Inter slice */
/*********************************************************/
if(u4_mbskip)
{
UWORD32 u4_ctx_inc = 2
- ((!!(ps_top_ctxt->u1_mb_type & CAB_SKIP_MASK))
+ (!!(ps_left_ctxt->u1_mb_type
& CAB_SKIP_MASK)));
dec_bit_stream_t * const ps_bitstrm = ps_dec->ps_bitstrm;
decoding_envirnoment_t *ps_cab_dec_env = &ps_dec->s_cab_dec_env;
bin_ctxt_model_t *p_mb_skip_flag_t = ps_dec->p_mb_skip_flag_t;
ps_dec->u4_next_mb_skip = 0;
u4_mbskip = ih264d_decode_bin(u4_ctx_inc, p_mb_skip_flag_t,
ps_bitstrm, ps_cab_dec_env);
if(u4_mbskip)
{
UWORD32 u4_next_mbskip;
ps_curr_ctxt->u1_mb_type = CAB_SKIP;
u4_ctx_inc =
2
- ((!!(ps_top_mb_of_bot->u1_mb_type
& CAB_SKIP_MASK))
+ (!!(ps_left_mb_of_bot->u1_mb_type
& CAB_SKIP_MASK)));
/* Decode the skip u4_flag of bottom Mb */
u4_next_mbskip = ih264d_decode_bin(u4_ctx_inc, p_mb_skip_flag_t,
ps_bitstrm,
ps_cab_dec_env);
ps_dec->u4_next_mb_skip = u4_next_mbskip;
if(!u4_next_mbskip)
{
u4_ctx_inc = u4_top_mb_pair_fld + u4_left_mb_pair_fld;
u1_cur_mb_field = ih264d_decode_bin(
u4_ctx_inc, ps_dec->p_mb_field_dec_flag_t,
ps_bitstrm, ps_cab_dec_env);
}
}
}
if(!u4_mbskip)
{
UWORD32 u4_ctx_inc = u4_top_mb_pair_fld + u4_left_mb_pair_fld;
u1_cur_mb_field = ih264d_decode_bin(u4_ctx_inc,
ps_dec->p_mb_field_dec_flag_t,
ps_dec->ps_bitstrm,
&ps_dec->s_cab_dec_env);
}
ps_dec->u1_cur_mb_fld_dec_flag = u1_cur_mb_field;
ps_dec->u2_top_left_mask = u2_top_left_mask;
ps_dec->u2_top_right_mask = u2_top_right_mask;
ps_dec->u2_mby = mb_y;
ps_dec->u2_mbx = mb_x;
}
else
{
u1_cur_mb_field = ps_dec->u1_cur_mb_fld_dec_flag;
u1_mb_ngbr_avail = ps_dec->u1_mb_ngbr_availablity;
u2_top_left_mask = ps_dec->u2_top_left_mask;
u2_top_right_mask = ps_dec->u2_top_right_mask;
ps_curr_ctxt = p_ctx_inc_mb_map + (mb_x << 1) + 1;
if(u1_mb_ngbr_avail & LEFT_MB_AVAILABLE_MASK)
{
u4_left_mb_pair_fld = ps_cur_mb_row[(mb_x << 1) - 1].u1_mb_fld;
/* point to A if top else A+1 */
ps_left_ctxt = ps_curr_ctxt - 2
- (u4_left_mb_pair_fld != u1_cur_mb_field);
}
if(u1_cur_mb_field)
{
if(u1_mb_ngbr_avail & TOP_MB_AVAILABLE_MASK)
{
/* point to MbAddrB + 1 */
ps_top_ctxt = ps_curr_ctxt;
}
}
else
{
/* Top is available */
u1_mb_ngbr_avail |= TOP_MB_AVAILABLE_MASK;
u2_top_right_mask |= TOP_RIGHT_TOP_AVAILABLE;
u2_top_left_mask |= TOP_LEFT_TOP_AVAILABLE;
/* Top Right not available */
u1_mb_ngbr_avail &= TOP_RT_SUBBLOCK_MASK_MOD;
u2_top_right_mask &= (~TOP_RIGHT_TOPR_AVAILABLE);
if(u1_mb_ngbr_avail & LEFT_MB_AVAILABLE_MASK)
{
u1_mb_ngbr_avail |= TOP_LEFT_MB_AVAILABLE_MASK;
u2_top_left_mask |= TOP_LEFT_LEFT_AVAILABLE;
u2_top_left_mask |= TOP_LEFT_TOPL_AVAILABLE;
}
/* CurMbAddr - 1 */
ps_top_ctxt = ps_curr_ctxt - 1;
}
if(u4_mbskip)
{
if(ps_curr_ctxt[-1].u1_mb_type & CAB_SKIP_MASK)
{
/* If previous mb is skipped, return value of next mb skip */
u4_mbskip = ps_dec->u4_next_mb_skip;
}
else
{
/* If previous mb is not skipped then call DecMbSkip */
UWORD32 u4_ctx_inc =
2
- ((!!(ps_top_ctxt->u1_mb_type
& CAB_SKIP_MASK))
+ (!!(ps_left_ctxt->u1_mb_type
& CAB_SKIP_MASK)));
u4_mbskip = ih264d_decode_bin(u4_ctx_inc,
ps_dec->p_mb_skip_flag_t,
ps_dec->ps_bitstrm,
&ps_dec->s_cab_dec_env);
}
}
}
ps_cur_mb_info->u2_mbx = mb_x;
ps_cur_mb_info->u2_mby = mb_y;
ps_cur_mb_info->u1_topmb = u1_top_mb;
ps_dec->i4_submb_ofst += SUB_BLK_SIZE;
ps_dec->u1_mb_ngbr_availablity = u1_mb_ngbr_avail;
ps_cur_mb_info->u1_mb_ngbr_availablity = u1_mb_ngbr_avail;
ps_cur_mb_info->u1_mb_field_decodingflag = u1_cur_mb_field;
ps_cur_mb_info->u2_top_left_avail_mask = u2_top_left_mask;
ps_cur_mb_info->u2_top_right_avail_mask = u2_top_right_mask;
ih264d_get_mbaff_neighbours(ps_dec, ps_cur_mb_info, u1_cur_mb_field);
{
ih264d_get_cabac_context_mbaff(ps_dec, ps_cur_mb_info, u4_mbskip);
}
{
bin_ctxt_model_t *p_cabac_ctxt_table_t = ps_dec->p_cabac_ctxt_table_t;
if(u1_cur_mb_field)
{
p_cabac_ctxt_table_t += SIGNIFICANT_COEFF_FLAG_FLD;
}
else
{
p_cabac_ctxt_table_t += SIGNIFICANT_COEFF_FLAG_FRAME;
}
{
bin_ctxt_model_t * * p_significant_coeff_flag_t =
ps_dec->p_significant_coeff_flag_t;
p_significant_coeff_flag_t[0] = p_cabac_ctxt_table_t
+ SIG_COEFF_CTXT_CAT_0_OFFSET;
p_significant_coeff_flag_t[1] = p_cabac_ctxt_table_t
+ SIG_COEFF_CTXT_CAT_1_OFFSET;
p_significant_coeff_flag_t[2] = p_cabac_ctxt_table_t
+ SIG_COEFF_CTXT_CAT_2_OFFSET;
p_significant_coeff_flag_t[3] = p_cabac_ctxt_table_t
+ SIG_COEFF_CTXT_CAT_3_OFFSET;
p_significant_coeff_flag_t[4] = p_cabac_ctxt_table_t
+ SIG_COEFF_CTXT_CAT_4_OFFSET;
p_significant_coeff_flag_t[5] = p_cabac_ctxt_table_t
+ SIG_COEFF_CTXT_CAT_5_OFFSET;
}
}
return (u4_mbskip);
}
/*****************************************************************************/
/* */
/* Function Name : ih264d_get_cabac_context_mbaff */
/* */
/* Description : Gets the current macroblock Cabac Context and sets the */
/* top and left cabac context ptrs in CtxIncMbMap */
/* 1. For Coss field left neigbours it alters coded block */
/* u4_flag , motion vectors, reference indices, cbp of */
/* the left neigbours which increases the code i4_size */
/* 2. For Coss field top neigbours it alters motion */
/* vectors reference indices of the top neigbours */
/* which further increases the code i4_size */
/* */
/* Inputs : 1. dec_struct_t */
/* 2. CurMbAddr used for Mbaff (only to see if curMB */
/* is top or bottom) */
/* 3. uc_curMbFldDecFlag only for Mbaff */
/* */
/* Returns : 0 */
/* */
/* Issues : code i4_size can be reduced if ui_CodedBlockFlag storage */
/* structure in context is changed. This change however */
/* would break the parseResidual4x4Cabac asm routine. */
/* */
/* Revision History: */
/* */
/* DD MM YYYY Author(s) Changes (Describe the changes made) */
/* 18 06 2005 Jay */
/* */
/*****************************************************************************/
UWORD32 ih264d_get_cabac_context_mbaff(dec_struct_t * ps_dec,
dec_mb_info_t *ps_cur_mb_info,
UWORD32 u4_mbskip)
{
const UWORD8 u1_mb_ngbr_availablity = ps_dec->u1_mb_ngbr_availablity;
ctxt_inc_mb_info_t * const p_ctx_inc_mb_map = ps_dec->p_ctxt_inc_mb_map;
UWORD8 (*pu1_left_mv_ctxt_inc_2d)[4] = &ps_dec->pu1_left_mv_ctxt_inc[0];
WORD8 (*pi1_left_ref_idx_ctxt_inc) = ps_dec->pi1_left_ref_idx_ctxt_inc;
const UWORD8 u1_cur_mb_fld_flag = ps_cur_mb_info->u1_mb_field_decodingflag;
const UWORD8 u1_topmb = ps_cur_mb_info->u1_topmb;
const UWORD8 uc_botMb = 1 - ps_cur_mb_info->u1_topmb;
ctxt_inc_mb_info_t * ps_leftMB;
ps_dec->ps_curr_ctxt_mb_info = p_ctx_inc_mb_map + (ps_dec->u2_mbx << 1);
ps_dec->p_top_ctxt_mb_info = ps_dec->ps_curr_ctxt_mb_info;
if(u1_topmb)
{
pu1_left_mv_ctxt_inc_2d = ps_dec->u1_left_mv_ctxt_inc_arr[0];
pi1_left_ref_idx_ctxt_inc = &ps_dec->i1_left_ref_idx_ctx_inc_arr[0][0];
ps_dec->pu1_left_yuv_dc_csbp = &ps_dec->u1_yuv_dc_csbp_topmb;
}
else
{
/* uc_botMb */
pu1_left_mv_ctxt_inc_2d = ps_dec->u1_left_mv_ctxt_inc_arr[1];
pi1_left_ref_idx_ctxt_inc = &ps_dec->i1_left_ref_idx_ctx_inc_arr[1][0];
ps_dec->pu1_left_yuv_dc_csbp = &ps_dec->u1_yuv_dc_csbp_bot_mb;
ps_dec->ps_curr_ctxt_mb_info += 1;
}
ps_dec->pu1_left_mv_ctxt_inc = pu1_left_mv_ctxt_inc_2d;
ps_dec->pi1_left_ref_idx_ctxt_inc = pi1_left_ref_idx_ctxt_inc;
if(u1_mb_ngbr_availablity & LEFT_MB_AVAILABLE_MASK)
{
const UWORD8 u1_left_mb_fld_flag = ps_cur_mb_info->ps_left_mb->u1_mb_fld;
ps_leftMB = ps_dec->ps_curr_ctxt_mb_info - 2;
if(u1_left_mb_fld_flag != u1_cur_mb_fld_flag)
{
ctxt_inc_mb_info_t *ps_tempLeft;
UWORD8 u1_cbp_t, u1_cbp_b;
UWORD8 u1_cr_cpb;
ps_leftMB -= uc_botMb;
ps_tempLeft = ps_dec->ps_left_mb_ctxt_info;
ps_tempLeft->u1_mb_type = ps_leftMB->u1_mb_type;
ps_tempLeft->u1_intra_chroma_pred_mode =
ps_leftMB->u1_intra_chroma_pred_mode;
ps_tempLeft->u1_transform8x8_ctxt = ps_leftMB->u1_transform8x8_ctxt;
u1_cr_cpb = ps_leftMB->u1_cbp;
/*****************************************************************/
/* reform RefIdx, CBP, MV and CBF ctxInc taking care of A and A+1*/
/*****************************************************************/
if(u1_cur_mb_fld_flag)
{
/* current MB is a FLD and left a FRM */
UWORD8 (* const pu1_left_mv_ctxt_inc_2d_arr_top)[4] =
ps_dec->u1_left_mv_ctxt_inc_arr[0];
UWORD8 (* const pu1_left_mv_ctxt_inc_2d_arr_bot)[4] =
ps_dec->u1_left_mv_ctxt_inc_arr[1];
WORD8 (* const i1_left_ref_idx_ctxt_inc_arr_top) =
&ps_dec->i1_left_ref_idx_ctx_inc_arr[0][0];
WORD8 (* const i1_left_ref_idx_ctxt_inc_arr_bot) =
&ps_dec->i1_left_ref_idx_ctx_inc_arr[1][0];
u1_cbp_t = ps_leftMB->u1_cbp;
u1_cbp_b = (ps_leftMB + 1)->u1_cbp;
ps_tempLeft->u1_cbp = (u1_cbp_t & 0x02)
| ((u1_cbp_b & 0x02) << 2);
// set motionvectors as
// 0T = 0T 0B = 0T
// 1T = 2T 1B = 2T
// 2T = 0B 2B = 0B
// 3T = 2B 3B = 2B
if(u1_topmb)
{
/********************************************/
/* Bottoms DC CBF = Top DC CBF */
/********************************************/
ps_dec->u1_yuv_dc_csbp_bot_mb =
ps_dec->u1_yuv_dc_csbp_topmb;
*(UWORD32 *)pu1_left_mv_ctxt_inc_2d[3] =
*(UWORD32 *)pu1_left_mv_ctxt_inc_2d_arr_bot[2];
*(UWORD32 *)pu1_left_mv_ctxt_inc_2d[1] =
*(UWORD32 *)pu1_left_mv_ctxt_inc_2d_arr_top[2];
*(UWORD32 *)pu1_left_mv_ctxt_inc_2d[2] =
*(UWORD32 *)pu1_left_mv_ctxt_inc_2d_arr_bot[0];
*(UWORD32 *)pu1_left_mv_ctxt_inc_2d[0] =
*(UWORD32 *)pu1_left_mv_ctxt_inc_2d_arr_top[0];
i1_left_ref_idx_ctxt_inc_arr_top[1] =
i1_left_ref_idx_ctxt_inc_arr_bot[0];
i1_left_ref_idx_ctxt_inc_arr_top[3] =
i1_left_ref_idx_ctxt_inc_arr_bot[2];
*(UWORD32 *)(i1_left_ref_idx_ctxt_inc_arr_bot) =
*(UWORD32 *)(i1_left_ref_idx_ctxt_inc_arr_top);
memcpy(pu1_left_mv_ctxt_inc_2d_arr_bot,
pu1_left_mv_ctxt_inc_2d_arr_top, 16);
}
{
UWORD8 i;
for(i = 0; i < 4; i++)
{
pu1_left_mv_ctxt_inc_2d[i][1] >>= 1;
pu1_left_mv_ctxt_inc_2d[i][3] >>= 1;
}
}
}
else
{
/* current MB is a FRM and left FLD */
if(u1_topmb)
{
u1_cbp_t = ps_leftMB->u1_cbp;
u1_cbp_t = (u1_cbp_t & 0x02);
ps_tempLeft->u1_cbp = (u1_cbp_t | (u1_cbp_t << 2));
/********************************************/
/* Bottoms DC CBF = Top DC CBF */
/********************************************/
ps_dec->u1_yuv_dc_csbp_bot_mb =
ps_dec->u1_yuv_dc_csbp_topmb;
// set motionvectors as
// 3B = 2B = 3T
// 1B = 0B = 2T
// 3T = 2T = 1T
// 1T = 0T = 0T
*(UWORD32 *)pu1_left_mv_ctxt_inc_2d[7] =
*(UWORD32 *)pu1_left_mv_ctxt_inc_2d[3];
*(UWORD32 *)pu1_left_mv_ctxt_inc_2d[6] =
*(UWORD32 *)pu1_left_mv_ctxt_inc_2d[3];
*(UWORD32 *)pu1_left_mv_ctxt_inc_2d[5] =
*(UWORD32 *)pu1_left_mv_ctxt_inc_2d[2];
*(UWORD32 *)pu1_left_mv_ctxt_inc_2d[4] =
*(UWORD32 *)pu1_left_mv_ctxt_inc_2d[2];
*(UWORD32 *)pu1_left_mv_ctxt_inc_2d[3] =
*(UWORD32 *)pu1_left_mv_ctxt_inc_2d[1];
*(UWORD32 *)pu1_left_mv_ctxt_inc_2d[2] =
*(UWORD32 *)pu1_left_mv_ctxt_inc_2d[1];
*(UWORD32 *)pu1_left_mv_ctxt_inc_2d[1] =
*(UWORD32 *)pu1_left_mv_ctxt_inc_2d[0];
pi1_left_ref_idx_ctxt_inc[7] = (pi1_left_ref_idx_ctxt_inc[3]
- 1);
pi1_left_ref_idx_ctxt_inc[6] = (pi1_left_ref_idx_ctxt_inc[3]
- 1);
pi1_left_ref_idx_ctxt_inc[5] = (pi1_left_ref_idx_ctxt_inc[1]
- 1);
pi1_left_ref_idx_ctxt_inc[4] = (pi1_left_ref_idx_ctxt_inc[1]
- 1);
pi1_left_ref_idx_ctxt_inc[3] = (pi1_left_ref_idx_ctxt_inc[2]
- 1);
pi1_left_ref_idx_ctxt_inc[2] = (pi1_left_ref_idx_ctxt_inc[2]
- 1);
pi1_left_ref_idx_ctxt_inc[1] = (pi1_left_ref_idx_ctxt_inc[0]
- 1);
pi1_left_ref_idx_ctxt_inc[0] = (pi1_left_ref_idx_ctxt_inc[0]
- 1);
}
else
{
u1_cbp_t = ps_leftMB->u1_cbp;
u1_cbp_t = (u1_cbp_t & 0x08);
ps_tempLeft->u1_cbp = (u1_cbp_t | (u1_cbp_t >> 2));
}
{
UWORD8 i;
for(i = 0; i < 4; i++)
{
pu1_left_mv_ctxt_inc_2d[i][1] <<= 1;
pu1_left_mv_ctxt_inc_2d[i][3] <<= 1;
}
}
}
ps_tempLeft->u1_cbp = ps_tempLeft->u1_cbp + ((u1_cr_cpb >> 4) << 4);
ps_leftMB = ps_tempLeft;
}
ps_dec->p_left_ctxt_mb_info = ps_leftMB;
}
else
{
ps_dec->p_left_ctxt_mb_info = p_ctx_inc_mb_map - 1;
if(!u4_mbskip)
{
*(ps_dec->pu1_left_yuv_dc_csbp) = 0;
MEMSET_16BYTES(&pu1_left_mv_ctxt_inc_2d[0][0], 0);
*(UWORD32 *)pi1_left_ref_idx_ctxt_inc = 0;
}
}
/*************************************************************************/
/* Now get the top context mb info */
/*************************************************************************/
{
UWORD8 (*u1_top_mv_ctxt_inc_arr_2d)[4] =
ps_dec->ps_curr_ctxt_mb_info->u1_mv;
WORD8 (*pi1_top_ref_idx_ctxt_inc) =
ps_dec->ps_curr_ctxt_mb_info->i1_ref_idx;
UWORD8 uc_topMbFldDecFlag = ps_cur_mb_info->ps_top_mb->u1_mb_fld;
if(u1_mb_ngbr_availablity & TOP_MB_AVAILABLE_MASK)
{
if(ps_cur_mb_info->i1_offset)
ps_dec->p_top_ctxt_mb_info += 1;
if(!u4_mbskip)
{
memcpy(u1_top_mv_ctxt_inc_arr_2d,
&ps_dec->p_top_ctxt_mb_info->u1_mv, 16);
memcpy(pi1_top_ref_idx_ctxt_inc,
&ps_dec->p_top_ctxt_mb_info->i1_ref_idx, 4);
if(uc_topMbFldDecFlag ^ u1_cur_mb_fld_flag)
{
UWORD8 i;
if(u1_cur_mb_fld_flag)
{
for(i = 0; i < 4; i++)
{
u1_top_mv_ctxt_inc_arr_2d[i][1] >>= 1;
u1_top_mv_ctxt_inc_arr_2d[i][3] >>= 1;
}
}
else
{
for(i = 0; i < 4; i++)
{
u1_top_mv_ctxt_inc_arr_2d[i][1] <<= 1;
u1_top_mv_ctxt_inc_arr_2d[i][3] <<= 1;
pi1_top_ref_idx_ctxt_inc[i] -= 1;
}
}
}
}
}
else
{
ps_dec->p_top_ctxt_mb_info = p_ctx_inc_mb_map - 1;
if(!u4_mbskip)
{
MEMSET_16BYTES(&u1_top_mv_ctxt_inc_arr_2d[0][0], 0);
memset(pi1_top_ref_idx_ctxt_inc, 0, 4);
}
}
}
return OK;
}
/*****************************************************************************/
/* */
/* Function Name : ih264d_update_mbaff_left_nnz */
/* */
/* Description : This function updates the left luma and chroma nnz for */
/* mbaff cases. */
/* */
/* Inputs : <What inputs does the function take?> */
/* Globals : <Does it use any global variables?> */
/* Processing : <Describe how the function operates - include algorithm */
/* description> */
/* Outputs : <What does the function produce?> */
/* Returns : <What does the function return?> */
/* */
/* Issues : <List any issues or problems with this function> */
/* */
/* Revision History: */
/* */
/* DD MM YYYY Author(s) Changes (Describe the changes made) */
/* 13 07 2002 Ittiam Draft */
/* */
/*****************************************************************************/
void ih264d_update_mbaff_left_nnz(dec_struct_t * ps_dec,
dec_mb_info_t * ps_cur_mb_info)
{
UWORD32 *pu4_buf;
UWORD8 *pu1_buf;
if(ps_cur_mb_info->u1_topmb)
{
pu1_buf = ps_dec->pu1_left_nnz_y;
pu4_buf = (UWORD32 *)pu1_buf;
ps_dec->u4_n_left_temp_y = *pu4_buf;
pu1_buf = ps_dec->pu1_left_nnz_uv;
pu4_buf = (UWORD32 *)pu1_buf;
ps_dec->u4_n_left_temp_uv = *pu4_buf;
}
else
{
ps_dec->u4_n_leftY[0] = ps_dec->u4_n_left_temp_y;
pu1_buf = ps_dec->pu1_left_nnz_y;
pu4_buf = (UWORD32 *)pu1_buf;
ps_dec->u4_n_leftY[1] = *pu4_buf;
ps_dec->u4_n_left_cr[0] = ps_dec->u4_n_left_temp_uv;
pu1_buf = ps_dec->pu1_left_nnz_uv;
pu4_buf = (UWORD32 *)pu1_buf;
ps_dec->u4_n_left_cr[1] = *pu4_buf;
}
}
/*!
**************************************************************************
* \if Function name : ih264d_get_mbaff_neighbours \endif
*
* \brief
* Gets the neighbors for the current MB if it is of type MB-AFF
* frame.
*
* \return
* None
*
**************************************************************************
*/
void ih264d_get_mbaff_neighbours(dec_struct_t * ps_dec,
dec_mb_info_t * ps_cur_mb_info,
UWORD8 uc_curMbFldDecFlag)
{
mb_neigbour_params_t *ps_left_mb;
mb_neigbour_params_t *ps_top_mb;
mb_neigbour_params_t *ps_top_right_mb = NULL;
mb_neigbour_params_t *ps_curmb;
const UWORD8 u1_topmb = ps_cur_mb_info->u1_topmb;
const UWORD8 uc_botMb = 1 - u1_topmb;
const UWORD32 u4_mb_x = ps_cur_mb_info->u2_mbx;
/* Current MbParams location in top row buffer */
ps_curmb = ps_dec->ps_cur_mb_row + (u4_mb_x << 1) + uc_botMb;
ps_left_mb = ps_curmb - 2;
/* point to A if top else A+1 */
if(uc_botMb && (ps_left_mb->u1_mb_fld != uc_curMbFldDecFlag))
{
/* move from A + 1 to A */
ps_left_mb--;
}
ps_cur_mb_info->i1_offset = 0;
if((uc_curMbFldDecFlag == 0) && uc_botMb)
{
mb_neigbour_params_t *ps_topleft_mb;
/* CurMbAddr - 1 */
ps_top_mb = ps_curmb - 1;
/* Mark Top right Not available */
/* point to A */
ps_topleft_mb = ps_curmb - 3;
if(ps_topleft_mb->u1_mb_fld)
{
/* point to A + 1 */
ps_topleft_mb++;
}
ps_cur_mb_info->u1_topleft_mb_fld = ps_topleft_mb->u1_mb_fld;
ps_cur_mb_info->u1_topleft_mbtype = ps_topleft_mb->u1_mb_type;
}
else
{
/* Top = B + 1 */
ps_top_mb = ps_dec->ps_top_mb_row + (u4_mb_x << 1) + 1;
ps_top_right_mb = ps_top_mb + 2;
ps_cur_mb_info->i1_offset = 4;
/* TopRight = C + 1 */
/* TopLeft = D+1 */
ps_cur_mb_info->u1_topleft_mb_fld = ps_dec->u1_topleft_mb_fld_bot;
ps_cur_mb_info->u1_topleft_mbtype = ps_dec->u1_topleft_mbtype_bot;
if(uc_curMbFldDecFlag && u1_topmb)
{
if(ps_top_mb->u1_mb_fld)
{
/* MbAddrB */
ps_top_mb--;
ps_cur_mb_info->i1_offset = 0;
}
/* If topright is field then point to C */
ps_top_right_mb -= ps_top_right_mb->u1_mb_fld ? 1 : 0;
if(ps_cur_mb_info->u1_topleft_mb_fld)
{
/* TopLeft = D */
ps_cur_mb_info->u1_topleft_mb_fld = ps_dec->u1_topleft_mb_fld;
ps_cur_mb_info->u1_topleft_mbtype = ps_dec->u1_topleft_mbtype;
}
}
}
if(u1_topmb)
{
/* Update the parameters of topleftmb*/
ps_dec->u1_topleft_mb_fld = ps_top_mb->u1_mb_fld;
ps_dec->u1_topleft_mbtype = ps_top_mb->u1_mb_type;
/* Set invscan and dequantMatrixScan*/
if(uc_curMbFldDecFlag)
{
ps_dec->pu1_inv_scan = (UWORD8 *)gau1_ih264d_inv_scan_fld;
}
else
{
ps_dec->pu1_inv_scan = (UWORD8 *)gau1_ih264d_inv_scan;
}
ps_dec->pu2_quant_scale_y =
gau2_ih264_iquant_scale_4x4[ps_dec->u1_qp_y_rem6];
ps_dec->pu2_quant_scale_u =
gau2_ih264_iquant_scale_4x4[ps_dec->u1_qp_u_rem6];
ps_dec->pu2_quant_scale_v =
gau2_ih264_iquant_scale_4x4[ps_dec->u1_qp_v_rem6];
}
else
{
/* Update the parameters of topleftmb*/
mb_neigbour_params_t *ps_top_mb_temp = ps_dec->ps_top_mb_row
+ (u4_mb_x << 1) + 1;
ps_dec->u1_topleft_mb_fld_bot = ps_top_mb_temp->u1_mb_fld;
ps_dec->u1_topleft_mbtype_bot = ps_top_mb_temp->u1_mb_type;
}
ps_cur_mb_info->ps_left_mb = ps_left_mb;
ps_cur_mb_info->ps_top_mb = ps_top_mb;
ps_cur_mb_info->ps_top_right_mb = ps_top_right_mb;
ps_cur_mb_info->ps_curmb = ps_curmb;
ps_curmb->u1_mb_fld = uc_curMbFldDecFlag;
{
/* Form Left NNZ */
UWORD8 u1_is_left_mb_fld = ps_left_mb->u1_mb_fld;
UWORD8 *pu1_left_mb_pair_nnz_y = (UWORD8 *)&ps_dec->u4_n_leftY[0];
UWORD8 *pu1_left_mb_pair_nnz_uv = (UWORD8 *)&ps_dec->u4_n_left_cr[0];
UWORD8 *pu1_left_nnz_y = ps_dec->pu1_left_nnz_y;
UWORD8 *pu1_left_nnz_uv = ps_dec->pu1_left_nnz_uv;
if(uc_curMbFldDecFlag == u1_is_left_mb_fld)
{
*(UWORD32 *)pu1_left_nnz_y = *(UWORD32 *)(pu1_left_mb_pair_nnz_y
+ (uc_botMb << 2));
*(UWORD32 *)pu1_left_nnz_uv = *(UWORD32 *)(pu1_left_mb_pair_nnz_uv
+ (uc_botMb << 2));
}
else if((uc_curMbFldDecFlag == 0) && u1_topmb && u1_is_left_mb_fld)
{
/* 0 0 1 1 of u4_n_leftY[0], 0 0 2 2 of u4_n_left_cr[0] */
pu1_left_nnz_y[0] = pu1_left_nnz_y[1] = pu1_left_mb_pair_nnz_y[0];
pu1_left_nnz_y[2] = pu1_left_nnz_y[3] = pu1_left_mb_pair_nnz_y[1];
pu1_left_nnz_uv[0] = pu1_left_nnz_uv[1] =
pu1_left_mb_pair_nnz_uv[0];
pu1_left_nnz_uv[2] = pu1_left_nnz_uv[3] =
pu1_left_mb_pair_nnz_uv[2];
}
else if((uc_curMbFldDecFlag == 0) && uc_botMb && u1_is_left_mb_fld)
{
/* 2 2 3 3 of u4_n_leftY[0] , 1 1 3 3 of u4_n_left_cr[0] */
pu1_left_nnz_y[0] = pu1_left_nnz_y[1] = pu1_left_mb_pair_nnz_y[2];
pu1_left_nnz_y[2] = pu1_left_nnz_y[3] = pu1_left_mb_pair_nnz_y[3];
pu1_left_nnz_uv[0] = pu1_left_nnz_uv[1] =
pu1_left_mb_pair_nnz_uv[1];
pu1_left_nnz_uv[2] = pu1_left_nnz_uv[3] =
pu1_left_mb_pair_nnz_uv[3];
}
else
{
/* 0 2 0 2 of u4_n_leftY[0], u4_n_leftY[1] */
pu1_left_nnz_y[0] = pu1_left_mb_pair_nnz_y[0];
pu1_left_nnz_y[1] = pu1_left_mb_pair_nnz_y[2];
pu1_left_nnz_y[2] = pu1_left_mb_pair_nnz_y[4 + 0];
pu1_left_nnz_y[3] = pu1_left_mb_pair_nnz_y[4 + 2];
/* 0 of u4_n_left_cr[0] and 0 u4_n_left_cr[1]
2 of u4_n_left_cr[0] and 2 u4_n_left_cr[1] */
pu1_left_nnz_uv[0] = pu1_left_mb_pair_nnz_uv[0];
pu1_left_nnz_uv[1] = pu1_left_mb_pair_nnz_uv[4 + 0];
pu1_left_nnz_uv[2] = pu1_left_mb_pair_nnz_uv[2];
pu1_left_nnz_uv[3] = pu1_left_mb_pair_nnz_uv[4 + 2];
}
}
}
/*
**************************************************************************
* \if Function name : ih264d_transfer_mb_group_data \endif
*
* \brief
* 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)
*
* \return
* None
*
**************************************************************************
*/
void ih264d_transfer_mb_group_data(dec_struct_t * ps_dec,
const UWORD8 u1_num_mbs,
const UWORD8 u1_end_of_row, /* Cur n-Mb End of Row Flag */
const UWORD8 u1_end_of_row_next /* Next n-Mb End of Row Flag */
)
{
dec_mb_info_t *ps_cur_mb_info = ps_dec->ps_nmb_info;
tfr_ctxt_t *ps_trns_addr = &ps_dec->s_tran_addrecon;
UWORD16 u2_mb_y;
UWORD32 y_offset;
UWORD32 u4_frame_stride;
mb_neigbour_params_t *ps_temp;
const UWORD8 u1_mbaff = ps_dec->ps_cur_slice->u1_mbaff_frame_flag;
UNUSED(u1_end_of_row_next);
ps_trns_addr->pu1_dest_y += ps_trns_addr->u4_inc_y[u1_end_of_row];
ps_trns_addr->pu1_dest_u += ps_trns_addr->u4_inc_uv[u1_end_of_row];
ps_trns_addr->pu1_dest_v += ps_trns_addr->u4_inc_uv[u1_end_of_row];
/* Swap top and current pointers */
if(u1_end_of_row)
{
if(ps_dec->u1_separate_parse)
{
u2_mb_y = ps_dec->i2_dec_thread_mb_y;
}
else
{
ps_temp = ps_dec->ps_cur_mb_row;
ps_dec->ps_cur_mb_row = ps_dec->ps_top_mb_row;
ps_dec->ps_top_mb_row = ps_temp;
u2_mb_y = ps_dec->u2_mby + (1 + u1_mbaff);
}
u4_frame_stride = ps_dec->u2_frm_wd_y
<< ps_dec->ps_cur_slice->u1_field_pic_flag;
y_offset = (u2_mb_y * u4_frame_stride) << 4;
ps_trns_addr->pu1_dest_y = ps_dec->s_cur_pic.pu1_buf1 + y_offset;
u4_frame_stride = ps_dec->u2_frm_wd_uv
<< ps_dec->ps_cur_slice->u1_field_pic_flag;
y_offset = (u2_mb_y * u4_frame_stride) << 3;
ps_trns_addr->pu1_dest_u = ps_dec->s_cur_pic.pu1_buf2 + y_offset;
ps_trns_addr->pu1_dest_v = ps_dec->s_cur_pic.pu1_buf3 + y_offset;
ps_trns_addr->pu1_mb_y = ps_trns_addr->pu1_dest_y;
ps_trns_addr->pu1_mb_u = ps_trns_addr->pu1_dest_u;
ps_trns_addr->pu1_mb_v = ps_trns_addr->pu1_dest_v;
}
/*
* The Slice boundary is also a valid condition to transfer. So recalculate
* the Left increment, in case the number of MBs is lesser than the
* N MB value. u1_num_mbs will be equal to N of N MB if the entire N Mb is
* decoded.
*/
ps_dec->s_tran_addrecon.u2_mv_left_inc = ((u1_num_mbs >> u1_mbaff) - 1)
<< (4 + u1_mbaff);
ps_dec->s_tran_addrecon.u2_mv_top_left_inc = (u1_num_mbs << 2) - 1
- (u1_mbaff << 2);
if(ps_dec->u1_separate_parse == 0)
{
/* reassign left MV and cur MV pointers */
ps_dec->ps_mv_left = ps_dec->ps_mv_cur
+ ps_dec->s_tran_addrecon.u2_mv_left_inc;
ps_dec->ps_mv_cur += (u1_num_mbs << 4);
}
/* Increment deblock parameters pointer in external memory */
if(ps_dec->u1_separate_parse == 0)
{
ps_dec->ps_deblk_mbn += u1_num_mbs;
}
}
Reference in New Issue View Git Blame Copy Permalink