/******************************************************************************
*
* Copyright (C) 2012 Ittiam Systems Pvt Ltd, Bangalore
*
* 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.
*
******************************************************************************/
/**
 *******************************************************************************
 * @file
 *  ihevc_iquant_recon.c
 *
 * @brief
 *  Contains function definitions for inverse  quantization and
 * reconstruction
 *
 * @author
 *  100470
 *
 * @par List of Functions:
 *  - ihevc_iquant_recon_4x4_ttype1()
 *  - ihevc_iquant_recon_4x4()
 *  - ihevc_iquant_recon_8x8()
 *  - ihevc_iquant_recon_16x16()
 *  - ihevc_iquant_recon_32x32()
 *
 * @remarks
 *  None
 *
 *******************************************************************************
 */
#include <stdio.h>
#include <string.h>
#include "ihevc_typedefs.h"
#include "ihevc_macros.h"
#include "ihevc_platform_macros.h"
#include "ihevc_defs.h"
#include "ihevc_trans_tables.h"
#include "ihevc_iquant_recon.h"
#include "ihevc_func_selector.h"
#include "ihevc_trans_macros.h"

/* All the functions here are replicated from ihevc_iquant_itrans_recon.c and modified to */
/* include reconstruction */

/**
 *******************************************************************************
 *
 * @brief
 *  This function performs inverse quantization type 1 and  reconstruction
 * for 4x4 input block
 *
 * @par Description:
 *  This function performs inverse quantization and  reconstruction for 4x4
 * input block
 *
 * @param[in] pi2_src
 *  Input 4x4 coefficients
 *
 * @param[in] pu1_pred
 *  Prediction 4x4 block
 *
 * @param[in] pi2_dequant_coeff
 *  Dequant Coeffs
 *
 * @param[out] pu1_dst
 *  Output 4x4 block
 *
 * @param[in] qp_div
 *  Quantization parameter / 6
 *
 * @param[in] qp_rem
 *  Quantization parameter % 6
 *
 * @param[in] src_strd
 *  Input stride
 *
 * @param[in] pred_strd
 *  Prediction stride
 *
 * @param[in] dst_strd
 *  Output Stride
 *
 * @param[in] zero_cols
 *  Zero columns in pi2_src
 *
 * @returns  Void
 *
 * @remarks
 *  None
 *
 *******************************************************************************
 */

void ihevc_iquant_recon_4x4_ttype1(WORD16 *pi2_src,
                                   UWORD8 *pu1_pred,
                                   WORD16 *pi2_dequant_coeff,
                                   UWORD8 *pu1_dst,
                                   WORD32 qp_div, /* qpscaled / 6 */
                                   WORD32 qp_rem, /* qpscaled % 6 */
                                   WORD32 src_strd,
                                   WORD32 pred_strd,
                                   WORD32 dst_strd,
                                   WORD32 zero_cols)
{

    {
        /* Inverse Quant and recon */
        {
            WORD32 i, j;
            WORD32 shift_iq;
            WORD32 trans_size;
            /* Inverse Quantization constants */
            {
                WORD32 log2_trans_size, bit_depth;

                log2_trans_size = 2;
                bit_depth = 8 + 0;
                shift_iq = bit_depth + log2_trans_size - 5;
            }

            trans_size = TRANS_SIZE_4;

            for(i = 0; i < trans_size; i++)
            {
                /* Checking for Zero Cols */
                if((zero_cols & 1) == 1)
                {
                    for(j = 0; j < trans_size; j++)
                        pu1_dst[j * dst_strd] = pu1_pred[j * pred_strd];
                }
                else
                {
                    for(j = 0; j < trans_size; j++)
                    {
                        WORD32 iquant_out;
                        IQUANT_4x4(iquant_out,
                                   pi2_src[j * src_strd],
                                   pi2_dequant_coeff[j * trans_size] * g_ihevc_iquant_scales[qp_rem],
                                   shift_iq, qp_div);

                        iquant_out = (iquant_out + 16) >> 5;
                        pu1_dst[j * dst_strd] =
                                        CLIP_U8(iquant_out + pu1_pred[j * pred_strd]);
                    }
                }
                pi2_src++;
                pi2_dequant_coeff++;
                pu1_pred++;
                pu1_dst++;

                zero_cols = zero_cols >> 1;
            }
        }
    }
}

