/******************************************************************************

 *

 * 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

*/

#include <stdio.h>

#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_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_vld_tables.h"

#include "impeg2d_vld.h"

#include "impeg2d_pic_proc.h"

#include "impeg2d_debug.h"

void impeg2d_init_function_ptr(void *pv_codec);

void impeg2d_format_convert(dec_state_t *ps_dec,

                            pic_buf_t *ps_src_pic,

                            iv_yuv_buf_t    *ps_disp_frm_buf,

                            UWORD32 u4_start_row, UWORD32 u4_num_rows)

{

    UWORD8  *pu1_src_y,*pu1_src_u,*pu1_src_v;

    UWORD8  *pu1_dst_y,*pu1_dst_u,*pu1_dst_v;

    if((NULL == ps_src_pic) || (NULL == ps_src_pic->pu1_y) || (0 == u4_num_rows))

            return;

    pu1_src_y   = ps_src_pic->pu1_y + (u4_start_row * ps_dec->u2_frame_width);

    pu1_src_u   = ps_src_pic->pu1_u + ((u4_start_row >> 1) * (ps_dec->u2_frame_width >> 1));

    pu1_src_v   = ps_src_pic->pu1_v + ((u4_start_row >> 1) *(ps_dec->u2_frame_width >> 1));

    pu1_dst_y  =  (UWORD8 *)ps_disp_frm_buf->pv_y_buf + (u4_start_row *  ps_dec->u4_frm_buf_stride);

    pu1_dst_u =   (UWORD8 *)ps_disp_frm_buf->pv_u_buf +((u4_start_row >> 1)*(ps_dec->u4_frm_buf_stride >> 1));

    pu1_dst_v =   (UWORD8 *)ps_disp_frm_buf->pv_v_buf +((u4_start_row >> 1)*(ps_dec->u4_frm_buf_stride >> 1));

    if (IV_YUV_420P == ps_dec->i4_chromaFormat)

    {

        ps_dec->pf_copy_yuv420p_buf(pu1_src_y, pu1_src_u, pu1_src_v, pu1_dst_y,

                                    pu1_dst_u, pu1_dst_v,

                                    ps_dec->u2_horizontal_size,

                                    u4_num_rows,

                                    ps_dec->u2_frame_width,

                                    (ps_dec->u2_frame_width >> 1),

                                    (ps_dec->u2_frame_width >> 1),

                                    ps_dec->u4_frm_buf_stride,

                                    (ps_dec->u4_frm_buf_stride >> 1),

                                    (ps_dec->u4_frm_buf_stride >> 1));

    }

    else if (IV_YUV_422ILE == ps_dec->i4_chromaFormat)

    {

        void    *pv_yuv422i;

        UWORD32 u2_height,u2_width,u2_stride_y,u2_stride_u,u2_stride_v;

        UWORD32 u2_stride_yuv422i;

        pv_yuv422i          = (UWORD8 *)ps_disp_frm_buf->pv_y_buf + ((ps_dec->u2_vertical_size)*(ps_dec->u4_frm_buf_stride));

        u2_height           = u4_num_rows;

        u2_width            = ps_dec->u2_horizontal_size;

        u2_stride_y         = ps_dec->u2_frame_width;

        u2_stride_u         = u2_stride_y >> 1;

        u2_stride_v         = u2_stride_u;

        u2_stride_yuv422i   = (0 == ps_dec->u4_frm_buf_stride) ? ps_dec->u2_horizontal_size : ps_dec->u4_frm_buf_stride;

        ps_dec->pf_fmt_conv_yuv420p_to_yuv422ile(pu1_src_y,

            pu1_src_u,

            pu1_src_v,

            pv_yuv422i,

            u2_width,

            u2_height,

            u2_stride_y,

            u2_stride_u,

            u2_stride_v,

            u2_stride_yuv422i);

    }

    else if((ps_dec->i4_chromaFormat == IV_YUV_420SP_UV) ||

            (ps_dec->i4_chromaFormat == IV_YUV_420SP_VU))

    {

        UWORD32 dest_inc_Y=0,dest_inc_UV=0;

        WORD32 convert_uv_only;

        pu1_dst_u =   (UWORD8 *)ps_disp_frm_buf->pv_u_buf +((u4_start_row >> 1)*(ps_dec->u4_frm_buf_stride));

        dest_inc_Y =    ps_dec->u4_frm_buf_stride;

        dest_inc_UV =   ((ps_dec->u4_frm_buf_stride + 1) >> 1) << 1;

        convert_uv_only = 0;

        if(1 == ps_dec->u4_share_disp_buf)

            convert_uv_only = 1;

        if(ps_src_pic->pu1_y == ps_disp_frm_buf->pv_y_buf)

            convert_uv_only = 1;

        if(ps_dec->i4_chromaFormat == IV_YUV_420SP_UV)

        {

            ps_dec->pf_fmt_conv_yuv420p_to_yuv420sp_uv(pu1_src_y,

                pu1_src_u,

                pu1_src_v,

                pu1_dst_y,

                pu1_dst_u,

                u4_num_rows,

                ps_dec->u2_horizontal_size,

                ps_dec->u2_frame_width,

                ps_dec->u2_frame_width >> 1,

                ps_dec->u2_frame_width >> 1,

                dest_inc_Y,

                dest_inc_UV,

                convert_uv_only);

        }

        else

        {

            ps_dec->pf_fmt_conv_yuv420p_to_yuv420sp_vu(pu1_src_y,

                    pu1_src_u,

                    pu1_src_v,

                    pu1_dst_y,

                    pu1_dst_u,

                    u4_num_rows,

                    ps_dec->u2_horizontal_size,

                    ps_dec->u2_frame_width,

                    ps_dec->u2_frame_width >> 1,

                    ps_dec->u2_frame_width >> 1,

                    dest_inc_Y,

                    dest_inc_UV,

                    convert_uv_only);

        }

    }

}

