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

 *

 * Copyright (C) 2015 The Android Open Source Project

 *

 * Licensed under the Apache License, Version 2.0 (the "License");

 * you may not use this file except in compliance with the License.

 * You may obtain a copy of the License at:

 *

 * http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 *

 *****************************************************************************

 * Originally developed and contributed by Ittiam Systems Pvt. Ltd, Bangalore

*/

/*!

 **************************************************************************

 * \file ih264d_parse_slice.c

 *

 * \brief

 *    Contains routines that decodes a slice NAL unit

 *

 * \date

 *    19/12/2002

 *

 * \author  AI

 **************************************************************************

 */

#include <string.h>

#include "ih264_typedefs.h"

#include "ih264_macros.h"

#include "ih264_platform_macros.h"

#include "ithread.h"

#include "ih264d_structs.h"

#include "ih264d_debug.h"

#include "ih264d_bitstrm.h"

#include "ih264d_parse_mb_header.h"

#include "ih264d_process_bslice.h"

#include "ih264d_process_pslice.h"

#include "ih264d_parse_cavlc.h"

#include "ih264d_utils.h"

#include "ih264d_deblocking.h"

#include "ih264d_defs.h"

#include "ih264d_error_handler.h"

#include "ih264d_tables.h"

#include "ih264d_defs.h"

#include "ih264d_mem_request.h"

#include "ih264d_parse_islice.h"

#include "ih264d_parse_slice.h"

#include "ih264d_mvpred.h"

#include "ih264d_mb_utils.h"

#include "ih264d_defs.h"

#include "ih264d_quant_scaling.h"

#include "ih264d_inter_pred.h"

#include "ih264d_sei.h"

#include "ih264d.h"

#include "ih264_error.h"

#include "ih264_disp_mgr.h"

#include "ih264_buf_mgr.h"

#include "ih264d_thread_parse_decode.h"

#include "ih264d_thread_compute_bs.h"

#include "ih264d_dpb_manager.h"

#include <assert.h>

#include "ih264d_parse_islice.h"

#define RET_LAST_SKIP  0x80000000

WORD32 check_app_out_buf_size(dec_struct_t *ps_dec);

/*!

 **************************************************************************

 * \if Function name : ih264d_form_pred_weight_matrix \endif

 *

 * \brief

 *    Forms pred weight matrix.

 *

 * \return

 *    None

 *

 **************************************************************************

 */

void ih264d_form_pred_weight_matrix(dec_struct_t *ps_dec)

{

    dec_slice_params_t *ps_cur_slice;

    UWORD8 uc_num_ref_idx_l0_active, uc_num_ref_idx_l1_active;

    UWORD8 i, j;

    UWORD32 *pu4_mat_iwt_ofst;

    UWORD16 i2_idx;

    UWORD32 *pui32_weight_offset_l0, *pui32_weight_offset_l1;

    UWORD32 u4_temp;

    ps_cur_slice = ps_dec->ps_cur_slice;

    uc_num_ref_idx_l0_active = ps_cur_slice->u1_num_ref_idx_lx_active[0];

    uc_num_ref_idx_l1_active = ps_cur_slice->u1_num_ref_idx_lx_active[1];

    pu4_mat_iwt_ofst = ps_dec->pu4_wts_ofsts_mat;

    if(ps_cur_slice->u1_slice_type == B_SLICE)

    {

        for(i = 0; i < uc_num_ref_idx_l0_active; i++)

        {

            pui32_weight_offset_l0 = ps_cur_slice->u4_wt_ofst_lx[0][i];

            for(j = 0; j < uc_num_ref_idx_l1_active; j++)

            {

                pui32_weight_offset_l1 = ps_cur_slice->u4_wt_ofst_lx[1][j];

                i2_idx = i * uc_num_ref_idx_l0_active + j;

                i2_idx = X3(i2_idx);

                /*        u4_temp = (pui32_weight_offset_l0[0] | (pui32_weight_offset_l1[0] << 16));

                 pu4_mat_iwt_ofst[0] = u4_temp;

                 u4_temp = (pui32_weight_offset_l0[1] | (pui32_weight_offset_l1[1] << 16));

                 pu4_mat_iwt_ofst[1] = u4_temp;

                 u4_temp = (pui32_weight_offset_l0[2] | (pui32_weight_offset_l1[2] << 16));

                 pu4_mat_iwt_ofst[2] = u4_temp;

                 pu4_mat_iwt_ofst += 3;*/

                pu4_mat_iwt_ofst[0] = pui32_weight_offset_l0[0];

                pu4_mat_iwt_ofst[1] = pui32_weight_offset_l1[0];

                pu4_mat_iwt_ofst[2] = pui32_weight_offset_l0[1];

                pu4_mat_iwt_ofst[3] = pui32_weight_offset_l1[1];

                pu4_mat_iwt_ofst[4] = pui32_weight_offset_l0[2];

                pu4_mat_iwt_ofst[5] = pui32_weight_offset_l1[2];

                pu4_mat_iwt_ofst += 6;

            }

        }

    }

    else

    {

        for(i = 0; i < uc_num_ref_idx_l0_active; i++)

        {

            pui32_weight_offset_l0 = ps_cur_slice->u4_wt_ofst_lx[0][i];

            i2_idx = X3(i);

            u4_temp = (UWORD32)pui32_weight_offset_l0[0];

            pu4_mat_iwt_ofst[0] = u4_temp;

            u4_temp = (UWORD32)pui32_weight_offset_l0[1];

            pu4_mat_iwt_ofst[2] = u4_temp;

            u4_temp = (UWORD32)pui32_weight_offset_l0[2];

            pu4_mat_iwt_ofst[4] = u4_temp;

            pu4_mat_iwt_ofst += 6;

        }

    }

}

