mirror of
https://github.com/pineappleEA/pineapple-src.git
synced 2024-12-04 23:28:24 -05:00
3078 lines
116 KiB
C
3078 lines
116 KiB
C
|
/*
|
||
|
* Copyright (c) 2015 Shivraj Patil (Shivraj.Patil@imgtec.com)
|
||
|
*
|
||
|
* This file is part of FFmpeg.
|
||
|
*
|
||
|
* FFmpeg is free software; you can redistribute it and/or
|
||
|
* modify it under the terms of the GNU Lesser General Public
|
||
|
* License as published by the Free Software Foundation; either
|
||
|
* version 2.1 of the License, or (at your option) any later version.
|
||
|
*
|
||
|
* FFmpeg is distributed in the hope that it will be useful,
|
||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||
|
* Lesser General Public License for more details.
|
||
|
*
|
||
|
* You should have received a copy of the GNU Lesser General Public
|
||
|
* License along with FFmpeg; if not, write to the Free Software
|
||
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
||
|
*/
|
||
|
|
||
|
#include "libavcodec/hevcdec.h"
|
||
|
#include "libavutil/mips/generic_macros_msa.h"
|
||
|
#include "hevcpred_mips.h"
|
||
|
|
||
|
static const int8_t intra_pred_angle_up[17] = {
|
||
|
-32, -26, -21, -17, -13, -9, -5, -2, 0, 2, 5, 9, 13, 17, 21, 26, 32
|
||
|
};
|
||
|
|
||
|
static const int8_t intra_pred_angle_low[16] = {
|
||
|
32, 26, 21, 17, 13, 9, 5, 2, 0, -2, -5, -9, -13, -17, -21, -26
|
||
|
};
|
||
|
|
||
|
#define HEVC_PRED_PLANAR_16x2(src0_r, src0_l, tmp0, tmp1, vec0, vec1, \
|
||
|
mul_val_h0, mul_val_h1, mul_val_h2, mul_val_h3, \
|
||
|
res0, res1, mul_val_b0, mul_val_b1, round) \
|
||
|
{ \
|
||
|
v8i16 res0_m, res1_m, res2_m, res3_m; \
|
||
|
\
|
||
|
MUL4(mul_val_h0, vec0, mul_val_h2, vec0, mul_val_h0, vec1, \
|
||
|
mul_val_h2, vec1, res0_m, res1_m, res2_m, res3_m); \
|
||
|
\
|
||
|
res0_m += mul_val_h1 * tmp0; \
|
||
|
res1_m += mul_val_h3 * tmp0; \
|
||
|
res2_m += mul_val_h1 * tmp0; \
|
||
|
res3_m += mul_val_h3 * tmp0; \
|
||
|
\
|
||
|
res0_m += mul_val_b0 * src0_r; \
|
||
|
res1_m += mul_val_b0 * src0_l; \
|
||
|
res2_m += (mul_val_b0 - 1) * src0_r; \
|
||
|
res3_m += (mul_val_b0 - 1) * src0_l; \
|
||
|
\
|
||
|
res0_m += mul_val_b1 * tmp1; \
|
||
|
res1_m += mul_val_b1 * tmp1; \
|
||
|
res2_m += (mul_val_b1 + 1) * tmp1; \
|
||
|
res3_m += (mul_val_b1 + 1) * tmp1; \
|
||
|
\
|
||
|
SRARI_H4_SH(res0_m, res1_m, res2_m, res3_m, round); \
|
||
|
PCKEV_B2_SH(res1_m, res0_m, res3_m, res2_m, res0, res1); \
|
||
|
}
|
||
|
|
||
|
static void hevc_intra_pred_vert_4x4_msa(const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
uint8_t *dst, int32_t stride,
|
||
|
int32_t flag)
|
||
|
{
|
||
|
uint32_t col;
|
||
|
uint32_t src_data;
|
||
|
v8i16 vec0, vec1, vec2;
|
||
|
v16i8 zero = { 0 };
|
||
|
|
||
|
src_data = LW(src_top);
|
||
|
SW4(src_data, src_data, src_data, src_data, dst, stride);
|
||
|
|
||
|
if (0 == flag) {
|
||
|
src_data = LW(src_left);
|
||
|
|
||
|
vec2 = (v8i16) __msa_insert_w((v4i32) vec2, 0, src_data);
|
||
|
|
||
|
vec0 = __msa_fill_h(src_left[-1]);
|
||
|
vec1 = __msa_fill_h(src_top[0]);
|
||
|
|
||
|
vec2 = (v8i16) __msa_ilvr_b(zero, (v16i8) vec2);
|
||
|
vec2 -= vec0;
|
||
|
vec2 >>= 1;
|
||
|
vec2 += vec1;
|
||
|
CLIP_SH_0_255(vec2);
|
||
|
|
||
|
for (col = 0; col < 4; col++) {
|
||
|
dst[stride * col] = (uint8_t) vec2[col];
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
static void hevc_intra_pred_vert_8x8_msa(const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
uint8_t *dst, int32_t stride,
|
||
|
int32_t flag)
|
||
|
{
|
||
|
uint8_t *tmp_dst = dst;
|
||
|
uint32_t row;
|
||
|
uint16_t val0, val1, val2, val3;
|
||
|
uint64_t src_data1;
|
||
|
v8i16 vec0, vec1, vec2;
|
||
|
v16i8 zero = { 0 };
|
||
|
|
||
|
src_data1 = LD(src_top);
|
||
|
|
||
|
for (row = 8; row--;) {
|
||
|
SD(src_data1, tmp_dst);
|
||
|
tmp_dst += stride;
|
||
|
}
|
||
|
|
||
|
if (0 == flag) {
|
||
|
src_data1 = LD(src_left);
|
||
|
|
||
|
vec2 = (v8i16) __msa_insert_d((v2i64) zero, 0, src_data1);
|
||
|
|
||
|
vec0 = __msa_fill_h(src_left[-1]);
|
||
|
vec1 = __msa_fill_h(src_top[0]);
|
||
|
|
||
|
vec2 = (v8i16) __msa_ilvr_b(zero, (v16i8) vec2);
|
||
|
vec2 -= vec0;
|
||
|
vec2 >>= 1;
|
||
|
vec2 += vec1;
|
||
|
CLIP_SH_0_255(vec2);
|
||
|
|
||
|
val0 = vec2[0];
|
||
|
val1 = vec2[1];
|
||
|
val2 = vec2[2];
|
||
|
val3 = vec2[3];
|
||
|
|
||
|
dst[0] = val0;
|
||
|
dst[stride] = val1;
|
||
|
dst[2 * stride] = val2;
|
||
|
dst[3 * stride] = val3;
|
||
|
|
||
|
val0 = vec2[4];
|
||
|
val1 = vec2[5];
|
||
|
val2 = vec2[6];
|
||
|
val3 = vec2[7];
|
||
|
|
||
|
dst[4 * stride] = val0;
|
||
|
dst[5 * stride] = val1;
|
||
|
dst[6 * stride] = val2;
|
||
|
dst[7 * stride] = val3;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
static void hevc_intra_pred_vert_16x16_msa(const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
uint8_t *dst, int32_t stride,
|
||
|
int32_t flag)
|
||
|
{
|
||
|
int32_t col;
|
||
|
uint8_t *tmp_dst = dst;
|
||
|
uint32_t row;
|
||
|
v16u8 src;
|
||
|
v8i16 vec0, vec1, vec2, vec3;
|
||
|
|
||
|
src = LD_UB(src_top);
|
||
|
|
||
|
for (row = 16; row--;) {
|
||
|
ST_UB(src, tmp_dst);
|
||
|
tmp_dst += stride;
|
||
|
}
|
||
|
|
||
|
if (0 == flag) {
|
||
|
src = LD_UB(src_left);
|
||
|
|
||
|
vec0 = __msa_fill_h(src_left[-1]);
|
||
|
vec1 = __msa_fill_h(src_top[0]);
|
||
|
|
||
|
UNPCK_UB_SH(src, vec2, vec3);
|
||
|
SUB2(vec2, vec0, vec3, vec0, vec2, vec3);
|
||
|
|
||
|
vec2 >>= 1;
|
||
|
vec3 >>= 1;
|
||
|
|
||
|
ADD2(vec2, vec1, vec3, vec1, vec2, vec3);
|
||
|
CLIP_SH2_0_255(vec2, vec3);
|
||
|
|
||
|
src = (v16u8) __msa_pckev_b((v16i8) vec3, (v16i8) vec2);
|
||
|
|
||
|
for (col = 0; col < 16; col++) {
|
||
|
dst[stride * col] = src[col];
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
static void hevc_intra_pred_horiz_4x4_msa(const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
uint8_t *dst, int32_t stride,
|
||
|
int32_t flag)
|
||
|
{
|
||
|
uint32_t val0, val1, val2, val3;
|
||
|
v16i8 src0;
|
||
|
v8i16 src0_r, src_top_val, src_left_val;
|
||
|
v16i8 zero = { 0 };
|
||
|
|
||
|
val0 = src_left[0] * 0x01010101;
|
||
|
val1 = src_left[1] * 0x01010101;
|
||
|
val2 = src_left[2] * 0x01010101;
|
||
|
val3 = src_left[3] * 0x01010101;
|
||
|
SW4(val0, val1, val2, val3, dst, stride);
|
||
|
|
||
|
if (0 == flag) {
|
||
|
val0 = LW(src_top);
|
||
|
src0 = (v16i8) __msa_insert_w((v4i32) src0, 0, val0);
|
||
|
src_top_val = __msa_fill_h(src_top[-1]);
|
||
|
src_left_val = __msa_fill_h(src_left[0]);
|
||
|
|
||
|
src0_r = (v8i16) __msa_ilvr_b(zero, src0);
|
||
|
|
||
|
src0_r -= src_top_val;
|
||
|
src0_r >>= 1;
|
||
|
src0_r += src_left_val;
|
||
|
CLIP_SH_0_255(src0_r);
|
||
|
src0 = __msa_pckev_b((v16i8) src0_r, (v16i8) src0_r);
|
||
|
val0 = __msa_copy_s_w((v4i32) src0, 0);
|
||
|
SW(val0, dst);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
static void hevc_intra_pred_horiz_8x8_msa(const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
uint8_t *dst, int32_t stride,
|
||
|
int32_t flag)
|
||
|
{
|
||
|
uint64_t val0, val1, val2, val3;
|
||
|
v16i8 src0;
|
||
|
v8i16 src0_r, src_top_val, src_left_val;
|
||
|
v16i8 zero = { 0 };
|
||
|
|
||
|
val0 = src_left[0] * 0x0101010101010101;
|
||
|
val1 = src_left[1] * 0x0101010101010101;
|
||
|
val2 = src_left[2] * 0x0101010101010101;
|
||
|
val3 = src_left[3] * 0x0101010101010101;
|
||
|
SD4(val0, val1, val2, val3, dst, stride);
|
||
|
|
||
|
val0 = src_left[4] * 0x0101010101010101;
|
||
|
val1 = src_left[5] * 0x0101010101010101;
|
||
|
val2 = src_left[6] * 0x0101010101010101;
|
||
|
val3 = src_left[7] * 0x0101010101010101;
|
||
|
SD4(val0, val1, val2, val3, dst + 4 * stride, stride);
|
||
|
|
||
|
if (0 == flag) {
|
||
|
val0 = LD(src_top);
|
||
|
src0 = (v16i8) __msa_insert_d((v2i64) src0, 0, val0);
|
||
|
src_top_val = __msa_fill_h(src_top[-1]);
|
||
|
src_left_val = __msa_fill_h(src_left[0]);
|
||
|
|
||
|
src0_r = (v8i16) __msa_ilvr_b(zero, src0);
|
||
|
|
||
|
src0_r -= src_top_val;
|
||
|
src0_r >>= 1;
|
||
|
src0_r += src_left_val;
|
||
|
CLIP_SH_0_255(src0_r);
|
||
|
src0 = __msa_pckev_b((v16i8) src0_r, (v16i8) src0_r);
|
||
|
val0 = __msa_copy_s_d((v2i64) src0, 0);
|
||
|
SD(val0, dst);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
static void hevc_intra_pred_horiz_16x16_msa(const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
uint8_t *dst, int32_t stride,
|
||
|
int32_t flag)
|
||
|
{
|
||
|
uint8_t *tmp_dst = dst;
|
||
|
uint32_t row;
|
||
|
uint8_t inp0, inp1, inp2, inp3;
|
||
|
v16i8 src0, src1, src2, src3;
|
||
|
v8i16 src0_r, src0_l, src_left_val, src_top_val;
|
||
|
|
||
|
src_left_val = __msa_fill_h(src_left[0]);
|
||
|
|
||
|
for (row = 4; row--;) {
|
||
|
inp0 = src_left[0];
|
||
|
inp1 = src_left[1];
|
||
|
inp2 = src_left[2];
|
||
|
inp3 = src_left[3];
|
||
|
src_left += 4;
|
||
|
|
||
|
src0 = __msa_fill_b(inp0);
|
||
|
src1 = __msa_fill_b(inp1);
|
||
|
src2 = __msa_fill_b(inp2);
|
||
|
src3 = __msa_fill_b(inp3);
|
||
|
|
||
|
ST_SB4(src0, src1, src2, src3, tmp_dst, stride);
|
||
|
tmp_dst += (4 * stride);
|
||
|
}
|
||
|
|
||
|
if (0 == flag) {
|
||
|
src0 = LD_SB(src_top);
|
||
|
src_top_val = __msa_fill_h(src_top[-1]);
|
||
|
|
||
|
UNPCK_UB_SH(src0, src0_r, src0_l);
|
||
|
SUB2(src0_r, src_top_val, src0_l, src_top_val, src0_r, src0_l);
|
||
|
|
||
|
src0_r >>= 1;
|
||
|
src0_l >>= 1;
|
||
|
|
||
|
ADD2(src0_r, src_left_val, src0_l, src_left_val, src0_r, src0_l);
|
||
|
CLIP_SH2_0_255(src0_r, src0_l);
|
||
|
src0 = __msa_pckev_b((v16i8) src0_l, (v16i8) src0_r);
|
||
|
ST_SB(src0, dst);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
static void hevc_intra_pred_horiz_32x32_msa(const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
uint8_t *dst, int32_t stride)
|
||
|
{
|
||
|
uint32_t row;
|
||
|
uint8_t inp0, inp1, inp2, inp3;
|
||
|
v16i8 src0, src1, src2, src3;
|
||
|
|
||
|
for (row = 0; row < 8; row++) {
|
||
|
inp0 = src_left[row * 4];
|
||
|
inp1 = src_left[row * 4 + 1];
|
||
|
inp2 = src_left[row * 4 + 2];
|
||
|
inp3 = src_left[row * 4 + 3];
|
||
|
|
||
|
src0 = __msa_fill_b(inp0);
|
||
|
src1 = __msa_fill_b(inp1);
|
||
|
src2 = __msa_fill_b(inp2);
|
||
|
src3 = __msa_fill_b(inp3);
|
||
|
|
||
|
ST_SB2(src0, src0, dst, 16);
|
||
|
dst += stride;
|
||
|
ST_SB2(src1, src1, dst, 16);
|
||
|
dst += stride;
|
||
|
ST_SB2(src2, src2, dst, 16);
|
||
|
dst += stride;
|
||
|
ST_SB2(src3, src3, dst, 16);
|
||
|
dst += stride;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
static void hevc_intra_pred_dc_4x4_msa(const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
uint8_t *dst, int32_t stride,
|
||
|
int32_t flag)
|
||
|
{
|
||
|
uint8_t *tmp_dst = dst;
|
||
|
uint32_t addition = 0;
|
||
|
uint32_t val0, val1, val2;
|
||
|
v16i8 src = { 0 };
|
||
|
v16u8 store;
|
||
|
v16i8 zero = { 0 };
|
||
|
v8u16 sum, vec0, vec1;
|
||
|
|
||
|
val0 = LW(src_top);
|
||
|
val1 = LW(src_left);
|
||
|
INSERT_W2_SB(val0, val1, src);
|
||
|
sum = __msa_hadd_u_h((v16u8) src, (v16u8) src);
|
||
|
sum = (v8u16) __msa_hadd_u_w(sum, sum);
|
||
|
sum = (v8u16) __msa_hadd_u_d((v4u32) sum, (v4u32) sum);
|
||
|
sum = (v8u16) __msa_srari_w((v4i32) sum, 3);
|
||
|
addition = __msa_copy_u_w((v4i32) sum, 0);
|
||
|
store = (v16u8) __msa_fill_b(addition);
|
||
|
val0 = __msa_copy_u_w((v4i32) store, 0);
|
||
|
SW4(val0, val0, val0, val0, dst, stride)
|
||
|
|
||
|
if (0 == flag) {
|
||
|
ILVR_B2_UH(zero, store, zero, src, vec0, vec1);
|
||
|
|
||
|
vec1 += vec0;
|
||
|
vec0 += vec0;
|
||
|
vec1 += vec0;
|
||
|
|
||
|
vec1 = (v8u16) __msa_srari_h((v8i16) vec1, 2);
|
||
|
store = (v16u8) __msa_pckev_b((v16i8) vec1, (v16i8) vec1);
|
||
|
val1 = (src_left[0] + 2 * addition + src_top[0] + 2) >> 2;
|
||
|
store = (v16u8) __msa_insert_b((v16i8) store, 0, val1);
|
||
|
val0 = __msa_copy_u_w((v4i32) store, 0);
|
||
|
SW(val0, tmp_dst);
|
||
|
|
||
|
val0 = src_left[1];
|
||
|
val1 = src_left[2];
|
||
|
val2 = src_left[3];
|
||
|
|
||
|
addition *= 3;
|
||
|
|
||
|
ADD2(val0, addition, val1, addition, val0, val1);
|
||
|
val2 += addition;
|
||
|
|
||
|
val0 += 2;
|
||
|
val1 += 2;
|
||
|
val2 += 2;
|
||
|
val0 >>= 2;
|
||
|
val1 >>= 2;
|
||
|
val2 >>= 2;
|
||
|
|
||
|
tmp_dst[stride * 1] = val0;
|
||
|
tmp_dst[stride * 2] = val1;
|
||
|
tmp_dst[stride * 3] = val2;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
static void hevc_intra_pred_dc_8x8_msa(const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
uint8_t *dst, int32_t stride,
|
||
|
int32_t flag)
|
||
|
{
|
||
|
uint8_t *tmp_dst = dst;
|
||
|
uint32_t row, col, val;
|
||
|
uint32_t addition = 0;
|
||
|
uint64_t val0, val1;
|
||
|
v16u8 src = { 0 };
|
||
|
v16u8 store;
|
||
|
v8u16 sum, vec0, vec1;
|
||
|
v16i8 zero = { 0 };
|
||
|
|
||
|
val0 = LD(src_top);
|
||
|
val1 = LD(src_left);
|
||
|
INSERT_D2_UB(val0, val1, src);
|
||
|
sum = __msa_hadd_u_h((v16u8) src, (v16u8) src);
|
||
|
sum = (v8u16) __msa_hadd_u_w(sum, sum);
|
||
|
sum = (v8u16) __msa_hadd_u_d((v4u32) sum, (v4u32) sum);
|
||
|
sum = (v8u16) __msa_pckev_w((v4i32) sum, (v4i32) sum);
|
||
|
sum = (v8u16) __msa_hadd_u_d((v4u32) sum, (v4u32) sum);
|
||
|
sum = (v8u16) __msa_srari_w((v4i32) sum, 4);
|
||
|
addition = __msa_copy_u_w((v4i32) sum, 0);
|
||
|
store = (v16u8) __msa_fill_b(addition);
|
||
|
val0 = __msa_copy_u_d((v2i64) store, 0);
|
||
|
|
||
|
for (row = 8; row--;) {
|
||
|
SD(val0, dst);
|
||
|
dst += stride;
|
||
|
}
|
||
|
|
||
|
if (0 == flag) {
|
||
|
ILVR_B2_UH(zero, store, zero, src, vec0, vec1);
|
||
|
|
||
|
vec1 += vec0;
|
||
|
vec0 += vec0;
|
||
|
vec1 += vec0;
|
||
|
vec1 = (v8u16) __msa_srari_h((v8i16) vec1, 2);
|
||
|
store = (v16u8) __msa_pckev_b((v16i8) vec1, (v16i8) vec1);
|
||
|
val = (src_left[0] + 2 * addition + src_top[0] + 2) >> 2;
|
||
|
store = (v16u8) __msa_insert_b((v16i8) store, 0, val);
|
||
|
val0 = __msa_copy_u_d((v2i64) store, 0);
|
||
|
SD(val0, tmp_dst);
|
||
|
|
||
|
val0 = LD(src_left);
|
||
|
src = (v16u8) __msa_insert_d((v2i64) src, 0, val0);
|
||
|
vec1 = (v8u16) __msa_ilvr_b(zero, (v16i8) src);
|
||
|
vec0 = (v8u16) __msa_fill_h(addition);
|
||
|
vec0 *= 3;
|
||
|
vec1 += vec0;
|
||
|
vec1 = (v8u16) __msa_srari_h((v8i16) vec1, 2);
|
||
|
|
||
|
for (col = 1; col < 8; col++) {
|
||
|
tmp_dst[stride * col] = vec1[col];
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
static void hevc_intra_pred_dc_16x16_msa(const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
uint8_t *dst, int32_t stride,
|
||
|
int32_t flag)
|
||
|
{
|
||
|
uint8_t *tmp_dst = dst;
|
||
|
uint32_t row, col, val;
|
||
|
uint32_t addition = 0;
|
||
|
v16u8 src_above1, store, src_left1;
|
||
|
v8u16 sum, sum_above, sum_left;
|
||
|
v8u16 vec0, vec1, vec2;
|
||
|
v16i8 zero = { 0 };
|
||
|
|
||
|
src_above1 = LD_UB(src_top);
|
||
|
src_left1 = LD_UB(src_left);
|
||
|
|
||
|
HADD_UB2_UH(src_above1, src_left1, sum_above, sum_left);
|
||
|
sum = sum_above + sum_left;
|
||
|
sum = (v8u16) __msa_hadd_u_w(sum, sum);
|
||
|
sum = (v8u16) __msa_hadd_u_d((v4u32) sum, (v4u32) sum);
|
||
|
sum = (v8u16) __msa_pckev_w((v4i32) sum, (v4i32) sum);
|
||
|
sum = (v8u16) __msa_hadd_u_d((v4u32) sum, (v4u32) sum);
|
||
|
sum = (v8u16) __msa_srari_w((v4i32) sum, 5);
|
||
|
addition = __msa_copy_u_w((v4i32) sum, 0);
|
||
|
store = (v16u8) __msa_fill_b(addition);
|
||
|
|
||
|
for (row = 16; row--;) {
|
||
|
ST_UB(store, dst);
|
||
|
dst += stride;
|
||
|
}
|
||
|
|
||
|
if (0 == flag) {
|
||
|
vec0 = (v8u16) __msa_ilvr_b(zero, (v16i8) store);
|
||
|
ILVRL_B2_UH(zero, src_above1, vec1, vec2);
|
||
|
ADD2(vec1, vec0, vec2, vec0, vec1, vec2);
|
||
|
vec0 += vec0;
|
||
|
ADD2(vec1, vec0, vec2, vec0, vec1, vec2);
|
||
|
SRARI_H2_UH(vec1, vec2, 2);
|
||
|
store = (v16u8) __msa_pckev_b((v16i8) vec2, (v16i8) vec1);
|
||
|
val = (src_left[0] + 2 * addition + src_top[0] + 2) >> 2;
|
||
|
store = (v16u8) __msa_insert_b((v16i8) store, 0, val);
|
||
|
ST_UB(store, tmp_dst);
|
||
|
|
||
|
ILVRL_B2_UH(zero, src_left1, vec1, vec2);
|
||
|
vec0 = (v8u16) __msa_fill_h(addition);
|
||
|
vec0 *= 3;
|
||
|
ADD2(vec1, vec0, vec2, vec0, vec1, vec2);
|
||
|
SRARI_H2_UH(vec1, vec2, 2);
|
||
|
store = (v16u8) __msa_pckev_b((v16i8) vec2, (v16i8) vec1);
|
||
|
|
||
|
for (col = 1; col < 16; col++) {
|
||
|
tmp_dst[stride * col] = store[col];
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
static void hevc_intra_pred_dc_32x32_msa(const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
uint8_t *dst, int32_t stride)
|
||
|
{
|
||
|
uint32_t row;
|
||
|
v16u8 src_above1, src_above2, store, src_left1, src_left2;
|
||
|
v8u16 sum_above1, sum_above2;
|
||
|
v8u16 sum_left1, sum_left2;
|
||
|
v8u16 sum, sum_above, sum_left;
|
||
|
|
||
|
LD_UB2(src_top, 16, src_above1, src_above2);
|
||
|
LD_UB2(src_left, 16, src_left1, src_left2);
|
||
|
HADD_UB2_UH(src_above1, src_above2, sum_above1, sum_above2);
|
||
|
HADD_UB2_UH(src_left1, src_left2, sum_left1, sum_left2);
|
||
|
sum_above = sum_above1 + sum_above2;
|
||
|
sum_left = sum_left1 + sum_left2;
|
||
|
sum = sum_above + sum_left;
|
||
|
sum = (v8u16) __msa_hadd_u_w(sum, sum);
|
||
|
sum = (v8u16) __msa_hadd_u_d((v4u32) sum, (v4u32) sum);
|
||
|
sum = (v8u16) __msa_pckev_w((v4i32) sum, (v4i32) sum);
|
||
|
sum = (v8u16) __msa_hadd_u_d((v4u32) sum, (v4u32) sum);
|
||
|
sum = (v8u16) __msa_srari_w((v4i32) sum, 6);
|
||
|
store = (v16u8) __msa_splati_b((v16i8) sum, 0);
|
||
|
|
||
|
for (row = 16; row--;) {
|
||
|
ST_UB2(store, store, dst, 16);
|
||
|
dst += stride;
|
||
|
ST_UB2(store, store, dst, 16);
|
||
|
dst += stride;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
static void hevc_intra_pred_plane_4x4_msa(const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
uint8_t *dst, int32_t stride)
|
||
|
{
|
||
|
uint32_t src0, src1;
|
||
|
v16i8 src_vec0, src_vec1;
|
||
|
v8i16 src_vec0_r, src1_r, tmp0, tmp1, mul_val1;
|
||
|
v8i16 vec0, vec1, vec2, vec3, res0, res1, res2, res3;
|
||
|
v8i16 mul_val0 = { 3, 2, 1, 0, 1, 2, 3, 4 };
|
||
|
v16i8 zero = { 0 };
|
||
|
|
||
|
src0 = LW(src_top);
|
||
|
src1 = LW(src_left);
|
||
|
|
||
|
mul_val1 = (v8i16) __msa_pckod_d((v2i64) mul_val0, (v2i64) mul_val0);
|
||
|
|
||
|
src_vec0 = (v16i8) __msa_insert_w((v4i32) zero, 0, src0);
|
||
|
src_vec1 = (v16i8) __msa_insert_w((v4i32) zero, 0, src1);
|
||
|
|
||
|
ILVR_B2_SH(zero, src_vec0, zero, src_vec1, src_vec0_r, src1_r);
|
||
|
SPLATI_H4_SH(src1_r, 0, 1, 2, 3, vec0, vec1, vec2, vec3);
|
||
|
|
||
|
tmp0 = __msa_fill_h(src_top[4]);
|
||
|
tmp1 = __msa_fill_h(src_left[4]);
|
||
|
|
||
|
MUL4(mul_val0, vec0, mul_val0, vec1, mul_val0, vec2, mul_val0, vec3,
|
||
|
res0, res1, res2, res3);
|
||
|
|
||
|
res0 += mul_val1 * tmp0;
|
||
|
res1 += mul_val1 * tmp0;
|
||
|
res2 += mul_val1 * tmp0;
|
||
|
res3 += mul_val1 * tmp0;
|
||
|
|
||
|
res0 += 3 * src_vec0_r;
|
||
|
res1 += 2 * src_vec0_r;
|
||
|
res2 += src_vec0_r;
|
||
|
res0 += tmp1;
|
||
|
res1 += 2 * tmp1;
|
||
|
res2 += 3 * tmp1;
|
||
|
res3 += 4 * tmp1;
|
||
|
|
||
|
PCKEV_D2_SH(res1, res0, res3, res2, res0, res1);
|
||
|
SRARI_H2_SH(res0, res1, 3);
|
||
|
src_vec0 = __msa_pckev_b((v16i8) res1, (v16i8) res0);
|
||
|
ST_W4(src_vec0, 0, 1, 2, 3, dst, stride);
|
||
|
}
|
||
|
|
||
|
static void hevc_intra_pred_plane_8x8_msa(const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
uint8_t *dst, int32_t stride)
|
||
|
{
|
||
|
uint64_t src0, src1;
|
||
|
v16i8 src_vec0, src_vec1, src_vec2, src_vec3;
|
||
|
v8i16 src_vec0_r, src_vec1_r;
|
||
|
v8i16 vec0, vec1, vec2, vec3, vec4, vec5, vec6, vec7;
|
||
|
v8i16 res0, res1, res2, res3, res4, res5, res6, res7;
|
||
|
v8i16 tmp0, tmp1, tmp2;
|
||
|
v8i16 mul_val1 = { 1, 2, 3, 4, 5, 6, 7, 8 };
|
||
|
v8i16 mul_val0 = { 7, 6, 5, 4, 3, 2, 1, 0 };
|
||
|
v16i8 zero = { 0 };
|
||
|
|
||
|
src0 = LD(src_top);
|
||
|
src1 = LD(src_left);
|
||
|
|
||
|
src_vec0 = (v16i8) __msa_insert_d((v2i64) zero, 0, src0);
|
||
|
src_vec1 = (v16i8) __msa_insert_d((v2i64) zero, 0, src1);
|
||
|
|
||
|
ILVR_B2_SH(zero, src_vec0, zero, src_vec1, src_vec0_r, src_vec1_r);
|
||
|
SPLATI_H4_SH(src_vec1_r, 0, 1, 2, 3, vec0, vec1, vec2, vec3);
|
||
|
SPLATI_H4_SH(src_vec1_r, 4, 5, 6, 7, vec4, vec5, vec6, vec7);
|
||
|
|
||
|
tmp0 = __msa_fill_h(src_top[8]);
|
||
|
tmp1 = __msa_fill_h(src_left[8]);
|
||
|
|
||
|
MUL4(mul_val0, vec0, mul_val0, vec1, mul_val0, vec2, mul_val0, vec3,
|
||
|
res0, res1, res2, res3);
|
||
|
MUL4(mul_val0, vec4, mul_val0, vec5, mul_val0, vec6, mul_val0, vec7,
|
||
|
res4, res5, res6, res7);
|
||
|
|
||
|
tmp2 = mul_val1 * tmp0;
|
||
|
res0 += tmp2;
|
||
|
res1 += tmp2;
|
||
|
res2 += tmp2;
|
||
|
res3 += tmp2;
|
||
|
res4 += tmp2;
|
||
|
res5 += tmp2;
|
||
|
res6 += tmp2;
|
||
|
res7 += tmp2;
|
||
|
|
||
|
res0 += 7 * src_vec0_r;
|
||
|
res1 += 6 * src_vec0_r;
|
||
|
res2 += 5 * src_vec0_r;
|
||
|
res3 += 4 * src_vec0_r;
|
||
|
res4 += 3 * src_vec0_r;
|
||
|
res5 += 2 * src_vec0_r;
|
||
|
res6 += src_vec0_r;
|
||
|
|
||
|
res0 += tmp1;
|
||
|
res1 += 2 * tmp1;
|
||
|
res2 += 3 * tmp1;
|
||
|
res3 += 4 * tmp1;
|
||
|
res4 += 5 * tmp1;
|
||
|
res5 += 6 * tmp1;
|
||
|
res6 += 7 * tmp1;
|
||
|
res7 += 8 * tmp1;
|
||
|
|
||
|
SRARI_H4_SH(res0, res1, res2, res3, 4);
|
||
|
SRARI_H4_SH(res4, res5, res6, res7, 4);
|
||
|
PCKEV_B4_SB(res1, res0, res3, res2, res5, res4, res7, res6,
|
||
|
src_vec0, src_vec1, src_vec2, src_vec3);
|
||
|
|
||
|
ST_D8(src_vec0, src_vec1, src_vec2, src_vec3, 0, 1, 0, 1,
|
||
|
0, 1, 0, 1, dst, stride);
|
||
|
}
|
||
|
|
||
|
static void hevc_intra_pred_plane_16x16_msa(const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
uint8_t *dst, int32_t stride)
|
||
|
{
|
||
|
v16u8 src0, src1;
|
||
|
v8i16 src0_r, src1_r, src0_l, src1_l;
|
||
|
v8i16 vec0, vec1;
|
||
|
v8i16 res0, res1, tmp0, tmp1;
|
||
|
v8i16 mul_val2, mul_val3;
|
||
|
v8i16 mul_val1 = { 1, 2, 3, 4, 5, 6, 7, 8 };
|
||
|
v8i16 mul_val0 = { 15, 14, 13, 12, 11, 10, 9, 8 };
|
||
|
|
||
|
src0 = LD_UB(src_top);
|
||
|
src1 = LD_UB(src_left);
|
||
|
|
||
|
UNPCK_UB_SH(src0, src0_r, src0_l);
|
||
|
UNPCK_UB_SH(src1, src1_r, src1_l);
|
||
|
|
||
|
mul_val2 = mul_val0 - 8;
|
||
|
mul_val3 = mul_val1 + 8;
|
||
|
|
||
|
tmp0 = __msa_fill_h(src_top[16]);
|
||
|
tmp1 = __msa_fill_h(src_left[16]);
|
||
|
|
||
|
SPLATI_H2_SH(src1_r, 0, 1, vec0, vec1);
|
||
|
HEVC_PRED_PLANAR_16x2(src0_r, src0_l, tmp0, tmp1, vec0, vec1,
|
||
|
mul_val0, mul_val1, mul_val2, mul_val3,
|
||
|
res0, res1, 15, 1, 5);
|
||
|
ST_SH2(res0, res1, dst, stride);
|
||
|
dst += (2 * stride);
|
||
|
|
||
|
SPLATI_H2_SH(src1_r, 2, 3, vec0, vec1);
|
||
|
HEVC_PRED_PLANAR_16x2(src0_r, src0_l, tmp0, tmp1, vec0, vec1,
|
||
|
mul_val0, mul_val1, mul_val2, mul_val3,
|
||
|
res0, res1, 13, 3, 5);
|
||
|
ST_SH2(res0, res1, dst, stride);
|
||
|
dst += (2 * stride);
|
||
|
|
||
|
SPLATI_H2_SH(src1_r, 4, 5, vec0, vec1);
|
||
|
HEVC_PRED_PLANAR_16x2(src0_r, src0_l, tmp0, tmp1, vec0, vec1,
|
||
|
mul_val0, mul_val1, mul_val2, mul_val3,
|
||
|
res0, res1, 11, 5, 5);
|
||
|
ST_SH2(res0, res1, dst, stride);
|
||
|
dst += (2 * stride);
|
||
|
|
||
|
SPLATI_H2_SH(src1_r, 6, 7, vec0, vec1);
|
||
|
HEVC_PRED_PLANAR_16x2(src0_r, src0_l, tmp0, tmp1, vec0, vec1,
|
||
|
mul_val0, mul_val1, mul_val2, mul_val3,
|
||
|
res0, res1, 9, 7, 5);
|
||
|
ST_SH2(res0, res1, dst, stride);
|
||
|
dst += (2 * stride);
|
||
|
|
||
|
SPLATI_H2_SH(src1_l, 0, 1, vec0, vec1);
|
||
|
HEVC_PRED_PLANAR_16x2(src0_r, src0_l, tmp0, tmp1, vec0, vec1,
|
||
|
mul_val0, mul_val1, mul_val2, mul_val3,
|
||
|
res0, res1, 7, 9, 5);
|
||
|
ST_SH2(res0, res1, dst, stride);
|
||
|
dst += (2 * stride);
|
||
|
|
||
|
SPLATI_H2_SH(src1_l, 2, 3, vec0, vec1);
|
||
|
HEVC_PRED_PLANAR_16x2(src0_r, src0_l, tmp0, tmp1, vec0, vec1,
|
||
|
mul_val0, mul_val1, mul_val2, mul_val3,
|
||
|
res0, res1, 5, 11, 5);
|
||
|
ST_SH2(res0, res1, dst, stride);
|
||
|
dst += (2 * stride);
|
||
|
|
||
|
SPLATI_H2_SH(src1_l, 4, 5, vec0, vec1);
|
||
|
HEVC_PRED_PLANAR_16x2(src0_r, src0_l, tmp0, tmp1, vec0, vec1,
|
||
|
mul_val0, mul_val1, mul_val2, mul_val3,
|
||
|
res0, res1, 3, 13, 5);
|
||
|
ST_SH2(res0, res1, dst, stride);
|
||
|
dst += (2 * stride);
|
||
|
|
||
|
SPLATI_H2_SH(src1_l, 6, 7, vec0, vec1);
|
||
|
HEVC_PRED_PLANAR_16x2(src0_r, src0_l, tmp0, tmp1, vec0, vec1,
|
||
|
mul_val0, mul_val1, mul_val2, mul_val3,
|
||
|
res0, res1, 1, 15, 5);
|
||
|
ST_SH2(res0, res1, dst, stride);
|
||
|
}
|
||
|
|
||
|
static void process_intra_upper_16x16_msa(const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
uint8_t *dst, int32_t stride,
|
||
|
uint8_t offset)
|
||
|
{
|
||
|
v16i8 src0, src1;
|
||
|
v8i16 src0_r, src1_r, src0_l, src1_l;
|
||
|
v8i16 vec0, vec1, res0, res1;
|
||
|
v8i16 tmp0, tmp1;
|
||
|
v8i16 mul_val2, mul_val3;
|
||
|
v8i16 mul_val1 = { 1, 2, 3, 4, 5, 6, 7, 8 };
|
||
|
v8i16 mul_val0 = { 31, 30, 29, 28, 27, 26, 25, 24 };
|
||
|
|
||
|
tmp0 = __msa_fill_h(src_top[32 - offset]);
|
||
|
tmp1 = __msa_fill_h(src_left[32]);
|
||
|
|
||
|
src0 = LD_SB(src_top);
|
||
|
src1 = LD_SB(src_left);
|
||
|
|
||
|
UNPCK_UB_SH(src0, src0_r, src0_l);
|
||
|
UNPCK_UB_SH(src1, src1_r, src1_l);
|
||
|
|
||
|
mul_val1 += offset;
|
||
|
mul_val0 -= offset;
|
||
|
mul_val2 = mul_val0 - 8;
|
||
|
mul_val3 = mul_val1 + 8;
|
||
|
|
||
|
SPLATI_H2_SH(src1_r, 0, 1, vec0, vec1);
|
||
|
HEVC_PRED_PLANAR_16x2(src0_r, src0_l, tmp0, tmp1, vec0, vec1,
|
||
|
mul_val0, mul_val1, mul_val2, mul_val3,
|
||
|
res0, res1, 31, 1, 6);
|
||
|
ST_SH2(res0, res1, dst, stride);
|
||
|
dst += (2 * stride);
|
||
|
|
||
|
SPLATI_H2_SH(src1_r, 2, 3, vec0, vec1);
|
||
|
HEVC_PRED_PLANAR_16x2(src0_r, src0_l, tmp0, tmp1, vec0, vec1,
|
||
|
mul_val0, mul_val1, mul_val2, mul_val3,
|
||
|
res0, res1, 29, 3, 6);
|
||
|
ST_SH2(res0, res1, dst, stride);
|
||
|
dst += (2 * stride);
|
||
|
|
||
|
SPLATI_H2_SH(src1_r, 4, 5, vec0, vec1);
|
||
|
HEVC_PRED_PLANAR_16x2(src0_r, src0_l, tmp0, tmp1, vec0, vec1,
|
||
|
mul_val0, mul_val1, mul_val2, mul_val3,
|
||
|
res0, res1, 27, 5, 6);
|
||
|
ST_SH2(res0, res1, dst, stride);
|
||
|
dst += (2 * stride);
|
||
|
|
||
|
SPLATI_H2_SH(src1_r, 6, 7, vec0, vec1);
|
||
|
HEVC_PRED_PLANAR_16x2(src0_r, src0_l, tmp0, tmp1, vec0, vec1,
|
||
|
mul_val0, mul_val1, mul_val2, mul_val3,
|
||
|
res0, res1, 25, 7, 6);
|
||
|
ST_SH2(res0, res1, dst, stride);
|
||
|
dst += (2 * stride);
|
||
|
|
||
|
SPLATI_H2_SH(src1_l, 0, 1, vec0, vec1);
|
||
|
HEVC_PRED_PLANAR_16x2(src0_r, src0_l, tmp0, tmp1, vec0, vec1,
|
||
|
mul_val0, mul_val1, mul_val2, mul_val3,
|
||
|
res0, res1, 23, 9, 6);
|
||
|
ST_SH2(res0, res1, dst, stride);
|
||
|
dst += (2 * stride);
|
||
|
|
||
|
SPLATI_H2_SH(src1_l, 2, 3, vec0, vec1);
|
||
|
HEVC_PRED_PLANAR_16x2(src0_r, src0_l, tmp0, tmp1, vec0, vec1,
|
||
|
mul_val0, mul_val1, mul_val2, mul_val3,
|
||
|
res0, res1, 21, 11, 6);
|
||
|
ST_SH2(res0, res1, dst, stride);
|
||
|
dst += (2 * stride);
|
||
|
|
||
|
SPLATI_H2_SH(src1_l, 4, 5, vec0, vec1);
|
||
|
HEVC_PRED_PLANAR_16x2(src0_r, src0_l, tmp0, tmp1, vec0, vec1,
|
||
|
mul_val0, mul_val1, mul_val2, mul_val3,
|
||
|
res0, res1, 19, 13, 6);
|
||
|
ST_SH2(res0, res1, dst, stride);
|
||
|
dst += (2 * stride);
|
||
|
|
||
|
SPLATI_H2_SH(src1_l, 6, 7, vec0, vec1);
|
||
|
HEVC_PRED_PLANAR_16x2(src0_r, src0_l, tmp0, tmp1, vec0, vec1,
|
||
|
mul_val0, mul_val1, mul_val2, mul_val3,
|
||
|
res0, res1, 17, 15, 6);
|
||
|
ST_SH2(res0, res1, dst, stride);
|
||
|
}
|
||
|
|
||
|
static void process_intra_lower_16x16_msa(const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
uint8_t *dst, int32_t stride,
|
||
|
uint8_t offset)
|
||
|
{
|
||
|
v16i8 src0, src1;
|
||
|
v8i16 src0_r, src1_r, src0_l, src1_l;
|
||
|
v8i16 vec0, vec1, res0, res1, tmp0, tmp1;
|
||
|
v8i16 mul_val2, mul_val3;
|
||
|
v8i16 mul_val1 = { 1, 2, 3, 4, 5, 6, 7, 8 };
|
||
|
v8i16 mul_val0 = { 31, 30, 29, 28, 27, 26, 25, 24 };
|
||
|
|
||
|
tmp0 = __msa_fill_h(src_top[32 - offset]);
|
||
|
tmp1 = __msa_fill_h(src_left[16]);
|
||
|
|
||
|
src0 = LD_SB(src_top);
|
||
|
src1 = LD_SB(src_left);
|
||
|
|
||
|
UNPCK_UB_SH(src0, src0_r, src0_l);
|
||
|
UNPCK_UB_SH(src1, src1_r, src1_l);
|
||
|
|
||
|
mul_val1 += offset;
|
||
|
mul_val0 -= offset;
|
||
|
mul_val2 = mul_val0 - 8;
|
||
|
mul_val3 = mul_val1 + 8;
|
||
|
|
||
|
SPLATI_H2_SH(src1_r, 0, 1, vec0, vec1);
|
||
|
HEVC_PRED_PLANAR_16x2(src0_r, src0_l, tmp0, tmp1, vec0, vec1,
|
||
|
mul_val0, mul_val1, mul_val2, mul_val3,
|
||
|
res0, res1, 15, 17, 6);
|
||
|
ST_SH2(res0, res1, dst, stride);
|
||
|
dst += (2 * stride);
|
||
|
|
||
|
SPLATI_H2_SH(src1_r, 2, 3, vec0, vec1);
|
||
|
HEVC_PRED_PLANAR_16x2(src0_r, src0_l, tmp0, tmp1, vec0, vec1,
|
||
|
mul_val0, mul_val1, mul_val2, mul_val3,
|
||
|
res0, res1, 13, 19, 6);
|
||
|
ST_SH2(res0, res1, dst, stride);
|
||
|
dst += (2 * stride);
|
||
|
|
||
|
SPLATI_H2_SH(src1_r, 4, 5, vec0, vec1);
|
||
|
HEVC_PRED_PLANAR_16x2(src0_r, src0_l, tmp0, tmp1, vec0, vec1,
|
||
|
mul_val0, mul_val1, mul_val2, mul_val3,
|
||
|
res0, res1, 11, 21, 6);
|
||
|
ST_SH2(res0, res1, dst, stride);
|
||
|
dst += (2 * stride);
|
||
|
|
||
|
SPLATI_H2_SH(src1_r, 6, 7, vec0, vec1);
|
||
|
HEVC_PRED_PLANAR_16x2(src0_r, src0_l, tmp0, tmp1, vec0, vec1,
|
||
|
mul_val0, mul_val1, mul_val2, mul_val3,
|
||
|
res0, res1, 9, 23, 6);
|
||
|
ST_SH2(res0, res1, dst, stride);
|
||
|
dst += (2 * stride);
|
||
|
|
||
|
SPLATI_H2_SH(src1_l, 0, 1, vec0, vec1);
|
||
|
HEVC_PRED_PLANAR_16x2(src0_r, src0_l, tmp0, tmp1, vec0, vec1,
|
||
|
mul_val0, mul_val1, mul_val2, mul_val3,
|
||
|
res0, res1, 7, 25, 6);
|
||
|
ST_SH2(res0, res1, dst, stride);
|
||
|
dst += (2 * stride);
|
||
|
|
||
|
SPLATI_H2_SH(src1_l, 2, 3, vec0, vec1);
|
||
|
HEVC_PRED_PLANAR_16x2(src0_r, src0_l, tmp0, tmp1, vec0, vec1,
|
||
|
mul_val0, mul_val1, mul_val2, mul_val3,
|
||
|
res0, res1, 5, 27, 6);
|
||
|
ST_SH2(res0, res1, dst, stride);
|
||
|
dst += (2 * stride);
|
||
|
|
||
|
SPLATI_H2_SH(src1_l, 4, 5, vec0, vec1);
|
||
|
HEVC_PRED_PLANAR_16x2(src0_r, src0_l, tmp0, tmp1, vec0, vec1,
|
||
|
mul_val0, mul_val1, mul_val2, mul_val3,
|
||
|
res0, res1, 3, 29, 6);
|
||
|
ST_SH2(res0, res1, dst, stride);
|
||
|
dst += (2 * stride);
|
||
|
|
||
|
SPLATI_H2_SH(src1_l, 6, 7, vec0, vec1);
|
||
|
HEVC_PRED_PLANAR_16x2(src0_r, src0_l, tmp0, tmp1, vec0, vec1,
|
||
|
mul_val0, mul_val1, mul_val2, mul_val3,
|
||
|
res0, res1, 1, 31, 6);
|
||
|
ST_SH2(res0, res1, dst, stride);
|
||
|
}
|
||
|
|
||
|
static void hevc_intra_pred_plane_32x32_msa(const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
uint8_t *dst, int32_t stride)
|
||
|
{
|
||
|
process_intra_upper_16x16_msa(src_top, src_left, dst, stride, 0);
|
||
|
process_intra_upper_16x16_msa((src_top + 16), src_left,
|
||
|
(dst + 16), stride, 16);
|
||
|
dst += (16 * stride);
|
||
|
src_left += 16;
|
||
|
|
||
|
process_intra_lower_16x16_msa(src_top, src_left, dst, stride, 0);
|
||
|
process_intra_lower_16x16_msa((src_top + 16), src_left,
|
||
|
(dst + 16), stride, 16);
|
||
|
}
|
||
|
|
||
|
static void hevc_intra_pred_angular_upper_4width_msa(const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
uint8_t *dst,
|
||
|
int32_t stride,
|
||
|
int32_t mode)
|
||
|
{
|
||
|
int16_t inv_angle[] = { -256, -315, -390, -482, -630, -910, -1638, -4096 };
|
||
|
uint8_t ref_array[3 * 32 + 4];
|
||
|
uint8_t *ref_tmp = ref_array + 4;
|
||
|
const uint8_t *ref;
|
||
|
int32_t last;
|
||
|
int32_t h_cnt, idx0, fact_val0, idx1, fact_val1;
|
||
|
int32_t idx2, fact_val2, idx3, fact_val3;
|
||
|
int32_t angle, angle_loop;
|
||
|
int32_t inv_angle_val, offset;
|
||
|
uint64_t tmp0;
|
||
|
v16i8 top0, top1, top2, top3;
|
||
|
v16i8 dst_val0;
|
||
|
v16i8 zero = { 0 };
|
||
|
v8i16 diff0, diff1, diff2, diff3, diff4, diff5, diff6, diff7;
|
||
|
v8i16 fact0, fact1, fact2, fact3, fact4, fact5, fact6, fact7;
|
||
|
|
||
|
angle = intra_pred_angle_up[mode - 18];
|
||
|
inv_angle_val = inv_angle[mode - 18];
|
||
|
last = (angle) >> 3;
|
||
|
angle_loop = angle;
|
||
|
|
||
|
ref = src_top - 1;
|
||
|
if (angle < 0 && last < -1) {
|
||
|
inv_angle_val = inv_angle[mode - 18];
|
||
|
|
||
|
tmp0 = LD(ref);
|
||
|
SD(tmp0, ref_tmp);
|
||
|
|
||
|
for (h_cnt = last; h_cnt <= -1; h_cnt++) {
|
||
|
offset = -1 + ((h_cnt * inv_angle_val + 128) >> 8);
|
||
|
ref_tmp[h_cnt] = src_left[offset];
|
||
|
}
|
||
|
|
||
|
ref = ref_tmp;
|
||
|
}
|
||
|
|
||
|
idx0 = angle_loop >> 5;
|
||
|
fact_val0 = angle_loop & 31;
|
||
|
angle_loop += angle;
|
||
|
|
||
|
idx1 = angle_loop >> 5;
|
||
|
fact_val1 = angle_loop & 31;
|
||
|
angle_loop += angle;
|
||
|
|
||
|
idx2 = angle_loop >> 5;
|
||
|
fact_val2 = angle_loop & 31;
|
||
|
angle_loop += angle;
|
||
|
|
||
|
idx3 = angle_loop >> 5;
|
||
|
fact_val3 = angle_loop & 31;
|
||
|
|
||
|
top0 = LD_SB(ref + idx0 + 1);
|
||
|
top1 = LD_SB(ref + idx1 + 1);
|
||
|
top2 = LD_SB(ref + idx2 + 1);
|
||
|
top3 = LD_SB(ref + idx3 + 1);
|
||
|
|
||
|
fact0 = __msa_fill_h(fact_val0);
|
||
|
fact1 = __msa_fill_h(32 - fact_val0);
|
||
|
|
||
|
fact2 = __msa_fill_h(fact_val1);
|
||
|
fact3 = __msa_fill_h(32 - fact_val1);
|
||
|
|
||
|
fact4 = __msa_fill_h(fact_val2);
|
||
|
fact5 = __msa_fill_h(32 - fact_val2);
|
||
|
|
||
|
fact6 = __msa_fill_h(fact_val3);
|
||
|
fact7 = __msa_fill_h(32 - fact_val3);
|
||
|
|
||
|
ILVR_D2_SH(fact2, fact0, fact6, fact4, fact0, fact2);
|
||
|
ILVR_D2_SH(fact3, fact1, fact7, fact5, fact1, fact3);
|
||
|
ILVR_B4_SH(zero, top0, zero, top1, zero, top2, zero, top3,
|
||
|
diff0, diff2, diff4, diff6);
|
||
|
SLDI_B4_SH(zero, diff0, zero, diff2, zero, diff4, zero, diff6, 2,
|
||
|
diff1, diff3, diff5, diff7);
|
||
|
ILVR_D2_SH(diff2, diff0, diff6, diff4, diff0, diff2);
|
||
|
ILVR_D2_SH(diff3, diff1, diff7, diff5, diff1, diff3);
|
||
|
MUL2(diff1, fact0, diff3, fact2, diff1, diff3);
|
||
|
|
||
|
diff1 += diff0 * fact1;
|
||
|
diff3 += diff2 * fact3;
|
||
|
|
||
|
SRARI_H2_SH(diff1, diff3, 5);
|
||
|
dst_val0 = __msa_pckev_b((v16i8) diff3, (v16i8) diff1);
|
||
|
ST_W4(dst_val0, 0, 1, 2, 3, dst, stride);
|
||
|
}
|
||
|
|
||
|
static void hevc_intra_pred_angular_upper_8width_msa(const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
uint8_t *dst,
|
||
|
int32_t stride,
|
||
|
int32_t mode)
|
||
|
{
|
||
|
int16_t inv_angle[] = { -256, -315, -390, -482, -630, -910, -1638, -4096 };
|
||
|
uint8_t ref_array[3 * 32 + 4];
|
||
|
uint8_t *ref_tmp = ref_array + 8;
|
||
|
const uint8_t *ref;
|
||
|
const uint8_t *src_left_tmp = src_left - 1;
|
||
|
int32_t last, offset;
|
||
|
int32_t h_cnt, v_cnt, idx0, fact_val0, idx1, fact_val1;
|
||
|
int32_t idx2, fact_val2, idx3, fact_val3;
|
||
|
int32_t angle, angle_loop;
|
||
|
int32_t inv_angle_val, inv_angle_val_loop;
|
||
|
int32_t tmp0, tmp1, tmp2;
|
||
|
v16i8 top0, top1, top2, top3;
|
||
|
v16u8 dst_val0, dst_val1;
|
||
|
v8i16 fact0, fact1, fact2, fact3, fact4, fact5, fact6, fact7;
|
||
|
v8i16 diff0, diff1, diff2, diff3, diff4, diff5, diff6, diff7;
|
||
|
|
||
|
angle = intra_pred_angle_up[mode - 18];
|
||
|
inv_angle_val = inv_angle[mode - 18];
|
||
|
last = (angle) >> 2;
|
||
|
angle_loop = angle;
|
||
|
|
||
|
ref = src_top - 1;
|
||
|
if (last < -1) {
|
||
|
inv_angle_val_loop = inv_angle_val * last;
|
||
|
|
||
|
tmp0 = LW(ref);
|
||
|
tmp1 = LW(ref + 4);
|
||
|
tmp2 = LW(ref + 8);
|
||
|
SW(tmp0, ref_tmp);
|
||
|
SW(tmp1, ref_tmp + 4);
|
||
|
SW(tmp2, ref_tmp + 8);
|
||
|
|
||
|
for (h_cnt = last; h_cnt <= -1; h_cnt++) {
|
||
|
offset = (inv_angle_val_loop + 128) >> 8;
|
||
|
ref_tmp[h_cnt] = src_left_tmp[offset];
|
||
|
inv_angle_val_loop += inv_angle_val;
|
||
|
}
|
||
|
ref = ref_tmp;
|
||
|
}
|
||
|
|
||
|
for (v_cnt = 0; v_cnt < 2; v_cnt++) {
|
||
|
idx0 = (angle_loop) >> 5;
|
||
|
fact_val0 = (angle_loop) & 31;
|
||
|
angle_loop += angle;
|
||
|
|
||
|
idx1 = (angle_loop) >> 5;
|
||
|
fact_val1 = (angle_loop) & 31;
|
||
|
angle_loop += angle;
|
||
|
|
||
|
idx2 = (angle_loop) >> 5;
|
||
|
fact_val2 = (angle_loop) & 31;
|
||
|
angle_loop += angle;
|
||
|
|
||
|
idx3 = (angle_loop) >> 5;
|
||
|
fact_val3 = (angle_loop) & 31;
|
||
|
angle_loop += angle;
|
||
|
|
||
|
top0 = LD_SB(ref + idx0 + 1);
|
||
|
top1 = LD_SB(ref + idx1 + 1);
|
||
|
top2 = LD_SB(ref + idx2 + 1);
|
||
|
top3 = LD_SB(ref + idx3 + 1);
|
||
|
|
||
|
fact0 = __msa_fill_h(fact_val0);
|
||
|
fact1 = __msa_fill_h(32 - fact_val0);
|
||
|
fact2 = __msa_fill_h(fact_val1);
|
||
|
fact3 = __msa_fill_h(32 - fact_val1);
|
||
|
fact4 = __msa_fill_h(fact_val2);
|
||
|
fact5 = __msa_fill_h(32 - fact_val2);
|
||
|
fact6 = __msa_fill_h(fact_val3);
|
||
|
fact7 = __msa_fill_h(32 - fact_val3);
|
||
|
|
||
|
UNPCK_UB_SH(top0, diff0, diff1);
|
||
|
UNPCK_UB_SH(top1, diff2, diff3);
|
||
|
UNPCK_UB_SH(top2, diff4, diff5);
|
||
|
UNPCK_UB_SH(top3, diff6, diff7);
|
||
|
|
||
|
SLDI_B4_SH(diff1, diff0, diff3, diff2, diff5, diff4, diff7, diff6, 2,
|
||
|
diff1, diff3, diff5, diff7);
|
||
|
MUL4(diff1, fact0, diff3, fact2, diff5, fact4, diff7, fact6,
|
||
|
diff1, diff3, diff5, diff7);
|
||
|
|
||
|
diff1 += diff0 * fact1;
|
||
|
diff3 += diff2 * fact3;
|
||
|
diff5 += diff4 * fact5;
|
||
|
diff7 += diff6 * fact7;
|
||
|
|
||
|
SRARI_H4_SH(diff1, diff3, diff5, diff7, 5);
|
||
|
PCKEV_B2_UB(diff3, diff1, diff7, diff5, dst_val0, dst_val1);
|
||
|
ST_D4(dst_val0, dst_val1, 0, 1, 0, 1, dst, stride);
|
||
|
dst += (4 * stride);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
static void hevc_intra_pred_angular_upper_16width_msa(const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
uint8_t *dst,
|
||
|
int32_t stride,
|
||
|
int32_t mode)
|
||
|
{
|
||
|
int16_t inv_angle[] = { -256, -315, -390, -482, -630, -910, -1638, -4096 };
|
||
|
int32_t h_cnt, v_cnt, idx0, fact_val0, idx1, fact_val1;
|
||
|
int32_t idx2, fact_val2, idx3, fact_val3;
|
||
|
int32_t tmp0;
|
||
|
int32_t angle, angle_loop, offset;
|
||
|
int32_t inv_angle_val, inv_angle_val_loop;
|
||
|
uint8_t ref_array[3 * 32 + 4];
|
||
|
uint8_t *ref_tmp = ref_array + 16;
|
||
|
const uint8_t *ref;
|
||
|
const uint8_t *src_left_tmp = src_left - 1;
|
||
|
int32_t last;
|
||
|
v16u8 top0, top1, top2, top3, top4, top5, top6, top7;
|
||
|
v16i8 dst0, dst1, dst2, dst3;
|
||
|
v8i16 fact0, fact1, fact2, fact3, fact4, fact5, fact6, fact7;
|
||
|
v8i16 diff0, diff1, diff2, diff3, diff4, diff5, diff6, diff7;
|
||
|
v8i16 diff8, diff9, diff10, diff11, diff12, diff13, diff14, diff15;
|
||
|
|
||
|
angle = intra_pred_angle_up[mode - 18];
|
||
|
inv_angle_val = inv_angle[mode - 18];
|
||
|
last = angle >> 1;
|
||
|
angle_loop = angle;
|
||
|
|
||
|
ref = src_top - 1;
|
||
|
if (last < -1) {
|
||
|
inv_angle_val_loop = inv_angle_val * last;
|
||
|
|
||
|
top0 = LD_UB(ref);
|
||
|
tmp0 = LW(ref + 16);
|
||
|
ST_UB(top0, ref_tmp);
|
||
|
SW(tmp0, ref_tmp + 16);
|
||
|
|
||
|
for (h_cnt = last; h_cnt <= -1; h_cnt++) {
|
||
|
offset = (inv_angle_val_loop + 128) >> 8;
|
||
|
ref_tmp[h_cnt] = src_left_tmp[offset];
|
||
|
inv_angle_val_loop += inv_angle_val;
|
||
|
}
|
||
|
ref = ref_tmp;
|
||
|
}
|
||
|
|
||
|
for (v_cnt = 4; v_cnt--;) {
|
||
|
idx0 = (angle_loop) >> 5;
|
||
|
fact_val0 = (angle_loop) & 31;
|
||
|
angle_loop += angle;
|
||
|
|
||
|
idx1 = (angle_loop) >> 5;
|
||
|
fact_val1 = (angle_loop) & 31;
|
||
|
angle_loop += angle;
|
||
|
|
||
|
idx2 = (angle_loop) >> 5;
|
||
|
fact_val2 = (angle_loop) & 31;
|
||
|
angle_loop += angle;
|
||
|
|
||
|
idx3 = (angle_loop) >> 5;
|
||
|
fact_val3 = (angle_loop) & 31;
|
||
|
angle_loop += angle;
|
||
|
|
||
|
LD_UB2(ref + idx0 + 1, 16, top0, top1);
|
||
|
LD_UB2(ref + idx1 + 1, 16, top2, top3);
|
||
|
LD_UB2(ref + idx2 + 1, 16, top4, top5);
|
||
|
LD_UB2(ref + idx3 + 1, 16, top6, top7);
|
||
|
|
||
|
fact0 = __msa_fill_h(fact_val0);
|
||
|
fact1 = __msa_fill_h(32 - fact_val0);
|
||
|
fact2 = __msa_fill_h(fact_val1);
|
||
|
fact3 = __msa_fill_h(32 - fact_val1);
|
||
|
fact4 = __msa_fill_h(fact_val2);
|
||
|
fact5 = __msa_fill_h(32 - fact_val2);
|
||
|
fact6 = __msa_fill_h(fact_val3);
|
||
|
fact7 = __msa_fill_h(32 - fact_val3);
|
||
|
|
||
|
SLDI_B4_UB(top1, top0, top3, top2, top5, top4, top7, top6, 1,
|
||
|
top1, top3, top5, top7);
|
||
|
UNPCK_UB_SH(top0, diff0, diff1);
|
||
|
UNPCK_UB_SH(top1, diff2, diff3);
|
||
|
UNPCK_UB_SH(top2, diff4, diff5);
|
||
|
UNPCK_UB_SH(top3, diff6, diff7);
|
||
|
UNPCK_UB_SH(top4, diff8, diff9);
|
||
|
UNPCK_UB_SH(top5, diff10, diff11);
|
||
|
UNPCK_UB_SH(top6, diff12, diff13);
|
||
|
UNPCK_UB_SH(top7, diff14, diff15);
|
||
|
|
||
|
MUL4(diff2, fact0, diff3, fact0, diff6, fact2, diff7, fact2,
|
||
|
diff2, diff3, diff6, diff7);
|
||
|
MUL4(diff10, fact4, diff11, fact4, diff14, fact6, diff15, fact6,
|
||
|
diff10, diff11, diff14, diff15);
|
||
|
|
||
|
diff2 += diff0 * fact1;
|
||
|
diff3 += diff1 * fact1;
|
||
|
diff6 += diff4 * fact3;
|
||
|
diff7 += diff5 * fact3;
|
||
|
diff10 += diff8 * fact5;
|
||
|
diff11 += diff9 * fact5;
|
||
|
diff14 += diff12 * fact7;
|
||
|
diff15 += diff13 * fact7;
|
||
|
|
||
|
SRARI_H4_SH(diff2, diff3, diff6, diff7, 5);
|
||
|
SRARI_H4_SH(diff10, diff11, diff14, diff15, 5);
|
||
|
PCKEV_B4_SB(diff3, diff2, diff7, diff6, diff11, diff10, diff15, diff14,
|
||
|
dst0, dst1, dst2, dst3);
|
||
|
ST_SB4(dst0, dst1, dst2, dst3, dst, stride);
|
||
|
dst += (4 * stride);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
static void hevc_intra_pred_angular_upper_32width_msa(const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
uint8_t *dst,
|
||
|
int32_t stride,
|
||
|
int32_t mode)
|
||
|
{
|
||
|
int16_t inv_angle[] = { -256, -315, -390, -482, -630, -910, -1638, -4096 };
|
||
|
uint8_t ref_array[3 * 32 + 4];
|
||
|
uint8_t *ref_tmp;
|
||
|
const uint8_t *ref;
|
||
|
const uint8_t *src_left_tmp = src_left - 1;
|
||
|
int32_t h_cnt, v_cnt, idx0, fact_val0, idx1, fact_val1;
|
||
|
int32_t tmp0, tmp1, tmp2, tmp3;
|
||
|
int32_t angle, angle_loop;
|
||
|
int32_t inv_angle_val, inv_angle_val_loop;
|
||
|
int32_t last, offset;
|
||
|
v16u8 top0, top1, top2, top3, top4, top5, top6, top7;
|
||
|
v16i8 dst0, dst1, dst2, dst3;
|
||
|
v8i16 fact0, fact1, fact2, fact3;
|
||
|
v8i16 diff0, diff1, diff2, diff3, diff4, diff5, diff6, diff7;
|
||
|
v8i16 diff8, diff9, diff10, diff11, diff12, diff13, diff14, diff15;
|
||
|
|
||
|
ref_tmp = ref_array + 32;
|
||
|
|
||
|
angle = intra_pred_angle_up[mode - 18];
|
||
|
inv_angle_val = inv_angle[mode - 18];
|
||
|
last = angle;
|
||
|
angle_loop = angle;
|
||
|
|
||
|
ref = src_top - 1;
|
||
|
if (last < -1) {
|
||
|
inv_angle_val_loop = inv_angle_val * last;
|
||
|
LD_UB2(ref, 16, top0, top1);
|
||
|
tmp0 = ref[32];
|
||
|
tmp1 = ref[33];
|
||
|
tmp2 = ref[34];
|
||
|
tmp3 = ref[35];
|
||
|
|
||
|
ST_UB2(top0, top1, ref_tmp, 16);
|
||
|
ref_tmp[32] = tmp0;
|
||
|
ref_tmp[33] = tmp1;
|
||
|
ref_tmp[34] = tmp2;
|
||
|
ref_tmp[35] = tmp3;
|
||
|
|
||
|
for (h_cnt = last; h_cnt <= -1; h_cnt++) {
|
||
|
offset = (inv_angle_val_loop + 128) >> 8;
|
||
|
ref_tmp[h_cnt] = src_left_tmp[offset];
|
||
|
inv_angle_val_loop += inv_angle_val;
|
||
|
}
|
||
|
|
||
|
ref = ref_tmp;
|
||
|
}
|
||
|
|
||
|
for (v_cnt = 16; v_cnt--;) {
|
||
|
idx0 = (angle_loop) >> 5;
|
||
|
fact_val0 = (angle_loop) & 31;
|
||
|
angle_loop += angle;
|
||
|
|
||
|
idx1 = (angle_loop) >> 5;
|
||
|
fact_val1 = (angle_loop) & 31;
|
||
|
angle_loop += angle;
|
||
|
|
||
|
top0 = LD_UB(ref + idx0 + 1);
|
||
|
top4 = LD_UB(ref + idx1 + 1);
|
||
|
top1 = LD_UB(ref + idx0 + 17);
|
||
|
top5 = LD_UB(ref + idx1 + 17);
|
||
|
top3 = LD_UB(ref + idx0 + 33);
|
||
|
top7 = LD_UB(ref + idx1 + 33);
|
||
|
|
||
|
fact0 = __msa_fill_h(fact_val0);
|
||
|
fact1 = __msa_fill_h(32 - fact_val0);
|
||
|
fact2 = __msa_fill_h(fact_val1);
|
||
|
fact3 = __msa_fill_h(32 - fact_val1);
|
||
|
|
||
|
top2 = top1;
|
||
|
top6 = top5;
|
||
|
|
||
|
SLDI_B4_UB(top1, top0, top3, top2, top5, top4, top7, top6, 1,
|
||
|
top1, top3, top5, top7);
|
||
|
UNPCK_UB_SH(top0, diff0, diff1);
|
||
|
UNPCK_UB_SH(top1, diff2, diff3);
|
||
|
UNPCK_UB_SH(top2, diff4, diff5);
|
||
|
UNPCK_UB_SH(top3, diff6, diff7);
|
||
|
UNPCK_UB_SH(top4, diff8, diff9);
|
||
|
UNPCK_UB_SH(top5, diff10, diff11);
|
||
|
UNPCK_UB_SH(top6, diff12, diff13);
|
||
|
UNPCK_UB_SH(top7, diff14, diff15);
|
||
|
|
||
|
MUL4(diff2, fact0, diff3, fact0, diff6, fact0, diff7, fact0,
|
||
|
diff2, diff3, diff6, diff7);
|
||
|
MUL4(diff10, fact2, diff11, fact2, diff14, fact2, diff15, fact2,
|
||
|
diff10, diff11, diff14, diff15);
|
||
|
|
||
|
diff2 += diff0 * fact1;
|
||
|
diff3 += diff1 * fact1;
|
||
|
diff6 += diff4 * fact1;
|
||
|
diff7 += diff5 * fact1;
|
||
|
diff10 += diff8 * fact3;
|
||
|
diff11 += diff9 * fact3;
|
||
|
diff14 += diff12 * fact3;
|
||
|
diff15 += diff13 * fact3;
|
||
|
|
||
|
SRARI_H4_SH(diff2, diff3, diff6, diff7, 5);
|
||
|
SRARI_H4_SH(diff10, diff11, diff14, diff15, 5);
|
||
|
PCKEV_B4_SB(diff3, diff2, diff7, diff6, diff11, diff10, diff15, diff14,
|
||
|
dst0, dst1, dst2, dst3);
|
||
|
|
||
|
ST_SB2(dst0, dst1, dst, 16);
|
||
|
dst += stride;
|
||
|
ST_SB2(dst2, dst3, dst, 16);
|
||
|
dst += stride;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
static void hevc_intra_pred_angular_lower_4width_msa(const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
uint8_t *dst,
|
||
|
int32_t stride,
|
||
|
int32_t mode)
|
||
|
{
|
||
|
int16_t inv_angle[] = { -4096, -1638, -910, -630, -482, -390, -315 };
|
||
|
uint8_t ref_array[3 * 32 + 4];
|
||
|
uint8_t *ref_tmp = ref_array + 4;
|
||
|
const uint8_t *ref;
|
||
|
int32_t last, offset;
|
||
|
int32_t h_cnt, idx0, fact_val0, idx1, fact_val1;
|
||
|
int32_t idx2, fact_val2, idx3, fact_val3;
|
||
|
int32_t angle, angle_loop, inv_angle_val;
|
||
|
uint64_t tmp0;
|
||
|
v16i8 dst_val0, dst_val1;
|
||
|
v16u8 top0, top1, top2, top3;
|
||
|
v16u8 zero = { 0 };
|
||
|
v8i16 diff0, diff1, diff2, diff3, diff4, diff5, diff6, diff7;
|
||
|
v8i16 fact0, fact1, fact2, fact3, fact4, fact5, fact6, fact7;
|
||
|
|
||
|
angle = intra_pred_angle_low[mode - 2];
|
||
|
last = angle >> 3;
|
||
|
angle_loop = angle;
|
||
|
|
||
|
ref = src_left - 1;
|
||
|
if (last < -1) {
|
||
|
inv_angle_val = inv_angle[mode - 11];
|
||
|
|
||
|
tmp0 = LD(ref);
|
||
|
SD(tmp0, ref_tmp);
|
||
|
|
||
|
for (h_cnt = last; h_cnt <= -1; h_cnt++) {
|
||
|
offset = -1 + ((h_cnt * inv_angle_val + 128) >> 8);
|
||
|
ref_tmp[h_cnt] = src_top[offset];
|
||
|
}
|
||
|
|
||
|
ref = ref_tmp;
|
||
|
}
|
||
|
|
||
|
idx0 = angle_loop >> 5;
|
||
|
fact_val0 = angle_loop & 31;
|
||
|
angle_loop += angle;
|
||
|
|
||
|
idx1 = angle_loop >> 5;
|
||
|
fact_val1 = angle_loop & 31;
|
||
|
angle_loop += angle;
|
||
|
|
||
|
idx2 = angle_loop >> 5;
|
||
|
fact_val2 = angle_loop & 31;
|
||
|
angle_loop += angle;
|
||
|
|
||
|
idx3 = angle_loop >> 5;
|
||
|
fact_val3 = angle_loop & 31;
|
||
|
|
||
|
top0 = LD_UB(ref + idx0 + 1);
|
||
|
top1 = LD_UB(ref + idx1 + 1);
|
||
|
top2 = LD_UB(ref + idx2 + 1);
|
||
|
top3 = LD_UB(ref + idx3 + 1);
|
||
|
|
||
|
fact0 = __msa_fill_h(fact_val0);
|
||
|
fact1 = __msa_fill_h(32 - fact_val0);
|
||
|
fact2 = __msa_fill_h(fact_val1);
|
||
|
fact3 = __msa_fill_h(32 - fact_val1);
|
||
|
fact4 = __msa_fill_h(fact_val2);
|
||
|
fact5 = __msa_fill_h(32 - fact_val2);
|
||
|
fact6 = __msa_fill_h(fact_val3);
|
||
|
fact7 = __msa_fill_h(32 - fact_val3);
|
||
|
|
||
|
ILVR_D2_SH(fact2, fact0, fact6, fact4, fact0, fact2);
|
||
|
ILVR_D2_SH(fact3, fact1, fact7, fact5, fact1, fact3);
|
||
|
ILVR_B4_SH(zero, top0, zero, top1, zero, top2, zero, top3,
|
||
|
diff0, diff2, diff4, diff6);
|
||
|
SLDI_B4_SH(zero, diff0, zero, diff2, zero, diff4, zero, diff6, 2,
|
||
|
diff1, diff3, diff5, diff7);
|
||
|
ILVR_D2_SH(diff2, diff0, diff6, diff4, diff0, diff2);
|
||
|
ILVR_D2_SH(diff3, diff1, diff7, diff5, diff1, diff3);
|
||
|
MUL2(diff1, fact0, diff3, fact2, diff1, diff3);
|
||
|
|
||
|
diff1 += diff0 * fact1;
|
||
|
diff3 += diff2 * fact3;
|
||
|
|
||
|
SRARI_H2_SH(diff1, diff3, 5);
|
||
|
PCKEV_B2_SB(diff1, diff1, diff3, diff3, dst_val0, dst_val1);
|
||
|
|
||
|
diff0 = (v8i16) __msa_pckev_b(dst_val1, dst_val0);
|
||
|
diff1 = (v8i16) __msa_pckod_b(dst_val1, dst_val0);
|
||
|
|
||
|
diff2 = (v8i16) __msa_pckev_w((v4i32) diff1, (v4i32) diff0);
|
||
|
|
||
|
dst_val0 = __msa_pckev_b((v16i8) diff2, (v16i8) diff2);
|
||
|
dst_val1 = __msa_pckod_b((v16i8) diff2, (v16i8) diff2);
|
||
|
|
||
|
ST_W2(dst_val0, 0, 1, dst, stride);
|
||
|
ST_W2(dst_val1, 0, 1, dst + 2 * stride, stride);
|
||
|
}
|
||
|
|
||
|
static void hevc_intra_pred_angular_lower_8width_msa(const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
uint8_t *dst,
|
||
|
int32_t stride,
|
||
|
int32_t mode)
|
||
|
{
|
||
|
int16_t inv_angle[] = { -4096, -1638, -910, -630, -482, -390, -315 };
|
||
|
uint8_t ref_array[3 * 32 + 4];
|
||
|
uint8_t *ref_tmp = ref_array + 8;
|
||
|
const uint8_t *ref;
|
||
|
const uint8_t *src_top_tmp = src_top - 1;
|
||
|
uint8_t *dst_org;
|
||
|
int32_t last, offset, tmp0, tmp1, tmp2;
|
||
|
int32_t h_cnt, v_cnt, idx0, fact_val0, idx1, fact_val1;
|
||
|
int32_t idx2, fact_val2, idx3, fact_val3;
|
||
|
int32_t angle, angle_loop, inv_angle_val;
|
||
|
v16i8 top0, top1, top2, top3;
|
||
|
v16i8 dst_val0, dst_val1, dst_val2, dst_val3;
|
||
|
v8i16 diff0, diff1, diff2, diff3, diff4, diff5, diff6, diff7;
|
||
|
v8i16 fact0, fact1, fact2, fact3, fact4, fact5, fact6, fact7;
|
||
|
|
||
|
angle = intra_pred_angle_low[mode - 2];
|
||
|
last = (angle) >> 2;
|
||
|
angle_loop = angle;
|
||
|
|
||
|
ref = src_left - 1;
|
||
|
if (last < -1) {
|
||
|
inv_angle_val = inv_angle[mode - 11];
|
||
|
|
||
|
tmp0 = LW(ref);
|
||
|
tmp1 = LW(ref + 4);
|
||
|
tmp2 = LW(ref + 8);
|
||
|
SW(tmp0, ref_tmp);
|
||
|
SW(tmp1, ref_tmp + 4);
|
||
|
SW(tmp2, ref_tmp + 8);
|
||
|
|
||
|
for (h_cnt = last; h_cnt <= -1; h_cnt++) {
|
||
|
offset = (h_cnt * inv_angle_val + 128) >> 8;
|
||
|
ref_tmp[h_cnt] = src_top_tmp[offset];
|
||
|
}
|
||
|
|
||
|
ref = ref_tmp;
|
||
|
}
|
||
|
|
||
|
for (v_cnt = 0; v_cnt < 2; v_cnt++) {
|
||
|
dst_org = dst;
|
||
|
|
||
|
idx0 = angle_loop >> 5;
|
||
|
fact_val0 = angle_loop & 31;
|
||
|
angle_loop += angle;
|
||
|
|
||
|
idx1 = angle_loop >> 5;
|
||
|
fact_val1 = angle_loop & 31;
|
||
|
angle_loop += angle;
|
||
|
|
||
|
idx2 = angle_loop >> 5;
|
||
|
fact_val2 = angle_loop & 31;
|
||
|
angle_loop += angle;
|
||
|
|
||
|
idx3 = angle_loop >> 5;
|
||
|
fact_val3 = angle_loop & 31;
|
||
|
angle_loop += angle;
|
||
|
|
||
|
top0 = LD_SB(ref + idx0 + 1);
|
||
|
top1 = LD_SB(ref + idx1 + 1);
|
||
|
top2 = LD_SB(ref + idx2 + 1);
|
||
|
top3 = LD_SB(ref + idx3 + 1);
|
||
|
|
||
|
fact0 = __msa_fill_h(fact_val0);
|
||
|
fact1 = __msa_fill_h(32 - fact_val0);
|
||
|
fact2 = __msa_fill_h(fact_val1);
|
||
|
fact3 = __msa_fill_h(32 - fact_val1);
|
||
|
fact4 = __msa_fill_h(fact_val2);
|
||
|
fact5 = __msa_fill_h(32 - fact_val2);
|
||
|
fact6 = __msa_fill_h(fact_val3);
|
||
|
fact7 = __msa_fill_h(32 - fact_val3);
|
||
|
|
||
|
UNPCK_UB_SH(top0, diff0, diff1);
|
||
|
UNPCK_UB_SH(top1, diff2, diff3);
|
||
|
UNPCK_UB_SH(top2, diff4, diff5);
|
||
|
UNPCK_UB_SH(top3, diff6, diff7);
|
||
|
SLDI_B4_SH(diff1, diff0, diff3, diff2, diff5, diff4, diff7, diff6, 2,
|
||
|
diff1, diff3, diff5, diff7);
|
||
|
MUL4(diff1, fact0, diff3, fact2, diff5, fact4, diff7, fact6,
|
||
|
diff1, diff3, diff5, diff7);
|
||
|
|
||
|
diff1 += diff0 * fact1;
|
||
|
diff3 += diff2 * fact3;
|
||
|
diff5 += diff4 * fact5;
|
||
|
diff7 += diff6 * fact7;
|
||
|
|
||
|
SRARI_H4_SH(diff1, diff3, diff5, diff7, 5);
|
||
|
PCKEV_B4_SB(diff1, diff1, diff3, diff3, diff5, diff5, diff7, diff7,
|
||
|
dst_val0, dst_val1, dst_val2, dst_val3);
|
||
|
ILVR_B2_SH(dst_val1, dst_val0, dst_val3, dst_val2, diff0, diff1);
|
||
|
ILVRL_H2_SH(diff1, diff0, diff3, diff4);
|
||
|
ST_W8(diff3, diff4, 0, 1, 2, 3, 0, 1, 2, 3, dst_org, stride);
|
||
|
dst += 4;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
static void hevc_intra_pred_angular_lower_16width_msa(const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
uint8_t *dst,
|
||
|
int32_t stride,
|
||
|
int32_t mode)
|
||
|
{
|
||
|
int16_t inv_angle[] = { -4096, -1638, -910, -630, -482, -390, -315 };
|
||
|
int32_t h_cnt, v_cnt, idx0, fact_val0, idx1, fact_val1;
|
||
|
int32_t idx2, fact_val2, idx3, fact_val3, tmp0;
|
||
|
v16i8 top0, top1, dst_val0, top2, top3, dst_val1;
|
||
|
v16i8 top4, top5, dst_val2, top6, top7, dst_val3;
|
||
|
v8i16 fact0, fact1, fact2, fact3, fact4, fact5, fact6, fact7;
|
||
|
v8i16 diff0, diff1, diff2, diff3, diff4, diff5, diff6, diff7;
|
||
|
v8i16 diff8, diff9, diff10, diff11, diff12, diff13, diff14, diff15;
|
||
|
int32_t angle, angle_loop, inv_angle_val, offset;
|
||
|
uint8_t ref_array[3 * 32 + 4];
|
||
|
uint8_t *ref_tmp = ref_array + 16;
|
||
|
const uint8_t *ref, *src_top_tmp = src_top - 1;
|
||
|
uint8_t *dst_org;
|
||
|
int32_t last;
|
||
|
|
||
|
angle = intra_pred_angle_low[mode - 2];
|
||
|
last = (angle) >> 1;
|
||
|
angle_loop = angle;
|
||
|
|
||
|
ref = src_left - 1;
|
||
|
if (last < -1) {
|
||
|
inv_angle_val = inv_angle[mode - 11];
|
||
|
|
||
|
top0 = LD_SB(ref);
|
||
|
tmp0 = LW(ref + 16);
|
||
|
ST_SB(top0, ref_tmp);
|
||
|
SW(tmp0, ref_tmp + 16);
|
||
|
|
||
|
for (h_cnt = last; h_cnt <= -1; h_cnt++) {
|
||
|
offset = (h_cnt * inv_angle_val + 128) >> 8;
|
||
|
ref_tmp[h_cnt] = src_top_tmp[offset];
|
||
|
}
|
||
|
|
||
|
ref = ref_tmp;
|
||
|
}
|
||
|
|
||
|
for (v_cnt = 0; v_cnt < 4; v_cnt++) {
|
||
|
dst_org = dst;
|
||
|
|
||
|
idx0 = angle_loop >> 5;
|
||
|
fact_val0 = angle_loop & 31;
|
||
|
angle_loop += angle;
|
||
|
|
||
|
idx1 = angle_loop >> 5;
|
||
|
fact_val1 = angle_loop & 31;
|
||
|
angle_loop += angle;
|
||
|
|
||
|
idx2 = angle_loop >> 5;
|
||
|
fact_val2 = angle_loop & 31;
|
||
|
angle_loop += angle;
|
||
|
|
||
|
idx3 = angle_loop >> 5;
|
||
|
fact_val3 = angle_loop & 31;
|
||
|
angle_loop += angle;
|
||
|
|
||
|
LD_SB2(ref + idx0 + 1, 16, top0, top1);
|
||
|
LD_SB2(ref + idx1 + 1, 16, top2, top3);
|
||
|
LD_SB2(ref + idx2 + 1, 16, top4, top5);
|
||
|
LD_SB2(ref + idx3 + 1, 16, top6, top7);
|
||
|
|
||
|
fact0 = __msa_fill_h(fact_val0);
|
||
|
fact1 = __msa_fill_h(32 - fact_val0);
|
||
|
fact2 = __msa_fill_h(fact_val1);
|
||
|
fact3 = __msa_fill_h(32 - fact_val1);
|
||
|
fact4 = __msa_fill_h(fact_val2);
|
||
|
fact5 = __msa_fill_h(32 - fact_val2);
|
||
|
fact6 = __msa_fill_h(fact_val3);
|
||
|
fact7 = __msa_fill_h(32 - fact_val3);
|
||
|
|
||
|
SLDI_B4_SB(top1, top0, top3, top2, top5, top4, top7, top6, 1,
|
||
|
top1, top3, top5, top7);
|
||
|
|
||
|
UNPCK_UB_SH(top0, diff0, diff1);
|
||
|
UNPCK_UB_SH(top1, diff2, diff3);
|
||
|
UNPCK_UB_SH(top2, diff4, diff5);
|
||
|
UNPCK_UB_SH(top3, diff6, diff7);
|
||
|
UNPCK_UB_SH(top4, diff8, diff9);
|
||
|
UNPCK_UB_SH(top5, diff10, diff11);
|
||
|
UNPCK_UB_SH(top6, diff12, diff13);
|
||
|
UNPCK_UB_SH(top7, diff14, diff15);
|
||
|
|
||
|
MUL4(diff2, fact0, diff3, fact0, diff6, fact2, diff7, fact2,
|
||
|
diff2, diff3, diff6, diff7);
|
||
|
MUL4(diff10, fact4, diff11, fact4, diff14, fact6, diff15, fact6,
|
||
|
diff10, diff11, diff14, diff15);
|
||
|
|
||
|
diff2 += diff0 * fact1;
|
||
|
diff3 += diff1 * fact1;
|
||
|
diff6 += diff4 * fact3;
|
||
|
diff7 += diff5 * fact3;
|
||
|
diff10 += diff8 * fact5;
|
||
|
diff11 += diff9 * fact5;
|
||
|
diff14 += diff12 * fact7;
|
||
|
diff15 += diff13 * fact7;
|
||
|
|
||
|
SRARI_H4_SH(diff2, diff3, diff6, diff7, 5);
|
||
|
SRARI_H4_SH(diff10, diff11, diff14, diff15, 5);
|
||
|
PCKEV_B4_SB(diff3, diff2, diff7, diff6, diff11, diff10, diff15, diff14,
|
||
|
dst_val0, dst_val1, dst_val2, dst_val3);
|
||
|
ILVR_B2_SH(dst_val1, dst_val0, dst_val3, dst_val2, diff0, diff1);
|
||
|
ILVL_B2_SH(dst_val1, dst_val0, dst_val3, dst_val2, diff2, diff3);
|
||
|
ILVRL_H2_SH(diff1, diff0, diff4, diff5);
|
||
|
ILVRL_H2_SH(diff3, diff2, diff6, diff7);
|
||
|
ST_W8(diff4, diff5, 0, 1, 2, 3, 0, 1, 2, 3, dst_org, stride);
|
||
|
dst_org += (8 * stride);
|
||
|
ST_W8(diff6, diff7, 0, 1, 2, 3, 0, 1, 2, 3, dst_org, stride);
|
||
|
dst += 4;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
static void hevc_intra_pred_angular_lower_32width_msa(const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
uint8_t *dst,
|
||
|
int32_t stride,
|
||
|
int32_t mode)
|
||
|
{
|
||
|
int16_t inv_angle[] = { -4096, -1638, -910, -630, -482, -390, -315 };
|
||
|
int32_t h_cnt, v_cnt, idx0, fact_val0, idx1, fact_val1, tmp0;
|
||
|
v16i8 top0, top1, dst_val0, top2, top3, dst_val1;
|
||
|
v16i8 top4, top5, dst_val2, top6, top7, dst_val3;
|
||
|
v8i16 fact0, fact1, fact2, fact3;
|
||
|
v8i16 diff0, diff1, diff2, diff3, diff4, diff5, diff6, diff7;
|
||
|
v8i16 diff8, diff9, diff10, diff11, diff12, diff13, diff14, diff15;
|
||
|
int32_t angle, angle_loop, inv_angle_val, offset;
|
||
|
uint8_t ref_array[3 * 32 + 4];
|
||
|
uint8_t *ref_tmp = ref_array + 32;
|
||
|
const uint8_t *ref, *src_top_tmp = src_top - 1;
|
||
|
uint8_t *dst_org;
|
||
|
int32_t last;
|
||
|
|
||
|
angle = intra_pred_angle_low[mode - 2];
|
||
|
last = angle;
|
||
|
angle_loop = angle;
|
||
|
|
||
|
ref = src_left - 1;
|
||
|
if (last < -1) {
|
||
|
inv_angle_val = inv_angle[mode - 11];
|
||
|
|
||
|
LD_SB2(ref, 16, top0, top1);
|
||
|
tmp0 = LW(ref + 32);
|
||
|
ST_SB2(top0, top1, ref_tmp, 16);
|
||
|
SW(tmp0, ref_tmp + 32);
|
||
|
|
||
|
for (h_cnt = last; h_cnt <= -1; h_cnt++) {
|
||
|
offset = (h_cnt * inv_angle_val + 128) >> 8;
|
||
|
ref_tmp[h_cnt] = src_top_tmp[offset];
|
||
|
}
|
||
|
|
||
|
ref = ref_tmp;
|
||
|
}
|
||
|
|
||
|
for (v_cnt = 0; v_cnt < 16; v_cnt++) {
|
||
|
dst_org = dst;
|
||
|
idx0 = angle_loop >> 5;
|
||
|
fact_val0 = angle_loop & 31;
|
||
|
angle_loop += angle;
|
||
|
|
||
|
idx1 = angle_loop >> 5;
|
||
|
fact_val1 = angle_loop & 31;
|
||
|
angle_loop += angle;
|
||
|
|
||
|
top0 = LD_SB(ref + idx0 + 1);
|
||
|
top4 = LD_SB(ref + idx1 + 1);
|
||
|
top1 = LD_SB(ref + idx0 + 17);
|
||
|
top5 = LD_SB(ref + idx1 + 17);
|
||
|
top3 = LD_SB(ref + idx0 + 33);
|
||
|
top7 = LD_SB(ref + idx1 + 33);
|
||
|
|
||
|
fact0 = __msa_fill_h(fact_val0);
|
||
|
fact1 = __msa_fill_h(32 - fact_val0);
|
||
|
fact2 = __msa_fill_h(fact_val1);
|
||
|
fact3 = __msa_fill_h(32 - fact_val1);
|
||
|
|
||
|
top2 = top1;
|
||
|
top6 = top5;
|
||
|
|
||
|
SLDI_B4_SB(top1, top0, top3, top2, top5, top4, top7, top6, 1,
|
||
|
top1, top3, top5, top7);
|
||
|
|
||
|
UNPCK_UB_SH(top0, diff0, diff1);
|
||
|
UNPCK_UB_SH(top1, diff2, diff3);
|
||
|
UNPCK_UB_SH(top2, diff4, diff5);
|
||
|
UNPCK_UB_SH(top3, diff6, diff7);
|
||
|
UNPCK_UB_SH(top4, diff8, diff9);
|
||
|
UNPCK_UB_SH(top5, diff10, diff11);
|
||
|
UNPCK_UB_SH(top6, diff12, diff13);
|
||
|
UNPCK_UB_SH(top7, diff14, diff15);
|
||
|
|
||
|
MUL4(diff2, fact0, diff3, fact0, diff6, fact0, diff7, fact0,
|
||
|
diff2, diff3, diff6, diff7);
|
||
|
MUL4(diff10, fact2, diff11, fact2, diff14, fact2, diff15, fact2,
|
||
|
diff10, diff11, diff14, diff15);
|
||
|
|
||
|
diff2 += diff0 * fact1;
|
||
|
diff3 += diff1 * fact1;
|
||
|
diff6 += diff4 * fact1;
|
||
|
diff7 += diff5 * fact1;
|
||
|
diff10 += diff8 * fact3;
|
||
|
diff11 += diff9 * fact3;
|
||
|
diff14 += diff12 * fact3;
|
||
|
diff15 += diff13 * fact3;
|
||
|
|
||
|
SRARI_H4_SH(diff2, diff3, diff6, diff7, 5);
|
||
|
SRARI_H4_SH(diff10, diff11, diff14, diff15, 5);
|
||
|
PCKEV_B4_SB(diff3, diff2, diff7, diff6, diff11, diff10, diff15, diff14,
|
||
|
dst_val0, dst_val1, dst_val2, dst_val3);
|
||
|
ILVRL_B2_SH(dst_val2, dst_val0, diff0, diff1);
|
||
|
ILVRL_B2_SH(dst_val3, dst_val1, diff2, diff3);
|
||
|
|
||
|
ST_H8(diff0, 0, 1, 2, 3, 4, 5, 6, 7, dst_org, stride)
|
||
|
dst_org += (8 * stride);
|
||
|
ST_H8(diff1, 0, 1, 2, 3, 4, 5, 6, 7, dst_org, stride)
|
||
|
dst_org += (8 * stride);
|
||
|
ST_H8(diff2, 0, 1, 2, 3, 4, 5, 6, 7, dst_org, stride)
|
||
|
dst_org += (8 * stride);
|
||
|
ST_H8(diff3, 0, 1, 2, 3, 4, 5, 6, 7, dst_org, stride)
|
||
|
dst_org += (8 * stride);
|
||
|
|
||
|
dst += 2;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
static void intra_predict_vert_32x32_msa(const uint8_t *src, uint8_t *dst,
|
||
|
int32_t dst_stride)
|
||
|
{
|
||
|
uint32_t row;
|
||
|
v16u8 src1, src2;
|
||
|
|
||
|
src1 = LD_UB(src);
|
||
|
src2 = LD_UB(src + 16);
|
||
|
|
||
|
for (row = 32; row--;) {
|
||
|
ST_UB2(src1, src2, dst, 16);
|
||
|
dst += dst_stride;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
void ff_hevc_intra_pred_planar_0_msa(uint8_t *dst,
|
||
|
const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
ptrdiff_t stride)
|
||
|
{
|
||
|
hevc_intra_pred_plane_4x4_msa(src_top, src_left, dst, stride);
|
||
|
}
|
||
|
|
||
|
void ff_hevc_intra_pred_planar_1_msa(uint8_t *dst,
|
||
|
const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
ptrdiff_t stride)
|
||
|
{
|
||
|
hevc_intra_pred_plane_8x8_msa(src_top, src_left, dst, stride);
|
||
|
}
|
||
|
|
||
|
void ff_hevc_intra_pred_planar_2_msa(uint8_t *dst,
|
||
|
const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
ptrdiff_t stride)
|
||
|
{
|
||
|
hevc_intra_pred_plane_16x16_msa(src_top, src_left, dst, stride);
|
||
|
}
|
||
|
|
||
|
void ff_hevc_intra_pred_planar_3_msa(uint8_t *dst,
|
||
|
const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
ptrdiff_t stride)
|
||
|
{
|
||
|
hevc_intra_pred_plane_32x32_msa(src_top, src_left, dst, stride);
|
||
|
}
|
||
|
|
||
|
void ff_hevc_intra_pred_dc_msa(uint8_t *dst, const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
ptrdiff_t stride, int log2, int c_idx)
|
||
|
{
|
||
|
switch (log2) {
|
||
|
case 2:
|
||
|
hevc_intra_pred_dc_4x4_msa(src_top, src_left, dst, stride, c_idx);
|
||
|
break;
|
||
|
|
||
|
case 3:
|
||
|
hevc_intra_pred_dc_8x8_msa(src_top, src_left, dst, stride, c_idx);
|
||
|
break;
|
||
|
|
||
|
case 4:
|
||
|
hevc_intra_pred_dc_16x16_msa(src_top, src_left, dst, stride, c_idx);
|
||
|
break;
|
||
|
|
||
|
case 5:
|
||
|
hevc_intra_pred_dc_32x32_msa(src_top, src_left, dst, stride);
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
void ff_pred_intra_pred_angular_0_msa(uint8_t *dst,
|
||
|
const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
ptrdiff_t stride, int c_idx, int mode)
|
||
|
{
|
||
|
if (mode == 10) {
|
||
|
hevc_intra_pred_horiz_4x4_msa(src_top, src_left, dst, stride, c_idx);
|
||
|
} else if (mode == 26) {
|
||
|
hevc_intra_pred_vert_4x4_msa(src_top, src_left, dst, stride, c_idx);
|
||
|
} else if (mode >= 18) {
|
||
|
hevc_intra_pred_angular_upper_4width_msa(src_top, src_left,
|
||
|
dst, stride, mode);
|
||
|
} else {
|
||
|
hevc_intra_pred_angular_lower_4width_msa(src_top, src_left,
|
||
|
dst, stride, mode);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
void ff_pred_intra_pred_angular_1_msa(uint8_t *dst,
|
||
|
const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
ptrdiff_t stride, int c_idx, int mode)
|
||
|
{
|
||
|
if (mode == 10) {
|
||
|
hevc_intra_pred_horiz_8x8_msa(src_top, src_left, dst, stride, c_idx);
|
||
|
} else if (mode == 26) {
|
||
|
hevc_intra_pred_vert_8x8_msa(src_top, src_left, dst, stride, c_idx);
|
||
|
} else if (mode >= 18) {
|
||
|
hevc_intra_pred_angular_upper_8width_msa(src_top, src_left,
|
||
|
dst, stride, mode);
|
||
|
} else {
|
||
|
hevc_intra_pred_angular_lower_8width_msa(src_top, src_left,
|
||
|
dst, stride, mode);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
void ff_pred_intra_pred_angular_2_msa(uint8_t *dst,
|
||
|
const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
ptrdiff_t stride, int c_idx, int mode)
|
||
|
{
|
||
|
if (mode == 10) {
|
||
|
hevc_intra_pred_horiz_16x16_msa(src_top, src_left, dst, stride, c_idx);
|
||
|
} else if (mode == 26) {
|
||
|
hevc_intra_pred_vert_16x16_msa(src_top, src_left, dst, stride, c_idx);
|
||
|
} else if (mode >= 18) {
|
||
|
hevc_intra_pred_angular_upper_16width_msa(src_top, src_left,
|
||
|
dst, stride, mode);
|
||
|
} else {
|
||
|
hevc_intra_pred_angular_lower_16width_msa(src_top, src_left,
|
||
|
dst, stride, mode);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
void ff_pred_intra_pred_angular_3_msa(uint8_t *dst,
|
||
|
const uint8_t *src_top,
|
||
|
const uint8_t *src_left,
|
||
|
ptrdiff_t stride, int c_idx, int mode)
|
||
|
{
|
||
|
if (mode == 10) {
|
||
|
hevc_intra_pred_horiz_32x32_msa(src_top, src_left, dst, stride);
|
||
|
} else if (mode == 26) {
|
||
|
intra_predict_vert_32x32_msa(src_top, dst, stride);
|
||
|
} else if (mode >= 18) {
|
||
|
hevc_intra_pred_angular_upper_32width_msa(src_top, src_left,
|
||
|
dst, stride, mode);
|
||
|
} else {
|
||
|
hevc_intra_pred_angular_lower_32width_msa(src_top, src_left,
|
||
|
dst, stride, mode);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
void ff_intra_pred_8_16x16_msa(HEVCContext *s, int x0, int y0, int c_idx)
|
||
|
{
|
||
|
v16u8 vec0;
|
||
|
HEVCLocalContext *lc = s->HEVClc;
|
||
|
int i;
|
||
|
int hshift = s->ps.sps->hshift[c_idx];
|
||
|
int vshift = s->ps.sps->vshift[c_idx];
|
||
|
int size_in_luma_h = 16 << hshift;
|
||
|
int size_in_tbs_h = size_in_luma_h >> s->ps.sps->log2_min_tb_size;
|
||
|
int size_in_luma_v = 16 << vshift;
|
||
|
int size_in_tbs_v = size_in_luma_v >> s->ps.sps->log2_min_tb_size;
|
||
|
int x = x0 >> hshift;
|
||
|
int y = y0 >> vshift;
|
||
|
int x_tb = (x0 >> s->ps.sps->log2_min_tb_size) & s->ps.sps->tb_mask;
|
||
|
int y_tb = (y0 >> s->ps.sps->log2_min_tb_size) & s->ps.sps->tb_mask;
|
||
|
|
||
|
int cur_tb_addr =
|
||
|
s->ps.pps->min_tb_addr_zs[(y_tb) * (s->ps.sps->tb_mask + 2) + (x_tb)];
|
||
|
|
||
|
ptrdiff_t stride = s->frame->linesize[c_idx] / sizeof(uint8_t);
|
||
|
uint8_t *src = (uint8_t *) s->frame->data[c_idx] + x + y * stride;
|
||
|
|
||
|
int min_pu_width = s->ps.sps->min_pu_width;
|
||
|
|
||
|
enum IntraPredMode mode = c_idx ? lc->tu.intra_pred_mode_c :
|
||
|
lc->tu.intra_pred_mode;
|
||
|
uint32_t a;
|
||
|
uint8_t left_array[2 * 32 + 1];
|
||
|
uint8_t filtered_left_array[2 * 32 + 1];
|
||
|
uint8_t top_array[2 * 32 + 1];
|
||
|
uint8_t filtered_top_array[2 * 32 + 1];
|
||
|
|
||
|
uint8_t *left = left_array + 1;
|
||
|
uint8_t *top = top_array + 1;
|
||
|
uint8_t *filtered_left = filtered_left_array + 1;
|
||
|
uint8_t *filtered_top = filtered_top_array + 1;
|
||
|
int cand_bottom_left = lc->na.cand_bottom_left
|
||
|
&& cur_tb_addr >
|
||
|
s->ps.pps->min_tb_addr_zs[((y_tb + size_in_tbs_v) & s->ps.sps->tb_mask) *
|
||
|
(s->ps.sps->tb_mask + 2) + (x_tb - 1)];
|
||
|
int cand_left = lc->na.cand_left;
|
||
|
int cand_up_left = lc->na.cand_up_left;
|
||
|
int cand_up = lc->na.cand_up;
|
||
|
int cand_up_right = lc->na.cand_up_right
|
||
|
&& cur_tb_addr >
|
||
|
s->ps.pps->min_tb_addr_zs[(y_tb - 1) * (s->ps.sps->tb_mask + 2) +
|
||
|
((x_tb + size_in_tbs_h) & s->ps.sps->tb_mask)];
|
||
|
|
||
|
int bottom_left_size =
|
||
|
(((y0 + 2 * size_in_luma_v) >
|
||
|
(s->ps.sps->height) ? (s->ps.sps->height) : (y0 +
|
||
|
2 * size_in_luma_v)) -
|
||
|
(y0 + size_in_luma_v)) >> vshift;
|
||
|
int top_right_size =
|
||
|
(((x0 + 2 * size_in_luma_h) >
|
||
|
(s->ps.sps->width) ? (s->ps.sps->width) : (x0 + 2 * size_in_luma_h)) -
|
||
|
(x0 + size_in_luma_h)) >> hshift;
|
||
|
|
||
|
if (s->ps.pps->constrained_intra_pred_flag == 1) {
|
||
|
int size_in_luma_pu_v = ((size_in_luma_v) >> s->ps.sps->log2_min_pu_size);
|
||
|
int size_in_luma_pu_h = ((size_in_luma_h) >> s->ps.sps->log2_min_pu_size);
|
||
|
int on_pu_edge_x = !(x0 & ((1 << s->ps.sps->log2_min_pu_size) - 1));
|
||
|
int on_pu_edge_y = !(y0 & ((1 << s->ps.sps->log2_min_pu_size) - 1));
|
||
|
if (!size_in_luma_pu_h)
|
||
|
size_in_luma_pu_h++;
|
||
|
if (cand_bottom_left == 1 && on_pu_edge_x) {
|
||
|
int x_left_pu = ((x0 - 1) >> s->ps.sps->log2_min_pu_size);
|
||
|
int y_bottom_pu =
|
||
|
((y0 + size_in_luma_v) >> s->ps.sps->log2_min_pu_size);
|
||
|
int max =
|
||
|
((size_in_luma_pu_v) >
|
||
|
(s->ps.sps->min_pu_height -
|
||
|
y_bottom_pu) ? (s->ps.sps->min_pu_height -
|
||
|
y_bottom_pu) : (size_in_luma_pu_v));
|
||
|
cand_bottom_left = 0;
|
||
|
for (i = 0; i < max; i += 2)
|
||
|
cand_bottom_left |=
|
||
|
((s->ref->tab_mvf[(x_left_pu) +
|
||
|
(y_bottom_pu +
|
||
|
i) * min_pu_width]).pred_flag ==
|
||
|
PF_INTRA);
|
||
|
}
|
||
|
if (cand_left == 1 && on_pu_edge_x) {
|
||
|
int x_left_pu = ((x0 - 1) >> s->ps.sps->log2_min_pu_size);
|
||
|
int y_left_pu = ((y0) >> s->ps.sps->log2_min_pu_size);
|
||
|
int max =
|
||
|
((size_in_luma_pu_v) >
|
||
|
(s->ps.sps->min_pu_height -
|
||
|
y_left_pu) ? (s->ps.sps->min_pu_height -
|
||
|
y_left_pu) : (size_in_luma_pu_v));
|
||
|
cand_left = 0;
|
||
|
for (i = 0; i < max; i += 2)
|
||
|
cand_left |=
|
||
|
((s->ref->tab_mvf[(x_left_pu) +
|
||
|
(y_left_pu +
|
||
|
i) * min_pu_width]).pred_flag ==
|
||
|
PF_INTRA);
|
||
|
}
|
||
|
if (cand_up_left == 1) {
|
||
|
int x_left_pu = ((x0 - 1) >> s->ps.sps->log2_min_pu_size);
|
||
|
int y_top_pu = ((y0 - 1) >> s->ps.sps->log2_min_pu_size);
|
||
|
cand_up_left =
|
||
|
(s->ref->tab_mvf[(x_left_pu) +
|
||
|
(y_top_pu) * min_pu_width]).pred_flag ==
|
||
|
PF_INTRA;
|
||
|
}
|
||
|
if (cand_up == 1 && on_pu_edge_y) {
|
||
|
int x_top_pu = ((x0) >> s->ps.sps->log2_min_pu_size);
|
||
|
int y_top_pu = ((y0 - 1) >> s->ps.sps->log2_min_pu_size);
|
||
|
int max =
|
||
|
((size_in_luma_pu_h) >
|
||
|
(s->ps.sps->min_pu_width -
|
||
|
x_top_pu) ? (s->ps.sps->min_pu_width -
|
||
|
x_top_pu) : (size_in_luma_pu_h));
|
||
|
cand_up = 0;
|
||
|
for (i = 0; i < max; i += 2)
|
||
|
cand_up |=
|
||
|
((s->ref->tab_mvf[(x_top_pu + i) +
|
||
|
(y_top_pu) *
|
||
|
min_pu_width]).pred_flag == PF_INTRA);
|
||
|
}
|
||
|
if (cand_up_right == 1 && on_pu_edge_y) {
|
||
|
int y_top_pu = ((y0 - 1) >> s->ps.sps->log2_min_pu_size);
|
||
|
int x_right_pu =
|
||
|
((x0 + size_in_luma_h) >> s->ps.sps->log2_min_pu_size);
|
||
|
int max =
|
||
|
((size_in_luma_pu_h) >
|
||
|
(s->ps.sps->min_pu_width -
|
||
|
x_right_pu) ? (s->ps.sps->min_pu_width -
|
||
|
x_right_pu) : (size_in_luma_pu_h));
|
||
|
cand_up_right = 0;
|
||
|
for (i = 0; i < max; i += 2)
|
||
|
cand_up_right |=
|
||
|
((s->ref->tab_mvf[(x_right_pu + i) +
|
||
|
(y_top_pu) *
|
||
|
min_pu_width]).pred_flag == PF_INTRA);
|
||
|
}
|
||
|
|
||
|
vec0 = (v16u8) __msa_ldi_b(128);
|
||
|
|
||
|
ST_UB4(vec0, vec0, vec0, vec0, left, 16);
|
||
|
|
||
|
ST_UB4(vec0, vec0, vec0, vec0, top, 16);
|
||
|
|
||
|
top[-1] = 128;
|
||
|
}
|
||
|
if (cand_up_left) {
|
||
|
left[-1] = src[(-1) + stride * (-1)];
|
||
|
top[-1] = left[-1];
|
||
|
}
|
||
|
if (cand_up) {
|
||
|
vec0 = LD_UB(src - stride);
|
||
|
ST_UB(vec0, top);
|
||
|
}
|
||
|
if (cand_up_right) {
|
||
|
vec0 = LD_UB(src - stride + 16);
|
||
|
ST_UB(vec0, (top + 16));
|
||
|
|
||
|
do {
|
||
|
uint32_t pix =
|
||
|
((src[(16 + top_right_size - 1) + stride * (-1)]) *
|
||
|
0x01010101U);
|
||
|
for (i = 0; i < (16 - top_right_size); i += 4)
|
||
|
((((union unaligned_32 *) (top + 16 + top_right_size +
|
||
|
i))->l) = (pix));
|
||
|
} while (0);
|
||
|
}
|
||
|
if (cand_left)
|
||
|
for (i = 0; i < 16; i++)
|
||
|
left[i] = src[(-1) + stride * (i)];
|
||
|
if (cand_bottom_left) {
|
||
|
for (i = 16; i < 16 + bottom_left_size; i++)
|
||
|
left[i] = src[(-1) + stride * (i)];
|
||
|
do {
|
||
|
uint32_t pix =
|
||
|
((src[(-1) + stride * (16 + bottom_left_size - 1)]) *
|
||
|
0x01010101U);
|
||
|
for (i = 0; i < (16 - bottom_left_size); i += 4)
|
||
|
((((union unaligned_32 *) (left + 16 + bottom_left_size +
|
||
|
i))->l) = (pix));
|
||
|
} while (0);
|
||
|
}
|
||
|
|
||
|
if (s->ps.pps->constrained_intra_pred_flag == 1) {
|
||
|
if (cand_bottom_left || cand_left || cand_up_left || cand_up
|
||
|
|| cand_up_right) {
|
||
|
int size_max_x =
|
||
|
x0 + ((2 * 16) << hshift) <
|
||
|
s->ps.sps->width ? 2 * 16 : (s->ps.sps->width - x0) >> hshift;
|
||
|
int size_max_y =
|
||
|
y0 + ((2 * 16) << vshift) <
|
||
|
s->ps.sps->height ? 2 * 16 : (s->ps.sps->height - y0) >> vshift;
|
||
|
int j = 16 + (cand_bottom_left ? bottom_left_size : 0) - 1;
|
||
|
if (!cand_up_right) {
|
||
|
size_max_x = x0 + ((16) << hshift) < s->ps.sps->width ?
|
||
|
16 : (s->ps.sps->width - x0) >> hshift;
|
||
|
}
|
||
|
if (!cand_bottom_left) {
|
||
|
size_max_y = y0 + ((16) << vshift) < s->ps.sps->height ?
|
||
|
16 : (s->ps.sps->height - y0) >> vshift;
|
||
|
}
|
||
|
if (cand_bottom_left || cand_left || cand_up_left) {
|
||
|
while (j > -1
|
||
|
&&
|
||
|
!((s->ref->tab_mvf[(((x0 +
|
||
|
((-1) << hshift)) >> s->ps.sps->
|
||
|
log2_min_pu_size)) + (((y0 +
|
||
|
((j) <<
|
||
|
vshift))
|
||
|
>> s->ps.sps->
|
||
|
log2_min_pu_size))
|
||
|
* min_pu_width]).pred_flag ==
|
||
|
PF_INTRA))
|
||
|
j--;
|
||
|
if (!
|
||
|
((s->ref->tab_mvf[(((x0 +
|
||
|
((-1) << hshift)) >> s->ps.sps->
|
||
|
log2_min_pu_size)) + (((y0 + ((j)
|
||
|
<<
|
||
|
vshift))
|
||
|
>> s->ps.sps->
|
||
|
log2_min_pu_size))
|
||
|
* min_pu_width]).pred_flag == PF_INTRA)) {
|
||
|
j = 0;
|
||
|
while (j < size_max_x
|
||
|
&&
|
||
|
!((s->ref->tab_mvf[(((x0 +
|
||
|
((j) << hshift)) >> s->ps.sps->
|
||
|
log2_min_pu_size)) + (((y0 +
|
||
|
((-1) <<
|
||
|
vshift))
|
||
|
>> s->
|
||
|
ps.sps->
|
||
|
log2_min_pu_size))
|
||
|
* min_pu_width]).pred_flag ==
|
||
|
PF_INTRA))
|
||
|
j++;
|
||
|
for (i = j; i > (j) - (j + 1); i--)
|
||
|
if (!
|
||
|
((s->ref->tab_mvf[(((x0 +
|
||
|
((i -
|
||
|
1) << hshift)) >> s->ps.sps->
|
||
|
log2_min_pu_size)) + (((y0 +
|
||
|
((-1) <<
|
||
|
vshift))
|
||
|
>> s->
|
||
|
ps.sps->
|
||
|
log2_min_pu_size))
|
||
|
* min_pu_width]).pred_flag ==
|
||
|
PF_INTRA))
|
||
|
top[i - 1] = top[i];
|
||
|
left[-1] = top[-1];
|
||
|
}
|
||
|
} else {
|
||
|
j = 0;
|
||
|
while (j < size_max_x
|
||
|
&&
|
||
|
!((s->ref->tab_mvf[(((x0 +
|
||
|
((j) << hshift)) >> s->ps.sps->
|
||
|
log2_min_pu_size)) + (((y0 + ((-1)
|
||
|
<<
|
||
|
vshift))
|
||
|
>> s->ps.sps->
|
||
|
log2_min_pu_size))
|
||
|
* min_pu_width]).pred_flag ==
|
||
|
PF_INTRA))
|
||
|
j++;
|
||
|
if (j > 0)
|
||
|
if (x0 > 0) {
|
||
|
for (i = j; i > (j) - (j + 1); i--)
|
||
|
if (!
|
||
|
((s->ref->tab_mvf[(((x0 +
|
||
|
((i -
|
||
|
1) << hshift)) >>
|
||
|
s->ps.sps->log2_min_pu_size))
|
||
|
+ (((y0 + ((-1)
|
||
|
<< vshift))
|
||
|
>>
|
||
|
s->ps.sps->log2_min_pu_size))
|
||
|
*
|
||
|
min_pu_width]).pred_flag ==
|
||
|
PF_INTRA))
|
||
|
top[i - 1] = top[i];
|
||
|
} else {
|
||
|
for (i = j; i > (j) - (j); i--)
|
||
|
if (!
|
||
|
((s->ref->tab_mvf[(((x0 +
|
||
|
((i -
|
||
|
1) << hshift)) >>
|
||
|
s->ps.sps->log2_min_pu_size))
|
||
|
+ (((y0 + ((-1)
|
||
|
<< vshift))
|
||
|
>>
|
||
|
s->ps.sps->log2_min_pu_size))
|
||
|
*
|
||
|
min_pu_width]).pred_flag ==
|
||
|
PF_INTRA))
|
||
|
top[i - 1] = top[i];
|
||
|
top[-1] = top[0];
|
||
|
}
|
||
|
left[-1] = top[-1];
|
||
|
}
|
||
|
left[-1] = top[-1];
|
||
|
if (cand_bottom_left || cand_left) {
|
||
|
a = ((left[-1]) * 0x01010101U);
|
||
|
for (i = 0; i < (0) + (size_max_y); i += 4)
|
||
|
if (!
|
||
|
((s->ref->tab_mvf[(((x0 +
|
||
|
((-1) << hshift)) >> s->ps.sps->
|
||
|
log2_min_pu_size)) + (((y0 +
|
||
|
((i) <<
|
||
|
vshift))
|
||
|
>> s->ps.sps->
|
||
|
log2_min_pu_size))
|
||
|
* min_pu_width]).pred_flag ==
|
||
|
PF_INTRA))
|
||
|
((((union unaligned_32 *) (&left[i]))->l) = (a));
|
||
|
else
|
||
|
a = ((left[i + 3]) * 0x01010101U);
|
||
|
}
|
||
|
if (!cand_left) {
|
||
|
vec0 = (v16u8) __msa_fill_b(left[-1]);
|
||
|
|
||
|
ST_UB(vec0, left);
|
||
|
}
|
||
|
if (!cand_bottom_left) {
|
||
|
|
||
|
vec0 = (v16u8) __msa_fill_b(left[15]);
|
||
|
|
||
|
ST_UB(vec0, (left + 16));
|
||
|
}
|
||
|
if (x0 != 0 && y0 != 0) {
|
||
|
a = ((left[size_max_y - 1]) * 0x01010101U);
|
||
|
for (i = (size_max_y - 1);
|
||
|
i > (size_max_y - 1) - (size_max_y); i -= 4)
|
||
|
if (!
|
||
|
((s->ref->tab_mvf[(((x0 +
|
||
|
((-1) << hshift)) >> s->ps.sps->
|
||
|
log2_min_pu_size)) + (((y0 +
|
||
|
((i -
|
||
|
3) <<
|
||
|
vshift))
|
||
|
>> s->ps.sps->
|
||
|
log2_min_pu_size))
|
||
|
* min_pu_width]).pred_flag ==
|
||
|
PF_INTRA))
|
||
|
((((union unaligned_32 *) (&left[i - 3]))->l) = (a));
|
||
|
else
|
||
|
a = ((left[i - 3]) * 0x01010101U);
|
||
|
if (!
|
||
|
((s->ref->tab_mvf[(((x0 +
|
||
|
((-1) << hshift)) >> s->ps.sps->
|
||
|
log2_min_pu_size)) + (((y0 + ((-1)
|
||
|
<<
|
||
|
vshift))
|
||
|
>> s->ps.sps->
|
||
|
log2_min_pu_size))
|
||
|
* min_pu_width]).pred_flag == PF_INTRA))
|
||
|
left[-1] = left[0];
|
||
|
} else if (x0 == 0) {
|
||
|
do {
|
||
|
uint32_t pix = ((0) * 0x01010101U);
|
||
|
for (i = 0; i < (size_max_y); i += 4)
|
||
|
((((union unaligned_32 *) (left + i))->l) = (pix));
|
||
|
} while (0);
|
||
|
} else {
|
||
|
a = ((left[size_max_y - 1]) * 0x01010101U);
|
||
|
for (i = (size_max_y - 1);
|
||
|
i > (size_max_y - 1) - (size_max_y); i -= 4)
|
||
|
if (!
|
||
|
((s->ref->tab_mvf[(((x0 +
|
||
|
((-1) << hshift)) >> s->ps.sps->
|
||
|
log2_min_pu_size)) + (((y0 +
|
||
|
((i -
|
||
|
3) <<
|
||
|
vshift))
|
||
|
>> s->ps.sps->
|
||
|
log2_min_pu_size))
|
||
|
* min_pu_width]).pred_flag ==
|
||
|
PF_INTRA))
|
||
|
((((union unaligned_32 *) (&left[i - 3]))->l) = (a));
|
||
|
else
|
||
|
a = ((left[i - 3]) * 0x01010101U);
|
||
|
}
|
||
|
top[-1] = left[-1];
|
||
|
if (y0 != 0) {
|
||
|
a = ((left[-1]) * 0x01010101U);
|
||
|
for (i = 0; i < (0) + (size_max_x); i += 4)
|
||
|
if (!
|
||
|
((s->ref->tab_mvf[(((x0 +
|
||
|
((i) << hshift)) >> s->ps.sps->
|
||
|
log2_min_pu_size)) + (((y0 + ((-1)
|
||
|
<<
|
||
|
vshift))
|
||
|
>> s->ps.sps->
|
||
|
log2_min_pu_size))
|
||
|
* min_pu_width]).pred_flag ==
|
||
|
PF_INTRA))
|
||
|
((((union unaligned_32 *) (&top[i]))->l) = (a));
|
||
|
else
|
||
|
a = ((top[i + 3]) * 0x01010101U);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (!cand_bottom_left) {
|
||
|
if (cand_left) {
|
||
|
vec0 = (v16u8) __msa_fill_b(left[15]);
|
||
|
|
||
|
ST_UB(vec0, (left + 16));
|
||
|
|
||
|
} else if (cand_up_left) {
|
||
|
vec0 = (v16u8) __msa_fill_b(left[-1]);
|
||
|
|
||
|
ST_UB2(vec0, vec0, left, 16);
|
||
|
|
||
|
cand_left = 1;
|
||
|
} else if (cand_up) {
|
||
|
left[-1] = top[0];
|
||
|
|
||
|
vec0 = (v16u8) __msa_fill_b(left[-1]);
|
||
|
|
||
|
ST_UB2(vec0, vec0, left, 16);
|
||
|
|
||
|
cand_up_left = 1;
|
||
|
cand_left = 1;
|
||
|
} else if (cand_up_right) {
|
||
|
vec0 = (v16u8) __msa_fill_b(top[16]);
|
||
|
|
||
|
ST_UB(vec0, top);
|
||
|
|
||
|
left[-1] = top[16];
|
||
|
|
||
|
ST_UB2(vec0, vec0, left, 16);
|
||
|
|
||
|
cand_up = 1;
|
||
|
cand_up_left = 1;
|
||
|
cand_left = 1;
|
||
|
} else {
|
||
|
left[-1] = 128;
|
||
|
vec0 = (v16u8) __msa_ldi_b(128);
|
||
|
|
||
|
ST_UB2(vec0, vec0, top, 16);
|
||
|
ST_UB2(vec0, vec0, left, 16);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (!cand_left) {
|
||
|
vec0 = (v16u8) __msa_fill_b(left[16]);
|
||
|
ST_UB(vec0, left);
|
||
|
}
|
||
|
if (!cand_up_left) {
|
||
|
left[-1] = left[0];
|
||
|
}
|
||
|
if (!cand_up) {
|
||
|
vec0 = (v16u8) __msa_fill_b(left[-1]);
|
||
|
ST_UB(vec0, top);
|
||
|
}
|
||
|
if (!cand_up_right) {
|
||
|
vec0 = (v16u8) __msa_fill_b(top[15]);
|
||
|
ST_UB(vec0, (top + 16));
|
||
|
}
|
||
|
|
||
|
top[-1] = left[-1];
|
||
|
|
||
|
|
||
|
if (!s->ps.sps->intra_smoothing_disabled_flag
|
||
|
&& (c_idx == 0 || s->ps.sps->chroma_format_idc == 3)) {
|
||
|
if (mode != INTRA_DC && 16 != 4) {
|
||
|
int intra_hor_ver_dist_thresh[] = { 7, 1, 0 };
|
||
|
int min_dist_vert_hor =
|
||
|
(((((int) (mode - 26U)) >=
|
||
|
0 ? ((int) (mode - 26U)) : (-((int) (mode - 26U))))) >
|
||
|
((((int) (mode - 10U)) >=
|
||
|
0 ? ((int) (mode - 10U)) : (-((int) (mode - 10U)))))
|
||
|
? ((((int) (mode - 10U)) >=
|
||
|
0 ? ((int) (mode - 10U)) : (-((int) (mode - 10U)))))
|
||
|
: ((((int) (mode - 26U)) >=
|
||
|
0 ? ((int) (mode - 26U)) : (-((int) (mode - 26U))))));
|
||
|
if (min_dist_vert_hor > intra_hor_ver_dist_thresh[4 - 3]) {
|
||
|
filtered_left[2 * 16 - 1] = left[2 * 16 - 1];
|
||
|
filtered_top[2 * 16 - 1] = top[2 * 16 - 1];
|
||
|
for (i = 2 * 16 - 2; i >= 0; i--)
|
||
|
filtered_left[i] = (left[i + 1] + 2 * left[i] +
|
||
|
left[i - 1] + 2) >> 2;
|
||
|
filtered_top[-1] =
|
||
|
filtered_left[-1] =
|
||
|
(left[0] + 2 * left[-1] + top[0] + 2) >> 2;
|
||
|
for (i = 2 * 16 - 2; i >= 0; i--)
|
||
|
filtered_top[i] = (top[i + 1] + 2 * top[i] +
|
||
|
top[i - 1] + 2) >> 2;
|
||
|
left = filtered_left;
|
||
|
top = filtered_top;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
switch (mode) {
|
||
|
case INTRA_PLANAR:
|
||
|
s->hpc.pred_planar[4 - 2] ((uint8_t *) src, (uint8_t *) top,
|
||
|
(uint8_t *) left, stride);
|
||
|
break;
|
||
|
case INTRA_DC:
|
||
|
s->hpc.pred_dc((uint8_t *) src, (uint8_t *) top,
|
||
|
(uint8_t *) left, stride, 4, c_idx);
|
||
|
break;
|
||
|
default:
|
||
|
s->hpc.pred_angular[4 - 2] ((uint8_t *) src, (uint8_t *) top,
|
||
|
(uint8_t *) left, stride, c_idx, mode);
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
void ff_intra_pred_8_32x32_msa(HEVCContext *s, int x0, int y0, int c_idx)
|
||
|
{
|
||
|
v16u8 vec0, vec1;
|
||
|
v8i16 tmp0, tmp1, tmp2, tmp3, tmp4, tmp5, tmp6, tmp7;
|
||
|
v8i16 res0, res1, res2, res3;
|
||
|
v8i16 mul_val0 = { 63, 62, 61, 60, 59, 58, 57, 56 };
|
||
|
v8i16 mul_val1 = { 1, 2, 3, 4, 5, 6, 7, 8 };
|
||
|
HEVCLocalContext *lc = s->HEVClc;
|
||
|
int i;
|
||
|
int hshift = s->ps.sps->hshift[c_idx];
|
||
|
int vshift = s->ps.sps->vshift[c_idx];
|
||
|
int size_in_luma_h = 32 << hshift;
|
||
|
int size_in_tbs_h = size_in_luma_h >> s->ps.sps->log2_min_tb_size;
|
||
|
int size_in_luma_v = 32 << vshift;
|
||
|
int size_in_tbs_v = size_in_luma_v >> s->ps.sps->log2_min_tb_size;
|
||
|
int x = x0 >> hshift;
|
||
|
int y = y0 >> vshift;
|
||
|
int x_tb = (x0 >> s->ps.sps->log2_min_tb_size) & s->ps.sps->tb_mask;
|
||
|
int y_tb = (y0 >> s->ps.sps->log2_min_tb_size) & s->ps.sps->tb_mask;
|
||
|
|
||
|
int cur_tb_addr =
|
||
|
s->ps.pps->min_tb_addr_zs[(y_tb) * (s->ps.sps->tb_mask + 2) + (x_tb)];
|
||
|
|
||
|
ptrdiff_t stride = s->frame->linesize[c_idx] / sizeof(uint8_t);
|
||
|
uint8_t *src = (uint8_t *) s->frame->data[c_idx] + x + y * stride;
|
||
|
|
||
|
int min_pu_width = s->ps.sps->min_pu_width;
|
||
|
|
||
|
enum IntraPredMode mode = c_idx ? lc->tu.intra_pred_mode_c :
|
||
|
lc->tu.intra_pred_mode;
|
||
|
uint32_t a;
|
||
|
uint8_t left_array[2 * 32 + 1];
|
||
|
uint8_t filtered_left_array[2 * 32 + 1];
|
||
|
uint8_t top_array[2 * 32 + 1];
|
||
|
uint8_t filtered_top_array[2 * 32 + 1];
|
||
|
|
||
|
uint8_t *left = left_array + 1;
|
||
|
uint8_t *top = top_array + 1;
|
||
|
uint8_t *filtered_left = filtered_left_array + 1;
|
||
|
uint8_t *filtered_top = filtered_top_array + 1;
|
||
|
int cand_bottom_left = lc->na.cand_bottom_left
|
||
|
&& cur_tb_addr >
|
||
|
s->ps.pps->min_tb_addr_zs[((y_tb + size_in_tbs_v) & s->ps.sps->tb_mask) *
|
||
|
(s->ps.sps->tb_mask + 2) + (x_tb - 1)];
|
||
|
int cand_left = lc->na.cand_left;
|
||
|
int cand_up_left = lc->na.cand_up_left;
|
||
|
int cand_up = lc->na.cand_up;
|
||
|
int cand_up_right = lc->na.cand_up_right
|
||
|
&& cur_tb_addr >
|
||
|
s->ps.pps->min_tb_addr_zs[(y_tb - 1) * (s->ps.sps->tb_mask + 2) +
|
||
|
((x_tb + size_in_tbs_h) & s->ps.sps->tb_mask)];
|
||
|
|
||
|
int bottom_left_size =
|
||
|
(((y0 + 2 * size_in_luma_v) >
|
||
|
(s->ps.sps->height) ? (s->ps.sps->height) : (y0 +
|
||
|
2 * size_in_luma_v)) -
|
||
|
(y0 + size_in_luma_v)) >> vshift;
|
||
|
int top_right_size =
|
||
|
(((x0 + 2 * size_in_luma_h) >
|
||
|
(s->ps.sps->width) ? (s->ps.sps->width) : (x0 + 2 * size_in_luma_h)) -
|
||
|
(x0 + size_in_luma_h)) >> hshift;
|
||
|
|
||
|
if (s->ps.pps->constrained_intra_pred_flag == 1) {
|
||
|
int size_in_luma_pu_v = ((size_in_luma_v) >> s->ps.sps->log2_min_pu_size);
|
||
|
int size_in_luma_pu_h = ((size_in_luma_h) >> s->ps.sps->log2_min_pu_size);
|
||
|
int on_pu_edge_x = !(x0 & ((1 << s->ps.sps->log2_min_pu_size) - 1));
|
||
|
int on_pu_edge_y = !(y0 & ((1 << s->ps.sps->log2_min_pu_size) - 1));
|
||
|
if (!size_in_luma_pu_h)
|
||
|
size_in_luma_pu_h++;
|
||
|
if (cand_bottom_left == 1 && on_pu_edge_x) {
|
||
|
int x_left_pu = ((x0 - 1) >> s->ps.sps->log2_min_pu_size);
|
||
|
int y_bottom_pu =
|
||
|
((y0 + size_in_luma_v) >> s->ps.sps->log2_min_pu_size);
|
||
|
int max =
|
||
|
((size_in_luma_pu_v) >
|
||
|
(s->ps.sps->min_pu_height -
|
||
|
y_bottom_pu) ? (s->ps.sps->min_pu_height -
|
||
|
y_bottom_pu) : (size_in_luma_pu_v));
|
||
|
cand_bottom_left = 0;
|
||
|
for (i = 0; i < max; i += 2)
|
||
|
cand_bottom_left |=
|
||
|
((s->ref->tab_mvf[(x_left_pu) +
|
||
|
(y_bottom_pu +
|
||
|
i) * min_pu_width]).pred_flag ==
|
||
|
PF_INTRA);
|
||
|
}
|
||
|
if (cand_left == 1 && on_pu_edge_x) {
|
||
|
int x_left_pu = ((x0 - 1) >> s->ps.sps->log2_min_pu_size);
|
||
|
int y_left_pu = ((y0) >> s->ps.sps->log2_min_pu_size);
|
||
|
int max =
|
||
|
((size_in_luma_pu_v) >
|
||
|
(s->ps.sps->min_pu_height -
|
||
|
y_left_pu) ? (s->ps.sps->min_pu_height -
|
||
|
y_left_pu) : (size_in_luma_pu_v));
|
||
|
cand_left = 0;
|
||
|
for (i = 0; i < max; i += 2)
|
||
|
cand_left |=
|
||
|
((s->ref->tab_mvf[(x_left_pu) +
|
||
|
(y_left_pu +
|
||
|
i) * min_pu_width]).pred_flag ==
|
||
|
PF_INTRA);
|
||
|
}
|
||
|
if (cand_up_left == 1) {
|
||
|
int x_left_pu = ((x0 - 1) >> s->ps.sps->log2_min_pu_size);
|
||
|
int y_top_pu = ((y0 - 1) >> s->ps.sps->log2_min_pu_size);
|
||
|
cand_up_left =
|
||
|
(s->ref->tab_mvf[(x_left_pu) +
|
||
|
(y_top_pu) * min_pu_width]).pred_flag ==
|
||
|
PF_INTRA;
|
||
|
}
|
||
|
if (cand_up == 1 && on_pu_edge_y) {
|
||
|
int x_top_pu = ((x0) >> s->ps.sps->log2_min_pu_size);
|
||
|
int y_top_pu = ((y0 - 1) >> s->ps.sps->log2_min_pu_size);
|
||
|
int max =
|
||
|
((size_in_luma_pu_h) >
|
||
|
(s->ps.sps->min_pu_width -
|
||
|
x_top_pu) ? (s->ps.sps->min_pu_width -
|
||
|
x_top_pu) : (size_in_luma_pu_h));
|
||
|
cand_up = 0;
|
||
|
for (i = 0; i < max; i += 2)
|
||
|
cand_up |=
|
||
|
((s->ref->tab_mvf[(x_top_pu + i) +
|
||
|
(y_top_pu) *
|
||
|
min_pu_width]).pred_flag == PF_INTRA);
|
||
|
}
|
||
|
if (cand_up_right == 1 && on_pu_edge_y) {
|
||
|
int y_top_pu = ((y0 - 1) >> s->ps.sps->log2_min_pu_size);
|
||
|
int x_right_pu =
|
||
|
((x0 + size_in_luma_h) >> s->ps.sps->log2_min_pu_size);
|
||
|
int max =
|
||
|
((size_in_luma_pu_h) >
|
||
|
(s->ps.sps->min_pu_width -
|
||
|
x_right_pu) ? (s->ps.sps->min_pu_width -
|
||
|
x_right_pu) : (size_in_luma_pu_h));
|
||
|
cand_up_right = 0;
|
||
|
for (i = 0; i < max; i += 2)
|
||
|
cand_up_right |=
|
||
|
((s->ref->tab_mvf[(x_right_pu + i) +
|
||
|
(y_top_pu) *
|
||
|
min_pu_width]).pred_flag == PF_INTRA);
|
||
|
}
|
||
|
vec0 = (v16u8) __msa_ldi_b(128);
|
||
|
|
||
|
ST_UB4(vec0, vec0, vec0, vec0, left, 16);
|
||
|
ST_UB4(vec0, vec0, vec0, vec0, top, 16);
|
||
|
|
||
|
top[-1] = 128;
|
||
|
}
|
||
|
if (cand_up_left) {
|
||
|
left[-1] = src[(-1) + stride * (-1)];
|
||
|
top[-1] = left[-1];
|
||
|
}
|
||
|
if (cand_up) {
|
||
|
LD_UB2(src - stride, 16, vec0, vec1);
|
||
|
ST_UB2(vec0, vec1, top, 16);
|
||
|
}
|
||
|
|
||
|
if (cand_up_right) {
|
||
|
LD_UB2(src - stride + 32, 16, vec0, vec1);
|
||
|
ST_UB2(vec0, vec1, (top + 32), 16);
|
||
|
do {
|
||
|
uint32_t pix =
|
||
|
((src[(32 + top_right_size - 1) + stride * (-1)]) *
|
||
|
0x01010101U);
|
||
|
for (i = 0; i < (32 - top_right_size); i += 4)
|
||
|
((((union unaligned_32 *) (top + 32 + top_right_size +
|
||
|
i))->l) = (pix));
|
||
|
} while (0);
|
||
|
}
|
||
|
if (cand_left)
|
||
|
for (i = 0; i < 32; i++)
|
||
|
left[i] = src[(-1) + stride * (i)];
|
||
|
if (cand_bottom_left) {
|
||
|
for (i = 32; i < 32 + bottom_left_size; i++)
|
||
|
left[i] = src[(-1) + stride * (i)];
|
||
|
do {
|
||
|
uint32_t pix =
|
||
|
((src[(-1) + stride * (32 + bottom_left_size - 1)]) *
|
||
|
0x01010101U);
|
||
|
for (i = 0; i < (32 - bottom_left_size); i += 4)
|
||
|
((((union unaligned_32 *) (left + 32 + bottom_left_size +
|
||
|
i))->l) = (pix));
|
||
|
} while (0);
|
||
|
}
|
||
|
|
||
|
if (s->ps.pps->constrained_intra_pred_flag == 1) {
|
||
|
if (cand_bottom_left || cand_left || cand_up_left || cand_up
|
||
|
|| cand_up_right) {
|
||
|
int size_max_x =
|
||
|
x0 + ((2 * 32) << hshift) <
|
||
|
s->ps.sps->width ? 2 * 32 : (s->ps.sps->width - x0) >> hshift;
|
||
|
int size_max_y =
|
||
|
y0 + ((2 * 32) << vshift) <
|
||
|
s->ps.sps->height ? 2 * 32 : (s->ps.sps->height - y0) >> vshift;
|
||
|
int j = 32 + (cand_bottom_left ? bottom_left_size : 0) - 1;
|
||
|
if (!cand_up_right) {
|
||
|
size_max_x = x0 + ((32) << hshift) < s->ps.sps->width ?
|
||
|
32 : (s->ps.sps->width - x0) >> hshift;
|
||
|
}
|
||
|
if (!cand_bottom_left) {
|
||
|
size_max_y = y0 + ((32) << vshift) < s->ps.sps->height ?
|
||
|
32 : (s->ps.sps->height - y0) >> vshift;
|
||
|
}
|
||
|
if (cand_bottom_left || cand_left || cand_up_left) {
|
||
|
while (j > -1
|
||
|
&&
|
||
|
!((s->ref->tab_mvf[(((x0 +
|
||
|
((-1) << hshift)) >> s->ps.sps->
|
||
|
log2_min_pu_size)) + (((y0 +
|
||
|
((j) <<
|
||
|
vshift))
|
||
|
>> s->ps.sps->
|
||
|
log2_min_pu_size))
|
||
|
* min_pu_width]).pred_flag ==
|
||
|
PF_INTRA))
|
||
|
j--;
|
||
|
if (!
|
||
|
((s->ref->tab_mvf[(((x0 +
|
||
|
((-1) << hshift)) >> s->ps.sps->
|
||
|
log2_min_pu_size)) + (((y0 + ((j)
|
||
|
<<
|
||
|
vshift))
|
||
|
>> s->ps.sps->
|
||
|
log2_min_pu_size))
|
||
|
* min_pu_width]).pred_flag == PF_INTRA)) {
|
||
|
j = 0;
|
||
|
while (j < size_max_x
|
||
|
&&
|
||
|
!((s->ref->tab_mvf[(((x0 +
|
||
|
((j) << hshift)) >> s->ps.sps->
|
||
|
log2_min_pu_size)) + (((y0 +
|
||
|
((-1) <<
|
||
|
vshift))
|
||
|
>> s->
|
||
|
ps.sps->
|
||
|
log2_min_pu_size))
|
||
|
* min_pu_width]).pred_flag ==
|
||
|
PF_INTRA))
|
||
|
j++;
|
||
|
for (i = j; i > (j) - (j + 1); i--)
|
||
|
if (!
|
||
|
((s->ref->tab_mvf[(((x0 +
|
||
|
((i -
|
||
|
1) << hshift)) >> s->ps.sps->
|
||
|
log2_min_pu_size)) + (((y0 +
|
||
|
((-1) <<
|
||
|
vshift))
|
||
|
>> s->
|
||
|
ps.sps->
|
||
|
log2_min_pu_size))
|
||
|
* min_pu_width]).pred_flag ==
|
||
|
PF_INTRA))
|
||
|
top[i - 1] = top[i];
|
||
|
left[-1] = top[-1];
|
||
|
}
|
||
|
} else {
|
||
|
j = 0;
|
||
|
while (j < size_max_x
|
||
|
&&
|
||
|
!((s->ref->tab_mvf[(((x0 +
|
||
|
((j) << hshift)) >> s->ps.sps->
|
||
|
log2_min_pu_size)) + (((y0 + ((-1)
|
||
|
<<
|
||
|
vshift))
|
||
|
>> s->ps.sps->
|
||
|
log2_min_pu_size))
|
||
|
* min_pu_width]).pred_flag ==
|
||
|
PF_INTRA))
|
||
|
j++;
|
||
|
if (j > 0)
|
||
|
if (x0 > 0) {
|
||
|
for (i = j; i > (j) - (j + 1); i--)
|
||
|
if (!
|
||
|
((s->ref->tab_mvf[(((x0 +
|
||
|
((i -
|
||
|
1) << hshift)) >>
|
||
|
s->ps.sps->log2_min_pu_size))
|
||
|
+ (((y0 + ((-1)
|
||
|
<< vshift))
|
||
|
>>
|
||
|
s->ps.sps->log2_min_pu_size))
|
||
|
*
|
||
|
min_pu_width]).pred_flag ==
|
||
|
PF_INTRA))
|
||
|
top[i - 1] = top[i];
|
||
|
} else {
|
||
|
for (i = j; i > (j) - (j); i--)
|
||
|
if (!
|
||
|
((s->ref->tab_mvf[(((x0 +
|
||
|
((i -
|
||
|
1) << hshift)) >>
|
||
|
s->ps.sps->log2_min_pu_size))
|
||
|
+ (((y0 + ((-1)
|
||
|
<< vshift))
|
||
|
>>
|
||
|
s->ps.sps->log2_min_pu_size))
|
||
|
*
|
||
|
min_pu_width]).pred_flag ==
|
||
|
PF_INTRA))
|
||
|
top[i - 1] = top[i];
|
||
|
top[-1] = top[0];
|
||
|
}
|
||
|
left[-1] = top[-1];
|
||
|
}
|
||
|
left[-1] = top[-1];
|
||
|
if (cand_bottom_left || cand_left) {
|
||
|
a = ((left[-1]) * 0x01010101U);
|
||
|
for (i = 0; i < (0) + (size_max_y); i += 4)
|
||
|
if (!
|
||
|
((s->ref->tab_mvf[(((x0 +
|
||
|
((-1) << hshift)) >> s->ps.sps->
|
||
|
log2_min_pu_size)) + (((y0 +
|
||
|
((i) <<
|
||
|
vshift))
|
||
|
>> s->ps.sps->
|
||
|
log2_min_pu_size))
|
||
|
* min_pu_width]).pred_flag ==
|
||
|
PF_INTRA))
|
||
|
((((union unaligned_32 *) (&left[i]))->l) = (a));
|
||
|
else
|
||
|
a = ((left[i + 3]) * 0x01010101U);
|
||
|
}
|
||
|
if (!cand_left) {
|
||
|
vec0 = (v16u8) __msa_fill_b(left[-1]);
|
||
|
|
||
|
ST_UB2(vec0, vec0, left, 16);
|
||
|
}
|
||
|
if (!cand_bottom_left) {
|
||
|
vec0 = (v16u8) __msa_fill_b(left[31]);
|
||
|
|
||
|
ST_UB2(vec0, vec0, (left + 32), 16);
|
||
|
}
|
||
|
if (x0 != 0 && y0 != 0) {
|
||
|
a = ((left[size_max_y - 1]) * 0x01010101U);
|
||
|
for (i = (size_max_y - 1);
|
||
|
i > (size_max_y - 1) - (size_max_y); i -= 4)
|
||
|
if (!
|
||
|
((s->ref->tab_mvf[(((x0 +
|
||
|
((-1) << hshift)) >> s->ps.sps->
|
||
|
log2_min_pu_size)) + (((y0 +
|
||
|
((i -
|
||
|
3) <<
|
||
|
vshift))
|
||
|
>> s->ps.sps->
|
||
|
log2_min_pu_size))
|
||
|
* min_pu_width]).pred_flag ==
|
||
|
PF_INTRA))
|
||
|
((((union unaligned_32 *) (&left[i - 3]))->l) = (a));
|
||
|
else
|
||
|
a = ((left[i - 3]) * 0x01010101U);
|
||
|
if (!
|
||
|
((s->ref->tab_mvf[(((x0 +
|
||
|
((-1) << hshift)) >> s->ps.sps->
|
||
|
log2_min_pu_size)) + (((y0 + ((-1)
|
||
|
<<
|
||
|
vshift))
|
||
|
>> s->ps.sps->
|
||
|
log2_min_pu_size))
|
||
|
* min_pu_width]).pred_flag == PF_INTRA))
|
||
|
left[-1] = left[0];
|
||
|
} else if (x0 == 0) {
|
||
|
do {
|
||
|
uint32_t pix = ((0) * 0x01010101U);
|
||
|
for (i = 0; i < (size_max_y); i += 4)
|
||
|
((((union unaligned_32 *) (left + i))->l) = (pix));
|
||
|
} while (0);
|
||
|
} else {
|
||
|
a = ((left[size_max_y - 1]) * 0x01010101U);
|
||
|
for (i = (size_max_y - 1);
|
||
|
i > (size_max_y - 1) - (size_max_y); i -= 4)
|
||
|
if (!
|
||
|
((s->ref->tab_mvf[(((x0 +
|
||
|
((-1) << hshift)) >> s->ps.sps->
|
||
|
log2_min_pu_size)) + (((y0 +
|
||
|
((i -
|
||
|
3) <<
|
||
|
vshift))
|
||
|
>> s->ps.sps->
|
||
|
log2_min_pu_size))
|
||
|
* min_pu_width]).pred_flag ==
|
||
|
PF_INTRA))
|
||
|
((((union unaligned_32 *) (&left[i - 3]))->l) = (a));
|
||
|
else
|
||
|
a = ((left[i - 3]) * 0x01010101U);
|
||
|
}
|
||
|
top[-1] = left[-1];
|
||
|
if (y0 != 0) {
|
||
|
a = ((left[-1]) * 0x01010101U);
|
||
|
for (i = 0; i < (0) + (size_max_x); i += 4)
|
||
|
if (!
|
||
|
((s->ref->tab_mvf[(((x0 +
|
||
|
((i) << hshift)) >> s->ps.sps->
|
||
|
log2_min_pu_size)) + (((y0 + ((-1)
|
||
|
<<
|
||
|
vshift))
|
||
|
>> s->ps.sps->
|
||
|
log2_min_pu_size))
|
||
|
* min_pu_width]).pred_flag ==
|
||
|
PF_INTRA))
|
||
|
((((union unaligned_32 *) (&top[i]))->l) = (a));
|
||
|
else
|
||
|
a = ((top[i + 3]) * 0x01010101U);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (!cand_bottom_left) {
|
||
|
if (cand_left) {
|
||
|
vec0 = (v16u8) __msa_fill_b(left[31]);
|
||
|
|
||
|
ST_UB2(vec0, vec0, (left + 32), 16);
|
||
|
} else if (cand_up_left) {
|
||
|
vec0 = (v16u8) __msa_fill_b(left[-1]);
|
||
|
|
||
|
ST_UB4(vec0, vec0, vec0, vec0, left, 16);
|
||
|
|
||
|
cand_left = 1;
|
||
|
} else if (cand_up) {
|
||
|
left[-1] = top[0];
|
||
|
|
||
|
vec0 = (v16u8) __msa_fill_b(left[-1]);
|
||
|
|
||
|
ST_UB4(vec0, vec0, vec0, vec0, left, 16);
|
||
|
|
||
|
cand_up_left = 1;
|
||
|
cand_left = 1;
|
||
|
} else if (cand_up_right) {
|
||
|
vec0 = (v16u8) __msa_fill_b(top[32]);
|
||
|
|
||
|
ST_UB2(vec0, vec0, top, 16);
|
||
|
|
||
|
left[-1] = top[32];
|
||
|
|
||
|
ST_UB4(vec0, vec0, vec0, vec0, left, 16);
|
||
|
|
||
|
cand_up = 1;
|
||
|
cand_up_left = 1;
|
||
|
cand_left = 1;
|
||
|
} else {
|
||
|
left[-1] = 128;
|
||
|
|
||
|
vec0 = (v16u8) __msa_ldi_b(128);
|
||
|
|
||
|
ST_UB4(vec0, vec0, vec0, vec0, top, 16);
|
||
|
ST_UB4(vec0, vec0, vec0, vec0, left, 16);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (!cand_left) {
|
||
|
vec0 = (v16u8) __msa_fill_b(left[32]);
|
||
|
|
||
|
ST_UB2(vec0, vec0, left, 16);
|
||
|
}
|
||
|
if (!cand_up_left) {
|
||
|
left[-1] = left[0];
|
||
|
}
|
||
|
if (!cand_up) {
|
||
|
vec0 = (v16u8) __msa_fill_b(left[-1]);
|
||
|
|
||
|
ST_UB2(vec0, vec0, top, 16);
|
||
|
}
|
||
|
if (!cand_up_right) {
|
||
|
vec0 = (v16u8) __msa_fill_b(top[31]);
|
||
|
|
||
|
ST_UB2(vec0, vec0, (top + 32), 16);
|
||
|
}
|
||
|
|
||
|
top[-1] = left[-1];
|
||
|
|
||
|
|
||
|
if (!s->ps.sps->intra_smoothing_disabled_flag
|
||
|
&& (c_idx == 0 || s->ps.sps->chroma_format_idc == 3)) {
|
||
|
if (mode != INTRA_DC && 32 != 4) {
|
||
|
int intra_hor_ver_dist_thresh[] = { 7, 1, 0 };
|
||
|
int min_dist_vert_hor =
|
||
|
(((((int) (mode - 26U)) >=
|
||
|
0 ? ((int) (mode - 26U)) : (-((int) (mode - 26U))))) >
|
||
|
((((int) (mode - 10U)) >=
|
||
|
0 ? ((int) (mode - 10U)) : (-((int) (mode - 10U)))))
|
||
|
? ((((int) (mode - 10U)) >=
|
||
|
0 ? ((int) (mode - 10U)) : (-((int) (mode - 10U)))))
|
||
|
: ((((int) (mode - 26U)) >=
|
||
|
0 ? ((int) (mode - 26U)) : (-((int) (mode - 26U))))));
|
||
|
if (min_dist_vert_hor > intra_hor_ver_dist_thresh[5 - 3]) {
|
||
|
int threshold = 1 << (8 - 5);
|
||
|
if (s->ps.sps->sps_strong_intra_smoothing_enable_flag
|
||
|
&& c_idx == 0
|
||
|
&& ((top[-1] + top[63] - 2 * top[31]) >=
|
||
|
0 ? (top[-1] + top[63] -
|
||
|
2 * top[31]) : (-(top[-1] + top[63] -
|
||
|
2 * top[31]))) < threshold
|
||
|
&& ((left[-1] + left[63] - 2 * left[31]) >=
|
||
|
0 ? (left[-1] + left[63] -
|
||
|
2 * left[31]) : (-(left[-1] + left[63] -
|
||
|
2 * left[31]))) < threshold) {
|
||
|
|
||
|
|
||
|
filtered_top[-1] = top[-1];
|
||
|
filtered_top[63] = top[63];
|
||
|
|
||
|
|
||
|
for (i = 0; i < 63; i++) {
|
||
|
filtered_top[i] =
|
||
|
((63 - i) * top[-1] + (i + 1) * top[63] + 32) >> 6;
|
||
|
}
|
||
|
|
||
|
tmp0 = __msa_fill_h(top[-1]);
|
||
|
tmp1 = __msa_fill_h(top[63]);
|
||
|
|
||
|
tmp2 = mul_val0 - 8;
|
||
|
tmp3 = mul_val0 - 16;
|
||
|
tmp4 = mul_val0 - 24;
|
||
|
tmp5 = mul_val1 + 8;
|
||
|
tmp6 = mul_val1 + 16;
|
||
|
tmp7 = mul_val1 + 24;
|
||
|
|
||
|
res0 = mul_val0 * tmp0;
|
||
|
res1 = tmp2 * tmp0;
|
||
|
res2 = tmp3 * tmp0;
|
||
|
res3 = tmp4 * tmp0;
|
||
|
res0 += mul_val1 * tmp1;
|
||
|
res1 += tmp5 * tmp1;
|
||
|
res2 += tmp6 * tmp1;
|
||
|
res3 += tmp7 * tmp1;
|
||
|
|
||
|
res0 = __msa_srari_h(res0, 6);
|
||
|
res1 = __msa_srari_h(res1, 6);
|
||
|
res2 = __msa_srari_h(res2, 6);
|
||
|
res3 = __msa_srari_h(res3, 6);
|
||
|
|
||
|
vec0 = (v16u8) __msa_pckev_b((v16i8) res1, (v16i8) res0);
|
||
|
vec1 = (v16u8) __msa_pckev_b((v16i8) res3, (v16i8) res2);
|
||
|
|
||
|
ST_UB2(vec0, vec1, filtered_top, 16);
|
||
|
|
||
|
res0 = mul_val0 - 32;
|
||
|
tmp2 = mul_val0 - 40;
|
||
|
tmp3 = mul_val0 - 48;
|
||
|
tmp4 = mul_val0 - 56;
|
||
|
res3 = mul_val1 + 32;
|
||
|
tmp5 = mul_val1 + 40;
|
||
|
tmp6 = mul_val1 + 48;
|
||
|
tmp7 = mul_val1 + 56;
|
||
|
|
||
|
res0 = res0 * tmp0;
|
||
|
res1 = tmp2 * tmp0;
|
||
|
res2 = tmp3 * tmp0;
|
||
|
res0 += res3 * tmp1;
|
||
|
res3 = tmp4 * tmp0;
|
||
|
res1 += tmp5 * tmp1;
|
||
|
res2 += tmp6 * tmp1;
|
||
|
res3 += tmp7 * tmp1;
|
||
|
|
||
|
res0 = __msa_srari_h(res0, 6);
|
||
|
res1 = __msa_srari_h(res1, 6);
|
||
|
res2 = __msa_srari_h(res2, 6);
|
||
|
res3 = __msa_srari_h(res3, 6);
|
||
|
|
||
|
vec0 = (v16u8) __msa_pckev_b((v16i8) res1, (v16i8) res0);
|
||
|
vec1 = (v16u8) __msa_pckev_b((v16i8) res3, (v16i8) res2);
|
||
|
|
||
|
ST_UB2(vec0, vec1, (filtered_top + 32), 16);
|
||
|
|
||
|
filtered_top[63] = top[63];
|
||
|
|
||
|
tmp0 = __msa_fill_h(left[-1]);
|
||
|
tmp1 = __msa_fill_h(left[63]);
|
||
|
|
||
|
tmp2 = mul_val0 - 8;
|
||
|
tmp3 = mul_val0 - 16;
|
||
|
tmp4 = mul_val0 - 24;
|
||
|
tmp5 = mul_val1 + 8;
|
||
|
tmp6 = mul_val1 + 16;
|
||
|
tmp7 = mul_val1 + 24;
|
||
|
|
||
|
res0 = mul_val0 * tmp0;
|
||
|
res1 = tmp2 * tmp0;
|
||
|
res2 = tmp3 * tmp0;
|
||
|
res3 = tmp4 * tmp0;
|
||
|
res0 += mul_val1 * tmp1;
|
||
|
res1 += tmp5 * tmp1;
|
||
|
res2 += tmp6 * tmp1;
|
||
|
res3 += tmp7 * tmp1;
|
||
|
|
||
|
res0 = __msa_srari_h(res0, 6);
|
||
|
res1 = __msa_srari_h(res1, 6);
|
||
|
res2 = __msa_srari_h(res2, 6);
|
||
|
res3 = __msa_srari_h(res3, 6);
|
||
|
|
||
|
vec0 = (v16u8) __msa_pckev_b((v16i8) res1, (v16i8) res0);
|
||
|
vec1 = (v16u8) __msa_pckev_b((v16i8) res3, (v16i8) res2);
|
||
|
|
||
|
ST_UB2(vec0, vec1, left, 16);
|
||
|
|
||
|
res0 = mul_val0 - 32;
|
||
|
tmp2 = mul_val0 - 40;
|
||
|
tmp3 = mul_val0 - 48;
|
||
|
tmp4 = mul_val0 - 56;
|
||
|
res3 = mul_val1 + 32;
|
||
|
tmp5 = mul_val1 + 40;
|
||
|
tmp6 = mul_val1 + 48;
|
||
|
tmp7 = mul_val1 + 56;
|
||
|
|
||
|
res0 = res0 * tmp0;
|
||
|
res1 = tmp2 * tmp0;
|
||
|
res2 = tmp3 * tmp0;
|
||
|
res0 += res3 * tmp1;
|
||
|
res3 = tmp4 * tmp0;
|
||
|
res1 += tmp5 * tmp1;
|
||
|
res2 += tmp6 * tmp1;
|
||
|
res3 += tmp7 * tmp1;
|
||
|
|
||
|
res0 = __msa_srari_h(res0, 6);
|
||
|
res1 = __msa_srari_h(res1, 6);
|
||
|
res2 = __msa_srari_h(res2, 6);
|
||
|
res3 = __msa_srari_h(res3, 6);
|
||
|
|
||
|
vec0 = (v16u8) __msa_pckev_b((v16i8) res1, (v16i8) res0);
|
||
|
vec1 = (v16u8) __msa_pckev_b((v16i8) res3, (v16i8) res2);
|
||
|
|
||
|
ST_UB2(vec0, vec1, (left + 32), 16);
|
||
|
|
||
|
left[63] = tmp1[0];
|
||
|
|
||
|
top = filtered_top;
|
||
|
} else {
|
||
|
filtered_left[2 * 32 - 1] = left[2 * 32 - 1];
|
||
|
filtered_top[2 * 32 - 1] = top[2 * 32 - 1];
|
||
|
for (i = 2 * 32 - 2; i >= 0; i--)
|
||
|
filtered_left[i] = (left[i + 1] + 2 * left[i] +
|
||
|
left[i - 1] + 2) >> 2;
|
||
|
filtered_top[-1] =
|
||
|
filtered_left[-1] =
|
||
|
(left[0] + 2 * left[-1] + top[0] + 2) >> 2;
|
||
|
for (i = 2 * 32 - 2; i >= 0; i--)
|
||
|
filtered_top[i] = (top[i + 1] + 2 * top[i] +
|
||
|
top[i - 1] + 2) >> 2;
|
||
|
left = filtered_left;
|
||
|
top = filtered_top;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
switch (mode) {
|
||
|
case INTRA_PLANAR:
|
||
|
s->hpc.pred_planar[3] ((uint8_t *) src, (uint8_t *) top,
|
||
|
(uint8_t *) left, stride);
|
||
|
break;
|
||
|
case INTRA_DC:
|
||
|
s->hpc.pred_dc((uint8_t *) src, (uint8_t *) top,
|
||
|
(uint8_t *) left, stride, 5, c_idx);
|
||
|
break;
|
||
|
default:
|
||
|
s->hpc.pred_angular[3] ((uint8_t *) src, (uint8_t *) top,
|
||
|
(uint8_t *) left, stride, c_idx, mode);
|
||
|
break;
|
||
|
}
|
||
|
}
|