| /****************************************************************************** |
| * |
| * 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_func_selector.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++) |
| { |
| 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; |
| } |
| |
| 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 32x32 pixels |
| * |
| * @param[in] pu1_pred |
| * Prediction data |
| * |
| * @param[in] pi2_tmp |
| * Temporary buffer of size 32x32 |
| * |
| * @param[out] pi2_dst |
| * Output 32x32 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_32x32(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[16], o[16]; |
| WORD32 ee[8], eo[8]; |
| WORD32 eee[4], eeo[4]; |
| WORD32 eeee[2], eeeo[2]; |
| 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_32; |
| /* Residue + Forward Transform 1st stage */ |
| /* Made to zero to match with intrinsics */ |
| shift = 0; // 4 : log2(iWidth) - 1 + g_uiBitIncrement |
| add = 0 ; //1 << (shift - 1); |
| |
| for(i = 0; i < trans_size; i++) |
| { |
| WORD32 resi_tmp_1, resi_tmp_2; |
| /* e and o*/ |
| for(k = 0; k < 16; k++) |
| { |
| resi_tmp_1 = pu1_src[k] - pu1_pred[k]; |
| resi_tmp_2 = pu1_src[31 - k] - pu1_pred[31 - k]; |
| 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 < 8; k++) |
| { |
| ee[k] = e[k] + e[15 - k]; |
| eo[k] = e[k] - e[15 - k]; |
| } |
| /* eee and eeo */ |
| for(k = 0; k < 4; k++) |
| { |
| eee[k] = ee[k] + ee[7 - k]; |
| eeo[k] = ee[k] - ee[7 - k]; |
| } |
| /* eeee and eeeo */ |
| eeee[0] = eee[0] + eee[3]; |
| eeeo[0] = eee[0] - eee[3]; |
| eeee[1] = eee[1] + eee[2]; |
| eeeo[1] = eee[1] - eee[2]; |
| |
| pi4_temp[0] = (g_ai2_ihevc_trans_32[0][0] * eeee[0] |
| + g_ai2_ihevc_trans_32[0][1] * eeee[1]);// + add) >> shift; |
| pi4_temp[16 * trans_size] = (g_ai2_ihevc_trans_32[16][0] * eeee[0] |
| + g_ai2_ihevc_trans_32[16][1] * eeee[1]);// + add) >> shift; |
| pi4_temp[8 * trans_size] = (g_ai2_ihevc_trans_32[8][0] * eeeo[0] |
| + g_ai2_ihevc_trans_32[8][1] * eeeo[1]);// + add) >> shift; |
| pi4_temp[24 * trans_size] = (g_ai2_ihevc_trans_32[24][0] * eeeo[0] |
| + g_ai2_ihevc_trans_32[24][1] * eeeo[1]);// + add) >> shift; |
| for(k = 4; k < 32; k += 8) |
| { |
| pi4_temp[k * trans_size] = (g_ai2_ihevc_trans_32[k][0] * eeo[0] |
| + g_ai2_ihevc_trans_32[k][1] * eeo[1] |
| + g_ai2_ihevc_trans_32[k][2] * eeo[2] |
| + g_ai2_ihevc_trans_32[k][3] * eeo[3]);// + add)>> shift; |
| } |
| for(k = 2; k < 32; k += 4) |
| { |
| pi4_temp[k * trans_size] = (g_ai2_ihevc_trans_32[k][0] * eo[0] |
| + g_ai2_ihevc_trans_32[k][1] * eo[1] |
| + g_ai2_ihevc_trans_32[k][2] * eo[2] |
| + g_ai2_ihevc_trans_32[k][3] * eo[3] |
| + g_ai2_ihevc_trans_32[k][4] * eo[4] |
| + g_ai2_ihevc_trans_32[k][5] * eo[5] |
| + g_ai2_ihevc_trans_32[k][6] * eo[6] |
| + g_ai2_ihevc_trans_32[k][7] * eo[7]);// + add)>> shift; |
| } |
| for(k = 1; k < 32; k += 2) |
| { |
| pi4_temp[k * trans_size] = (g_ai2_ihevc_trans_32[k][0] * o[0] |
| + g_ai2_ihevc_trans_32[k][1] * o[1] |
| + g_ai2_ihevc_trans_32[k][2] * o[2] |
| + g_ai2_ihevc_trans_32[k][3] * o[3] |
| + g_ai2_ihevc_trans_32[k][4] * o[4] |
| + g_ai2_ihevc_trans_32[k][5] * o[5] |
| + g_ai2_ihevc_trans_32[k][6] * o[6] |
| + g_ai2_ihevc_trans_32[k][7] * o[7] |
| + g_ai2_ihevc_trans_32[k][8] * o[8] |
| + g_ai2_ihevc_trans_32[k][9] * o[9] |
| + g_ai2_ihevc_trans_32[k][10] * o[10] |
| + g_ai2_ihevc_trans_32[k][11] * o[11] |
| + g_ai2_ihevc_trans_32[k][12] * o[12] |
| + g_ai2_ihevc_trans_32[k][13] * o[13] |
| + g_ai2_ihevc_trans_32[k][14] * o[14] |
| + g_ai2_ihevc_trans_32[k][15] * o[15]);// + add) >> shift; |
| } |
| pu1_src += src_strd; |
| pu1_pred += pred_strd; |
| pi4_temp++; |
| } |
| |
| pi4_temp = pi4_tmp_orig; |
| /* Forward Transform 2nd stage */ |
| shift = 15; // log2(iHeight) + 6 |
| add = 1 << (shift - 1); |
| |
| for(i = 0; i < TRANS_SIZE_32; i++) |
| { |
| /* e and o*/ |
| for(k = 0; k < 16; k++) |
| { |
| e[k] = pi4_temp[k] + pi4_temp[31 - k]; |
| o[k] = pi4_temp[k] - pi4_temp[31 - k]; |
| } |
| /* ee and eo */ |
| for(k = 0; k < 8; k++) |
| { |
| ee[k] = e[k] + e[15 - k]; |
| eo[k] = e[k] - e[15 - k]; |
| } |
| /* eee and eeo */ |
| for(k = 0; k < 4; k++) |
| { |
| eee[k] = ee[k] + ee[7 - k]; |
| eeo[k] = ee[k] - ee[7 - k]; |
| } |
| /* eeee and eeeo */ |
| eeee[0] = eee[0] + eee[3]; |
| eeeo[0] = eee[0] - eee[3]; |
| eeee[1] = eee[1] + eee[2]; |
| eeeo[1] = eee[1] - eee[2]; |
| |
| pi2_dst[0] = (g_ai2_ihevc_trans_32[0][0] * eeee[0] |
| + g_ai2_ihevc_trans_32[0][1] * eeee[1] + add) >> shift; |
| pi2_dst[16 * dst_strd] = (g_ai2_ihevc_trans_32[16][0] * eeee[0] |
| + g_ai2_ihevc_trans_32[16][1] * eeee[1] + add) >> shift; |
| pi2_dst[8 * dst_strd] = (g_ai2_ihevc_trans_32[8][0] * eeeo[0] |
| + g_ai2_ihevc_trans_32[8][1] * eeeo[1] + add) >> shift; |
| pi2_dst[24 * dst_strd] = (g_ai2_ihevc_trans_32[24][0] * eeeo[0] |
| + g_ai2_ihevc_trans_32[24][1] * eeeo[1] + add) >> shift; |
| for(k = 4; k < 32; k += 8) |
| { |
| pi2_dst[k * dst_strd] = (g_ai2_ihevc_trans_32[k][0] * eeo[0] |
| + g_ai2_ihevc_trans_32[k][1] * eeo[1] |
| + g_ai2_ihevc_trans_32[k][2] * eeo[2] |
| + g_ai2_ihevc_trans_32[k][3] * eeo[3] + add) |
| >> shift; |
| } |
| for(k = 2; k < 32; k += 4) |
| { |
| pi2_dst[k * dst_strd] = (g_ai2_ihevc_trans_32[k][0] * eo[0] |
| + g_ai2_ihevc_trans_32[k][1] * eo[1] |
| + g_ai2_ihevc_trans_32[k][2] * eo[2] |
| + g_ai2_ihevc_trans_32[k][3] * eo[3] |
| + g_ai2_ihevc_trans_32[k][4] * eo[4] |
| + g_ai2_ihevc_trans_32[k][5] * eo[5] |
| + g_ai2_ihevc_trans_32[k][6] * eo[6] |
| + g_ai2_ihevc_trans_32[k][7] * eo[7] + add) |
| >> shift; |
| } |
| for(k = 1; k < 32; k += 2) |
| { |
| pi2_dst[k * dst_strd] = (g_ai2_ihevc_trans_32[k][0] * o[0] |
| + g_ai2_ihevc_trans_32[k][1] * o[1] |
| + g_ai2_ihevc_trans_32[k][2] * o[2] |
| + g_ai2_ihevc_trans_32[k][3] * o[3] |
| + g_ai2_ihevc_trans_32[k][4] * o[4] |
| + g_ai2_ihevc_trans_32[k][5] * o[5] |
| + g_ai2_ihevc_trans_32[k][6] * o[6] |
| + g_ai2_ihevc_trans_32[k][7] * o[7] |
| + g_ai2_ihevc_trans_32[k][8] * o[8] |
| + g_ai2_ihevc_trans_32[k][9] * o[9] |
| + g_ai2_ihevc_trans_32[k][10] * o[10] |
| + g_ai2_ihevc_trans_32[k][11] * o[11] |
| + g_ai2_ihevc_trans_32[k][12] * o[12] |
| + g_ai2_ihevc_trans_32[k][13] * o[13] |
| + g_ai2_ihevc_trans_32[k][14] * o[14] |
| + g_ai2_ihevc_trans_32[k][15] * o[15] + add) |
| >> shift; |
| } |
| |
| pi4_temp += trans_size; |
| pi2_dst++; |
| } |
| |
| return u4_blk_sad; |
| } |
| |
| |
| |
| void ihevc_resi_trans_32x32_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[16], o[16]; |
| WORD32 ee[8], eo[8]; |
| WORD32 eee[4], eeo[4]; |
| WORD32 eeee[2], eeeo[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_32; |
| /* Residue + Forward Transform 1st stage */ |
| shift = 4; // 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 < 16; k++) |
| { |
| resi_tmp_1 = pi2_src[k] - pu1_pred[k]; |
| resi_tmp_2 = pi2_src[31 - k] - pu1_pred[31 - k]; |
| e[k] = resi_tmp_1 + resi_tmp_2; |
| o[k] = resi_tmp_1 - resi_tmp_2; |
| } |
| /* ee and eo */ |
| for(k = 0; k < 8; k++) |
| { |
| ee[k] = e[k] + e[15 - k]; |
| eo[k] = e[k] - e[15 - k]; |
| } |
| /* eee and eeo */ |
| for(k = 0; k < 4; k++) |
| { |
| eee[k] = ee[k] + ee[7 - k]; |
| eeo[k] = ee[k] - ee[7 - k]; |
| } |
| /* eeee and eeeo */ |
| eeee[0] = eee[0] + eee[3]; |
| eeeo[0] = eee[0] - eee[3]; |
| eeee[1] = eee[1] + eee[2]; |
| eeeo[1] = eee[1] - eee[2]; |
| |
| pi2_tmp[0] = (g_ai2_ihevc_trans_32[0][0] * eeee[0] |
| + g_ai2_ihevc_trans_32[0][1] * eeee[1] + add) >> shift; |
| pi2_tmp[16 * trans_size] = (g_ai2_ihevc_trans_32[16][0] * eeee[0] |
| + g_ai2_ihevc_trans_32[16][1] * eeee[1] + add) >> shift; |
| pi2_tmp[8 * trans_size] = (g_ai2_ihevc_trans_32[8][0] * eeeo[0] |
| + g_ai2_ihevc_trans_32[8][1] * eeeo[1] + add) >> shift; |
| pi2_tmp[24 * trans_size] = (g_ai2_ihevc_trans_32[24][0] * eeeo[0] |
| + g_ai2_ihevc_trans_32[24][1] * eeeo[1] + add) >> shift; |
| for(k = 4; k < 32; k += 8) |
| { |
| pi2_tmp[k * trans_size] = (g_ai2_ihevc_trans_32[k][0] * eeo[0] |
| + g_ai2_ihevc_trans_32[k][1] * eeo[1] |
| + g_ai2_ihevc_trans_32[k][2] * eeo[2] |
| + g_ai2_ihevc_trans_32[k][3] * eeo[3] + add) |
| >> shift; |
| } |
| for(k = 2; k < 32; k += 4) |
| { |
| pi2_tmp[k * trans_size] = (g_ai2_ihevc_trans_32[k][0] * eo[0] |
| + g_ai2_ihevc_trans_32[k][1] * eo[1] |
| + g_ai2_ihevc_trans_32[k][2] * eo[2] |
| + g_ai2_ihevc_trans_32[k][3] * eo[3] |
| + g_ai2_ihevc_trans_32[k][4] * eo[4] |
| + g_ai2_ihevc_trans_32[k][5] * eo[5] |
| + g_ai2_ihevc_trans_32[k][6] * eo[6] |
| + g_ai2_ihevc_trans_32[k][7] * eo[7] + add) |
| >> shift; |
| } |
| for(k = 1; k < 32; k += 2) |
| { |
| pi2_tmp[k * trans_size] = (g_ai2_ihevc_trans_32[k][0] * o[0] |
| + g_ai2_ihevc_trans_32[k][1] * o[1] |
| + g_ai2_ihevc_trans_32[k][2] * o[2] |
| + g_ai2_ihevc_trans_32[k][3] * o[3] |
| + g_ai2_ihevc_trans_32[k][4] * o[4] |
| + g_ai2_ihevc_trans_32[k][5] * o[5] |
| + g_ai2_ihevc_trans_32[k][6] * o[6] |
| + g_ai2_ihevc_trans_32[k][7] * o[7] |
| + g_ai2_ihevc_trans_32[k][8] * o[8] |
| + g_ai2_ihevc_trans_32[k][9] * o[9] |
| + g_ai2_ihevc_trans_32[k][10] * o[10] |
| + g_ai2_ihevc_trans_32[k][11] * o[11] |
| + g_ai2_ihevc_trans_32[k][12] * o[12] |
| + g_ai2_ihevc_trans_32[k][13] * o[13] |
| + g_ai2_ihevc_trans_32[k][14] * o[14] |
| + g_ai2_ihevc_trans_32[k][15] * o[15] + add) |
| >> shift; |
| } |
| pi2_src += src_strd; |
| pu1_pred += pred_strd; |
| pi2_tmp++; |
| } |
| |
| pi2_tmp = pi2_tmp_orig; |
| /* Forward Transform 2nd stage */ |
| shift = 11; // log2(iHeight) + 6 |
| add = 1 << (shift - 1); |
| |
| for(i = 0; i < TRANS_SIZE_32; i++) |
| { |
| /* e and o*/ |
| for(k = 0; k < 16; k++) |
| { |
| e[k] = pi2_tmp[k] + pi2_tmp[31 - k]; |
| o[k] = pi2_tmp[k] - pi2_tmp[31 - k]; |
| } |
| /* ee and eo */ |
| for(k = 0; k < 8; k++) |
| { |
| ee[k] = e[k] + e[15 - k]; |
| eo[k] = e[k] - e[15 - k]; |
| } |
| /* eee and eeo */ |
| for(k = 0; k < 4; k++) |
| { |
| eee[k] = ee[k] + ee[7 - k]; |
| eeo[k] = ee[k] - ee[7 - k]; |
| } |
| /* eeee and eeeo */ |
| eeee[0] = eee[0] + eee[3]; |
| eeeo[0] = eee[0] - eee[3]; |
| eeee[1] = eee[1] + eee[2]; |
| eeeo[1] = eee[1] - eee[2]; |
| |
| pi2_dst[0] = (g_ai2_ihevc_trans_32[0][0] * eeee[0] |
| + g_ai2_ihevc_trans_32[0][1] * eeee[1] + add) >> shift; |
| pi2_dst[16 * dst_strd] = (g_ai2_ihevc_trans_32[16][0] * eeee[0] |
| + g_ai2_ihevc_trans_32[16][1] * eeee[1] + add) >> shift; |
| pi2_dst[8 * dst_strd] = (g_ai2_ihevc_trans_32[8][0] * eeeo[0] |
| + g_ai2_ihevc_trans_32[8][1] * eeeo[1] + add) >> shift; |
| pi2_dst[24 * dst_strd] = (g_ai2_ihevc_trans_32[24][0] * eeeo[0] |
| + g_ai2_ihevc_trans_32[24][1] * eeeo[1] + add) >> shift; |
| for(k = 4; k < 32; k += 8) |
| { |
| pi2_dst[k * dst_strd] = (g_ai2_ihevc_trans_32[k][0] * eeo[0] |
| + g_ai2_ihevc_trans_32[k][1] * eeo[1] |
| + g_ai2_ihevc_trans_32[k][2] * eeo[2] |
| + g_ai2_ihevc_trans_32[k][3] * eeo[3] + add) |
| >> shift; |
| } |
| for(k = 2; k < 32; k += 4) |
| { |
| pi2_dst[k * dst_strd] = (g_ai2_ihevc_trans_32[k][0] * eo[0] |
| + g_ai2_ihevc_trans_32[k][1] * eo[1] |
| + g_ai2_ihevc_trans_32[k][2] * eo[2] |
| + g_ai2_ihevc_trans_32[k][3] * eo[3] |
| + g_ai2_ihevc_trans_32[k][4] * eo[4] |
| + g_ai2_ihevc_trans_32[k][5] * eo[5] |
| + g_ai2_ihevc_trans_32[k][6] * eo[6] |
| + g_ai2_ihevc_trans_32[k][7] * eo[7] + add) |
| >> shift; |
| } |
| for(k = 1; k < 32; k += 2) |
| { |
| pi2_dst[k * dst_strd] = (g_ai2_ihevc_trans_32[k][0] * o[0] |
| + g_ai2_ihevc_trans_32[k][1] * o[1] |
| + g_ai2_ihevc_trans_32[k][2] * o[2] |
| + g_ai2_ihevc_trans_32[k][3] * o[3] |
| + g_ai2_ihevc_trans_32[k][4] * o[4] |
| + g_ai2_ihevc_trans_32[k][5] * o[5] |
| + g_ai2_ihevc_trans_32[k][6] * o[6] |
| + g_ai2_ihevc_trans_32[k][7] * o[7] |
| + g_ai2_ihevc_trans_32[k][8] * o[8] |
| + g_ai2_ihevc_trans_32[k][9] * o[9] |
| + g_ai2_ihevc_trans_32[k][10] * o[10] |
| + g_ai2_ihevc_trans_32[k][11] * o[11] |
| + g_ai2_ihevc_trans_32[k][12] * o[12] |
| + g_ai2_ihevc_trans_32[k][13] * o[13] |
| + g_ai2_ihevc_trans_32[k][14] * o[14] |
| + g_ai2_ihevc_trans_32[k][15] * o[15] + add) |
| >> shift; |
| } |
| |
| pi2_tmp += trans_size; |
| pi2_dst++; |
| } |
| } |
| |