/**
 *******************************************************************************
 *
 * @brief
 *  This function performs inverse quantization and  reconstruction for 4x4
 * input block
 *
 * @par Description:
 *  This function performs inverse quantization and  reconstruction for 4x4
 * input block
 *
 * @param[in] pi2_src
 *  Input 4x4 coefficients
 *
 * @param[in] pu1_pred
 *  Prediction 4x4 block
 *
 * @param[in] pi2_dequant_coeff
 *  Dequant Coeffs
 *
 * @param[out] pu1_dst
 *  Output 4x4 block
 *
 * @param[in] qp_div
 *  Quantization parameter / 6
 *
 * @param[in] qp_rem
 *  Quantization parameter % 6
 *
 * @param[in] src_strd
 *  Input stride
 *
 * @param[in] pred_strd
 *  Prediction stride
 *
 * @param[in] dst_strd
 *  Output Stride
 *
 * @param[in] zero_cols
 *  Zero columns in pi2_src
 *
 * @returns  Void
 *
 * @remarks
 *  None
 *
 *******************************************************************************
 */

void ihevc_iquant_recon_4x4(WORD16 *pi2_src,
                            UWORD8 *pu1_pred,
                            WORD16 *pi2_dequant_coeff,
                            UWORD8 *pu1_dst,
                            WORD32 qp_div, /* qpscaled / 6 */
                            WORD32 qp_rem, /* qpscaled % 6 */
                            WORD32 src_strd,
                            WORD32 pred_strd,
                            WORD32 dst_strd,
                            WORD32 zero_cols)
{

    {
        /* Inverse Quant and recon */
        {
            WORD32 i, j;
            WORD32 shift_iq;
            WORD32 trans_size;
            /* Inverse Quantization constants */
            {
                WORD32 log2_trans_size, bit_depth;

                log2_trans_size = 2;
                bit_depth = 8 + 0;
                shift_iq = bit_depth + log2_trans_size - 5;
            }

            trans_size = TRANS_SIZE_4;

            for(i = 0; i < trans_size; i++)
            {
                /* Checking for Zero Cols */
                if((zero_cols & 1) == 1)
                {
                    for(j = 0; j < trans_size; j++)
                        pu1_dst[j * dst_strd] = pu1_pred[j * pred_strd];
                }
                else
                {
                    for(j = 0; j < trans_size; j++)
                    {
                        WORD32 iquant_out;
                        IQUANT_4x4(iquant_out,
                                   pi2_src[j * src_strd],
                                   pi2_dequant_coeff[j * trans_size] * g_ihevc_iquant_scales[qp_rem],
                                   shift_iq, qp_div);
                        iquant_out = (iquant_out + 16) >> 5;
                        pu1_dst[j * dst_strd] =
                                        CLIP_U8(iquant_out + pu1_pred[j * pred_strd]);
                    }
                }
                pi2_src++;
                pi2_dequant_coeff++;
                pu1_pred++;
                pu1_dst++;

                zero_cols = zero_cols >> 1;
            }
        }
    }
}

/**
 *******************************************************************************
 *
 * @brief
 *  This function performs inverse quantization and  reconstruction for 8x8
 * input block
 *
 * @par Description:
 *  This function performs inverse quantization and  reconstruction for 8x8
 * input block
 *
 * @param[in] pi2_src
 *  Input 8x8 coefficients
 *
 * @param[in] pu1_pred
 *  Prediction 8x8 block
 *
 * @param[in] pi2_dequant_coeff
 *  Dequant Coeffs
 *
 * @param[out] pu1_dst
 *  Output 8x8 block
 *
 * @param[in] qp_div
 *  Quantization parameter / 6
 *
 * @param[in] qp_rem
 *  Quantization parameter % 6
 *
 * @param[in] src_strd
 *  Input stride
 *
 * @param[in] pred_strd
 *  Prediction stride
 *
 * @param[in] dst_strd
 *  Output Stride
 *
 * @param[in] zero_cols
 *  Zero columns in pi2_src
 *
 * @returns  Void
 *
 * @remarks
 *  None
 *
 *******************************************************************************
 */

