/******************************************************************************
 *
 * 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;
    }

}