/******************************************************************************
 *
 * 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
*  impeg2d_mc.c
*
* @brief
*  Contains MC function definitions for MPEG2 decoder
*
* @author
*  Harish
*
* @par List of Functions:
* - impeg2d_motion_comp()
* - impeg2d_motion_comp_recon_buf()
* - impeg2d_mc_1mv()
* - impeg2d_mc_fw_or_bk_mb()
* - impeg2d_mc_frm_dual_prime()
* - impeg2d_mc_fld_dual_prime()
* - impeg2d_mc_4mv()
* - impeg2d_mc_2mv()
* - impeg2d_dec_intra_mb()
* - impeg2d_dec_skip_p_mb()
* - impeg2d_dec_skip_b_mb()
* - impeg2d_dec_skip_mbs()
* - impeg2d_dec_0mv_coded_mb()
* - impeg2d_mc_halfx_halfy()
* - impeg2d_mc_halfx_fully()
* - impeg2d_mc_fullx_halfy()
* - impeg2d_mc_fullx_fully()
* - impeg2d_set_mc_params()
*
* @remarks
*  None
*
*******************************************************************************
*/
#include <string.h>

#include "iv_datatypedef.h"
#include "iv.h"

#include "impeg2_buf_mgr.h"
#include "impeg2_disp_mgr.h"
#include "impeg2_defs.h"
#include "impeg2_platform_macros.h"
#include "impeg2_inter_pred.h"
#include "impeg2_idct.h"
#include "impeg2_globals.h"
#include "impeg2_mem_func.h"
#include "impeg2_format_conv.h"
#include "impeg2_macros.h"

#include "ivd.h"
#include "impeg2d.h"
#include "impeg2d_bitstream.h"
#include "impeg2d_structs.h"
#include "impeg2d_globals.h"
#include "impeg2d_pic_proc.h"
#include "impeg2d_debug.h"
#include "impeg2d_mv_dec.h"
#include "impeg2d_mc.h"

/*****************************************************************************/
/*                                                                           */
/*  Function Name : impeg2d_motion_comp                                      */
/*                                                                           */
/*  Description   : Perform motion compensation and store the resulting block*/
/*                  in the buf                                               */
/*                                                                           */
/*  Inputs        : params - Parameters required to do motion compensation   */
/*                                                                           */
/*  Globals       :                                                          */
/*                                                                           */
/*  Processing    : Calls appropriate functions depending on the mode of     */
/*                  compensation                                             */
/*                                                                           */
/*  Outputs       : buf       - Buffer for the motion compensation result    */
/*                                                                           */
/*  Returns       : None                                                     */
/*                                                                           */
/*  Issues        : None                                                     */
/*                                                                           */
/*  Revision History:                                                        */
/*                                                                           */
/*         DD MM YYYY   Author(s)       Changes                              */
/*         14 09 2005   Hairsh M        First Version                        */
/*                                                                           */
/*****************************************************************************/
void impeg2d_motion_comp(dec_state_t *ps_dec, mb_mc_params_t *ps_params,yuv_buf_t *ps_buf)
{

    PROFILE_DISABLE_MC_RETURN;

    /* Perform motion compensation for Y */
    ps_dec->pf_mc[ps_params->s_luma.u4_mode]((void *)ps_dec, ps_params->s_ref.pu1_y + ps_params->s_luma.u4_src_offset,
                ps_params->s_luma.u4_src_wd,
                ps_buf->pu1_y + ps_params->s_luma.u4_dst_offset_res_buf,
                ps_params->s_luma.u4_dst_wd_res_buf,
                ps_params->s_luma.u4_cols,
                ps_params->s_luma.u4_rows);
    /* Perform motion compensation for U */
    ps_dec->pf_mc[ps_params->s_chroma.u4_mode]((void *)ps_dec, ps_params->s_ref.pu1_u + ps_params->s_chroma.u4_src_offset,
                ps_params->s_chroma.u4_src_wd,
                ps_buf->pu1_u + ps_params->s_chroma.u4_dst_offset_res_buf,
                ps_params->s_chroma.u4_dst_wd_res_buf,
                ps_params->s_chroma.u4_cols,
                ps_params->s_chroma.u4_rows);

    /* Perform motion compensation for V */
    ps_dec->pf_mc[ps_params->s_chroma.u4_mode]((void *)ps_dec, ps_params->s_ref.pu1_v + ps_params->s_chroma.u4_src_offset,
                ps_params->s_chroma.u4_src_wd,
                ps_buf->pu1_v + ps_params->s_chroma.u4_dst_offset_res_buf,
                ps_params->s_chroma.u4_dst_wd_res_buf,
                ps_params->s_chroma.u4_cols,
                ps_params->s_chroma.u4_rows);
}


/*****************************************************************************/
/*                                                                           */
/*  Function Name : impeg2d_motion_comp_recon_buf                          */
/*                                                                           */
/*  Description   : Perform motion compensation and store the resulting block*/
/*                  in the buf                                               */
/*                                                                           */
/*  Inputs        : params - Parameters required to do motion compensation   */
/*                                                                           */
/*  Globals       :                                                          */
/*                                                                           */
/*  Processing    : Calls appropriate functions depending on the mode of     */
/*                  compensation                                             */
/*                                                                           */
/*  Outputs       : buf       - Buffer for the motion compensation result    */
/*                                                                           */
/*  Returns       : None                                                     */
/*                                                                           */
/*  Issues        : None                                                     */
/*                                                                           */
/*  Revision History:                                                        */
/*                                                                           */
/*         DD MM YYYY   Author(s)       Changes                              */
/*         14 09 2005   Harish M        First Version                        */
/*                                                                           */
/*****************************************************************************/
void impeg2d_motion_comp_recon_buf(dec_state_t *ps_dec,
                                     mb_mc_params_t *ps_params,
                                     yuv_buf_t *ps_dest_buf)
{

    PROFILE_DISABLE_MC_RETURN;

    /* Perform motion compensation for Y */
    ps_dec->pf_mc[ps_params->s_luma.u4_mode](ps_dec, ps_params->s_ref.pu1_y + ps_params->s_luma.u4_src_offset,
                                        ps_params->s_luma.u4_src_wd,
                                        ps_dest_buf->pu1_y + ps_params->s_luma.u4_dst_offset_cur_frm,
                                        ps_params->s_luma.u4_dst_wd_cur_frm,
                                        ps_params->s_luma.u4_cols,
                                        ps_params->s_luma.u4_rows);

    /* Perform motion compensation for U */

    ps_dec->pf_mc[ps_params->s_chroma.u4_mode](ps_dec, ps_params->s_ref.pu1_u + ps_params->s_chroma.u4_src_offset,
                                        ps_params->s_chroma.u4_src_wd,
                                        ps_dest_buf->pu1_u + ps_params->s_chroma.u4_dst_offset_cur_frm,
                                        ps_params->s_chroma.u4_dst_wd_cur_frm,
                                        ps_params->s_chroma.u4_cols,
                                        ps_params->s_chroma.u4_rows);

    /* Perform motion compensation for V */
    ps_dec->pf_mc[ps_params->s_chroma.u4_mode](ps_dec, ps_params->s_ref.pu1_v + ps_params->s_chroma.u4_src_offset,
                                        ps_params->s_chroma.u4_src_wd,
                                        ps_dest_buf->pu1_v + ps_params->s_chroma.u4_dst_offset_cur_frm,
                                        ps_params->s_chroma.u4_dst_wd_cur_frm,
                                        ps_params->s_chroma.u4_cols,
                                        ps_params->s_chroma.u4_rows);
}


