/******************************************************************************
 *
 * 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 "ih264_typedefs.h"
#include "ih264_macros.h"
#include "ih264_platform_macros.h"
#include "ih264d_bitstrm.h"
#include "ih264d_structs.h"
#include "ih264d_parse_cavlc.h"
#include "ih264d_defs.h"
#include "ih264d_defs.h"
#include "ih264d_defs.h"

#include "ih264d_parse_slice.h"
#include "ih264d_tables.h"
#include "ih264d_utils.h"
#include "ih264d_nal.h"
#include "ih264d_deblocking.h"

#include "ih264d_mem_request.h"
#include "ih264d_debug.h"

#include "ih264d_error_handler.h"
#include "ih264d_mb_utils.h"
#include "ih264d_sei.h"
#include "ih264d_vui.h"
#include "ih264d_tables.h"

#define IDCT_BLOCK_WIDTH8X8  8

void ih264d_scaling_list(WORD16 *pi2_scaling_list,
                         WORD32 i4_size_of_scalinglist,
                         UWORD8 *pu1_use_default_scaling_matrix_flag,
                         dec_bit_stream_t *ps_bitstrm)
{
    WORD32 i4_j, i4_delta_scale, i4_lastScale = 8, i4_nextScale = 8;
    UWORD32 *pu4_bitstrm_buf = ps_bitstrm->pu4_buffer;
    UWORD32 *pu4_bitstrm_ofst = &ps_bitstrm->u4_ofst;

    *pu1_use_default_scaling_matrix_flag = 0;

    for(i4_j = 0; i4_j < i4_size_of_scalinglist; i4_j++)
    {
        if(i4_nextScale != 0)
        {
            i4_delta_scale = ih264d_sev(pu4_bitstrm_ofst,
                                        pu4_bitstrm_buf);

            i4_nextScale = ((i4_lastScale + i4_delta_scale + 256) & 0xff);

            *pu1_use_default_scaling_matrix_flag = ((i4_j == 0)
                            && (i4_nextScale == 0));

        }
        pi2_scaling_list[i4_j] =
                        (i4_nextScale == 0) ? (i4_lastScale) : (i4_nextScale);
        i4_lastScale = pi2_scaling_list[i4_j];
    }
}

WORD32 ih264d_form_default_scaling_matrix(dec_struct_t *ps_dec)
{

    /*************************************************************************/
    /* perform the inverse scanning for the frame and field scaling matrices */
    /*************************************************************************/
    {
        UWORD8 *pu1_inv_scan;
        WORD32 i4_i, i4_j;

        pu1_inv_scan = (UWORD8 *)gau1_ih264d_inv_scan;

        /* for all 4x4 matrices */
        for(i4_i = 0; i4_i < 6; i4_i++)
        {
            for(i4_j = 0; i4_j < 16; i4_j++)
            {
                ps_dec->s_high_profile.i2_scalinglist4x4[i4_i][pu1_inv_scan[i4_j]] =
                                16;

            }
        }

        /* for all 8x8 matrices */
        for(i4_i = 0; i4_i < 2; i4_i++)
        {
            for(i4_j = 0; i4_j < 64; i4_j++)
            {
                ps_dec->s_high_profile.i2_scalinglist8x8[i4_i][gau1_ih264d_inv_scan_prog8x8_cabac[i4_j]] =
                                16;

            }
        }
    }
    return OK;
}