void ihevc_iquant_recon_8x8(WORD16 *pi2_src,
                            UWORD8 *pu1_pred,
                            WORD16 *pi2_dequant_coeff,
                            UWORD8 *pu1_dst,
                            WORD32 qp_div, /* qpscaled / 6 */
                            WORD32 qp_rem, /* qpscaled % 6 */
                            WORD32 src_strd,
                            WORD32 pred_strd,
                            WORD32 dst_strd,
                            WORD32 zero_cols)
{

    {
        /* Inverse Quant and recon */
        {
            WORD32 i, j;
            WORD32 shift_iq;
            WORD32 trans_size;
            /* Inverse Quantization constants */
            {
                WORD32 log2_trans_size, bit_depth;

                log2_trans_size = 3;
                bit_depth = 8 + 0;
                shift_iq = bit_depth + log2_trans_size - 5;
            }

            trans_size = TRANS_SIZE_8;

            for(i = 0; i < trans_size; i++)
            {
                /* Checking for Zero Cols */
                if((zero_cols & 1) == 1)
                {
                    for(j = 0; j < trans_size; j++)
                        pu1_dst[j * dst_strd] = pu1_pred[j * pred_strd];
                }
                else
                {
                    for(j = 0; j < trans_size; j++)
                    {
                        WORD32 iquant_out;
                        IQUANT(iquant_out,
                               pi2_src[j * src_strd],
                               pi2_dequant_coeff[j * trans_size] * g_ihevc_iquant_scales[qp_rem],
                               shift_iq, qp_div);
                        iquant_out = (iquant_out + 16) >> 5;
                        pu1_dst[j * dst_strd] =
                                        CLIP_U8(iquant_out + pu1_pred[j * pred_strd]);
                    }
                }
                pi2_src++;
                pi2_dequant_coeff++;
                pu1_pred++;
                pu1_dst++;

                zero_cols = zero_cols >> 1;
            }
        }
    }
}

/**
 *******************************************************************************
 *
 * @brief
 *  This function performs inverse quantization and  reconstruction for 16x16
 * input block
 *
 * @par Description:
 *  This function performs inverse quantization and  reconstruction for 16x16
 * input block
 *
 * @param[in] pi2_src
 *  Input 16x16 coefficients
 *
 * @param[in] pu1_pred
 *  Prediction 16x16 block
 *
 * @param[in] pi2_dequant_coeff
 *  Dequant Coeffs
 *
 * @param[out] pu1_dst
 *  Output 16x16 block
 *
 * @param[in] qp_div
 *  Quantization parameter / 6
 *
 * @param[in] qp_rem
 *  Quantization parameter % 6
 *
 * @param[in] src_strd
 *  Input stride
 *
 * @param[in] pred_strd
 *  Prediction stride
 *
 * @param[in] dst_strd
 *  Output Stride
 *
 * @param[in] zero_cols
 *  Zero columns in pi2_src
 *
 * @returns  Void
 *
 * @remarks
 *  None
 *
 *******************************************************************************
 */

void ihevc_iquant_recon_16x16(WORD16 *pi2_src,
                              UWORD8 *pu1_pred,
                              WORD16 *pi2_dequant_coeff,
                              UWORD8 *pu1_dst,
                              WORD32 qp_div, /* qpscaled / 6 */
                              WORD32 qp_rem, /* qpscaled % 6 */
                              WORD32 src_strd,
                              WORD32 pred_strd,
                              WORD32 dst_strd,
                              WORD32 zero_cols)