/*****************************************************************************/
/*                                                                           */
/*  Function Name : impeg2d_mc_1mv                                           */
/*                                                                           */
/*  Description   : Perform motion compensation and store the resulting block*/
/*                  in the buf                                               */
/*                                                                           */
/*  Inputs        : params - Parameters required to do motion compensation   */
/*                                                                           */
/*  Globals       :                                                          */
/*                                                                           */
/*  Processing    : Calls appropriate functions depending on the mode of     */
/*                  compensation                                             */
/*                                                                           */
/*  Outputs       : buf       - Buffer for the motion compensation result    */
/*                                                                           */
/*  Returns       : None                                                     */
/*                                                                           */
/*  Issues        : None                                                     */
/*                                                                           */
/*  Revision History:                                                        */
/*                                                                           */
/*         DD MM YYYY   Author(s)       Changes                              */
/*         14 09 2005   Hairsh M        First Version                        */
/*                                                                           */
/*****************************************************************************/
void impeg2d_mc_1mv(dec_state_t *ps_dec)
{

    impeg2d_motion_comp_recon_buf(ps_dec, &ps_dec->as_mb_mc_params[ps_dec->e_mb_pred][FIRST], &ps_dec->s_dest_buf);
}


/*****************************************************************************/
/*                                                                           */
/*  Function Name : impeg2d_mc_fw_or_bk_mb                                   */
/*                                                                           */
/*  Description   : Perform motion compensation and store the resulting block*/
/*                  in the buf                                               */
/*                                                                           */
/*  Inputs        : params - Parameters required to do motion compensation   */
/*                                                                           */
/*  Globals       :                                                          */
/*                                                                           */
/*  Processing    : Calls appropriate functions depending on the mode of     */
/*                  compensation                                             */
/*                                                                           */
/*  Outputs       : buf       - Buffer for the motion compensation result    */
/*                                                                           */
/*  Returns       : None                                                     */
/*                                                                           */
/*  Issues        : None                                                     */
/*                                                                           */
/*  Revision History:                                                        */
/*                                                                           */
/*         DD MM YYYY   Author(s)       Changes                              */
/*         14 09 2005   Hairsh M        First Version                        */
/*                                                                           */
/*****************************************************************************/
void impeg2d_mc_fw_or_bk_mb(dec_state_t *ps_dec)
{
    impeg2d_motion_comp_recon_buf(ps_dec, &ps_dec->as_mb_mc_params[FORW][FIRST], &ps_dec->s_dest_buf);
    impeg2d_motion_comp_recon_buf(ps_dec, &ps_dec->as_mb_mc_params[FORW][SECOND], &ps_dec->s_dest_buf);
}


/*****************************************************************************/
/*                                                                           */
/*  Function Name : impeg2d_mc_frm_dual_prime                                */
/*                                                                           */
/*  Description   : Perform motion compensation and store the resulting block*/
/*                  in the buf                                               */
/*                                                                           */
/*  Inputs        : params - Parameters required to do motion compensation   */
/*                                                                           */
/*  Globals       :                                                          */
/*                                                                           */
/*  Processing    : Calls appropriate functions depending on the mode of     */
/*                  compensation                                             */
/*                                                                           */
/*  Outputs       : buf       - Buffer for the motion compensation result    */
/*                                                                           */
/*  Returns       : None                                                     */
/*                                                                           */
/*  Issues        : None                                                     */
/*                                                                           */
/*  Revision History:                                                        */
/*                                                                           */
/*         DD MM YYYY   Author(s)       Changes                              */
/*         14 09 2005   Hairsh M        First Version                        */
/*                                                                           */
/*****************************************************************************/
void impeg2d_mc_frm_dual_prime(dec_state_t *ps_dec)
{
    /************************************************************************/
    /* Perform Motion Compensation                                          */
    /************************************************************************/
    impeg2d_motion_comp(ps_dec, &ps_dec->as_mb_mc_params[FORW][FIRST], &ps_dec->s_mc_fw_buf);
    impeg2d_motion_comp(ps_dec, &ps_dec->as_mb_mc_params[BACK][FIRST], &ps_dec->s_mc_bk_buf);

    impeg2d_motion_comp(ps_dec, &ps_dec->as_mb_mc_params[FORW][SECOND], &ps_dec->s_mc_fw_buf);
    impeg2d_motion_comp(ps_dec, &ps_dec->as_mb_mc_params[BACK][SECOND], &ps_dec->s_mc_bk_buf);


    ps_dec->pf_interpolate(&ps_dec->s_mc_fw_buf,&ps_dec->s_mc_bk_buf,&ps_dec->s_dest_buf,ps_dec->u2_picture_width);
}


/*****************************************************************************/
/*                                                                           */
/*  Function Name : impeg2d_mc_fld_dual_prime                                */
/*                                                                           */
/*  Description   : Perform motion compensation and store the resulting block*/
/*                  in the buf                                               */
/*                                                                           */
/*  Inputs        : params - Parameters required to do motion compensation   */
/*                                                                           */
/*  Globals       :                                                          */
/*                                                                           */
/*  Processing    : Calls appropriate functions depending on the mode of     */
/*                  compensation                                             */
/*                                                                           */
/*  Outputs       : buf       - Buffer for the motion compensation result    */
/*                                                                           */
/*  Returns       : None                                                     */
/*                                                                           */
/*  Issues        : None                                                     */
/*                                                                           */
/*  Revision History:                                                        */
/*                                                                           */
/*         DD MM YYYY   Author(s)       Changes                              */
/*         14 09 2005   Hairsh M        First Version                        */
/*                                                                           */
/*****************************************************************************/
void impeg2d_mc_fld_dual_prime(dec_state_t *ps_dec)
{
    /************************************************************************/
    /* Perform Motion Compensation                                          */
    /************************************************************************/
    impeg2d_motion_comp(ps_dec, &ps_dec->as_mb_mc_params[FORW][FIRST], &ps_dec->s_mc_fw_buf);
    impeg2d_motion_comp(ps_dec, &ps_dec->as_mb_mc_params[FORW][SECOND], &ps_dec->s_mc_bk_buf);


    ps_dec->pf_interpolate(&ps_dec->s_mc_fw_buf,&ps_dec->s_mc_bk_buf,&ps_dec->s_dest_buf,ps_dec->u2_picture_width);
}