WORD32 ih264d_form_scaling_matrix_picture(dec_seq_params_t *ps_seq,
                                          dec_pic_params_t *ps_pic,
                                          dec_struct_t *ps_dec)
{
    /* default scaling matrices */
    WORD32 i4_i;

    /* check the SPS first */
    if(ps_seq->i4_seq_scaling_matrix_present_flag)
    {
        for(i4_i = 0; i4_i < 8; i4_i++)
        {
            if(i4_i < 6)
            {
                /* fall-back rule A */
                if(!ps_seq->u1_seq_scaling_list_present_flag[i4_i])
                {
                    if((i4_i == 0) || (i4_i == 3))
                    {
                        ps_dec->s_high_profile.pi2_scale_mat[i4_i] =
                                        (i4_i == 0) ? (WORD16 *)(gai2_ih264d_default_intra4x4) : (WORD16 *)(gai2_ih264d_default_inter4x4);
                    }
                    else
                    {
                        ps_dec->s_high_profile.pi2_scale_mat[i4_i] =
                                        ps_dec->s_high_profile.pi2_scale_mat[i4_i
                                                        - 1];
                    }
                }
                else
                {
                    if(ps_seq->u1_use_default_scaling_matrix_flag[i4_i])
                    {
                        ps_dec->s_high_profile.pi2_scale_mat[i4_i] =
                                        (i4_i < 3) ? (WORD16 *)(gai2_ih264d_default_intra4x4) : (WORD16 *)(gai2_ih264d_default_inter4x4);
                    }
                    else
                    {
                        ps_dec->s_high_profile.pi2_scale_mat[i4_i] =
                                        ps_seq->i2_scalinglist4x4[i4_i];
                    }
                }

            }
            else
            {
                /* fall-back rule A */
                if((!ps_seq->u1_seq_scaling_list_present_flag[i4_i])
                                || (ps_seq->u1_use_default_scaling_matrix_flag[i4_i]))
                {
                    ps_dec->s_high_profile.pi2_scale_mat[i4_i] =
                                    (i4_i == 6) ? ((WORD16*)gai2_ih264d_default_intra8x8) : ((WORD16*)gai2_ih264d_default_inter8x8);
                }
                else
                {
                    ps_dec->s_high_profile.pi2_scale_mat[i4_i] =
                                    ps_seq->i2_scalinglist8x8[i4_i - 6];
                }
            }
        }
    }

    /* checking for the PPS */

    if(ps_pic->i4_pic_scaling_matrix_present_flag)
    {
        for(i4_i = 0; i4_i < 8; i4_i++)
        {
            if(i4_i < 6)
            {
                /* fall back rule B */
                if(!ps_pic->u1_pic_scaling_list_present_flag[i4_i])
                {
                    if((i4_i == 0) || (i4_i == 3))
                    {
                        if(!ps_seq->i4_seq_scaling_matrix_present_flag)
                        {
                            ps_dec->s_high_profile.pi2_scale_mat[i4_i] =
                                            (i4_i == 0) ? (WORD16 *)(gai2_ih264d_default_intra4x4) : (WORD16 *)(gai2_ih264d_default_inter4x4);
                        }
                    }
                    else
                    {
                        ps_dec->s_high_profile.pi2_scale_mat[i4_i] =
                                        ps_dec->s_high_profile.pi2_scale_mat[i4_i
                                                        - 1];
                    }
                }
                else
                {
                    if(ps_pic->u1_pic_use_default_scaling_matrix_flag[i4_i])
                    {
                        ps_dec->s_high_profile.pi2_scale_mat[i4_i] =
                                        (i4_i < 3) ? (WORD16 *)(gai2_ih264d_default_intra4x4) : (WORD16 *)(gai2_ih264d_default_inter4x4);
                    }
                    else
                    {
                        ps_dec->s_high_profile.pi2_scale_mat[i4_i] =
                                        ps_pic->i2_pic_scalinglist4x4[i4_i];
                    }
                }
            }
            else
            {
                if(!ps_pic->u1_pic_scaling_list_present_flag[i4_i])
                {
                    if(!ps_seq->i4_seq_scaling_matrix_present_flag)
                    {
                        ps_dec->s_high_profile.pi2_scale_mat[i4_i] =
                                        (i4_i == 6) ? ((WORD16*)gai2_ih264d_default_intra8x8) : ((WORD16*)gai2_ih264d_default_inter8x8);
                    }
                }
                else
                {
                    if(ps_pic->u1_pic_use_default_scaling_matrix_flag[i4_i])
                    {
                        ps_dec->s_high_profile.pi2_scale_mat[i4_i] =
                                        (i4_i == 6) ? (WORD16 *)(gai2_ih264d_default_intra8x8) : (WORD16 *)(gai2_ih264d_default_inter8x8);
                    }
                    else
                    {
                        ps_dec->s_high_profile.pi2_scale_mat[i4_i] =
                                        ps_pic->i2_pic_scalinglist8x8[i4_i - 6];
                    }
                }
            }
        }
    }

    /*************************************************************************/
    /* perform the inverse scanning for the frame and field scaling matrices */
    /*************************************************************************/
    {
        UWORD8 *pu1_inv_scan_4x4;
        WORD32 i4_i, i4_j;

        pu1_inv_scan_4x4 = (UWORD8 *)gau1_ih264d_inv_scan;

        /* for all 4x4 matrices */
        for(i4_i = 0; i4_i < 6; i4_i++)
        {
            if(ps_dec->s_high_profile.pi2_scale_mat[i4_i] == NULL)
                return ERROR_CORRUPTED_SLICE;

            for(i4_j = 0; i4_j < 16; i4_j++)
            {
                ps_dec->s_high_profile.i2_scalinglist4x4[i4_i][pu1_inv_scan_4x4[i4_j]] =
                                ps_dec->s_high_profile.pi2_scale_mat[i4_i][i4_j];

            }
        }

        /* for all 8x8 matrices */
        for(i4_i = 0; i4_i < 2; i4_i++)
        {
            if(ps_dec->s_high_profile.pi2_scale_mat[i4_i + 6] == NULL)
                return ERROR_CORRUPTED_SLICE;

            for(i4_j = 0; i4_j < 64; i4_j++)
            {
                ps_dec->s_high_profile.i2_scalinglist8x8[i4_i][gau1_ih264d_inv_scan_prog8x8_cabac[i4_j]] =
                                ps_dec->s_high_profile.pi2_scale_mat[i4_i + 6][i4_j];

            }
        }
    }
    return OK;
}