| /* |
| * Copyright (c) 2021, Alliance for Open Media. All rights reserved |
| * |
| * This source code is subject to the terms of the BSD 3-Clause Clear License |
| * and the Alliance for Open Media Patent License 1.0. If the BSD 3-Clause Clear |
| * License was not distributed with this source code in the LICENSE file, you |
| * can obtain it at aomedia.org/license/software-license/bsd-3-c-c/. 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 |
| * aomedia.org/license/patent-license/. |
| */ |
| |
| #include <emmintrin.h> |
| #include <stdio.h> |
| |
| #include "av1/common/common.h" |
| |
| int64_t av1_highbd_block_error_sse2(const tran_low_t *coeff, |
| const tran_low_t *dqcoeff, |
| intptr_t block_size, int64_t *ssz, |
| int bps) { |
| int i, j, test; |
| uint32_t temp[4]; |
| __m128i max, min, cmp0, cmp1, cmp2, cmp3; |
| int64_t error = 0, sqcoeff = 0; |
| const int shift = 2 * (bps - 8); |
| const int rounding = shift > 0 ? 1 << (shift - 1) : 0; |
| |
| for (i = 0; i < block_size; i += 8) { |
| // Load the data into xmm registers |
| __m128i mm_coeff = _mm_load_si128((__m128i *)(coeff + i)); |
| __m128i mm_coeff2 = _mm_load_si128((__m128i *)(coeff + i + 4)); |
| __m128i mm_dqcoeff = _mm_load_si128((__m128i *)(dqcoeff + i)); |
| __m128i mm_dqcoeff2 = _mm_load_si128((__m128i *)(dqcoeff + i + 4)); |
| // Check if any values require more than 15 bit |
| max = _mm_set1_epi32(0x3fff); |
| min = _mm_set1_epi32(0xffffc000); |
| cmp0 = _mm_xor_si128(_mm_cmpgt_epi32(mm_coeff, max), |
| _mm_cmplt_epi32(mm_coeff, min)); |
| cmp1 = _mm_xor_si128(_mm_cmpgt_epi32(mm_coeff2, max), |
| _mm_cmplt_epi32(mm_coeff2, min)); |
| cmp2 = _mm_xor_si128(_mm_cmpgt_epi32(mm_dqcoeff, max), |
| _mm_cmplt_epi32(mm_dqcoeff, min)); |
| cmp3 = _mm_xor_si128(_mm_cmpgt_epi32(mm_dqcoeff2, max), |
| _mm_cmplt_epi32(mm_dqcoeff2, min)); |
| test = _mm_movemask_epi8( |
| _mm_or_si128(_mm_or_si128(cmp0, cmp1), _mm_or_si128(cmp2, cmp3))); |
| |
| if (!test) { |
| __m128i mm_diff, error_sse2, sqcoeff_sse2; |
| mm_coeff = _mm_packs_epi32(mm_coeff, mm_coeff2); |
| mm_dqcoeff = _mm_packs_epi32(mm_dqcoeff, mm_dqcoeff2); |
| mm_diff = _mm_sub_epi16(mm_coeff, mm_dqcoeff); |
| error_sse2 = _mm_madd_epi16(mm_diff, mm_diff); |
| sqcoeff_sse2 = _mm_madd_epi16(mm_coeff, mm_coeff); |
| _mm_storeu_si128((__m128i *)temp, error_sse2); |
| error = error + temp[0] + temp[1] + temp[2] + temp[3]; |
| _mm_storeu_si128((__m128i *)temp, sqcoeff_sse2); |
| sqcoeff += temp[0] + temp[1] + temp[2] + temp[3]; |
| } else { |
| for (j = 0; j < 8; j++) { |
| const int64_t diff = coeff[i + j] - dqcoeff[i + j]; |
| error += diff * diff; |
| sqcoeff += (int64_t)coeff[i + j] * (int64_t)coeff[i + j]; |
| } |
| } |
| } |
| assert(error >= 0 && sqcoeff >= 0); |
| error = (error + rounding) >> shift; |
| sqcoeff = (sqcoeff + rounding) >> shift; |
| |
| *ssz = sqcoeff; |
| return error; |
| } |