/*****************************************************************************/
/*                                                                           */
/*  Function Name : impeg2d_mc_4mv                                      */
/*                                                                           */
/*  Description   : Perform motion compensation and store the resulting block*/
/*                  in the buf                                               */
/*                                                                           */
/*  Inputs        : params - Parameters required to do motion compensation   */
/*                                                                           */
/*  Globals       :                                                          */
/*                                                                           */
/*  Processing    : Calls appropriate functions depending on the mode of     */
/*                  compensation                                             */
/*                                                                           */
/*  Outputs       : buf       - Buffer for the motion compensation result    */
/*                                                                           */
/*  Returns       : None                                                     */
/*                                                                           */
/*  Issues        : None                                                     */
/*                                                                           */
/*  Revision History:                                                        */
/*                                                                           */
/*         DD MM YYYY   Author(s)       Changes                              */
/*         14 09 2005   Hairsh M        First Version                        */
/*                                                                           */
/*****************************************************************************/
void impeg2d_mc_4mv(dec_state_t *ps_dec)
{
    /************************************************************************/
    /* Perform Motion Compensation                                          */
    /************************************************************************/
    impeg2d_motion_comp(ps_dec, &ps_dec->as_mb_mc_params[FORW][FIRST], &ps_dec->s_mc_fw_buf);
    impeg2d_motion_comp(ps_dec, &ps_dec->as_mb_mc_params[BACK][FIRST], &ps_dec->s_mc_bk_buf);
    impeg2d_motion_comp(ps_dec, &ps_dec->as_mb_mc_params[FORW][SECOND], &ps_dec->s_mc_fw_buf);
    impeg2d_motion_comp(ps_dec, &ps_dec->as_mb_mc_params[BACK][SECOND], &ps_dec->s_mc_bk_buf);

    ps_dec->pf_interpolate(&ps_dec->s_mc_fw_buf,&ps_dec->s_mc_bk_buf,&ps_dec->s_dest_buf,ps_dec->u2_picture_width);
}

/*****************************************************************************/
/*                                                                           */
/*  Function Name : impeg2d_mc_2mv                                         */
/*                                                                           */
/*  Description   : Perform motion compensation and store the resulting block*/
/*                  in the buf                                               */
/*                                                                           */
/*  Inputs        : params - Parameters required to do motion compensation   */
/*                                                                           */
/*  Globals       :                                                          */
/*                                                                           */
/*  Processing    : Calls appropriate functions depending on the mode of     */
/*                  compensation                                             */
/*                                                                           */
/*  Outputs       : buf       - Buffer for the motion compensation result    */
/*                                                                           */
/*  Returns       : None                                                     */
/*                                                                           */
/*  Issues        : None                                                     */
/*                                                                           */
/*  Revision History:                                                        */
/*                                                                           */
/*         DD MM YYYY   Author(s)       Changes                              */
/*         14 09 2005   Hairsh M        First Version                        */
/*                                                                           */
/*****************************************************************************/
void impeg2d_mc_2mv(dec_state_t *ps_dec)
{
   /************************************************************************/
    /* Perform Motion Compensation                                          */
    /************************************************************************/
    impeg2d_motion_comp(ps_dec, &ps_dec->as_mb_mc_params[FORW][FIRST], &ps_dec->s_mc_fw_buf);
    impeg2d_motion_comp(ps_dec, &ps_dec->as_mb_mc_params[BACK][FIRST], &ps_dec->s_mc_bk_buf);

    ps_dec->pf_interpolate(&ps_dec->s_mc_fw_buf,&ps_dec->s_mc_bk_buf,&ps_dec->s_dest_buf,ps_dec->u2_picture_width);
}

/*****************************************************************************
*  Function Name   : impeg2d_dec_intra_mb
*
*  Description     : Performs decoding of Intra MB
*
*  Arguments       :
*  dec             : Decoder state
*
*  Values Returned : None
*****************************************************************************/
void impeg2d_dec_intra_mb(dec_state_t *ps_dec)
{

    ps_dec->u2_cbp = 0x3F;
    if(ps_dec->u2_concealment_motion_vectors)
    {

        stream_t *ps_stream;

        ps_stream = &ps_dec->s_bit_stream;
        /* Decode the concealment motion vector */
        impeg2d_dec_mv(ps_stream,ps_dec->ai2_pred_mv[FORW][FIRST],ps_dec->ai2_mv[FORW][FIRST],
        ps_dec->au2_f_code[FORW],0,ps_dec->u2_fld_pic);


        /* Set the second motion vector predictor */
        ps_dec->ai2_pred_mv[FORW][SECOND][MV_X] = ps_dec->ai2_pred_mv[FORW][FIRST][MV_X];
        ps_dec->ai2_pred_mv[FORW][SECOND][MV_Y] = ps_dec->ai2_pred_mv[FORW][FIRST][MV_Y];

        /* Flush the marker bit */
        if(0 == (impeg2d_bit_stream_get(ps_stream,1)))
        {
            /* Ignore marker bit error */
        }
    }
    else
    {
        /* Reset the motion vector predictors */
        memset(ps_dec->ai2_pred_mv,0,sizeof(ps_dec->ai2_pred_mv));
    }
}

/*****************************************************************************
*  Function Name   : impeg2d_dec_skip_p_mb
*
*  Description     : Performs decoding needed for Skipped MB encountered in
*                    P Pictures and B Pictures with previous MB not bi-predicted
*
*  Arguments       :
*  dec             : Decoder state
*
*  Values Returned : None
*****************************************************************************/
void impeg2d_dec_skip_p_mb(dec_state_t *ps_dec, WORD32 u4_num_of_mbs)
{
    WORD16  *pi2_mv;

    e_mb_type_t e_mb_type;
    mb_mc_params_t *ps_mc;


    WORD32 i4_iter;
    UWORD32 u4_dst_wd;
    UWORD32  u4_dst_offset_x;
    UWORD32  u4_dst_offset_y;
    UWORD32 u4_frm_offset = 0;
    yuv_buf_t s_dst;

    u4_dst_wd = ps_dec->u2_frame_width;

    if(ps_dec->u2_picture_structure != FRAME_PICTURE)
    {
        u4_dst_wd <<= 1;
        if(ps_dec->u2_picture_structure == BOTTOM_FIELD)
        {
            u4_frm_offset = ps_dec->u2_frame_width;
        }
    }

    for (i4_iter = u4_num_of_mbs; i4_iter > 0; i4_iter--)
    {
        if(ps_dec->u2_picture_structure == FRAME_PICTURE)
        {
            e_mb_type = MC_FRM_FW_AND_BK_2MV;
        }
        else
        {
            e_mb_type = MC_FLD_FW_AND_BK_2MV;
        }

        ps_dec->u2_prev_intra_mb = 0;
        pi2_mv               = (WORD16 *)&(ps_dec->ai2_mv[FORW][FIRST]);

        /* Reset the motion vector predictors */
        if(ps_dec->e_pic_type == P_PIC)
        {
            memset(ps_dec->ai2_pred_mv,0,sizeof(ps_dec->ai2_pred_mv));
            pi2_mv[MV_X]    = pi2_mv[MV_Y] = 0;

            ps_dec->u2_cbp     = 0;

            pi2_mv           = (WORD16 *)&ps_dec->ai2_mv[FORW][FIRST];
            ps_mc           = &ps_dec->as_mb_mc_params[FORW][FIRST];
            ps_mc->s_ref      = ps_dec->as_ref_buf[ps_dec->e_mb_pred][ps_dec->u2_fld_parity];

            impeg2d_set_mc_params(&ps_mc->s_luma, &ps_mc->s_chroma, e_mb_type, 0,
                      pi2_mv, ps_dec->u2_mb_x, ps_dec->u2_mb_y, ps_dec->u2_frame_width, ps_dec->u2_frame_height,ps_dec->u2_picture_width);


            u4_dst_offset_x = (ps_dec->u2_mb_x << 4) + u4_frm_offset;
            u4_dst_offset_y = (ps_dec->u2_mb_y << 4) * u4_dst_wd;

            s_dst.pu1_y = ps_dec->s_cur_frm_buf.pu1_y + u4_dst_offset_x + u4_dst_offset_y;

            u4_dst_offset_x = u4_dst_offset_x >> 1;
            u4_dst_offset_y = u4_dst_offset_y >> 2;

            s_dst.pu1_u = ps_dec->s_cur_frm_buf.pu1_u + u4_dst_offset_x + u4_dst_offset_y;
            s_dst.pu1_v = ps_dec->s_cur_frm_buf.pu1_v + u4_dst_offset_x + u4_dst_offset_y;


            ps_mc->s_ref.pu1_y += ps_mc->s_luma.u4_src_offset;
            ps_mc->s_ref.pu1_u += ps_mc->s_chroma.u4_src_offset;
            ps_mc->s_ref.pu1_v += ps_mc->s_chroma.u4_src_offset;

            ps_dec->pf_copy_mb(&ps_mc->s_ref, &s_dst, ps_mc->s_luma.u4_src_wd, u4_dst_wd);
        }

        else
        {
            pi2_mv[MV_X]    = ps_dec->ai2_pred_mv[ps_dec->e_mb_pred][FIRST][MV_X];
            pi2_mv[MV_Y]    = ps_dec->ai2_pred_mv[ps_dec->e_mb_pred][FIRST][MV_Y];

            ps_dec->u2_cbp     = 0;

            pi2_mv           = (WORD16 *)&ps_dec->ai2_mv[FORW][FIRST];
            ps_mc           = &ps_dec->as_mb_mc_params[FORW][FIRST];
            ps_mc->s_ref      = ps_dec->as_ref_buf[ps_dec->e_mb_pred][ps_dec->u2_fld_parity];

            impeg2d_set_mc_params(&ps_mc->s_luma, &ps_mc->s_chroma, e_mb_type, 0,
                      pi2_mv, ps_dec->u2_mb_x, ps_dec->u2_mb_y, ps_dec->u2_frame_width, ps_dec->u2_frame_height,ps_dec->u2_picture_width);

            u4_dst_offset_x = (ps_dec->u2_mb_x << 4) + u4_frm_offset;
            u4_dst_offset_y = (ps_dec->u2_mb_y << 4) * u4_dst_wd;

            ps_mc->s_luma.u4_dst_offset_res_buf = u4_dst_offset_x + u4_dst_offset_y;
            ps_mc->s_luma.u4_dst_wd_res_buf = u4_dst_wd;

            u4_dst_offset_x = u4_dst_offset_x >> 1;
            u4_dst_offset_y = u4_dst_offset_y >> 2;

            ps_mc->s_chroma.u4_dst_offset_res_buf = u4_dst_offset_x + u4_dst_offset_y;
            ps_mc->s_chroma.u4_dst_wd_res_buf = u4_dst_wd >> 1;

            impeg2d_motion_comp(ps_dec, ps_mc, &ps_dec->s_cur_frm_buf);
        }


        /********************************************************************/
        /* Common MB processing tasks                                       */
        /********************************************************************/
        ps_dec->u2_mb_x++;
        ps_dec->u2_num_mbs_left--;

        if (ps_dec->u2_mb_x == ps_dec->u2_num_horiz_mb)
        {
            ps_dec->u2_mb_x = 0;
            ps_dec->u2_mb_y++;
        }
    }

}

