| /* |
| * Copyright (c) 2022 The WebM project authors. All Rights Reserved. |
| * |
| * Use of this source code is governed by a BSD-style license |
| * that can be found in the LICENSE file in the root of the source |
| * tree. An additional intellectual property rights grant can be found |
| * in the file PATENTS. All contributing project authors may |
| * be found in the AUTHORS file in the root of the source tree. |
| */ |
| |
| #include <assert.h> |
| #include "./vpx_dsp_rtcd.h" |
| #include "vpx_dsp/loongarch/vpx_convolve_lsx.h" |
| |
| static void common_vt_8t_4w_lsx(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 = height >> 2; |
| int32_t src_stride2 = src_stride << 1; |
| int32_t src_stride3 = src_stride + src_stride2; |
| int32_t src_stride4 = src_stride2 << 1; |
| __m128i src0, src1, src2, src3, src4, src5, src6, src7, src8, src9, src10; |
| __m128i tmp0, tmp1, tmp2, tmp3, tmp4, tmp5; |
| __m128i reg0, reg1, reg2, reg3, reg4; |
| __m128i filter0, filter1, filter2, filter3; |
| __m128i out0, out1; |
| uint8_t *_src = (uint8_t *)src - src_stride3; |
| |
| DUP4_ARG2(__lsx_vldrepl_h, filter, 0, filter, 2, filter, 4, filter, 6, |
| filter0, filter1, filter2, filter3); |
| src0 = __lsx_vld(_src, 0); |
| DUP2_ARG2(__lsx_vldx, _src, src_stride, _src, src_stride2, src1, src2); |
| src3 = __lsx_vldx(_src, src_stride3); |
| _src += src_stride4; |
| src4 = __lsx_vld(_src, 0); |
| DUP2_ARG2(__lsx_vldx, _src, src_stride, _src, src_stride2, src5, src6); |
| _src += src_stride3; |
| DUP4_ARG2(__lsx_vilvl_b, src1, src0, src3, src2, src5, src4, src2, src1, tmp0, |
| tmp1, tmp2, tmp3); |
| DUP2_ARG2(__lsx_vilvl_b, src4, src3, src6, src5, tmp4, tmp5); |
| DUP2_ARG2(__lsx_vilvl_d, tmp3, tmp0, tmp4, tmp1, reg0, reg1); |
| reg2 = __lsx_vilvl_d(tmp5, tmp2); |
| DUP2_ARG2(__lsx_vxori_b, reg0, 128, reg1, 128, reg0, reg1); |
| reg2 = __lsx_vxori_b(reg2, 128); |
| |
| for (; loop_cnt--;) { |
| src7 = __lsx_vld(_src, 0); |
| DUP2_ARG2(__lsx_vldx, _src, src_stride, _src, src_stride2, src8, src9); |
| src10 = __lsx_vldx(_src, src_stride3); |
| _src += src_stride4; |
| DUP4_ARG2(__lsx_vilvl_b, src7, src6, src8, src7, src9, src8, src10, src9, |
| tmp0, tmp1, tmp2, tmp3); |
| DUP2_ARG2(__lsx_vilvl_d, tmp1, tmp0, tmp3, tmp2, reg3, reg4); |
| DUP2_ARG2(__lsx_vxori_b, reg3, 128, reg4, 128, reg3, reg4); |
| out0 = FILT_8TAP_DPADD_S_H(reg0, reg1, reg2, reg3, filter0, filter1, |
| filter2, filter3); |
| out1 = FILT_8TAP_DPADD_S_H(reg1, reg2, reg3, reg4, filter0, filter1, |
| filter2, filter3); |
| out0 = __lsx_vssrarni_b_h(out1, out0, 7); |
| out0 = __lsx_vxori_b(out0, 128); |
| __lsx_vstelm_w(out0, dst, 0, 0); |
| dst += dst_stride; |
| __lsx_vstelm_w(out0, dst, 0, 1); |
| dst += dst_stride; |
| __lsx_vstelm_w(out0, dst, 0, 2); |
| dst += dst_stride; |
| __lsx_vstelm_w(out0, dst, 0, 3); |
| dst += dst_stride; |
| |
| reg0 = reg2; |
| reg1 = reg3; |
| reg2 = reg4; |
| src6 = src10; |
| } |
| } |
| |
| static void common_vt_8t_8w_lsx(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 = height >> 2; |
| __m128i src0, src1, src2, src3, src4, src5, src6, src7, src8, src9, src10; |
| __m128i tmp0, tmp1, tmp2, tmp3; |
| __m128i reg0, reg1, reg2, reg3, reg4, reg5; |
| __m128i filter0, filter1, filter2, filter3; |
| __m128i out0, out1, out2, out3; |
| int32_t src_stride2 = src_stride << 1; |
| int32_t src_stride3 = src_stride + src_stride2; |
| int32_t src_stride4 = src_stride2 << 1; |
| src = src - src_stride3; |
| |
| DUP4_ARG2(__lsx_vldrepl_h, filter, 0, filter, 2, filter, 4, filter, 6, |
| filter0, filter1, filter2, filter3); |
| |
| src0 = __lsx_vld(src, 0); |
| DUP2_ARG2(__lsx_vldx, src, src_stride, src, src_stride2, src1, src2); |
| src3 = __lsx_vldx(src, src_stride3); |
| src += src_stride4; |
| src4 = __lsx_vld(src, 0); |
| DUP2_ARG2(__lsx_vldx, src, src_stride, src, src_stride2, src5, src6); |
| src += src_stride3; |
| |
| DUP4_ARG2(__lsx_vxori_b, src0, 128, src1, 128, src2, 128, src3, 128, src0, |
| src1, src2, src3); |
| DUP2_ARG2(__lsx_vxori_b, src4, 128, src5, 128, src4, src5); |
| src6 = __lsx_vxori_b(src6, 128); |
| DUP4_ARG2(__lsx_vilvl_b, src1, src0, src3, src2, src5, src4, src2, src1, reg0, |
| reg1, reg2, reg3); |
| DUP2_ARG2(__lsx_vilvl_b, src4, src3, src6, src5, reg4, reg5); |
| |
| for (; loop_cnt--;) { |
| src7 = __lsx_vld(src, 0); |
| DUP2_ARG2(__lsx_vldx, src, src_stride, src, src_stride2, src8, src9); |
| src10 = __lsx_vldx(src, src_stride3); |
| src += src_stride4; |
| DUP4_ARG2(__lsx_vxori_b, src7, 128, src8, 128, src9, 128, src10, 128, src7, |
| src8, src9, src10); |
| DUP4_ARG2(__lsx_vilvl_b, src7, src6, src8, src7, src9, src8, src10, src9, |
| tmp0, tmp1, tmp2, tmp3); |
| out0 = FILT_8TAP_DPADD_S_H(reg0, reg1, reg2, tmp0, filter0, filter1, |
| filter2, filter3); |
| out1 = FILT_8TAP_DPADD_S_H(reg3, reg4, reg5, tmp1, filter0, filter1, |
| filter2, filter3); |
| out2 = FILT_8TAP_DPADD_S_H(reg1, reg2, tmp0, tmp2, filter0, filter1, |
| filter2, filter3); |
| out3 = FILT_8TAP_DPADD_S_H(reg4, reg5, tmp1, tmp3, filter0, filter1, |
| filter2, filter3); |
| DUP2_ARG3(__lsx_vssrarni_b_h, out1, out0, 7, out3, out2, 7, out0, out1); |
| DUP2_ARG2(__lsx_vxori_b, out0, 128, out1, 128, out0, out1); |
| __lsx_vstelm_d(out0, dst, 0, 0); |
| dst += dst_stride; |
| __lsx_vstelm_d(out0, dst, 0, 1); |
| dst += dst_stride; |
| __lsx_vstelm_d(out1, dst, 0, 0); |
| dst += dst_stride; |
| __lsx_vstelm_d(out1, dst, 0, 1); |
| dst += dst_stride; |
| |
| reg0 = reg2; |
| reg1 = tmp0; |
| reg2 = tmp2; |
| reg3 = reg5; |
| reg4 = tmp1; |
| reg5 = tmp3; |
| src6 = src10; |
| } |
| } |
| |
| static void common_vt_8t_16w_lsx(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 = height >> 2; |
| __m128i src0, src1, src2, src3, src4, src5, src6, src7, src8, src9, src10; |
| __m128i filter0, filter1, filter2, filter3; |
| __m128i reg0, reg1, reg2, reg3, reg4, reg5; |
| __m128i reg6, reg7, reg8, reg9, reg10, reg11; |
| __m128i tmp0, tmp1, tmp2, tmp3; |
| int32_t src_stride2 = src_stride << 1; |
| int32_t src_stride3 = src_stride + src_stride2; |
| int32_t src_stride4 = src_stride2 << 1; |
| // uint8_t *_src = (uint8_t *)src - src_stride3; |
| src -= src_stride3; |
| DUP4_ARG2(__lsx_vldrepl_h, filter, 0, filter, 2, filter, 4, filter, 6, |
| filter0, filter1, filter2, filter3); |
| |
| src0 = __lsx_vld(src, 0); |
| DUP4_ARG2(__lsx_vldx, src, src_stride, src, src_stride2, src, src_stride3, |
| src, src_stride4, src1, src2, src3, src4); |
| src += src_stride4; |
| DUP2_ARG2(__lsx_vldx, src, src_stride, src, src_stride2, src5, src6); |
| src += src_stride3; |
| |
| DUP4_ARG2(__lsx_vxori_b, src0, 128, src1, 128, src2, 128, src3, 128, src0, |
| src1, src2, src3); |
| DUP2_ARG2(__lsx_vxori_b, src4, 128, src5, 128, src4, src5); |
| src6 = __lsx_vxori_b(src6, 128); |
| DUP4_ARG2(__lsx_vilvl_b, src1, src0, src3, src2, src5, src4, src2, src1, reg0, |
| reg1, reg2, reg3); |
| DUP2_ARG2(__lsx_vilvl_b, src4, src3, src6, src5, reg4, reg5); |
| DUP4_ARG2(__lsx_vilvh_b, src1, src0, src3, src2, src5, src4, src2, src1, reg6, |
| reg7, reg8, reg9); |
| DUP2_ARG2(__lsx_vilvh_b, src4, src3, src6, src5, reg10, reg11); |
| |
| for (; loop_cnt--;) { |
| src7 = __lsx_vld(src, 0); |
| DUP2_ARG2(__lsx_vldx, src, src_stride, src, src_stride2, src8, src9); |
| src10 = __lsx_vldx(src, src_stride3); |
| src += src_stride4; |
| |
| DUP4_ARG2(__lsx_vxori_b, src7, 128, src8, 128, src9, 128, src10, 128, src7, |
| src8, src9, src10); |
| DUP4_ARG2(__lsx_vilvl_b, src7, src6, src8, src7, src9, src8, src10, src9, |
| src0, src1, src2, src3); |
| DUP4_ARG2(__lsx_vilvh_b, src7, src6, src8, src7, src9, src8, src10, src9, |
| src4, src5, src7, src8); |
| tmp0 = FILT_8TAP_DPADD_S_H(reg0, reg1, reg2, src0, filter0, filter1, |
| filter2, filter3); |
| tmp1 = FILT_8TAP_DPADD_S_H(reg3, reg4, reg5, src1, filter0, filter1, |
| filter2, filter3); |
| tmp2 = FILT_8TAP_DPADD_S_H(reg6, reg7, reg8, src4, filter0, filter1, |
| filter2, filter3); |
| tmp3 = FILT_8TAP_DPADD_S_H(reg9, reg10, reg11, src5, filter0, filter1, |
| filter2, filter3); |
| DUP2_ARG3(__lsx_vssrarni_b_h, tmp2, tmp0, 7, tmp3, tmp1, 7, tmp0, tmp1); |
| DUP2_ARG2(__lsx_vxori_b, tmp0, 128, tmp1, 128, tmp0, tmp1); |
| __lsx_vst(tmp0, dst, 0); |
| dst += dst_stride; |
| __lsx_vst(tmp1, dst, 0); |
| dst += dst_stride; |
| tmp0 = FILT_8TAP_DPADD_S_H(reg1, reg2, src0, src2, filter0, filter1, |
| filter2, filter3); |
| tmp1 = FILT_8TAP_DPADD_S_H(reg4, reg5, src1, src3, filter0, filter1, |
| filter2, filter3); |
| tmp2 = FILT_8TAP_DPADD_S_H(reg7, reg8, src4, src7, filter0, filter1, |
| filter2, filter3); |
| tmp3 = FILT_8TAP_DPADD_S_H(reg10, reg11, src5, src8, filter0, filter1, |
| filter2, filter3); |
| DUP2_ARG3(__lsx_vssrarni_b_h, tmp2, tmp0, 7, tmp3, tmp1, 7, tmp0, tmp1); |
| DUP2_ARG2(__lsx_vxori_b, tmp0, 128, tmp1, 128, tmp0, tmp1); |
| __lsx_vst(tmp0, dst, 0); |
| dst += dst_stride; |
| __lsx_vst(tmp1, dst, 0); |
| dst += dst_stride; |
| |
| reg0 = reg2; |
| reg1 = src0; |
| reg2 = src2; |
| reg3 = reg5; |
| reg4 = src1; |
| reg5 = src3; |
| reg6 = reg8; |
| reg7 = src4; |
| reg8 = src7; |
| reg9 = reg11; |
| reg10 = src5; |
| reg11 = src8; |
| src6 = src10; |
| } |
| } |
| |
| static void common_vt_8t_16w_mult_lsx(const uint8_t *src, int32_t src_stride, |
| uint8_t *dst, int32_t dst_stride, |
| int8_t *filter, int32_t height, |
| int32_t width) { |
| uint8_t *src_tmp; |
| uint8_t *dst_tmp; |
| uint32_t cnt = width >> 4; |
| __m128i src0, src1, src2, src3, src4, src5, src6, src7, src8, src9, src10; |
| __m128i filter0, filter1, filter2, filter3; |
| __m128i reg0, reg1, reg2, reg3, reg4, reg5; |
| __m128i reg6, reg7, reg8, reg9, reg10, reg11; |
| __m128i tmp0, tmp1, tmp2, tmp3; |
| int32_t src_stride2 = src_stride << 1; |
| int32_t src_stride3 = src_stride + src_stride2; |
| int32_t src_stride4 = src_stride2 << 1; |
| int32_t dst_stride2 = dst_stride << 1; |
| int32_t dst_stride3 = dst_stride2 + dst_stride; |
| int32_t dst_stride4 = dst_stride2 << 1; |
| src -= src_stride3; |
| DUP4_ARG2(__lsx_vldrepl_h, filter, 0, filter, 2, filter, 4, filter, 6, |
| filter0, filter1, filter2, filter3); |
| |
| for (; cnt--;) { |
| uint32_t loop_cnt = height >> 2; |
| |
| src_tmp = src; |
| dst_tmp = dst; |
| |
| src0 = __lsx_vld(src_tmp, 0); |
| DUP2_ARG2(__lsx_vldx, src_tmp, src_stride, src_tmp, src_stride2, src1, |
| src2); |
| src3 = __lsx_vldx(src_tmp, src_stride3); |
| src_tmp += src_stride4; |
| src4 = __lsx_vld(src_tmp, 0); |
| DUP2_ARG2(__lsx_vldx, src_tmp, src_stride, src_tmp, src_stride2, src5, |
| src6); |
| src_tmp += src_stride3; |
| |
| DUP4_ARG2(__lsx_vxori_b, src0, 128, src1, 128, src2, 128, src3, 128, src0, |
| src1, src2, src3); |
| DUP2_ARG2(__lsx_vxori_b, src4, 128, src5, 128, src4, src5); |
| src6 = __lsx_vxori_b(src6, 128); |
| DUP4_ARG2(__lsx_vilvl_b, src1, src0, src3, src2, src5, src4, src2, src1, |
| reg0, reg1, reg2, reg3); |
| DUP2_ARG2(__lsx_vilvl_b, src4, src3, src6, src5, reg4, reg5); |
| DUP4_ARG2(__lsx_vilvh_b, src1, src0, src3, src2, src5, src4, src2, src1, |
| reg6, reg7, reg8, reg9); |
| DUP2_ARG2(__lsx_vilvh_b, src4, src3, src6, src5, reg10, reg11); |
| |
| for (; loop_cnt--;) { |
| src7 = __lsx_vld(src_tmp, 0); |
| DUP2_ARG2(__lsx_vldx, src_tmp, src_stride, src_tmp, src_stride2, src8, |
| src9); |
| src10 = __lsx_vldx(src_tmp, src_stride3); |
| src_tmp += src_stride4; |
| DUP4_ARG2(__lsx_vxori_b, src7, 128, src8, 128, src9, 128, src10, 128, |
| src7, src8, src9, src10); |
| DUP4_ARG2(__lsx_vilvl_b, src7, src6, src8, src7, src9, src8, src10, src9, |
| src0, src1, src2, src3); |
| DUP4_ARG2(__lsx_vilvh_b, src7, src6, src8, src7, src9, src8, src10, src9, |
| src4, src5, src7, src8); |
| tmp0 = FILT_8TAP_DPADD_S_H(reg0, reg1, reg2, src0, filter0, filter1, |
| filter2, filter3); |
| tmp1 = FILT_8TAP_DPADD_S_H(reg3, reg4, reg5, src1, filter0, filter1, |
| filter2, filter3); |
| tmp2 = FILT_8TAP_DPADD_S_H(reg6, reg7, reg8, src4, filter0, filter1, |
| filter2, filter3); |
| tmp3 = FILT_8TAP_DPADD_S_H(reg9, reg10, reg11, src5, filter0, filter1, |
| filter2, filter3); |
| DUP2_ARG3(__lsx_vssrarni_b_h, tmp2, tmp0, 7, tmp3, tmp1, 7, tmp0, tmp1); |
| DUP2_ARG2(__lsx_vxori_b, tmp0, 128, tmp1, 128, tmp0, tmp1); |
| __lsx_vst(tmp0, dst_tmp, 0); |
| __lsx_vstx(tmp1, dst_tmp, dst_stride); |
| tmp0 = FILT_8TAP_DPADD_S_H(reg1, reg2, src0, src2, filter0, filter1, |
| filter2, filter3); |
| tmp1 = FILT_8TAP_DPADD_S_H(reg4, reg5, src1, src3, filter0, filter1, |
| filter2, filter3); |
| tmp2 = FILT_8TAP_DPADD_S_H(reg7, reg8, src4, src7, filter0, filter1, |
| filter2, filter3); |
| tmp3 = FILT_8TAP_DPADD_S_H(reg10, reg11, src5, src8, filter0, filter1, |
| filter2, filter3); |
| DUP2_ARG3(__lsx_vssrarni_b_h, tmp2, tmp0, 7, tmp3, tmp1, 7, tmp0, tmp1); |
| DUP2_ARG2(__lsx_vxori_b, tmp0, 128, tmp1, 128, tmp0, tmp1); |
| __lsx_vstx(tmp0, dst_tmp, dst_stride2); |
| __lsx_vstx(tmp1, dst_tmp, dst_stride3); |
| dst_tmp += dst_stride4; |
| |
| reg0 = reg2; |
| reg1 = src0; |
| reg2 = src2; |
| reg3 = reg5; |
| reg4 = src1; |
| reg5 = src3; |
| reg6 = reg8; |
| reg7 = src4; |
| reg8 = src7; |
| reg9 = reg11; |
| reg10 = src5; |
| reg11 = src8; |
| src6 = src10; |
| } |
| src += 16; |
| dst += 16; |
| } |
| } |
| |
| static void common_vt_8t_32w_lsx(const uint8_t *src, int32_t src_stride, |
| uint8_t *dst, int32_t dst_stride, |
| int8_t *filter, int32_t height) { |
| common_vt_8t_16w_mult_lsx(src, src_stride, dst, dst_stride, filter, height, |
| 32); |
| } |
| |
| static void common_vt_8t_64w_lsx(const uint8_t *src, int32_t src_stride, |
| uint8_t *dst, int32_t dst_stride, |
| int8_t *filter, int32_t height) { |
| common_vt_8t_16w_mult_lsx(src, src_stride, dst, dst_stride, filter, height, |
| 64); |
| } |
| |
| static void common_vt_2t_4x4_lsx(const uint8_t *src, int32_t src_stride, |
| uint8_t *dst, int32_t dst_stride, |
| int8_t *filter) { |
| __m128i src0, src1, src2, src3, src4; |
| __m128i vec0, vec1, vec2, vec3, vec4, vec5; |
| __m128i filt0, tmp0, tmp1; |
| |
| int32_t src_stride2 = src_stride << 1; |
| int32_t src_stride3 = src_stride2 + src_stride; |
| int32_t src_stride4 = src_stride2 << 1; |
| int32_t dst_stride2 = dst_stride << 1; |
| int32_t dst_stride3 = dst_stride2 + dst_stride; |
| |
| filt0 = __lsx_vldrepl_h(filter, 0); |
| |
| src0 = __lsx_vld(src, 0); |
| DUP4_ARG2(__lsx_vldx, src, src_stride, src, src_stride2, src, src_stride3, |
| src, src_stride4, src1, src2, src3, src4); |
| src += (src_stride4 + src_stride); |
| |
| DUP4_ARG2(__lsx_vilvl_b, src1, src0, src2, src1, src3, src2, src4, src3, vec0, |
| vec1, vec2, vec3); |
| DUP2_ARG2(__lsx_vilvl_d, vec1, vec0, vec3, vec2, vec4, vec5); |
| DUP2_ARG2(__lsx_vdp2_h_bu, vec4, filt0, vec5, filt0, tmp0, tmp1); |
| tmp0 = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS); |
| |
| __lsx_vstelm_w(tmp0, dst, 0, 0); |
| __lsx_vstelm_w(tmp0, dst + dst_stride, 0, 1); |
| __lsx_vstelm_w(tmp0, dst + dst_stride2, 0, 2); |
| __lsx_vstelm_w(tmp0, dst + dst_stride3, 0, 3); |
| } |
| |
| static void common_vt_2t_4x8_lsx(const uint8_t *src, int32_t src_stride, |
| uint8_t *dst, int32_t dst_stride, |
| int8_t *filter) { |
| __m128i src0, src1, src2, src3, src4, src5, src6, src7, src8; |
| __m128i vec0, vec1, vec2, vec3, vec4, vec5; |
| __m128i vec6, vec7, vec8, vec9, vec10, vec11; |
| __m128i tmp0, tmp1, tmp2, tmp3; |
| __m128i filt0; |
| |
| int32_t src_stride2 = src_stride << 1; |
| int32_t src_stride3 = src_stride2 + src_stride; |
| int32_t src_stride4 = src_stride2 << 1; |
| int32_t dst_stride2 = dst_stride << 1; |
| int32_t dst_stride3 = dst_stride2 + dst_stride; |
| int32_t dst_stride4 = dst_stride2 << 1; |
| uint8_t *dst_tmp1 = dst + dst_stride4; |
| |
| filt0 = __lsx_vldrepl_h(filter, 0); |
| src0 = __lsx_vld(src, 0); |
| DUP4_ARG2(__lsx_vldx, src, src_stride, src, src_stride2, src, src_stride3, |
| src, src_stride4, src1, src2, src3, src4); |
| src += src_stride4; |
| DUP4_ARG2(__lsx_vldx, src, src_stride, src, src_stride2, src, src_stride3, |
| src, src_stride4, src5, src6, src7, src8); |
| src += (src_stride4 + src_stride); |
| |
| DUP4_ARG2(__lsx_vilvl_b, src1, src0, src2, src1, src3, src2, src4, src3, vec0, |
| vec1, vec2, vec3); |
| DUP4_ARG2(__lsx_vilvl_b, src5, src4, src6, src5, src7, src6, src8, src7, vec4, |
| vec5, vec6, vec7); |
| DUP4_ARG2(__lsx_vilvl_d, vec1, vec0, vec3, vec2, vec5, vec4, vec7, vec6, vec8, |
| vec9, vec10, vec11); |
| |
| DUP4_ARG2(__lsx_vdp2_h_bu, vec8, filt0, vec9, filt0, vec10, filt0, vec11, |
| filt0, tmp0, tmp1, tmp2, tmp3); |
| DUP2_ARG3(__lsx_vssrarni_bu_h, tmp1, tmp0, FILTER_BITS, tmp3, tmp2, |
| FILTER_BITS, tmp0, tmp1); |
| |
| __lsx_vstelm_w(tmp0, dst, 0, 0); |
| __lsx_vstelm_w(tmp0, dst + dst_stride, 0, 1); |
| __lsx_vstelm_w(tmp0, dst + dst_stride2, 0, 2); |
| __lsx_vstelm_w(tmp0, dst + dst_stride3, 0, 3); |
| |
| __lsx_vstelm_w(tmp1, dst_tmp1, 0, 0); |
| __lsx_vstelm_w(tmp1, dst_tmp1 + dst_stride, 0, 1); |
| __lsx_vstelm_w(tmp1, dst_tmp1 + dst_stride2, 0, 2); |
| __lsx_vstelm_w(tmp1, dst_tmp1 + dst_stride3, 0, 3); |
| } |
| |
| static void common_vt_2t_4w_lsx(const uint8_t *src, int32_t src_stride, |
| uint8_t *dst, int32_t dst_stride, |
| int8_t *filter, int32_t height) { |
| if (height == 4) { |
| common_vt_2t_4x4_lsx(src, src_stride, dst, dst_stride, filter); |
| } else if (height == 8) { |
| common_vt_2t_4x8_lsx(src, src_stride, dst, dst_stride, filter); |
| } |
| } |
| |
| static void common_vt_2t_8x4_lsx(const uint8_t *src, int32_t src_stride, |
| uint8_t *dst, int32_t dst_stride, |
| int8_t *filter) { |
| __m128i src0, src1, src2, src3, src4, vec0, vec1, vec2, vec3, filt0; |
| __m128i out0, out1, tmp0, tmp1, tmp2, tmp3; |
| |
| int32_t src_stride2 = src_stride << 1; |
| int32_t src_stride3 = src_stride2 + src_stride; |
| int32_t src_stride4 = src_stride2 << 1; |
| int32_t dst_stride2 = dst_stride << 1; |
| int32_t dst_stride3 = dst_stride2 + dst_stride; |
| |
| filt0 = __lsx_vldrepl_h(filter, 0); |
| |
| src0 = __lsx_vld(src, 0); |
| DUP4_ARG2(__lsx_vldx, src, src_stride, src, src_stride2, src, src_stride3, |
| src, src_stride4, src1, src2, src3, src4); |
| |
| DUP4_ARG2(__lsx_vilvl_b, src1, src0, src2, src1, src3, src2, src4, src3, vec0, |
| vec1, vec2, vec3); |
| DUP4_ARG2(__lsx_vdp2_h_bu, vec0, filt0, vec1, filt0, vec2, filt0, vec3, filt0, |
| tmp0, tmp1, tmp2, tmp3); |
| DUP2_ARG3(__lsx_vssrarni_bu_h, tmp1, tmp0, FILTER_BITS, tmp3, tmp2, |
| FILTER_BITS, out0, out1); |
| |
| __lsx_vstelm_d(out0, dst, 0, 0); |
| __lsx_vstelm_d(out0, dst + dst_stride, 0, 1); |
| __lsx_vstelm_d(out1, dst + dst_stride2, 0, 0); |
| __lsx_vstelm_d(out1, dst + dst_stride3, 0, 1); |
| } |
| |
| static void common_vt_2t_8x8mult_lsx(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 = (height >> 3); |
| __m128i src0, src1, src2, src3, src4, src5, src6, src7, src8; |
| __m128i vec0, vec1, vec2, vec3, vec4, vec5, vec6, vec7, filt0; |
| __m128i out0, out1, tmp0, tmp1, tmp2, tmp3; |
| |
| int32_t src_stride2 = src_stride << 1; |
| int32_t src_stride3 = src_stride2 + src_stride; |
| int32_t src_stride4 = src_stride2 << 1; |
| int32_t dst_stride2 = dst_stride << 1; |
| int32_t dst_stride3 = dst_stride2 + dst_stride; |
| int32_t dst_stride4 = dst_stride2 << 1; |
| |
| filt0 = __lsx_vldrepl_h(filter, 0); |
| src0 = __lsx_vld(src, 0); |
| src += src_stride; |
| |
| for (; loop_cnt--;) { |
| src1 = __lsx_vld(src, 0); |
| DUP2_ARG2(__lsx_vldx, src, src_stride, src, src_stride2, src2, src3); |
| src4 = __lsx_vldx(src, src_stride3); |
| src += src_stride4; |
| src5 = __lsx_vld(src, 0); |
| DUP2_ARG2(__lsx_vldx, src, src_stride, src, src_stride2, src6, src7) |
| src8 = __lsx_vldx(src, src_stride3); |
| src += src_stride4; |
| |
| DUP4_ARG2(__lsx_vilvl_b, src1, src0, src2, src1, src3, src2, src4, src3, |
| vec0, vec1, vec2, vec3); |
| DUP4_ARG2(__lsx_vilvl_b, src5, src4, src6, src5, src7, src6, src8, src7, |
| vec4, vec5, vec6, vec7); |
| DUP4_ARG2(__lsx_vdp2_h_bu, vec0, filt0, vec1, filt0, vec2, filt0, vec3, |
| filt0, tmp0, tmp1, tmp2, tmp3); |
| DUP2_ARG3(__lsx_vssrarni_bu_h, tmp1, tmp0, FILTER_BITS, tmp3, tmp2, |
| FILTER_BITS, out0, out1); |
| |
| __lsx_vstelm_d(out0, dst, 0, 0); |
| __lsx_vstelm_d(out0, dst + dst_stride, 0, 1); |
| __lsx_vstelm_d(out1, dst + dst_stride2, 0, 0); |
| __lsx_vstelm_d(out1, dst + dst_stride3, 0, 1); |
| dst += dst_stride4; |
| |
| DUP4_ARG2(__lsx_vdp2_h_bu, vec4, filt0, vec5, filt0, vec6, filt0, vec7, |
| filt0, tmp0, tmp1, tmp2, tmp3); |
| DUP2_ARG3(__lsx_vssrarni_bu_h, tmp1, tmp0, FILTER_BITS, tmp3, tmp2, |
| FILTER_BITS, out0, out1); |
| |
| __lsx_vstelm_d(out0, dst, 0, 0); |
| __lsx_vstelm_d(out0, dst + dst_stride, 0, 1); |
| __lsx_vstelm_d(out1, dst + dst_stride2, 0, 0); |
| __lsx_vstelm_d(out1, dst + dst_stride3, 0, 1); |
| dst += dst_stride4; |
| |
| src0 = src8; |
| } |
| } |
| |
| static void common_vt_2t_8w_lsx(const uint8_t *src, int32_t src_stride, |
| uint8_t *dst, int32_t dst_stride, |
| int8_t *filter, int32_t height) { |
| if (height == 4) { |
| common_vt_2t_8x4_lsx(src, src_stride, dst, dst_stride, filter); |
| } else { |
| common_vt_2t_8x8mult_lsx(src, src_stride, dst, dst_stride, filter, height); |
| } |
| } |
| |
| static void common_vt_2t_16w_lsx(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 = (height >> 2); |
| __m128i src0, src1, src2, src3, src4, tmp, tmp0, tmp1; |
| __m128i vec0, vec1, vec2, vec3, vec4, vec5, vec6, vec7, filt0; |
| int32_t src_stride2 = src_stride << 1; |
| int32_t src_stride3 = src_stride2 + src_stride; |
| int32_t src_stride4 = src_stride2 << 1; |
| |
| filt0 = __lsx_vldrepl_h(filter, 0); |
| |
| src0 = __lsx_vld(src, 0); |
| src += src_stride; |
| |
| for (; loop_cnt--;) { |
| src1 = __lsx_vld(src, 0); |
| DUP2_ARG2(__lsx_vldx, src, src_stride, src, src_stride2, src2, src3); |
| src4 = __lsx_vldx(src, src_stride3); |
| src += src_stride4; |
| |
| DUP2_ARG2(__lsx_vilvl_b, src1, src0, src2, src1, vec0, vec2); |
| DUP2_ARG2(__lsx_vilvh_b, src1, src0, src2, src1, vec1, vec3); |
| DUP2_ARG2(__lsx_vdp2_h_bu, vec0, filt0, vec1, filt0, tmp0, tmp1); |
| tmp = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS); |
| __lsx_vst(tmp, dst, 0); |
| dst += dst_stride; |
| |
| DUP2_ARG2(__lsx_vilvl_b, src3, src2, src4, src3, vec4, vec6); |
| DUP2_ARG2(__lsx_vilvh_b, src3, src2, src4, src3, vec5, vec7); |
| DUP2_ARG2(__lsx_vdp2_h_bu, vec2, filt0, vec3, filt0, tmp0, tmp1); |
| tmp = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS); |
| __lsx_vst(tmp, dst, 0); |
| dst += dst_stride; |
| |
| DUP2_ARG2(__lsx_vdp2_h_bu, vec4, filt0, vec5, filt0, tmp0, tmp1); |
| tmp = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS); |
| __lsx_vst(tmp, dst, 0); |
| dst += dst_stride; |
| |
| DUP2_ARG2(__lsx_vdp2_h_bu, vec6, filt0, vec7, filt0, tmp0, tmp1); |
| tmp = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS); |
| __lsx_vst(tmp, dst, 0); |
| dst += dst_stride; |
| |
| src0 = src4; |
| } |
| } |
| |
| static void common_vt_2t_32w_lsx(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 = (height >> 2); |
| __m128i src0, src1, src2, src3, src4, src5, src6, src7, src8, src9; |
| __m128i vec0, vec1, vec2, vec3, vec4, vec5, vec6, vec7, filt0; |
| __m128i tmp, tmp0, tmp1; |
| |
| int32_t src_stride2 = src_stride << 1; |
| int32_t src_stride3 = src_stride2 + src_stride; |
| int32_t src_stride4 = src_stride2 << 1; |
| int32_t dst_stride2 = dst_stride << 1; |
| int32_t dst_stride3 = dst_stride2 + dst_stride; |
| uint8_t *src_tmp; |
| |
| filt0 = __lsx_vldrepl_h(filter, 0); |
| |
| DUP2_ARG2(__lsx_vld, src, 0, src, 16, src0, src5); |
| src += src_stride; |
| src_tmp = src + 16; |
| |
| for (; loop_cnt--;) { |
| DUP2_ARG2(__lsx_vld, src, 0, src_tmp, 0, src1, src6); |
| DUP4_ARG2(__lsx_vldx, src, src_stride, src_tmp, src_stride, src, |
| src_stride2, src_tmp, src_stride2, src2, src7, src3, src8); |
| DUP2_ARG2(__lsx_vldx, src, src_stride3, src_tmp, src_stride3, src4, src9); |
| DUP2_ARG2(__lsx_vilvl_b, src1, src0, src2, src1, vec0, vec2); |
| DUP2_ARG2(__lsx_vilvh_b, src1, src0, src2, src1, vec1, vec3); |
| src += src_stride4; |
| src_tmp += src_stride4; |
| |
| DUP2_ARG2(__lsx_vdp2_h_bu, vec0, filt0, vec1, filt0, tmp0, tmp1); |
| tmp = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS); |
| __lsx_vst(tmp, dst, 0); |
| |
| DUP2_ARG2(__lsx_vdp2_h_bu, vec2, filt0, vec3, filt0, tmp0, tmp1); |
| tmp = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS); |
| __lsx_vstx(tmp, dst, dst_stride); |
| |
| DUP2_ARG2(__lsx_vilvl_b, src3, src2, src4, src3, vec4, vec6); |
| DUP2_ARG2(__lsx_vilvh_b, src3, src2, src4, src3, vec5, vec7); |
| DUP2_ARG2(__lsx_vdp2_h_bu, vec4, filt0, vec5, filt0, tmp0, tmp1); |
| tmp = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS); |
| __lsx_vstx(tmp, dst, dst_stride2); |
| |
| DUP2_ARG2(__lsx_vdp2_h_bu, vec6, filt0, vec7, filt0, tmp0, tmp1); |
| tmp = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS); |
| __lsx_vstx(tmp, dst, dst_stride3); |
| |
| DUP2_ARG2(__lsx_vilvl_b, src6, src5, src7, src6, vec0, vec2); |
| DUP2_ARG2(__lsx_vilvh_b, src6, src5, src7, src6, vec1, vec3); |
| DUP2_ARG2(__lsx_vdp2_h_bu, vec0, filt0, vec1, filt0, tmp0, tmp1); |
| tmp = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS); |
| __lsx_vst(tmp, dst, 16); |
| |
| DUP2_ARG2(__lsx_vdp2_h_bu, vec2, filt0, vec3, filt0, tmp0, tmp1); |
| tmp = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS); |
| dst += dst_stride; |
| __lsx_vst(tmp, dst, 16); |
| |
| DUP2_ARG2(__lsx_vilvl_b, src8, src7, src9, src8, vec4, vec6); |
| DUP2_ARG2(__lsx_vilvh_b, src8, src7, src9, src8, vec5, vec7); |
| DUP2_ARG2(__lsx_vdp2_h_bu, vec4, filt0, vec5, filt0, tmp0, tmp1); |
| tmp = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS); |
| dst += dst_stride; |
| __lsx_vst(tmp, dst, 16); |
| |
| DUP2_ARG2(__lsx_vdp2_h_bu, vec6, filt0, vec7, filt0, tmp0, tmp1); |
| tmp = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS); |
| dst += dst_stride; |
| __lsx_vst(tmp, dst, 16); |
| |
| dst += dst_stride; |
| |
| src0 = src4; |
| src5 = src9; |
| } |
| } |
| |
| static void common_vt_2t_64w_lsx(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 = (height >> 1); |
| __m128i src0, src1, src2, src3, src4, src5, src6, src7, src8, src9, src10; |
| __m128i src11, vec0, vec1, vec2, vec3, vec4, vec5, vec6, vec7, filt0; |
| __m128i tmp, tmp0, tmp1; |
| |
| int32_t src_stride2 = src_stride << 1; |
| int32_t dst_stride2 = dst_stride << 1; |
| uint8_t *dst_tmp1 = dst + dst_stride; |
| |
| filt0 = __lsx_vldrepl_h(filter, 0); |
| |
| DUP4_ARG2(__lsx_vld, src, 0, src, 16, src, 32, src, 48, src0, src3, src6, |
| src9); |
| src += src_stride; |
| |
| for (; loop_cnt--;) { |
| uint8_t *src_tmp0 = src + src_stride; |
| |
| DUP4_ARG2(__lsx_vld, src, 0, src, 16, src, 32, src, 48, src1, src4, src7, |
| src10); |
| DUP4_ARG2(__lsx_vld, src_tmp0, 0, src_tmp0, 16, src_tmp0, 32, src_tmp0, 48, |
| src2, src5, src8, src11); |
| src += src_stride2; |
| |
| DUP2_ARG2(__lsx_vilvl_b, src1, src0, src2, src1, vec0, vec2); |
| DUP2_ARG2(__lsx_vilvh_b, src1, src0, src2, src1, vec1, vec3); |
| DUP2_ARG2(__lsx_vdp2_h_bu, vec0, filt0, vec1, filt0, tmp0, tmp1); |
| tmp = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS); |
| __lsx_vst(tmp, dst, 0); |
| |
| DUP2_ARG2(__lsx_vdp2_h_bu, vec2, filt0, vec3, filt0, tmp0, tmp1); |
| tmp = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS); |
| __lsx_vst(tmp, dst_tmp1, 0); |
| |
| DUP2_ARG2(__lsx_vilvl_b, src4, src3, src5, src4, vec4, vec6); |
| DUP2_ARG2(__lsx_vilvh_b, src4, src3, src5, src4, vec5, vec7); |
| DUP2_ARG2(__lsx_vdp2_h_bu, vec4, filt0, vec5, filt0, tmp0, tmp1); |
| tmp = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS); |
| __lsx_vst(tmp, dst, 16); |
| |
| DUP2_ARG2(__lsx_vdp2_h_bu, vec6, filt0, vec7, filt0, tmp0, tmp1); |
| tmp = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS); |
| __lsx_vst(tmp, dst_tmp1, 16); |
| |
| DUP2_ARG2(__lsx_vilvl_b, src7, src6, src8, src7, vec0, vec2); |
| DUP2_ARG2(__lsx_vilvh_b, src7, src6, src8, src7, vec1, vec3); |
| DUP2_ARG2(__lsx_vdp2_h_bu, vec0, filt0, vec1, filt0, tmp0, tmp1); |
| tmp = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS); |
| __lsx_vst(tmp, dst, 32); |
| |
| DUP2_ARG2(__lsx_vdp2_h_bu, vec2, filt0, vec3, filt0, tmp0, tmp1); |
| tmp = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS); |
| __lsx_vst(tmp, dst_tmp1, 32); |
| |
| DUP2_ARG2(__lsx_vilvl_b, src10, src9, src11, src10, vec4, vec6); |
| DUP2_ARG2(__lsx_vilvh_b, src10, src9, src11, src10, vec5, vec7); |
| DUP2_ARG2(__lsx_vdp2_h_bu, vec4, filt0, vec5, filt0, tmp0, tmp1); |
| tmp = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS); |
| __lsx_vst(tmp, dst, 48); |
| |
| DUP2_ARG2(__lsx_vdp2_h_bu, vec6, filt0, vec7, filt0, tmp0, tmp1); |
| tmp = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS); |
| __lsx_vst(tmp, dst_tmp1, 48); |
| dst += dst_stride2; |
| dst_tmp1 += dst_stride2; |
| |
| src0 = src2; |
| src3 = src5; |
| src6 = src8; |
| src9 = src11; |
| } |
| } |
| |
| void vpx_convolve8_vert_lsx(const uint8_t *src, ptrdiff_t src_stride, |
| uint8_t *dst, ptrdiff_t dst_stride, |
| const InterpKernel *filter, int x0_q4, |
| int32_t x_step_q4, int y0_q4, int y_step_q4, int w, |
| int h) { |
| const int16_t *const filter_y = filter[y0_q4]; |
| int8_t cnt, filt_ver[8]; |
| |
| assert(y_step_q4 == 16); |
| assert(((const int32_t *)filter_y)[1] != 0x800000); |
| |
| for (cnt = 8; cnt--;) { |
| filt_ver[cnt] = filter_y[cnt]; |
| } |
| |
| if (vpx_get_filter_taps(filter_y) == 2) { |
| switch (w) { |
| case 4: |
| common_vt_2t_4w_lsx(src, (int32_t)src_stride, dst, (int32_t)dst_stride, |
| &filt_ver[3], h); |
| break; |
| case 8: |
| common_vt_2t_8w_lsx(src, (int32_t)src_stride, dst, (int32_t)dst_stride, |
| &filt_ver[3], h); |
| break; |
| case 16: |
| common_vt_2t_16w_lsx(src, (int32_t)src_stride, dst, (int32_t)dst_stride, |
| &filt_ver[3], h); |
| break; |
| case 32: |
| common_vt_2t_32w_lsx(src, (int32_t)src_stride, dst, (int32_t)dst_stride, |
| &filt_ver[3], h); |
| break; |
| case 64: |
| common_vt_2t_64w_lsx(src, (int32_t)src_stride, dst, (int32_t)dst_stride, |
| &filt_ver[3], h); |
| break; |
| default: |
| vpx_convolve8_vert_c(src, src_stride, dst, dst_stride, filter, x0_q4, |
| x_step_q4, y0_q4, y_step_q4, w, h); |
| break; |
| } |
| } else { |
| switch (w) { |
| case 4: |
| common_vt_8t_4w_lsx(src, (int32_t)src_stride, dst, (int32_t)dst_stride, |
| filt_ver, h); |
| break; |
| case 8: |
| common_vt_8t_8w_lsx(src, (int32_t)src_stride, dst, (int32_t)dst_stride, |
| filt_ver, h); |
| break; |
| case 16: |
| common_vt_8t_16w_lsx(src, (int32_t)src_stride, dst, (int32_t)dst_stride, |
| filt_ver, h); |
| break; |
| case 32: |
| common_vt_8t_32w_lsx(src, (int32_t)src_stride, dst, (int32_t)dst_stride, |
| filt_ver, h); |
| break; |
| case 64: |
| common_vt_8t_64w_lsx(src, (int32_t)src_stride, dst, (int32_t)dst_stride, |
| filt_ver, h); |
| break; |
| default: |
| vpx_convolve8_vert_c(src, src_stride, dst, dst_stride, filter, x0_q4, |
| x_step_q4, y0_q4, y_step_q4, w, h); |
| break; |
| } |
| } |
| } |