/*******************************************************************************

*

*  Function Name   : impeg2d_get_frm_buf

*

*  Description     : Gets YUV component buffers for the frame

*

*  Arguments       :

*  frm_buf         : YUV buffer

*  frm             : Reference frame

*  width           : Width of the frame

*  Height          : Height of the frame

*

*  Values Returned : None

*******************************************************************************/

void impeg2d_get_frm_buf(yuv_buf_t *ps_frm_buf,UWORD8 *pu1_frm,UWORD32 u4_width,UWORD32 u4_height)

{

   UWORD32 u4_luma_size = u4_width * u4_height;

   UWORD32 u4_chroma_size = (u4_width * u4_height)>>2;

   ps_frm_buf->pu1_y = pu1_frm;

   ps_frm_buf->pu1_u = pu1_frm + u4_luma_size;

   ps_frm_buf->pu1_v = pu1_frm + u4_luma_size + u4_chroma_size;

}

/*******************************************************************************

*

*  Function Name   : impeg2d_get_bottom_field_buf

*

*  Description     : Gets YUV component buffers for bottom field of the frame

*

*  Arguments       :

*  frm_buf         : YUV buffer

*  frm             : Reference frame

*  width           : Width of the frame

*  Height          : Height of the frame

*

*  Values Returned : None

*******************************************************************************/

void impeg2d_get_bottom_field_buf(yuv_buf_t *ps_src_buf,yuv_buf_t *ps_dst_buf,

                      UWORD32 u4_width)

{

   ps_dst_buf->pu1_y = ps_src_buf->pu1_y + u4_width;

   ps_dst_buf->pu1_u = ps_src_buf->pu1_u + (u4_width>>1);

   ps_dst_buf->pu1_v = ps_src_buf->pu1_v + (u4_width>>1);

}

/*******************************************************************************

*  Function Name   : impeg2d_get_mb_addr_incr

*

*  Description     : Decodes the Macroblock address increment

*

*  Arguments       :

*  stream          : Bitstream

*

*  Values Returned : Macroblock address increment

*******************************************************************************/

UWORD16 impeg2d_get_mb_addr_incr(stream_t *ps_stream)

{

    UWORD16 u2_mb_addr_incr = 0;

    while (impeg2d_bit_stream_nxt(ps_stream,MB_ESCAPE_CODE_LEN) == MB_ESCAPE_CODE &&

            ps_stream->u4_offset < ps_stream->u4_max_offset)

    {

        impeg2d_bit_stream_flush(ps_stream,MB_ESCAPE_CODE_LEN);

        u2_mb_addr_incr += 33;

    }

    u2_mb_addr_incr += impeg2d_dec_vld_symbol(ps_stream,gai2_impeg2d_mb_addr_incr,MB_ADDR_INCR_LEN) +

        MB_ADDR_INCR_OFFSET;

    return(u2_mb_addr_incr);

}

