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

 *

 * Copyright (C) 2018 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

 *  ihevc_resi_trans.c

 *

 * @brief

 *  Contains function definitions for residual and  forward transform

 *

 * @author

 *  100470

 *

 * @par List of Functions:

 *  - ihevc_resi_trans_4x4_ttype1()

 *  - ihevc_resi_trans_4x4()

 *  - ihevc_resi_trans_8x8()

 *  - ihevc_resi_trans_16x16()

 *  - ihevc_resi_trans_32x32()

 *

 * @remarks

 *  None

 *

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

 */

#include <stdio.h>

#include <string.h>

#include <stdlib.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_resi_trans.h"

#include "ihevc_trans_macros.h"

/**

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

 *

 * @brief

 *  This function performs residue calculation and forward  transform type 1

 * on input pixels

 *

 * @par Description:

 *  Performs residue calculation by subtracting source and  prediction and

 * followed by forward transform

 *

 * @param[in] pu1_src

 *  Input 4x4 pixels

 *

 * @param[in] pu1_pred

 *  Prediction data

 *

 * @param[in] pi2_tmp

 *  Temporary buffer of size 4x4

 *

 * @param[out] pi2_dst

 *  Output 4x4 coefficients

 *

 * @param[in] src_strd

 *  Input stride

 *

 * @param[in] pred_strd

 *  Prediction Stride

 *

 * @param[in] dst_strd

 *  Output Stride

 *

 * @param[in] e_chroma_plane

 *  Enum singalling chroma plane

 *

 *

 * @returns  Void

 *

 * @remarks

 *  None

 *

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

 */

UWORD32 ihevc_resi_trans_4x4_ttype1(UWORD8 *pu1_src,

                                 UWORD8 *pu1_pred,

                                    WORD32 *pi4_temp,

                                 WORD16 *pi2_dst,

                                 WORD32 src_strd,

                                 WORD32 pred_strd,

                                 WORD32 dst_strd,

                                 CHROMA_PLANE_ID_T e_chroma_plane)

{

    WORD32 i, c[4];

    WORD32 add, shift;

    WORD32 trans_size;

    WORD32 *pi4_tmp_orig;

    WORD16 *pi2_dst_orig;

    UWORD32  u4_blk_sad = 0;

    UNUSED(e_chroma_plane);

    pi2_dst_orig = pi2_dst;

    pi4_tmp_orig = pi4_temp;

    trans_size = TRANS_SIZE_4;

    /* Residue + Forward Transform 1st stage */

    shift = 1; // log2(iWidth) - 1 + g_uiBitIncrement

    add = 1 << (shift - 1);

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

    {

        WORD32 resi_tmp_1, resi_tmp_2, resi_tmp_3;

        // Intermediate Variables

        resi_tmp_1 = pu1_src[0] - pu1_pred[0];

        resi_tmp_2 = pu1_src[3] - pu1_pred[3];

        c[0] = resi_tmp_1 + resi_tmp_2;

        u4_blk_sad += abs(resi_tmp_1) + abs(resi_tmp_2);

        resi_tmp_1 = pu1_src[1] - pu1_pred[1];

        resi_tmp_2 = pu1_src[3] - pu1_pred[3];

        c[1] = resi_tmp_1 + resi_tmp_2;

        u4_blk_sad += abs(resi_tmp_1);

        resi_tmp_1 = pu1_src[0] - pu1_pred[0];

        resi_tmp_2 = pu1_src[1] - pu1_pred[1];

        c[2] = resi_tmp_1 - resi_tmp_2;

        resi_tmp_1 = pu1_src[2] - pu1_pred[2];

        c[3] = 74 * resi_tmp_1;

        u4_blk_sad += abs(resi_tmp_1);

        pi4_temp[0] = (29 * c[0] + 55 * c[1] + c[3] + add) >> shift;

        resi_tmp_1 = pu1_src[0] - pu1_pred[0];

        resi_tmp_2 = pu1_src[1] - pu1_pred[1];

        resi_tmp_3 = pu1_src[3] - pu1_pred[3];

        pi4_temp[trans_size] =

                        (74 * (resi_tmp_1 + resi_tmp_2 - resi_tmp_3) + add)

                                        >> shift;

        pi4_temp[2 * trans_size] = (29 * c[2] + 55 * c[0] - c[3] + add) >> shift;

        pi4_temp[3 * trans_size] = (55 * c[2] - 29 * c[1] + c[3] + add) >> shift;

        pu1_src += src_strd;

        pu1_pred += pred_strd;

        pi4_temp++;

    }

    pi4_temp = pi4_tmp_orig;

    /* Forward transform 2nd stage */

    shift = 8; // log2(iHeight) + 6

    add = 1 << (shift - 1);

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

    {

        // Intermediate Variables

        c[0] = pi4_temp[0] + pi4_temp[3];

        c[1] = pi4_temp[1] + pi4_temp[3];

        c[2] = pi4_temp[0] - pi4_temp[1];

        c[3] = 74 * pi4_temp[2];

        pi2_dst[0] = (29 * c[0] + 55 * c[1] + c[3] + add) >> shift;

        pi2_dst[dst_strd] = (74 * (pi4_temp[0] + pi4_temp[1] - pi4_temp[3]) + add)

                        >> shift;

        pi2_dst[2 * dst_strd] = (29 * c[2] + 55 * c[0] - c[3] + add) >> shift;

        pi2_dst[3 * dst_strd] = (55 * c[2] - 29 * c[1] + c[3] + add) >> shift;

        pi4_temp += trans_size;

        pi2_dst++;

    }

    return u4_blk_sad;

}

