blob: bc81692d933930e31d32ec27c595367351c23a8e [file] [log] [blame]
/*
* Copyright (c) 2019, Alliance for Open Media. 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 <arm_neon.h>
#include <assert.h>
#include "av1/common/arm/mem_neon.h"
#include "aom_dsp/aom_dsp_common.h"
int64_t av1_block_error_neon(const tran_low_t *coeff, const tran_low_t *dqcoeff,
intptr_t block_size, int64_t *ssz) {
int64x2_t error = vdupq_n_s64(0);
int64x2_t sqcoeff = vdupq_n_s64(0);
assert(block_size >= 8);
assert((block_size % 8) == 0);
do {
const int16x8_t c = load_tran_low_to_s16q(coeff);
const int16x8_t d = load_tran_low_to_s16q(dqcoeff);
const int16x8_t diff = vsubq_s16(c, d);
const int16x4_t diff_lo = vget_low_s16(diff);
const int16x4_t diff_hi = vget_high_s16(diff);
// diff is 15-bits, the squares 30, so we can store 2 in 31-bits before
// accumulating them in 64-bits.
const int32x4_t err0 = vmull_s16(diff_lo, diff_lo);
const int32x4_t err1 = vmlal_s16(err0, diff_hi, diff_hi);
const int64x2_t err2 = vaddl_s32(vget_low_s32(err1), vget_high_s32(err1));
error = vaddq_s64(error, err2);
const int16x4_t coeff_lo = vget_low_s16(c);
const int16x4_t coeff_hi = vget_high_s16(c);
const int32x4_t sqcoeff0 = vmull_s16(coeff_lo, coeff_lo);
const int32x4_t sqcoeff1 = vmlal_s16(sqcoeff0, coeff_hi, coeff_hi);
const int64x2_t sqcoeff2 =
vaddl_s32(vget_low_s32(sqcoeff1), vget_high_s32(sqcoeff1));
sqcoeff = vaddq_s64(sqcoeff, sqcoeff2);
coeff += 8;
dqcoeff += 8;
block_size -= 8;
} while (block_size != 0);
#if defined(__aarch64__)
*ssz = vaddvq_s64(sqcoeff);
return vaddvq_s64(error);
#else
*ssz = vgetq_lane_s64(sqcoeff, 0) + vgetq_lane_s64(sqcoeff, 1);
return vgetq_lane_s64(error, 0) + vgetq_lane_s64(error, 1);
#endif
}