/*******************************************************************************

*

*  Function Name   : impeg2d_init_video_state

*

*  Description     : Initializes the Video decoder state

*

*  Arguments       :

*  dec             : Decoder context

*  videoType       : MPEG_2_Video / MPEG_1_Video

*

*  Values Returned : None

*******************************************************************************/

IMPEG2D_ERROR_CODES_T impeg2d_init_video_state(dec_state_t *ps_dec, e_video_type_t e_video_type)

{

    /*-----------------------------------------------------------------------*/

    /* Bit Stream  that conforms to MPEG-1 <ISO/IEC 11172-2> standard        */

    /*-----------------------------------------------------------------------*/

    if(e_video_type == MPEG_1_VIDEO)

    {

        ps_dec->u2_is_mpeg2 = 0;

        /*-------------------------------------------------------------------*/

        /* force MPEG-1 parameters for proper decoder behavior               */

        /* see ISO/IEC 13818-2 section D.9.14                                */

        /*-------------------------------------------------------------------*/

        ps_dec->u2_progressive_sequence         = 1;

        ps_dec->u2_intra_dc_precision           = 0;

        ps_dec->u2_picture_structure            = FRAME_PICTURE;

        ps_dec->u2_frame_pred_frame_dct         = 1;

        ps_dec->u2_concealment_motion_vectors   = 0;

        ps_dec->u2_q_scale_type                 = 0;

        ps_dec->u2_intra_vlc_format             = 0;

        ps_dec->u2_alternate_scan               = 0;

        ps_dec->u2_repeat_first_field           = 0;

        ps_dec->u2_progressive_frame            = 1;

        ps_dec->u2_frame_rate_extension_n       = 0;

        ps_dec->u2_frame_rate_extension_d       = 0;

        ps_dec->u2_forw_f_code                  = 7;

        ps_dec->u2_back_f_code                  = 7;

        ps_dec->pf_vld_inv_quant                  = impeg2d_vld_inv_quant_mpeg1;

        /*-------------------------------------------------------------------*/

        /* Setting of parameters other than those mentioned in MPEG2 standard*/

        /* but used in decoding process.                                     */

        /*-------------------------------------------------------------------*/

    }

    /*-----------------------------------------------------------------------*/

    /* Bit Stream  that conforms to MPEG-2                                   */

    /*-----------------------------------------------------------------------*/

    else

    {

        ps_dec->u2_is_mpeg2                  = 1;

        ps_dec->u2_full_pel_forw_vector   = 0;

        ps_dec->u2_forw_f_code            = 7;

        ps_dec->u2_full_pel_back_vector   = 0;

        ps_dec->u2_back_f_code            = 7;

        ps_dec->pf_vld_inv_quant       = impeg2d_vld_inv_quant_mpeg2;

    }

    impeg2d_init_function_ptr(ps_dec);

    /* Set the frame Width and frame Height */

    ps_dec->u2_frame_height        = ALIGN16(ps_dec->u2_vertical_size);

    ps_dec->u2_frame_width         = ALIGN16(ps_dec->u2_horizontal_size);

    ps_dec->u2_num_horiz_mb         = (ps_dec->u2_horizontal_size + 15) >> 4;

   // dec->u4_frm_buf_stride    = dec->frameWidth;

    if (ps_dec->u2_frame_height > ps_dec->u2_create_max_height || ps_dec->u2_frame_width > ps_dec->u2_create_max_width)

    {

        return IMPEG2D_PIC_SIZE_NOT_SUPPORTED;

    }

    ps_dec->u2_num_flds_decoded = 0;

    /* Calculate the frame period */

    {

        UWORD32 numer;

        UWORD32 denom;

        numer = (UWORD32)gau2_impeg2_frm_rate_code[ps_dec->u2_frame_rate_code][1] *

                                (UWORD32)(ps_dec->u2_frame_rate_extension_d + 1);

        denom = (UWORD32)gau2_impeg2_frm_rate_code[ps_dec->u2_frame_rate_code][0] *

                                (UWORD32)(ps_dec->u2_frame_rate_extension_n + 1);

        ps_dec->u2_framePeriod =  (numer * 1000 * 100) / denom;

    }

   if(VERTICAL_SCAN == ps_dec->u2_alternate_scan)

   {

    ps_dec->pu1_inv_scan_matrix = (UWORD8 *)gau1_impeg2_inv_scan_vertical;

   }

   else

   {

    ps_dec->pu1_inv_scan_matrix = (UWORD8 *)gau1_impeg2_inv_scan_zig_zag;

   }

   return (IMPEG2D_ERROR_CODES_T)IVD_ERROR_NONE;

}

