xref: /aosp_15_r20/external/libaom/aom_dsp/x86/intrapred_x86.h (revision 77c1e3ccc04c968bd2bc212e87364f250e820521)
1 /*
2  * Copyright (c) 2020, Alliance for Open Media. All rights reserved.
3  *
4  * This source code is subject to the terms of the BSD 2 Clause License and
5  * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
6  * was not distributed with this source code in the LICENSE file, you can
7  * obtain it at www.aomedia.org/license/software. If the Alliance for Open
8  * Media Patent License 1.0 was not distributed with this source code in the
9  * PATENTS file, you can obtain it at www.aomedia.org/license/patent.
10  */
11 
12 #ifndef AOM_AOM_DSP_X86_INTRAPRED_X86_H_
13 #define AOM_AOM_DSP_X86_INTRAPRED_X86_H_
14 
15 #include <emmintrin.h>  // SSE2
16 #include "aom/aom_integer.h"
17 #include "config/aom_config.h"
18 
dc_sum_16_sse2(const uint8_t * ref)19 static inline __m128i dc_sum_16_sse2(const uint8_t *ref) {
20   __m128i x = _mm_load_si128((__m128i const *)ref);
21   const __m128i zero = _mm_setzero_si128();
22   x = _mm_sad_epu8(x, zero);
23   const __m128i high = _mm_unpackhi_epi64(x, x);
24   return _mm_add_epi16(x, high);
25 }
26 
dc_sum_32_sse2(const uint8_t * ref)27 static inline __m128i dc_sum_32_sse2(const uint8_t *ref) {
28   __m128i x0 = _mm_load_si128((__m128i const *)ref);
29   __m128i x1 = _mm_load_si128((__m128i const *)(ref + 16));
30   const __m128i zero = _mm_setzero_si128();
31   x0 = _mm_sad_epu8(x0, zero);
32   x1 = _mm_sad_epu8(x1, zero);
33   x0 = _mm_add_epi16(x0, x1);
34   const __m128i high = _mm_unpackhi_epi64(x0, x0);
35   return _mm_add_epi16(x0, high);
36 }
37 
38 #endif  // AOM_AOM_DSP_X86_INTRAPRED_X86_H_
39