/**

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

 *

 * @brief

 *  This function performs residue calculation and forward  transform on

 * input pixels

 *

 * @par Description:

 *  Performs residue calculation by subtracting source and  prediction and

 * followed by forward transform

 *

 * @param[in] pu1_src

 *  Input 4x4 pixels

 *

 * @param[in] pu1_pred

 *  Prediction data

 *

 * @param[in] pi2_tmp

 *  Temporary buffer of size 4x4

 *

 * @param[out] pi2_dst

 *  Output 4x4 coefficients

 *

 * @param[in] src_strd

 *  Input stride

 *

 * @param[in] pred_strd

 *  Prediction Stride

 *

 * @param[in] dst_strd

 *  Output Stride

 *

 * @param[in] e_chroma_plane

 *  Enum singalling chroma plane

 *

 * @returns  Void

 *

 * @remarks

 *  None

 *

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

 */

UWORD32 ihevc_resi_trans_4x4(UWORD8 *pu1_src,

                          UWORD8 *pu1_pred,

                          WORD32 *pi4_temp,

                          WORD16 *pi2_dst,

                          WORD32 src_strd,

                          WORD32 pred_strd,

                          WORD32 dst_strd,

                          CHROMA_PLANE_ID_T e_chroma_plane)

{

    WORD32 i;

    WORD32 e[2], o[2];

    WORD32 add, shift;

    WORD32 trans_size;

    WORD32 *pi4_tmp_orig;

    WORD16 *pi2_dst_orig;

    UWORD32 u4_blk_sad=0;

    WORD32 chroma_flag = 0;

    if (e_chroma_plane != NULL_PLANE)

    {

        chroma_flag = 1;

        pu1_src += e_chroma_plane;

        pu1_pred += e_chroma_plane;

    }

    pi2_dst_orig = pi2_dst;

    pi4_tmp_orig = pi4_temp;

    trans_size = TRANS_SIZE_4;

    /* Residue + Forward Transform 1st stage */

    shift = 1; // log2(iWidth) - 1 + g_uiBitIncrement

    add = 1 << (shift - 1);

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

    {

        WORD32 resi_tmp_1, resi_tmp_2;

        /* e and o */

        resi_tmp_1 = pu1_src[0 + 0*chroma_flag] - pu1_pred[0 + 0*chroma_flag];

        resi_tmp_2 = pu1_src[3 + 3*chroma_flag] - pu1_pred[3 + 3*chroma_flag];

        e[0] = resi_tmp_1 + resi_tmp_2;

        o[0] = resi_tmp_1 - resi_tmp_2;

        u4_blk_sad += abs(resi_tmp_1);

        u4_blk_sad += abs(resi_tmp_2);

        resi_tmp_1 = pu1_src[1 + 1*chroma_flag] - pu1_pred[1 + 1*chroma_flag];

        resi_tmp_2 = pu1_src[2 + 2*chroma_flag] - pu1_pred[2 + 2*chroma_flag];

        e[1] = resi_tmp_1 + resi_tmp_2;

        o[1] = resi_tmp_1 - resi_tmp_2;

        u4_blk_sad += abs(resi_tmp_1);

        u4_blk_sad += abs(resi_tmp_2);

        pi4_temp[0] = (g_ai2_ihevc_trans_4[0][0] * e[0]

                        + g_ai2_ihevc_trans_4[0][1] * e[1]);// + add) >> shift;

        pi4_temp[2 * trans_size] = (g_ai2_ihevc_trans_4[2][0] * e[0]

                        + g_ai2_ihevc_trans_4[2][1] * e[1]);// + add) >> shift;

        pi4_temp[trans_size] = (g_ai2_ihevc_trans_4[1][0] * o[0]

                        + g_ai2_ihevc_trans_4[1][1] * o[1]);// + add) >> shift;

        pi4_temp[3 * trans_size] = (g_ai2_ihevc_trans_4[3][0] * o[0]

                        + g_ai2_ihevc_trans_4[3][1] * o[1]);// + add) >> shift;

        pu1_src += src_strd;

        pu1_pred += pred_strd;

        pi4_temp++;

    }

    pi4_temp = pi4_tmp_orig;

    /* Forward Transform 2nd stage */

    shift = 9; // log2(iHeight) + 6

    add = 1 << (shift - 1);

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

    {

        /* e and o */

        e[0] = pi4_temp[0] + pi4_temp[3];

        o[0] = pi4_temp[0] - pi4_temp[3];

        e[1] = pi4_temp[1] + pi4_temp[2];

        o[1] = pi4_temp[1] - pi4_temp[2];

        pi2_dst[0] = (g_ai2_ihevc_trans_4[0][0] * e[0]

                        + g_ai2_ihevc_trans_4[0][1] * e[1] + add) >> shift;

        pi2_dst[2 * dst_strd] = (g_ai2_ihevc_trans_4[2][0] * e[0]

                        + g_ai2_ihevc_trans_4[2][1] * e[1] + add) >> shift;

        pi2_dst[dst_strd] = (g_ai2_ihevc_trans_4[1][0] * o[0]

                        + g_ai2_ihevc_trans_4[1][1] * o[1] + add) >> shift;

        pi2_dst[3 * dst_strd] = (g_ai2_ihevc_trans_4[3][0] * o[0]

                        + g_ai2_ihevc_trans_4[3][1] * o[1] + add) >> shift;

        pi4_temp += trans_size;

        pi2_dst++;

    }

    return u4_blk_sad;

}

