Source code
Revision control
Copy as Markdown
Other Tools
/*
* 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 "./vpx_dsp_rtcd.h"
#include "vpx_ports/mem.h"
#include "vpx_dsp/loongarch/variance_lsx.h"
#include "vpx_dsp/variance.h"
static const uint8_t bilinear_filters_lsx[8][2] = {
{ 128, 0 }, { 112, 16 }, { 96, 32 }, { 80, 48 },
{ 64, 64 }, { 48, 80 }, { 32, 96 }, { 16, 112 },
};
#define VARIANCE_WxH(sse, diff, shift) \
(sse) - (((uint32_t)(diff) * (diff)) >> (shift))
#define VARIANCE_LARGE_WxH(sse, diff, shift) \
(sse) - (((int64_t)(diff) * (diff)) >> (shift))
static uint32_t avg_sse_diff_64x64_lsx(const uint8_t *src_ptr,
int32_t src_stride,
const uint8_t *ref_ptr,
int32_t ref_stride,
const uint8_t *sec_pred, int32_t *diff) {
int32_t res, ht_cnt = 32;
__m128i src0, src1, src2, src3, ref0, ref1, ref2, ref3;
__m128i pred0, pred1, pred2, pred3, vec, vec_tmp;
__m128i avg0, avg1, avg2, avg3;
__m128i var = __lsx_vldi(0);
avg0 = var;
avg1 = var;
avg2 = var;
avg3 = var;
for (; ht_cnt--;) {
DUP4_ARG2(__lsx_vld, sec_pred, 0, sec_pred, 16, sec_pred, 32, sec_pred, 48,
pred0, pred1, pred2, pred3);
sec_pred += 64;
DUP4_ARG2(__lsx_vld, src_ptr, 0, src_ptr, 16, src_ptr, 32, src_ptr, 48,
src0, src1, src2, src3);
src_ptr += src_stride;
DUP4_ARG2(__lsx_vld, ref_ptr, 0, ref_ptr, 16, ref_ptr, 32, ref_ptr, 48,
ref0, ref1, ref2, ref3);
ref_ptr += ref_stride;
DUP4_ARG2(__lsx_vavgr_bu, src0, pred0, src1, pred1, src2, pred2, src3,
pred3, src0, src1, src2, src3);
CALC_MSE_AVG_B(src0, ref0, var, avg0);
CALC_MSE_AVG_B(src1, ref1, var, avg1);
CALC_MSE_AVG_B(src2, ref2, var, avg2);
CALC_MSE_AVG_B(src3, ref3, var, avg3);
DUP4_ARG2(__lsx_vld, sec_pred, 0, sec_pred, 16, sec_pred, 32, sec_pred, 48,
pred0, pred1, pred2, pred3);
sec_pred += 64;
DUP4_ARG2(__lsx_vld, src_ptr, 0, src_ptr, 16, src_ptr, 32, src_ptr, 48,
src0, src1, src2, src3);
src_ptr += src_stride;
DUP4_ARG2(__lsx_vld, ref_ptr, 0, ref_ptr, 16, ref_ptr, 32, ref_ptr, 48,
ref0, ref1, ref2, ref3);
ref_ptr += ref_stride;
DUP4_ARG2(__lsx_vavgr_bu, src0, pred0, src1, pred1, src2, pred2, src3,
pred3, src0, src1, src2, src3);
CALC_MSE_AVG_B(src0, ref0, var, avg0);
CALC_MSE_AVG_B(src1, ref1, var, avg1);
CALC_MSE_AVG_B(src2, ref2, var, avg2);
CALC_MSE_AVG_B(src3, ref3, var, avg3);
}
vec = __lsx_vhaddw_w_h(avg0, avg0);
vec_tmp = __lsx_vhaddw_w_h(avg1, avg1);
vec = __lsx_vadd_w(vec, vec_tmp);
vec_tmp = __lsx_vhaddw_w_h(avg2, avg2);
vec = __lsx_vadd_w(vec, vec_tmp);
vec_tmp = __lsx_vhaddw_w_h(avg3, avg3);
vec = __lsx_vadd_w(vec, vec_tmp);
HADD_SW_S32(vec, *diff);
HADD_SW_S32(var, res);
return res;
}
static uint32_t sub_pixel_sse_diff_8width_h_lsx(
const uint8_t *src, int32_t src_stride, const uint8_t *dst,
int32_t dst_stride, const uint8_t *filter, int32_t height, int32_t *diff) {
uint32_t loop_cnt = (height >> 2);
int32_t res;
__m128i src0, src1, src2, src3, ref0, ref1, ref2, ref3;
__m128i vec0, vec1, vec2, vec3, filt0, out, vec;
__m128i mask = { 0x0403030202010100, 0x0807070606050504 };
__m128i avg = __lsx_vldi(0);
__m128i var = avg;
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);
for (; loop_cnt--;) {
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;
ref0 = __lsx_vld(dst, 0);
DUP2_ARG2(__lsx_vldx, dst, dst_stride, dst, dst_stride2, ref1, ref2);
ref3 = __lsx_vldx(dst, dst_stride3);
dst += dst_stride4;
DUP2_ARG2(__lsx_vpickev_d, ref1, ref0, ref3, ref2, ref0, ref1);
DUP2_ARG3(__lsx_vshuf_b, src0, src0, mask, src1, src1, mask, vec0, vec1);
DUP2_ARG3(__lsx_vshuf_b, src2, src2, mask, src3, src3, mask, vec2, vec3);
DUP4_ARG2(__lsx_vdp2_h_bu, vec0, filt0, vec1, filt0, vec2, filt0, vec3,
filt0, vec0, vec1, vec2, vec3);
DUP4_ARG3(__lsx_vssrarni_bu_h, vec0, vec0, FILTER_BITS, vec1, vec1,
FILTER_BITS, vec2, vec2, FILTER_BITS, vec3, vec3, FILTER_BITS,
src0, src1, src2, src3);
out = __lsx_vpackev_d(src1, src0);
CALC_MSE_AVG_B(out, ref0, var, avg);
out = __lsx_vpackev_d(src3, src2);
CALC_MSE_AVG_B(out, ref1, var, avg);
}
vec = __lsx_vhaddw_w_h(avg, avg);
HADD_SW_S32(vec, *diff);
HADD_SW_S32(var, res);
return res;
}
static uint32_t sub_pixel_sse_diff_16width_h_lsx(
const uint8_t *src, int32_t src_stride, const uint8_t *dst,
int32_t dst_stride, const uint8_t *filter, int32_t height, int32_t *diff) {
uint32_t loop_cnt = (height >> 2);
int32_t res;
__m128i src0, src1, src2, src3, src4, src5, src6, src7;
__m128i dst0, dst1, dst2, dst3, filt0;
__m128i vec0, vec1, vec2, vec3, vec4, vec5, vec6, vec7;
__m128i out0, out1, out2, out3, out4, out5, out6, out7;
__m128i vec, var = __lsx_vldi(0);
__m128i avg = var;
__m128i mask = { 0x0403030202010100, 0x0807070606050504 };
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);
for (; loop_cnt--;) {
DUP2_ARG2(__lsx_vld, src, 0, src, 8, src0, src1);
src += src_stride;
DUP2_ARG2(__lsx_vld, src, 0, src, 8, src2, src3);
src += src_stride;
DUP2_ARG2(__lsx_vld, src, 0, src, 8, src4, src5);
src += src_stride;
DUP2_ARG2(__lsx_vld, src, 0, src, 8, src6, src7);
src += src_stride;
dst0 = __lsx_vld(dst, 0);
DUP2_ARG2(__lsx_vldx, dst, dst_stride, dst, dst_stride2, dst1, dst2);
dst3 = __lsx_vldx(dst, dst_stride3);
dst += dst_stride4;
DUP2_ARG3(__lsx_vshuf_b, src0, src0, mask, src1, src1, mask, vec0, vec1);
DUP2_ARG3(__lsx_vshuf_b, src2, src2, mask, src3, src3, mask, vec2, vec3);
DUP2_ARG3(__lsx_vshuf_b, src4, src4, mask, src5, src5, mask, vec4, vec5);
DUP2_ARG3(__lsx_vshuf_b, src6, src6, mask, src7, src7, mask, vec6, vec7);
DUP4_ARG2(__lsx_vdp2_h_bu, vec0, filt0, vec1, filt0, vec2, filt0, vec3,
filt0, out0, out1, out2, out3);
DUP4_ARG2(__lsx_vdp2_h_bu, vec4, filt0, vec5, filt0, vec6, filt0, vec7,
filt0, out4, out5, out6, out7);
DUP4_ARG3(__lsx_vssrarni_bu_h, out1, out0, FILTER_BITS, out3, out2,
FILTER_BITS, out5, out4, FILTER_BITS, out7, out6, FILTER_BITS,
src0, src1, src2, src3);
CALC_MSE_AVG_B(src0, dst0, var, avg);
CALC_MSE_AVG_B(src1, dst1, var, avg);
CALC_MSE_AVG_B(src2, dst2, var, avg);
CALC_MSE_AVG_B(src3, dst3, var, avg);
}
vec = __lsx_vhaddw_w_h(avg, avg);
HADD_SW_S32(vec, *diff);
HADD_SW_S32(var, res);
return res;
}
static uint32_t sub_pixel_sse_diff_32width_h_lsx(
const uint8_t *src, int32_t src_stride, const uint8_t *dst,
int32_t dst_stride, const uint8_t *filter, int32_t height, int32_t *diff) {
uint32_t sse = 0;
int32_t diff0[2];
sse += sub_pixel_sse_diff_16width_h_lsx(src, src_stride, dst, dst_stride,
filter, height, &diff0[0]);
src += 16;
dst += 16;
sse += sub_pixel_sse_diff_16width_h_lsx(src, src_stride, dst, dst_stride,
filter, height, &diff0[1]);
*diff = diff0[0] + diff0[1];
return sse;
}
static uint32_t sub_pixel_sse_diff_8width_v_lsx(
const uint8_t *src, int32_t src_stride, const uint8_t *dst,
int32_t dst_stride, const uint8_t *filter, int32_t height, int32_t *diff) {
uint32_t loop_cnt = (height >> 2);
int32_t res;
__m128i ref0, ref1, ref2, ref3, src0, src1, src2, src3, src4;
__m128i vec, vec0, vec1, vec2, vec3, tmp0, tmp1, tmp2, tmp3, filt0;
__m128i avg = __lsx_vldi(0);
__m128i var = avg;
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;
ref0 = __lsx_vld(dst, 0);
DUP2_ARG2(__lsx_vldx, dst, dst_stride, dst, dst_stride2, ref1, ref2);
ref3 = __lsx_vldx(dst, dst_stride3);
dst += dst_stride4;
DUP2_ARG2(__lsx_vpickev_d, ref1, ref0, ref3, ref2, ref0, ref1);
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, src0, src1);
CALC_MSE_AVG_B(src0, ref0, var, avg);
CALC_MSE_AVG_B(src1, ref1, var, avg);
src0 = src4;
}
vec = __lsx_vhaddw_w_h(avg, avg);
HADD_SW_S32(vec, *diff);
HADD_SW_S32(var, res);
return res;
}
static uint32_t sub_pixel_sse_diff_16width_v_lsx(
const uint8_t *src, int32_t src_stride, const uint8_t *dst,
int32_t dst_stride, const uint8_t *filter, int32_t height, int32_t *diff) {
uint32_t loop_cnt = (height >> 2);
int32_t res;
__m128i ref0, ref1, ref2, ref3, src0, src1, src2, src3, src4;
__m128i out0, out1, out2, out3, tmp0, tmp1, filt0, vec;
__m128i vec0, vec1, vec2, vec3, vec4, vec5, vec6, vec7;
__m128i var = __lsx_vldi(0);
__m128i avg = var;
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;
ref0 = __lsx_vld(dst, 0);
DUP2_ARG2(__lsx_vldx, dst, dst_stride, dst, dst_stride2, ref1, ref2);
ref3 = __lsx_vldx(dst, dst_stride3);
dst += dst_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);
out0 = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS);
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);
out1 = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS);
DUP2_ARG2(__lsx_vdp2_h_bu, vec4, filt0, vec5, filt0, tmp0, tmp1);
out2 = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS);
DUP2_ARG2(__lsx_vdp2_h_bu, vec6, filt0, vec7, filt0, tmp0, tmp1);
out3 = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS);
src0 = src4;
CALC_MSE_AVG_B(out0, ref0, var, avg);
CALC_MSE_AVG_B(out1, ref1, var, avg);
CALC_MSE_AVG_B(out2, ref2, var, avg);
CALC_MSE_AVG_B(out3, ref3, var, avg);
}
vec = __lsx_vhaddw_w_h(avg, avg);
HADD_SW_S32(vec, *diff);
HADD_SW_S32(var, res);
return res;
}
static uint32_t sub_pixel_sse_diff_32width_v_lsx(
const uint8_t *src, int32_t src_stride, const uint8_t *dst,
int32_t dst_stride, const uint8_t *filter, int32_t height, int32_t *diff) {
uint32_t sse = 0;
int32_t diff0[2];
sse += sub_pixel_sse_diff_16width_v_lsx(src, src_stride, dst, dst_stride,
filter, height, &diff0[0]);
src += 16;
dst += 16;
sse += sub_pixel_sse_diff_16width_v_lsx(src, src_stride, dst, dst_stride,
filter, height, &diff0[1]);
*diff = diff0[0] + diff0[1];
return sse;
}
static uint32_t sub_pixel_sse_diff_8width_hv_lsx(
const uint8_t *src, int32_t src_stride, const uint8_t *dst,
int32_t dst_stride, const uint8_t *filter_horiz, const uint8_t *filter_vert,
int32_t height, int32_t *diff) {
uint32_t loop_cnt = (height >> 2);
int32_t res;
__m128i ref0, ref1, ref2, ref3, src0, src1, src2, src3, src4, out0, out1;
__m128i hz_out0, hz_out1, tmp0, tmp1, tmp2, tmp3, vec, vec0, filt_hz, filt_vt;
__m128i mask = { 0x0403030202010100, 0x0807070606050504 };
__m128i avg = __lsx_vldi(0);
__m128i var = avg;
filt_hz = __lsx_vldrepl_h(filter_horiz, 0);
filt_vt = __lsx_vldrepl_h(filter_vert, 0);
src0 = __lsx_vld(src, 0);
src += src_stride;
HORIZ_2TAP_FILT_UH(src0, src0, mask, filt_hz, FILTER_BITS, hz_out0);
for (; loop_cnt--;) {
DUP2_ARG2(__lsx_vld, src, 0, dst, 0, src1, ref0);
src += src_stride;
dst += dst_stride;
DUP2_ARG2(__lsx_vld, src, 0, dst, 0, src2, ref1);
src += src_stride;
dst += dst_stride;
DUP2_ARG2(__lsx_vld, src, 0, dst, 0, src3, ref2);
src += src_stride;
dst += dst_stride;
DUP2_ARG2(__lsx_vld, src, 0, dst, 0, src4, ref3);
src += src_stride;
dst += dst_stride;
DUP2_ARG2(__lsx_vpickev_d, ref1, ref0, ref3, ref2, ref0, ref1);
HORIZ_2TAP_FILT_UH(src1, src1, mask, filt_hz, FILTER_BITS, hz_out1);
vec0 = __lsx_vpackev_b(hz_out1, hz_out0);
tmp0 = __lsx_vdp2_h_bu(vec0, filt_vt);
HORIZ_2TAP_FILT_UH(src2, src2, mask, filt_hz, FILTER_BITS, hz_out0);
vec0 = __lsx_vpackev_b(hz_out0, hz_out1);
tmp1 = __lsx_vdp2_h_bu(vec0, filt_vt);
HORIZ_2TAP_FILT_UH(src3, src3, mask, filt_hz, FILTER_BITS, hz_out1);
vec0 = __lsx_vpackev_b(hz_out1, hz_out0);
tmp2 = __lsx_vdp2_h_bu(vec0, filt_vt);
HORIZ_2TAP_FILT_UH(src4, src4, mask, filt_hz, FILTER_BITS, hz_out0);
vec0 = __lsx_vpackev_b(hz_out0, hz_out1);
tmp3 = __lsx_vdp2_h_bu(vec0, filt_vt);
DUP2_ARG3(__lsx_vssrarni_bu_h, tmp1, tmp0, FILTER_BITS, tmp3, tmp2,
FILTER_BITS, out0, out1);
CALC_MSE_AVG_B(out0, ref0, var, avg);
CALC_MSE_AVG_B(out1, ref1, var, avg);
}
vec = __lsx_vhaddw_w_h(avg, avg);
HADD_SW_S32(vec, *diff);
HADD_SW_S32(var, res);
return res;
}
static uint32_t sub_pixel_sse_diff_16width_hv_lsx(
const uint8_t *src, int32_t src_stride, const uint8_t *dst,
int32_t dst_stride, const uint8_t *filter_horiz, const uint8_t *filter_vert,
int32_t height, int32_t *diff) {
uint32_t loop_cnt = (height >> 2);
int32_t res;
__m128i src0, src1, src2, src3, src4, src5, src6, src7;
__m128i ref0, ref1, ref2, ref3, filt_hz, filt_vt, vec0, vec1;
__m128i hz_out0, hz_out1, hz_out2, hz_out3, tmp0, tmp1, vec;
__m128i var = __lsx_vldi(0);
__m128i avg = var;
__m128i mask = { 0x0403030202010100, 0x0807070606050504 };
int32_t dst_stride2 = dst_stride << 1;
int32_t dst_stride3 = dst_stride2 + dst_stride;
int32_t dst_stride4 = dst_stride2 << 1;
filt_hz = __lsx_vldrepl_h(filter_horiz, 0);
filt_vt = __lsx_vldrepl_h(filter_vert, 0);
DUP2_ARG2(__lsx_vld, src, 0, src, 8, src0, src1);
src += src_stride;
HORIZ_2TAP_FILT_UH(src0, src0, mask, filt_hz, FILTER_BITS, hz_out0);
HORIZ_2TAP_FILT_UH(src1, src1, mask, filt_hz, FILTER_BITS, hz_out2);
for (; loop_cnt--;) {
DUP2_ARG2(__lsx_vld, src, 0, src, 8, src0, src1);
src += src_stride;
DUP2_ARG2(__lsx_vld, src, 0, src, 8, src2, src3);
src += src_stride;
DUP2_ARG2(__lsx_vld, src, 0, src, 8, src4, src5);
src += src_stride;
DUP2_ARG2(__lsx_vld, src, 0, src, 8, src6, src7);
src += src_stride;
ref0 = __lsx_vld(dst, 0);
DUP2_ARG2(__lsx_vldx, dst, dst_stride, dst, dst_stride2, ref1, ref2);
ref3 = __lsx_vldx(dst, dst_stride3);
dst += dst_stride4;
HORIZ_2TAP_FILT_UH(src0, src0, mask, filt_hz, FILTER_BITS, hz_out1);
HORIZ_2TAP_FILT_UH(src1, src1, mask, filt_hz, FILTER_BITS, hz_out3);
DUP2_ARG2(__lsx_vpackev_b, hz_out1, hz_out0, hz_out3, hz_out2, vec0, vec1);
DUP2_ARG2(__lsx_vdp2_h_bu, vec0, filt_vt, vec1, filt_vt, tmp0, tmp1);
src0 = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS);
HORIZ_2TAP_FILT_UH(src2, src2, mask, filt_hz, FILTER_BITS, hz_out0);
HORIZ_2TAP_FILT_UH(src3, src3, mask, filt_hz, FILTER_BITS, hz_out2);
DUP2_ARG2(__lsx_vpackev_b, hz_out0, hz_out1, hz_out2, hz_out3, vec0, vec1);
DUP2_ARG2(__lsx_vdp2_h_bu, vec0, filt_vt, vec1, filt_vt, tmp0, tmp1);
src1 = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS);
HORIZ_2TAP_FILT_UH(src4, src4, mask, filt_hz, FILTER_BITS, hz_out1);
HORIZ_2TAP_FILT_UH(src5, src5, mask, filt_hz, FILTER_BITS, hz_out3);
DUP2_ARG2(__lsx_vpackev_b, hz_out1, hz_out0, hz_out3, hz_out2, vec0, vec1);
DUP2_ARG2(__lsx_vdp2_h_bu, vec0, filt_vt, vec1, filt_vt, tmp0, tmp1);
src2 = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS);
HORIZ_2TAP_FILT_UH(src6, src6, mask, filt_hz, FILTER_BITS, hz_out0);
HORIZ_2TAP_FILT_UH(src7, src7, mask, filt_hz, FILTER_BITS, hz_out2);
DUP2_ARG2(__lsx_vpackev_b, hz_out0, hz_out1, hz_out2, hz_out3, vec0, vec1);
DUP2_ARG2(__lsx_vdp2_h_bu, vec0, filt_vt, vec1, filt_vt, tmp0, tmp1);
src3 = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS);
CALC_MSE_AVG_B(src0, ref0, var, avg);
CALC_MSE_AVG_B(src1, ref1, var, avg);
CALC_MSE_AVG_B(src2, ref2, var, avg);
CALC_MSE_AVG_B(src3, ref3, var, avg);
}
vec = __lsx_vhaddw_w_h(avg, avg);
HADD_SW_S32(vec, *diff);
HADD_SW_S32(var, res);
return res;
}
static uint32_t sub_pixel_sse_diff_32width_hv_lsx(
const uint8_t *src, int32_t src_stride, const uint8_t *dst,
int32_t dst_stride, const uint8_t *filter_horiz, const uint8_t *filter_vert,
int32_t height, int32_t *diff) {
uint32_t sse = 0;
int32_t diff0[2];
sse += sub_pixel_sse_diff_16width_hv_lsx(src, src_stride, dst, dst_stride,
filter_horiz, filter_vert, height,
&diff0[0]);
src += 16;
dst += 16;
sse += sub_pixel_sse_diff_16width_hv_lsx(src, src_stride, dst, dst_stride,
filter_horiz, filter_vert, height,
&diff0[1]);
*diff = diff0[0] + diff0[1];
return sse;
}
static uint32_t subpel_avg_ssediff_16w_h_lsx(
const uint8_t *src, int32_t src_stride, const uint8_t *dst,
int32_t dst_stride, const uint8_t *sec_pred, const uint8_t *filter,
int32_t height, int32_t *diff, int32_t width) {
uint32_t loop_cnt = (height >> 2);
int32_t res;
__m128i src0, src1, src2, src3, src4, src5, src6, src7;
__m128i dst0, dst1, dst2, dst3, tmp0, tmp1, tmp2, tmp3;
__m128i pred0, pred1, pred2, pred3, filt0, vec;
__m128i vec0, vec1, vec2, vec3, vec4, vec5, vec6, vec7;
__m128i out0, out1, out2, out3, out4, out5, out6, out7;
__m128i mask = { 0x403030202010100, 0x807070606050504 };
__m128i avg = __lsx_vldi(0);
__m128i var = avg;
filt0 = __lsx_vldrepl_h(filter, 0);
for (; loop_cnt--;) {
DUP2_ARG2(__lsx_vld, src, 0, src, 8, src0, src1);
src += src_stride;
DUP2_ARG2(__lsx_vld, src, 0, src, 8, src2, src3);
src += src_stride;
DUP2_ARG2(__lsx_vld, src, 0, src, 8, src4, src5);
src += src_stride;
DUP2_ARG2(__lsx_vld, src, 0, src, 8, src6, src7);
src += src_stride;
dst0 = __lsx_vld(dst, 0);
dst += dst_stride;
dst1 = __lsx_vld(dst, 0);
dst += dst_stride;
dst2 = __lsx_vld(dst, 0);
dst += dst_stride;
dst3 = __lsx_vld(dst, 0);
dst += dst_stride;
pred0 = __lsx_vld(sec_pred, 0);
sec_pred += width;
pred1 = __lsx_vld(sec_pred, 0);
sec_pred += width;
pred2 = __lsx_vld(sec_pred, 0);
sec_pred += width;
pred3 = __lsx_vld(sec_pred, 0);
sec_pred += width;
DUP2_ARG3(__lsx_vshuf_b, src0, src0, mask, src1, src1, mask, vec0, vec1);
DUP2_ARG3(__lsx_vshuf_b, src2, src2, mask, src3, src3, mask, vec2, vec3);
DUP2_ARG3(__lsx_vshuf_b, src4, src4, mask, src5, src5, mask, vec4, vec5);
DUP2_ARG3(__lsx_vshuf_b, src6, src6, mask, src7, src7, mask, vec6, vec7);
DUP4_ARG2(__lsx_vdp2_h_bu, vec0, filt0, vec1, filt0, vec2, filt0, vec3,
filt0, out0, out1, out2, out3);
DUP4_ARG2(__lsx_vdp2_h_bu, vec4, filt0, vec5, filt0, vec6, filt0, vec7,
filt0, out4, out5, out6, out7);
DUP4_ARG3(__lsx_vssrarni_bu_h, out1, out0, FILTER_BITS, out3, out2,
FILTER_BITS, out5, out4, FILTER_BITS, out7, out6, FILTER_BITS,
tmp0, tmp1, tmp2, tmp3);
DUP4_ARG2(__lsx_vavgr_bu, tmp0, pred0, tmp1, pred1, tmp2, pred2, tmp3,
pred3, tmp0, tmp1, tmp2, tmp3);
CALC_MSE_AVG_B(tmp0, dst0, var, avg);
CALC_MSE_AVG_B(tmp1, dst1, var, avg);
CALC_MSE_AVG_B(tmp2, dst2, var, avg);
CALC_MSE_AVG_B(tmp3, dst3, var, avg);
}
vec = __lsx_vhaddw_w_h(avg, avg);
HADD_SW_S32(vec, *diff);
HADD_SW_S32(var, res);
return res;
}
static uint32_t subpel_avg_ssediff_16w_v_lsx(
const uint8_t *src, int32_t src_stride, const uint8_t *dst,
int32_t dst_stride, const uint8_t *sec_pred, const uint8_t *filter,
int32_t height, int32_t *diff, int32_t width) {
uint32_t loop_cnt = (height >> 2);
int32_t res;
__m128i ref0, ref1, ref2, ref3, pred0, pred1, pred2, pred3;
__m128i src0, src1, src2, src3, src4, out0, out1, out2, out3;
__m128i vec0, vec1, vec2, vec3, vec4, vec5, vec6, vec7;
__m128i tmp0, tmp1, vec, filt0;
__m128i avg = __lsx_vldi(0);
__m128i var = avg;
filt0 = __lsx_vldrepl_h(filter, 0);
src0 = __lsx_vld(src, 0);
src += src_stride;
for (; loop_cnt--;) {
src1 = __lsx_vld(src, 0);
src += src_stride;
src2 = __lsx_vld(src, 0);
src += src_stride;
src3 = __lsx_vld(src, 0);
src += src_stride;
src4 = __lsx_vld(src, 0);
src += src_stride;
pred0 = __lsx_vld(sec_pred, 0);
sec_pred += width;
pred1 = __lsx_vld(sec_pred, 0);
sec_pred += width;
pred2 = __lsx_vld(sec_pred, 0);
sec_pred += width;
pred3 = __lsx_vld(sec_pred, 0);
sec_pred += width;
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);
out0 = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS);
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);
out1 = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS);
DUP2_ARG2(__lsx_vdp2_h_bu, vec4, filt0, vec5, filt0, tmp0, tmp1);
out2 = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS);
DUP2_ARG2(__lsx_vdp2_h_bu, vec6, filt0, vec7, filt0, tmp0, tmp1);
out3 = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS);
src0 = src4;
ref0 = __lsx_vld(dst, 0);
dst += dst_stride;
ref1 = __lsx_vld(dst, 0);
dst += dst_stride;
ref2 = __lsx_vld(dst, 0);
dst += dst_stride;
ref3 = __lsx_vld(dst, 0);
dst += dst_stride;
DUP4_ARG2(__lsx_vavgr_bu, out0, pred0, out1, pred1, out2, pred2, out3,
pred3, out0, out1, out2, out3);
CALC_MSE_AVG_B(out0, ref0, var, avg);
CALC_MSE_AVG_B(out1, ref1, var, avg);
CALC_MSE_AVG_B(out2, ref2, var, avg);
CALC_MSE_AVG_B(out3, ref3, var, avg);
}
vec = __lsx_vhaddw_w_h(avg, avg);
HADD_SW_S32(vec, *diff);
HADD_SW_S32(var, res);
return res;
}
static uint32_t subpel_avg_ssediff_16w_hv_lsx(
const uint8_t *src, int32_t src_stride, const uint8_t *dst,
int32_t dst_stride, const uint8_t *sec_pred, const uint8_t *filter_horiz,
const uint8_t *filter_vert, int32_t height, int32_t *diff, int32_t width) {
uint32_t loop_cnt = (height >> 2);
int32_t res;
__m128i src0, src1, src2, src3, src4, src5, src6, src7;
__m128i ref0, ref1, ref2, ref3, pred0, pred1, pred2, pred3;
__m128i hz_out0, hz_out1, hz_out2, hz_out3, tmp0, tmp1;
__m128i out0, out1, out2, out3, filt_hz, filt_vt, vec, vec0, vec1;
__m128i mask = { 0x403030202010100, 0x807070606050504 };
__m128i avg = __lsx_vldi(0);
__m128i var = avg;
filt_hz = __lsx_vldrepl_h(filter_horiz, 0);
filt_vt = __lsx_vldrepl_h(filter_vert, 0);
DUP2_ARG2(__lsx_vld, src, 0, src, 8, src0, src1);
src += src_stride;
HORIZ_2TAP_FILT_UH(src0, src0, mask, filt_hz, FILTER_BITS, hz_out0);
HORIZ_2TAP_FILT_UH(src1, src1, mask, filt_hz, FILTER_BITS, hz_out2);
for (; loop_cnt--;) {
DUP2_ARG2(__lsx_vld, src, 0, src, 8, src0, src1);
src += src_stride;
DUP2_ARG2(__lsx_vld, src, 0, src, 8, src2, src3);
src += src_stride;
DUP2_ARG2(__lsx_vld, src, 0, src, 8, src4, src5);
src += src_stride;
DUP2_ARG2(__lsx_vld, src, 0, src, 8, src6, src7);
src += src_stride;
pred0 = __lsx_vld(sec_pred, 0);
sec_pred += width;
pred1 = __lsx_vld(sec_pred, 0);
sec_pred += width;
pred2 = __lsx_vld(sec_pred, 0);
sec_pred += width;
pred3 = __lsx_vld(sec_pred, 0);
sec_pred += width;
HORIZ_2TAP_FILT_UH(src0, src0, mask, filt_hz, FILTER_BITS, hz_out1);
HORIZ_2TAP_FILT_UH(src1, src1, mask, filt_hz, FILTER_BITS, hz_out3);
DUP2_ARG2(__lsx_vpackev_b, hz_out1, hz_out0, hz_out3, hz_out2, vec0, vec1);
DUP2_ARG2(__lsx_vdp2_h_bu, vec0, filt_vt, vec1, filt_vt, tmp0, tmp1);
out0 = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS);
HORIZ_2TAP_FILT_UH(src2, src2, mask, filt_hz, FILTER_BITS, hz_out0);
HORIZ_2TAP_FILT_UH(src3, src3, mask, filt_hz, FILTER_BITS, hz_out2);
DUP2_ARG2(__lsx_vpackev_b, hz_out0, hz_out1, hz_out2, hz_out3, vec0, vec1);
DUP2_ARG2(__lsx_vdp2_h_bu, vec0, filt_vt, vec1, filt_vt, tmp0, tmp1);
out1 = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS);
HORIZ_2TAP_FILT_UH(src4, src4, mask, filt_hz, FILTER_BITS, hz_out1);
HORIZ_2TAP_FILT_UH(src5, src5, mask, filt_hz, FILTER_BITS, hz_out3);
DUP2_ARG2(__lsx_vpackev_b, hz_out1, hz_out0, hz_out3, hz_out2, vec0, vec1);
DUP2_ARG2(__lsx_vdp2_h_bu, vec0, filt_vt, vec1, filt_vt, tmp0, tmp1);
out2 = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS);
HORIZ_2TAP_FILT_UH(src6, src6, mask, filt_hz, FILTER_BITS, hz_out0);
HORIZ_2TAP_FILT_UH(src7, src7, mask, filt_hz, FILTER_BITS, hz_out2);
DUP2_ARG2(__lsx_vpackev_b, hz_out0, hz_out1, hz_out2, hz_out3, vec0, vec1);
DUP2_ARG2(__lsx_vdp2_h_bu, vec0, filt_vt, vec1, filt_vt, tmp0, tmp1);
out3 = __lsx_vssrarni_bu_h(tmp1, tmp0, FILTER_BITS);
ref0 = __lsx_vld(dst, 0);
dst += dst_stride;
ref1 = __lsx_vld(dst, 0);
dst += dst_stride;
ref2 = __lsx_vld(dst, 0);
dst += dst_stride;
ref3 = __lsx_vld(dst, 0);
dst += dst_stride;
DUP4_ARG2(__lsx_vavgr_bu, out0, pred0, out1, pred1, out2, pred2, out3,
pred3, out0, out1, out2, out3);
CALC_MSE_AVG_B(out0, ref0, var, avg);
CALC_MSE_AVG_B(out1, ref1, var, avg);
CALC_MSE_AVG_B(out2, ref2, var, avg);
CALC_MSE_AVG_B(out3, ref3, var, avg);
}
vec = __lsx_vhaddw_w_h(avg, avg);
HADD_SW_S32(vec, *diff);
HADD_SW_S32(var, res);
return res;
}
static uint32_t sub_pixel_avg_sse_diff_64width_h_lsx(
const uint8_t *src, int32_t src_stride, const uint8_t *dst,
int32_t dst_stride, const uint8_t *sec_pred, const uint8_t *filter,
int32_t height, int32_t *diff) {
uint32_t loop_cnt, sse = 0;
int32_t diff0[4];
for (loop_cnt = 0; loop_cnt < 4; ++loop_cnt) {
sse +=
subpel_avg_ssediff_16w_h_lsx(src, src_stride, dst, dst_stride, sec_pred,
filter, height, &diff0[loop_cnt], 64);
src += 16;
dst += 16;
sec_pred += 16;
}
*diff = diff0[0] + diff0[1] + diff0[2] + diff0[3];
return sse;
}
static uint32_t sub_pixel_avg_sse_diff_64width_v_lsx(
const uint8_t *src, int32_t src_stride, const uint8_t *dst,
int32_t dst_stride, const uint8_t *sec_pred, const uint8_t *filter,
int32_t height, int32_t *diff) {
uint32_t loop_cnt, sse = 0;
int32_t diff0[4];
for (loop_cnt = 0; loop_cnt < 4; ++loop_cnt) {
sse +=
subpel_avg_ssediff_16w_v_lsx(src, src_stride, dst, dst_stride, sec_pred,
filter, height, &diff0[loop_cnt], 64);
src += 16;
dst += 16;
sec_pred += 16;
}
*diff = diff0[0] + diff0[1] + diff0[2] + diff0[3];
return sse;
}
static uint32_t sub_pixel_avg_sse_diff_64width_hv_lsx(
const uint8_t *src, int32_t src_stride, const uint8_t *dst,
int32_t dst_stride, const uint8_t *sec_pred, const uint8_t *filter_horiz,
const uint8_t *filter_vert, int32_t height, int32_t *diff) {
uint32_t loop_cnt, sse = 0;
int32_t diff0[4];
for (loop_cnt = 0; loop_cnt < 4; ++loop_cnt) {
sse += subpel_avg_ssediff_16w_hv_lsx(src, src_stride, dst, dst_stride,
sec_pred, filter_horiz, filter_vert,
height, &diff0[loop_cnt], 64);
src += 16;
dst += 16;
sec_pred += 16;
}
*diff = diff0[0] + diff0[1] + diff0[2] + diff0[3];
return sse;
}
#define VARIANCE_8Wx8H(sse, diff) VARIANCE_WxH(sse, diff, 6)
#define VARIANCE_16Wx16H(sse, diff) VARIANCE_WxH(sse, diff, 8)
#define VARIANCE_32Wx32H(sse, diff) VARIANCE_LARGE_WxH(sse, diff, 10)
#define VARIANCE_64Wx64H(sse, diff) VARIANCE_LARGE_WxH(sse, diff, 12)
#define VPX_SUB_PIXEL_VARIANCE_WDXHT_LSX(wd, ht) \
uint32_t vpx_sub_pixel_variance##wd##x##ht##_lsx( \
const uint8_t *src, int32_t src_stride, int32_t x_offset, \
int32_t y_offset, const uint8_t *ref, int32_t ref_stride, \
uint32_t *sse) { \
int32_t diff; \
uint32_t var; \
const uint8_t *h_filter = bilinear_filters_lsx[x_offset]; \
const uint8_t *v_filter = bilinear_filters_lsx[y_offset]; \
\
if (y_offset) { \
if (x_offset) { \
*sse = sub_pixel_sse_diff_##wd##width_hv_lsx( \
src, src_stride, ref, ref_stride, h_filter, v_filter, ht, &diff); \
} else { \
*sse = sub_pixel_sse_diff_##wd##width_v_lsx( \
src, src_stride, ref, ref_stride, v_filter, ht, &diff); \
} \
\
var = VARIANCE_##wd##Wx##ht##H(*sse, diff); \
} else { \
if (x_offset) { \
*sse = sub_pixel_sse_diff_##wd##width_h_lsx( \
src, src_stride, ref, ref_stride, h_filter, ht, &diff); \
\
var = VARIANCE_##wd##Wx##ht##H(*sse, diff); \
} else { \
var = vpx_variance##wd##x##ht##_lsx(src, src_stride, ref, ref_stride, \
sse); \
} \
} \
\
return var; \
}
VPX_SUB_PIXEL_VARIANCE_WDXHT_LSX(8, 8)
VPX_SUB_PIXEL_VARIANCE_WDXHT_LSX(16, 16)
VPX_SUB_PIXEL_VARIANCE_WDXHT_LSX(32, 32)
#define VPX_SUB_PIXEL_AVG_VARIANCE64XHEIGHT_LSX(ht) \
uint32_t vpx_sub_pixel_avg_variance64x##ht##_lsx( \
const uint8_t *src_ptr, int32_t src_stride, int32_t x_offset, \
int32_t y_offset, const uint8_t *ref_ptr, int32_t ref_stride, \
uint32_t *sse, const uint8_t *sec_pred) { \
int32_t diff; \
const uint8_t *h_filter = bilinear_filters_lsx[x_offset]; \
const uint8_t *v_filter = bilinear_filters_lsx[y_offset]; \
\
if (y_offset) { \
if (x_offset) { \
*sse = sub_pixel_avg_sse_diff_64width_hv_lsx( \
src_ptr, src_stride, ref_ptr, ref_stride, sec_pred, h_filter, \
v_filter, ht, &diff); \
} else { \
*sse = sub_pixel_avg_sse_diff_64width_v_lsx( \
src_ptr, src_stride, ref_ptr, ref_stride, sec_pred, v_filter, ht, \
&diff); \
} \
} else { \
if (x_offset) { \
*sse = sub_pixel_avg_sse_diff_64width_h_lsx( \
src_ptr, src_stride, ref_ptr, ref_stride, sec_pred, h_filter, ht, \
&diff); \
} else { \
*sse = avg_sse_diff_64x##ht##_lsx(src_ptr, src_stride, ref_ptr, \
ref_stride, sec_pred, &diff); \
} \
} \
\
return VARIANCE_64Wx##ht##H(*sse, diff); \
}
VPX_SUB_PIXEL_AVG_VARIANCE64XHEIGHT_LSX(64)