/*!

 **************************************************************************

 * \if Function name :  init_firstSliceParam \endif

 *

 * \brief

 *    Initialize the Parameter required for all the slices for a picture

 *

 * \return           : Nothing

 *

 **************************************************************************

 */

WORD32 ih264d_start_of_pic(dec_struct_t *ps_dec,

                         WORD32 i4_poc,

                         pocstruct_t *ps_temp_poc,

                         UWORD16 u2_frame_num,

                         dec_pic_params_t *ps_pps)

{

    pocstruct_t *ps_prev_poc = &ps_dec->s_cur_pic_poc;

    pocstruct_t *ps_cur_poc = ps_temp_poc;

    pic_buffer_t *pic_buf;

    ivd_video_decode_op_t * ps_dec_output =

                    (ivd_video_decode_op_t *)ps_dec->pv_dec_out;

    dec_slice_params_t *ps_cur_slice = ps_dec->ps_cur_slice;

    dec_seq_params_t *ps_seq = ps_pps->ps_sps;

    UWORD8 u1_bottom_field_flag = ps_cur_slice->u1_bottom_field_flag;

    UWORD8 u1_field_pic_flag = ps_cur_slice->u1_field_pic_flag;

    /* high profile related declarations */

    high_profile_tools_t s_high_profile;

    WORD32 ret;

    H264_MUTEX_LOCK(&ps_dec->process_disp_mutex);

    /* check output buffer size given by the application */

    if(check_app_out_buf_size(ps_dec) != IV_SUCCESS)

        return IVD_DISP_FRM_ZERO_OP_BUF_SIZE;

    ps_prev_poc->i4_pic_order_cnt_lsb = ps_cur_poc->i4_pic_order_cnt_lsb;

    ps_prev_poc->i4_pic_order_cnt_msb = ps_cur_poc->i4_pic_order_cnt_msb;

    ps_prev_poc->i4_delta_pic_order_cnt_bottom =

                    ps_cur_poc->i4_delta_pic_order_cnt_bottom;

    ps_prev_poc->i4_delta_pic_order_cnt[0] =

                    ps_cur_poc->i4_delta_pic_order_cnt[0];

    ps_prev_poc->i4_delta_pic_order_cnt[1] =

                    ps_cur_poc->i4_delta_pic_order_cnt[1];

    ps_prev_poc->u1_bot_field = ps_dec->ps_cur_slice->u1_bottom_field_flag;

    ps_prev_poc->i4_prev_frame_num_ofst = ps_cur_poc->i4_prev_frame_num_ofst;

    ps_prev_poc->u2_frame_num = u2_frame_num;

    ps_dec->i1_prev_mb_qp_delta = 0;

    ps_dec->i1_next_ctxt_idx = 0;

    ps_dec->u4_nmb_deblk = 0;

    if(ps_dec->u4_num_cores == 1)

       ps_dec->u4_nmb_deblk = 1;

    if(ps_seq->u1_mb_aff_flag == 1)

    {

        ps_dec->u4_nmb_deblk = 0;

        if(ps_dec->u4_num_cores > 2)

            ps_dec->u4_num_cores = 2;

    }

        ps_dec->u4_use_intrapred_line_copy = 0;

    if (ps_seq->u1_mb_aff_flag == 0)

    {

        ps_dec->u4_use_intrapred_line_copy = 1;

    }

    ps_dec->u4_app_disable_deblk_frm = 0;

    /* If degrade is enabled, set the degrade flags appropriately */

    if(ps_dec->i4_degrade_type && ps_dec->i4_degrade_pics)

    {

        WORD32 degrade_pic;

        ps_dec->i4_degrade_pic_cnt++;

        degrade_pic = 0;

        /* If degrade is to be done in all frames, then do not check further */

        switch(ps_dec->i4_degrade_pics)

        {

            case 4:

            {

                degrade_pic = 1;

                break;

            }

            case 3:

            {

                if(ps_cur_slice->u1_slice_type != I_SLICE)

                    degrade_pic = 1;

                break;

            }

            case 2:

            {

                /* If pic count hits non-degrade interval or it is an islice, then do not degrade */

                if((ps_cur_slice->u1_slice_type != I_SLICE)

                                && (ps_dec->i4_degrade_pic_cnt

                                                != ps_dec->i4_nondegrade_interval))

                    degrade_pic = 1;

                break;

            }

            case 1:

            {

                /* Check if the current picture is non-ref */

                if(0 == ps_cur_slice->u1_nal_ref_idc)

                {

                    degrade_pic = 1;

                }

                break;

            }

        }

        if(degrade_pic)

        {

            if(ps_dec->i4_degrade_type & 0x2)

                ps_dec->u4_app_disable_deblk_frm = 1;

            /* MC degrading is done only for non-ref pictures */

            if(0 == ps_cur_slice->u1_nal_ref_idc)

            {

                if(ps_dec->i4_degrade_type & 0x4)

                    ps_dec->i4_mv_frac_mask = 0;

                if(ps_dec->i4_degrade_type & 0x8)

                    ps_dec->i4_mv_frac_mask = 0;

            }

        }

        else

            ps_dec->i4_degrade_pic_cnt = 0;

    }

    {

        dec_err_status_t * ps_err = ps_dec->ps_dec_err_status;

        if((ps_cur_slice->u1_slice_type == I_SLICE)

                        || (ps_cur_slice->u1_slice_type == SI_SLICE))

            ps_err->u1_cur_pic_type = PIC_TYPE_I;

        else

            ps_err->u1_cur_pic_type = PIC_TYPE_UNKNOWN;

        if(ps_err->u1_pic_aud_i == PIC_TYPE_I)

        {

            ps_err->u1_cur_pic_type = PIC_TYPE_I;

            ps_err->u1_pic_aud_i = PIC_TYPE_UNKNOWN;

        }

        if(ps_cur_slice->u1_nal_unit_type == IDR_SLICE_NAL)

        {

            if(ps_err->u1_err_flag)

                ih264d_reset_ref_bufs(ps_dec->ps_dpb_mgr);

            ps_err->u1_err_flag = ACCEPT_ALL_PICS;

        }

    }

    if(ps_dec->u1_init_dec_flag && ps_dec->s_prev_seq_params.u1_eoseq_pending)

    {

        /* Reset the decoder picture buffers */

        WORD32 j;

        for(j = 0; j < MAX_DISP_BUFS_NEW; j++)

        {

            ih264_buf_mgr_release((buf_mgr_t *)ps_dec->pv_pic_buf_mgr,

                                  j,

                                  BUF_MGR_REF);

            ih264_buf_mgr_release((buf_mgr_t *)ps_dec->pv_mv_buf_mgr,

                                  ps_dec->as_buf_id_info_map[j].mv_buf_id,

                                  BUF_MGR_REF);

            ih264_buf_mgr_release((buf_mgr_t *)ps_dec->pv_pic_buf_mgr,

                                  j,

                                  BUF_MGR_IO);

        }

        /* reset the decoder structure parameters related to buffer handling */

        ps_dec->u1_second_field = 0;

        ps_dec->i4_cur_display_seq = 0;

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

        /* indicate in the decoder output i4_status that some frames are being */

        /* dropped, so that it resets timestamp and wait for a new sequence */

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

        ps_dec->s_prev_seq_params.u1_eoseq_pending = 0;

    }

    ret = ih264d_init_pic(ps_dec, u2_frame_num, i4_poc, ps_pps);

    if(ret != OK)

        return ret;

    ps_dec->pv_parse_tu_coeff_data = ps_dec->pv_pic_tu_coeff_data;

    ps_dec->pv_proc_tu_coeff_data  = ps_dec->pv_pic_tu_coeff_data;

    ps_dec->ps_nmb_info = ps_dec->ps_frm_mb_info;

    if(ps_dec->u1_separate_parse)

    {

        UWORD16 pic_wd;

        UWORD16 pic_ht;

        UWORD32 num_mbs;

        pic_wd = ps_dec->u2_pic_wd;

        pic_ht = ps_dec->u2_pic_ht;

        num_mbs = (pic_wd * pic_ht) >> 8;

        if(ps_dec->pu1_dec_mb_map)

        {

            memset((void *)ps_dec->pu1_dec_mb_map, 0, num_mbs);

        }

        if(ps_dec->pu1_recon_mb_map)

        {

            memset((void *)ps_dec->pu1_recon_mb_map, 0, num_mbs);

        }

        if(ps_dec->pu2_slice_num_map)

        {

            memset((void *)ps_dec->pu2_slice_num_map, 0,

                   (num_mbs * sizeof(UWORD16)));

        }

    }

    ps_dec->ps_parse_cur_slice = &(ps_dec->ps_dec_slice_buf[0]);

    ps_dec->ps_decode_cur_slice = &(ps_dec->ps_dec_slice_buf[0]);

    ps_dec->ps_computebs_cur_slice = &(ps_dec->ps_dec_slice_buf[0]);

    ps_dec->u2_cur_slice_num = 0;

    /* Initialize all the HP toolsets to zero */

    ps_dec->s_high_profile.u1_scaling_present = 0;

    ps_dec->s_high_profile.u1_transform8x8_present = 0;

    /* Get Next Free Picture */

    if(1 == ps_dec->u4_share_disp_buf)

    {

        UWORD32 i;

        /* Free any buffer that is in the queue to be freed */

        for(i = 0; i < MAX_DISP_BUFS_NEW; i++)

        {

            if(0 == ps_dec->u4_disp_buf_to_be_freed[i])

                continue;

            ih264_buf_mgr_release((buf_mgr_t *)ps_dec->pv_pic_buf_mgr, i,

            BUF_MGR_IO);

            ps_dec->u4_disp_buf_to_be_freed[i] = 0;

            ps_dec->u4_disp_buf_mapping[i] = 0;

        }

    }

    if(!(u1_field_pic_flag && 0 != ps_dec->u1_top_bottom_decoded)) //ps_dec->u1_second_field))

    {

        pic_buffer_t *ps_cur_pic;

        WORD32 cur_pic_buf_id, cur_mv_buf_id;

        col_mv_buf_t *ps_col_mv;

        while(1)

        {

            ps_cur_pic = (pic_buffer_t *)ih264_buf_mgr_get_next_free(

                            (buf_mgr_t *)ps_dec->pv_pic_buf_mgr,

                            &cur_pic_buf_id);

            /* In case of IDR slices, if there is no free picture buffer, then release

             * all buffers from display and reference

             */

            if((ps_cur_pic == NULL) && (ps_cur_slice->u1_nal_unit_type == IDR_SLICE_NAL))

            {

                WORD32 j;

                for(j = 0; j < MAX_DISP_BUFS_NEW; j++)

                {

                    ih264_buf_mgr_release((buf_mgr_t *)ps_dec->pv_pic_buf_mgr,

                                          j,

                                          BUF_MGR_REF);

                    ih264_buf_mgr_release((buf_mgr_t *)ps_dec->pv_mv_buf_mgr,

                                          ps_dec->as_buf_id_info_map[j].mv_buf_id,

                                          BUF_MGR_REF);

                    ih264_buf_mgr_release((buf_mgr_t *)ps_dec->pv_pic_buf_mgr,

                                          j,

                                          BUF_MGR_IO);

                }

                ps_cur_pic = (pic_buffer_t *)ih264_buf_mgr_get_next_free(

                                (buf_mgr_t *)ps_dec->pv_pic_buf_mgr,

                                &cur_pic_buf_id);

            }

            if(ps_cur_pic == NULL)

            {

                ps_dec->i4_error_code = ERROR_UNAVAIL_PICBUF_T;

                ps_dec->ps_dec_err_status->u1_err_flag |= REJECT_CUR_PIC;

                return ERROR_UNAVAIL_PICBUF_T;

            }

            if(0 == ps_dec->u4_disp_buf_mapping[cur_pic_buf_id])

            {

                break;

            }

        }

        ps_col_mv = (col_mv_buf_t *)ih264_buf_mgr_get_next_free((buf_mgr_t *)ps_dec->pv_mv_buf_mgr,

                                                               &cur_mv_buf_id);

        if(ps_col_mv == NULL)

        {

            ps_dec->i4_error_code = ERROR_UNAVAIL_MVBUF_T;

            ps_dec->ps_dec_err_status->u1_err_flag |= REJECT_CUR_PIC;

            return ERROR_UNAVAIL_MVBUF_T;

        }

        ps_dec->ps_cur_pic = ps_cur_pic;

        ps_dec->u1_pic_buf_id = cur_pic_buf_id;

        ps_cur_pic->u4_ts = ps_dec->u4_ts;

        /* Update POC and inter_pic_id in sei structure,

         * later can be used by application for grain synthesis(SMPTE-RDD5) */

        if(ps_dec->ps_sei->u1_sei_fgc_params_present_flag)

        {

            ps_dec->ps_sei->s_sei_fgc_params.i4_poc = i4_poc;

            ps_dec->ps_sei->s_sei_fgc_params.u4_idr_pic_id = ps_cur_slice->u4_idr_pic_id;

        }

        memcpy(&ps_cur_pic->s_sei_pic, ps_dec->ps_sei, sizeof(sei));

        ps_cur_pic->u1_mv_buf_id = cur_mv_buf_id;

        ps_dec->as_buf_id_info_map[cur_pic_buf_id].mv_buf_id = cur_mv_buf_id;

        if(ps_dec->u1_enable_mb_info)

        {

            UWORD32 mb_info_map_size = ps_dec->u4_total_mbs << 2;

            ps_dec->as_buf_id_info_map[cur_pic_buf_id].pu1_qp_map

                = ps_dec->pu1_qp_map_base + cur_pic_buf_id * mb_info_map_size;

            ps_dec->as_buf_id_info_map[cur_pic_buf_id].pu1_mb_type_map

                = ps_dec->pu1_mb_type_map_base + cur_pic_buf_id * mb_info_map_size;

            memset(ps_dec->as_buf_id_info_map[cur_pic_buf_id].pu1_qp_map, 0, mb_info_map_size);

            memset(ps_dec->as_buf_id_info_map[cur_pic_buf_id].pu1_mb_type_map, 0, mb_info_map_size);

        }

        ps_cur_pic->pu1_col_zero_flag = (UWORD8 *)ps_col_mv->pv_col_zero_flag;

        ps_cur_pic->ps_mv = (mv_pred_t *)ps_col_mv->pv_mv;

        ps_dec->au1_pic_buf_ref_flag[cur_pic_buf_id] = 0;

        {

            /*make first entry of list0 and list1 point to cur pic,

             *so that if first slice is in error, ref pic struct will have valid entries*/

            ps_dec->ps_ref_pic_buf_lx[0] = ps_dec->ps_dpb_mgr->ps_init_dpb[0];

            ps_dec->ps_ref_pic_buf_lx[1] = ps_dec->ps_dpb_mgr->ps_init_dpb[1];

            *(ps_dec->ps_dpb_mgr->ps_init_dpb[0][0]) = *ps_cur_pic;

            /* Initialize for field reference as well */

            *(ps_dec->ps_dpb_mgr->ps_init_dpb[0][MAX_REF_BUFS]) = *ps_cur_pic;

            *(ps_dec->ps_dpb_mgr->ps_mod_dpb[0][0]) = *ps_cur_pic;

            /* Initialize for field reference as well */

            *(ps_dec->ps_dpb_mgr->ps_mod_dpb[0][MAX_REF_BUFS]) = *ps_cur_pic;

            *(ps_dec->ps_dpb_mgr->ps_init_dpb[1][0]) = *ps_cur_pic;

            /* Initialize for field reference as well */

            *(ps_dec->ps_dpb_mgr->ps_init_dpb[1][MAX_REF_BUFS]) = *ps_cur_pic;

            *(ps_dec->ps_dpb_mgr->ps_mod_dpb[1][0]) = *ps_cur_pic;

            /* Initialize for field reference as well */

            *(ps_dec->ps_dpb_mgr->ps_mod_dpb[1][MAX_REF_BUFS]) = *ps_cur_pic;

        }

        ps_dec->ps_cur_pic->u1_picturetype = u1_field_pic_flag;

        ps_dec->ps_cur_pic->u4_pack_slc_typ = SKIP_NONE;

        H264_DEC_DEBUG_PRINT("got a buffer\n");

    }

    else

    {

        H264_DEC_DEBUG_PRINT("did not get a buffer\n");

    }

    ps_dec->u4_pic_buf_got = 1;

    ps_dec->ps_cur_pic->i4_poc = i4_poc;

    ps_dec->ps_cur_pic->i4_frame_num = u2_frame_num;

    ps_dec->ps_cur_pic->i4_pic_num = u2_frame_num;

    ps_dec->ps_cur_pic->i4_top_field_order_cnt = ps_pps->i4_top_field_order_cnt;

    ps_dec->ps_cur_pic->i4_bottom_field_order_cnt =

                    ps_pps->i4_bottom_field_order_cnt;

    ps_dec->ps_cur_pic->i4_avg_poc = ps_pps->i4_avg_poc;

    ps_dec->ps_cur_pic->u4_time_stamp = ps_dec->u4_pts;

    ps_dec->s_cur_pic = *(ps_dec->ps_cur_pic);

    if(u1_field_pic_flag && u1_bottom_field_flag)

    {

        WORD32 i4_temp_poc;

        WORD32 i4_top_field_order_poc, i4_bot_field_order_poc;

        /* Point to odd lines, since it's bottom field */

        ps_dec->s_cur_pic.pu1_buf1 += ps_dec->s_cur_pic.u2_frm_wd_y;

        ps_dec->s_cur_pic.pu1_buf2 += ps_dec->s_cur_pic.u2_frm_wd_uv;

        ps_dec->s_cur_pic.pu1_buf3 += ps_dec->s_cur_pic.u2_frm_wd_uv;

        ps_dec->s_cur_pic.ps_mv +=

                        ((ps_dec->u2_pic_ht * ps_dec->u2_pic_wd) >> 5);

        ps_dec->s_cur_pic.pu1_col_zero_flag += ((ps_dec->u2_pic_ht

                        * ps_dec->u2_pic_wd) >> 5);

        ps_dec->ps_cur_pic->u1_picturetype |= BOT_FLD;

        i4_top_field_order_poc = ps_dec->ps_cur_pic->i4_top_field_order_cnt;

        i4_bot_field_order_poc = ps_dec->ps_cur_pic->i4_bottom_field_order_cnt;

        i4_temp_poc = MIN(i4_top_field_order_poc,

                                 i4_bot_field_order_poc);

        ps_dec->ps_cur_pic->i4_avg_poc = i4_temp_poc;

    }

    ps_cur_slice->u1_mbaff_frame_flag = ps_seq->u1_mb_aff_flag

                    && (!u1_field_pic_flag);

    ps_dec->ps_cur_pic->u1_picturetype |= (ps_cur_slice->u1_mbaff_frame_flag

                    << 2);

    ps_dec->ps_cur_mb_row = ps_dec->ps_nbr_mb_row; //[0];

    //Increment by 2 ,so that left mb (mbaff decrements by 2)  will always be valid

    ps_dec->ps_cur_mb_row += 2;

    ps_dec->ps_top_mb_row = ps_dec->ps_nbr_mb_row;

    ps_dec->ps_top_mb_row += ((ps_dec->u2_frm_wd_in_mbs + 2) << (1 - ps_dec->ps_cur_sps->u1_frame_mbs_only_flag));

    //Increment by 2 ,so that left mb (mbaff decrements by 2)  will always be valid

    ps_dec->ps_top_mb_row += 2;

    /* CHANGED CODE */

    ps_dec->ps_mv_cur = ps_dec->s_cur_pic.ps_mv;

    ps_dec->ps_mv_top = ps_dec->ps_mv_top_p[0];

    /* CHANGED CODE */

    ps_dec->u1_mv_top_p = 0;

    ps_dec->u4_mb_idx = 0;

    /* CHANGED CODE */

    ps_dec->ps_mv_left = ps_dec->s_cur_pic.ps_mv;

    ps_dec->u4_total_mbs_coded = 0;

    ps_dec->i4_submb_ofst = -(SUB_BLK_SIZE);

    ps_dec->u4_pred_info_idx = 0;

    ps_dec->u4_pred_info_pkd_idx = 0;

    ps_dec->u4_dma_buf_idx = 0;

    ps_dec->ps_mv = ps_dec->s_cur_pic.ps_mv;

    ps_dec->ps_mv_bank_cur = ps_dec->s_cur_pic.ps_mv;

    ps_dec->pu1_col_zero_flag = ps_dec->s_cur_pic.pu1_col_zero_flag;

    ps_dec->ps_part = ps_dec->ps_parse_part_params;

    ps_dec->i2_prev_slice_mbx = -1;

    ps_dec->i2_prev_slice_mby = 0;

    ps_dec->u2_mv_2mb[0] = 0;

    ps_dec->u2_mv_2mb[1] = 0;

    ps_dec->u1_last_pic_not_decoded = 0;

    ps_dec->u2_cur_slice_num_dec_thread = 0;

    ps_dec->u2_cur_slice_num_bs = 0;

    ps_dec->u4_intra_pred_line_ofst = 0;

    ps_dec->pu1_cur_y_intra_pred_line = ps_dec->pu1_y_intra_pred_line;

    ps_dec->pu1_cur_u_intra_pred_line = ps_dec->pu1_u_intra_pred_line;

    ps_dec->pu1_cur_v_intra_pred_line = ps_dec->pu1_v_intra_pred_line;

    ps_dec->pu1_cur_y_intra_pred_line_base = ps_dec->pu1_y_intra_pred_line;

    ps_dec->pu1_cur_u_intra_pred_line_base = ps_dec->pu1_u_intra_pred_line;

    ps_dec->pu1_cur_v_intra_pred_line_base = ps_dec->pu1_v_intra_pred_line;

    ps_dec->pu1_prev_y_intra_pred_line = ps_dec->pu1_y_intra_pred_line

                    + (ps_dec->u2_frm_wd_in_mbs * MB_SIZE);

    ps_dec->pu1_prev_u_intra_pred_line = ps_dec->pu1_u_intra_pred_line

                    + ps_dec->u2_frm_wd_in_mbs * BLK8x8SIZE * YUV420SP_FACTOR;

    ps_dec->pu1_prev_v_intra_pred_line = ps_dec->pu1_v_intra_pred_line

                    + ps_dec->u2_frm_wd_in_mbs * BLK8x8SIZE;

    ps_dec->ps_deblk_mbn = ps_dec->ps_deblk_pic;

    /* Initialize The Function Pointer Depending Upon the Entropy and MbAff Flag */

    {

        if(ps_cur_slice->u1_mbaff_frame_flag)

        {

            ps_dec->pf_compute_bs = ih264d_compute_bs_mbaff;

            ps_dec->pf_mvpred = ih264d_mvpred_mbaff;

        }

        else

        {

            ps_dec->pf_compute_bs = ih264d_compute_bs_non_mbaff;

            ps_dec->u1_cur_mb_fld_dec_flag = ps_cur_slice->u1_field_pic_flag;

        }

    }

    /* Set up the Parameter for DMA transfer */

    {

        UWORD8 u1_field_pic_flag = ps_dec->ps_cur_slice->u1_field_pic_flag;

        UWORD8 u1_mbaff = ps_cur_slice->u1_mbaff_frame_flag;

        UWORD16 uc_lastmbs = (((ps_dec->u2_pic_wd) >> 4)

                        % (ps_dec->u4_recon_mb_grp >> u1_mbaff));

        UWORD16 ui16_lastmbs_widthY =

                        (uc_lastmbs ? (uc_lastmbs << 4) : ((ps_dec->u4_recon_mb_grp

                                        >> u1_mbaff) << 4));

        UWORD16 ui16_lastmbs_widthUV =

                        uc_lastmbs ? (uc_lastmbs << 3) : ((ps_dec->u4_recon_mb_grp

                                        >> u1_mbaff) << 3);

        ps_dec->s_tran_addrecon.pu1_dest_y = ps_dec->s_cur_pic.pu1_buf1;

        ps_dec->s_tran_addrecon.pu1_dest_u = ps_dec->s_cur_pic.pu1_buf2;

        ps_dec->s_tran_addrecon.pu1_dest_v = ps_dec->s_cur_pic.pu1_buf3;

        ps_dec->s_tran_addrecon.u2_frm_wd_y = ps_dec->u2_frm_wd_y

                        << u1_field_pic_flag;

        ps_dec->s_tran_addrecon.u2_frm_wd_uv = ps_dec->u2_frm_wd_uv

                        << u1_field_pic_flag;

        if(u1_field_pic_flag)

        {

            ui16_lastmbs_widthY += ps_dec->u2_frm_wd_y;

            ui16_lastmbs_widthUV += ps_dec->u2_frm_wd_uv;

        }

        /* Normal Increment of Pointer */

        ps_dec->s_tran_addrecon.u4_inc_y[0] = ((ps_dec->u4_recon_mb_grp << 4)

                        >> u1_mbaff);

        ps_dec->s_tran_addrecon.u4_inc_uv[0] = ((ps_dec->u4_recon_mb_grp << 4)

                        >> u1_mbaff);

        /* End of Row Increment */

        ps_dec->s_tran_addrecon.u4_inc_y[1] = (ui16_lastmbs_widthY

                        + (PAD_LEN_Y_H << 1)

                        + ps_dec->s_tran_addrecon.u2_frm_wd_y

                                        * ((15 << u1_mbaff) + u1_mbaff));

        ps_dec->s_tran_addrecon.u4_inc_uv[1] = (ui16_lastmbs_widthUV

                        + (PAD_LEN_UV_H << 2)

                        + ps_dec->s_tran_addrecon.u2_frm_wd_uv

                                        * ((15 << u1_mbaff) + u1_mbaff));

        /* Assign picture numbers to each frame/field  */

        /* only once per picture.                      */

        ih264d_assign_pic_num(ps_dec);

        ps_dec->s_tran_addrecon.u2_mv_top_left_inc = (ps_dec->u4_recon_mb_grp

                        << 2) - 1 - (u1_mbaff << 2);

        ps_dec->s_tran_addrecon.u2_mv_left_inc = ((ps_dec->u4_recon_mb_grp

                        >> u1_mbaff) - 1) << (4 + u1_mbaff);

    }

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

    /* High profile related initialization at pictrue level               */

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

    if(ps_seq->u1_profile_idc == HIGH_PROFILE_IDC)

    {

        if((ps_seq->i4_seq_scaling_matrix_present_flag)

                        || (ps_pps->i4_pic_scaling_matrix_present_flag))

        {

            ret = ih264d_form_scaling_matrix_picture(ps_seq, ps_pps, ps_dec);

            ps_dec->s_high_profile.u1_scaling_present = 1;

        }

        else

        {

            ret = ih264d_form_default_scaling_matrix(ps_dec);

        }

    }

    else

    {

        ret = ih264d_form_default_scaling_matrix(ps_dec);

    }

    ps_dec->s_high_profile.u1_transform8x8_present = ps_pps->i4_transform_8x8_mode_flag;

    if(ret != OK)

        return ret;

    /* required while reading the transform_size_8x8 u4_flag */

    ps_dec->s_high_profile.u1_direct_8x8_inference_flag =

                    ps_seq->u1_direct_8x8_inference_flag;

    ps_dec->s_high_profile.s_cavlc_ctxt = ps_dec->s_cavlc_ctxt;

    ps_dec->i1_recon_in_thread3_flag = 1;

    ps_dec->ps_frame_buf_ip_recon = &ps_dec->s_tran_addrecon;

    if(ps_dec->u1_separate_parse)

    {

        memcpy(&ps_dec->s_tran_addrecon_parse, &ps_dec->s_tran_addrecon,

               sizeof(tfr_ctxt_t));

        if(ps_dec->u4_num_cores >= 3 && ps_dec->i1_recon_in_thread3_flag)

        {

            memcpy(&ps_dec->s_tran_iprecon, &ps_dec->s_tran_addrecon,

                   sizeof(tfr_ctxt_t));

            ps_dec->ps_frame_buf_ip_recon = &ps_dec->s_tran_iprecon;

        }

    }

    ih264d_init_deblk_tfr_ctxt(ps_dec,&(ps_dec->s_pad_mgr), &(ps_dec->s_tran_addrecon),

                               ps_dec->u2_frm_wd_in_mbs, 0);

    ps_dec->ps_cur_deblk_mb = ps_dec->ps_deblk_pic;

    ps_dec->u4_cur_deblk_mb_num = 0;

    ps_dec->u4_deblk_mb_x = 0;

    ps_dec->u4_deblk_mb_y = 0;

    ps_dec->pu4_wt_ofsts = ps_dec->pu4_wts_ofsts_mat;

    ps_dec->u4_first_slice_in_pic = 0;

    H264_MUTEX_UNLOCK(&ps_dec->process_disp_mutex);

    return OK;

}