void ihevc_resi_trans_4x4_16bit(WORD16 *pi2_src,

                          UWORD8 *pu1_pred,

                          WORD16 *pi2_tmp,

                          WORD16 *pi2_dst,

                          WORD32 src_strd,

                          WORD32 pred_strd,

                          WORD32 dst_strd)

{

    WORD32 i;

    WORD32 e[2], o[2];

    WORD32 add, shift;

    WORD32 trans_size;

    WORD16 *pi2_tmp_orig;

    WORD16 *pi2_dst_orig;

    pi2_dst_orig = pi2_dst;

    pi2_tmp_orig = pi2_tmp;

    trans_size = TRANS_SIZE_4;

    /* Residue + Forward Transform 1st stage */

    shift = 1; // log2(iWidth) - 1 + g_uiBitIncrement

    add = 1 << (shift - 1);

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

    {

        WORD32 resi_tmp_1, resi_tmp_2;

        /* e and o */

        resi_tmp_1 = pi2_src[0] - pu1_pred[0];

        resi_tmp_2 = pi2_src[3] - pu1_pred[3];

        e[0] = resi_tmp_1 + resi_tmp_2;

        o[0] = resi_tmp_1 - resi_tmp_2;

        resi_tmp_1 = pi2_src[1] - pu1_pred[1];

        resi_tmp_2 = pi2_src[2] - pu1_pred[2];

        e[1] = resi_tmp_1 + resi_tmp_2;

        o[1] = resi_tmp_1 - resi_tmp_2;

        pi2_tmp[0] = (g_ai2_ihevc_trans_4[0][0] * e[0]

                        + g_ai2_ihevc_trans_4[0][1] * e[1] + add) >> shift;

        pi2_tmp[2 * trans_size] = (g_ai2_ihevc_trans_4[2][0] * e[0]

                        + g_ai2_ihevc_trans_4[2][1] * e[1] + add) >> shift;

        pi2_tmp[trans_size] = (g_ai2_ihevc_trans_4[1][0] * o[0]

                        + g_ai2_ihevc_trans_4[1][1] * o[1] + add) >> shift;

        pi2_tmp[3 * trans_size] = (g_ai2_ihevc_trans_4[3][0] * o[0]

                        + g_ai2_ihevc_trans_4[3][1] * o[1] + add) >> shift;

        pi2_src += src_strd;

        pu1_pred += pred_strd;

        pi2_tmp++;

    }

    pi2_tmp = pi2_tmp_orig;

    /* Forward Transform 2nd stage */

    shift = 8; // log2(iHeight) + 6

    add = 1 << (shift - 1);

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

    {

        /* e and o */

        e[0] = pi2_tmp[0] + pi2_tmp[3];

        o[0] = pi2_tmp[0] - pi2_tmp[3];

        e[1] = pi2_tmp[1] + pi2_tmp[2];

        o[1] = pi2_tmp[1] - pi2_tmp[2];

        pi2_dst[0] = (g_ai2_ihevc_trans_4[0][0] * e[0]

                        + g_ai2_ihevc_trans_4[0][1] * e[1] + add) >> shift;

        pi2_dst[2 * dst_strd] = (g_ai2_ihevc_trans_4[2][0] * e[0]

                        + g_ai2_ihevc_trans_4[2][1] * e[1] + add) >> shift;

        pi2_dst[dst_strd] = (g_ai2_ihevc_trans_4[1][0] * o[0]

                        + g_ai2_ihevc_trans_4[1][1] * o[1] + add) >> shift;

        pi2_dst[3 * dst_strd] = (g_ai2_ihevc_trans_4[3][0] * o[0]

                        + g_ai2_ihevc_trans_4[3][1] * o[1] + add) >> shift;

        pi2_tmp += trans_size;

        pi2_dst++;

    }

}

/**

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

 *

 * @brief

 *  This function performs residue calculation and forward  transform on

 * input pixels

 *

 * @par Description:

 *  Performs residue calculation by subtracting source and  prediction and

 * followed by forward transform

 *

 * @param[in] pu1_src

 *  Input 8x8 pixels

 *

 * @param[in] pu1_pred

 *  Prediction data

 *

 * @param[in] pi2_tmp

 *  Temporary buffer of size 8x8

 *

 * @param[out] pi2_dst

 *  Output 8x8 coefficients

 *

 * @param[in] src_strd

 *  Input stride

 *

 * @param[in] pred_strd

 *  Prediction Stride

 *

 * @param[in] dst_strd

 *  Output Stride

 *

 * @param[in] e_chroma_plane

 *  Enum singalling chroma plane

 *

 * @returns  Void

 *

 * @remarks

 *  None

 *

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

 */

UWORD32 ihevc_resi_trans_8x8(UWORD8 *pu1_src,

                          UWORD8 *pu1_pred,

                          WORD32 *pi4_temp,

                          WORD16 *pi2_dst,

                          WORD32 src_strd,

                          WORD32 pred_strd,

                          WORD32 dst_strd,

                          CHROMA_PLANE_ID_T e_chroma_plane)

