1262 lines
48 KiB
C
1262 lines
48 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_intra_modes_eval_ssse3.c
|
|
*
|
|
* @brief
|
|
* This file contains definitions of routines that perform rate distortion
|
|
* analysis on a macroblock if they are to be coded as intra.
|
|
*
|
|
* @author
|
|
* Ittiam
|
|
*
|
|
* @par List of Functions:
|
|
* ih264e_evaluate_intra16x16_modes_ssse3
|
|
* ih264e_evaluate_intra_4x4_modes_ssse3
|
|
* ih264e_evaluate_intra_chroma_modes_ssse3
|
|
*
|
|
* @remarks
|
|
* None
|
|
*
|
|
*******************************************************************************
|
|
*/
|
|
|
|
/*****************************************************************************/
|
|
/* File Includes */
|
|
/*****************************************************************************/
|
|
|
|
/* System include files */
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <limits.h>
|
|
#include <assert.h>
|
|
#include <immintrin.h>
|
|
|
|
/* User include files */
|
|
#include "ih264e_config.h"
|
|
#include "ih264_typedefs.h"
|
|
#include "ih264e_defs.h"
|
|
#include "iv2.h"
|
|
#include "ive2.h"
|
|
#include "ih264_debug.h"
|
|
#include "ih264_defs.h"
|
|
#include "ih264_macros.h"
|
|
#include "ih264_intra_pred_filters.h"
|
|
#include "ih264_structs.h"
|
|
#include "ih264_common_tables.h"
|
|
#include "ih264_trans_quant_itrans_iquant.h"
|
|
#include "ih264_inter_pred_filters.h"
|
|
#include "ih264_mem_fns.h"
|
|
#include "ih264_padding.h"
|
|
#include "ih264_deblk_edge_filters.h"
|
|
#include "ime_distortion_metrics.h"
|
|
#include "ih264e_error.h"
|
|
#include "ih264e_bitstream.h"
|
|
#include "ime_defs.h"
|
|
#include "ime_structs.h"
|
|
#include "ih264_cabac_tables.h"
|
|
#include "irc_cntrl_param.h"
|
|
#include "irc_frame_info_collector.h"
|
|
#include "ih264e_rate_control.h"
|
|
|
|
#include "ih264e_cabac_structs.h"
|
|
#include "ih264e_structs.h"
|
|
#include "ih264e_cabac.h"
|
|
#include "ih264e_intra_modes_eval.h"
|
|
#include "ih264e_globals.h"
|
|
#include "ime_platform_macros.h"
|
|
|
|
|
|
/*****************************************************************************/
|
|
/* Function Definitions */
|
|
/*****************************************************************************/
|
|
/**
|
|
******************************************************************************
|
|
*
|
|
* @brief
|
|
* evaluate best intra 16x16 mode (among VERT, HORZ and DC) and do the
|
|
* prediction.
|
|
*
|
|
* @par Description
|
|
* This function evaluates first three 16x16 modes and compute corresponding
|
|
* SAD and returns the buffer predicted with best mode.
|
|
*
|
|
* @param[in] pu1_src
|
|
* UWORD8 pointer to the source
|
|
*
|
|
* @param[in] pu1_ngbr_pels_i16
|
|
* UWORD8 pointer to neighbouring pels
|
|
*
|
|
* @param[out] pu1_dst
|
|
* UWORD8 pointer to the destination
|
|
*
|
|
* @param[in] src_strd
|
|
* integer source stride
|
|
*
|
|
* @param[in] dst_strd
|
|
* integer destination stride
|
|
*
|
|
* @param[in] u4_n_avblty
|
|
* availability of neighbouring pixels
|
|
*
|
|
* @param[in] u4_intra_mode
|
|
* pointer to the variable in which best mode is returned
|
|
*
|
|
* @param[in] pu4_sadmin
|
|
* pointer to the variable in which minimum sad is returned
|
|
*
|
|
* @param[in] u4_valid_intra_modes
|
|
* says what all modes are valid
|
|
*
|
|
* @return
|
|
* None
|
|
*
|
|
******************************************************************************
|
|
*/
|
|
void ih264e_evaluate_intra16x16_modes_ssse3(UWORD8 *pu1_src,
|
|
UWORD8 *pu1_ngbr_pels_i16,
|
|
UWORD8 *pu1_dst,
|
|
UWORD32 src_strd,
|
|
UWORD32 dst_strd,
|
|
WORD32 n_avblty,
|
|
UWORD32 *u4_intra_mode,
|
|
WORD32 *pu4_sadmin,
|
|
UWORD32 u4_valid_intra_modes)
|
|
{
|
|
UWORD8 *pu1_src_temp;
|
|
|
|
WORD32 left, top, horz_flag, vert_flag, dc_flag;
|
|
WORD32 sad_vert, sad_horz, sad_dc, min_sad;
|
|
|
|
WORD32 cnt, dcval;
|
|
WORD32 src_strd2, src_strd3, src_strd4;
|
|
WORD32 dst_strd2, dst_strd3, dst_strd4;
|
|
|
|
__m128i src1_16x8b, src2_16x8b, src3_16x8b, src4_16x8b;
|
|
__m128i val1_16x8b, val2_16x8b, val3_16x8b, val4_16x8b;
|
|
__m128i sad1_8x16b, sad2_8x16b, sad3_8x16b, sad4_8x16b;
|
|
|
|
__m128i sad_8x16b, val_16x8b, zero_vector;
|
|
|
|
sad_vert = INT_MAX;
|
|
sad_horz = INT_MAX;
|
|
sad_dc = INT_MAX;
|
|
|
|
src_strd2 = src_strd << 1;
|
|
src_strd4 = src_strd << 2;
|
|
src_strd3 = src_strd + src_strd2;
|
|
|
|
dst_strd2 = dst_strd << 1;
|
|
dst_strd4 = dst_strd << 2;
|
|
dst_strd3 = dst_strd + dst_strd2;
|
|
|
|
left = (n_avblty & LEFT_MB_AVAILABLE_MASK);
|
|
top = (n_avblty & TOP_MB_AVAILABLE_MASK) >> 2;
|
|
|
|
zero_vector = _mm_setzero_si128();
|
|
|
|
horz_flag = left && ((u4_valid_intra_modes & 02) != 0);
|
|
vert_flag = top && ((u4_valid_intra_modes & 01) != 0);
|
|
dc_flag = (u4_valid_intra_modes & 04) != 0;
|
|
|
|
if(horz_flag)
|
|
{
|
|
pu1_src_temp = pu1_src;
|
|
|
|
val1_16x8b = _mm_set1_epi8(pu1_ngbr_pels_i16[15]);
|
|
val2_16x8b = _mm_set1_epi8(pu1_ngbr_pels_i16[14]);
|
|
val3_16x8b = _mm_set1_epi8(pu1_ngbr_pels_i16[13]);
|
|
val4_16x8b = _mm_set1_epi8(pu1_ngbr_pels_i16[12]);
|
|
|
|
src1_16x8b = _mm_loadu_si128((__m128i *)pu1_src_temp);
|
|
src2_16x8b = _mm_loadu_si128((__m128i *)(pu1_src_temp + src_strd));
|
|
src3_16x8b = _mm_loadu_si128((__m128i *)(pu1_src_temp + src_strd2));
|
|
src4_16x8b = _mm_loadu_si128((__m128i *)(pu1_src_temp + src_strd3));
|
|
|
|
sad1_8x16b = _mm_sad_epu8(val1_16x8b, src1_16x8b);
|
|
sad2_8x16b = _mm_sad_epu8(val2_16x8b, src2_16x8b);
|
|
sad3_8x16b = _mm_sad_epu8(val3_16x8b, src3_16x8b);
|
|
sad4_8x16b = _mm_sad_epu8(val4_16x8b, src4_16x8b);
|
|
|
|
sad1_8x16b = _mm_packs_epi32(sad1_8x16b, sad2_8x16b);
|
|
sad3_8x16b = _mm_packs_epi32(sad3_8x16b, sad4_8x16b);
|
|
|
|
cnt = 11;
|
|
sad_8x16b = _mm_packs_epi32(sad1_8x16b, sad3_8x16b);
|
|
do
|
|
{
|
|
pu1_src_temp += src_strd4;
|
|
|
|
val1_16x8b = _mm_set1_epi8(pu1_ngbr_pels_i16[cnt]);
|
|
val2_16x8b = _mm_set1_epi8(pu1_ngbr_pels_i16[cnt - 1]);
|
|
val3_16x8b = _mm_set1_epi8(pu1_ngbr_pels_i16[cnt - 2]);
|
|
val4_16x8b = _mm_set1_epi8(pu1_ngbr_pels_i16[cnt - 3]);
|
|
|
|
src1_16x8b = _mm_loadu_si128((__m128i *)pu1_src_temp);
|
|
src2_16x8b = _mm_loadu_si128((__m128i *)(pu1_src_temp + src_strd));
|
|
src3_16x8b = _mm_loadu_si128((__m128i *)(pu1_src_temp + src_strd2));
|
|
src4_16x8b = _mm_loadu_si128((__m128i *)(pu1_src_temp + src_strd3));
|
|
|
|
sad1_8x16b = _mm_sad_epu8(val1_16x8b, src1_16x8b);
|
|
sad2_8x16b = _mm_sad_epu8(val2_16x8b, src2_16x8b);
|
|
sad3_8x16b = _mm_sad_epu8(val3_16x8b, src3_16x8b);
|
|
sad4_8x16b = _mm_sad_epu8(val4_16x8b, src4_16x8b);
|
|
|
|
sad1_8x16b = _mm_packs_epi32(sad1_8x16b, sad2_8x16b);
|
|
sad3_8x16b = _mm_packs_epi32(sad3_8x16b, sad4_8x16b);
|
|
sad1_8x16b = _mm_packs_epi32(sad1_8x16b, sad3_8x16b);
|
|
|
|
cnt -= 4;
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, sad1_8x16b);
|
|
}
|
|
while(cnt >= 0);
|
|
|
|
sad_8x16b = _mm_hadd_epi16(sad_8x16b, sad_8x16b);
|
|
sad_8x16b = _mm_hadd_epi16(sad_8x16b, sad_8x16b);
|
|
sad_8x16b = _mm_hadd_epi16(sad_8x16b, sad_8x16b);
|
|
|
|
sad_horz = _mm_extract_epi16(sad_8x16b, 0);
|
|
}
|
|
|
|
if(vert_flag)
|
|
{
|
|
pu1_src_temp = pu1_src;
|
|
|
|
val1_16x8b = _mm_loadu_si128((__m128i *)(pu1_ngbr_pels_i16 + 17));
|
|
|
|
src1_16x8b = _mm_loadu_si128((__m128i *)pu1_src_temp);
|
|
src2_16x8b = _mm_loadu_si128((__m128i *)(pu1_src_temp + src_strd));
|
|
src3_16x8b = _mm_loadu_si128((__m128i *)(pu1_src_temp + src_strd2));
|
|
src4_16x8b = _mm_loadu_si128((__m128i *)(pu1_src_temp + src_strd3));
|
|
|
|
sad1_8x16b = _mm_sad_epu8(val1_16x8b, src1_16x8b);
|
|
sad2_8x16b = _mm_sad_epu8(val1_16x8b, src2_16x8b);
|
|
sad3_8x16b = _mm_sad_epu8(val1_16x8b, src3_16x8b);
|
|
sad4_8x16b = _mm_sad_epu8(val1_16x8b, src4_16x8b);
|
|
|
|
sad1_8x16b = _mm_packs_epi32(sad1_8x16b, sad2_8x16b);
|
|
sad3_8x16b = _mm_packs_epi32(sad3_8x16b, sad4_8x16b);
|
|
|
|
cnt = 11;
|
|
sad_8x16b = _mm_packs_epi32(sad1_8x16b, sad3_8x16b);
|
|
do
|
|
{
|
|
pu1_src_temp += src_strd4;
|
|
|
|
src1_16x8b = _mm_loadu_si128((__m128i *)pu1_src_temp);
|
|
src2_16x8b = _mm_loadu_si128((__m128i *)(pu1_src_temp + src_strd));
|
|
src3_16x8b = _mm_loadu_si128((__m128i *)(pu1_src_temp + src_strd2));
|
|
src4_16x8b = _mm_loadu_si128((__m128i *)(pu1_src_temp + src_strd3));
|
|
|
|
sad1_8x16b = _mm_sad_epu8(val1_16x8b, src1_16x8b);
|
|
sad2_8x16b = _mm_sad_epu8(val1_16x8b, src2_16x8b);
|
|
sad3_8x16b = _mm_sad_epu8(val1_16x8b, src3_16x8b);
|
|
sad4_8x16b = _mm_sad_epu8(val1_16x8b, src4_16x8b);
|
|
|
|
sad1_8x16b = _mm_packs_epi32(sad1_8x16b, sad2_8x16b);
|
|
sad3_8x16b = _mm_packs_epi32(sad3_8x16b, sad4_8x16b);
|
|
sad1_8x16b = _mm_packs_epi32(sad1_8x16b, sad3_8x16b);
|
|
|
|
cnt -= 4;
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, sad1_8x16b);
|
|
}
|
|
while(cnt >= 0);
|
|
|
|
sad_8x16b = _mm_hadd_epi16(sad_8x16b, sad_8x16b);
|
|
sad_8x16b = _mm_hadd_epi16(sad_8x16b, sad_8x16b);
|
|
sad_8x16b = _mm_hadd_epi16(sad_8x16b, sad_8x16b);
|
|
|
|
sad_vert = _mm_extract_epi16(sad_8x16b, 0);
|
|
}
|
|
|
|
dcval = 0;
|
|
|
|
if(left)
|
|
{
|
|
val_16x8b = _mm_loadu_si128((__m128i *)pu1_ngbr_pels_i16);
|
|
dcval += 8;
|
|
|
|
sad1_8x16b = _mm_sad_epu8(val_16x8b, zero_vector);
|
|
dcval += _mm_extract_epi16(sad1_8x16b, 0);
|
|
dcval += _mm_extract_epi16(sad1_8x16b, 4);
|
|
}
|
|
if(top)
|
|
{
|
|
val_16x8b = _mm_loadu_si128((__m128i *)(pu1_ngbr_pels_i16 + 17));
|
|
dcval += 8;
|
|
|
|
sad1_8x16b = _mm_sad_epu8(val_16x8b, zero_vector);
|
|
dcval += _mm_extract_epi16(sad1_8x16b, 0);
|
|
dcval += _mm_extract_epi16(sad1_8x16b, 4);
|
|
}
|
|
dcval = dcval >> (3 + left + top);
|
|
dcval += ((left == 0) & (top == 0)) << 7;
|
|
|
|
if(dc_flag)
|
|
{
|
|
pu1_src_temp = pu1_src;
|
|
val1_16x8b = _mm_set1_epi8(dcval);
|
|
|
|
src1_16x8b = _mm_loadu_si128((__m128i *)pu1_src_temp);
|
|
src2_16x8b = _mm_loadu_si128((__m128i *)(pu1_src_temp + src_strd));
|
|
src3_16x8b = _mm_loadu_si128((__m128i *)(pu1_src_temp + src_strd2));
|
|
src4_16x8b = _mm_loadu_si128((__m128i *)(pu1_src_temp + src_strd3));
|
|
|
|
sad1_8x16b = _mm_sad_epu8(val1_16x8b, src1_16x8b);
|
|
sad2_8x16b = _mm_sad_epu8(val1_16x8b, src2_16x8b);
|
|
sad3_8x16b = _mm_sad_epu8(val1_16x8b, src3_16x8b);
|
|
sad4_8x16b = _mm_sad_epu8(val1_16x8b, src4_16x8b);
|
|
|
|
sad1_8x16b = _mm_packs_epi32(sad1_8x16b, sad2_8x16b);
|
|
sad3_8x16b = _mm_packs_epi32(sad3_8x16b, sad4_8x16b);
|
|
|
|
cnt = 12;
|
|
sad_8x16b = _mm_packs_epi32(sad1_8x16b, sad3_8x16b);
|
|
do
|
|
{
|
|
pu1_src_temp += src_strd4;
|
|
|
|
src1_16x8b = _mm_loadu_si128((__m128i *)pu1_src_temp);
|
|
src2_16x8b = _mm_loadu_si128((__m128i *)(pu1_src_temp + src_strd));
|
|
src3_16x8b = _mm_loadu_si128((__m128i *)(pu1_src_temp + src_strd2));
|
|
src4_16x8b = _mm_loadu_si128((__m128i *)(pu1_src_temp + src_strd3));
|
|
|
|
sad1_8x16b = _mm_sad_epu8(val1_16x8b, src1_16x8b);
|
|
sad2_8x16b = _mm_sad_epu8(val1_16x8b, src2_16x8b);
|
|
sad3_8x16b = _mm_sad_epu8(val1_16x8b, src3_16x8b);
|
|
sad4_8x16b = _mm_sad_epu8(val1_16x8b, src4_16x8b);
|
|
|
|
sad1_8x16b = _mm_packs_epi32(sad1_8x16b, sad2_8x16b);
|
|
sad3_8x16b = _mm_packs_epi32(sad3_8x16b, sad4_8x16b);
|
|
sad1_8x16b = _mm_packs_epi32(sad1_8x16b, sad3_8x16b);
|
|
|
|
cnt -= 4;
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, sad1_8x16b);
|
|
}
|
|
while(cnt > 0);
|
|
|
|
sad_8x16b = _mm_hadd_epi16(sad_8x16b, sad_8x16b);
|
|
sad_8x16b = _mm_hadd_epi16(sad_8x16b, sad_8x16b);
|
|
sad_8x16b = _mm_hadd_epi16(sad_8x16b, sad_8x16b);
|
|
|
|
sad_dc = _mm_extract_epi16(sad_8x16b, 0);
|
|
}
|
|
|
|
// Doing prediction for minimum SAD
|
|
min_sad = MIN3(sad_horz, sad_vert, sad_dc);
|
|
if(min_sad < *pu4_sadmin)
|
|
{
|
|
*pu4_sadmin = min_sad;
|
|
if(min_sad == sad_vert)
|
|
{
|
|
*u4_intra_mode = VERT_I16x16;
|
|
val1_16x8b = _mm_loadu_si128((__m128i *)(pu1_ngbr_pels_i16 + 17));
|
|
cnt = 15;
|
|
do
|
|
{
|
|
_mm_storeu_si128((__m128i *)pu1_dst, val1_16x8b);
|
|
_mm_storeu_si128((__m128i *)(pu1_dst + dst_strd), val1_16x8b);
|
|
_mm_storeu_si128((__m128i *)(pu1_dst + dst_strd2), val1_16x8b);
|
|
_mm_storeu_si128((__m128i *)(pu1_dst + dst_strd3), val1_16x8b);
|
|
|
|
cnt -= 4;
|
|
pu1_dst += dst_strd4;
|
|
}
|
|
while(cnt > 0);
|
|
}
|
|
else if(min_sad == sad_horz)
|
|
{
|
|
*u4_intra_mode = HORZ_I16x16;
|
|
cnt = 15;
|
|
do
|
|
{
|
|
val1_16x8b = _mm_set1_epi8(pu1_ngbr_pels_i16[cnt]);
|
|
val2_16x8b = _mm_set1_epi8(pu1_ngbr_pels_i16[cnt - 1]);
|
|
val3_16x8b = _mm_set1_epi8(pu1_ngbr_pels_i16[cnt - 2]);
|
|
val4_16x8b = _mm_set1_epi8(pu1_ngbr_pels_i16[cnt - 3]);
|
|
|
|
_mm_storeu_si128((__m128i *)pu1_dst, val1_16x8b);
|
|
_mm_storeu_si128((__m128i *)(pu1_dst + dst_strd), val2_16x8b);
|
|
_mm_storeu_si128((__m128i *)(pu1_dst + dst_strd2), val3_16x8b);
|
|
_mm_storeu_si128((__m128i *)(pu1_dst + dst_strd3), val4_16x8b);
|
|
|
|
cnt -= 4;
|
|
pu1_dst += dst_strd4;
|
|
}
|
|
while(cnt >= 0);
|
|
}
|
|
else
|
|
{
|
|
*u4_intra_mode = DC_I16x16;
|
|
val1_16x8b = _mm_set1_epi8(dcval);
|
|
cnt = 15;
|
|
do
|
|
{
|
|
_mm_storeu_si128((__m128i *)pu1_dst, val1_16x8b);
|
|
_mm_storeu_si128((__m128i *)(pu1_dst + dst_strd), val1_16x8b);
|
|
_mm_storeu_si128((__m128i *)(pu1_dst + dst_strd2), val1_16x8b);
|
|
_mm_storeu_si128((__m128i *)(pu1_dst + dst_strd3), val1_16x8b);
|
|
|
|
cnt -= 4;
|
|
pu1_dst += dst_strd4;
|
|
}
|
|
while(cnt > 0);
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
******************************************************************************
|
|
*
|
|
* @brief :Evaluate best intra 4x4 mode and do the prediction.
|
|
*
|
|
* @par Description
|
|
* This function evaluates intra 4x4 modes, computes corresponding sad
|
|
* and returns the buffer predicted with best mode.
|
|
*
|
|
* @param[in] pu1_src
|
|
* UWORD8 pointer to the source
|
|
*
|
|
** @param[in] pu1_ngbr_pels
|
|
* UWORD8 pointer to neighbouring pels
|
|
*
|
|
* @param[out] pu1_dst
|
|
* UWORD8 pointer to the destination
|
|
*
|
|
* @param[in] src_strd
|
|
* integer source stride
|
|
*
|
|
* @param[in] dst_strd
|
|
* integer destination stride
|
|
*
|
|
* @param[in] u4_n_avblty
|
|
* availability of neighbouring pixels
|
|
*
|
|
* @param[in] u4_intra_mode
|
|
* Pointer to the variable in which best mode is returned
|
|
*
|
|
* @param[in] pu4_sadmin
|
|
* Pointer to the variable in which minimum cost is returned
|
|
*
|
|
* @param[in] u4_valid_intra_modes
|
|
* Says what all modes are valid
|
|
*
|
|
* * @param[in] u4_lambda
|
|
* Lamda value for computing cost from SAD
|
|
*
|
|
* @param[in] u4_predictd_mode
|
|
* Predicted mode for cost computation
|
|
*
|
|
* @return none
|
|
*
|
|
******************************************************************************
|
|
*/
|
|
void ih264e_evaluate_intra_4x4_modes_ssse3(UWORD8 *pu1_src,
|
|
UWORD8 *pu1_ngbr_pels,
|
|
UWORD8 *pu1_dst,
|
|
UWORD32 src_strd,
|
|
UWORD32 dst_strd,
|
|
WORD32 u4_n_avblty,
|
|
UWORD32 *u4_intra_mode,
|
|
WORD32 *pu4_sadmin,
|
|
UWORD32 u4_valid_intra_modes,
|
|
UWORD32 u4_lambda,
|
|
UWORD32 u4_predictd_mode)
|
|
{
|
|
WORD32 left, top;
|
|
WORD32 sad[MAX_I4x4] = { INT_MAX, INT_MAX, INT_MAX, INT_MAX, INT_MAX,
|
|
INT_MAX, INT_MAX, INT_MAX, INT_MAX };
|
|
WORD32 cost[MAX_I4x4] = { INT_MAX, INT_MAX, INT_MAX, INT_MAX, INT_MAX,
|
|
INT_MAX, INT_MAX, INT_MAX, INT_MAX };
|
|
|
|
WORD32 min_cost;
|
|
UWORD32 lambda4 = u4_lambda << 2;
|
|
WORD32 dst_strd2, dst_strd3;
|
|
|
|
__m128i left_top_16x8b, src_16x8b, pred0_16x8b, sad_8x16b;
|
|
__m128i pred1_16x8b, pred2_16x8b, pred3_16x8b, pred4_16x8b;
|
|
__m128i pred5_16x8b, pred6_16x8b, pred7_16x8b, pred8_16x8b;
|
|
__m128i shuffle_16x8b, zero_vector, mask_low_32b;
|
|
|
|
left = (u4_n_avblty & LEFT_MB_AVAILABLE_MASK);
|
|
top = (u4_n_avblty & TOP_MB_AVAILABLE_MASK) >> 2;
|
|
|
|
dst_strd2 = dst_strd << 1;
|
|
dst_strd3 = dst_strd + dst_strd2;
|
|
|
|
// loading the 4x4 source block and neighbouring pixels
|
|
{
|
|
__m128i row1_16x8b, row2_16x8b;
|
|
|
|
row1_16x8b = _mm_loadl_epi64((__m128i *)pu1_src);
|
|
row2_16x8b = _mm_loadl_epi64((__m128i *)(pu1_src + src_strd));
|
|
left_top_16x8b = _mm_loadu_si128((__m128i *)pu1_ngbr_pels);
|
|
|
|
pu1_src += src_strd << 1;
|
|
src_16x8b = _mm_unpacklo_epi32(row1_16x8b, row2_16x8b);
|
|
|
|
row1_16x8b = _mm_loadl_epi64((__m128i *)pu1_src);
|
|
row2_16x8b = _mm_loadl_epi64((__m128i *)(pu1_src + src_strd));
|
|
zero_vector = _mm_setzero_si128();
|
|
|
|
row1_16x8b = _mm_unpacklo_epi32(row1_16x8b, row2_16x8b);
|
|
src_16x8b = _mm_unpacklo_epi64(src_16x8b, row1_16x8b);
|
|
}
|
|
|
|
/* Computing SADs*/
|
|
if(u4_valid_intra_modes & 1)/* VERT mode valid ????*/
|
|
{
|
|
pred0_16x8b = _mm_srli_si128(left_top_16x8b, 5);
|
|
pred0_16x8b = _mm_shuffle_epi32(pred0_16x8b, 0);
|
|
sad_8x16b = _mm_sad_epu8(src_16x8b, pred0_16x8b);
|
|
|
|
sad[VERT_I4x4] = _mm_extract_epi16(sad_8x16b, 0) + _mm_extract_epi16(sad_8x16b, 4);
|
|
cost[VERT_I4x4] = sad[VERT_I4x4] + ((u4_predictd_mode == VERT_I4x4) ? u4_lambda: lambda4);
|
|
}
|
|
|
|
if(u4_valid_intra_modes & 2)/* HORZ mode valid ????*/
|
|
{
|
|
shuffle_16x8b = _mm_setr_epi8(3, 3, 3, 3, 2, 2, 2, 2, 1, 1, 1, 1, 0, 0, 0, 0);
|
|
pred1_16x8b = _mm_shuffle_epi8(left_top_16x8b, shuffle_16x8b);
|
|
|
|
sad_8x16b = _mm_sad_epu8(src_16x8b, pred1_16x8b);
|
|
|
|
sad[HORZ_I4x4] = _mm_extract_epi16(sad_8x16b, 0) + _mm_extract_epi16(sad_8x16b, 4);
|
|
cost[HORZ_I4x4] = sad[HORZ_I4x4] + ((u4_predictd_mode == HORZ_I4x4) ? u4_lambda: lambda4);
|
|
}
|
|
|
|
if(u4_valid_intra_modes & 4)/* DC mode valid ????*/
|
|
{
|
|
if(top + left)
|
|
{
|
|
WORD32 shft = 1, dcval = 0;
|
|
|
|
__m128i val_16x8b, temp_16x8b, temp_8x16b;
|
|
|
|
val_16x8b = _mm_setzero_si128();
|
|
|
|
if(top)
|
|
{
|
|
temp_16x8b = _mm_srli_si128(left_top_16x8b, 5);
|
|
val_16x8b = _mm_alignr_epi8(temp_16x8b, val_16x8b, 4);
|
|
shft ++;
|
|
dcval += 2;
|
|
}
|
|
if(left)
|
|
{
|
|
val_16x8b = _mm_alignr_epi8(left_top_16x8b, val_16x8b, 4);
|
|
shft++;
|
|
dcval += 2;
|
|
}
|
|
|
|
temp_8x16b = _mm_sad_epu8(val_16x8b, zero_vector);
|
|
dcval += _mm_extract_epi16(temp_8x16b, 4);
|
|
dcval = dcval >> shft;
|
|
pred2_16x8b = _mm_set1_epi8(dcval);
|
|
}
|
|
else
|
|
pred2_16x8b = _mm_set1_epi8(-128);
|
|
|
|
sad_8x16b = _mm_sad_epu8(src_16x8b, pred2_16x8b);
|
|
|
|
sad[DC_I4x4] = _mm_extract_epi16(sad_8x16b, 0) + _mm_extract_epi16(sad_8x16b, 4);
|
|
cost[DC_I4x4] = sad[DC_I4x4] + ((u4_predictd_mode == DC_I4x4) ? u4_lambda: lambda4);
|
|
}
|
|
|
|
if(u4_valid_intra_modes > 7)/* if modes other than VERT, HORZ and DC are valid ????*/
|
|
{
|
|
__m128i w11_16x8b, w121_16x8b;
|
|
__m128i temp1_16x8b, temp2_16x8b;
|
|
|
|
/* Performing FILT121 and FILT11 operation for all neighbour values*/
|
|
{
|
|
__m128i temp1_8x16b, temp2_8x16b, temp3_8x16b;
|
|
__m128i const_2_8x16b;
|
|
|
|
const_2_8x16b = _mm_set1_epi16(2);
|
|
|
|
temp1_8x16b = _mm_unpacklo_epi8(left_top_16x8b, zero_vector); //l3 l2 l1 l0 tl t0 t1 t2
|
|
temp2_8x16b = _mm_slli_si128(temp1_8x16b, 2); // 0 l3 l2 l1 l0 tl t0 t1
|
|
temp2_8x16b = _mm_shufflelo_epi16(temp2_8x16b, 0xe5); //l3 l3 l2 l1 l0 tl t0 t1
|
|
|
|
temp1_8x16b = _mm_add_epi16(temp1_8x16b, temp2_8x16b); //l3+l3 l3+l2 l2+l1... t1+t2
|
|
temp2_8x16b = _mm_slli_si128(temp1_8x16b, 2); //l3+l3 l3+l3 l3+l2... t0+t1
|
|
temp2_8x16b = _mm_shufflelo_epi16(temp2_8x16b, 0xe5);
|
|
temp1_8x16b = _mm_add_epi16(temp1_8x16b, temp2_8x16b); //4*l3 l3+2*l3+l2 l3+2*l2+l1... t0+2*t1+t2
|
|
|
|
temp1_8x16b = _mm_add_epi16(const_2_8x16b, temp1_8x16b); //4*l3+2 3*l3+l2+2 l3+2*l2+l1+2.. t0+2*t1+t2+2
|
|
temp1_8x16b = _mm_srli_epi16(temp1_8x16b, 2);
|
|
|
|
temp1_16x8b = _mm_srli_si128(left_top_16x8b, 1);
|
|
w11_16x8b = _mm_avg_epu8(left_top_16x8b, temp1_16x8b);
|
|
|
|
temp2_16x8b = _mm_srli_si128(left_top_16x8b, 6);
|
|
temp2_8x16b = _mm_unpacklo_epi8(temp2_16x8b, zero_vector); //t1 t2 t3 t4 t5 t6 t7 0
|
|
temp3_8x16b = _mm_srli_si128(temp2_8x16b, 2); //t2 t3 t4 t5 t6 t7 0 0
|
|
temp3_8x16b = _mm_shufflehi_epi16(temp3_8x16b, 0xd4); //t2 t3 t4 t5 t6 t7 t7 0
|
|
|
|
temp2_8x16b = _mm_add_epi16(temp2_8x16b, temp3_8x16b); //t1+t2 t2+t3... t6+t7 t7+t7 0
|
|
temp3_8x16b = _mm_srli_si128(temp2_8x16b, 2); //t2+t3 t3+t4... t7+t7 0 0
|
|
temp2_8x16b = _mm_add_epi16(temp2_8x16b, temp3_8x16b); //t1+2*t2+t3 t2+2*t3+t4.. t6+2*t7+t7 t7+t7 0
|
|
|
|
temp2_8x16b = _mm_add_epi16(const_2_8x16b, temp2_8x16b); //t1+2*t2+t3+2 t2+2*t3+t4+2 t3+2*t4+t5+2... t6+2*t7+t7+2 t7+t7+2 2
|
|
temp2_8x16b = _mm_srli_epi16(temp2_8x16b, 2);
|
|
|
|
w121_16x8b = _mm_packus_epi16(temp1_8x16b, temp2_8x16b);
|
|
}
|
|
|
|
if(u4_valid_intra_modes & 8)/* DIAG_DL */
|
|
{
|
|
shuffle_16x8b = _mm_setr_epi8( 7, 8, 9, 10,
|
|
8, 9, 10, 11,
|
|
9, 10, 11, 12,
|
|
10, 11, 12, 13);
|
|
pred3_16x8b = _mm_shuffle_epi8(w121_16x8b, shuffle_16x8b);
|
|
sad_8x16b = _mm_sad_epu8(src_16x8b, pred3_16x8b);
|
|
|
|
sad[DIAG_DL_I4x4] = _mm_extract_epi16(sad_8x16b, 0) + _mm_extract_epi16(sad_8x16b, 4);
|
|
cost[DIAG_DL_I4x4] = sad[DIAG_DL_I4x4] + ((u4_predictd_mode == DIAG_DL_I4x4) ? u4_lambda: lambda4);
|
|
}
|
|
|
|
if(u4_valid_intra_modes & 16)/* DIAG_DR */
|
|
{
|
|
shuffle_16x8b = _mm_setr_epi8(5, 6, 7, 8,
|
|
4, 5, 6, 7,
|
|
3, 4, 5, 6,
|
|
2, 3, 4, 5);
|
|
pred4_16x8b = _mm_shuffle_epi8(w121_16x8b, shuffle_16x8b);
|
|
sad_8x16b = _mm_sad_epu8(src_16x8b, pred4_16x8b);
|
|
|
|
sad[DIAG_DR_I4x4] = _mm_extract_epi16(sad_8x16b, 0) + _mm_extract_epi16(sad_8x16b, 4);
|
|
cost[DIAG_DR_I4x4] = sad[DIAG_DR_I4x4] + ((u4_predictd_mode == DIAG_DR_I4x4) ? u4_lambda: lambda4);
|
|
}
|
|
|
|
if(u4_valid_intra_modes & 32)/* VERT_R mode valid ????*/
|
|
{
|
|
temp1_16x8b = _mm_srli_si128(w121_16x8b, 1);
|
|
temp1_16x8b = _mm_unpacklo_epi64(temp1_16x8b, w11_16x8b);
|
|
shuffle_16x8b = _mm_setr_epi8(12, 13, 14, 15,
|
|
4, 5, 6, 7,
|
|
3, 12, 13, 14,
|
|
2, 4, 5, 6);
|
|
pred5_16x8b = _mm_shuffle_epi8(temp1_16x8b, shuffle_16x8b);
|
|
sad_8x16b = _mm_sad_epu8(src_16x8b, pred5_16x8b);
|
|
|
|
sad[VERT_R_I4x4] = _mm_extract_epi16(sad_8x16b, 0) + _mm_extract_epi16(sad_8x16b, 4);
|
|
cost[VERT_R_I4x4] = sad[VERT_R_I4x4] + ((u4_predictd_mode == VERT_R_I4x4) ? u4_lambda: lambda4);
|
|
}
|
|
|
|
if(u4_valid_intra_modes & 64)/* HORZ_D mode valid ????*/
|
|
{
|
|
temp1_16x8b = _mm_unpacklo_epi64(w121_16x8b, w11_16x8b);
|
|
shuffle_16x8b = _mm_setr_epi8(11, 5, 6, 7,
|
|
10, 4, 11, 5,
|
|
9, 3, 10, 4,
|
|
8, 2, 9, 3);
|
|
pred6_16x8b = _mm_shuffle_epi8(temp1_16x8b, shuffle_16x8b);
|
|
sad_8x16b = _mm_sad_epu8(src_16x8b, pred6_16x8b);
|
|
|
|
sad[HORZ_D_I4x4] = _mm_extract_epi16(sad_8x16b, 0) + _mm_extract_epi16(sad_8x16b, 4);
|
|
cost[HORZ_D_I4x4] = sad[HORZ_D_I4x4] + ((u4_predictd_mode == HORZ_D_I4x4) ? u4_lambda: lambda4);
|
|
}
|
|
|
|
if(u4_valid_intra_modes & 128)/* VERT_L mode valid ????*/
|
|
{
|
|
temp1_16x8b = _mm_srli_si128(w121_16x8b, 5);
|
|
temp2_16x8b = _mm_srli_si128(w11_16x8b, 5);
|
|
temp1_16x8b = _mm_unpacklo_epi64(temp1_16x8b, temp2_16x8b);
|
|
shuffle_16x8b = _mm_setr_epi8(8, 9, 10, 11,
|
|
2, 3, 4, 5,
|
|
9, 10, 11, 12,
|
|
3, 4, 5, 6);
|
|
pred7_16x8b = _mm_shuffle_epi8(temp1_16x8b, shuffle_16x8b);
|
|
sad_8x16b = _mm_sad_epu8(src_16x8b, pred7_16x8b);
|
|
|
|
sad[VERT_L_I4x4] = _mm_extract_epi16(sad_8x16b, 0) + _mm_extract_epi16(sad_8x16b, 4);
|
|
cost[VERT_L_I4x4] = sad[VERT_L_I4x4] + ((u4_predictd_mode == VERT_L_I4x4) ? u4_lambda: lambda4);
|
|
}
|
|
|
|
if(u4_valid_intra_modes & 256)/* HORZ_U mode valid ????*/
|
|
{
|
|
temp1_16x8b = _mm_unpacklo_epi64(w121_16x8b, w11_16x8b);
|
|
shuffle_16x8b = _mm_setr_epi8(10, 3, 9, 2,
|
|
9, 2, 8, 1,
|
|
8, 1, 0, 0,
|
|
0, 0, 0, 0);
|
|
pred8_16x8b = _mm_shuffle_epi8(temp1_16x8b, shuffle_16x8b);
|
|
sad_8x16b = _mm_sad_epu8(src_16x8b, pred8_16x8b);
|
|
|
|
sad[HORZ_U_I4x4] = _mm_extract_epi16(sad_8x16b, 0) + _mm_extract_epi16(sad_8x16b, 4);
|
|
cost[HORZ_U_I4x4] = sad[HORZ_U_I4x4] + ((u4_predictd_mode == HORZ_U_I4x4) ? u4_lambda: lambda4);
|
|
}
|
|
|
|
min_cost = MIN3(MIN3(cost[0], cost[1], cost[2]),
|
|
MIN3(cost[3], cost[4], cost[5]),
|
|
MIN3(cost[6], cost[7], cost[8]));
|
|
}
|
|
else
|
|
{ /*Only first three modes valid*/
|
|
min_cost = MIN3(cost[0], cost[1], cost[2]);
|
|
}
|
|
|
|
*pu4_sadmin = min_cost;
|
|
|
|
if(min_cost == cost[0])
|
|
{
|
|
*u4_intra_mode = VERT_I4x4;
|
|
}
|
|
else if(min_cost == cost[1])
|
|
{
|
|
*u4_intra_mode = HORZ_I4x4;
|
|
pred0_16x8b = pred1_16x8b;
|
|
}
|
|
else if(min_cost == cost[2])
|
|
{
|
|
*u4_intra_mode = DC_I4x4;
|
|
pred0_16x8b = pred2_16x8b;
|
|
}
|
|
else if(min_cost == cost[3])
|
|
{
|
|
*u4_intra_mode = DIAG_DL_I4x4;
|
|
pred0_16x8b = pred3_16x8b;
|
|
}
|
|
else if(min_cost == cost[4])
|
|
{
|
|
*u4_intra_mode = DIAG_DR_I4x4;
|
|
pred0_16x8b = pred4_16x8b;
|
|
}
|
|
else if(min_cost == cost[5])
|
|
{
|
|
*u4_intra_mode = VERT_R_I4x4;
|
|
pred0_16x8b = pred5_16x8b;
|
|
}
|
|
else if(min_cost == cost[6])
|
|
{
|
|
*u4_intra_mode = HORZ_D_I4x4;
|
|
pred0_16x8b = pred6_16x8b;
|
|
}
|
|
else if(min_cost == cost[7])
|
|
{
|
|
*u4_intra_mode = VERT_L_I4x4;
|
|
pred0_16x8b = pred7_16x8b;
|
|
}
|
|
else if(min_cost == cost[8])
|
|
{
|
|
*u4_intra_mode = HORZ_U_I4x4;
|
|
pred0_16x8b = pred8_16x8b;
|
|
}
|
|
|
|
mask_low_32b = _mm_set1_epi8(0xff);
|
|
mask_low_32b = _mm_srli_si128(mask_low_32b, 12);
|
|
|
|
_mm_maskmoveu_si128(pred0_16x8b, mask_low_32b, (char*)pu1_dst);
|
|
pred0_16x8b = _mm_srli_si128(pred0_16x8b, 4);
|
|
_mm_maskmoveu_si128(pred0_16x8b, mask_low_32b, (char*)(pu1_dst + dst_strd));
|
|
pred0_16x8b = _mm_srli_si128(pred0_16x8b, 4);
|
|
_mm_maskmoveu_si128(pred0_16x8b, mask_low_32b, (char*)(pu1_dst + dst_strd2));
|
|
pred0_16x8b = _mm_srli_si128(pred0_16x8b, 4);
|
|
_mm_maskmoveu_si128(pred0_16x8b, mask_low_32b, (char*)(pu1_dst + dst_strd3));
|
|
|
|
}
|
|
|
|
/**
|
|
******************************************************************************
|
|
*
|
|
* @brief
|
|
* Evaluate best intra chroma mode (among VERT, HORZ and DC) and do the prediction.
|
|
*
|
|
* @par Description
|
|
* This function evaluates first three intra chroma modes and compute corresponding sad
|
|
* and return the buffer predicted with best mode.
|
|
*
|
|
* @param[in] pu1_src
|
|
* UWORD8 pointer to the source
|
|
*
|
|
** @param[in] pu1_ngbr_pels
|
|
* UWORD8 pointer to neighbouring pels
|
|
*
|
|
* @param[out] pu1_dst
|
|
* UWORD8 pointer to the destination
|
|
*
|
|
* @param[in] src_strd
|
|
* integer source stride
|
|
*
|
|
* @param[in] dst_strd
|
|
* integer destination stride
|
|
*
|
|
* @param[in] u4_n_avblty
|
|
* availability of neighbouring pixels
|
|
*
|
|
* @param[in] u4_intra_mode
|
|
* pointer to the variable in which best mode is returned
|
|
*
|
|
* @param[in] pu4_sadmin
|
|
* pointer to the variable in which minimum sad is returned
|
|
*
|
|
* @param[in] u4_valid_intra_modes
|
|
* says what all modes are valid
|
|
*
|
|
* @return
|
|
* none
|
|
*
|
|
******************************************************************************
|
|
*/
|
|
|
|
void ih264e_evaluate_intra_chroma_modes_ssse3(UWORD8 *pu1_src,
|
|
UWORD8 *pu1_ngbr_pels,
|
|
UWORD8 *pu1_dst,
|
|
UWORD32 src_strd,
|
|
UWORD32 dst_strd,
|
|
WORD32 u4_n_avblty,
|
|
UWORD32 *u4_intra_mode,
|
|
WORD32 *pu4_sadmin,
|
|
UWORD32 u4_valid_intra_modes)
|
|
{
|
|
WORD32 left, top;
|
|
WORD32 sad_vert = INT_MAX, sad_horz = INT_MAX, sad_dc = INT_MAX, min_sad;
|
|
|
|
__m128i src1_16x8b, src2_16x8b, src3_16x8b, src4_16x8b;
|
|
__m128i src5_16x8b, src6_16x8b, src7_16x8b, src8_16x8b;
|
|
|
|
__m128i top_16x8b, left_16x8b;
|
|
__m128i pred1_16x8b, pred2_16x8b;
|
|
__m128i tmp1_8x16b, tmp2_8x16b, sad_8x16b;
|
|
|
|
left = (u4_n_avblty & LEFT_MB_AVAILABLE_MASK);
|
|
top = (u4_n_avblty & TOP_MB_AVAILABLE_MASK) >> 2;
|
|
|
|
//Loading source
|
|
{
|
|
src1_16x8b = _mm_loadu_si128((__m128i *)pu1_src);
|
|
pu1_src += src_strd;
|
|
src2_16x8b = _mm_loadu_si128((__m128i *)pu1_src);
|
|
pu1_src += src_strd;
|
|
src3_16x8b = _mm_loadu_si128((__m128i *)pu1_src);
|
|
pu1_src += src_strd;
|
|
src4_16x8b = _mm_loadu_si128((__m128i *)pu1_src);
|
|
pu1_src += src_strd;
|
|
src5_16x8b = _mm_loadu_si128((__m128i *)pu1_src);
|
|
pu1_src += src_strd;
|
|
src6_16x8b = _mm_loadu_si128((__m128i *)pu1_src);
|
|
pu1_src += src_strd;
|
|
src7_16x8b = _mm_loadu_si128((__m128i *)pu1_src);
|
|
pu1_src += src_strd;
|
|
src8_16x8b = _mm_loadu_si128((__m128i *)pu1_src);
|
|
}
|
|
|
|
if(left)
|
|
{
|
|
left_16x8b = _mm_loadu_si128((__m128i *)pu1_ngbr_pels);
|
|
|
|
if(u4_valid_intra_modes & 02) //If HORZ mode is valid
|
|
{
|
|
__m128i left_tmp_16x8b, left_sh_16x8b;
|
|
__m128i const_14_15_16x8b;
|
|
|
|
const_14_15_16x8b = _mm_set1_epi16(0x0f0e);
|
|
left_sh_16x8b = _mm_slli_si128(left_16x8b, 2);
|
|
|
|
pred1_16x8b = _mm_shuffle_epi8(left_16x8b, const_14_15_16x8b); //row 1
|
|
pred2_16x8b = _mm_shuffle_epi8(left_sh_16x8b, const_14_15_16x8b); //row 2
|
|
tmp1_8x16b = _mm_sad_epu8(src1_16x8b, pred1_16x8b);
|
|
tmp2_8x16b = _mm_sad_epu8(src2_16x8b, pred2_16x8b);
|
|
|
|
left_tmp_16x8b = _mm_slli_si128(left_16x8b, 4);
|
|
left_sh_16x8b = _mm_slli_si128(left_sh_16x8b, 4);
|
|
sad_8x16b = _mm_add_epi16(tmp1_8x16b, tmp2_8x16b);
|
|
|
|
pred1_16x8b = _mm_shuffle_epi8(left_tmp_16x8b, const_14_15_16x8b); //row 3
|
|
pred2_16x8b = _mm_shuffle_epi8(left_sh_16x8b, const_14_15_16x8b); //row 4
|
|
tmp1_8x16b = _mm_sad_epu8(src3_16x8b, pred1_16x8b);
|
|
tmp2_8x16b = _mm_sad_epu8(src4_16x8b, pred2_16x8b);
|
|
|
|
left_tmp_16x8b = _mm_slli_si128(left_tmp_16x8b, 4);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp1_8x16b);
|
|
left_sh_16x8b = _mm_slli_si128(left_sh_16x8b, 4);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp2_8x16b);
|
|
|
|
pred1_16x8b = _mm_shuffle_epi8(left_tmp_16x8b, const_14_15_16x8b); //row 5
|
|
pred2_16x8b = _mm_shuffle_epi8(left_sh_16x8b, const_14_15_16x8b); //row 6
|
|
tmp1_8x16b = _mm_sad_epu8(src5_16x8b, pred1_16x8b);
|
|
tmp2_8x16b = _mm_sad_epu8(src6_16x8b, pred2_16x8b);
|
|
|
|
left_tmp_16x8b = _mm_slli_si128(left_tmp_16x8b, 4);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp1_8x16b);
|
|
left_sh_16x8b = _mm_slli_si128(left_sh_16x8b, 4);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp2_8x16b);
|
|
|
|
pred1_16x8b = _mm_shuffle_epi8(left_tmp_16x8b, const_14_15_16x8b); //row 7
|
|
pred2_16x8b = _mm_shuffle_epi8(left_sh_16x8b, const_14_15_16x8b); //row 8
|
|
tmp1_8x16b = _mm_sad_epu8(src7_16x8b, pred1_16x8b);
|
|
tmp2_8x16b = _mm_sad_epu8(src8_16x8b, pred2_16x8b);
|
|
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp1_8x16b);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp2_8x16b);
|
|
|
|
sad_horz = _mm_extract_epi16(sad_8x16b, 0) + _mm_extract_epi16(sad_8x16b, 4);
|
|
}
|
|
}
|
|
|
|
if(top)
|
|
{
|
|
UWORD8 *pu1_top;
|
|
|
|
pu1_top = pu1_ngbr_pels + 2 * BLK8x8SIZE + 2;
|
|
top_16x8b = _mm_loadu_si128((__m128i *)pu1_top);
|
|
|
|
if(u4_valid_intra_modes & 04) //If VERT mode is valid
|
|
{
|
|
tmp1_8x16b = _mm_sad_epu8(src1_16x8b, top_16x8b);
|
|
tmp2_8x16b = _mm_sad_epu8(src2_16x8b, top_16x8b);
|
|
sad_8x16b = _mm_add_epi16(tmp1_8x16b, tmp2_8x16b);
|
|
|
|
tmp1_8x16b = _mm_sad_epu8(src3_16x8b, top_16x8b);
|
|
tmp2_8x16b = _mm_sad_epu8(src4_16x8b, top_16x8b);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp1_8x16b);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp2_8x16b);
|
|
|
|
tmp1_8x16b = _mm_sad_epu8(src5_16x8b, top_16x8b);
|
|
tmp2_8x16b = _mm_sad_epu8(src6_16x8b, top_16x8b);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp1_8x16b);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp2_8x16b);
|
|
|
|
tmp1_8x16b = _mm_sad_epu8(src7_16x8b, top_16x8b);
|
|
tmp2_8x16b = _mm_sad_epu8(src8_16x8b, top_16x8b);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp1_8x16b);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp2_8x16b);
|
|
|
|
sad_vert = _mm_extract_epi16(sad_8x16b, 0) + _mm_extract_epi16(sad_8x16b, 4);
|
|
}
|
|
}
|
|
|
|
if(u4_valid_intra_modes & 01) //If DC mode is valid
|
|
{
|
|
if(left && top)
|
|
{
|
|
WORD32 left_up_u, left_down_u, left_up_v, left_down_v;
|
|
WORD32 top_left_u, top_right_u, top_left_v, top_right_v;
|
|
WORD32 dc_1u, dc_1v, dc_2u, dc_2v;
|
|
|
|
__m128i val_sh_16x8b;
|
|
__m128i intrlv_mask_8x16b, zero_vector;
|
|
|
|
intrlv_mask_8x16b = _mm_set1_epi16(0x00ff);
|
|
zero_vector = _mm_setzero_si128();
|
|
|
|
val_sh_16x8b = _mm_srli_si128(left_16x8b, 1);
|
|
|
|
tmp1_8x16b = _mm_and_si128(intrlv_mask_8x16b, left_16x8b);
|
|
tmp2_8x16b = _mm_and_si128(intrlv_mask_8x16b, val_sh_16x8b);
|
|
tmp1_8x16b = _mm_sad_epu8(zero_vector, tmp1_8x16b);
|
|
tmp2_8x16b = _mm_sad_epu8(zero_vector, tmp2_8x16b);
|
|
|
|
left_up_u = _mm_extract_epi16(tmp1_8x16b, 4);
|
|
left_up_v = _mm_extract_epi16(tmp2_8x16b, 4);
|
|
left_down_u = _mm_extract_epi16(tmp1_8x16b, 0);
|
|
left_down_v = _mm_extract_epi16(tmp2_8x16b, 0);
|
|
|
|
val_sh_16x8b = _mm_srli_si128(top_16x8b, 1);
|
|
|
|
tmp1_8x16b = _mm_and_si128(intrlv_mask_8x16b, top_16x8b);
|
|
tmp2_8x16b = _mm_and_si128(intrlv_mask_8x16b, val_sh_16x8b);
|
|
tmp1_8x16b = _mm_sad_epu8(zero_vector, tmp1_8x16b);
|
|
tmp2_8x16b = _mm_sad_epu8(zero_vector, tmp2_8x16b);
|
|
|
|
top_left_u = _mm_extract_epi16(tmp1_8x16b, 0);
|
|
top_left_v = _mm_extract_epi16(tmp2_8x16b, 0);
|
|
top_right_u = _mm_extract_epi16(tmp1_8x16b, 4);
|
|
top_right_v = _mm_extract_epi16(tmp2_8x16b, 4);
|
|
|
|
// First four rows
|
|
dc_1u = (left_up_u + top_left_u + 4) >> 3;
|
|
dc_1v = (left_up_v + top_left_v + 4) >> 3;
|
|
dc_2u = (top_right_u + 2) >> 2;
|
|
dc_2v = (top_right_v + 2) >> 2;
|
|
|
|
pred1_16x8b = _mm_setr_epi8(dc_1u, dc_1v, dc_1u, dc_1v, dc_1u, dc_1v, dc_1u, dc_1v,
|
|
dc_2u, dc_2v, dc_2u, dc_2v, dc_2u, dc_2v, dc_2u, dc_2v);
|
|
|
|
tmp1_8x16b = _mm_sad_epu8(src1_16x8b, pred1_16x8b);
|
|
tmp2_8x16b = _mm_sad_epu8(src2_16x8b, pred1_16x8b);
|
|
sad_8x16b = _mm_add_epi16(tmp1_8x16b, tmp2_8x16b);
|
|
|
|
tmp1_8x16b = _mm_sad_epu8(src3_16x8b, pred1_16x8b);
|
|
tmp2_8x16b = _mm_sad_epu8(src4_16x8b, pred1_16x8b);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp1_8x16b);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp2_8x16b);
|
|
|
|
// Second four rows
|
|
dc_1u = (left_down_u + 2) >> 2;
|
|
dc_1v = (left_down_v + 2) >> 2;
|
|
dc_2u = (left_down_u + top_right_u + 4) >> 3;
|
|
dc_2v = (left_down_v + top_right_v + 4) >> 3;
|
|
|
|
pred2_16x8b = _mm_setr_epi8(dc_1u, dc_1v, dc_1u, dc_1v, dc_1u, dc_1v, dc_1u, dc_1v,
|
|
dc_2u, dc_2v, dc_2u, dc_2v, dc_2u, dc_2v, dc_2u, dc_2v);
|
|
|
|
tmp1_8x16b = _mm_sad_epu8(src5_16x8b, pred2_16x8b);
|
|
tmp2_8x16b = _mm_sad_epu8(src6_16x8b, pred2_16x8b);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp1_8x16b);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp2_8x16b);
|
|
|
|
tmp1_8x16b = _mm_sad_epu8(src7_16x8b, pred2_16x8b);
|
|
tmp2_8x16b = _mm_sad_epu8(src8_16x8b, pred2_16x8b);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp1_8x16b);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp2_8x16b);
|
|
|
|
sad_dc = _mm_extract_epi16(sad_8x16b, 0) + _mm_extract_epi16(sad_8x16b, 4);
|
|
}
|
|
else if(left)
|
|
{
|
|
WORD32 left_up_u, left_down_u, left_up_v, left_down_v;
|
|
WORD32 dc_u, dc_v;
|
|
|
|
__m128i left_sh_16x8b;
|
|
__m128i intrlv_mask_8x16b, zero_vector;
|
|
|
|
intrlv_mask_8x16b = _mm_set1_epi16(0x00ff);
|
|
zero_vector = _mm_setzero_si128();
|
|
|
|
left_sh_16x8b = _mm_srli_si128(left_16x8b, 1);
|
|
|
|
tmp1_8x16b = _mm_and_si128(intrlv_mask_8x16b, left_16x8b);
|
|
tmp2_8x16b = _mm_and_si128(intrlv_mask_8x16b, left_sh_16x8b);
|
|
tmp1_8x16b = _mm_sad_epu8(zero_vector, tmp1_8x16b);
|
|
tmp2_8x16b = _mm_sad_epu8(zero_vector, tmp2_8x16b);
|
|
|
|
left_up_u = _mm_extract_epi16(tmp1_8x16b, 4);
|
|
left_up_v = _mm_extract_epi16(tmp2_8x16b, 4);
|
|
left_down_u = _mm_extract_epi16(tmp1_8x16b, 0);
|
|
left_down_v = _mm_extract_epi16(tmp2_8x16b, 0);
|
|
|
|
// First four rows
|
|
dc_u = (left_up_u + 2) >> 2;
|
|
dc_v = (left_up_v + 2) >> 2;
|
|
|
|
pred1_16x8b = _mm_set1_epi16(dc_u | (dc_v << 8));
|
|
|
|
tmp1_8x16b = _mm_sad_epu8(src1_16x8b, pred1_16x8b);
|
|
tmp2_8x16b = _mm_sad_epu8(src2_16x8b, pred1_16x8b);
|
|
sad_8x16b = _mm_add_epi16(tmp1_8x16b, tmp2_8x16b);
|
|
|
|
tmp1_8x16b = _mm_sad_epu8(src3_16x8b, pred1_16x8b);
|
|
tmp2_8x16b = _mm_sad_epu8(src4_16x8b, pred1_16x8b);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp1_8x16b);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp2_8x16b);
|
|
|
|
// Second four rows
|
|
dc_u = (left_down_u + 2) >> 2;
|
|
dc_v = (left_down_v + 2) >> 2;
|
|
|
|
pred2_16x8b = _mm_set1_epi16(dc_u | (dc_v << 8));
|
|
|
|
tmp1_8x16b = _mm_sad_epu8(src5_16x8b, pred2_16x8b);
|
|
tmp2_8x16b = _mm_sad_epu8(src6_16x8b, pred2_16x8b);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp1_8x16b);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp2_8x16b);
|
|
|
|
tmp1_8x16b = _mm_sad_epu8(src7_16x8b, pred2_16x8b);
|
|
tmp2_8x16b = _mm_sad_epu8(src8_16x8b, pred2_16x8b);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp1_8x16b);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp2_8x16b);
|
|
|
|
sad_dc = _mm_extract_epi16(sad_8x16b, 0) + _mm_extract_epi16(sad_8x16b, 4);
|
|
}
|
|
else if(top)
|
|
{
|
|
WORD32 top_left_u, top_right_u, top_left_v, top_right_v;
|
|
WORD32 dc_1u, dc_1v, dc_2u, dc_2v;
|
|
|
|
__m128i top_sh_16x8b;
|
|
__m128i intrlv_mask_8x16b, zero_vector;
|
|
|
|
intrlv_mask_8x16b = _mm_set1_epi16(0x00ff);
|
|
zero_vector = _mm_setzero_si128();
|
|
|
|
top_sh_16x8b = _mm_srli_si128(top_16x8b, 1);
|
|
|
|
tmp1_8x16b = _mm_and_si128(intrlv_mask_8x16b, top_16x8b);
|
|
tmp2_8x16b = _mm_and_si128(intrlv_mask_8x16b, top_sh_16x8b);
|
|
tmp1_8x16b = _mm_sad_epu8(zero_vector, tmp1_8x16b);
|
|
tmp2_8x16b = _mm_sad_epu8(zero_vector, tmp2_8x16b);
|
|
|
|
top_left_u = _mm_extract_epi16(tmp1_8x16b, 0);
|
|
top_left_v = _mm_extract_epi16(tmp2_8x16b, 0);
|
|
top_right_u = _mm_extract_epi16(tmp1_8x16b, 4);
|
|
top_right_v = _mm_extract_epi16(tmp2_8x16b, 4);
|
|
|
|
dc_1u = (top_left_u + 2) >> 2;
|
|
dc_1v = (top_left_v + 2) >> 2;
|
|
dc_2u = (top_right_u + 2) >> 2;
|
|
dc_2v = (top_right_v + 2) >> 2;
|
|
|
|
pred1_16x8b = _mm_setr_epi8(dc_1u, dc_1v, dc_1u, dc_1v, dc_1u, dc_1v, dc_1u, dc_1v,
|
|
dc_2u, dc_2v, dc_2u, dc_2v, dc_2u, dc_2v, dc_2u, dc_2v);
|
|
|
|
tmp1_8x16b = _mm_sad_epu8(src1_16x8b, pred1_16x8b);
|
|
tmp2_8x16b = _mm_sad_epu8(src2_16x8b, pred1_16x8b);
|
|
sad_8x16b = _mm_add_epi16(tmp1_8x16b, tmp2_8x16b);
|
|
|
|
tmp1_8x16b = _mm_sad_epu8(src3_16x8b, pred1_16x8b);
|
|
tmp2_8x16b = _mm_sad_epu8(src4_16x8b, pred1_16x8b);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp1_8x16b);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp2_8x16b);
|
|
|
|
tmp1_8x16b = _mm_sad_epu8(src5_16x8b, pred1_16x8b);
|
|
tmp2_8x16b = _mm_sad_epu8(src6_16x8b, pred1_16x8b);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp1_8x16b);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp2_8x16b);
|
|
|
|
tmp1_8x16b = _mm_sad_epu8(src7_16x8b, pred1_16x8b);
|
|
tmp2_8x16b = _mm_sad_epu8(src8_16x8b, pred1_16x8b);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp1_8x16b);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp2_8x16b);
|
|
|
|
sad_dc = _mm_extract_epi16(sad_8x16b, 0) + _mm_extract_epi16(sad_8x16b, 4);
|
|
}
|
|
else
|
|
{
|
|
pred1_16x8b = _mm_set1_epi8(-128);
|
|
|
|
tmp1_8x16b = _mm_sad_epu8(src1_16x8b, pred1_16x8b);
|
|
tmp2_8x16b = _mm_sad_epu8(src2_16x8b, pred1_16x8b);
|
|
sad_8x16b = _mm_add_epi16(tmp1_8x16b, tmp2_8x16b);
|
|
|
|
tmp1_8x16b = _mm_sad_epu8(src3_16x8b, pred1_16x8b);
|
|
tmp2_8x16b = _mm_sad_epu8(src4_16x8b, pred1_16x8b);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp1_8x16b);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp2_8x16b);
|
|
|
|
tmp1_8x16b = _mm_sad_epu8(src5_16x8b, pred1_16x8b);
|
|
tmp2_8x16b = _mm_sad_epu8(src6_16x8b, pred1_16x8b);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp1_8x16b);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp2_8x16b);
|
|
|
|
tmp1_8x16b = _mm_sad_epu8(src7_16x8b, pred1_16x8b);
|
|
tmp2_8x16b = _mm_sad_epu8(src8_16x8b, pred1_16x8b);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp1_8x16b);
|
|
sad_8x16b = _mm_add_epi16(sad_8x16b, tmp2_8x16b);
|
|
|
|
sad_dc = _mm_extract_epi16(sad_8x16b, 0) + _mm_extract_epi16(sad_8x16b, 4);
|
|
}
|
|
}
|
|
|
|
min_sad = MIN3(sad_horz, sad_vert, sad_dc);
|
|
|
|
/* Finding minimum SAD and doing corresponding prediction*/
|
|
if(min_sad < *pu4_sadmin)
|
|
{
|
|
*pu4_sadmin = min_sad;
|
|
|
|
if(min_sad == sad_dc)
|
|
{
|
|
*u4_intra_mode = DC_CH_I8x8;
|
|
|
|
if(!left)
|
|
pred2_16x8b = pred1_16x8b;
|
|
|
|
_mm_storeu_si128((__m128i *)pu1_dst, pred1_16x8b);
|
|
pu1_dst += dst_strd;
|
|
_mm_storeu_si128((__m128i *)pu1_dst, pred1_16x8b);
|
|
pu1_dst += dst_strd;
|
|
_mm_storeu_si128((__m128i *)pu1_dst, pred1_16x8b);
|
|
pu1_dst += dst_strd;
|
|
_mm_storeu_si128((__m128i *)pu1_dst, pred1_16x8b);
|
|
pu1_dst += dst_strd;
|
|
|
|
_mm_storeu_si128((__m128i *)pu1_dst, pred2_16x8b);
|
|
pu1_dst += dst_strd;
|
|
_mm_storeu_si128((__m128i *)pu1_dst, pred2_16x8b);
|
|
pu1_dst += dst_strd;
|
|
_mm_storeu_si128((__m128i *)pu1_dst, pred2_16x8b);
|
|
pu1_dst += dst_strd;
|
|
_mm_storeu_si128((__m128i *)pu1_dst, pred2_16x8b);
|
|
}
|
|
else if(min_sad == sad_horz)
|
|
{
|
|
__m128i left_sh_16x8b, const_14_15_16x8b;
|
|
|
|
*u4_intra_mode = HORZ_CH_I8x8;
|
|
|
|
const_14_15_16x8b = _mm_set1_epi16(0x0f0e);
|
|
|
|
left_sh_16x8b = _mm_slli_si128(left_16x8b, 2);
|
|
pred1_16x8b = _mm_shuffle_epi8(left_16x8b, const_14_15_16x8b); //row 1
|
|
pred2_16x8b = _mm_shuffle_epi8(left_sh_16x8b, const_14_15_16x8b); //row 2
|
|
|
|
_mm_storeu_si128((__m128i *)pu1_dst, pred1_16x8b);
|
|
pu1_dst += dst_strd;
|
|
_mm_storeu_si128((__m128i *)pu1_dst, pred2_16x8b);
|
|
|
|
left_16x8b = _mm_slli_si128(left_16x8b, 4);
|
|
left_sh_16x8b = _mm_slli_si128(left_sh_16x8b, 4);
|
|
pred1_16x8b = _mm_shuffle_epi8(left_16x8b, const_14_15_16x8b); //row 3
|
|
pred2_16x8b = _mm_shuffle_epi8(left_sh_16x8b, const_14_15_16x8b); //row 4
|
|
|
|
pu1_dst += dst_strd;
|
|
_mm_storeu_si128((__m128i *)pu1_dst, pred1_16x8b);
|
|
pu1_dst += dst_strd;
|
|
_mm_storeu_si128((__m128i *)pu1_dst, pred2_16x8b);
|
|
|
|
left_16x8b = _mm_slli_si128(left_16x8b, 4);
|
|
left_sh_16x8b = _mm_slli_si128(left_sh_16x8b, 4);
|
|
pred1_16x8b = _mm_shuffle_epi8(left_16x8b, const_14_15_16x8b); //row 5
|
|
pred2_16x8b = _mm_shuffle_epi8(left_sh_16x8b, const_14_15_16x8b); //row 6
|
|
|
|
pu1_dst += dst_strd;
|
|
_mm_storeu_si128((__m128i *)pu1_dst, pred1_16x8b);
|
|
pu1_dst += dst_strd;
|
|
_mm_storeu_si128((__m128i *)pu1_dst, pred2_16x8b);
|
|
|
|
left_16x8b = _mm_slli_si128(left_16x8b, 4);
|
|
left_sh_16x8b = _mm_slli_si128(left_sh_16x8b, 4);
|
|
pred1_16x8b = _mm_shuffle_epi8(left_16x8b, const_14_15_16x8b); //row 7
|
|
pred2_16x8b = _mm_shuffle_epi8(left_sh_16x8b, const_14_15_16x8b); //row 8
|
|
|
|
pu1_dst += dst_strd;
|
|
_mm_storeu_si128((__m128i *)pu1_dst, pred1_16x8b);
|
|
pu1_dst += dst_strd;
|
|
_mm_storeu_si128((__m128i *)pu1_dst, pred2_16x8b);
|
|
}
|
|
else
|
|
{
|
|
*u4_intra_mode = VERT_CH_I8x8;
|
|
|
|
_mm_storeu_si128((__m128i *)pu1_dst, top_16x8b);
|
|
pu1_dst += dst_strd;
|
|
_mm_storeu_si128((__m128i *)pu1_dst, top_16x8b);
|
|
pu1_dst += dst_strd;
|
|
_mm_storeu_si128((__m128i *)pu1_dst, top_16x8b);
|
|
pu1_dst += dst_strd;
|
|
_mm_storeu_si128((__m128i *)pu1_dst, top_16x8b);
|
|
pu1_dst += dst_strd;
|
|
_mm_storeu_si128((__m128i *)pu1_dst, top_16x8b);
|
|
pu1_dst += dst_strd;
|
|
_mm_storeu_si128((__m128i *)pu1_dst, top_16x8b);
|
|
pu1_dst += dst_strd;
|
|
_mm_storeu_si128((__m128i *)pu1_dst, top_16x8b);
|
|
pu1_dst += dst_strd;
|
|
_mm_storeu_si128((__m128i *)pu1_dst, top_16x8b);
|
|
}
|
|
}
|
|
}
|