/*!

 **************************************************************************

 * \if Function name :  ih264d_deblock_display \endif

 *

 * \brief            :  The function callls the deblocking routine and manages

 :  the Recon buffers and displays .

 * \return           :  Nothing

 *

 **************************************************************************

 */

WORD32 ih264d_end_of_pic_dispbuf_mgr(dec_struct_t * ps_dec)

{

    dec_slice_params_t *ps_cur_slice = ps_dec->ps_cur_slice;

    UWORD8 u1_num_of_users = 0;

    WORD32 ret;

    H264_MUTEX_LOCK(&ps_dec->process_disp_mutex);

    if(1)

    {

        {

            ih264d_delete_nonref_nondisplay_pics(ps_dec->ps_dpb_mgr);

            if(ps_cur_slice->u1_mmco_equalto5

                            || (ps_cur_slice->u1_nal_unit_type == IDR_SLICE_NAL))

            {

                ps_dec->ps_cur_pic->i4_poc = 0;

                if(ps_dec->u4_total_mbs_coded

                                == (ps_dec->ps_cur_sps->u4_max_mb_addr + 1))

                    ih264d_reset_ref_bufs(ps_dec->ps_dpb_mgr);

                ih264d_release_display_bufs(ps_dec);

            }

            if(IVD_DECODE_FRAME_OUT != ps_dec->e_frm_out_mode)

            {

                ret = ih264d_assign_display_seq(ps_dec);

                if(ret != OK)

                    return ret;

            }

        }

        if(ps_cur_slice->u1_nal_ref_idc)

        {

            /* Mark pic buf as needed for reference */

            ih264_buf_mgr_set_status((buf_mgr_t *)ps_dec->pv_pic_buf_mgr,

                                     ps_dec->u1_pic_buf_id,

                                     BUF_MGR_REF);

            /* Mark mv buf as needed for reference */

            ih264_buf_mgr_set_status((buf_mgr_t *)ps_dec->pv_mv_buf_mgr,

                                     ps_dec->as_buf_id_info_map[ps_dec->u1_pic_buf_id].mv_buf_id,

                                     BUF_MGR_REF);

            ps_dec->au1_pic_buf_ref_flag[ps_dec->u1_pic_buf_id] = 1;

        }

        /* 420 consumer */

        /* Increment the number of users by 1 for display based upon */

        /*the SEEK KEY FRAME control sent to decoder                 */

        if(((0 == ps_dec->u1_last_pic_not_decoded)

                        && (0

                                        == (ps_dec->ps_cur_pic->u4_pack_slc_typ

                                                        & ps_dec->u4_skip_frm_mask)))

                        || (ps_cur_slice->u1_nal_unit_type == IDR_SLICE_NAL))

        {

            /* Mark pic buf as needed for display */

            ih264_buf_mgr_set_status((buf_mgr_t *)ps_dec->pv_pic_buf_mgr,

                                     ps_dec->u1_pic_buf_id,

                                     BUF_MGR_IO);

        }

        if(!ps_cur_slice->u1_field_pic_flag

                        || ((TOP_FIELD_ONLY | BOT_FIELD_ONLY)

                                        != ps_dec->u1_top_bottom_decoded))

        {

            pic_buffer_t *ps_cur_pic = ps_dec->ps_cur_pic;

            ps_cur_pic->u2_disp_width = ps_dec->u2_disp_width;

            ps_cur_pic->u2_disp_height = ps_dec->u2_disp_height >> 1;

            ps_cur_pic->u2_crop_offset_y = ps_dec->u2_crop_offset_y;

            ps_cur_pic->u2_crop_offset_uv = ps_dec->u2_crop_offset_uv;

            ps_cur_pic->u1_pic_type = 0;

            {

                WORD64 i8_display_poc;

                i8_display_poc = (WORD64)ps_dec->i4_prev_max_display_seq +

                            ps_dec->ps_cur_pic->i4_poc;

                if(IS_OUT_OF_RANGE_S32(i8_display_poc))

                {

                    ps_dec->i4_prev_max_display_seq = 0;

                }

            }

            ret = ih264d_insert_pic_in_display_list(

                            ps_dec->ps_dpb_mgr,

                            ps_dec->u1_pic_buf_id,

                            ps_dec->i4_prev_max_display_seq

                                            + ps_dec->ps_cur_pic->i4_poc,

                            ps_dec->ps_cur_pic->i4_frame_num);

            if(ret != OK)

                return ret;

            {

                ivd_video_decode_op_t * ps_dec_output =

                                (ivd_video_decode_op_t *)ps_dec->pv_dec_out;

                ps_dec_output->u4_frame_decoded_flag = 1;

            }

            if(ps_dec->au1_pic_buf_ref_flag[ps_dec->u1_pic_buf_id] == 0)

            {

                ih264_buf_mgr_release((buf_mgr_t *)ps_dec->pv_mv_buf_mgr,

                                      ps_dec->as_buf_id_info_map[ps_dec->u1_pic_buf_id].mv_buf_id,

                                      BUF_MGR_REF);

                ps_dec->au1_pic_buf_ref_flag[ps_dec->u1_pic_buf_id] = 0;

            }

        }

        else

        {

            H264_DEC_DEBUG_PRINT("pic not inserted display %d %d\n",

                                 ps_cur_slice->u1_field_pic_flag,

                                 ps_dec->u1_second_field);

        }

        if(!ps_cur_slice->u1_field_pic_flag

                        || ((TOP_FIELD_ONLY | BOT_FIELD_ONLY)

                                        == ps_dec->u1_top_bottom_decoded))

        {

            if(IVD_DECODE_FRAME_OUT == ps_dec->e_frm_out_mode)

            {

                ret = ih264d_assign_display_seq(ps_dec);

                if(ret != OK)

                    return ret;

            }

        }

    }

    H264_MUTEX_UNLOCK(&ps_dec->process_disp_mutex);

    return OK;

}