/*******************************************************************************
*  Function Name   : impeg2d_dec_skip_b_mb
*
*  Description     : Performs processing needed for Skipped MB encountered in
*                    B Pictures with previous MB bi-predicted.
*
*  Arguments       :
*  dec             : Decoder state
*
*  Values Returned : None
*******************************************************************************/
void impeg2d_dec_skip_b_mb(dec_state_t *ps_dec, WORD32 u4_num_of_mbs)
{


    WORD16  *pi2_mv;

    UWORD32 i;
    e_mb_type_t e_mb_type;
    mb_mc_params_t *ps_mc;

    WORD32 i4_iter;
    UWORD32 u4_dst_wd;
    yuv_buf_t s_dst;
    UWORD32  u4_dst_offset_x;
    UWORD32  u4_dst_offset_y;
    UWORD32 u4_frm_offset = 0;

    u4_dst_wd = ps_dec->u2_frame_width;
    s_dst = ps_dec->s_cur_frm_buf;

    if(ps_dec->u2_picture_structure != FRAME_PICTURE)
    {
        u4_dst_wd <<= 1;
        if(ps_dec->u2_picture_structure == BOTTOM_FIELD)
        {
            u4_frm_offset = ps_dec->u2_frame_width;
        }
    }

    for (i4_iter = u4_num_of_mbs; i4_iter > 0; i4_iter--)
    {
        ps_dec->u2_prev_intra_mb = 0;

        if(ps_dec->u2_picture_structure == FRAME_PICTURE)
        {
            e_mb_type = MC_FRM_FW_AND_BK_2MV;
        }
        else
        {
            e_mb_type = MC_FLD_FW_AND_BK_2MV;
        }

        /************************************************************************/
        /* Setting of first motion vector for B MB                              */
        /************************************************************************/
        pi2_mv               = (WORD16 *)&(ps_dec->ai2_mv[FORW][FIRST]);
        {
            pi2_mv[MV_X]         = ps_dec->ai2_pred_mv[FORW][FIRST][MV_X];
            pi2_mv[MV_Y]         = ps_dec->ai2_pred_mv[FORW][FIRST][MV_Y];
        }
        /************************************************************************/
        /* Setting of second motion vector for B MB                             */
        /************************************************************************/
        pi2_mv               = (WORD16 *)&(ps_dec->ai2_mv[BACK][FIRST]);
        {
            pi2_mv[MV_X]         = ps_dec->ai2_pred_mv[BACK][FIRST][MV_X];
            pi2_mv[MV_Y]         = ps_dec->ai2_pred_mv[BACK][FIRST][MV_Y];
        }
        ps_dec->u2_cbp  = 0;

        for(i = 0; i < 2; i++)
        {
            pi2_mv          = (WORD16 *)&ps_dec->ai2_mv[i][FIRST];
            ps_mc          = &ps_dec->as_mb_mc_params[i][FIRST];
            ps_mc->s_ref     = ps_dec->as_ref_buf[i][ps_dec->u2_fld_parity];

            impeg2d_set_mc_params(&ps_mc->s_luma, &ps_mc->s_chroma, e_mb_type, 0, pi2_mv, ps_dec->u2_mb_x,
                          ps_dec->u2_mb_y, ps_dec->u2_frame_width, ps_dec->u2_frame_height,ps_dec->u2_picture_width);
        }

        impeg2d_motion_comp(ps_dec, &ps_dec->as_mb_mc_params[FORW][FIRST], &ps_dec->s_mc_fw_buf);
        impeg2d_motion_comp(ps_dec, &ps_dec->as_mb_mc_params[BACK][FIRST], &ps_dec->s_mc_bk_buf);

        u4_dst_offset_x = (ps_dec->u2_mb_x << 4) + u4_frm_offset;
        u4_dst_offset_y = (ps_dec->u2_mb_y << 4) * u4_dst_wd;

        s_dst.pu1_y = ps_dec->s_cur_frm_buf.pu1_y + u4_dst_offset_x + u4_dst_offset_y;

        u4_dst_offset_x = u4_dst_offset_x >> 1;
        u4_dst_offset_y = u4_dst_offset_y >> 2;

        s_dst.pu1_u = ps_dec->s_cur_frm_buf.pu1_u + u4_dst_offset_x + u4_dst_offset_y;
        s_dst.pu1_v = ps_dec->s_cur_frm_buf.pu1_v + u4_dst_offset_x + u4_dst_offset_y;

        ps_dec->pf_interpolate(&ps_dec->s_mc_fw_buf,&ps_dec->s_mc_bk_buf,&s_dst, u4_dst_wd);
//        dec->pf_copy_mb(&dec->mc_buf, &dst, MB_SIZE, dst_wd);

        /********************************************************************/
        /* Common MB processing tasks                                       */
        /********************************************************************/
        ps_dec->u2_mb_x++;
        ps_dec->u2_num_mbs_left--;

        if (ps_dec->u2_mb_x == ps_dec->u2_num_horiz_mb)
        {
            ps_dec->u2_mb_x = 0;
            ps_dec->u2_mb_y++;
        }
    }
}
/*******************************************************************************
*  Function Name   : impeg2d_dec_skip_mbs
*
*  Description     : Performs processing needed for Skipped MB encountered in
*                    B Pictures with previous MB bi-predicted.
*
*  Arguments       :
*  dec             : Decoder state
*
*  Values Returned : None
*******************************************************************************/
void impeg2d_dec_skip_mbs(dec_state_t *ps_dec, UWORD16 u2_num_skip_mbs)
{
    PROFILE_DISABLE_SKIP_MB();

    if(ps_dec->e_mb_pred == BIDIRECT)
    {
        impeg2d_dec_skip_b_mb(ps_dec, u2_num_skip_mbs);
    }
    else
    {
        impeg2d_dec_skip_p_mb(ps_dec, u2_num_skip_mbs);
    }

    ps_dec->u2_def_dc_pred[Y_LUMA] = 128 << ps_dec->u2_intra_dc_precision;
    ps_dec->u2_def_dc_pred[U_CHROMA] = 128 << ps_dec->u2_intra_dc_precision;
    ps_dec->u2_def_dc_pred[V_CHROMA] = 128 << ps_dec->u2_intra_dc_precision;
}


