/******************************************************************************
*
* 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_chroma_intra_pred_filters.c
*
* @brief
* Contains function Definition for intra prediction interpolation filters
*
*
* @author
* Ittiam
*
* @par List of Functions:
* ihevc_intra_pred_chroma_planar()
*
* ihevc_intra_pred_chroma_dc()
*
* ihevc_intra_pred_chroma_horz()
*
* ihevc_intra_pred_chroma_ver()
*
* ihevc_intra_pred_chroma_mode2()
*
* ihevc_intra_pred_chroma_mode_18_34()
*
* ihevc_intra_pred_chroma_mode_3_to_9()
*
* ihevc_intra_pred_chroma_mode_11_to_17()
*
* ihevc_intra_pred_chroma_mode_19_to_25()
*
* ihevc_intra_pred_chroma_mode_27_to_33()
*
* ihevc_intra_pred_chroma_ref_substitution()
*
*
* @remarks
* None
*
*******************************************************************************
*/
/*****************************************************************************/
/* File Includes */
/*****************************************************************************/
#include "ihevc_typedefs.h"
#include "ihevc_macros.h"
#include "ihevc_func_selector.h"
#include "ihevc_platform_macros.h"
#include "ihevc_intra_pred.h"
#include "ihevc_mem_fns.h"
#include "ihevc_chroma_intra_pred.h"
#include "ihevc_common_tables.h"
/****************************************************************************/
/* Constant Macros */
/****************************************************************************/
#define MAX_CU_SIZE 64
#define BIT_DEPTH 8
#define T32_4NT 128
#define T16_4NT 64
#define T16C_4NT 64
#define T8C_4NT 32
/****************************************************************************/
/* Function Macros */
/****************************************************************************/
#define GET_BIT(y,x) ((y) & (1 << x)) && (1 << x)
/*****************************************************************************/
/* Function Definition */
/*****************************************************************************/
/**
*******************************************************************************
*
* @brief
* Reference substitution process for samples unavailable for prediction
* Refer to section 8.4.4.2.2
*
* @par Description:
*
*
* @param[in] pu1_top_left
* UWORD8 pointer to the top-left
*
* @param[in] pu1_top
* UWORD8 pointer to the top
*
* @param[in] pu1_left
* UWORD8 pointer to the left
*
* @param[in] src_strd
* WORD32 Source stride
*
* @param[in] nbr_flags
* WORD32 neighbor availability flags
*
* @param[in] nt
* WORD32 transform Block size
*
* @param[in] dst_strd
* WORD32 Destination stride
*
* @returns
*
* @remarks
* None
*
*******************************************************************************
*/
void ihevc_intra_pred_chroma_ref_substitution(UWORD8 *pu1_top_left,
UWORD8 *pu1_top,
UWORD8 *pu1_left,
WORD32 src_strd,
WORD32 nt,
WORD32 nbr_flags,
UWORD8 *pu1_dst,
WORD32 dst_strd)
{
UWORD8 pu1_ref_u, pu1_ref_v;
WORD32 dc_val, i, j;
WORD32 total_samples = (4 * nt) + 1;
WORD32 get_bits;
WORD32 next;
WORD32 bot_left, left, top, tp_right, tp_left;
WORD32 idx, nbr_id_from_bl, frwd_nbr_flag;
WORD32 a_nbr_flag[5];
UNUSED(dst_strd);
/* Neighbor Flag Structure*/
/* WORD32 nbr_flags MSB-->LSB TOP LEFT | TOP-RIGHT | TOP | LEFT | BOTTOM LEFT*/
/* (1 bit) (4 bits) (4 bits) (4 bits) (4 bits) */
if(nbr_flags == 0)
{
/* If no neighbor flags are present, fill the neighbor samples with DC value */
/*dc_val = 1 << (BIT_DEPTH - 1);*/
dc_val = 1 << (8 - 1);
for(i = 0; i < (2 * total_samples); i++)
{
pu1_dst[i] = dc_val;
}
}
else
{
/* Else fill the corresponding samples */
/* Check for the neighbors availibility */
tp_left = (nbr_flags & 0x10000);
tp_right = (nbr_flags & 0x0f000);
top = (nbr_flags & 0x00f00);
left = (nbr_flags & 0x000f0);
bot_left = (nbr_flags & 0x0000f);
/* Fill nbrs depending on avalibility */
/* Top -Left nbrs */
if(0 != tp_left)
{
pu1_dst[(4 * nt)] = *pu1_top_left; // U top-left sample
pu1_dst[(4 * nt) + 1] = *(pu1_top_left + 1); // V top-left sample
}
/* Left nbrs */
if(0 != left)
{
for(i = 0, j = 0; i < (2 * nt); i += 2)
{
pu1_dst[(4 * nt) - 2 - i] = pu1_left[j * src_strd]; // U left samples
pu1_dst[(4 * nt) - 1 - i] = pu1_left[(j * src_strd) + 1]; // V left samples
j++;
}
}
/* Bottom - Left nbrs */
if(0 != bot_left)
{
for(i = (2 * nt), j = nt; i < (4 * nt); i += 2)
{
pu1_dst[(4 * nt) - 2 - i] = pu1_left[j * src_strd]; // U left samples
pu1_dst[(4 * nt) - 1 - i] = pu1_left[(j * src_strd) + 1]; // V left samples
j++;
}
}
/* Top nbrs */
if(0 != top)
{
ihevc_memcpy_mul_8(&pu1_dst[(4 * nt) + 2], pu1_top, 2 * nt);
// U-V interleaved Top-top right samples
}
/* Top - Right nbrs */
if(0 != tp_right)
{
ihevc_memcpy_mul_8(&pu1_dst[(4 * nt) + 2 + 2 * nt], pu1_top + 2 * nt, 2 * nt);
// U-V interleaved Top-top right samples
}
if(nt == 4)
{
/* 1 bit extraction for all the neighboring blocks */
tp_left = (nbr_flags & 0x10000) >> 16;
bot_left = (nbr_flags & 0x8) >> 3;
left = (nbr_flags & 0x80) >> 7;
top = (nbr_flags & 0x100) >> 8;
tp_right = (nbr_flags & 0x1000) >> 12;
next = 1;
a_nbr_flag[0] = bot_left;
a_nbr_flag[1] = left;
a_nbr_flag[2] = tp_left;
a_nbr_flag[3] = top;
a_nbr_flag[4] = tp_right;
/* If bottom -left is not available, reverse substitution process*/
if(bot_left == 0)
{
/* Check for the 1st available sample from bottom-left*/
while(!a_nbr_flag[next])
next++;
/* If Left, top-left are available*/
if(next <= 2)
{
UWORD16 *pu2_dst;
idx = (nt * next);
pu2_dst = (UWORD16 *)&pu1_dst[2 * idx];
ihevc_memset_16bit((UWORD16 *)pu1_dst, pu2_dst[0], idx);
}
else /* If top, top-right are available */
{
UWORD16 *pu2_dst;
/* Idx is changed to copy 1 pixel value for top-left ,if top-left is not available*/
idx = (nt * (next - 1)) + 1;
pu2_dst = (UWORD16 *)&pu1_dst[2 * idx];
ihevc_memset_16bit((UWORD16 *)pu1_dst, pu2_dst[0], idx);
}
}
if(left == 0)
{
UWORD16 *pu2_dst = (UWORD16 *)&pu1_dst[(2 * nt) - 2];
ihevc_memset_16bit((UWORD16 *)&pu1_dst[(2 * nt)], pu2_dst[0], nt);
}
if(tp_left == 0)
{
pu1_dst[4 * nt] = pu1_dst[(4 * nt) - 2];
pu1_dst[(4 * nt) + 1] = pu1_dst[(4 * nt) - 1];
}
if(top == 0)
{
UWORD16 *pu2_dst = (UWORD16 *)&pu1_dst[(4 * nt)];
ihevc_memset_16bit((UWORD16 *)&pu1_dst[(4 * nt) + 2], pu2_dst[0], nt);
}
if(tp_right == 0)
{
UWORD16 *pu2_dst = (UWORD16 *)&pu1_dst[(6 * nt)];
ihevc_memset_16bit((UWORD16 *)&pu1_dst[(6 * nt) + 2], pu2_dst[0], nt);
}
}
else if(nt == 8)
{
WORD32 nbr_flags_temp = 0;
nbr_flags_temp = ((nbr_flags & 0xC) >> 2) + ((nbr_flags & 0xC0) >> 4)
+ ((nbr_flags & 0x300) >> 4)
+ ((nbr_flags & 0x3000) >> 6)
+ ((nbr_flags & 0x10000) >> 8);
/* compute trailing zeors based on nbr_flag for substitution process of below left see section .*/
/* as each bit in nbr flags corresponds to 8 pels for bot_left, left, top and topright but 1 pel for topleft */
{
nbr_id_from_bl = look_up_trailing_zeros(nbr_flags_temp & 0XF) * 4; /* for bottom left and left */
if(nbr_id_from_bl == 32)
nbr_id_from_bl = 16;
if(nbr_id_from_bl == 16)
{
/* for top left : 1 pel per nbr bit */
if(!((nbr_flags_temp >> 8) & 0x1))
{
nbr_id_from_bl++;
nbr_id_from_bl += look_up_trailing_zeros((nbr_flags_temp >> 4) & 0xF) * 4; /* top and top right; 8 pels per nbr bit */
}
}
/* Reverse Substitution Process*/
if(nbr_id_from_bl)
{
/* Replicate the bottom-left and subsequent unavailable pixels with the 1st available pixel above */
pu1_ref_u = pu1_dst[2 * nbr_id_from_bl];
pu1_ref_v = pu1_dst[(2 * nbr_id_from_bl) + 1];
for(i = 2 * (nbr_id_from_bl - 1); i >= 0; i -= 2)
{
pu1_dst[i] = pu1_ref_u;
pu1_dst[i + 1] = pu1_ref_v;
}
}
}
/* for the loop of 4*Nt+1 pixels (excluding pixels computed from reverse substitution) */
while(nbr_id_from_bl < ((T8C_4NT)+1))
{
/* To Obtain the next unavailable idx flag after reverse neighbor substitution */
/* Divide by 8 to obtain the original index */
frwd_nbr_flag = (nbr_id_from_bl >> 2); /*+ (nbr_id_from_bl & 0x1);*/
/* The Top-left flag is at the last bit location of nbr_flags*/
if(nbr_id_from_bl == (T8C_4NT / 2))
{
get_bits = GET_BIT(nbr_flags_temp, 8);
/* only pel substitution for TL */
if(!get_bits)
{
pu1_dst[2 * nbr_id_from_bl] = pu1_dst[(2 * nbr_id_from_bl) - 2];
pu1_dst[(2 * nbr_id_from_bl) + 1] = pu1_dst[(2 * nbr_id_from_bl) - 1];
}
}
else
{
get_bits = GET_BIT(nbr_flags_temp, frwd_nbr_flag);
if(!get_bits)
{
UWORD16 *pu2_dst;
/* 8 pel substitution (other than TL) */
pu2_dst = (UWORD16 *)&pu1_dst[(2 * nbr_id_from_bl) - 2];
ihevc_memset_16bit((UWORD16 *)(pu1_dst + (2 * nbr_id_from_bl)), pu2_dst[0], 4);
}
}
nbr_id_from_bl += (nbr_id_from_bl == (T8C_4NT / 2)) ? 1 : 4;
}
}
else if(nt == 16)
{
/* compute trailing ones based on mbr_flag for substitution process of below left see section .*/
/* as each bit in nbr flags corresponds to 4 pels for bot_left, left, top and topright but 1 pel for topleft */
{
nbr_id_from_bl = look_up_trailing_zeros((nbr_flags & 0XFF)) * 4; /* for bottom left and left */
if(nbr_id_from_bl == 32)
{
/* for top left : 1 pel per nbr bit */
if(!((nbr_flags >> 16) & 0x1))
{
/* top left not available */
nbr_id_from_bl++;
/* top and top right; 4 pels per nbr bit */
nbr_id_from_bl += look_up_trailing_zeros((nbr_flags >> 8) & 0xFF) * 4;
}
}
/* Reverse Substitution Process*/
if(nbr_id_from_bl)
{
/* Replicate the bottom-left and subsequent unavailable pixels with the 1st available pixel above */
pu1_ref_u = pu1_dst[2 * nbr_id_from_bl];
pu1_ref_v = pu1_dst[2 * nbr_id_from_bl + 1];
for(i = (2 * (nbr_id_from_bl - 1)); i >= 0; i -= 2)
{
pu1_dst[i] = pu1_ref_u;
pu1_dst[i + 1] = pu1_ref_v;
}
}
}
/* for the loop of 4*Nt+1 pixels (excluding pixels computed from reverse substitution) */
while(nbr_id_from_bl < ((T16C_4NT)+1))
{
/* To Obtain the next unavailable idx flag after reverse neighbor substitution */
/* Devide by 4 to obtain the original index */
frwd_nbr_flag = (nbr_id_from_bl >> 2); /*+ (nbr_id_from_bl & 0x1);*/
/* The Top-left flag is at the last bit location of nbr_flags*/
if(nbr_id_from_bl == (T16C_4NT / 2))
{
get_bits = GET_BIT(nbr_flags, 16);
/* only pel substitution for TL */
if(!get_bits)
{
pu1_dst[2 * nbr_id_from_bl] = pu1_dst[(2 * nbr_id_from_bl) - 2];
pu1_dst[(2 * nbr_id_from_bl) + 1] = pu1_dst[(2 * nbr_id_from_bl) - 1];
}
}
else
{
get_bits = GET_BIT(nbr_flags, frwd_nbr_flag);
if(!get_bits)
{
UWORD16 *pu2_dst;
/* 4 pel substitution (other than TL) */
pu2_dst = (UWORD16 *)&pu1_dst[(2 * nbr_id_from_bl) - 2];
ihevc_memset_16bit((UWORD16 *)(pu1_dst + (2 * nbr_id_from_bl)), pu2_dst[0], 4);
}
}
nbr_id_from_bl += (nbr_id_from_bl == (T16C_4NT / 2)) ? 1 : 4;
}
}
}
}
/**
*******************************************************************************
*
* @brief
* Planar Intraprediction with reference neighboring samples location
* pointed by 'pu1_ref' to the TU block location pointed by 'pu1_dst' Refer
* to section 8.4.4.2.4 in the standard
*
* @par Description:
*
*
* @param[in] pu1_src
* UWORD8 pointer to the source
*
* @param[in] pu1_dst
* UWORD8 pointer to the destination
*
* @param[in] src_strd
* integer source stride
*
* @param[in] dst_strd
* integer destination stride
*
* @param[in] nt
* integer Transform Block size
*
* @param[in] mode
* integer intraprediction mode
*
* @returns
*
* @remarks
* None
*
*******************************************************************************
*/
void ihevc_intra_pred_chroma_planar(UWORD8 *pu1_ref,
WORD32 src_strd,
UWORD8 *pu1_dst,
WORD32 dst_strd,
WORD32 nt,
WORD32 mode)
{
WORD32 row, col;
WORD32 log2nt = 5;
WORD32 two_nt, three_nt;
UNUSED(src_strd);
UNUSED(mode);
switch(nt)
{
case 16:
log2nt = 4;
break;
case 8:
log2nt = 3;
break;
case 4:
log2nt = 2;
break;
default:
break;
}
two_nt = 2 * nt;
three_nt = 3 * nt;
/* Planar filtering */
for(row = 0; row < nt; row++)
{
for(col = 0; col < (2 * nt); col += 2)
{
pu1_dst[row * dst_strd + col] = ((nt - 1 - col / 2)
* pu1_ref[2 * (two_nt - 1 - row)]
+ (col / 2 + 1) * pu1_ref[2 * (three_nt + 1)]
+ (nt - 1 - row) * pu1_ref[2 * (two_nt + 1) + col]
+ (row + 1) * pu1_ref[2 * (nt - 1)] + nt) >> (log2nt + 1);
pu1_dst[row * dst_strd + col + 1] = ((nt - 1 - col / 2)
* pu1_ref[2 * (two_nt - 1 - row) + 1]
+ (col / 2 + 1) * pu1_ref[2 * (three_nt + 1) + 1]
+ (nt - 1 - row) * pu1_ref[2 * (two_nt + 1) + col + 1]
+ (row + 1) * pu1_ref[2 * (nt - 1) + 1] + nt) >> (log2nt + 1);
}
}
}
/**
*******************************************************************************
*
* @brief
* Intraprediction for DC mode with reference neighboring samples location
* pointed by 'pu1_ref' to the TU block location pointed by 'pu1_dst' Refer
* to section 8.4.4.2.5 in the standard
*
* @par Description:
*
*
* @param[in] pu1_src
* UWORD8 pointer to the source
*
* @param[in] pu1_dst
* UWORD8 pointer to the destination
*
* @param[in] src_strd
* integer source stride
*
* @param[in] dst_strd
* integer destination stride
*
* @param[in] nt
* integer Transform Block size (Chroma)
*
* @param[in] mode
* integer intraprediction mode
*
* @returns
*
* @remarks
* None
*
*******************************************************************************
*/
void ihevc_intra_pred_chroma_dc(UWORD8 *pu1_ref,
WORD32 src_strd,
UWORD8 *pu1_dst,
WORD32 dst_strd,
WORD32 nt,
WORD32 mode)
{
WORD32 acc_dc_u, acc_dc_v;
WORD32 dc_val_u, dc_val_v;
WORD32 i;
WORD32 row, col;
WORD32 log2nt = 5;
UNUSED(mode);
UNUSED(src_strd);
switch(nt)
{
case 32:
log2nt = 5;
break;
case 16:
log2nt = 4;
break;
case 8:
log2nt = 3;
break;
case 4:
log2nt = 2;
break;
default:
break;
}
acc_dc_u = 0;
acc_dc_v = 0;
/* Calculate DC value for the transform block */
for(i = (2 * nt); i < (4 * nt); i += 2)
{
acc_dc_u += pu1_ref[i];
acc_dc_v += pu1_ref[i + 1];
}
for(i = ((4 * nt) + 2); i < ((6 * nt) + 2); i += 2)
{
acc_dc_u += pu1_ref[i];
acc_dc_v += pu1_ref[i + 1];
}
dc_val_u = (acc_dc_u + nt) >> (log2nt + 1);
dc_val_v = (acc_dc_v + nt) >> (log2nt + 1);
/* Fill the remaining rows with DC value*/
for(row = 0; row < nt; row++)
{
for(col = 0; col < (2 * nt); col += 2)
{
pu1_dst[(row * dst_strd) + col] = dc_val_u;
pu1_dst[(row * dst_strd) + col + 1] = dc_val_v;
}
}
}
/**
*******************************************************************************
*
* @brief
* Horizontal intraprediction(mode 10) with reference samples location
* pointed by 'pu1_ref' to the TU block location pointed by 'pu1_dst' Refer
* to section 8.4.4.2.6 in the standard (Special case)
*
* @par Description:
*
*
* @param[in] pu1_src
* UWORD8 pointer to the source
*
* @param[in] pu1_dst
* UWORD8 pointer to the destination
*
* @param[in] src_strd
* integer source stride
*
* @param[in] dst_strd
* integer destination stride
*
* @param[in] nt
* integer Transform Block size
*
* @param[in] mode
* integer intraprediction mode
*
* @returns
*
* @remarks
* None
*
*******************************************************************************
*/
void ihevc_intra_pred_chroma_horz(UWORD8 *pu1_ref,
WORD32 src_strd,
UWORD8 *pu1_dst,
WORD32 dst_strd,
WORD32 nt,
WORD32 mode)
{
WORD32 row, col;
UNUSED(mode);
UNUSED(src_strd);
/* Replication to next rows*/
for(row = 0; row < nt; row++)
{
for(col = 0; col < (2 * nt); col += 2)
{
pu1_dst[(row * dst_strd) + col] = pu1_ref[(4 * nt) - 2 - 2 * row];
pu1_dst[(row * dst_strd) + col + 1] = pu1_ref[(4 * nt) - 1 - 2 * row];
}
}
}
/**
*******************************************************************************
*
* @brief
* Horizontal intraprediction with reference neighboring samples location
* pointed by 'pu1_ref' to the TU block location pointed by 'pu1_dst' Refer
* to section 8.4.4.2.6 in the standard (Special case)
*
* @par Description:
*
*
* @param[in] pu1_src
* UWORD8 pointer to the source
*
* @param[in] pu1_dst
* UWORD8 pointer to the destination
*
* @param[in] src_strd
* integer source stride
*
* @param[in] dst_strd
* integer destination stride
*
* @param[in] nt
* integer Transform Block size
*
* @param[in] mode
* integer intraprediction mode
*
* @returns
*
* @remarks
* None
*
*******************************************************************************
*/
void ihevc_intra_pred_chroma_ver(UWORD8 *pu1_ref,
WORD32 src_strd,
UWORD8 *pu1_dst,
WORD32 dst_strd,
WORD32 nt,
WORD32 mode)
{
WORD32 row, col;
UNUSED(mode);
UNUSED(src_strd);
/* Replication to next columns*/
for(row = 0; row < nt; row++)
{
for(col = 0; col < (2 * nt); col += 2)
{
pu1_dst[(row * dst_strd) + col] = pu1_ref[(4 * nt) + 2 + col];
pu1_dst[(row * dst_strd) + col + 1] = pu1_ref[(4 * nt) + 3 + col];
}
}
}
/**
*******************************************************************************
*
* @brief
* Intraprediction for mode 2 (sw angle) with reference neighboring samples
* location pointed by 'pu1_ref' to the TU block location pointed by
* 'pu1_dst' Refer to section 8.4.4.2.6 in the standard
*
* @par Description:
*
*
* @param[in] pu1_src
* UWORD8 pointer to the source
*
* @param[in] pu1_dst
* UWORD8 pointer to the destination
*
* @param[in] src_strd
* integer source stride
*
* @param[in] dst_strd
* integer destination stride
*
* @param[in] nt
* integer Transform Block size
*
* @param[in] mode
* integer intraprediction mode
*
* @returns
*
* @remarks
* None
*
*******************************************************************************
*/
void ihevc_intra_pred_chroma_mode2(UWORD8 *pu1_ref,
WORD32 src_strd,
UWORD8 *pu1_dst,
WORD32 dst_strd,
WORD32 nt,
WORD32 mode)
{
WORD32 row, col;
WORD32 intra_pred_ang = 32;
WORD32 idx_u, idx_v;
UNUSED(src_strd);
UNUSED(mode);
/* For the angle 45, replication is done from the corresponding angle */
/* intra_pred_ang = tan(angle) in q5 format */
for(col = 0; col < (2 * nt); col += 2)
{
idx_u = ((col + 1) * intra_pred_ang) >> 5; /* Use idx++ */
idx_v = (((col + 1) + 1) * intra_pred_ang) >> 5; /* Use idx++ */
for(row = 0; row < nt; row++)
{
pu1_dst[col + (row * dst_strd)] = pu1_ref[(4 * nt) - 2 * row - idx_u - 3];
pu1_dst[(col + 1) + (row * dst_strd)] = pu1_ref[(4 * nt) - 2 * row - idx_v - 1];
}
}
}
/**
*******************************************************************************
*
* @brief
* Intraprediction for mode 34 (ne angle) and mode 18 (nw angle) with
* reference neighboring samples location pointed by 'pu1_ref' to the TU
* block location pointed by 'pu1_dst'
*
* @par Description:
*
*
* @param[in] pu1_src
* UWORD8 pointer to the source
*
* @param[in] pu1_dst
* UWORD8 pointer to the destination
*
* @param[in] src_strd
* integer source stride
*
* @param[in] dst_strd
* integer destination stride
*
* @param[in] nt
* integer Transform Block size
*
* @param[in] mode
* integer intraprediction mode
*
* @returns
*
* @remarks
* None
*
*******************************************************************************
*/
void ihevc_intra_pred_chroma_mode_18_34(UWORD8 *pu1_ref,
WORD32 src_strd,
UWORD8 *pu1_dst,
WORD32 dst_strd,
WORD32 nt,
WORD32 mode)
{
WORD32 row, col;
WORD32 intra_pred_ang;
WORD32 idx = 0;
UNUSED(src_strd);
intra_pred_ang = 32; /*Default value*/
/* For mode 18, angle is -45degree */
if(mode == 18)
intra_pred_ang = -32;
/* For mode 34, angle is 45degree */
else if(mode == 34)
intra_pred_ang = 32;
/* For the angle 45 and -45, replication is done from the corresponding angle */
/* No interpolation is done for 45 degree*/
for(row = 0; row < nt; row++)
{
idx = ((row + 1) * intra_pred_ang) >> 5;
for(col = 0; col < (2 * nt); col += 2)
{
pu1_dst[col + (row * dst_strd)] = pu1_ref[(4 * nt) + col + 2 * idx + 2];
pu1_dst[(col + 1) + (row * dst_strd)] = pu1_ref[(4 * nt) + (col + 1) + 2 * idx + 2];
}
}
}
/**
*******************************************************************************
*
* @brief
* Intraprediction for mode 3 to 9 (positive angle, horizontal mode ) with
* reference neighboring samples location pointed by 'pu1_ref' to the TU
* block location pointed by 'pu1_dst'
*
* @par Description:
*
*
* @param[in] pu1_src
* UWORD8 pointer to the source
*
* @param[in] pu1_dst
* UWORD8 pointer to the destination
*
* @param[in] src_strd
* integer source stride
*
* @param[in] dst_strd
* integer destination stride
*
* @param[in] nt
* integer Transform Block size
*
* @param[in] mode
* integer intraprediction mode
*
* @returns
*
* @remarks
* None
*
*******************************************************************************
*/
void ihevc_intra_pred_chroma_mode_3_to_9(UWORD8 *pu1_ref,
WORD32 src_strd,
UWORD8 *pu1_dst,
WORD32 dst_strd,
WORD32 nt,
WORD32 mode)
{
WORD32 row, col;
WORD32 intra_pred_ang;
WORD32 idx_u, ref_main_idx_u;
WORD32 idx_v, ref_main_idx_v;
WORD32 pos_u, fract_u;
WORD32 pos_v, fract_v;
UNUSED(src_strd);
/* Intra Pred Angle according to the mode */
intra_pred_ang = gai4_ihevc_ang_table[mode];
/* For the angles other then 45 degree, interpolation btw 2 neighboring */
/* samples dependent on distance to obtain destination sample */
for(col = 0; col < (2 * nt); col += 2)
{
pos_u = ((col / 2 + 1) * intra_pred_ang);
pos_v = ((col / 2 + 1) * intra_pred_ang);
idx_u = pos_u >> 5;
fract_u = pos_u & (31);
idx_v = pos_v >> 5;
fract_v = pos_v & (31);
// Do linear filtering
for(row = 0; row < nt; row++)
{
ref_main_idx_u = (4 * nt) - 2 * row - 2 * idx_u - 2;
ref_main_idx_v = (4 * nt) - 2 * row - 2 * idx_v - 1;
pu1_dst[col + (row * dst_strd)] = (((32 - fract_u)
* pu1_ref[ref_main_idx_u]
+ fract_u * pu1_ref[ref_main_idx_u - 2] + 16) >> 5);
pu1_dst[(col + 1) + (row * dst_strd)] = (((32 - fract_v)
* pu1_ref[ref_main_idx_v]
+ fract_v * pu1_ref[ref_main_idx_v - 2] + 16) >> 5);
}
}
}
/**
*******************************************************************************
*
* @brief
* Intraprediction for mode 11 to 17 (negative angle, horizontal mode )
* with reference neighboring samples location pointed by 'pu1_ref' to the
* TU block location pointed by 'pu1_dst'
*
* @par Description:
*
*
* @param[in] pu1_src
* UWORD8 pointer to the source
*
* @param[in] pu1_dst
* UWORD8 pointer to the destination
*
* @param[in] src_strd
* integer source stride
*
* @param[in] dst_strd
* integer destination stride
*
* @param[in] nt
* integer Transform Block size
*
* @param[in] mode
* integer intraprediction mode
*
* @returns
*
* @remarks
* None
*
*******************************************************************************
*/
void ihevc_intra_pred_chroma_mode_11_to_17(UWORD8 *pu1_ref,
WORD32 src_strd,
UWORD8 *pu1_dst,
WORD32 dst_strd,
WORD32 nt,
WORD32 mode)
{
/* This function and ihevc_intra_pred_CHROMA_mode_19_to_25 are same except*/
/* for ref main & side samples assignment,can be combined for */
/* optimzation*/
WORD32 row, col, k;
WORD32 intra_pred_ang, inv_ang, inv_ang_sum;
WORD32 idx_u, idx_v, ref_main_idx_u, ref_main_idx_v, ref_idx;
WORD32 pos_u, pos_v, fract_u, fract_v;
UWORD8 ref_temp[2 * MAX_CU_SIZE + 2];
UWORD8 *ref_main;
UNUSED(src_strd);
inv_ang_sum = 128;
intra_pred_ang = gai4_ihevc_ang_table[mode];
inv_ang = gai4_ihevc_inv_ang_table[mode - 11];
/* Intermediate reference samples for negative angle modes */
/* This have to be removed during optimization*/
/* For horizontal modes, (ref main = ref left) (ref side = ref above) */
ref_main = ref_temp + 2 * nt;
for(k = 0; k < (2 * (nt + 1)); k += 2)
{
ref_temp[k + (2 * (nt - 1))] = pu1_ref[(4 * nt) - k];
ref_temp[k + 1 + (2 * (nt - 1))] = pu1_ref[(4 * nt) - k + 1];
}
ref_main = ref_temp + (2 * (nt - 1));
ref_idx = (nt * intra_pred_ang) >> 5;
/* SIMD Optimization can be done using look-up table for the loop */
/* For negative angled derive the main reference samples from side */
/* reference samples refer to section 8.4.4.2.6 */
for(k = -2; k > (2 * ref_idx); k -= 2)
{
inv_ang_sum += inv_ang;
ref_main[k] = pu1_ref[(4 * nt) + ((inv_ang_sum >> 8) << 1)];
ref_main[k + 1] = pu1_ref[((4 * nt) + 1) + ((inv_ang_sum >> 8) << 1)];
}
/* For the angles other then 45 degree, interpolation btw 2 neighboring */
/* samples dependent on distance to obtain destination sample */
for(col = 0; col < (2 * nt); col += 2)
{
pos_u = ((col / 2 + 1) * intra_pred_ang);
pos_v = ((col / 2 + 1) * intra_pred_ang);
idx_u = pos_u >> 5;
idx_v = pos_v >> 5;
fract_u = pos_u & (31);
fract_v = pos_v & (31);
// Do linear filtering
for(row = 0; row < nt; row++)
{
ref_main_idx_u = 2 * (row + idx_u + 1);
ref_main_idx_v = 2 * (row + idx_v + 1) + 1;
pu1_dst[col + (dst_strd * row)] = (UWORD8)(((32 - fract_u)
* ref_main[ref_main_idx_u]
+ fract_u * ref_main[ref_main_idx_u + 2] + 16) >> 5);
pu1_dst[(col + 1) + (dst_strd * row)] = (UWORD8)(((32 - fract_v)
* ref_main[ref_main_idx_v]
+ fract_v * ref_main[ref_main_idx_v + 2] + 16) >> 5);
}
}
}
/**
*******************************************************************************
*
* @brief
* Intraprediction for mode 19 to 25 (negative angle, vertical mode ) with
* reference neighboring samples location pointed by 'pu1_ref' to the TU
* block location pointed by 'pu1_dst'
*
* @par Description:
*
*
* @param[in] pu1_src
* UWORD8 pointer to the source
*
* @param[in] pu1_dst
* UWORD8 pointer to the destination
*
* @param[in] src_strd
* integer source stride
*
* @param[in] dst_strd
* integer destination stride
*
* @param[in] nt
* integer Transform Block size
*
* @param[in] mode
* integer intraprediction mode
*
* @returns
*
* @remarks
* None
*
*******************************************************************************
*/
void ihevc_intra_pred_chroma_mode_19_to_25(UWORD8 *pu1_ref,
WORD32 src_strd,
UWORD8 *pu1_dst,
WORD32 dst_strd,
WORD32 nt,
WORD32 mode)
{
WORD32 row, col, k;
WORD32 intra_pred_ang, idx;
WORD32 inv_ang, inv_ang_sum, pos, fract;
WORD32 ref_main_idx_u, ref_main_idx_v, ref_idx;
UWORD8 ref_temp[(2 * MAX_CU_SIZE) + 2];
UWORD8 *ref_main;
UNUSED(src_strd);
intra_pred_ang = gai4_ihevc_ang_table_chroma[mode];
inv_ang = gai4_ihevc_inv_ang_table_chroma[mode - 12];
/* Intermediate reference samples for negative angle modes */
/* This have to be removed during optimization*/
/* For horizontal modes, (ref main = ref above) (ref side = ref left) */
ref_main = ref_temp + 2 * nt;
for(k = 0; k < (2 * (nt + 1)); k += 2)
{
ref_temp[k + (2 * (nt - 1))] = pu1_ref[(4 * nt) + k];
ref_temp[k + 1 + (2 * (nt - 1))] = pu1_ref[(4 * nt) + k + 1];
}
ref_idx = (nt * intra_pred_ang) >> 5;
inv_ang_sum = 128;
ref_main = ref_temp + (2 * (nt - 1));
/* SIMD Optimization can be done using look-up table for the loop */
/* For negative angled derive the main reference samples from side */
/* reference samples refer to section 8.4.4.2.6 */
for(k = -2; k > (2 * ref_idx); k -= 2)
{
inv_ang_sum += inv_ang;
ref_main[k] = pu1_ref[(4 * nt) - (inv_ang_sum >> 8) * 2];
ref_main[k + 1] = pu1_ref[((4 * nt) + 1) - (inv_ang_sum >> 8) * 2];
}
for(row = 0; row < nt; row++)
{
pos = ((row + 1) * intra_pred_ang);
idx = pos >> 5;
fract = pos & (31);
// Do linear filtering
for(col = 0; col < (2 * nt); col += 2)
{
ref_main_idx_u = col + 2 * idx + 2;
ref_main_idx_v = (col + 1) + 2 * idx + 2;
pu1_dst[(row * dst_strd) + col] = (UWORD8)(((32 - fract)
* ref_main[ref_main_idx_u]
+ fract * ref_main[ref_main_idx_u + 2] + 16) >> 5);
pu1_dst[(row * dst_strd) + (col + 1)] = (UWORD8)(((32 - fract)
* ref_main[ref_main_idx_v]
+ fract * ref_main[ref_main_idx_v + 2] + 16) >> 5);
}
}
}
/**
*******************************************************************************
*
* @brief
* Intraprediction for mode 27 to 33 (positive angle, vertical mode ) with
* reference neighboring samples location pointed by 'pu1_ref' to the TU
* block location pointed by 'pu1_dst'
*
* @par Description:
*
*
* @param[in] pu1_src
* UWORD8 pointer to the source
*
* @param[in] pu1_dst
* UWORD8 pointer to the destination
*
* @param[in] src_strd
* integer source stride
*
* @param[in] dst_strd
* integer destination stride
*
* @param[in] nt
* integer Transform Block size
*
* @param[in] mode
* integer intraprediction mode
*
* @returns
*
* @remarks
* None
*
*******************************************************************************
*/
void ihevc_intra_pred_chroma_mode_27_to_33(UWORD8 *pu1_ref,
WORD32 src_strd,
UWORD8 *pu1_dst,
WORD32 dst_strd,
WORD32 nt,
WORD32 mode)
{
WORD32 row, col;
WORD32 pos, fract;
WORD32 intra_pred_ang;
WORD32 idx, ref_main_idx_u, ref_main_idx_v;
UNUSED(src_strd);
intra_pred_ang = gai4_ihevc_ang_table_chroma[mode];
for(row = 0; row < nt; row++)
{
pos = ((row + 1) * intra_pred_ang);
idx = pos >> 5;
fract = pos & (31);
// Do linear filtering
for(col = 0; col < (2 * nt); col += 2)
{
ref_main_idx_u = (4 * nt) + col + 2 * idx + 2;
ref_main_idx_v = (4 * nt) + (col + 1) + 2 * idx + 2;
pu1_dst[col + (row * dst_strd)] = (((32 - fract)
* pu1_ref[ref_main_idx_u]
+ fract * pu1_ref[ref_main_idx_u + 2] + 16) >> 5);
pu1_dst[(col + 1) + (row * dst_strd)] = (((32 - fract)
* pu1_ref[ref_main_idx_v]
+ fract * pu1_ref[ref_main_idx_v + 2] + 16) >> 5);
}
}
}