void ih264d_err_pic_dispbuf_mgr(dec_struct_t *ps_dec)

{

    dec_slice_params_t *ps_cur_slice = ps_dec->ps_cur_slice;

    ivd_video_decode_op_t * ps_dec_output =

                    (ivd_video_decode_op_t *)ps_dec->pv_dec_out;

    ih264_buf_mgr_release((buf_mgr_t *)ps_dec->pv_pic_buf_mgr,

                          ps_dec->u1_pic_buf_id,

                          BUF_MGR_REF);

    ih264_buf_mgr_release((buf_mgr_t *)ps_dec->pv_mv_buf_mgr,

                          ps_dec->as_buf_id_info_map[ps_dec->u1_pic_buf_id].mv_buf_id,

                          BUF_MGR_REF);

    ih264_buf_mgr_release((buf_mgr_t *)ps_dec->pv_pic_buf_mgr,

                          ps_dec->u1_pic_buf_id,

                          BUF_MGR_IO);

}

void ih264d_deblock_picture(void *ptr)

{

    dec_struct_t *ps_dec = (dec_struct_t *)ptr;

    {

        /*Deblock picture only if all the mb's in the frame have been decoded*/

        if(ps_dec->u1_pic_decode_done == 1)

        {

            if(ps_dec->ps_cur_slice->u1_mbaff_frame_flag

                            || ps_dec->ps_cur_slice->u1_field_pic_flag)

            {

                ps_dec->p_DeblockPicture[ps_dec->ps_cur_slice->u1_mbaff_frame_flag](

                                ps_dec);

            }

            else

            {

                ih264d_deblock_picture_progressive(ps_dec);

            }

        }

    }

}