/*****************************************************************************
*  Function Name   : impeg2d_dec_0mv_coded_mb
*
*  Description     : Decodes the MB with 0 MV but coded. This can occur in P
*                    pictures only
*
*  Arguments       :
*  dec             : Decoder state
*
*  Values Returned : None
*****************************************************************************/
void impeg2d_dec_0mv_coded_mb(dec_state_t *ps_dec)
{


    WORD16   *pi2_mv;
    e_mb_type_t e_mb_type;
    mb_mc_params_t *ps_mc;

    if(ps_dec->u2_picture_structure == FRAME_PICTURE)
    {
        e_mb_type = MC_FRM_FW_AND_BK_2MV;
    }
    else
    {
        e_mb_type = MC_FLD_FW_AND_BK_2MV;
    }


    /* Reset the motion vector predictors */
    memset(ps_dec->ai2_pred_mv,0,sizeof(ps_dec->ai2_pred_mv));

    pi2_mv           = (WORD16 *)&ps_dec->ai2_mv[FORW][FIRST];
    ps_mc           = &ps_dec->as_mb_mc_params[FORW][FIRST];
    ps_mc->s_ref      = ps_dec->as_ref_buf[FORW][ps_dec->u2_fld_parity];

    pi2_mv[MV_X] = 0;
    pi2_mv[MV_Y] = 0;

    impeg2d_set_mc_params(&ps_mc->s_luma, &ps_mc->s_chroma, e_mb_type, 0,
              pi2_mv, ps_dec->u2_mb_x, ps_dec->u2_mb_y, ps_dec->u2_frame_width, ps_dec->u2_frame_height,ps_dec->u2_picture_width);
}

/*****************************************************************************/
/*                                                                           */
/*  Function Name : impeg2d_mc_halfx_halfy()                                 */
/*                                                                           */
/*  Description   : Gets the buffer from (0.5,0.5) to (8.5,8.5)              */
/*                  and the above block of size 8 x 8 will be placed as a    */
/*                  block from the current position of out_buf               */
/*                                                                           */
/*  Inputs        : ref - Reference frame from which the block will be       */
/*                        block will be extracted.                           */
/*                  ref_wid - WIdth of reference frame                       */
/*                  out_wid - WIdth of the output frame                      */
/*                  blk_width  - width of the block                          */
/*                  blk_width  - height of the block                         */
/*                                                                           */
/*  Globals       : None                                                     */
/*                                                                           */
/*  Processing    : Point to the (0,0),(1,0),(0,1),(1,1) position in         */
/*                  the ref frame.Interpolate these four values to get the   */
/*                  value at(0.5,0.5).Repeat this to get an 8 x 8 block      */
/*                  using 9 x 9 block from reference frame                   */
/*                                                                           */
/*  Outputs       : out -  Output containing the extracted block             */
/*                                                                           */
/*  Returns       : None                                                     */
/*                                                                           */
/*  Issues        : None                                                     */
/*                                                                           */
/*  Revision History:                                                        */
/*                                                                           */
/*         DD MM YYYY   Author(s)       Changes                              */
/*         05 09 2005   Harish M        First Version                        */
/*                                                                           */
/*****************************************************************************/
void impeg2d_mc_halfx_halfy(void *pv_dec,
                           UWORD8 *pu1_ref,
                           UWORD32 u4_ref_wid,
                           UWORD8 *pu1_out,
                           UWORD32 u4_out_wid,
                           UWORD32 u4_blk_width,
                           UWORD32 u4_blk_height)
{
   UWORD8 *pu1_out_ptr,*pu1_ref_ptr;
   dec_state_t *ps_dec = (dec_state_t *)pv_dec;

        pu1_out_ptr = pu1_out;
        pu1_ref_ptr = pu1_ref;

    if((u4_blk_width == MB_SIZE) && (u4_blk_height == MB_SIZE))
    {

        /*luma 16 x 16*/

        /*block 0*/
        ps_dec->pf_halfx_halfy_8x8(pu1_out_ptr,pu1_ref_ptr,u4_ref_wid,u4_out_wid);

        /*block1*/
        pu1_out_ptr = (pu1_out + BLK_SIZE);
        pu1_ref_ptr = (pu1_ref + BLK_SIZE);
        ps_dec->pf_halfx_halfy_8x8(pu1_out_ptr,pu1_ref_ptr,u4_ref_wid,u4_out_wid);

        /*block 2*/
        pu1_out_ptr = pu1_out + BLK_SIZE * u4_out_wid;
        pu1_ref_ptr = pu1_ref + BLK_SIZE * u4_ref_wid;
        ps_dec->pf_halfx_halfy_8x8(pu1_out_ptr,pu1_ref_ptr,u4_ref_wid,u4_out_wid);

        /*block 3*/
        pu1_out_ptr = pu1_out + BLK_SIZE * u4_out_wid + BLK_SIZE;
        pu1_ref_ptr = pu1_ref + BLK_SIZE * u4_ref_wid + BLK_SIZE;
        ps_dec->pf_halfx_halfy_8x8(pu1_out_ptr,pu1_ref_ptr,u4_ref_wid,u4_out_wid);


    }
    else if ((u4_blk_width == BLK_SIZE) && (u4_blk_height == BLK_SIZE))
    {
        /*chroma 8 x 8*/
        ps_dec->pf_halfx_halfy_8x8(pu1_out_ptr,pu1_ref_ptr,u4_ref_wid,u4_out_wid);
    }
    else if ((u4_blk_width == MB_SIZE) && (u4_blk_height == BLK_SIZE))
    {
        /*block 0*/
        ps_dec->pf_halfx_halfy_8x8(pu1_out_ptr,pu1_ref_ptr,u4_ref_wid,u4_out_wid);

        /*block 1*/
        pu1_out_ptr = (pu1_out + BLK_SIZE);
        pu1_ref_ptr = (pu1_ref + BLK_SIZE);
        ps_dec->pf_halfx_halfy_8x8(pu1_out_ptr,pu1_ref_ptr,u4_ref_wid,u4_out_wid);

    }

    else
    {
        UWORD8 *ref_p0,*ref_p1,*ref_p2,*ref_p3;
        UWORD32 i,j;
        /* P0-P3 are the pixels in the reference frame and Q is the value being */
        /* estimated                                                            */
        /*
           P0 P1
             Q
           P2 P3
        */

        ref_p0 = pu1_ref;
        ref_p1 = pu1_ref + 1;
        ref_p2 = pu1_ref + u4_ref_wid;
        ref_p3 = pu1_ref + u4_ref_wid + 1;

        for(i = 0; i < u4_blk_height; i++)
        {
            for(j = 0; j < u4_blk_width; j++)
            {
                *pu1_out++ =   (( (*ref_p0++ )
                            + (*ref_p1++ )
                            + (*ref_p2++ )
                            + (*ref_p3++ ) + 2 ) >> 2);
            }
            ref_p0 += u4_ref_wid - u4_blk_width;
            ref_p1 += u4_ref_wid - u4_blk_width;
            ref_p2 += u4_ref_wid - u4_blk_width;
            ref_p3 += u4_ref_wid - u4_blk_width;

            pu1_out    += u4_out_wid - u4_blk_width;
        }
    }
    return;
}

