| /* |
| * 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/hevc.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; |
| vec2 = 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; |
| vec2 = 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; |
| src0_r = 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; |
| src0_r = 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); |
| ST4x4_UB(src_vec0, 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); |
| |
| ST8x8_UB(src_vec0, src_vec1, src_vec2, src_vec3, 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_0_SH(diff0, diff2, diff4, diff6, diff1, diff3, diff5, diff7, 2); |
| 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); |
| ST4x4_UB(dst_val0, 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_B2_SH(diff1, diff3, diff0, diff2, diff1, diff3, 2); |
| SLDI_B2_SH(diff5, diff7, diff4, diff6, diff5, diff7, 2); |
| 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); |
| ST8x4_UB(dst_val0, dst_val1, 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_B2_UB(top1, top3, top0, top2, top1, top3, 1); |
| SLDI_B2_UB(top5, top7, top4, top6, top5, top7, 1); |
| 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_B2_UB(top1, top3, top0, top2, top1, top3, 1); |
| SLDI_B2_UB(top5, top7, top4, top6, top5, top7, 1); |
| 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_0_SH(diff0, diff2, diff4, diff6, diff1, diff3, diff5, diff7, 2); |
| 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); |
| |
| ST4x2_UB(dst_val0, dst, stride); |
| dst += (2 * stride); |
| ST4x2_UB(dst_val1, dst, 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_B2_SH(diff1, diff3, diff0, diff2, diff1, diff3, 2); |
| SLDI_B2_SH(diff5, diff7, diff4, diff6, diff5, diff7, 2); |
| 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); |
| ST4x8_UB(diff3, diff4, 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_B2_SB(top1, top3, top0, top2, top1, top3, 1); |
| SLDI_B2_SB(top5, top7, top4, top6, top5, top7, 1); |
| |
| 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); |
| ST4x8_UB(diff4, diff5, dst_org, stride); |
| dst_org += (8 * stride); |
| ST4x8_UB(diff6, diff7, 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_B2_SB(top1, top3, top0, top2, top1, top3, 1); |
| SLDI_B2_SB(top5, top7, top4, top6, top5, top7, 1); |
| |
| 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); |
| |
| ST2x4_UB(diff0, 0, dst_org, stride); |
| dst_org += (4 * stride); |
| ST2x4_UB(diff0, 4, dst_org, stride); |
| dst_org += (4 * stride); |
| ST2x4_UB(diff1, 0, dst_org, stride); |
| dst_org += (4 * stride); |
| ST2x4_UB(diff1, 4, dst_org, stride); |
| dst_org += (4 * stride); |
| |
| ST2x4_UB(diff2, 0, dst_org, stride); |
| dst_org += (4 * stride); |
| ST2x4_UB(diff2, 4, dst_org, stride); |
| dst_org += (4 * stride); |
| ST2x4_UB(diff3, 0, dst_org, stride); |
| dst_org += (4 * stride); |
| ST2x4_UB(diff3, 4, dst_org, stride); |
| dst_org += (4 * 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 *to
|