/*!

 **************************************************************************

 * \if Function name :  ih264d_deblock_display \endif

 *

 * \brief            :  The function callls the deblocking routine and manages

 :  the Recon buffers and displays .

 * \return           :  Nothing

 *

 **************************************************************************

 */

WORD32 ih264d_deblock_display(dec_struct_t *ps_dec)

{

    WORD32 ret;

    /* Call deblocking */

    ih264d_deblock_picture(ps_dec);

    ret = ih264d_end_of_pic_dispbuf_mgr(ps_dec);

    if(ret != OK)

        return ret;

    return OK;

}

/*

 *!

 **************************************************************************

 * \if Function name : EndofPoc \endif

 *

 * \brief

 *    EndofPoc Processing

 *

 * \return

 *    0 on Success and Error code otherwise

 **************************************************************************

 */

WORD32 ih264d_end_of_pic(dec_struct_t *ps_dec)

{

    dec_slice_params_t *ps_cur_slice = ps_dec->ps_cur_slice;

    WORD32 ret;

    {

        dec_err_status_t * ps_err = ps_dec->ps_dec_err_status;

        if(ps_err->u1_err_flag & REJECT_CUR_PIC)

        {

            ih264d_err_pic_dispbuf_mgr(ps_dec);

            return ERROR_NEW_FRAME_EXPECTED;

        }

    }

    H264_MUTEX_LOCK(&ps_dec->process_disp_mutex);

    ret = ih264d_end_of_pic_processing(ps_dec);

    if(ret != OK)

        return ret;

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

    /* ih264d_decode_pic_order_cnt - calculate the Pic Order Cnt                    */

    /* Needed to detect end of picture                                    */

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

    H264_MUTEX_UNLOCK(&ps_dec->process_disp_mutex);

    return OK;

}