/*****************************************************************************/
/*                                                                           */
/*  Function Name : impeg2d_mc_halfx_fully()                                 */
/*                                                                           */
/*  Description   : Gets the buffer from (0.5,0) to (8.5,8)                  */
/*                  and the above block of size 8 x 8 will be placed as a    */
/*                  block from the current position of out_buf               */
/*                                                                           */
/*  Inputs        : ref - Reference frame from which the block will be       */
/*                        block will be extracted.                           */
/*                  ref_wid - WIdth of reference frame                       */
/*                  out_wid - WIdth of the output frame                      */
/*                  blk_width  - width of the block                          */
/*                  blk_width  - height of the block                         */
/*                                                                           */
/*  Globals       : None                                                     */
/*                                                                           */
/*  Processing    : Point to the (0,0) and (1,0) position in the ref frame   */
/*                  Interpolate these two values to get the value at(0.5,0)  */
/*                  Repeat this to get an 8 x 8 block using 9 x 8 block from */
/*                  reference frame                                          */
/*                                                                           */
/*  Outputs       : out -  Output containing the extracted block             */
/*                                                                           */
/*  Returns       : None                                                     */
/*                                                                           */
/*  Issues        : None                                                     */
/*                                                                           */
/*  Revision History:                                                        */
/*                                                                           */
/*         DD MM YYYY   Author(s)       Changes                              */
/*         05 09 2005   Harish M        First Version                        */
/*                                                                           */
/*****************************************************************************/

void impeg2d_mc_halfx_fully(void *pv_dec,
                            UWORD8 *pu1_ref,
                            UWORD32 u4_ref_wid,
                            UWORD8 *pu1_out,
                            UWORD32 u4_out_wid,
                            UWORD32 u4_blk_width,
                            UWORD32 u4_blk_height)
{
    UWORD8 *pu1_out_ptr,*pu1_ref_ptr;
    dec_state_t *ps_dec = (dec_state_t *)pv_dec;

        pu1_out_ptr = pu1_out;
        pu1_ref_ptr = pu1_ref;

    if((u4_blk_width == MB_SIZE) && (u4_blk_height == MB_SIZE))
    {

        /*luma 16 x 16*/

        /*block 0*/
        ps_dec->pf_halfx_fully_8x8(pu1_out_ptr,pu1_ref_ptr,u4_ref_wid,u4_out_wid);

        /*block1*/
        pu1_out_ptr = (pu1_out + BLK_SIZE);
        pu1_ref_ptr = (pu1_ref + BLK_SIZE);
        ps_dec->pf_halfx_fully_8x8(pu1_out_ptr,pu1_ref_ptr,u4_ref_wid,u4_out_wid);

        /*block 2*/
        pu1_out_ptr = pu1_out + BLK_SIZE * u4_out_wid;
        pu1_ref_ptr = pu1_ref + BLK_SIZE * u4_ref_wid;
        ps_dec->pf_halfx_fully_8x8(pu1_out_ptr,pu1_ref_ptr,u4_ref_wid,u4_out_wid);

        /*block 3*/
        pu1_out_ptr = pu1_out + BLK_SIZE * u4_out_wid + BLK_SIZE;
        pu1_ref_ptr = pu1_ref + BLK_SIZE * u4_ref_wid + BLK_SIZE;
        ps_dec->pf_halfx_fully_8x8(pu1_out_ptr,pu1_ref_ptr,u4_ref_wid,u4_out_wid);


    }
    else if ((u4_blk_width == BLK_SIZE) && (u4_blk_height == BLK_SIZE))
    {
        /*chroma 8 x 8*/
        ps_dec->pf_halfx_fully_8x8(pu1_out_ptr,pu1_ref_ptr,u4_ref_wid,u4_out_wid);
    }
    else if ((u4_blk_width == MB_SIZE) && (u4_blk_height == BLK_SIZE))
    {
        /*block 0*/
        ps_dec->pf_halfx_fully_8x8(pu1_out_ptr,pu1_ref_ptr,u4_ref_wid,u4_out_wid);

        /*block 1*/
        pu1_out_ptr = (pu1_out + BLK_SIZE);
        pu1_ref_ptr = (pu1_ref + BLK_SIZE);
        ps_dec->pf_halfx_fully_8x8(pu1_out_ptr,pu1_ref_ptr,u4_ref_wid,u4_out_wid);

    }

    else
    {
        UWORD8 *ref_p0,*ref_p1;
        UWORD32 i,j;

        /* P0-P3 are the pixels in the reference frame and Q is the value being */
        /* estimated                                                            */
        /*
           P0 Q P1
        */

        ref_p0 = pu1_ref;
        ref_p1 = pu1_ref + 1;

        for(i = 0; i < u4_blk_height; i++)
        {
            for(j = 0; j < u4_blk_width; j++)
            {
                *pu1_out++ =   ((( *ref_p0++ )
                            + (*ref_p1++) + 1 ) >> 1);
            }
            ref_p0 += u4_ref_wid - u4_blk_width;
            ref_p1 += u4_ref_wid - u4_blk_width;

            pu1_out    += u4_out_wid - u4_blk_width;
        }
    }
    return;
}