{

    WORD32 i, k;

    WORD32 e[4], o[4];

    WORD32 ee[2], eo[2];

    WORD32 add, shift;

    WORD32 trans_size;

    WORD32 *pi4_tmp_orig;

//    WORD16 *pi2_tmp;

    WORD16 *pi2_dst_orig;

    UWORD32 u4_blk_sad=0;

    WORD32 chroma_flag = 0;

    if (e_chroma_plane != NULL_PLANE)

    {

        chroma_flag = 1;

        pu1_src += e_chroma_plane;

        pu1_pred += e_chroma_plane;

    }

    pi2_dst_orig = pi2_dst;

    pi4_tmp_orig = pi4_temp;

    trans_size = TRANS_SIZE_8;

    /* Residue + Forward Transform 1st stage */

    shift = 2; // log2(iWidth) - 1 + g_uiBitIncrement

    add = 1 << (shift - 1);

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

    {

        WORD32 resi_tmp_1, resi_tmp_2;

        /* Utilizing symmetry properties to the maximum to minimize the number of multiplications */

        /* e and o*/

        for(k = 0; k < 4; k++)

        {

            resi_tmp_1 = pu1_src[k*(1+chroma_flag)] - pu1_pred[k*(1+chroma_flag)];

            resi_tmp_2 = pu1_src[(7-k)*(1+chroma_flag)] - pu1_pred[(7-k)*(1+chroma_flag)];

            e[k] = resi_tmp_1 + resi_tmp_2;

            o[k] = resi_tmp_1 - resi_tmp_2;

            u4_blk_sad += abs(resi_tmp_1) + abs(resi_tmp_2);

        }

        /* ee and eo */

        ee[0] = e[0] + e[3];

        eo[0] = e[0] - e[3];

        ee[1] = e[1] + e[2];

        eo[1] = e[1] - e[2];

        pi4_temp[0] = (g_ai2_ihevc_trans_8[0][0] * ee[0]

                        + g_ai2_ihevc_trans_8[0][1] * ee[1]);// + add) >> shift;

        pi4_temp[4 * trans_size] = (g_ai2_ihevc_trans_8[4][0] * ee[0]

                        + g_ai2_ihevc_trans_8[4][1] * ee[1]);// + add) >> shift;

        pi4_temp[2 * trans_size] = (g_ai2_ihevc_trans_8[2][0] * eo[0]

                        + g_ai2_ihevc_trans_8[2][1] * eo[1]);// + add) >> shift;

        pi4_temp[6 * trans_size] = (g_ai2_ihevc_trans_8[6][0] * eo[0]

                        + g_ai2_ihevc_trans_8[6][1] * eo[1]);// + add) >> shift;

        pi4_temp[trans_size] = (g_ai2_ihevc_trans_8[1][0] * o[0]

                        + g_ai2_ihevc_trans_8[1][1] * o[1]

                        + g_ai2_ihevc_trans_8[1][2] * o[2]

                        + g_ai2_ihevc_trans_8[1][3] * o[3]);// + add) >> shift;

        pi4_temp[3 * trans_size] = (g_ai2_ihevc_trans_8[3][0] * o[0]

                        + g_ai2_ihevc_trans_8[3][1] * o[1]

                        + g_ai2_ihevc_trans_8[3][2] * o[2]

                        + g_ai2_ihevc_trans_8[3][3] * o[3]);// + add) >> shift;

        pi4_temp[5 * trans_size] = (g_ai2_ihevc_trans_8[5][0] * o[0]

                        + g_ai2_ihevc_trans_8[5][1] * o[1]

                        + g_ai2_ihevc_trans_8[5][2] * o[2]

                        + g_ai2_ihevc_trans_8[5][3] * o[3]);// + add) >> shift;

        pi4_temp[7 * trans_size] = (g_ai2_ihevc_trans_8[7][0] * o[0]

                        + g_ai2_ihevc_trans_8[7][1] * o[1]

                        + g_ai2_ihevc_trans_8[7][2] * o[2]

                        + g_ai2_ihevc_trans_8[7][3] * o[3]);// + add) >> shift;

        pu1_src += src_strd;

        pu1_pred += pred_strd;

        pi4_temp++;

    }

    pi4_temp = pi4_tmp_orig;

    /* Forward Transform 2nd stage */

    shift = 11; // log2(iHeight) + 6

    add = 1 << (shift - 1);

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

    {

        /* Utilizing symmetry properties to the maximum to minimize the number of multiplications */

        /* e and o*/

        for(k = 0; k < 4; k++)

        {

            e[k] = pi4_temp[k] + pi4_temp[7 - k];

            o[k] = pi4_temp[k] - pi4_temp[7 - k];

        }

        /* ee and eo */

        ee[0] = e[0] + e[3];

        eo[0] = e[0] - e[3];

        ee[1] = e[1] + e[2];

        eo[1] = e[1] - e[2];

        pi2_dst[0] = (g_ai2_ihevc_trans_8[0][0] * ee[0]

                        + g_ai2_ihevc_trans_8[0][1] * ee[1] + add) >> shift;

        pi2_dst[4 * dst_strd] = (g_ai2_ihevc_trans_8[4][0] * ee[0]

                        + g_ai2_ihevc_trans_8[4][1] * ee[1] + add) >> shift;

        pi2_dst[2 * dst_strd] = (g_ai2_ihevc_trans_8[2][0] * eo[0]

                        + g_ai2_ihevc_trans_8[2][1] * eo[1] + add) >> shift;

        pi2_dst[6 * dst_strd] = (g_ai2_ihevc_trans_8[6][0] * eo[0]

                        + g_ai2_ihevc_trans_8[6][1] * eo[1] + add) >> shift;

        pi2_dst[dst_strd] = (g_ai2_ihevc_trans_8[1][0] * o[0]

                        + g_ai2_ihevc_trans_8[1][1] * o[1]

                        + g_ai2_ihevc_trans_8[1][2] * o[2]

                        + g_ai2_ihevc_trans_8[1][3] * o[3] + add) >> shift;

        pi2_dst[3 * dst_strd] = (g_ai2_ihevc_trans_8[3][0] * o[0]

                        + g_ai2_ihevc_trans_8[3][1] * o[1]

                        + g_ai2_ihevc_trans_8[3][2] * o[2]

                        + g_ai2_ihevc_trans_8[3][3] * o[3] + add) >> shift;

        pi2_dst[5 * dst_strd] = (g_ai2_ihevc_trans_8[5][0] * o[0]

                        + g_ai2_ihevc_trans_8[5][1] * o[1]

                        + g_ai2_ihevc_trans_8[5][2] * o[2]

                        + g_ai2_ihevc_trans_8[5][3] * o[3] + add) >> shift;

        pi2_dst[7 * dst_strd] = (g_ai2_ihevc_trans_8[7][0] * o[0]

                        + g_ai2_ihevc_trans_8[7][1] * o[1]

                        + g_ai2_ihevc_trans_8[7][2] * o[2]

                        + g_ai2_ihevc_trans_8[7][3] * o[3] + add) >> shift;

        pi4_temp += trans_size;

        pi2_dst++;

    }

    return u4_blk_sad;

}