/*******************************************************************************

*

*  Function Name   : impeg2d_pre_pic_dec_proc

*

*  Description     : Does the processing neccessary before picture decoding

*

*  Arguments       :

*  dec             : Decoder context

*

*  Values Returned : None

*******************************************************************************/

IMPEG2D_ERROR_CODES_T impeg2d_pre_pic_dec_proc(dec_state_t *ps_dec)

{

    WORD32 u4_get_disp;

    pic_buf_t *ps_disp_pic;

    IMPEG2D_ERROR_CODES_T e_error = (IMPEG2D_ERROR_CODES_T)IVD_ERROR_NONE;

    u4_get_disp = 0;

    ps_disp_pic = NULL;

    /* Field Picture */

    if(ps_dec->u2_picture_structure != FRAME_PICTURE)

    {

        ps_dec->u2_num_vert_mb       = (ps_dec->u2_vertical_size + 31) >> 5;

        if(ps_dec->u2_num_flds_decoded == 0)

        {

            pic_buf_t *ps_pic_buf;

            u4_get_disp = 1;

            ps_pic_buf = impeg2_buf_mgr_get_next_free(ps_dec->pv_pic_buf_mg, &ps_dec->i4_cur_buf_id);

            if (NULL == ps_pic_buf)

            {

                return IMPEG2D_NO_FREE_BUF_ERR;

            }

            impeg2_buf_mgr_set_status((buf_mgr_t *)ps_dec->pv_pic_buf_mg, ps_dec->i4_cur_buf_id, BUF_MGR_DISP);

            impeg2_buf_mgr_set_status((buf_mgr_t *)ps_dec->pv_pic_buf_mg, ps_dec->i4_cur_buf_id, BUF_MGR_REF);

            if(ps_dec->u4_deinterlace)

                impeg2_buf_mgr_set_status((buf_mgr_t *)ps_dec->pv_pic_buf_mg, ps_dec->i4_cur_buf_id, MPEG2_BUF_MGR_DEINT);

            ps_pic_buf->e_pic_type = ps_dec->e_pic_type;

            ps_dec->ps_cur_pic = ps_pic_buf;

            ps_dec->s_cur_frm_buf.pu1_y = ps_pic_buf->pu1_y;

            ps_dec->s_cur_frm_buf.pu1_u = ps_pic_buf->pu1_u;

            ps_dec->s_cur_frm_buf.pu1_v = ps_pic_buf->pu1_v;

        }

        else

        {

            /* Timestamp associated with second field is associated with the current picture */

            ps_dec->ps_cur_pic->u4_ts = ps_dec->u4_inp_ts;

        }

        if(ps_dec->u2_picture_structure == TOP_FIELD)

        {

            ps_dec->u2_fld_parity = TOP;

        }

        else

        {

            ps_dec->u2_fld_parity = BOTTOM;

        }

        ps_dec->u2_field_dct           = 0;

        ps_dec->u2_read_dct_type        = 0;

        ps_dec->u2_read_motion_type     = 1;

        ps_dec->u2_fld_pic             = 1;

        ps_dec->u2_frm_pic             = 0;

        ps_dec->ps_func_forw_or_back     = gas_impeg2d_func_fld_fw_or_bk;

        ps_dec->ps_func_bi_direct       = gas_impeg2d_func_fld_bi_direct;

   }

    /* Frame Picture */

    else

    {

        pic_buf_t *ps_pic_buf;

        ps_dec->u2_num_vert_mb       = (ps_dec->u2_vertical_size + 15) >> 4;

        u4_get_disp = 1;

        ps_pic_buf = impeg2_buf_mgr_get_next_free(ps_dec->pv_pic_buf_mg, &ps_dec->i4_cur_buf_id);

        if (NULL == ps_pic_buf)

        {

            return IMPEG2D_NO_FREE_BUF_ERR;

        }

        impeg2_buf_mgr_set_status((buf_mgr_t *)ps_dec->pv_pic_buf_mg, ps_dec->i4_cur_buf_id, BUF_MGR_DISP);

        impeg2_buf_mgr_set_status((buf_mgr_t *)ps_dec->pv_pic_buf_mg, ps_dec->i4_cur_buf_id, BUF_MGR_REF);

        if(ps_dec->u4_deinterlace)

            impeg2_buf_mgr_set_status((buf_mgr_t *)ps_dec->pv_pic_buf_mg, ps_dec->i4_cur_buf_id, MPEG2_BUF_MGR_DEINT);

        ps_pic_buf->u4_ts = ps_dec->u4_inp_ts;

        ps_pic_buf->e_pic_type = ps_dec->e_pic_type;

        ps_dec->ps_cur_pic = ps_pic_buf;

        ps_dec->s_cur_frm_buf.pu1_y = ps_pic_buf->pu1_y;

        ps_dec->s_cur_frm_buf.pu1_u = ps_pic_buf->pu1_u;

        ps_dec->s_cur_frm_buf.pu1_v = ps_pic_buf->pu1_v;

        if(ps_dec->u2_frame_pred_frame_dct == 0)

        {

            ps_dec->u2_read_dct_type    = 1;

            ps_dec->u2_read_motion_type = 1;

        }

        else

        {

            ps_dec->u2_read_dct_type    = 0;

            ps_dec->u2_read_motion_type = 0;

            ps_dec->u2_motion_type     = 2;

            ps_dec->u2_field_dct       = 0;

        }

        ps_dec->u2_fld_parity          = TOP;

        ps_dec->u2_fld_pic             = 0;

        ps_dec->u2_frm_pic             = 1;

        ps_dec->ps_func_forw_or_back     = gas_impeg2d_func_frm_fw_or_bk;

        ps_dec->ps_func_bi_direct       = gas_impeg2d_func_frm_bi_direct;

   }

    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;

    ps_dec->u2_num_mbs_left  = ps_dec->u2_num_horiz_mb * ps_dec->u2_num_vert_mb;

    if(u4_get_disp)

    {

        if(ps_dec->u4_num_frames_decoded > 1)

        {

            ps_disp_pic = impeg2_disp_mgr_get(&ps_dec->s_disp_mgr, &ps_dec->i4_disp_buf_id);

        }

        ps_dec->ps_disp_pic = ps_disp_pic;

        if(ps_disp_pic)

        {

            if(1 == ps_dec->u4_share_disp_buf)

            {

                ps_dec->ps_disp_frm_buf->pv_y_buf  = ps_disp_pic->pu1_y;

                if(IV_YUV_420P == ps_dec->i4_chromaFormat)

                {

                    ps_dec->ps_disp_frm_buf->pv_u_buf  = ps_disp_pic->pu1_u;

                    ps_dec->ps_disp_frm_buf->pv_v_buf  = ps_disp_pic->pu1_v;

                }

                else

                {

                    UWORD8 *pu1_buf;

                    pu1_buf = ps_dec->as_disp_buffers[ps_disp_pic->i4_buf_id].pu1_bufs[1];

                    ps_dec->ps_disp_frm_buf->pv_u_buf  = pu1_buf;

                    pu1_buf = ps_dec->as_disp_buffers[ps_disp_pic->i4_buf_id].pu1_bufs[2];

                    ps_dec->ps_disp_frm_buf->pv_v_buf  = pu1_buf;

                }

            }

        }

    }

    switch(ps_dec->e_pic_type)

    {

    case I_PIC:

        {

            ps_dec->pf_decode_slice = impeg2d_dec_i_slice;

            break;

        }

    case D_PIC:

        {

            ps_dec->pf_decode_slice = impeg2d_dec_d_slice;

            break;

        }

    case P_PIC:

        {

            ps_dec->pf_decode_slice = impeg2d_dec_p_b_slice;

            ps_dec->pu2_mb_type       = gau2_impeg2d_p_mb_type;

            break;

        }

    case B_PIC:

        {

            ps_dec->pf_decode_slice = impeg2d_dec_p_b_slice;

            ps_dec->pu2_mb_type       = gau2_impeg2d_b_mb_type;

            break;

        }

    default:

        return IMPEG2D_INVALID_PIC_TYPE;

    }

    /*************************************************************************/

    /* Set the reference pictures                                            */

    /*************************************************************************/

    /* Error resilience: If forward and backward pictures are going to be NULL*/

    /* then assign both to the current                                        */

    /* if one of them NULL then we will assign the non null to the NULL one   */

    if(ps_dec->e_pic_type == P_PIC)

    {

        if (NULL == ps_dec->as_recent_fld[1][0].pu1_y)

        {

            ps_dec->as_recent_fld[1][0] = ps_dec->s_cur_frm_buf;

        }

        if (NULL == ps_dec->as_recent_fld[1][1].pu1_y)

        {

            impeg2d_get_bottom_field_buf(&ps_dec->s_cur_frm_buf, &ps_dec->as_recent_fld[1][1],

                ps_dec->u2_frame_width);

        }

        ps_dec->as_ref_buf[FORW][TOP]    = ps_dec->as_recent_fld[1][0];

        ps_dec->as_ref_buf[FORW][BOTTOM] = ps_dec->as_recent_fld[1][1];

    }

    else if(ps_dec->e_pic_type == B_PIC)

    {

        if((NULL == ps_dec->as_recent_fld[1][0].pu1_y) && (NULL == ps_dec->as_recent_fld[0][0].pu1_y))

        {

            // assign the current picture to both

            ps_dec->as_recent_fld[1][0] = ps_dec->s_cur_frm_buf;

            impeg2d_get_bottom_field_buf(&ps_dec->s_cur_frm_buf, &ps_dec->as_recent_fld[1][1],

                ps_dec->u2_frame_width);

            ps_dec->as_recent_fld[0][0] = ps_dec->s_cur_frm_buf;

            ps_dec->as_recent_fld[0][1] = ps_dec->as_recent_fld[1][1];

        }

        //Assign the non-null picture to the null picture

        else if ((NULL != ps_dec->as_recent_fld[1][0].pu1_y) && (NULL == ps_dec->as_recent_fld[0][0].pu1_y))

        {

            ps_dec->as_recent_fld[0][0] = ps_dec->as_recent_fld[1][0];

            ps_dec->as_recent_fld[0][1] = ps_dec->as_recent_fld[1][1];

        }

        else if ((NULL == ps_dec->as_recent_fld[1][0].pu1_y) && (NULL != ps_dec->as_recent_fld[0][0].pu1_y))

        {

            ps_dec->as_recent_fld[1][0] = ps_dec->as_recent_fld[0][0];

            ps_dec->as_recent_fld[1][1] = ps_dec->as_recent_fld[0][1];

        }

        /* Error resilience: If forward and backward pictures are going to be NULL*/

        /* then assign both to the current                                        */

        /* if one of them NULL then we will assign the non null to the NULL one   */

        if((NULL == ps_dec->as_recent_fld[0][1].pu1_y) && (NULL == ps_dec->as_recent_fld[1][1].pu1_y))

        {

            // assign the current picture to both

            ps_dec->as_recent_fld[1][0] = ps_dec->s_cur_frm_buf;

            impeg2d_get_bottom_field_buf(&ps_dec->s_cur_frm_buf, &ps_dec->as_recent_fld[1][1],

                                         ps_dec->u2_frame_width);

            ps_dec->as_recent_fld[0][0] = ps_dec->s_cur_frm_buf;

            ps_dec->as_recent_fld[0][1] = ps_dec->as_recent_fld[1][1];

        }

        //Assign the non-null picture to the null picture

        else if((NULL == ps_dec->as_recent_fld[0][1].pu1_y) && (NULL != ps_dec->as_recent_fld[1][1].pu1_y))

        {

            ps_dec->as_recent_fld[0][0] = ps_dec->as_recent_fld[1][0];

            ps_dec->as_recent_fld[0][1] = ps_dec->as_recent_fld[1][1];

        }

        else if((NULL == ps_dec->as_recent_fld[1][1].pu1_y) && (NULL != ps_dec->as_recent_fld[0][1].pu1_y))

        {

            ps_dec->as_recent_fld[1][0] = ps_dec->as_recent_fld[0][0];

            ps_dec->as_recent_fld[1][1] = ps_dec->as_recent_fld[0][1];

        }

        ps_dec->as_ref_buf[FORW][TOP]    = ps_dec->as_recent_fld[0][0];

        ps_dec->as_ref_buf[FORW][BOTTOM] = ps_dec->as_recent_fld[0][1];

        ps_dec->as_ref_buf[BACK][TOP]    = ps_dec->as_recent_fld[1][0];

        ps_dec->as_ref_buf[BACK][BOTTOM] = ps_dec->as_recent_fld[1][1];

    }

    return e_error;

}