/*!

 **************************************************************************

 * \if Function name : ih264d_fix_error_in_dpb \endif

 *

 * \brief

 *    fix error in DPB

 *

 * \return

 *    Number of node(s) deleted

 **************************************************************************

 */

WORD32 ih264d_fix_error_in_dpb(dec_struct_t *ps_dec)

{

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

    /* If there is common node in lt_list and st_list then delete it from */

    /* st_list                                                            */

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

    UWORD8 no_of_nodes_deleted = 0;

    UWORD8 lt_ref_num = ps_dec->ps_dpb_mgr->u1_num_lt_ref_bufs;

    struct dpb_info_t *ps_lt_curr_dpb = ps_dec->ps_dpb_mgr->ps_dpb_ht_head;

    while(lt_ref_num && ps_lt_curr_dpb)

    {

        if(ps_dec->ps_dpb_mgr->ps_dpb_st_head

                && ((ps_lt_curr_dpb->s_bot_field.u1_reference_info

                        | ps_lt_curr_dpb->s_top_field.u1_reference_info)

                        == (IS_SHORT_TERM | IS_LONG_TERM)))

        {

            struct dpb_info_t *ps_st_next_dpb = ps_dec->ps_dpb_mgr->ps_dpb_st_head;

            struct dpb_info_t *ps_st_curr_dpb = ps_dec->ps_dpb_mgr->ps_dpb_st_head;

            UWORD8 st_ref_num = ps_dec->ps_dpb_mgr->u1_num_st_ref_bufs;

            while(st_ref_num && ps_st_curr_dpb)

            {