aosp12/external/libavc/encoder/ih264e_half_pel.c

226 lines
6.7 KiB
C

/******************************************************************************
*
* Copyright (C) 2015 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at:
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*****************************************************************************
* Originally developed and contributed by Ittiam Systems Pvt. Ltd, Bangalore
*/
/**
*******************************************************************************
* @file
* ih264e_half_pel.c
*
* @brief
* This file contains functions that are used for computing subpixel planes
*
* @author
* ittiam
*
* @par List of Functions:
* - ih264e_sixtapfilter_horz
* - ih264e_sixtap_filter_2dvh_vert
*
* @remarks
* None
*
*******************************************************************************
*/
/*****************************************************************************/
/* File Includes */
/*****************************************************************************/
/* System include files */
#include <stdio.h>
#include <assert.h>
#include <limits.h>
/* User include files */
#include "ih264_typedefs.h"
#include "ithread.h"
#include "ih264_platform_macros.h"
#include "ih264_defs.h"
#include "ih264e_half_pel.h"
#include "ih264_macros.h"
#include "ih264e_debug.h"
#include "ih264_inter_pred_filters.h"
#include "ih264_mem_fns.h"
#include "ih264_padding.h"
#include "ih264_intra_pred_filters.h"
#include "ih264_deblk_edge_filters.h"
/*****************************************************************************/
/* Function Definitions */
/*****************************************************************************/
/**
*******************************************************************************
*
* @brief
* Interprediction luma filter for horizontal input (Filter run for width = 17
* and height =16)
*
* @par Description:
* Applies a 6 tap horizontal filter .The output is clipped to 8 bits
* sec 8.4.2.2.1 titled "Luma sample interpolation process"
*
* @param[in] pu1_src
* UWORD8 pointer to the source
*
* @param[out] pu1_dst
* UWORD8 pointer to the destination
*
* @param[in] src_strd
* integer source stride
*
* @param[in] dst_strd
* integer destination stride
*
* @returns
*
* @remarks
* None
*
*******************************************************************************
*/
void ih264e_sixtapfilter_horz(UWORD8 *pu1_src,
UWORD8 *pu1_dst,
WORD32 src_strd,
WORD32 dst_strd)
{
UWORD32 u4_i, u4_j;
UWORD32 u4_w, u4_h;
/* width and height of interpolation */
u4_w = HP_PL_WD;
u4_h = MB_SIZE;
pu1_src -= 2;
for (u4_i = 0; u4_i < u4_h; u4_i++)
{
for (u4_j = 0; u4_j < u4_w; u4_j++, pu1_dst++, pu1_src++)
{
WORD16 i16_temp;
i16_temp = ih264_g_six_tap[0] * (*pu1_src + pu1_src[5])
+ ih264_g_six_tap[1] * (pu1_src[1] + pu1_src[4])
+ ih264_g_six_tap[2] * (pu1_src[2] + pu1_src[3]);
i16_temp = (i16_temp + 16) >> 5;
*pu1_dst = CLIP_U8(i16_temp);
}
pu1_src += src_strd - u4_w;
pu1_dst += dst_strd - u4_w;
}
}
/**
*******************************************************************************
*
* @brief
* This function implements a two stage cascaded six tap filter. It applies
* the six tap filter in the vertical direction on the predictor values,
* followed by applying the same filter in the horizontal direction on the
* output of the first stage. The six tap filtering operation is described in
* sec 8.4.2.2.1 titled "Luma sample interpolation process" (Filter run for
* width = 17 and height = 17)
*
* @par Description:
* The function interpolates the predictors first in the vertical direction and
* then in the horizontal direction to output the (1/2,1/2). The output of the
* first stage of the filter is stored in the buffer pointed to by
* pi16_pred1(only in C) in 16 bit precision.
*
* @param[in] pu1_src
* UWORD8 pointer to the source
*
* @param[out] pu1_dst1
* UWORD8 pointer to the destination (Horizontal filtered output)
*
* @param[out] pu1_dst2
* UWORD8 pointer to the destination (output after applying vertical filter to
* the intermediate horizontal output)
*
* @param[in] src_strd
* integer source stride
* @param[in] dst_strd
* integer destination stride of pu1_dst
*
* @param[in] pi4_pred
* Pointer to 16bit intermediate buffer (used only in c)
*
* @param[in] i4_pred_strd
* integer destination stride of pi16_pred1
*
* @returns
*
* @remarks
* None
*
*******************************************************************************
*/
void ih264e_sixtap_filter_2dvh_vert(UWORD8 *pu1_src,
UWORD8 *pu1_dst1,
UWORD8 *pu1_dst2,
WORD32 src_strd,
WORD32 dst_strd,
WORD32 *pi4_pred,
WORD32 i4_pred_strd)
{
WORD32 row, col;
WORD32 tmp;
WORD32 *pi4_pred_temp = pi4_pred;
WORD32 ht = HP_PL_HT, wd = HP_PL_WD;
for (row = 0; row < ht; row++)
{
for (col = -2; col < wd + 3; col++)
{
tmp = ih264_g_six_tap[0] * (pu1_src[col - 2 * src_strd] + pu1_src[col + 3 * src_strd]) +
ih264_g_six_tap[1] * (pu1_src[col - 1 * src_strd] + pu1_src[col + 2 * src_strd]) +
ih264_g_six_tap[2] * (pu1_src[col] + pu1_src[col + 1 * src_strd]);
pi4_pred_temp[col] = tmp;
}
pu1_src += src_strd;
pi4_pred_temp += i4_pred_strd;
}
for (row = 0; row < ht; row++)
{
for (col = 0; col < wd; col++)
{
tmp = (pi4_pred[col - 2] + pi4_pred[col + 3]) +
ih264_g_six_tap[1] * (pi4_pred[col - 1] + pi4_pred[col + 2]) +
ih264_g_six_tap[2] * (pi4_pred[col] + pi4_pred[col + 1]);
tmp = (tmp + 512) >> 10;
pu1_dst2[col] = CLIP_U8(tmp);
pu1_dst1[col] = CLIP_U8((pi4_pred[col] + 16) >> 5);
}
pi4_pred += i4_pred_strd;
pu1_dst2 += dst_strd;
pu1_dst1 += dst_strd;
}
}