/*******************************************************************************

*

*  Function Name   : impeg2d_post_pic_dec_proc

*

*  Description     : Performs processing that is needed at the end of picture

*                    decode

*

*  Arguments       :

*  dec             : Decoder context

*

*  Values Returned : None

*******************************************************************************/

void impeg2d_post_pic_dec_proc(dec_state_t *ps_dec)

{

   WORD32 u4_update_pic_buf = 0;

    /*************************************************************************/

    /* Processing at the end of picture                                      */

    /*************************************************************************/

    if(ps_dec->u2_picture_structure != FRAME_PICTURE)

    {

        ps_dec->u2_num_vert_mb       = (ps_dec->u2_vertical_size + 31) >> 5;

        if(ps_dec->u2_num_flds_decoded == 1)

        {

            ps_dec->u2_num_flds_decoded = 0;

            u4_update_pic_buf = 1;

        }

        else

        {

            ps_dec->u2_num_flds_decoded = 1;

        }

    }

    else

    {

        u4_update_pic_buf = 1;

    }

    if(u4_update_pic_buf)

    {

        ps_dec->i4_frame_decoded = 1;

        if(ps_dec->e_pic_type != B_PIC)

        {

            /* In any sequence first two pictures have to be reference pictures */

            /* Adding of first picture in the sequence */

            if(ps_dec->aps_ref_pics[0] == NULL)

            {

                ps_dec->aps_ref_pics[0] = ps_dec->ps_cur_pic;

            }

            /* Adding of second picture in the sequence */

            else if(ps_dec->aps_ref_pics[1] == NULL)

            {

                ps_dec->aps_ref_pics[1] = ps_dec->ps_cur_pic;

                impeg2_disp_mgr_add(&ps_dec->s_disp_mgr, ps_dec->aps_ref_pics[0], ps_dec->aps_ref_pics[0]->i4_buf_id);

            }

            else

            {

                impeg2_disp_mgr_add(&ps_dec->s_disp_mgr, ps_dec->aps_ref_pics[1], ps_dec->aps_ref_pics[1]->i4_buf_id);

                impeg2_buf_mgr_release(ps_dec->pv_pic_buf_mg, ps_dec->aps_ref_pics[0]->i4_buf_id, BUF_MGR_REF);

                ps_dec->aps_ref_pics[0] = ps_dec->aps_ref_pics[1];

                ps_dec->aps_ref_pics[1] = ps_dec->ps_cur_pic;

            }

        }

        else

        {

            impeg2_disp_mgr_add(&ps_dec->s_disp_mgr, ps_dec->ps_cur_pic, ps_dec->ps_cur_pic->i4_buf_id);

            impeg2_buf_mgr_release(ps_dec->pv_pic_buf_mg, ps_dec->ps_cur_pic->i4_buf_id, BUF_MGR_REF);

        }

    }

    /*************************************************************************/

    /* Update the list of recent reference pictures                          */

    /*************************************************************************/

    if(ps_dec->e_pic_type != B_PIC)

    {

        switch(ps_dec->u2_picture_structure)

        {

        case FRAME_PICTURE:

            {

                ps_dec->as_recent_fld[0][0] = ps_dec->as_recent_fld[1][0];

                ps_dec->as_recent_fld[0][1] = ps_dec->as_recent_fld[1][1];

                ps_dec->as_recent_fld[1][0] = ps_dec->s_cur_frm_buf;

                impeg2d_get_bottom_field_buf(&ps_dec->s_cur_frm_buf, &ps_dec->as_recent_fld[1][1],

                ps_dec->u2_frame_width);

                break;

            }

        case TOP_FIELD:

            {

                ps_dec->as_recent_fld[0][0] = ps_dec->as_recent_fld[1][0];

                ps_dec->as_recent_fld[1][0] = ps_dec->s_cur_frm_buf;

                break;

            }

        case BOTTOM_FIELD:

            {

                ps_dec->as_recent_fld[0][1] = ps_dec->as_recent_fld[1][1];

                impeg2d_get_bottom_field_buf(&ps_dec->s_cur_frm_buf, &ps_dec->as_recent_fld[1][1],

                ps_dec->u2_frame_width);

                break;

            }

        }

    }

}