void ihevc_resi_trans_8x8_16bit(WORD16 *pi2_src,

                          UWORD8 *pu1_pred,

                          WORD16 *pi2_tmp,

                          WORD16 *pi2_dst,

                          WORD32 src_strd,

                          WORD32 pred_strd,

                          WORD32 dst_strd)

{

    WORD32 i, k;

    WORD32 e[4], o[4];

    WORD32 ee[2], eo[2];

    WORD32 add, shift;

    WORD32 trans_size;

    WORD16 *pi2_tmp_orig;

    WORD16 *pi2_dst_orig;

    pi2_dst_orig = pi2_dst;

    pi2_tmp_orig = pi2_tmp;

    trans_size = TRANS_SIZE_8;

    /* Residue + Forward Transform 1st stage */

    shift = 2; // log2(iWidth) - 1 + g_uiBitIncrement

    add = 1 << (shift - 1);

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

    {

        WORD32 resi_tmp_1, resi_tmp_2;

        /* Utilizing symmetry properties to the maximum to minimize the number of multiplications */

        /* e and o*/

        for(k = 0; k < 4; k++)

        {

            resi_tmp_1 = pi2_src[k] - pu1_pred[k];

            resi_tmp_2 = pi2_src[7 - k] - pu1_pred[7 - k];

            e[k] = resi_tmp_1 + resi_tmp_2;

            o[k] = resi_tmp_1 - resi_tmp_2;

        }

        /* ee and eo */

        ee[0] = e[0] + e[3];

        eo[0] = e[0] - e[3];

        ee[1] = e[1] + e[2];

        eo[1] = e[1] - e[2];

        pi2_tmp[0] = (g_ai2_ihevc_trans_8[0][0] * ee[0]

                        + g_ai2_ihevc_trans_8[0][1] * ee[1] + add) >> shift;

        pi2_tmp[4 * trans_size] = (g_ai2_ihevc_trans_8[4][0] * ee[0]

                        + g_ai2_ihevc_trans_8[4][1] * ee[1] + add) >> shift;

        pi2_tmp[2 * trans_size] = (g_ai2_ihevc_trans_8[2][0] * eo[0]

                        + g_ai2_ihevc_trans_8[2][1] * eo[1] + add) >> shift;

        pi2_tmp[6 * trans_size] = (g_ai2_ihevc_trans_8[6][0] * eo[0]

                        + g_ai2_ihevc_trans_8[6][1] * eo[1] + add) >> shift;

        pi2_tmp[trans_size] = (g_ai2_ihevc_trans_8[1][0] * o[0]

                        + g_ai2_ihevc_trans_8[1][1] * o[1]

                        + g_ai2_ihevc_trans_8[1][2] * o[2]

                        + g_ai2_ihevc_trans_8[1][3] * o[3] + add) >> shift;

        pi2_tmp[3 * trans_size] = (g_ai2_ihevc_trans_8[3][0] * o[0]

                        + g_ai2_ihevc_trans_8[3][1] * o[1]

                        + g_ai2_ihevc_trans_8[3][2] * o[2]

                        + g_ai2_ihevc_trans_8[3][3] * o[3] + add) >> shift;

        pi2_tmp[5 * trans_size] = (g_ai2_ihevc_trans_8[5][0] * o[0]

                        + g_ai2_ihevc_trans_8[5][1] * o[1]

                        + g_ai2_ihevc_trans_8[5][2] * o[2]

                        + g_ai2_ihevc_trans_8[5][3] * o[3] + add) >> shift;

        pi2_tmp[7 * trans_size] = (g_ai2_ihevc_trans_8[7][0] * o[0]

                        + g_ai2_ihevc_trans_8[7][1] * o[1]

                        + g_ai2_ihevc_trans_8[7][2] * o[2]

                        + g_ai2_ihevc_trans_8[7][3] * o[3] + add) >> shift;

        pi2_src += src_strd;

        pu1_pred += pred_strd;

        pi2_tmp++;

    }

    pi2_tmp = pi2_tmp_orig;

    /* Forward Transform 2nd stage */

    shift = 9; // log2(iHeight) + 6

    add = 1 << (shift - 1);

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

    {

        /* Utilizing symmetry properties to the maximum to minimize the number of multiplications */

        /* e and o*/

        for(k = 0; k < 4; k++)

        {

            e[k] = pi2_tmp[k] + pi2_tmp[7 - k];

            o[k] = pi2_tmp[k] - pi2_tmp[7 - k];

        }

        /* ee and eo */

        ee[0] = e[0] + e[3];

        eo[0] = e[0] - e[3];

        ee[1] = e[1] + e[2];

        eo[1] = e[1] - e[2];

        pi2_dst[0] = (g_ai2_ihevc_trans_8[0][0] * ee[0]

                        + g_ai2_ihevc_trans_8[0][1] * ee[1] + add) >> shift;

        pi2_dst[4 * dst_strd] = (g_ai2_ihevc_trans_8[4][0] * ee[0]

                        + g_ai2_ihevc_trans_8[4][1] * ee[1] + add) >> shift;

        pi2_dst[2 * dst_strd] = (g_ai2_ihevc_trans_8[2][0] * eo[0]

                        + g_ai2_ihevc_trans_8[2][1] * eo[1] + add) >> shift;

        pi2_dst[6 * dst_strd] = (g_ai2_ihevc_trans_8[6][0] * eo[0]

                        + g_ai2_ihevc_trans_8[6][1] * eo[1] + add) >> shift;

        pi2_dst[dst_strd] = (g_ai2_ihevc_trans_8[1][0] * o[0]

                        + g_ai2_ihevc_trans_8[1][1] * o[1]

                        + g_ai2_ihevc_trans_8[1][2] * o[2]

                        + g_ai2_ihevc_trans_8[1][3] * o[3] + add) >> shift;

        pi2_dst[3 * dst_strd] = (g_ai2_ihevc_trans_8[3][0] * o[0]

                        + g_ai2_ihevc_trans_8[3][1] * o[1]

                        + g_ai2_ihevc_trans_8[3][2] * o[2]

                        + g_ai2_ihevc_trans_8[3][3] * o[3] + add) >> shift;

        pi2_dst[5 * dst_strd] = (g_ai2_ihevc_trans_8[5][0] * o[0]

                        + g_ai2_ihevc_trans_8[5][1] * o[1]

                        + g_ai2_ihevc_trans_8[5][2] * o[2]

                        + g_ai2_ihevc_trans_8[5][3] * o[3] + add) >> shift;

        pi2_dst[7 * dst_strd] = (g_ai2_ihevc_trans_8[7][0] * o[0]

                        + g_ai2_ihevc_trans_8[7][1] * o[1]

                        + g_ai2_ihevc_trans_8[7][2] * o[2]

                        + g_ai2_ihevc_trans_8[7][3] * o[3] + add) >> shift;

        pi2_tmp += trans_size;

        pi2_dst++;

    }

}