/*****************************************************************************/
/*                                                                           */
/*  Function Name : impeg2d_mc_fullx_halfy()                                 */
/*                                                                           */
/*  Description   : Gets the buffer from (0,0.5) to (8,8.5)                  */
/*                  and the above block of size 8 x 8 will be placed as a    */
/*                  block from the current position of out_buf               */
/*                                                                           */
/*  Inputs        : ref - Reference frame from which the block will be       */
/*                        block will be extracted.                           */
/*                  ref_wid - WIdth of reference frame                       */
/*                  out_wid - WIdth of the output frame                      */
/*                  blk_width  - width of the block                          */
/*                  blk_width  - height of the block                         */
/*                                                                           */
/*  Globals       : None                                                     */
/*                                                                           */
/*  Processing    : Point to the (0,0) and (0,1)   position in the ref frame */
/*                  Interpolate these two values to get the value at(0,0.5)  */
/*                  Repeat this to get an 8 x 8 block using 8 x 9 block from */
/*                  reference frame                                          */
/*                                                                           */
/*  Outputs       : out -  Output containing the extracted block             */
/*                                                                           */
/*  Returns       : None                                                     */
/*                                                                           */
/*  Issues        : None                                                     */
/*                                                                           */
/*  Revision History:                                                        */
/*                                                                           */
/*         DD MM YYYY   Author(s)       Changes                              */
/*         05 09 2005   Harish M        First Version                        */
/*                                                                           */
/*****************************************************************************/
void impeg2d_mc_fullx_halfy(void *pv_dec,
                            UWORD8 *pu1_ref,
                            UWORD32 u4_ref_wid,
                            UWORD8 *pu1_out,
                            UWORD32 u4_out_wid,
                            UWORD32 u4_blk_width,
                            UWORD32 u4_blk_height)
{

    UWORD8 *pu1_out_ptr,*pu1_ref_ptr;
    dec_state_t *ps_dec = (dec_state_t *)pv_dec;
        pu1_out_ptr = pu1_out;
        pu1_ref_ptr = pu1_ref;

    if((u4_blk_width == MB_SIZE) && (u4_blk_height == MB_SIZE))
    {

        /*luma 16 x 16*/

        /*block 0*/
        ps_dec->pf_fullx_halfy_8x8(pu1_out_ptr,pu1_ref_ptr,u4_ref_wid,u4_out_wid);

        /*block1*/
        pu1_out_ptr = (pu1_out + BLK_SIZE);
        pu1_ref_ptr = (pu1_ref + BLK_SIZE);
        ps_dec->pf_fullx_halfy_8x8(pu1_out_ptr,pu1_ref_ptr,u4_ref_wid,u4_out_wid);

        /*block 2*/
        pu1_out_ptr = pu1_out + BLK_SIZE * u4_out_wid;
        pu1_ref_ptr = pu1_ref + BLK_SIZE * u4_ref_wid;
        ps_dec->pf_fullx_halfy_8x8(pu1_out_ptr,pu1_ref_ptr,u4_ref_wid,u4_out_wid);

        /*block 3*/
        pu1_out_ptr = pu1_out + BLK_SIZE * u4_out_wid + BLK_SIZE;
        pu1_ref_ptr = pu1_ref + BLK_SIZE * u4_ref_wid + BLK_SIZE;
        ps_dec->pf_fullx_halfy_8x8(pu1_out_ptr,pu1_ref_ptr,u4_ref_wid,u4_out_wid);


    }
    else if ((u4_blk_width == BLK_SIZE) && (u4_blk_height == BLK_SIZE))
    {
        /*chroma 8 x 8*/
        ps_dec->pf_fullx_halfy_8x8(pu1_out_ptr,pu1_ref_ptr,u4_ref_wid,u4_out_wid);
    }
    else if ((u4_blk_width == MB_SIZE) && (u4_blk_height == BLK_SIZE))
    {
        /*block 0*/
        ps_dec->pf_fullx_halfy_8x8(pu1_out_ptr,pu1_ref_ptr,u4_ref_wid,u4_out_wid);

        /*block 1*/
        pu1_out_ptr = (pu1_out + BLK_SIZE);
        pu1_ref_ptr = (pu1_ref + BLK_SIZE);
        ps_dec->pf_fullx_halfy_8x8(pu1_out_ptr,pu1_ref_ptr,u4_ref_wid,u4_out_wid);

    }

    else if ((u4_blk_width == BLK_SIZE) && (u4_blk_height == (BLK_SIZE / 2)))
    {
        UWORD8 *ref_p0,*ref_p1;
        UWORD32 i,j;
        /* P0-P3 are the pixels in the reference frame and Q is the value being */
        /* estimated                                                            */
        /*
           P0
            x
           P1
        */
        ref_p0 = pu1_ref;
        ref_p1 = pu1_ref + u4_ref_wid;

        for(i = 0; i < u4_blk_height; i++)
        {
            for(j = 0; j < u4_blk_width; j++)
            {
                *pu1_out++ =   ((( *ref_p0++)
                            + (*ref_p1++) + 1 ) >> 1);
            }
            ref_p0 += u4_ref_wid - u4_blk_width;
            ref_p1 += u4_ref_wid - u4_blk_width;

            pu1_out    += u4_out_wid - u4_blk_width;
        }
    }
    return;
}

/*****************************************************************************/
/*                                                                           */
/*  Function Name : impeg2d_mc_fullx_fully()                                 */
/*                                                                           */
/*  Description   : Gets the buffer from (x,y) to (x+8,y+8)                  */
/*                  and the above block of size 8 x 8 will be placed as a    */
/*                  block from the current position of out_buf               */
/*                                                                           */
/*  Inputs        : ref - Reference frame from which the block will be       */
/*                        block will be extracted.                           */
/*                  ref_wid - WIdth of reference frame                       */
/*                  out_wid - WIdth of the output frame                      */
/*                  blk_width  - width of the block                          */
/*                  blk_width  - height of the block                         */
/*                                                                           */
/*  Globals       : None                                                     */
/*                                                                           */
/*  Processing    : Point to the (0,0) position in the ref frame             */
/*                  Get an 8 x 8 block from reference frame                  */
/*                                                                           */
/*  Outputs       : out -  Output containing the extracted block             */
/*                                                                           */
/*  Returns       : None                                                     */
/*                                                                           */
/*  Issues        : None                                                     */
/*                                                                           */
/*  Revision History:                                                        */
/*                                                                           */
/*         DD MM YYYY   Author(s)       Changes                              */
/*         05 09 2005   Harish M        First Version                        */
/*                                                                           */
/*****************************************************************************/

