678 строки
27 KiB
C
678 строки
27 KiB
C
/*
|
|
* Copyright (c) 2016, Alliance for Open Media. All rights reserved
|
|
*
|
|
* This source code is subject to the terms of the BSD 2 Clause License and
|
|
* the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
|
|
* was not distributed with this source code in the LICENSE file, you can
|
|
* obtain it at www.aomedia.org/license/software. If the Alliance for Open
|
|
* Media Patent License 1.0 was not distributed with this source code in the
|
|
* PATENTS file, you can obtain it at www.aomedia.org/license/patent.
|
|
*/
|
|
|
|
#include <assert.h>
|
|
#include "./aom_dsp_rtcd.h"
|
|
#include "aom_dsp/mips/aom_convolve_msa.h"
|
|
|
|
static void common_vt_8t_and_aver_dst_4w_msa(const uint8_t *src,
|
|
int32_t src_stride, uint8_t *dst,
|
|
int32_t dst_stride, int8_t *filter,
|
|
int32_t height) {
|
|
uint32_t loop_cnt;
|
|
v16i8 src0, src1, src2, src3, src4, src5, src6, src7, src8, src9, src10;
|
|
v16u8 dst0, dst1, dst2, dst3, out;
|
|
v16i8 src10_r, src32_r, src54_r, src76_r, src98_r, src21_r, src43_r;
|
|
v16i8 src65_r, src87_r, src109_r, src2110, src4332, src6554, src8776;
|
|
v16i8 src10998, filt0, filt1, filt2, filt3;
|
|
v8i16 filt, out10, out32;
|
|
|
|
src -= (3 * src_stride);
|
|
|
|
filt = LD_SH(filter);
|
|
SPLATI_H4_SB(filt, 0, 1, 2, 3, filt0, filt1, filt2, filt3);
|
|
|
|
LD_SB7(src, src_stride, src0, src1, src2, src3, src4, src5, src6);
|
|
src += (7 * src_stride);
|
|
|
|
ILVR_B4_SB(src1, src0, src3, src2, src5, src4, src2, src1, src10_r, src32_r,
|
|
src54_r, src21_r);
|
|
ILVR_B2_SB(src4, src3, src6, src5, src43_r, src65_r);
|
|
ILVR_D3_SB(src21_r, src10_r, src43_r, src32_r, src65_r, src54_r, src2110,
|
|
src4332, src6554);
|
|
XORI_B3_128_SB(src2110, src4332, src6554);
|
|
|
|
for (loop_cnt = (height >> 2); loop_cnt--;) {
|
|
LD_SB4(src, src_stride, src7, src8, src9, src10);
|
|
src += (4 * src_stride);
|
|
|
|
LD_UB4(dst, dst_stride, dst0, dst1, dst2, dst3);
|
|
ILVR_B4_SB(src7, src6, src8, src7, src9, src8, src10, src9, src76_r,
|
|
src87_r, src98_r, src109_r);
|
|
ILVR_D2_SB(src87_r, src76_r, src109_r, src98_r, src8776, src10998);
|
|
XORI_B2_128_SB(src8776, src10998);
|
|
out10 = FILT_8TAP_DPADD_S_H(src2110, src4332, src6554, src8776, filt0,
|
|
filt1, filt2, filt3);
|
|
out32 = FILT_8TAP_DPADD_S_H(src4332, src6554, src8776, src10998, filt0,
|
|
filt1, filt2, filt3);
|
|
SRARI_H2_SH(out10, out32, FILTER_BITS);
|
|
SAT_SH2_SH(out10, out32, 7);
|
|
out = PCKEV_XORI128_UB(out10, out32);
|
|
ILVR_W2_UB(dst1, dst0, dst3, dst2, dst0, dst2);
|
|
|
|
dst0 = (v16u8)__msa_ilvr_d((v2i64)dst2, (v2i64)dst0);
|
|
out = __msa_aver_u_b(out, dst0);
|
|
|
|
ST4x4_UB(out, out, 0, 1, 2, 3, dst, dst_stride);
|
|
dst += (4 * dst_stride);
|
|
|
|
src2110 = src6554;
|
|
src4332 = src8776;
|
|
src6554 = src10998;
|
|
src6 = src10;
|
|
}
|
|
}
|
|
|
|
static void common_vt_8t_and_aver_dst_8w_msa(const uint8_t *src,
|
|
int32_t src_stride, uint8_t *dst,
|
|
int32_t dst_stride, int8_t *filter,
|
|
int32_t height) {
|
|
uint32_t loop_cnt;
|
|
v16i8 src0, src1, src2, src3, src4, src5, src6, src7, src8, src9, src10;
|
|
v16u8 dst0, dst1, dst2, dst3;
|
|
v16i8 src10_r, src32_r, src54_r, src76_r, src98_r, src21_r, src43_r;
|
|
v16i8 src65_r, src87_r, src109_r, filt0, filt1, filt2, filt3;
|
|
v8i16 filt, out0, out1, out2, out3;
|
|
|
|
src -= (3 * src_stride);
|
|
|
|
filt = LD_SH(filter);
|
|
SPLATI_H4_SB(filt, 0, 1, 2, 3, filt0, filt1, filt2, filt3);
|
|
|
|
LD_SB7(src, src_stride, src0, src1, src2, src3, src4, src5, src6);
|
|
src += (7 * src_stride);
|
|
|
|
XORI_B7_128_SB(src0, src1, src2, src3, src4, src5, src6);
|
|
ILVR_B4_SB(src1, src0, src3, src2, src5, src4, src2, src1, src10_r, src32_r,
|
|
src54_r, src21_r);
|
|
ILVR_B2_SB(src4, src3, src6, src5, src43_r, src65_r);
|
|
|
|
for (loop_cnt = (height >> 2); loop_cnt--;) {
|
|
LD_SB4(src, src_stride, src7, src8, src9, src10);
|
|
src += (4 * src_stride);
|
|
|
|
LD_UB4(dst, dst_stride, dst0, dst1, dst2, dst3);
|
|
XORI_B4_128_SB(src7, src8, src9, src10);
|
|
ILVR_B4_SB(src7, src6, src8, src7, src9, src8, src10, src9, src76_r,
|
|
src87_r, src98_r, src109_r);
|
|
out0 = FILT_8TAP_DPADD_S_H(src10_r, src32_r, src54_r, src76_r, filt0, filt1,
|
|
filt2, filt3);
|
|
out1 = FILT_8TAP_DPADD_S_H(src21_r, src43_r, src65_r, src87_r, filt0, filt1,
|
|
filt2, filt3);
|
|
out2 = FILT_8TAP_DPADD_S_H(src32_r, src54_r, src76_r, src98_r, filt0, filt1,
|
|
filt2, filt3);
|
|
out3 = FILT_8TAP_DPADD_S_H(src43_r, src65_r, src87_r, src109_r, filt0,
|
|
filt1, filt2, filt3);
|
|
SRARI_H4_SH(out0, out1, out2, out3, FILTER_BITS);
|
|
SAT_SH4_SH(out0, out1, out2, out3, 7);
|
|
CONVERT_UB_AVG_ST8x4_UB(out0, out1, out2, out3, dst0, dst1, dst2, dst3, dst,
|
|
dst_stride);
|
|
dst += (4 * dst_stride);
|
|
|
|
src10_r = src54_r;
|
|
src32_r = src76_r;
|
|
src54_r = src98_r;
|
|
src21_r = src65_r;
|
|
src43_r = src87_r;
|
|
src65_r = src109_r;
|
|
src6 = src10;
|
|
}
|
|
}
|
|
|
|
static void common_vt_8t_and_aver_dst_16w_mult_msa(
|
|
const uint8_t *src, int32_t src_stride, uint8_t *dst, int32_t dst_stride,
|
|
int8_t *filter, int32_t height, int32_t width) {
|
|
const uint8_t *src_tmp;
|
|
uint8_t *dst_tmp;
|
|
uint32_t loop_cnt, cnt;
|
|
v16i8 src0, src1, src2, src3, src4, src5, src6, src7, src8, src9, src10;
|
|
v16i8 src10_r, src32_r, src54_r, src76_r, src98_r, src21_r, src43_r;
|
|
v16i8 src65_r, src87_r, src109_r, src10_l, src32_l, src54_l, src76_l;
|
|
v16i8 src98_l, src21_l, src43_l, src65_l, src87_l, src109_l;
|
|
v16i8 filt0, filt1, filt2, filt3;
|
|
v16u8 dst0, dst1, dst2, dst3, tmp0, tmp1, tmp2, tmp3;
|
|
v8i16 out0_r, out1_r, out2_r, out3_r, out0_l, out1_l, out2_l, out3_l, filt;
|
|
|
|
src -= (3 * src_stride);
|
|
|
|
filt = LD_SH(filter);
|
|
SPLATI_H4_SB(filt, 0, 1, 2, 3, filt0, filt1, filt2, filt3);
|
|
|
|
for (cnt = (width >> 4); cnt--;) {
|
|
src_tmp = src;
|
|
dst_tmp = dst;
|
|
|
|
LD_SB7(src_tmp, src_stride, src0, src1, src2, src3, src4, src5, src6);
|
|
XORI_B7_128_SB(src0, src1, src2, src3, src4, src5, src6);
|
|
src_tmp += (7 * src_stride);
|
|
|
|
ILVR_B4_SB(src1, src0, src3, src2, src5, src4, src2, src1, src10_r, src32_r,
|
|
src54_r, src21_r);
|
|
ILVR_B2_SB(src4, src3, src6, src5, src43_r, src65_r);
|
|
ILVL_B4_SB(src1, src0, src3, src2, src5, src4, src2, src1, src10_l, src32_l,
|
|
src54_l, src21_l);
|
|
ILVL_B2_SB(src4, src3, src6, src5, src43_l, src65_l);
|
|
|
|
for (loop_cnt = (height >> 2); loop_cnt--;) {
|
|
LD_SB4(src_tmp, src_stride, src7, src8, src9, src10);
|
|
src_tmp += (4 * src_stride);
|
|
|
|
LD_UB4(dst_tmp, dst_stride, dst0, dst1, dst2, dst3);
|
|
XORI_B4_128_SB(src7, src8, src9, src10);
|
|
ILVR_B4_SB(src7, src6, src8, src7, src9, src8, src10, src9, src76_r,
|
|
src87_r, src98_r, src109_r);
|
|
ILVL_B4_SB(src7, src6, src8, src7, src9, src8, src10, src9, src76_l,
|
|
src87_l, src98_l, src109_l);
|
|
out0_r = FILT_8TAP_DPADD_S_H(src10_r, src32_r, src54_r, src76_r, filt0,
|
|
filt1, filt2, filt3);
|
|
out1_r = FILT_8TAP_DPADD_S_H(src21_r, src43_r, src65_r, src87_r, filt0,
|
|
filt1, filt2, filt3);
|
|
out2_r = FILT_8TAP_DPADD_S_H(src32_r, src54_r, src76_r, src98_r, filt0,
|
|
filt1, filt2, filt3);
|
|
out3_r = FILT_8TAP_DPADD_S_H(src43_r, src65_r, src87_r, src109_r, filt0,
|
|
filt1, filt2, filt3);
|
|
out0_l = FILT_8TAP_DPADD_S_H(src10_l, src32_l, src54_l, src76_l, filt0,
|
|
filt1, filt2, filt3);
|
|
out1_l = FILT_8TAP_DPADD_S_H(src21_l, src43_l, src65_l, src87_l, filt0,
|
|
filt1, filt2, filt3);
|
|
out2_l = FILT_8TAP_DPADD_S_H(src32_l, src54_l, src76_l, src98_l, filt0,
|
|
filt1, filt2, filt3);
|
|
out3_l = FILT_8TAP_DPADD_S_H(src43_l, src65_l, src87_l, src109_l, filt0,
|
|
filt1, filt2, filt3);
|
|
SRARI_H4_SH(out0_r, out1_r, out2_r, out3_r, FILTER_BITS);
|
|
SRARI_H4_SH(out0_l, out1_l, out2_l, out3_l, FILTER_BITS);
|
|
SAT_SH4_SH(out0_r, out1_r, out2_r, out3_r, 7);
|
|
SAT_SH4_SH(out0_l, out1_l, out2_l, out3_l, 7);
|
|
PCKEV_B4_UB(out0_l, out0_r, out1_l, out1_r, out2_l, out2_r, out3_l,
|
|
out3_r, tmp0, tmp1, tmp2, tmp3);
|
|
XORI_B4_128_UB(tmp0, tmp1, tmp2, tmp3);
|
|
AVER_UB4_UB(tmp0, dst0, tmp1, dst1, tmp2, dst2, tmp3, dst3, dst0, dst1,
|
|
dst2, dst3);
|
|
ST_UB4(dst0, dst1, dst2, dst3, dst_tmp, dst_stride);
|
|
dst_tmp += (4 * dst_stride);
|
|
|
|
src10_r = src54_r;
|
|
src32_r = src76_r;
|
|
src54_r = src98_r;
|
|
src21_r = src65_r;
|
|
src43_r = src87_r;
|
|
src65_r = src109_r;
|
|
src10_l = src54_l;
|
|
src32_l = src76_l;
|
|
src54_l = src98_l;
|
|
src21_l = src65_l;
|
|
src43_l = src87_l;
|
|
src65_l = src109_l;
|
|
src6 = src10;
|
|
}
|
|
|
|
src += 16;
|
|
dst += 16;
|
|
}
|
|
}
|
|
|
|
static void common_vt_8t_and_aver_dst_16w_msa(const uint8_t *src,
|
|
int32_t src_stride, uint8_t *dst,
|
|
int32_t dst_stride,
|
|
int8_t *filter, int32_t height) {
|
|
common_vt_8t_and_aver_dst_16w_mult_msa(src, src_stride, dst, dst_stride,
|
|
filter, height, 16);
|
|
}
|
|
|
|
static void common_vt_8t_and_aver_dst_32w_msa(const uint8_t *src,
|
|
int32_t src_stride, uint8_t *dst,
|
|
int32_t dst_stride,
|
|
int8_t *filter, int32_t height) {
|
|
common_vt_8t_and_aver_dst_16w_mult_msa(src, src_stride, dst, dst_stride,
|
|
filter, height, 32);
|
|
}
|
|
|
|
static void common_vt_8t_and_aver_dst_64w_msa(const uint8_t *src,
|
|
int32_t src_stride, uint8_t *dst,
|
|
int32_t dst_stride,
|
|
int8_t *filter, int32_t height) {
|
|
common_vt_8t_and_aver_dst_16w_mult_msa(src, src_stride, dst, dst_stride,
|
|
filter, height, 64);
|
|
}
|
|
|
|
static void common_vt_2t_and_aver_dst_4x4_msa(const uint8_t *src,
|
|
int32_t src_stride, uint8_t *dst,
|
|
int32_t dst_stride,
|
|
int8_t *filter) {
|
|
v16i8 src0, src1, src2, src3, src4;
|
|
v16u8 dst0, dst1, dst2, dst3, out, filt0, src2110, src4332;
|
|
v16i8 src10_r, src32_r, src21_r, src43_r;
|
|
v8i16 filt;
|
|
v8u16 tmp0, tmp1;
|
|
|
|
filt = LD_SH(filter);
|
|
filt0 = (v16u8)__msa_splati_h(filt, 0);
|
|
|
|
LD_SB4(src, src_stride, src0, src1, src2, src3);
|
|
src += (4 * src_stride);
|
|
|
|
src4 = LD_SB(src);
|
|
src += src_stride;
|
|
|
|
LD_UB4(dst, dst_stride, dst0, dst1, dst2, dst3);
|
|
ILVR_W2_UB(dst1, dst0, dst3, dst2, dst0, dst1);
|
|
dst0 = (v16u8)__msa_ilvr_d((v2i64)dst1, (v2i64)dst0);
|
|
ILVR_B4_SB(src1, src0, src2, src1, src3, src2, src4, src3, src10_r, src21_r,
|
|
src32_r, src43_r);
|
|
ILVR_D2_UB(src21_r, src10_r, src43_r, src32_r, src2110, src4332);
|
|
DOTP_UB2_UH(src2110, src4332, filt0, filt0, tmp0, tmp1);
|
|
SRARI_H2_UH(tmp0, tmp1, FILTER_BITS);
|
|
|
|
out = (v16u8)__msa_pckev_b((v16i8)tmp1, (v16i8)tmp0);
|
|
out = __msa_aver_u_b(out, dst0);
|
|
|
|
ST4x4_UB(out, out, 0, 1, 2, 3, dst, dst_stride);
|
|
}
|
|
|
|
static void common_vt_2t_and_aver_dst_4x8_msa(const uint8_t *src,
|
|
int32_t src_stride, uint8_t *dst,
|
|
int32_t dst_stride,
|
|
int8_t *filter) {
|
|
v16u8 dst0, dst1, dst2, dst3, dst4, dst5, dst6, dst7;
|
|
v16i8 src0, src1, src2, src3, src4, src5, src6, src7, src8, src87_r;
|
|
v16i8 src10_r, src32_r, src54_r, src76_r, src21_r, src43_r, src65_r;
|
|
v16u8 src2110, src4332, src6554, src8776, filt0;
|
|
v8u16 tmp0, tmp1, tmp2, tmp3;
|
|
v8i16 filt;
|
|
|
|
filt = LD_SH(filter);
|
|
filt0 = (v16u8)__msa_splati_h(filt, 0);
|
|
|
|
LD_SB8(src, src_stride, src0, src1, src2, src3, src4, src5, src6, src7);
|
|
src += (8 * src_stride);
|
|
src8 = LD_SB(src);
|
|
|
|
LD_UB8(dst, dst_stride, dst0, dst1, dst2, dst3, dst4, dst5, dst6, dst7);
|
|
ILVR_W4_UB(dst1, dst0, dst3, dst2, dst5, dst4, dst7, dst6, dst0, dst1, dst2,
|
|
dst3);
|
|
ILVR_D2_UB(dst1, dst0, dst3, dst2, dst0, dst1);
|
|
ILVR_B4_SB(src1, src0, src2, src1, src3, src2, src4, src3, src10_r, src21_r,
|
|
src32_r, src43_r);
|
|
ILVR_B4_SB(src5, src4, src6, src5, src7, src6, src8, src7, src54_r, src65_r,
|
|
src76_r, src87_r);
|
|
ILVR_D4_UB(src21_r, src10_r, src43_r, src32_r, src65_r, src54_r, src87_r,
|
|
src76_r, src2110, src4332, src6554, src8776);
|
|
DOTP_UB4_UH(src2110, src4332, src6554, src8776, filt0, filt0, filt0, filt0,
|
|
tmp0, tmp1, tmp2, tmp3);
|
|
SRARI_H4_UH(tmp0, tmp1, tmp2, tmp3, FILTER_BITS);
|
|
PCKEV_B2_UB(tmp1, tmp0, tmp3, tmp2, src2110, src4332);
|
|
AVER_UB2_UB(src2110, dst0, src4332, dst1, src2110, src4332);
|
|
ST4x4_UB(src2110, src2110, 0, 1, 2, 3, dst, dst_stride);
|
|
dst += (4 * dst_stride);
|
|
ST4x4_UB(src4332, src4332, 0, 1, 2, 3, dst, dst_stride);
|
|
}
|
|
|
|
static void common_vt_2t_and_aver_dst_4w_msa(const uint8_t *src,
|
|
int32_t src_stride, uint8_t *dst,
|
|
int32_t dst_stride, int8_t *filter,
|
|
int32_t height) {
|
|
if (4 == height) {
|
|
common_vt_2t_and_aver_dst_4x4_msa(src, src_stride, dst, dst_stride, filter);
|
|
} else if (8 == height) {
|
|
common_vt_2t_and_aver_dst_4x8_msa(src, src_stride, dst, dst_stride, filter);
|
|
}
|
|
}
|
|
|
|
static void common_vt_2t_and_aver_dst_8x4_msa(const uint8_t *src,
|
|
int32_t src_stride, uint8_t *dst,
|
|
int32_t dst_stride,
|
|
int8_t *filter) {
|
|
v16u8 src0, src1, src2, src3, src4;
|
|
v16u8 dst0, dst1, dst2, dst3, vec0, vec1, vec2, vec3, filt0;
|
|
v8u16 tmp0, tmp1, tmp2, tmp3;
|
|
v8i16 filt;
|
|
|
|
/* rearranging filter_y */
|
|
filt = LD_SH(filter);
|
|
filt0 = (v16u8)__msa_splati_h(filt, 0);
|
|
|
|
LD_UB5(src, src_stride, src0, src1, src2, src3, src4);
|
|
LD_UB4(dst, dst_stride, dst0, dst1, dst2, dst3);
|
|
ILVR_B2_UB(src1, src0, src2, src1, vec0, vec1);
|
|
ILVR_B2_UB(src3, src2, src4, src3, vec2, vec3);
|
|
DOTP_UB4_UH(vec0, vec1, vec2, vec3, filt0, filt0, filt0, filt0, tmp0, tmp1,
|
|
tmp2, tmp3);
|
|
SRARI_H4_UH(tmp0, tmp1, tmp2, tmp3, FILTER_BITS);
|
|
PCKEV_AVG_ST8x4_UB(tmp0, dst0, tmp1, dst1, tmp2, dst2, tmp3, dst3, dst,
|
|
dst_stride);
|
|
}
|
|
|
|
static void common_vt_2t_and_aver_dst_8x8mult_msa(
|
|
const uint8_t *src, int32_t src_stride, uint8_t *dst, int32_t dst_stride,
|
|
int8_t *filter, int32_t height) {
|
|
uint32_t loop_cnt;
|
|
v16u8 src0, src1, src2, src3, src4, src5, src6, src7, src8;
|
|
v16u8 dst1, dst2, dst3, dst4, dst5, dst6, dst7, dst8;
|
|
v16u8 vec0, vec1, vec2, vec3, vec4, vec5, vec6, vec7, filt0;
|
|
v8u16 tmp0, tmp1, tmp2, tmp3;
|
|
v8i16 filt;
|
|
|
|
/* rearranging filter_y */
|
|
filt = LD_SH(filter);
|
|
filt0 = (v16u8)__msa_splati_h(filt, 0);
|
|
|
|
src0 = LD_UB(src);
|
|
src += src_stride;
|
|
|
|
for (loop_cnt = (height >> 3); loop_cnt--;) {
|
|
LD_UB8(src, src_stride, src1, src2, src3, src4, src5, src6, src7, src8);
|
|
src += (8 * src_stride);
|
|
LD_UB8(dst, dst_stride, dst1, dst2, dst3, dst4, dst5, dst6, dst7, dst8);
|
|
|
|
ILVR_B4_UB(src1, src0, src2, src1, src3, src2, src4, src3, vec0, vec1, vec2,
|
|
vec3);
|
|
ILVR_B4_UB(src5, src4, src6, src5, src7, src6, src8, src7, vec4, vec5, vec6,
|
|
vec7);
|
|
DOTP_UB4_UH(vec0, vec1, vec2, vec3, filt0, filt0, filt0, filt0, tmp0, tmp1,
|
|
tmp2, tmp3);
|
|
SRARI_H4_UH(tmp0, tmp1, tmp2, tmp3, FILTER_BITS);
|
|
PCKEV_AVG_ST8x4_UB(tmp0, dst1, tmp1, dst2, tmp2, dst3, tmp3, dst4, dst,
|
|
dst_stride);
|
|
dst += (4 * dst_stride);
|
|
|
|
DOTP_UB4_UH(vec4, vec5, vec6, vec7, filt0, filt0, filt0, filt0, tmp0, tmp1,
|
|
tmp2, tmp3);
|
|
SRARI_H4_UH(tmp0, tmp1, tmp2, tmp3, FILTER_BITS);
|
|
PCKEV_AVG_ST8x4_UB(tmp0, dst5, tmp1, dst6, tmp2, dst7, tmp3, dst8, dst,
|
|
dst_stride);
|
|
dst += (4 * dst_stride);
|
|
|
|
src0 = src8;
|
|
}
|
|
}
|
|
|
|
static void common_vt_2t_and_aver_dst_8w_msa(const uint8_t *src,
|
|
int32_t src_stride, uint8_t *dst,
|
|
int32_t dst_stride, int8_t *filter,
|
|
int32_t height) {
|
|
if (4 == height) {
|
|
common_vt_2t_and_aver_dst_8x4_msa(src, src_stride, dst, dst_stride, filter);
|
|
} else {
|
|
common_vt_2t_and_aver_dst_8x8mult_msa(src, src_stride, dst, dst_stride,
|
|
filter, height);
|
|
}
|
|
}
|
|
|
|
static void common_vt_2t_and_aver_dst_16w_msa(const uint8_t *src,
|
|
int32_t src_stride, uint8_t *dst,
|
|
int32_t dst_stride,
|
|
int8_t *filter, int32_t height) {
|
|
uint32_t loop_cnt;
|
|
v16u8 src0, src1, src2, src3, src4, dst0, dst1, dst2, dst3, filt0;
|
|
v16u8 vec0, vec1, vec2, vec3, vec4, vec5, vec6, vec7;
|
|
v8u16 tmp0, tmp1, tmp2, tmp3, filt;
|
|
|
|
/* rearranging filter_y */
|
|
filt = LD_UH(filter);
|
|
filt0 = (v16u8)__msa_splati_h((v8i16)filt, 0);
|
|
|
|
src0 = LD_UB(src);
|
|
src += src_stride;
|
|
|
|
for (loop_cnt = (height >> 2); loop_cnt--;) {
|
|
LD_UB4(src, src_stride, src1, src2, src3, src4);
|
|
src += (4 * src_stride);
|
|
|
|
LD_UB4(dst, dst_stride, dst0, dst1, dst2, dst3);
|
|
ILVR_B2_UB(src1, src0, src2, src1, vec0, vec2);
|
|
ILVL_B2_UB(src1, src0, src2, src1, vec1, vec3);
|
|
DOTP_UB2_UH(vec0, vec1, filt0, filt0, tmp0, tmp1);
|
|
SRARI_H2_UH(tmp0, tmp1, FILTER_BITS);
|
|
PCKEV_AVG_ST_UB(tmp1, tmp0, dst0, dst);
|
|
dst += dst_stride;
|
|
|
|
ILVR_B2_UB(src3, src2, src4, src3, vec4, vec6);
|
|
ILVL_B2_UB(src3, src2, src4, src3, vec5, vec7);
|
|
DOTP_UB2_UH(vec2, vec3, filt0, filt0, tmp2, tmp3);
|
|
SRARI_H2_UH(tmp2, tmp3, FILTER_BITS);
|
|
PCKEV_AVG_ST_UB(tmp3, tmp2, dst1, dst);
|
|
dst += dst_stride;
|
|
|
|
DOTP_UB2_UH(vec4, vec5, filt0, filt0, tmp0, tmp1);
|
|
SRARI_H2_UH(tmp0, tmp1, FILTER_BITS);
|
|
PCKEV_AVG_ST_UB(tmp1, tmp0, dst2, dst);
|
|
dst += dst_stride;
|
|
|
|
DOTP_UB2_UH(vec6, vec7, filt0, filt0, tmp2, tmp3);
|
|
SRARI_H2_UH(tmp2, tmp3, FILTER_BITS);
|
|
PCKEV_AVG_ST_UB(tmp3, tmp2, dst3, dst);
|
|
dst += dst_stride;
|
|
|
|
src0 = src4;
|
|
}
|
|
}
|
|
|
|
static void common_vt_2t_and_aver_dst_32w_msa(const uint8_t *src,
|
|
int32_t src_stride, uint8_t *dst,
|
|
int32_t dst_stride,
|
|
int8_t *filter, int32_t height) {
|
|
uint32_t loop_cnt;
|
|
v16u8 src0, src1, src2, src3, src4, src5, src6, src7, src8, src9;
|
|
v16u8 dst0, dst1, dst2, dst3, dst4, dst5, dst6, dst7;
|
|
v16u8 vec0, vec1, vec2, vec3, vec4, vec5, vec6, vec7, filt0;
|
|
v8u16 tmp0, tmp1, tmp2, tmp3, filt;
|
|
|
|
/* rearranging filter_y */
|
|
filt = LD_UH(filter);
|
|
filt0 = (v16u8)__msa_splati_h((v8i16)filt, 0);
|
|
|
|
LD_UB2(src, 16, src0, src5);
|
|
src += src_stride;
|
|
|
|
for (loop_cnt = (height >> 2); loop_cnt--;) {
|
|
LD_UB4(src, src_stride, src1, src2, src3, src4);
|
|
LD_UB4(dst, dst_stride, dst0, dst1, dst2, dst3);
|
|
ILVR_B2_UB(src1, src0, src2, src1, vec0, vec2);
|
|
ILVL_B2_UB(src1, src0, src2, src1, vec1, vec3);
|
|
|
|
LD_UB4(src + 16, src_stride, src6, src7, src8, src9);
|
|
LD_UB4(dst + 16, dst_stride, dst4, dst5, dst6, dst7);
|
|
src += (4 * src_stride);
|
|
|
|
DOTP_UB2_UH(vec0, vec1, filt0, filt0, tmp0, tmp1);
|
|
SRARI_H2_UH(tmp0, tmp1, FILTER_BITS);
|
|
PCKEV_AVG_ST_UB(tmp1, tmp0, dst0, dst);
|
|
|
|
DOTP_UB2_UH(vec2, vec3, filt0, filt0, tmp2, tmp3);
|
|
SRARI_H2_UH(tmp2, tmp3, FILTER_BITS);
|
|
PCKEV_AVG_ST_UB(tmp3, tmp2, dst1, dst + dst_stride);
|
|
|
|
ILVR_B2_UB(src3, src2, src4, src3, vec4, vec6);
|
|
ILVL_B2_UB(src3, src2, src4, src3, vec5, vec7);
|
|
DOTP_UB2_UH(vec4, vec5, filt0, filt0, tmp0, tmp1);
|
|
SRARI_H2_UH(tmp0, tmp1, FILTER_BITS);
|
|
PCKEV_AVG_ST_UB(tmp1, tmp0, dst2, dst + 2 * dst_stride);
|
|
|
|
DOTP_UB2_UH(vec6, vec7, filt0, filt0, tmp2, tmp3);
|
|
SRARI_H2_UH(tmp2, tmp3, FILTER_BITS);
|
|
PCKEV_AVG_ST_UB(tmp3, tmp2, dst3, dst + 3 * dst_stride);
|
|
|
|
ILVR_B2_UB(src6, src5, src7, src6, vec0, vec2);
|
|
ILVL_B2_UB(src6, src5, src7, src6, vec1, vec3);
|
|
DOTP_UB2_UH(vec0, vec1, filt0, filt0, tmp0, tmp1);
|
|
SRARI_H2_UH(tmp0, tmp1, FILTER_BITS);
|
|
PCKEV_AVG_ST_UB(tmp1, tmp0, dst4, dst + 16);
|
|
|
|
DOTP_UB2_UH(vec2, vec3, filt0, filt0, tmp2, tmp3);
|
|
SRARI_H2_UH(tmp2, tmp3, FILTER_BITS);
|
|
PCKEV_AVG_ST_UB(tmp3, tmp2, dst5, dst + 16 + dst_stride);
|
|
|
|
ILVR_B2_UB(src8, src7, src9, src8, vec4, vec6);
|
|
ILVL_B2_UB(src8, src7, src9, src8, vec5, vec7);
|
|
DOTP_UB2_UH(vec4, vec5, filt0, filt0, tmp0, tmp1);
|
|
SRARI_H2_UH(tmp0, tmp1, FILTER_BITS);
|
|
PCKEV_AVG_ST_UB(tmp1, tmp0, dst6, dst + 16 + 2 * dst_stride);
|
|
|
|
DOTP_UB2_UH(vec6, vec7, filt0, filt0, tmp2, tmp3);
|
|
SRARI_H2_UH(tmp2, tmp3, FILTER_BITS);
|
|
PCKEV_AVG_ST_UB(tmp3, tmp2, dst7, dst + 16 + 3 * dst_stride);
|
|
dst += (4 * dst_stride);
|
|
|
|
src0 = src4;
|
|
src5 = src9;
|
|
}
|
|
}
|
|
|
|
static void common_vt_2t_and_aver_dst_64w_msa(const uint8_t *src,
|
|
int32_t src_stride, uint8_t *dst,
|
|
int32_t dst_stride,
|
|
int8_t *filter, int32_t height) {
|
|
uint32_t loop_cnt;
|
|
v16u8 src0, src1, src2, src3, src4, src5;
|
|
v16u8 src6, src7, src8, src9, src10, src11, filt0;
|
|
v16u8 dst0, dst1, dst2, dst3, dst4, dst5, dst6, dst7;
|
|
v16u8 vec0, vec1, vec2, vec3, vec4, vec5, vec6, vec7;
|
|
v8u16 tmp0, tmp1, tmp2, tmp3, tmp4, tmp5, tmp6, tmp7;
|
|
v8u16 filt;
|
|
|
|
/* rearranging filter_y */
|
|
filt = LD_UH(filter);
|
|
filt0 = (v16u8)__msa_splati_h((v8i16)filt, 0);
|
|
|
|
LD_UB4(src, 16, src0, src3, src6, src9);
|
|
src += src_stride;
|
|
|
|
for (loop_cnt = (height >> 1); loop_cnt--;) {
|
|
LD_UB2(src, src_stride, src1, src2);
|
|
LD_UB2(dst, dst_stride, dst0, dst1);
|
|
LD_UB2(src + 16, src_stride, src4, src5);
|
|
LD_UB2(dst + 16, dst_stride, dst2, dst3);
|
|
LD_UB2(src + 32, src_stride, src7, src8);
|
|
LD_UB2(dst + 32, dst_stride, dst4, dst5);
|
|
LD_UB2(src + 48, src_stride, src10, src11);
|
|
LD_UB2(dst + 48, dst_stride, dst6, dst7);
|
|
src += (2 * src_stride);
|
|
|
|
ILVR_B2_UB(src1, src0, src2, src1, vec0, vec2);
|
|
ILVL_B2_UB(src1, src0, src2, src1, vec1, vec3);
|
|
DOTP_UB2_UH(vec0, vec1, filt0, filt0, tmp0, tmp1);
|
|
SRARI_H2_UH(tmp0, tmp1, FILTER_BITS);
|
|
PCKEV_AVG_ST_UB(tmp1, tmp0, dst0, dst);
|
|
|
|
DOTP_UB2_UH(vec2, vec3, filt0, filt0, tmp2, tmp3);
|
|
SRARI_H2_UH(tmp2, tmp3, FILTER_BITS);
|
|
PCKEV_AVG_ST_UB(tmp3, tmp2, dst1, dst + dst_stride);
|
|
|
|
ILVR_B2_UB(src4, src3, src5, src4, vec4, vec6);
|
|
ILVL_B2_UB(src4, src3, src5, src4, vec5, vec7);
|
|
DOTP_UB2_UH(vec4, vec5, filt0, filt0, tmp4, tmp5);
|
|
SRARI_H2_UH(tmp4, tmp5, FILTER_BITS);
|
|
PCKEV_AVG_ST_UB(tmp5, tmp4, dst2, dst + 16);
|
|
|
|
DOTP_UB2_UH(vec6, vec7, filt0, filt0, tmp6, tmp7);
|
|
SRARI_H2_UH(tmp6, tmp7, FILTER_BITS);
|
|
PCKEV_AVG_ST_UB(tmp7, tmp6, dst3, dst + 16 + dst_stride);
|
|
|
|
ILVR_B2_UB(src7, src6, src8, src7, vec0, vec2);
|
|
ILVL_B2_UB(src7, src6, src8, src7, vec1, vec3);
|
|
DOTP_UB2_UH(vec0, vec1, filt0, filt0, tmp0, tmp1);
|
|
SRARI_H2_UH(tmp0, tmp1, FILTER_BITS);
|
|
PCKEV_AVG_ST_UB(tmp1, tmp0, dst4, dst + 32);
|
|
|
|
DOTP_UB2_UH(vec2, vec3, filt0, filt0, tmp2, tmp3);
|
|
SRARI_H2_UH(tmp2, tmp3, FILTER_BITS);
|
|
PCKEV_AVG_ST_UB(tmp3, tmp2, dst5, dst + 32 + dst_stride);
|
|
|
|
ILVR_B2_UB(src10, src9, src11, src10, vec4, vec6);
|
|
ILVL_B2_UB(src10, src9, src11, src10, vec5, vec7);
|
|
DOTP_UB2_UH(vec4, vec5, filt0, filt0, tmp4, tmp5);
|
|
SRARI_H2_UH(tmp4, tmp5, FILTER_BITS);
|
|
PCKEV_AVG_ST_UB(tmp5, tmp4, dst6, (dst + 48));
|
|
|
|
DOTP_UB2_UH(vec6, vec7, filt0, filt0, tmp6, tmp7);
|
|
SRARI_H2_UH(tmp6, tmp7, FILTER_BITS);
|
|
PCKEV_AVG_ST_UB(tmp7, tmp6, dst7, dst + 48 + dst_stride);
|
|
dst += (2 * dst_stride);
|
|
|
|
src0 = src2;
|
|
src3 = src5;
|
|
src6 = src8;
|
|
src9 = src11;
|
|
}
|
|
}
|
|
|
|
void aom_convolve8_avg_vert_msa(const uint8_t *src, ptrdiff_t src_stride,
|
|
uint8_t *dst, ptrdiff_t dst_stride,
|
|
const int16_t *filter_x, int x_step_q4,
|
|
const int16_t *filter_y, int y_step_q4, int w,
|
|
int h) {
|
|
int8_t cnt, filt_ver[8];
|
|
|
|
assert(y_step_q4 == 16);
|
|
assert(((const int32_t *)filter_y)[1] != 0x800000);
|
|
|
|
for (cnt = 0; cnt < 8; ++cnt) {
|
|
filt_ver[cnt] = filter_y[cnt];
|
|
}
|
|
|
|
if (((const int32_t *)filter_y)[0] == 0) {
|
|
switch (w) {
|
|
case 4:
|
|
common_vt_2t_and_aver_dst_4w_msa(src, (int32_t)src_stride, dst,
|
|
(int32_t)dst_stride, &filt_ver[3], h);
|
|
break;
|
|
case 8:
|
|
common_vt_2t_and_aver_dst_8w_msa(src, (int32_t)src_stride, dst,
|
|
(int32_t)dst_stride, &filt_ver[3], h);
|
|
break;
|
|
case 16:
|
|
common_vt_2t_and_aver_dst_16w_msa(src, (int32_t)src_stride, dst,
|
|
(int32_t)dst_stride, &filt_ver[3], h);
|
|
break;
|
|
case 32:
|
|
common_vt_2t_and_aver_dst_32w_msa(src, (int32_t)src_stride, dst,
|
|
(int32_t)dst_stride, &filt_ver[3], h);
|
|
break;
|
|
case 64:
|
|
common_vt_2t_and_aver_dst_64w_msa(src, (int32_t)src_stride, dst,
|
|
(int32_t)dst_stride, &filt_ver[3], h);
|
|
break;
|
|
default:
|
|
aom_convolve8_avg_vert_c(src, src_stride, dst, dst_stride, filter_x,
|
|
x_step_q4, filter_y, y_step_q4, w, h);
|
|
break;
|
|
}
|
|
} else {
|
|
switch (w) {
|
|
case 4:
|
|
common_vt_8t_and_aver_dst_4w_msa(src, (int32_t)src_stride, dst,
|
|
(int32_t)dst_stride, filt_ver, h);
|
|
break;
|
|
case 8:
|
|
common_vt_8t_and_aver_dst_8w_msa(src, (int32_t)src_stride, dst,
|
|
(int32_t)dst_stride, filt_ver, h);
|
|
break;
|
|
case 16:
|
|
common_vt_8t_and_aver_dst_16w_msa(src, (int32_t)src_stride, dst,
|
|
(int32_t)dst_stride, filt_ver, h);
|
|
|
|
break;
|
|
case 32:
|
|
common_vt_8t_and_aver_dst_32w_msa(src, (int32_t)src_stride, dst,
|
|
(int32_t)dst_stride, filt_ver, h);
|
|
break;
|
|
case 64:
|
|
common_vt_8t_and_aver_dst_64w_msa(src, (int32_t)src_stride, dst,
|
|
(int32_t)dst_stride, filt_ver, h);
|
|
break;
|
|
default:
|
|
aom_convolve8_avg_vert_c(src, src_stride, dst, dst_stride, filter_x,
|
|
x_step_q4, filter_y, y_step_q4, w, h);
|
|
break;
|
|
}
|
|
}
|
|
}
|