/**

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

 *

 * @brief

 *  This function performs residue calculation and forward  transform on

 * input pixels

 *

 * @par Description:

 *  Performs residue calculation by subtracting source and  prediction and

 * followed by forward transform

 *

 * @param[in] pu1_src

 *  Input 16x16 pixels

 *

 * @param[in] pu1_pred

 *  Prediction data

 *

 * @param[in] pi2_tmp

 *  Temporary buffer of size 16x16

 *

 * @param[out] pi2_dst

 *  Output 16x16 coefficients

 *

 * @param[in] src_strd

 *  Input stride

 *

 * @param[in] pred_strd

 *  Prediction Stride

 *

 * @param[in] dst_strd

 *  Output Stride

 *

 * @param[in] e_chroma_plane

 *  Enum singalling chroma plane

 *

 * @returns  Void

 *

 * @remarks

 *  None

 *

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

 */

UWORD32 ihevc_resi_trans_16x16(UWORD8 *pu1_src,

                            UWORD8 *pu1_pred,

                            WORD32 *pi4_temp,

                            WORD16 *pi2_dst,

                            WORD32 src_strd,

                            WORD32 pred_strd,

                            WORD32 dst_strd,

                            CHROMA_PLANE_ID_T e_chroma_plane)

{

    WORD32 i, k;

    WORD32 e[8], o[8];

    WORD32 ee[4], eo[4];

    WORD32 eee[2], eeo[2];

    WORD32 add, shift;

    WORD32 trans_size;

    WORD32 *pi4_tmp_orig;

    WORD16 *pi2_dst_orig;

    UWORD32 u4_blk_sad = 0;

    WORD32 chroma_flag = 0;

    if (e_chroma_plane != NULL_PLANE)

    {

        chroma_flag = 1;

        pu1_src += e_chroma_plane;

        pu1_pred += e_chroma_plane;

    }

    pi2_dst_orig = pi2_dst;

    pi4_tmp_orig = pi4_temp;

    trans_size = TRANS_SIZE_16;

    /* Residue + Forward Transform 1st stage */

    shift = 3; // log2(iWidth) - 1 + g_uiBitIncrement

    add = 1 << (shift - 1);

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

    {

        WORD32 resi_tmp_1, resi_tmp_2;

        /* e and o*/

        for(k = 0; k < 8; k++)

        {

            resi_tmp_1 = pu1_src[k*(1+chroma_flag)] - pu1_pred[k*(1+chroma_flag)];

            resi_tmp_2 = pu1_src[(15-k)*(1+chroma_flag)] - pu1_pred[(15-k)*(1+chroma_flag)];

            e[k] = resi_tmp_1 + resi_tmp_2;

            o[k] = resi_tmp_1 - resi_tmp_2;

            u4_blk_sad += abs(resi_tmp_1) + abs(resi_tmp_2);

        }

        /* ee and eo */

        for(k = 0; k < 4; k++)

        {

            ee[k] = e[k] + e[7 - k];

            eo[k] = e[k] - e[7 - k];

        }

        /* eee and eeo */

        eee[0] = ee[0] + ee[3];

        eeo[0] = ee[0] - ee[3];

        eee[1] = ee[1] + ee[2];

        eeo[1] = ee[1] - ee[2];

        pi4_temp[0] = (g_ai2_ihevc_trans_16[0][0] * eee[0]

                        + g_ai2_ihevc_trans_16[0][1] * eee[1]);// + add) >> shift;

        pi4_temp[8 * trans_size] = (g_ai2_ihevc_trans_16[8][0] * eee[0]

                        + g_ai2_ihevc_trans_16[8][1] * eee[1]);// + add) >> shift;

        pi4_temp[4 * trans_size] = (g_ai2_ihevc_trans_16[4][0] * eeo[0]

                        + g_ai2_ihevc_trans_16[4][1] * eeo[1]);// + add) >> shift;

        pi4_temp[12 * trans_size] = (g_ai2_ihevc_trans_16[12][0] * eeo[0]

                        + g_ai2_ihevc_trans_16[12][1] * eeo[1]);// + add) >> shift;

        for(k = 2; k < 16; k += 4)

        {

            pi4_temp[k * trans_size] = (g_ai2_ihevc_trans_16[k][0] * eo[0]

                            + g_ai2_ihevc_trans_16[k][1] * eo[1]

                            + g_ai2_ihevc_trans_16[k][2] * eo[2]

                            + g_ai2_ihevc_trans_16[k][3] * eo[3]);// + add)>> shift;

        }

        for(k = 1; k < 16; k += 2)

        {

            pi4_temp[k * trans_size] = (g_ai2_ihevc_trans_16[k][0] * o[0]

                            + g_ai2_ihevc_trans_16[k][1] * o[1]

                            + g_ai2_ihevc_trans_16[k][2] * o[2]

                            + g_ai2_ihevc_trans_16[k][3] * o[3]

                            + g_ai2_ihevc_trans_16[k][4] * o[4]

                            + g_ai2_ihevc_trans_16[k][5] * o[5]

                            + g_ai2_ihevc_trans_16[k][6] * o[6]

                            + g_ai2_ihevc_trans_16[k][7] * o[7]);// + add) >> shift;

        }

        pu1_src += src_strd;

        pu1_pred += pred_strd;

        pi4_temp++;

    }

    pi4_temp = pi4_tmp_orig;

    /* Forward Transform 2nd stage */

    shift = 13; // log2(iHeight) + 6

    add = 1 << (shift - 1);

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

    {

        /* e and o*/

        for(k = 0; k < 8; k++)

        {

            e[k] = pi4_temp[k] + pi4_temp[15 - k];

            o[k] = pi4_temp[k] - pi4_temp[15 - k];

        }

        /* ee and eo */

        for(k = 0; k < 4; k++)

        {

            ee[k] = e[k] + e[7 - k];

            eo[k] = e[k] - e[7 - k];

        }

        /* eee and eeo */

        eee[0] = ee[0] + ee[3];

        eeo[0] = ee[0] - ee[3];

        eee[1] = ee[1] + ee[2];

        eeo[1] = ee[1] - ee[2];

        pi2_dst[0] = (g_ai2_ihevc_trans_16[0][0] * eee[0]

                        + g_ai2_ihevc_trans_16[0][1] * eee[1] + add) >> shift;

        pi2_dst[8 * dst_strd] = (g_ai2_ihevc_trans_16[8][0] * eee[0]

                        + g_ai2_ihevc_trans_16[8][1] * eee[1] + add) >> shift;

        pi2_dst[4 * dst_strd] = (g_ai2_ihevc_trans_16[4][0] * eeo[0]

                        + g_ai2_ihevc_trans_16[4][1] * eeo[1] + add) >> shift;

        pi2_dst[12 * dst_strd] = (g_ai2_ihevc_trans_16[12][0] * eeo[0]

                        + g_ai2_ihevc_trans_16[12][1] * eeo[1] + add) >> shift;

        for(k = 2; k < 16; k += 4)

        {

            pi2_dst[k * dst_strd] = (g_ai2_ihevc_trans_16[k][0] * eo[0]

                            + g_ai2_ihevc_trans_16[k][1] * eo[1]

                            + g_ai2_ihevc_trans_16[k][2] * eo[2]

                            + g_ai2_ihevc_trans_16[k][3] * eo[3] + add)

                            >> shift;

        }

        for(k = 1; k < 16; k += 2)

        {

            pi2_dst[k * dst_strd] = (g_ai2_ihevc_trans_16[k][0] * o[0]

                            + g_ai2_ihevc_trans_16[k][1] * o[1]

                            + g_ai2_ihevc_trans_16[k][2] * o[2]

                            + g_ai2_ihevc_trans_16[k][3] * o[3]

                            + g_ai2_ihevc_trans_16[k][4] * o[4]

                            + g_ai2_ihevc_trans_16[k][5] * o[5]

                            + g_ai2_ihevc_trans_16[k][6] * o[6]

                            + g_ai2_ihevc_trans_16[k][7] * o[7] + add) >> shift;

        }

        pi4_temp += trans_size;

        pi2_dst++;

    }

    return u4_blk_sad;

}