{

    {
        /* Inverse Quant and recon */
        {
            WORD32 i, j;
            WORD32 shift_iq;
            WORD32 trans_size;
            /* Inverse Quantization constants */
            {
                WORD32 log2_trans_size, bit_depth;

                log2_trans_size = 4;
                bit_depth = 8 + 0;
                shift_iq = bit_depth + log2_trans_size - 5;
            }

            trans_size = TRANS_SIZE_16;

            for(i = 0; i < trans_size; i++)
            {
                /* Checking for Zero Cols */
                if((zero_cols & 1) == 1)
                {
                    for(j = 0; j < trans_size; j++)
                        pu1_dst[j * dst_strd] = pu1_pred[j * pred_strd];
                }
                else
                {
                    for(j = 0; j < trans_size; j++)
                    {
                        WORD32 iquant_out;
                        IQUANT(iquant_out,
                               pi2_src[j * src_strd],
                               pi2_dequant_coeff[j * trans_size] * g_ihevc_iquant_scales[qp_rem],
                               shift_iq, qp_div);
                        iquant_out = (iquant_out + 16) >> 5;
                        pu1_dst[j * dst_strd] =
                                        CLIP_U8(iquant_out + pu1_pred[j * pred_strd]);
                    }
                }
                pi2_src++;
                pi2_dequant_coeff++;
                pu1_pred++;
                pu1_dst++;

                zero_cols = zero_cols >> 1;
            }
        }
    }
}

/**
 *******************************************************************************
 *
 * @brief
 *  This function performs inverse quantization and  reconstruction for 32x32
 * input block
 *
 * @par Description:
 *  This function performs inverse quantization and  reconstruction for 32x32
 * input block
 *
 * @param[in] pi2_src
 *  Input 32x32 coefficients
 *
 * @param[in] pu1_pred
 *  Prediction 32x32 block
 *
 * @param[in] pi2_dequant_coeff
 *  Dequant Coeffs
 *
 * @param[out] pu1_dst
 *  Output 32x32 block
 *
 * @param[in] qp_div
 *  Quantization parameter / 6
 *
 * @param[in] qp_rem
 *  Quantization parameter % 6
 *
 * @param[in] src_strd
 *  Input stride
 *
 * @param[in] pred_strd
 *  Prediction stride
 *
 * @param[in] dst_strd
 *  Output Stride
 *
 * @param[in] zero_cols
 *  Zero columns in pi2_src
 *
 * @returns  Void
 *
 * @remarks
 *  None
 *
 *******************************************************************************
 */

void ihevc_iquant_recon_32x32(WORD16 *pi2_src,
                              UWORD8 *pu1_pred,
                              WORD16 *pi2_dequant_coeff,
                              UWORD8 *pu1_dst,
                              WORD32 qp_div, /* qpscaled / 6 */
                              WORD32 qp_rem, /* qpscaled % 6 */
                              WORD32 src_strd,
                              WORD32 pred_strd,
                              WORD32 dst_strd,
                              WORD32 zero_cols)
{

    {
        /* Inverse Quant and recon */
        {
            WORD32 i, j;
            WORD32 shift_iq;
            WORD32 trans_size;
            /* Inverse Quantization constants */
            {
                WORD32 log2_trans_size, bit_depth;

                log2_trans_size = 5;
                bit_depth = 8 + 0;
                shift_iq = bit_depth + log2_trans_size - 5;
            }

            trans_size = TRANS_SIZE_32;

            for(i = 0; i < trans_size; i++)
            {
                /* Checking for Zero Cols */
                if((zero_cols & 1) == 1)
                {
                    for(j = 0; j < trans_size; j++)
                        pu1_dst[j * dst_strd] = pu1_pred[j * pred_strd];
                }
                else
                {
                    for(j = 0; j < trans_size; j++)
                    {
                        WORD32 iquant_out;
                        IQUANT(iquant_out,
                               pi2_src[j * src_strd],
                               pi2_dequant_coeff[j * trans_size] * g_ihevc_iquant_scales[qp_rem],
                               shift_iq, qp_div);
                        iquant_out = (iquant_out + 16) >> 5;
                        pu1_dst[j * dst_strd] =
                                        CLIP_U8(iquant_out + pu1_pred[j * pred_strd]);
                    }
                }
                pi2_src++;
                pi2_dequant_coeff++;
                pu1_pred++;
                pu1_dst++;

                zero_cols = zero_cols >> 1;
            }
        }
    }
}