void impeg2d_mc_fullx_fully(void *pv_dec,
                            UWORD8 *pu1_ref,
                            UWORD32 u4_ref_wid,
                            UWORD8 *pu1_out,
                            UWORD32 u4_out_wid,
                            UWORD32 u4_blk_width,
                            UWORD32 u4_blk_height)
{

    UWORD8 *pu1_out_ptr,*pu1_ref_ptr;
    dec_state_t *ps_dec = (dec_state_t *)pv_dec;

        pu1_out_ptr = pu1_out;
        pu1_ref_ptr = pu1_ref;

    if((u4_blk_width == MB_SIZE) && (u4_blk_height == MB_SIZE))
    {

        /*luma 16 x 16*/

        /*block 0*/
        ps_dec->pf_fullx_fully_8x8(pu1_out_ptr,pu1_ref_ptr,u4_ref_wid,u4_out_wid);

        /*block1*/
        pu1_out_ptr = (pu1_out + BLK_SIZE);
        pu1_ref_ptr = (pu1_ref + BLK_SIZE);
        ps_dec->pf_fullx_fully_8x8(pu1_out_ptr,pu1_ref_ptr,u4_ref_wid,u4_out_wid);

        /*block 2*/
        pu1_out_ptr = pu1_out + BLK_SIZE * u4_out_wid;
        pu1_ref_ptr = pu1_ref + BLK_SIZE * u4_ref_wid;
        ps_dec->pf_fullx_fully_8x8(pu1_out_ptr,pu1_ref_ptr,u4_ref_wid,u4_out_wid);

        /*block 3*/
        pu1_out_ptr = pu1_out + BLK_SIZE * u4_out_wid + BLK_SIZE;
        pu1_ref_ptr = pu1_ref + BLK_SIZE * u4_ref_wid + BLK_SIZE;
        ps_dec->pf_fullx_fully_8x8(pu1_out_ptr,pu1_ref_ptr,u4_ref_wid,u4_out_wid);


    }
    else if ((u4_blk_width == BLK_SIZE) && (u4_blk_height == BLK_SIZE))
    {
        /*chroma 8 x 8*/
        ps_dec->pf_fullx_fully_8x8(pu1_out_ptr,pu1_ref_ptr,u4_ref_wid,u4_out_wid);
    }
    else if ((u4_blk_width == MB_SIZE) && (u4_blk_height == BLK_SIZE))
    {
        /*block 0*/
        ps_dec->pf_fullx_fully_8x8(pu1_out_ptr,pu1_ref_ptr,u4_ref_wid,u4_out_wid);

        /*block 1*/
        pu1_out_ptr = (pu1_out + BLK_SIZE);
        pu1_ref_ptr = (pu1_ref + BLK_SIZE);
        ps_dec->pf_fullx_fully_8x8(pu1_out_ptr,pu1_ref_ptr,u4_ref_wid,u4_out_wid);

    }
    else
    {
        UWORD32 i;

        for(i = 0; i < u4_blk_height; i++)
        {
            memmove(pu1_out, pu1_ref, u4_blk_width);
            pu1_ref += u4_ref_wid;
            pu1_out += u4_out_wid;
        }
    }
    return;
}

/*******************************************************************************
*  Function Name   : impeg2d_set_mc_params
*
*  Description     : Sets the parameters for Motion Compensation
*
*  Arguments       :
*  luma            : Parameters for luma blocks
*  chroma          : Parameters for chroma blocks
*  type            : Motion compensation type
*  mv_num          : Number of motion vectors
*  mv              : Motion Vectors
*  mb_x            : X co-ordinate of MB
*  mb_y            : Y co-ordinate of MB
*  frm_wd          : Width of the frame
*
*  Values Returned : None
*******************************************************************************/
void impeg2d_set_mc_params(comp_mc_params_t *ps_luma,
                           comp_mc_params_t *ps_chroma,
                           e_mb_type_t e_type,
                           UWORD16 u2_mv_num,
                           WORD16 ai2_mv[],
                           UWORD16 u2_mb_x,
                           UWORD16 u2_mb_y,
                           UWORD16 u2_frm_wd,
                           UWORD16 u2_frm_ht,
                           UWORD16 u2_picture_width)
{
    WORD16 i2_mvy_round;
    WORD16 i2_mvx_round;
    const mc_type_consts_t *ps_mc_params;
    WORD16 i2_mvx_fullp_round;
    WORD16 i2_mvy_fullp_round;
    UWORD32 u4_frm_chroma_wd;
    WORD16 i2_pix_x, i2_pix_y;

    ps_mc_params = &gas_impeg2d_mc_params_luma[e_type][u2_mv_num];
    /****************************************************************************/
    /* get luma mc params                                                       */
    /****************************************************************************/
    i2_pix_x = MB_SIZE * u2_mb_x + (ai2_mv[MV_X]>>1);
    i2_pix_y = (MB_SIZE * u2_mb_y  +
        (ai2_mv[MV_Y]>>1) * ps_mc_params->mvy_cf + u2_mv_num * ps_mc_params->mv_num_cf) * ps_mc_params->frm_wd_cf;

    // clip pix_x and pix_y so as it falls inside the frame boundary
    CLIP(i2_pix_x, (u2_frm_wd-16), 0);
    CLIP(i2_pix_y, (u2_frm_ht-16), 0);

    ps_luma->u4_src_offset = i2_pix_x +  i2_pix_y * u2_frm_wd;


    /* keep offset  in full pel */
    ps_luma->u4_rows          = ps_mc_params->rows;
    ps_luma->u4_cols          = MB_SIZE;
    ps_luma->u4_dst_wd_res_buf        = ps_mc_params->dst_wd;
    ps_luma->u4_src_wd        = u2_frm_wd * ps_mc_params->src_wd_cf;
    ps_luma->u4_dst_offset_res_buf    = ps_mc_params->dst_offset_scale * MB_SIZE;
    ps_luma->u4_dst_offset_cur_frm    = ps_mc_params->dst_offset_scale * u2_picture_width;
    ps_luma->u4_mode          = ((ai2_mv[MV_X] & 1) << 1) | (ai2_mv[MV_Y] & 1);

    /****************************************************************************/
    /* get chroma mc params                                                     */
    /****************************************************************************/
    ps_mc_params   = &gas_impeg2d_mc_params_chroma[e_type][u2_mv_num];
    i2_mvx_round   = ((ai2_mv[MV_X] + IS_NEG(ai2_mv[MV_X]))>>1);
    i2_mvy_round   = ((ai2_mv[MV_Y] + IS_NEG(ai2_mv[MV_Y]))>>1);

    i2_mvx_fullp_round = (i2_mvx_round>>1);
    i2_mvy_fullp_round = (i2_mvy_round>>1)*ps_mc_params->mvy_cf;

    u4_frm_chroma_wd = (u2_frm_wd>>1);

    i2_pix_x = (MB_SIZE/2) * u2_mb_x + i2_mvx_fullp_round;
    i2_pix_y = ((MB_SIZE/2) * u2_mb_y + i2_mvy_fullp_round + u2_mv_num *
                           ps_mc_params->mv_num_cf)*ps_mc_params->frm_wd_cf;

    CLIP(i2_pix_x, ((u2_frm_wd / 2)-8), 0);
    CLIP(i2_pix_y, ((u2_frm_ht / 2)-8), 0);
    ps_chroma->u4_src_offset = i2_pix_x + i2_pix_y * u4_frm_chroma_wd;


    /* keep offset  in full pel */
    ps_chroma->u4_rows = ps_mc_params->rows;
    ps_chroma->u4_cols        = (MB_SIZE >> 1);
    ps_chroma->u4_dst_wd_res_buf = ps_mc_params->dst_wd;
    ps_chroma->u4_src_wd = (u2_frm_wd>>1) * ps_mc_params->src_wd_cf;
    ps_chroma->u4_dst_offset_res_buf = ps_mc_params->dst_offset_scale * MB_CHROMA_SIZE;
    ps_chroma->u4_dst_offset_cur_frm = ps_mc_params->dst_offset_scale * (u2_picture_width >> 1);
    ps_chroma->u4_mode = ((i2_mvx_round & 1) << 1) | (i2_mvy_round & 1);


    ps_luma->u4_dst_wd_cur_frm = u2_picture_width;
    ps_chroma->u4_dst_wd_cur_frm = u2_picture_width >> 1;

    if(ps_luma->u4_dst_wd_res_buf == MB_SIZE * 2)
    {
        ps_luma->u4_dst_wd_cur_frm = u2_frm_wd << 1;
        ps_chroma->u4_dst_wd_cur_frm = u2_frm_wd;
    }
}