void ihevc_resi_trans_16x16_16bit(WORD16 *pi2_src,

                            UWORD8 *pu1_pred,

                            WORD16 *pi2_tmp,

                            WORD16 *pi2_dst,

                            WORD32 src_strd,

                            WORD32 pred_strd,

                            WORD32 dst_strd)

{

    WORD32 i, k;

    WORD32 e[8], o[8];

    WORD32 ee[4], eo[4];

    WORD32 eee[2], eeo[2];

    WORD32 add, shift;

    WORD32 trans_size;

    WORD16 *pi2_tmp_orig;

    WORD16 *pi2_dst_orig;

    pi2_dst_orig = pi2_dst;

    pi2_tmp_orig = pi2_tmp;

    trans_size = TRANS_SIZE_16;

    /* Residue + Forward Transform 1st stage */

    shift = 3; // log2(iWidth) - 1 + g_uiBitIncrement

    add = 1 << (shift - 1);

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

    {

        WORD32 resi_tmp_1, resi_tmp_2;

        /* e and o*/

        for(k = 0; k < 8; k++)

        {

            resi_tmp_1 = pi2_src[k] - pu1_pred[k];

            resi_tmp_2 = pi2_src[15 - k] - pu1_pred[15 - k];

            e[k] = resi_tmp_1 + resi_tmp_2;

            o[k] = resi_tmp_1 - resi_tmp_2;

        }

        /* ee and eo */

        for(k = 0; k < 4; k++)

        {

            ee[k] = e[k] + e[7 - k];

            eo[k] = e[k] - e[7 - k];

        }

        /* eee and eeo */

        eee[0] = ee[0] + ee[3];

        eeo[0] = ee[0] - ee[3];

        eee[1] = ee[1] + ee[2];

        eeo[1] = ee[1] - ee[2];

        pi2_tmp[0] = (g_ai2_ihevc_trans_16[0][0] * eee[0]

                        + g_ai2_ihevc_trans_16[0][1] * eee[1] + add) >> shift;

        pi2_tmp[8 * trans_size] = (g_ai2_ihevc_trans_16[8][0] * eee[0]

                        + g_ai2_ihevc_trans_16[8][1] * eee[1] + add) >> shift;

        pi2_tmp[4 * trans_size] = (g_ai2_ihevc_trans_16[4][0] * eeo[0]

                        + g_ai2_ihevc_trans_16[4][1] * eeo[1] + add) >> shift;

        pi2_tmp[12 * trans_size] = (g_ai2_ihevc_trans_16[12][0] * eeo[0]

                        + g_ai2_ihevc_trans_16[12][1] * eeo[1] + add) >> shift;

        for(k = 2; k < 16; k += 4)

        {

            pi2_tmp[k * trans_size] = (g_ai2_ihevc_trans_16[k][0] * eo[0]

                            + g_ai2_ihevc_trans_16[k][1] * eo[1]

                            + g_ai2_ihevc_trans_16[k][2] * eo[2]

                            + g_ai2_ihevc_trans_16[k][3] * eo[3] + add)

                            >> shift;

        }

        for(k = 1; k < 16; k += 2)

        {

            pi2_tmp[k * trans_size] = (g_ai2_ihevc_trans_16[k][0] * o[0]

                            + g_ai2_ihevc_trans_16[k][1] * o[1]

                            + g_ai2_ihevc_trans_16[k][2] * o[2]

                            + g_ai2_ihevc_trans_16[k][3] * o[3]

                            + g_ai2_ihevc_trans_16[k][4] * o[4]

                            + g_ai2_ihevc_trans_16[k][5] * o[5]

                            + g_ai2_ihevc_trans_16[k][6] * o[6]

                            + g_ai2_ihevc_trans_16[k][7] * o[7] + add) >> shift;

        }

        pi2_src += src_strd;

        pu1_pred += pred_strd;

        pi2_tmp++;

    }

    pi2_tmp = pi2_tmp_orig;

    /* Forward Transform 2nd stage */

    shift = 10; // log2(iHeight) + 6

    add = 1 << (shift - 1);

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

    {

        /* e and o*/

        for(k = 0; k < 8; k++)

        {

            e[k] = pi2_tmp[k] + pi2_tmp[15 - k];

            o[k] = pi2_tmp[k] - pi2_tmp[15 - k];

        }

        /* ee and eo */

        for(k = 0; k < 4; k++)

        {