/* * Copyright (c) 2016, Alliance for Open Media. All rights reserved * * This source code is subject to the terms of the BSD 2 Clause License and * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License * was not distributed with this source code in the LICENSE file, you can * obtain it at www.aomedia.org/license/software. 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 www.aomedia.org/license/patent. */ #include // SSE2 #include "config/aom_dsp_rtcd.h" #include "aom_dsp/x86/synonyms.h" #include "aom_ports/mem.h" #include "aom_ports/emmintrin_compat.h" #include "aom_dsp/x86/lpf_common_sse2.h" static INLINE __m128i abs_diff(__m128i a, __m128i b) { return _mm_or_si128(_mm_subs_epu8(a, b), _mm_subs_epu8(b, a)); } // this function treats its input as 2 parallel 8x4 matrices, transposes each of // them to 4x8 independently while flipping the second matrix horizontally. // Used for 14 taps pq pairs creation static INLINE void transpose_pq_14_sse2(__m128i *x0, __m128i *x1, __m128i *x2, __m128i *x3, __m128i *q0p0, __m128i *q1p1, __m128i *q2p2, __m128i *q3p3, __m128i *q4p4, __m128i *q5p5, __m128i *q6p6, __m128i *q7p7) { __m128i w0, w1, ww0, ww1, w2, w3, ww2, ww3; w0 = _mm_unpacklo_epi8( *x0, *x1); // 00 10 01 11 02 12 03 13 04 14 05 15 06 16 07 17 w1 = _mm_unpacklo_epi8( *x2, *x3); // 20 30 21 31 22 32 23 33 24 34 25 35 26 36 27 37 w2 = _mm_unpackhi_epi8( *x0, *x1); // 08 18 09 19 010 110 011 111 012 112 013 113 014 114 015 115 w3 = _mm_unpackhi_epi8( *x2, *x3); // 28 38 29 39 210 310 211 311 212 312 213 313 214 314 215 315 ww0 = _mm_unpacklo_epi16( w0, w1); // 00 10 20 30 01 11 21 31 02 12 22 32 03 13 23 33 ww1 = _mm_unpackhi_epi16( w0, w1); // 04 14 24 34 05 15 25 35 06 16 26 36 07 17 27 37 ww2 = _mm_unpacklo_epi16( w2, w3); // 08 18 28 38 09 19 29 39 010 110 210 310 011 111 211 311 ww3 = _mm_unpackhi_epi16( w2, w3); // 012 112 212 312 013 113 213 313 014 114 214 314 015 115 215 315 *q7p7 = _mm_unpacklo_epi32( ww0, _mm_srli_si128( ww3, 12)); // 00 10 20 30 015 115 215 315 xx xx xx xx xx xx xx xx *q6p6 = _mm_unpackhi_epi32( _mm_slli_si128(ww0, 4), ww3); // 01 11 21 31 014 114 214 314 xx xx xx xxxx xx xx xx *q5p5 = _mm_unpackhi_epi32( ww0, _mm_slli_si128( ww3, 4)); // 02 12 22 32 013 113 213 313 xx xx xx x xx xx xx xxx *q4p4 = _mm_unpacklo_epi32( _mm_srli_si128(ww0, 12), ww3); // 03 13 23 33 012 112 212 312 xx xx xx xx xx xx xx xx *q3p3 = _mm_unpacklo_epi32( ww1, _mm_srli_si128( ww2, 12)); // 04 14 24 34 011 111 211 311 xx xx xx xx xx xx xx xx *q2p2 = _mm_unpackhi_epi32( _mm_slli_si128(ww1, 4), ww2); // 05 15 25 35 010 110 210 310 xx xx xx xx xx xx xx xx *q1p1 = _mm_unpackhi_epi32( ww1, _mm_slli_si128( ww2, 4)); // 06 16 26 36 09 19 29 39 xx xx xx xx xx xx xx xx *q0p0 = _mm_unpacklo_epi32( _mm_srli_si128(ww1, 12), ww2); // 07 17 27 37 08 18 28 38 xx xx xx xx xx xx xx xx } // this function treats its input as 2 parallel 8x4 matrices, transposes each of // them independently while flipping the second matrix horizontaly Used for 14 // taps filter pq pairs inverse static INLINE void transpose_pq_14_inv_sse2(__m128i *x0, __m128i *x1, __m128i *x2, __m128i *x3, __m128i *x4, __m128i *x5, __m128i *x6, __m128i *x7, __m128i *pq0, __m128i *pq1, __m128i *pq2, __m128i *pq3) { __m128i w10, w11, w12, w13; __m128i w0, w1, w2, w3, w4, w5; __m128i d0, d1, d2, d3; w0 = _mm_unpacklo_epi8( *x0, *x1); // p 00 10 01 11 02 12 03 13 04 14 05 15 06 16 07 17 w1 = _mm_unpacklo_epi8( *x2, *x3); // p 20 30 21 31 22 32 23 33 24 34 25 35 26 36 27 37 w2 = _mm_unpacklo_epi8( *x4, *x5); // p 40 50 41 51 42 52 43 53 44 54 45 55 46 56 47 57 w3 = _mm_unpacklo_epi8( *x6, *x7); // p 60 70 61 71 62 72 63 73 64 74 65 75 66 76 67 77 w4 = _mm_unpacklo_epi16( w0, w1); // 00 10 20 30 01 11 21 31 02 12 22 32 03 13 23 33 w5 = _mm_unpacklo_epi16( w2, w3); // 40 50 60 70 41 51 61 71 42 52 62 72 43 53 63 73 d0 = _mm_unpacklo_epi32( w4, w5); // 00 10 20 30 40 50 60 70 01 11 21 31 41 51 61 71 d2 = _mm_unpackhi_epi32( w4, w5); // 02 12 22 32 42 52 62 72 03 13 23 33 43 53 63 73 w10 = _mm_unpacklo_epi8( *x7, *x6); // q xx xx xx xx xx xx xx xx 00 10 01 11 02 12 03 13 w11 = _mm_unpacklo_epi8( *x5, *x4); // q xx xx xx xx xx xx xx xx 20 30 21 31 22 32 23 33 w12 = _mm_unpacklo_epi8( *x3, *x2); // q xx xx xx xx xx xx xx xx 40 50 41 51 42 52 43 53 w13 = _mm_unpacklo_epi8( *x1, *x0); // q xx xx xx xx xx xx xx xx 60 70 61 71 62 72 63 73 w4 = _mm_unpackhi_epi16( w10, w11); // 00 10 20 30 01 11 21 31 02 12 22 32 03 13 23 33 w5 = _mm_unpackhi_epi16( w12, w13); // 40 50 60 70 41 51 61 71 42 52 62 72 43 53 63 73 d1 = _mm_unpacklo_epi32( w4, w5); // 00 10 20 30 40 50 60 70 01 11 21 31 41 51 61 71 d3 = _mm_unpackhi_epi32( w4, w5); // 02 12 22 32 42 52 62 72 03 13 23 33 43 53 63 73 *pq0 = _mm_unpacklo_epi64(d0, d1); // pq *pq1 = _mm_unpackhi_epi64(d0, d1); // pq *pq2 = _mm_unpacklo_epi64(d2, d3); // pq *pq3 = _mm_unpackhi_epi64(d2, d3); // pq } static AOM_FORCE_INLINE void filter4_sse2(__m128i *p1p0, __m128i *q1q0, __m128i *hev, __m128i *mask, __m128i *qs1qs0, __m128i *ps1ps0) { __m128i filter, filter2filter1, work; __m128i ps1ps0_work, qs1qs0_work; __m128i hev1; const __m128i t3t4 = _mm_set_epi8(0, 0, 0, 0, 0, 0, 0, 0, 3, 3, 3, 3, 4, 4, 4, 4); const __m128i t80 = _mm_set1_epi8((char)0x80); const __m128i ff = _mm_cmpeq_epi8(t80, t80); ps1ps0_work = _mm_xor_si128(*p1p0, t80); /* ^ 0x80 */ qs1qs0_work = _mm_xor_si128(*q1q0, t80); /* int8_t filter = signed_char_clamp(ps1 - qs1) & hev; */ work = _mm_subs_epi8(ps1ps0_work, qs1qs0_work); filter = _mm_and_si128(_mm_srli_si128(work, 4), *hev); /* filter = signed_char_clamp(filter + 3 * (qs0 - ps0)) & mask; */ filter = _mm_subs_epi8(filter, work); filter = _mm_subs_epi8(filter, work); filter = _mm_subs_epi8(filter, work); /* + 3 * (qs0 - ps0) */ filter = _mm_and_si128(filter, *mask); /* & mask */ filter = _mm_unpacklo_epi32(filter, filter); /* filter1 = signed_char_clamp(filter + 4) >> 3; */ /* filter2 = signed_char_clamp(filter + 3) >> 3; */ filter2filter1 = _mm_adds_epi8(filter, t3t4); /* signed_char_clamp */ filter2filter1 = _mm_unpacklo_epi8(filter2filter1, filter2filter1); // goto 16 bit filter2filter1 = _mm_srai_epi16(filter2filter1, 11); /* >> 3 */ filter2filter1 = _mm_packs_epi16(filter2filter1, filter2filter1); /* filter = ROUND_POWER_OF_TWO(filter1, 1) & ~hev; */ filter = _mm_subs_epi8(filter2filter1, ff); /* + 1 */ filter = _mm_unpacklo_epi8(filter, filter); // goto 16 bit filter = _mm_srai_epi16(filter, 9); /* round */ filter = _mm_packs_epi16(filter, filter); filter = _mm_andnot_si128(*hev, filter); filter = _mm_unpacklo_epi32(filter, filter); filter2filter1 = _mm_unpacklo_epi32(filter2filter1, filter); hev1 = _mm_srli_si128(filter2filter1, 8); /* signed_char_clamp(qs1 - filter), signed_char_clamp(qs0 - filter1) */ qs1qs0_work = _mm_subs_epi8(qs1qs0_work, filter2filter1); /* signed_char_clamp(ps1 + filter), signed_char_clamp(ps0 + filter2) */ ps1ps0_work = _mm_adds_epi8(ps1ps0_work, hev1); *qs1qs0 = _mm_xor_si128(qs1qs0_work, t80); /* ^ 0x80 */ *ps1ps0 = _mm_xor_si128(ps1ps0_work, t80); /* ^ 0x80 */ } static AOM_FORCE_INLINE void filter4_dual_sse2(__m128i *p1p0, __m128i *q1q0, __m128i *hev, __m128i *mask, __m128i *qs1qs0, __m128i *ps1ps0) { const __m128i t3t4 = _mm_set_epi8(3, 3, 3, 3, 3, 3, 3, 3, 4, 4, 4, 4, 4, 4, 4, 4); const __m128i t80 = _mm_set1_epi8((char)0x80); __m128i filter, filter2filter1, work; __m128i ps1ps0_work, qs1qs0_work; __m128i hev1; const __m128i ff = _mm_cmpeq_epi8(t80, t80); ps1ps0_work = _mm_xor_si128(*p1p0, t80); /* ^ 0x80 */ qs1qs0_work = _mm_xor_si128(*q1q0, t80); /* int8_t filter = signed_char_clamp(ps1 - qs1) & hev; */ work = _mm_subs_epi8(ps1ps0_work, qs1qs0_work); filter = _mm_and_si128(_mm_srli_si128(work, 8), *hev); /* filter = signed_char_clamp(filter + 3 * (qs0 - ps0)) & mask; */ filter = _mm_subs_epi8(filter, work); filter = _mm_subs_epi8(filter, work); filter = _mm_subs_epi8(filter, work); /* + 3 * (qs0 - ps0) */ filter = _mm_and_si128(filter, *mask); /* & mask */ filter = _mm_unpacklo_epi64(filter, filter); /* filter1 = signed_char_clamp(filter + 4) >> 3; */ /* filter2 = signed_char_clamp(filter + 3) >> 3; */ filter2filter1 = _mm_adds_epi8(filter, t3t4); /* signed_char_clamp */ filter = _mm_unpackhi_epi8(filter2filter1, filter2filter1); filter2filter1 = _mm_unpacklo_epi8(filter2filter1, filter2filter1); filter2filter1 = _mm_srai_epi16(filter2filter1, 11); /* >> 3 */ filter = _mm_srai_epi16(filter, 11); /* >> 3 */ filter2filter1 = _mm_packs_epi16(filter2filter1, filter); /* filter = ROUND_POWER_OF_TWO(filter1, 1) & ~hev; */ filter = _mm_subs_epi8(filter2filter1, ff); /* + 1 */ filter = _mm_unpacklo_epi8(filter, filter); filter = _mm_srai_epi16(filter, 9); /* round */ filter = _mm_packs_epi16(filter, filter); filter = _mm_andnot_si128(*hev, filter); hev1 = _mm_unpackhi_epi64(filter2filter1, filter); filter2filter1 = _mm_unpacklo_epi64(filter2filter1, filter); /* signed_char_clamp(qs1 - filter), signed_char_clamp(qs0 - filter1) */ qs1qs0_work = _mm_subs_epi8(qs1qs0_work, filter2filter1); /* signed_char_clamp(ps1 + filter), signed_char_clamp(ps0 + filter2) */ ps1ps0_work = _mm_adds_epi8(ps1ps0_work, hev1); *qs1qs0 = _mm_xor_si128(qs1qs0_work, t80); /* ^ 0x80 */ *ps1ps0 = _mm_xor_si128(ps1ps0_work, t80); /* ^ 0x80 */ } static AOM_FORCE_INLINE void lpf_internal_4_sse2( __m128i *p1, __m128i *p0, __m128i *q0, __m128i *q1, __m128i *limit, __m128i *thresh, __m128i *q1q0_out, __m128i *p1p0_out) { __m128i q1p1, q0p0, p1p0, q1q0; __m128i abs_p0q0, abs_p1q1; __m128i mask, flat, hev; const __m128i zero = _mm_setzero_si128(); q1p1 = _mm_unpacklo_epi32(*p1, *q1); q0p0 = _mm_unpacklo_epi32(*p0, *q0); p1p0 = _mm_unpacklo_epi32(q0p0, q1p1); q1q0 = _mm_srli_si128(p1p0, 8); /* (abs(q1 - q0), abs(p1 - p0) */ flat = abs_diff(q1p1, q0p0); /* abs(p1 - q1), abs(p0 - q0) */ __m128i abs_p1q1p0q0 = abs_diff(p1p0, q1q0); /* const uint8_t hev = hev_mask(thresh, *op1, *op0, *oq0, *oq1); */ flat = _mm_max_epu8(flat, _mm_srli_si128(flat, 4)); hev = _mm_unpacklo_epi8(flat, zero); hev = _mm_cmpgt_epi16(hev, *thresh); hev = _mm_packs_epi16(hev, hev); hev = _mm_unpacklo_epi32(hev, hev); abs_p0q0 = _mm_adds_epu8(abs_p1q1p0q0, abs_p1q1p0q0); /* abs(p0 - q0) * 2 */ abs_p1q1 = _mm_srli_si128(abs_p1q1p0q0, 4); /* abs(p1 - q1) */ abs_p1q1 = _mm_unpacklo_epi8(abs_p1q1, abs_p1q1); abs_p1q1 = _mm_srli_epi16(abs_p1q1, 9); abs_p1q1 = _mm_packs_epi16(abs_p1q1, abs_p1q1); /* abs(p1 - q1) / 2 */ /* abs(p0 - q0) * 2 + abs(p1 - q1) / 2 */ mask = _mm_adds_epu8(abs_p0q0, abs_p1q1); mask = _mm_unpacklo_epi32(mask, flat); mask = _mm_subs_epu8(mask, *limit); mask = _mm_cmpeq_epi8(mask, zero); mask = _mm_and_si128(mask, _mm_srli_si128(mask, 4)); filter4_sse2(&p1p0, &q1q0, &hev, &mask, q1q0_out, p1p0_out); } static AOM_FORCE_INLINE void lpf_internal_4_dual_sse2( __m128i *p1, __m128i *p0, __m128i *q0, __m128i *q1, __m128i *limit, __m128i *thresh, __m128i *q1q0_out, __m128i *p1p0_out) { __m128i q1p1, q0p0, p1p0, q1q0; __m128i abs_p0q0, abs_p1q1; __m128i mask, hev; const __m128i zero = _mm_setzero_si128(); q1p1 = _mm_unpacklo_epi64(*p1, *q1); q0p0 = _mm_unpacklo_epi64(*p0, *q0); p1p0 = _mm_unpacklo_epi64(q0p0, q1p1); q1q0 = _mm_unpackhi_epi64(q0p0, q1p1); /* (abs(q1 - q0), abs(p1 - p0) */ __m128i flat = abs_diff(q1p1, q0p0); /* abs(p1 - q1), abs(p0 - q0) */ const __m128i abs_p1q1p0q0 = abs_diff(p1p0, q1q0); /* const uint8_t hev = hev_mask(thresh, *op1, *op0, *oq0, *oq1); */ flat = _mm_max_epu8(flat, _mm_srli_si128(flat, 8)); hev = _mm_unpacklo_epi8(flat, zero); hev = _mm_cmpgt_epi16(hev, *thresh); hev = _mm_packs_epi16(hev, hev); /* const int8_t mask = filter_mask2(*limit, *blimit, */ /* p1, p0, q0, q1); */ abs_p0q0 = _mm_adds_epu8(abs_p1q1p0q0, abs_p1q1p0q0); /* abs(p0 - q0) * 2 */ abs_p1q1 = _mm_unpackhi_epi8(abs_p1q1p0q0, abs_p1q1p0q0); /* abs(p1 - q1) */ abs_p1q1 = _mm_srli_epi16(abs_p1q1, 9); abs_p1q1 = _mm_packs_epi16(abs_p1q1, abs_p1q1); /* abs(p1 - q1) / 2 */ /* abs(p0 - q0) * 2 + abs(p1 - q1) / 2 */ mask = _mm_adds_epu8(abs_p0q0, abs_p1q1); mask = _mm_unpacklo_epi64(mask, flat); mask = _mm_subs_epu8(mask, *limit); mask = _mm_cmpeq_epi8(mask, zero); mask = _mm_and_si128(mask, _mm_srli_si128(mask, 8)); filter4_dual_sse2(&p1p0, &q1q0, &hev, &mask, q1q0_out, p1p0_out); } void aom_lpf_horizontal_4_sse2(uint8_t *s, int p /* pitch */, const uint8_t *_blimit, const uint8_t *_limit, const uint8_t *_thresh) { const __m128i zero = _mm_setzero_si128(); __m128i limit = _mm_unpacklo_epi32(_mm_loadl_epi64((const __m128i *)_blimit), _mm_loadl_epi64((const __m128i *)_limit)); __m128i thresh = _mm_unpacklo_epi8(_mm_loadl_epi64((const __m128i *)_thresh), zero); __m128i qs1qs0, ps1ps0; __m128i p1, p0, q0, q1; p1 = xx_loadl_32(s - 2 * p); p0 = xx_loadl_32(s - 1 * p); q0 = xx_loadl_32(s - 0 * p); q1 = xx_loadl_32(s + 1 * p); lpf_internal_4_sse2(&p1, &p0, &q0, &q1, &limit, &thresh, &qs1qs0, &ps1ps0); xx_storel_32(s - 1 * p, ps1ps0); xx_storel_32(s - 2 * p, _mm_srli_si128(ps1ps0, 4)); xx_storel_32(s + 0 * p, qs1qs0); xx_storel_32(s + 1 * p, _mm_srli_si128(qs1qs0, 4)); } void aom_lpf_vertical_4_sse2(uint8_t *s, int p /* pitch */, const uint8_t *_blimit, const uint8_t *_limit, const uint8_t *_thresh) { __m128i p1p0, q1q0; __m128i p1, p0, q0, q1; const __m128i zero = _mm_setzero_si128(); __m128i limit = _mm_unpacklo_epi32(_mm_loadl_epi64((const __m128i *)_blimit), _mm_loadl_epi64((const __m128i *)_limit)); __m128i thresh = _mm_unpacklo_epi8(_mm_loadl_epi64((const __m128i *)_thresh), zero); __m128i x0, x1, x2, x3; __m128i d0, d1, d2, d3; x0 = _mm_loadl_epi64((__m128i *)(s - 2 + 0 * p)); x1 = _mm_loadl_epi64((__m128i *)(s - 2 + 1 * p)); x2 = _mm_loadl_epi64((__m128i *)(s - 2 + 2 * p)); x3 = _mm_loadl_epi64((__m128i *)(s - 2 + 3 * p)); transpose4x8_8x4_low_sse2(&x0, &x1, &x2, &x3, &p1, &p0, &q0, &q1); lpf_internal_4_sse2(&p1, &p0, &q0, &q1, &limit, &thresh, &q1q0, &p1p0); // Transpose 8x4 to 4x8 p1 = _mm_srli_si128(p1p0, 4); q1 = _mm_srli_si128(q1q0, 4); transpose4x8_8x4_low_sse2(&p1, &p1p0, &q1q0, &q1, &d0, &d1, &d2, &d3); xx_storel_32(s + 0 * p - 2, d0); xx_storel_32(s + 1 * p - 2, d1); xx_storel_32(s + 2 * p - 2, d2); xx_storel_32(s + 3 * p - 2, d3); } static INLINE void store_buffer_horz_8(__m128i x, int p, int num, uint8_t *s) { xx_storel_32(s - (num + 1) * p, x); xx_storel_32(s + num * p, _mm_srli_si128(x, 4)); } static AOM_FORCE_INLINE void lpf_internal_14_dual_sse2( __m128i *q6p6, __m128i *q5p5, __m128i *q4p4, __m128i *q3p3, __m128i *q2p2, __m128i *q1p1, __m128i *q0p0, __m128i *blimit, __m128i *limit, __m128i *thresh) { const __m128i zero = _mm_setzero_si128(); const __m128i one = _mm_set1_epi8(1); __m128i mask, hev, flat, flat2; __m128i qs0ps0, qs1ps1; __m128i p1p0, q1q0, qs1qs0, ps1ps0; __m128i abs_p1p0; p1p0 = _mm_unpacklo_epi64(*q0p0, *q1p1); q1q0 = _mm_unpackhi_epi64(*q0p0, *q1p1); { __m128i abs_p1q1, abs_p0q0, abs_q1q0; __m128i fe, ff, work; abs_p1p0 = abs_diff(*q1p1, *q0p0); abs_q1q0 = _mm_srli_si128(abs_p1p0, 8); fe = _mm_set1_epi8((char)0xfe); ff = _mm_cmpeq_epi8(abs_p1p0, abs_p1p0); abs_p0q0 = abs_diff(p1p0, q1q0); abs_p1q1 = _mm_srli_si128(abs_p0q0, 8); abs_p0q0 = _mm_unpacklo_epi64(abs_p0q0, zero); flat = _mm_max_epu8(abs_p1p0, abs_q1q0); hev = _mm_subs_epu8(flat, *thresh); hev = _mm_xor_si128(_mm_cmpeq_epi8(hev, zero), ff); // replicate for the further "merged variables" usage hev = _mm_unpacklo_epi64(hev, hev); abs_p0q0 = _mm_adds_epu8(abs_p0q0, abs_p0q0); abs_p1q1 = _mm_srli_epi16(_mm_and_si128(abs_p1q1, fe), 1); mask = _mm_subs_epu8(_mm_adds_epu8(abs_p0q0, abs_p1q1), *blimit); mask = _mm_xor_si128(_mm_cmpeq_epi8(mask, zero), ff); // mask |= (abs(p0 - q0) * 2 + abs(p1 - q1) / 2 > blimit) * -1; mask = _mm_max_epu8(abs_p1p0, mask); // mask |= (abs(p1 - p0) > limit) * -1; // mask |= (abs(q1 - q0) > limit) * -1; work = _mm_max_epu8(abs_diff(*q2p2, *q1p1), abs_diff(*q3p3, *q2p2)); mask = _mm_max_epu8(work, mask); mask = _mm_max_epu8(mask, _mm_srli_si128(mask, 8)); mask = _mm_subs_epu8(mask, *limit); mask = _mm_cmpeq_epi8(mask, zero); } // lp filter - the same for 6, 8 and 14 versions filter4_dual_sse2(&p1p0, &q1q0, &hev, &mask, &qs1qs0, &ps1ps0); qs0ps0 = _mm_unpacklo_epi64(ps1ps0, qs1qs0); qs1ps1 = _mm_unpackhi_epi64(ps1ps0, qs1qs0); // loopfilter done __m128i flat2_q5p5, flat2_q4p4, flat2_q3p3, flat2_q2p2; __m128i flat2_q1p1, flat2_q0p0, flat_q2p2, flat_q1p1, flat_q0p0; __m128i work; flat = _mm_max_epu8(abs_diff(*q2p2, *q0p0), abs_diff(*q3p3, *q0p0)); flat = _mm_max_epu8(abs_p1p0, flat); flat = _mm_max_epu8(flat, _mm_srli_si128(flat, 8)); flat = _mm_subs_epu8(flat, one); flat = _mm_cmpeq_epi8(flat, zero); flat = _mm_and_si128(flat, mask); // if flat ==0 then flat2 is zero as well and we don't need any calc below // sse4.1 if (0==_mm_test_all_zeros(flat,ff)) if (0xffff != _mm_movemask_epi8(_mm_cmpeq_epi8(flat, zero))) { // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ // flat and wide flat calculations const __m128i eight = _mm_set1_epi16(8); const __m128i four = _mm_set1_epi16(4); __m128i p6_16, p5_16, p4_16, p3_16, p2_16, p1_16, p0_16; __m128i q6_16, q5_16, q4_16, q3_16, q2_16, q1_16, q0_16; __m128i pixelFilter_p, pixelFilter_q; __m128i pixetFilter_p2p1p0, pixetFilter_q2q1q0; __m128i sum_p6, sum_q6; __m128i sum_p3, sum_q3, res_p, res_q; p6_16 = _mm_unpacklo_epi8(*q6p6, zero); p5_16 = _mm_unpacklo_epi8(*q5p5, zero); p4_16 = _mm_unpacklo_epi8(*q4p4, zero); p3_16 = _mm_unpacklo_epi8(*q3p3, zero); p2_16 = _mm_unpacklo_epi8(*q2p2, zero); p1_16 = _mm_unpacklo_epi8(*q1p1, zero); p0_16 = _mm_unpacklo_epi8(*q0p0, zero); q0_16 = _mm_unpackhi_epi8(*q0p0, zero); q1_16 = _mm_unpackhi_epi8(*q1p1, zero); q2_16 = _mm_unpackhi_epi8(*q2p2, zero); q3_16 = _mm_unpackhi_epi8(*q3p3, zero); q4_16 = _mm_unpackhi_epi8(*q4p4, zero); q5_16 = _mm_unpackhi_epi8(*q5p5, zero); q6_16 = _mm_unpackhi_epi8(*q6p6, zero); pixelFilter_p = _mm_add_epi16(p5_16, _mm_add_epi16(p4_16, p3_16)); pixelFilter_q = _mm_add_epi16(q5_16, _mm_add_epi16(q4_16, q3_16)); pixetFilter_p2p1p0 = _mm_add_epi16(p0_16, _mm_add_epi16(p2_16, p1_16)); pixelFilter_p = _mm_add_epi16(pixelFilter_p, pixetFilter_p2p1p0); pixetFilter_q2q1q0 = _mm_add_epi16(q0_16, _mm_add_epi16(q2_16, q1_16)); pixelFilter_q = _mm_add_epi16(pixelFilter_q, pixetFilter_q2q1q0); pixelFilter_p = _mm_add_epi16(eight, _mm_add_epi16(pixelFilter_p, pixelFilter_q)); pixetFilter_p2p1p0 = _mm_add_epi16( four, _mm_add_epi16(pixetFilter_p2p1p0, pixetFilter_q2q1q0)); res_p = _mm_srli_epi16( _mm_add_epi16(pixelFilter_p, _mm_add_epi16(_mm_add_epi16(p6_16, p0_16), _mm_add_epi16(p1_16, q0_16))), 4); res_q = _mm_srli_epi16( _mm_add_epi16(pixelFilter_p, _mm_add_epi16(_mm_add_epi16(q6_16, q0_16), _mm_add_epi16(p0_16, q1_16))), 4); flat2_q0p0 = _mm_packus_epi16(res_p, res_q); res_p = _mm_srli_epi16( _mm_add_epi16(pixetFilter_p2p1p0, _mm_add_epi16(p3_16, p0_16)), 3); res_q = _mm_srli_epi16( _mm_add_epi16(pixetFilter_p2p1p0, _mm_add_epi16(q3_16, q0_16)), 3); flat_q0p0 = _mm_packus_epi16(res_p, res_q); sum_p6 = _mm_add_epi16(p6_16, p6_16); sum_q6 = _mm_add_epi16(q6_16, q6_16); sum_p3 = _mm_add_epi16(p3_16, p3_16); sum_q3 = _mm_add_epi16(q3_16, q3_16); pixelFilter_q = _mm_sub_epi16(pixelFilter_p, p5_16); pixelFilter_p = _mm_sub_epi16(pixelFilter_p, q5_16); res_p = _mm_srli_epi16( _mm_add_epi16( pixelFilter_p, _mm_add_epi16(sum_p6, _mm_add_epi16(p1_16, _mm_add_epi16(p2_16, p0_16)))), 4); res_q = _mm_srli_epi16( _mm_add_epi16( pixelFilter_q, _mm_add_epi16(sum_q6, _mm_add_epi16(q1_16, _mm_add_epi16(q0_16, q2_16)))), 4); flat2_q1p1 = _mm_packus_epi16(res_p, res_q); pixetFilter_q2q1q0 = _mm_sub_epi16(pixetFilter_p2p1p0, p2_16); pixetFilter_p2p1p0 = _mm_sub_epi16(pixetFilter_p2p1p0, q2_16); res_p = _mm_srli_epi16( _mm_add_epi16(pixetFilter_p2p1p0, _mm_add_epi16(sum_p3, p1_16)), 3); res_q = _mm_srli_epi16( _mm_add_epi16(pixetFilter_q2q1q0, _mm_add_epi16(sum_q3, q1_16)), 3); flat_q1p1 = _mm_packus_epi16(res_p, res_q); pixetFilter_p2p1p0 = _mm_sub_epi16(pixetFilter_p2p1p0, q1_16); pixetFilter_q2q1q0 = _mm_sub_epi16(pixetFilter_q2q1q0, p1_16); sum_p3 = _mm_add_epi16(sum_p3, p3_16); sum_q3 = _mm_add_epi16(sum_q3, q3_16); res_p = _mm_srli_epi16( _mm_add_epi16(pixetFilter_p2p1p0, _mm_add_epi16(sum_p3, p2_16)), 3); res_q = _mm_srli_epi16( _mm_add_epi16(pixetFilter_q2q1q0, _mm_add_epi16(sum_q3, q2_16)), 3); flat_q2p2 = _mm_packus_epi16(res_p, res_q); // work with flat2 flat2 = _mm_max_epu8(abs_diff(*q4p4, *q0p0), abs_diff(*q5p5, *q0p0)); work = abs_diff(*q6p6, *q0p0); flat2 = _mm_max_epu8(work, flat2); flat2 = _mm_max_epu8(flat2, _mm_srli_si128(flat2, 8)); flat2 = _mm_subs_epu8(flat2, one); flat2 = _mm_cmpeq_epi8(flat2, zero); flat2 = _mm_and_si128(flat2, flat); // flat2 & flat & mask // ~~~~~~~~~~ apply flat ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ flat = _mm_unpacklo_epi64(flat, flat); *q2p2 = _mm_andnot_si128(flat, *q2p2); flat_q2p2 = _mm_and_si128(flat, flat_q2p2); *q2p2 = _mm_or_si128(*q2p2, flat_q2p2); qs1ps1 = _mm_andnot_si128(flat, qs1ps1); flat_q1p1 = _mm_and_si128(flat, flat_q1p1); *q1p1 = _mm_or_si128(qs1ps1, flat_q1p1); qs0ps0 = _mm_andnot_si128(flat, qs0ps0); flat_q0p0 = _mm_and_si128(flat, flat_q0p0); *q0p0 = _mm_or_si128(qs0ps0, flat_q0p0); if (0xffff != _mm_movemask_epi8(_mm_cmpeq_epi8(flat2, zero))) { pixelFilter_p = _mm_sub_epi16(pixelFilter_p, q4_16); pixelFilter_q = _mm_sub_epi16(pixelFilter_q, p4_16); sum_p6 = _mm_add_epi16(sum_p6, p6_16); sum_q6 = _mm_add_epi16(sum_q6, q6_16); res_p = _mm_srli_epi16( _mm_add_epi16( pixelFilter_p, _mm_add_epi16(sum_p6, _mm_add_epi16(p2_16, _mm_add_epi16(p3_16, p1_16)))), 4); res_q = _mm_srli_epi16( _mm_add_epi16( pixelFilter_q, _mm_add_epi16(sum_q6, _mm_add_epi16(q2_16, _mm_add_epi16(q1_16, q3_16)))), 4); flat2_q2p2 = _mm_packus_epi16(res_p, res_q); sum_p6 = _mm_add_epi16(sum_p6, p6_16); sum_q6 = _mm_add_epi16(sum_q6, q6_16); pixelFilter_p = _mm_sub_epi16(pixelFilter_p, q3_16); pixelFilter_q = _mm_sub_epi16(pixelFilter_q, p3_16); res_p = _mm_srli_epi16( _mm_add_epi16( pixelFilter_p, _mm_add_epi16(sum_p6, _mm_add_epi16(p3_16, _mm_add_epi16(p4_16, p2_16)))), 4); res_q = _mm_srli_epi16( _mm_add_epi16( pixelFilter_q, _mm_add_epi16(sum_q6, _mm_add_epi16(q3_16, _mm_add_epi16(q2_16, q4_16)))), 4); flat2_q3p3 = _mm_packus_epi16(res_p, res_q); sum_p6 = _mm_add_epi16(sum_p6, p6_16); sum_q6 = _mm_add_epi16(sum_q6, q6_16); pixelFilter_p = _mm_sub_epi16(pixelFilter_p, q2_16); pixelFilter_q = _mm_sub_epi16(pixelFilter_q, p2_16); res_p = _mm_srli_epi16( _mm_add_epi16( pixelFilter_p, _mm_add_epi16(sum_p6, _mm_add_epi16(p4_16, _mm_add_epi16(p5_16, p3_16)))), 4); res_q = _mm_srli_epi16( _mm_add_epi16( pixelFilter_q, _mm_add_epi16(sum_q6, _mm_add_epi16(q4_16, _mm_add_epi16(q3_16, q5_16)))), 4); flat2_q4p4 = _mm_packus_epi16(res_p, res_q); sum_p6 = _mm_add_epi16(sum_p6, p6_16); sum_q6 = _mm_add_epi16(sum_q6, q6_16); pixelFilter_p = _mm_sub_epi16(pixelFilter_p, q1_16); pixelFilter_q = _mm_sub_epi16(pixelFilter_q, p1_16); res_p = _mm_srli_epi16( _mm_add_epi16( pixelFilter_p, _mm_add_epi16(sum_p6, _mm_add_epi16(p5_16, _mm_add_epi16(p6_16, p4_16)))), 4); res_q = _mm_srli_epi16( _mm_add_epi16( pixelFilter_q, _mm_add_epi16(sum_q6, _mm_add_epi16(q5_16, _mm_add_epi16(q6_16, q4_16)))), 4); flat2_q5p5 = _mm_packus_epi16(res_p, res_q); // wide flat // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ flat2 = _mm_unpacklo_epi64(flat2, flat2); *q5p5 = _mm_andnot_si128(flat2, *q5p5); flat2_q5p5 = _mm_and_si128(flat2, flat2_q5p5); *q5p5 = _mm_or_si128(*q5p5, flat2_q5p5); *q4p4 = _mm_andnot_si128(flat2, *q4p4); flat2_q4p4 = _mm_and_si128(flat2, flat2_q4p4); *q4p4 = _mm_or_si128(*q4p4, flat2_q4p4); *q3p3 = _mm_andnot_si128(flat2, *q3p3); flat2_q3p3 = _mm_and_si128(flat2, flat2_q3p3); *q3p3 = _mm_or_si128(*q3p3, flat2_q3p3); *q2p2 = _mm_andnot_si128(flat2, *q2p2); flat2_q2p2 = _mm_and_si128(flat2, flat2_q2p2); *q2p2 = _mm_or_si128(*q2p2, flat2_q2p2); *q1p1 = _mm_andnot_si128(flat2, *q1p1); flat2_q1p1 = _mm_and_si128(flat2, flat2_q1p1); *q1p1 = _mm_or_si128(*q1p1, flat2_q1p1); *q0p0 = _mm_andnot_si128(flat2, *q0p0); flat2_q0p0 = _mm_and_si128(flat2, flat2_q0p0); *q0p0 = _mm_or_si128(*q0p0, flat2_q0p0); } } else { *q0p0 = qs0ps0; *q1p1 = qs1ps1; } } static AOM_FORCE_INLINE void lpf_internal_14_sse2( __m128i *q6p6, __m128i *q5p5, __m128i *q4p4, __m128i *q3p3, __m128i *q2p2, __m128i *q1p1, __m128i *q0p0, __m128i *blimit, __m128i *limit, __m128i *thresh) { const __m128i zero = _mm_setzero_si128(); const __m128i one = _mm_set1_epi8(1); __m128i mask, hev, flat, flat2; __m128i flat2_pq[6], flat_pq[3]; __m128i qs0ps0, qs1ps1; __m128i p1p0, q1q0, qs1qs0, ps1ps0; __m128i abs_p1p0; p1p0 = _mm_unpacklo_epi32(*q0p0, *q1p1); q1q0 = _mm_srli_si128(p1p0, 8); __m128i fe, ff, work; { __m128i abs_p1q1, abs_p0q0, abs_q1q0; abs_p1p0 = abs_diff(*q1p1, *q0p0); abs_q1q0 = _mm_srli_si128(abs_p1p0, 4); fe = _mm_set1_epi8((char)0xfe); ff = _mm_cmpeq_epi8(fe, fe); abs_p0q0 = abs_diff(p1p0, q1q0); abs_p1q1 = _mm_srli_si128(abs_p0q0, 4); flat = _mm_max_epu8(abs_p1p0, abs_q1q0); hev = _mm_subs_epu8(flat, *thresh); hev = _mm_xor_si128(_mm_cmpeq_epi8(hev, zero), ff); // replicate for the further "merged variables" usage hev = _mm_unpacklo_epi32(hev, hev); abs_p0q0 = _mm_adds_epu8(abs_p0q0, abs_p0q0); abs_p1q1 = _mm_srli_epi16(_mm_and_si128(abs_p1q1, fe), 1); mask = _mm_subs_epu8(_mm_adds_epu8(abs_p0q0, abs_p1q1), *blimit); mask = _mm_unpacklo_epi32(mask, zero); mask = _mm_xor_si128(_mm_cmpeq_epi8(mask, zero), ff); // mask |= (abs(p0 - q0) * 2 + abs(p1 - q1) / 2 > blimit) * -1; mask = _mm_max_epu8(abs_p1p0, mask); // mask |= (abs(p1 - p0) > limit) * -1; // mask |= (abs(q1 - q0) > limit) * -1; work = _mm_max_epu8(abs_diff(*q2p2, *q1p1), abs_diff(*q3p3, *q2p2)); mask = _mm_max_epu8(work, mask); mask = _mm_max_epu8(mask, _mm_srli_si128(mask, 4)); mask = _mm_subs_epu8(mask, *limit); mask = _mm_cmpeq_epi8(mask, zero); } // lp filter - the same for 6, 8 and 14 versions filter4_sse2(&p1p0, &q1q0, &hev, &mask, &qs1qs0, &ps1ps0); qs0ps0 = _mm_unpacklo_epi32(ps1ps0, qs1qs0); qs1ps1 = _mm_srli_si128(qs0ps0, 8); // loopfilter done flat = _mm_max_epu8(abs_diff(*q2p2, *q0p0), abs_diff(*q3p3, *q0p0)); flat = _mm_max_epu8(abs_p1p0, flat); flat = _mm_max_epu8(flat, _mm_srli_si128(flat, 4)); flat = _mm_subs_epu8(flat, one); flat = _mm_cmpeq_epi8(flat, zero); flat = _mm_and_si128(flat, mask); flat = _mm_unpacklo_epi32(flat, flat); flat = _mm_unpacklo_epi64(flat, flat); // if flat ==0 then flat2 is zero as well and we don't need any calc below // sse4.1 if (0==_mm_test_all_zeros(flat,ff)) if (0xffff != _mm_movemask_epi8(_mm_cmpeq_epi8(flat, zero))) { // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ // flat and wide flat calculations __m128i q5_16, q4_16, q3_16, q2_16, q1_16, q0_16; __m128i pq_16[7]; const __m128i eight = _mm_set1_epi16(8); const __m128i four = _mm_set1_epi16(4); __m128i sum_p6; __m128i sum_p3; pq_16[0] = _mm_unpacklo_epi8(*q0p0, zero); pq_16[1] = _mm_unpacklo_epi8(*q1p1, zero); pq_16[2] = _mm_unpacklo_epi8(*q2p2, zero); pq_16[3] = _mm_unpacklo_epi8(*q3p3, zero); pq_16[4] = _mm_unpacklo_epi8(*q4p4, zero); pq_16[5] = _mm_unpacklo_epi8(*q5p5, zero); pq_16[6] = _mm_unpacklo_epi8(*q6p6, zero); q0_16 = _mm_srli_si128(pq_16[0], 8); q1_16 = _mm_srli_si128(pq_16[1], 8); q2_16 = _mm_srli_si128(pq_16[2], 8); q3_16 = _mm_srli_si128(pq_16[3], 8); q4_16 = _mm_srli_si128(pq_16[4], 8); q5_16 = _mm_srli_si128(pq_16[5], 8); __m128i flat_p[3], flat_q[3]; __m128i flat2_p[6], flat2_q[6]; __m128i work0, work0_0, work0_1, sum_p_0; __m128i sum_p = _mm_add_epi16(pq_16[5], _mm_add_epi16(pq_16[4], pq_16[3])); __m128i sum_lp = _mm_add_epi16(pq_16[0], _mm_add_epi16(pq_16[2], pq_16[1])); sum_p = _mm_add_epi16(sum_p, sum_lp); __m128i sum_lq = _mm_srli_si128(sum_lp, 8); __m128i sum_q = _mm_srli_si128(sum_p, 8); sum_p_0 = _mm_add_epi16(eight, _mm_add_epi16(sum_p, sum_q)); sum_lp = _mm_add_epi16(four, _mm_add_epi16(sum_lp, sum_lq)); flat_p[0] = _mm_add_epi16(sum_lp, _mm_add_epi16(pq_16[3], pq_16[0])); flat_q[0] = _mm_add_epi16(sum_lp, _mm_add_epi16(q3_16, q0_16)); sum_p6 = _mm_add_epi16(pq_16[6], pq_16[6]); sum_p3 = _mm_add_epi16(pq_16[3], pq_16[3]); sum_q = _mm_sub_epi16(sum_p_0, pq_16[5]); sum_p = _mm_sub_epi16(sum_p_0, q5_16); work0_0 = _mm_add_epi16(_mm_add_epi16(pq_16[6], pq_16[0]), pq_16[1]); work0_1 = _mm_add_epi16( sum_p6, _mm_add_epi16(pq_16[1], _mm_add_epi16(pq_16[2], pq_16[0]))); sum_lq = _mm_sub_epi16(sum_lp, pq_16[2]); sum_lp = _mm_sub_epi16(sum_lp, q2_16); work0 = _mm_add_epi16(sum_p3, pq_16[1]); flat_p[1] = _mm_add_epi16(sum_lp, work0); flat_q[1] = _mm_add_epi16(sum_lq, _mm_srli_si128(work0, 8)); flat_pq[0] = _mm_srli_epi16(_mm_unpacklo_epi64(flat_p[0], flat_q[0]), 3); flat_pq[1] = _mm_srli_epi16(_mm_unpacklo_epi64(flat_p[1], flat_q[1]), 3); flat_pq[0] = _mm_packus_epi16(flat_pq[0], flat_pq[0]); flat_pq[1] = _mm_packus_epi16(flat_pq[1], flat_pq[1]); sum_lp = _mm_sub_epi16(sum_lp, q1_16); sum_lq = _mm_sub_epi16(sum_lq, pq_16[1]); sum_p3 = _mm_add_epi16(sum_p3, pq_16[3]); work0 = _mm_add_epi16(sum_p3, pq_16[2]); flat_p[2] = _mm_add_epi16(sum_lp, work0); flat_q[2] = _mm_add_epi16(sum_lq, _mm_srli_si128(work0, 8)); flat_pq[2] = _mm_srli_epi16(_mm_unpacklo_epi64(flat_p[2], flat_q[2]), 3); flat_pq[2] = _mm_packus_epi16(flat_pq[2], flat_pq[2]); // ~~~~~~~~~~~~~~~~~~~~~~~~~~~ flat 2 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ flat2 = _mm_max_epu8(abs_diff(*q4p4, *q0p0), abs_diff(*q5p5, *q0p0)); work = abs_diff(*q6p6, *q0p0); flat2 = _mm_max_epu8(work, flat2); flat2 = _mm_max_epu8(flat2, _mm_srli_si128(flat2, 4)); flat2 = _mm_subs_epu8(flat2, one); flat2 = _mm_cmpeq_epi8(flat2, zero); flat2 = _mm_and_si128(flat2, flat); // flat2 & flat & mask flat2 = _mm_unpacklo_epi32(flat2, flat2); // ~~~~~~~~~~ apply flat ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ qs0ps0 = _mm_andnot_si128(flat, qs0ps0); flat_pq[0] = _mm_and_si128(flat, flat_pq[0]); *q0p0 = _mm_or_si128(qs0ps0, flat_pq[0]); qs1ps1 = _mm_andnot_si128(flat, qs1ps1); flat_pq[1] = _mm_and_si128(flat, flat_pq[1]); *q1p1 = _mm_or_si128(qs1ps1, flat_pq[1]); *q2p2 = _mm_andnot_si128(flat, *q2p2); flat_pq[2] = _mm_and_si128(flat, flat_pq[2]); *q2p2 = _mm_or_si128(*q2p2, flat_pq[2]); if (0xffff != _mm_movemask_epi8(_mm_cmpeq_epi8(flat2, zero))) { flat2_p[0] = _mm_add_epi16(sum_p_0, _mm_add_epi16(work0_0, q0_16)); flat2_q[0] = _mm_add_epi16( sum_p_0, _mm_add_epi16(_mm_srli_si128(work0_0, 8), pq_16[0])); flat2_p[1] = _mm_add_epi16(sum_p, work0_1); flat2_q[1] = _mm_add_epi16(sum_q, _mm_srli_si128(work0_1, 8)); flat2_pq[0] = _mm_srli_epi16(_mm_unpacklo_epi64(flat2_p[0], flat2_q[0]), 4); flat2_pq[1] = _mm_srli_epi16(_mm_unpacklo_epi64(flat2_p[1], flat2_q[1]), 4); flat2_pq[0] = _mm_packus_epi16(flat2_pq[0], flat2_pq[0]); flat2_pq[1] = _mm_packus_epi16(flat2_pq[1], flat2_pq[1]); sum_p = _mm_sub_epi16(sum_p, q4_16); sum_q = _mm_sub_epi16(sum_q, pq_16[4]); sum_p6 = _mm_add_epi16(sum_p6, pq_16[6]); work0 = _mm_add_epi16( sum_p6, _mm_add_epi16(pq_16[2], _mm_add_epi16(pq_16[3], pq_16[1]))); flat2_p[2] = _mm_add_epi16(sum_p, work0); flat2_q[2] = _mm_add_epi16(sum_q, _mm_srli_si128(work0, 8)); flat2_pq[2] = _mm_srli_epi16(_mm_unpacklo_epi64(flat2_p[2], flat2_q[2]), 4); flat2_pq[2] = _mm_packus_epi16(flat2_pq[2], flat2_pq[2]); sum_p6 = _mm_add_epi16(sum_p6, pq_16[6]); sum_p = _mm_sub_epi16(sum_p, q3_16); sum_q = _mm_sub_epi16(sum_q, pq_16[3]); work0 = _mm_add_epi16( sum_p6, _mm_add_epi16(pq_16[3], _mm_add_epi16(pq_16[4], pq_16[2]))); flat2_p[3] = _mm_add_epi16(sum_p, work0); flat2_q[3] = _mm_add_epi16(sum_q, _mm_srli_si128(work0, 8)); flat2_pq[3] = _mm_srli_epi16(_mm_unpacklo_epi64(flat2_p[3], flat2_q[3]), 4); flat2_pq[3] = _mm_packus_epi16(flat2_pq[3], flat2_pq[3]); sum_p6 = _mm_add_epi16(sum_p6, pq_16[6]); sum_p = _mm_sub_epi16(sum_p, q2_16); sum_q = _mm_sub_epi16(sum_q, pq_16[2]); work0 = _mm_add_epi16( sum_p6, _mm_add_epi16(pq_16[4], _mm_add_epi16(pq_16[5], pq_16[3]))); flat2_p[4] = _mm_add_epi16(sum_p, work0); flat2_q[4] = _mm_add_epi16(sum_q, _mm_srli_si128(work0, 8)); flat2_pq[4] = _mm_srli_epi16(_mm_unpacklo_epi64(flat2_p[4], flat2_q[4]), 4); flat2_pq[4] = _mm_packus_epi16(flat2_pq[4], flat2_pq[4]); sum_p6 = _mm_add_epi16(sum_p6, pq_16[6]); sum_p = _mm_sub_epi16(sum_p, q1_16); sum_q = _mm_sub_epi16(sum_q, pq_16[1]); work0 = _mm_add_epi16( sum_p6, _mm_add_epi16(pq_16[5], _mm_add_epi16(pq_16[6], pq_16[4]))); flat2_p[5] = _mm_add_epi16(sum_p, work0); flat2_q[5] = _mm_add_epi16(sum_q, _mm_srli_si128(work0, 8)); flat2_pq[5] = _mm_srli_epi16(_mm_unpacklo_epi64(flat2_p[5], flat2_q[5]), 4); flat2_pq[5] = _mm_packus_epi16(flat2_pq[5], flat2_pq[5]); // wide flat // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ *q0p0 = _mm_andnot_si128(flat2, *q0p0); flat2_pq[0] = _mm_and_si128(flat2, flat2_pq[0]); *q0p0 = _mm_or_si128(*q0p0, flat2_pq[0]); *q1p1 = _mm_andnot_si128(flat2, *q1p1); flat2_pq[1] = _mm_and_si128(flat2, flat2_pq[1]); *q1p1 = _mm_or_si128(*q1p1, flat2_pq[1]); *q2p2 = _mm_andnot_si128(flat2, *q2p2); flat2_pq[2] = _mm_and_si128(flat2, flat2_pq[2]); *q2p2 = _mm_or_si128(*q2p2, flat2_pq[2]); *q3p3 = _mm_andnot_si128(flat2, *q3p3); flat2_pq[3] = _mm_and_si128(flat2, flat2_pq[3]); *q3p3 = _mm_or_si128(*q3p3, flat2_pq[3]); *q4p4 = _mm_andnot_si128(flat2, *q4p4); flat2_pq[4] = _mm_and_si128(flat2, flat2_pq[4]); *q4p4 = _mm_or_si128(*q4p4, flat2_pq[4]); *q5p5 = _mm_andnot_si128(flat2, *q5p5); flat2_pq[5] = _mm_and_si128(flat2, flat2_pq[5]); *q5p5 = _mm_or_si128(*q5p5, flat2_pq[5]); } } else { *q0p0 = qs0ps0; *q1p1 = qs1ps1; } } void aom_lpf_horizontal_14_sse2(unsigned char *s, int p, const unsigned char *_blimit, const unsigned char *_limit, const unsigned char *_thresh) { __m128i q6p6, q5p5, q4p4, q3p3, q2p2, q1p1, q0p0; __m128i blimit = _mm_load_si128((const __m128i *)_blimit); __m128i limit = _mm_load_si128((const __m128i *)_limit); __m128i thresh = _mm_load_si128((const __m128i *)_thresh); q4p4 = _mm_unpacklo_epi32(xx_loadl_32(s - 5 * p), xx_loadl_32(s + 4 * p)); q3p3 = _mm_unpacklo_epi32(xx_loadl_32(s - 4 * p), xx_loadl_32(s + 3 * p)); q2p2 = _mm_unpacklo_epi32(xx_loadl_32(s - 3 * p), xx_loadl_32(s + 2 * p)); q1p1 = _mm_unpacklo_epi32(xx_loadl_32(s - 2 * p), xx_loadl_32(s + 1 * p)); q0p0 = _mm_unpacklo_epi32(xx_loadl_32(s - 1 * p), xx_loadl_32(s - 0 * p)); q5p5 = _mm_unpacklo_epi32(xx_loadl_32(s - 6 * p), xx_loadl_32(s + 5 * p)); q6p6 = _mm_unpacklo_epi32(xx_loadl_32(s - 7 * p), xx_loadl_32(s + 6 * p)); lpf_internal_14_sse2(&q6p6, &q5p5, &q4p4, &q3p3, &q2p2, &q1p1, &q0p0, &blimit, &limit, &thresh); store_buffer_horz_8(q0p0, p, 0, s); store_buffer_horz_8(q1p1, p, 1, s); store_buffer_horz_8(q2p2, p, 2, s); store_buffer_horz_8(q3p3, p, 3, s); store_buffer_horz_8(q4p4, p, 4, s); store_buffer_horz_8(q5p5, p, 5, s); } static AOM_FORCE_INLINE void lpf_internal_6_dual_sse2( __m128i *p2, __m128i *q2, __m128i *p1, __m128i *q1, __m128i *p0, __m128i *q0, __m128i *q1q0, __m128i *p1p0, __m128i *blimit, __m128i *limit, __m128i *thresh) { const __m128i zero = _mm_setzero_si128(); __m128i mask, hev, flat; __m128i q2p2, q1p1, q0p0, flat_p1p0, flat_q0q1; __m128i p2_16, q2_16, p1_16, q1_16, p0_16, q0_16; __m128i ps1ps0, qs1qs0; q2p2 = _mm_unpacklo_epi64(*p2, *q2); q1p1 = _mm_unpacklo_epi64(*p1, *q1); q0p0 = _mm_unpacklo_epi64(*p0, *q0); *p1p0 = _mm_unpacklo_epi64(q0p0, q1p1); *q1q0 = _mm_unpackhi_epi64(q0p0, q1p1); const __m128i one = _mm_set1_epi8(1); const __m128i fe = _mm_set1_epi8((char)0xfe); const __m128i ff = _mm_cmpeq_epi8(fe, fe); { // filter_mask and hev_mask __m128i abs_p1q1, abs_p0q0, abs_q1q0, abs_p1p0, work; abs_p1p0 = abs_diff(q1p1, q0p0); abs_q1q0 = _mm_srli_si128(abs_p1p0, 8); abs_p0q0 = abs_diff(*p1p0, *q1q0); abs_p1q1 = _mm_srli_si128(abs_p0q0, 8); abs_p0q0 = _mm_unpacklo_epi64(abs_p0q0, zero); // considering sse doesn't have unsigned elements comparison the idea is // to find at least one case when X > limit, it means the corresponding // mask bit is set. // to achieve that we find global max value of all inputs of abs(x-y) or // (abs(p0 - q0) * 2 + abs(p1 - q1) / 2 If it is > limit the mask is set // otherwise - not flat = _mm_max_epu8(abs_p1p0, abs_q1q0); hev = _mm_subs_epu8(flat, *thresh); hev = _mm_xor_si128(_mm_cmpeq_epi8(hev, zero), ff); // replicate for the further "merged variables" usage hev = _mm_unpacklo_epi64(hev, hev); abs_p0q0 = _mm_adds_epu8(abs_p0q0, abs_p0q0); abs_p1q1 = _mm_srli_epi16(_mm_and_si128(abs_p1q1, fe), 1); mask = _mm_subs_epu8(_mm_adds_epu8(abs_p0q0, abs_p1q1), *blimit); mask = _mm_xor_si128(_mm_cmpeq_epi8(mask, zero), ff); // mask |= (abs(p0 - q0) * 2 + abs(p1 - q1) / 2 > blimit) * -1; mask = _mm_max_epu8(abs_p1p0, mask); // mask |= (abs(p1 - p0) > limit) * -1; // mask |= (abs(q1 - q0) > limit) * -1; work = abs_diff(q2p2, q1p1); mask = _mm_max_epu8(work, mask); mask = _mm_max_epu8(mask, _mm_srli_si128(mask, 8)); mask = _mm_subs_epu8(mask, *limit); mask = _mm_cmpeq_epi8(mask, zero); // lp filter - the same for 6, 8 and 14 versions filter4_dual_sse2(p1p0, q1q0, &hev, &mask, q1q0, p1p0); // flat_mask flat = _mm_max_epu8(abs_diff(q2p2, q0p0), abs_p1p0); flat = _mm_max_epu8(flat, _mm_srli_si128(flat, 8)); flat = _mm_subs_epu8(flat, one); flat = _mm_cmpeq_epi8(flat, zero); flat = _mm_and_si128(flat, mask); // replicate for the further "merged variables" usage flat = _mm_unpacklo_epi64(flat, flat); } // 5 tap filter // need it only if flat !=0 if (0xffff != _mm_movemask_epi8(_mm_cmpeq_epi8(flat, zero))) { const __m128i four = _mm_set1_epi16(4); __m128i workp_a, workp_b, workp_shft0, workp_shft1; p2_16 = _mm_unpacklo_epi8(*p2, zero); p1_16 = _mm_unpacklo_epi8(*p1, zero); p0_16 = _mm_unpacklo_epi8(*p0, zero); q0_16 = _mm_unpacklo_epi8(*q0, zero); q1_16 = _mm_unpacklo_epi8(*q1, zero); q2_16 = _mm_unpacklo_epi8(*q2, zero); // op1 workp_a = _mm_add_epi16(_mm_add_epi16(p0_16, p0_16), _mm_add_epi16(p1_16, p1_16)); // p0 *2 + p1 * 2 workp_a = _mm_add_epi16(_mm_add_epi16(workp_a, four), p2_16); // p2 + p0 * 2 + p1 * 2 + 4 workp_b = _mm_add_epi16(_mm_add_epi16(p2_16, p2_16), q0_16); workp_shft0 = _mm_srli_epi16(_mm_add_epi16(workp_a, workp_b), 3); // p2 * 3 + p1 * 2 + p0 * 2 + q0 + 4 // op0 workp_b = _mm_add_epi16(_mm_add_epi16(q0_16, q0_16), q1_16); // q0 * 2 + q1 workp_a = _mm_add_epi16(workp_a, workp_b); // p2 + p0 * 2 + p1 * 2 + q0 * 2 + q1 + 4 workp_shft1 = _mm_srli_epi16(workp_a, 3); flat_p1p0 = _mm_packus_epi16(workp_shft1, workp_shft0); // oq0 workp_a = _mm_sub_epi16(_mm_sub_epi16(workp_a, p2_16), p1_16); // p0 * 2 + p1 + q0 * 2 + q1 + 4 workp_b = _mm_add_epi16(q1_16, q2_16); workp_a = _mm_add_epi16( workp_a, workp_b); // p0 * 2 + p1 + q0 * 2 + q1 * 2 + q2 + 4 workp_shft0 = _mm_srli_epi16(workp_a, 3); // oq1 workp_a = _mm_sub_epi16(_mm_sub_epi16(workp_a, p1_16), p0_16); // p0 + q0 * 2 + q1 * 2 + q2 + 4 workp_b = _mm_add_epi16(q2_16, q2_16); workp_shft1 = _mm_srli_epi16(_mm_add_epi16(workp_a, workp_b), 3); // p0 + q0 * 2 + q1 * 2 + q2 * 3 + 4 flat_q0q1 = _mm_packus_epi16(workp_shft0, workp_shft1); qs1qs0 = _mm_andnot_si128(flat, *q1q0); *q1q0 = _mm_and_si128(flat, flat_q0q1); *q1q0 = _mm_or_si128(qs1qs0, *q1q0); ps1ps0 = _mm_andnot_si128(flat, *p1p0); *p1p0 = _mm_and_si128(flat, flat_p1p0); *p1p0 = _mm_or_si128(ps1ps0, *p1p0); } } static AOM_FORCE_INLINE void lpf_internal_6_sse2( __m128i *p2, __m128i *q2, __m128i *p1, __m128i *q1, __m128i *p0, __m128i *q0, __m128i *q1q0, __m128i *p1p0, __m128i *blimit, __m128i *limit, __m128i *thresh) { const __m128i zero = _mm_setzero_si128(); __m128i mask, hev, flat; __m128i q2p2, q1p1, q0p0, flat_p1p0, flat_q0q1; __m128i pq2_16, q2_16, pq1_16, pq0_16, q0_16; __m128i ps1ps0, qs1qs0; q2p2 = _mm_unpacklo_epi32(*p2, *q2); q1p1 = _mm_unpacklo_epi32(*p1, *q1); q0p0 = _mm_unpacklo_epi32(*p0, *q0); *p1p0 = _mm_unpacklo_epi32(*p0, *p1); *q1q0 = _mm_unpacklo_epi32(*q0, *q1); const __m128i one = _mm_set1_epi8(1); const __m128i fe = _mm_set1_epi8((char)0xfe); const __m128i ff = _mm_cmpeq_epi8(fe, fe); { // filter_mask and hev_mask __m128i abs_p1q1, abs_p0q0, abs_q1q0, abs_p1p0, work; abs_p1p0 = abs_diff(q1p1, q0p0); abs_q1q0 = _mm_srli_si128(abs_p1p0, 4); abs_p0q0 = abs_diff(*p1p0, *q1q0); abs_p1q1 = _mm_srli_si128(abs_p0q0, 4); // considering sse doesn't have unsigned elements comparison the idea is // to find at least one case when X > limit, it means the corresponding // mask bit is set. // to achieve that we find global max value of all inputs of abs(x-y) or // (abs(p0 - q0) * 2 + abs(p1 - q1) / 2 If it is > limit the mask is set // otherwise - not flat = _mm_max_epu8(abs_p1p0, abs_q1q0); hev = _mm_subs_epu8(flat, *thresh); hev = _mm_xor_si128(_mm_cmpeq_epi8(hev, zero), ff); // replicate for the further "merged variables" usage hev = _mm_unpacklo_epi32(hev, hev); abs_p0q0 = _mm_adds_epu8(abs_p0q0, abs_p0q0); abs_p1q1 = _mm_srli_epi16(_mm_and_si128(abs_p1q1, fe), 1); mask = _mm_subs_epu8(_mm_adds_epu8(abs_p0q0, abs_p1q1), *blimit); mask = _mm_unpacklo_epi32(mask, zero); mask = _mm_xor_si128(_mm_cmpeq_epi8(mask, zero), ff); // mask |= (abs(p0 - q0) * 2 + abs(p1 - q1) / 2 > blimit) * -1; mask = _mm_max_epu8(abs_p1p0, mask); // mask |= (abs(p1 - p0) > limit) * -1; // mask |= (abs(q1 - q0) > limit) * -1; work = abs_diff(q2p2, q1p1); mask = _mm_max_epu8(work, mask); mask = _mm_max_epu8(mask, _mm_srli_si128(mask, 4)); mask = _mm_subs_epu8(mask, *limit); mask = _mm_cmpeq_epi8(mask, zero); // lp filter - the same for 6, 8 and 14 versions filter4_sse2(p1p0, q1q0, &hev, &mask, q1q0, p1p0); // flat_mask flat = _mm_max_epu8(abs_diff(q2p2, q0p0), abs_p1p0); flat = _mm_max_epu8(flat, _mm_srli_si128(flat, 4)); flat = _mm_subs_epu8(flat, one); flat = _mm_cmpeq_epi8(flat, zero); flat = _mm_and_si128(flat, mask); // replicate for the further "merged variables" usage flat = _mm_unpacklo_epi32(flat, flat); flat = _mm_unpacklo_epi64(flat, flat); } // 5 tap filter // need it only if flat !=0 if (0xffff != _mm_movemask_epi8(_mm_cmpeq_epi8(flat, zero))) { const __m128i four = _mm_set1_epi16(4); __m128i workp_a, workp_b, workp_c; __m128i pq0x2_pq1, pq1_pq2; pq2_16 = _mm_unpacklo_epi8(q2p2, zero); pq1_16 = _mm_unpacklo_epi8(q1p1, zero); pq0_16 = _mm_unpacklo_epi8(q0p0, zero); q0_16 = _mm_srli_si128(pq0_16, 8); q2_16 = _mm_srli_si128(pq2_16, 8); // op1 pq0x2_pq1 = _mm_add_epi16(_mm_add_epi16(pq0_16, pq0_16), pq1_16); // p0 *2 + p1 pq1_pq2 = _mm_add_epi16(pq1_16, pq2_16); // p1 + p2 workp_a = _mm_add_epi16(_mm_add_epi16(pq0x2_pq1, four), pq1_pq2); // p2 + p0 * 2 + p1 * 2 + 4 workp_b = _mm_add_epi16(_mm_add_epi16(pq2_16, pq2_16), q0_16); workp_b = _mm_add_epi16(workp_a, workp_b); // p2 * 3 + p1 * 2 + p0 * 2 + q0 + 4 // op0 workp_c = _mm_srli_si128(pq0x2_pq1, 8); // q0 * 2 + q1 workp_a = _mm_add_epi16(workp_a, workp_c); // p2 + p0 * 2 + p1 * 2 + q0 * 2 + q1 + 4 workp_b = _mm_unpacklo_epi64(workp_a, workp_b); workp_b = _mm_srli_epi16(workp_b, 3); flat_p1p0 = _mm_packus_epi16(workp_b, workp_b); // oq0 workp_a = _mm_sub_epi16(_mm_sub_epi16(workp_a, pq2_16), pq1_16); // p0 * 2 + p1 + q0 * 2 + q1 + 4 workp_b = _mm_srli_si128(pq1_pq2, 8); workp_a = _mm_add_epi16( workp_a, workp_b); // p0 * 2 + p1 + q0 * 2 + q1 * 2 + q2 + 4 // workp_shft0 = _mm_srli_epi16(workp_a, 3); // oq1 workp_c = _mm_sub_epi16(_mm_sub_epi16(workp_a, pq1_16), pq0_16); // p0 + q0 * 2 + q1 * 2 + q2 + 4 workp_b = _mm_add_epi16(q2_16, q2_16); workp_b = _mm_add_epi16(workp_c, workp_b); // p0 + q0 * 2 + q1 * 2 + q2 * 3 + 4 workp_a = _mm_unpacklo_epi64(workp_a, workp_b); workp_a = _mm_srli_epi16(workp_a, 3); flat_q0q1 = _mm_packus_epi16(workp_a, workp_a); qs1qs0 = _mm_andnot_si128(flat, *q1q0); *q1q0 = _mm_and_si128(flat, flat_q0q1); *q1q0 = _mm_or_si128(qs1qs0, *q1q0); ps1ps0 = _mm_andnot_si128(flat, *p1p0); *p1p0 = _mm_and_si128(flat, flat_p1p0); *p1p0 = _mm_or_si128(ps1ps0, *p1p0); } } void aom_lpf_horizontal_6_sse2(unsigned char *s, int p, const unsigned char *_blimit, const unsigned char *_limit, const unsigned char *_thresh) { __m128i p2, p1, p0, q0, q1, q2; __m128i p1p0, q1q0; __m128i blimit = _mm_load_si128((__m128i *)_blimit); __m128i limit = _mm_load_si128((__m128i *)_limit); __m128i thresh = _mm_load_si128((__m128i *)_thresh); p2 = xx_loadl_32(s - 3 * p); p1 = xx_loadl_32(s - 2 * p); p0 = xx_loadl_32(s - 1 * p); q0 = xx_loadl_32(s - 0 * p); q1 = xx_loadl_32(s + 1 * p); q2 = xx_loadl_32(s + 2 * p); lpf_internal_6_sse2(&p2, &q2, &p1, &q1, &p0, &q0, &q1q0, &p1p0, &blimit, &limit, &thresh); xx_storel_32(s - 1 * p, p1p0); xx_storel_32(s - 2 * p, _mm_srli_si128(p1p0, 4)); xx_storel_32(s + 0 * p, q1q0); xx_storel_32(s + 1 * p, _mm_srli_si128(q1q0, 4)); } void aom_lpf_horizontal_6_dual_sse2(unsigned char *s, int p, const unsigned char *_blimit0, const unsigned char *_limit0, const unsigned char *_thresh0, const unsigned char *_blimit1, const unsigned char *_limit1, const unsigned char *_thresh1) { __m128i blimit = _mm_unpacklo_epi32(_mm_load_si128((__m128i *)_blimit0), _mm_load_si128((__m128i *)_blimit1)); __m128i limit = _mm_unpacklo_epi32(_mm_load_si128((__m128i *)_limit0), _mm_load_si128((__m128i *)_limit1)); __m128i thresh = _mm_unpacklo_epi32(_mm_load_si128((__m128i *)_thresh0), _mm_load_si128((__m128i *)_thresh1)); __m128i p2, p1, p0, q0, q1, q2; __m128i p1p0, q1q0; p2 = _mm_loadl_epi64((__m128i *)(s - 3 * p)); p1 = _mm_loadl_epi64((__m128i *)(s - 2 * p)); p0 = _mm_loadl_epi64((__m128i *)(s - 1 * p)); q0 = _mm_loadl_epi64((__m128i *)(s - 0 * p)); q1 = _mm_loadl_epi64((__m128i *)(s + 1 * p)); q2 = _mm_loadl_epi64((__m128i *)(s + 2 * p)); lpf_internal_6_dual_sse2(&p2, &q2, &p1, &q1, &p0, &q0, &q1q0, &p1p0, &blimit, &limit, &thresh); _mm_storel_epi64((__m128i *)(s - 1 * p), p1p0); _mm_storel_epi64((__m128i *)(s - 2 * p), _mm_srli_si128(p1p0, 8)); _mm_storel_epi64((__m128i *)(s + 0 * p), q1q0); _mm_storel_epi64((__m128i *)(s + 1 * p), _mm_srli_si128(q1q0, 8)); } static AOM_FORCE_INLINE void lpf_internal_8_sse2( __m128i *p3, __m128i *q3, __m128i *p2, __m128i *q2, __m128i *p1, __m128i *q1, __m128i *p0, __m128i *q0, __m128i *q1q0_out, __m128i *p1p0_out, __m128i *blimit, __m128i *limit, __m128i *thresh) { const __m128i zero = _mm_setzero_si128(); __m128i mask, hev, flat; __m128i p2_16, q2_16, p1_16, p0_16, q0_16, q1_16, p3_16, q3_16, q3p3, flat_p1p0, flat_q0q1; __m128i q2p2, q1p1, q0p0; __m128i q1q0, p1p0, ps1ps0, qs1qs0; __m128i work_pq, opq2, pq2; q3p3 = _mm_unpacklo_epi32(*p3, *q3); q2p2 = _mm_unpacklo_epi32(*p2, *q2); q1p1 = _mm_unpacklo_epi32(*p1, *q1); q0p0 = _mm_unpacklo_epi32(*p0, *q0); p1p0 = _mm_unpacklo_epi32(q0p0, q1p1); // p1p0 q1q0 q1q0 = _mm_srli_si128(p1p0, 8); // filter_mask and hev_mask // considering sse doesn't have unsigned elements comparison the idea is to // find at least one case when X > limit, it means the corresponding mask // bit is set. // to achieve that we find global max value of all inputs of abs(x-y) or // (abs(p0 - q0) * 2 + abs(p1 - q1) / 2 If it is > limit the mask is set // otherwise - not const __m128i one = _mm_set1_epi8(1); const __m128i fe = _mm_set1_epi8((char)0xfe); const __m128i ff = _mm_cmpeq_epi8(fe, fe); __m128i abs_p1q1, abs_p0q0, abs_q1q0, abs_p1p0, work; abs_p1p0 = abs_diff(q1p1, q0p0); abs_q1q0 = _mm_srli_si128(abs_p1p0, 4); abs_p0q0 = abs_diff(p1p0, q1q0); abs_p1q1 = _mm_srli_si128(abs_p0q0, 4); flat = _mm_max_epu8(abs_p1p0, abs_q1q0); hev = _mm_subs_epu8(flat, *thresh); hev = _mm_xor_si128(_mm_cmpeq_epi8(hev, zero), ff); // replicate for the further "merged variables" usage hev = _mm_unpacklo_epi32(hev, hev); abs_p0q0 = _mm_adds_epu8(abs_p0q0, abs_p0q0); abs_p1q1 = _mm_srli_epi16(_mm_and_si128(abs_p1q1, fe), 1); mask = _mm_subs_epu8(_mm_adds_epu8(abs_p0q0, abs_p1q1), *blimit); mask = _mm_unpacklo_epi32(mask, zero); mask = _mm_xor_si128(_mm_cmpeq_epi8(mask, zero), ff); // mask |= (abs(p0 - q0) * 2 + abs(p1 - q1) / 2 > blimit) * -1; mask = _mm_max_epu8(abs_p1p0, mask); // mask |= (abs(p1 - p0) > limit) * -1; // mask |= (abs(q1 - q0) > limit) * -1; work = _mm_max_epu8(abs_diff(q2p2, q1p1), abs_diff(q3p3, q2p2)); mask = _mm_max_epu8(work, mask); mask = _mm_max_epu8(mask, _mm_srli_si128(mask, 4)); mask = _mm_subs_epu8(mask, *limit); mask = _mm_cmpeq_epi8(mask, zero); // lp filter - the same for 6, 8 and 14 versions filter4_sse2(&p1p0, &q1q0, &hev, &mask, q1q0_out, p1p0_out); // flat_mask4 flat = _mm_max_epu8(abs_diff(q2p2, q0p0), abs_diff(q3p3, q0p0)); flat = _mm_max_epu8(abs_p1p0, flat); flat = _mm_max_epu8(flat, _mm_srli_si128(flat, 4)); flat = _mm_subs_epu8(flat, one); flat = _mm_cmpeq_epi8(flat, zero); flat = _mm_and_si128(flat, mask); // replicate for the further "merged variables" usage flat = _mm_unpacklo_epi32(flat, flat); flat = _mm_unpacklo_epi64(flat, flat); // filter8 need it only if flat !=0 if (0xffff != _mm_movemask_epi8(_mm_cmpeq_epi8(flat, zero))) { const __m128i four = _mm_set1_epi16(4); __m128i workp_a, workp_b, workp_c, workp_d, workp_shft1, workp_shft2; p2_16 = _mm_unpacklo_epi8(*p2, zero); p1_16 = _mm_unpacklo_epi8(*p1, zero); p0_16 = _mm_unpacklo_epi8(*p0, zero); q0_16 = _mm_unpacklo_epi8(*q0, zero); q1_16 = _mm_unpacklo_epi8(*q1, zero); q2_16 = _mm_unpacklo_epi8(*q2, zero); p3_16 = _mm_unpacklo_epi8(*p3, zero); q3_16 = _mm_unpacklo_epi8(*q3, zero); // op2 workp_a = _mm_add_epi16(_mm_add_epi16(p3_16, p3_16), _mm_add_epi16(p2_16, p1_16)); workp_a = _mm_add_epi16(_mm_add_epi16(workp_a, four), p0_16); workp_b = _mm_add_epi16(_mm_add_epi16(q0_16, p2_16), p3_16); workp_shft2 = _mm_add_epi16(workp_a, workp_b); // op1 workp_b = _mm_add_epi16(_mm_add_epi16(q0_16, q1_16), p1_16); workp_c = _mm_add_epi16(workp_a, workp_b); // workp_shft0 = _mm_srli_epi16(_mm_add_epi16(workp_a, workp_b), 3); // op0 workp_a = _mm_add_epi16(_mm_sub_epi16(workp_a, p3_16), q2_16); workp_b = _mm_add_epi16(_mm_sub_epi16(workp_b, p1_16), p0_16); workp_d = _mm_add_epi16(workp_a, workp_b); // workp_shft1 = _mm_srli_epi16(_mm_add_epi16(workp_a, workp_b), 3); workp_c = _mm_unpacklo_epi64(workp_d, workp_c); workp_c = _mm_srli_epi16(workp_c, 3); flat_p1p0 = _mm_packus_epi16(workp_c, workp_c); // oq0 workp_a = _mm_add_epi16(_mm_sub_epi16(workp_a, p3_16), q3_16); workp_b = _mm_add_epi16(_mm_sub_epi16(workp_b, p0_16), q0_16); // workp_shft0 = _mm_srli_epi16(_mm_add_epi16(workp_a, workp_b), 3); workp_c = _mm_add_epi16(workp_a, workp_b); // oq1 workp_a = _mm_add_epi16(_mm_sub_epi16(workp_a, p2_16), q3_16); workp_b = _mm_add_epi16(_mm_sub_epi16(workp_b, q0_16), q1_16); workp_d = _mm_add_epi16(workp_a, workp_b); // workp_shft1 = _mm_srli_epi16(_mm_add_epi16(workp_a, workp_b), 3); workp_c = _mm_unpacklo_epi64(workp_c, workp_d); workp_c = _mm_srli_epi16(workp_c, 3); flat_q0q1 = _mm_packus_epi16(workp_c, workp_c); // oq2 workp_a = _mm_add_epi16(_mm_sub_epi16(workp_a, p1_16), q3_16); workp_b = _mm_add_epi16(_mm_sub_epi16(workp_b, q1_16), q2_16); workp_shft1 = _mm_add_epi16(workp_a, workp_b); workp_c = _mm_unpacklo_epi64(workp_shft2, workp_shft1); workp_c = _mm_srli_epi16(workp_c, 3); opq2 = _mm_packus_epi16(workp_c, workp_c); work_pq = _mm_andnot_si128(flat, q2p2); pq2 = _mm_and_si128(flat, opq2); *p2 = _mm_or_si128(work_pq, pq2); *q2 = _mm_srli_si128(*p2, 4); qs1qs0 = _mm_andnot_si128(flat, *q1q0_out); q1q0 = _mm_and_si128(flat, flat_q0q1); *q1q0_out = _mm_or_si128(qs1qs0, q1q0); ps1ps0 = _mm_andnot_si128(flat, *p1p0_out); p1p0 = _mm_and_si128(flat, flat_p1p0); *p1p0_out = _mm_or_si128(ps1ps0, p1p0); } } static AOM_FORCE_INLINE void lpf_internal_8_dual_sse2( __m128i *p3, __m128i *q3, __m128i *p2, __m128i *q2, __m128i *p1, __m128i *q1, __m128i *p0, __m128i *q0, __m128i *q1q0_out, __m128i *p1p0_out, __m128i *blimit, __m128i *limit, __m128i *thresh) { const __m128i zero = _mm_setzero_si128(); __m128i mask, hev, flat; __m128i p2_16, q2_16, p1_16, p0_16, q0_16, q1_16, p3_16, q3_16, q3p3, flat_p1p0, flat_q0q1; __m128i q2p2, q1p1, q0p0; __m128i q1q0, p1p0, ps1ps0, qs1qs0; __m128i work_pq, opq2, pq2; q3p3 = _mm_unpacklo_epi64(*p3, *q3); q2p2 = _mm_unpacklo_epi64(*p2, *q2); q1p1 = _mm_unpacklo_epi64(*p1, *q1); q0p0 = _mm_unpacklo_epi64(*p0, *q0); p1p0 = _mm_unpacklo_epi64(q0p0, q1p1); q1q0 = _mm_unpackhi_epi64(q0p0, q1p1); { // filter_mask and hev_mask // considering sse doesn't have unsigned elements comparison the idea is to // find at least one case when X > limit, it means the corresponding mask // bit is set. // to achieve that we find global max value of all inputs of abs(x-y) or // (abs(p0 - q0) * 2 + abs(p1 - q1) / 2 If it is > limit the mask is set // otherwise - not const __m128i one = _mm_set1_epi8(1); const __m128i fe = _mm_set1_epi8((char)0xfe); const __m128i ff = _mm_cmpeq_epi8(fe, fe); __m128i abs_p1q1, abs_p0q0, abs_q1q0, abs_p1p0, work; abs_p1p0 = abs_diff(q1p1, q0p0); abs_q1q0 = _mm_srli_si128(abs_p1p0, 8); abs_p0q0 = abs_diff(p1p0, q1q0); abs_p1q1 = _mm_srli_si128(abs_p0q0, 8); abs_p0q0 = _mm_unpacklo_epi64(abs_p0q0, abs_p0q0); flat = _mm_max_epu8(abs_p1p0, abs_q1q0); hev = _mm_subs_epu8(flat, *thresh); hev = _mm_xor_si128(_mm_cmpeq_epi8(hev, zero), ff); // replicate for the further "merged variables" usage hev = _mm_unpacklo_epi64(hev, hev); abs_p0q0 = _mm_adds_epu8(abs_p0q0, abs_p0q0); abs_p1q1 = _mm_srli_epi16(_mm_and_si128(abs_p1q1, fe), 1); mask = _mm_subs_epu8(_mm_adds_epu8(abs_p0q0, abs_p1q1), *blimit); mask = _mm_xor_si128(_mm_cmpeq_epi8(mask, zero), ff); // mask |= (abs(p0 - q0) * 2 + abs(p1 - q1) / 2 > blimit) * -1; mask = _mm_max_epu8(abs_p1p0, mask); // mask |= (abs(p1 - p0) > limit) * -1; // mask |= (abs(q1 - q0) > limit) * -1; work = _mm_max_epu8(abs_diff(q2p2, q1p1), abs_diff(q3p3, q2p2)); mask = _mm_max_epu8(work, mask); mask = _mm_max_epu8(mask, _mm_srli_si128(mask, 8)); mask = _mm_subs_epu8(mask, *limit); mask = _mm_cmpeq_epi8(mask, zero); // lp filter - the same for 6, 8 and 14 versions filter4_dual_sse2(&p1p0, &q1q0, &hev, &mask, q1q0_out, p1p0_out); // flat_mask4 flat = _mm_max_epu8(abs_diff(q2p2, q0p0), abs_diff(q3p3, q0p0)); flat = _mm_max_epu8(abs_p1p0, flat); flat = _mm_max_epu8(flat, _mm_srli_si128(flat, 8)); flat = _mm_subs_epu8(flat, one); flat = _mm_cmpeq_epi8(flat, zero); flat = _mm_and_si128(flat, mask); // replicate for the further "merged variables" usage flat = _mm_unpacklo_epi64(flat, flat); } // filter8 need it only if flat !=0 if (0xffff != _mm_movemask_epi8(_mm_cmpeq_epi8(flat, zero))) { const __m128i four = _mm_set1_epi16(4); __m128i workp_a, workp_b, workp_shft0, workp_shft1, workp_shft2; p2_16 = _mm_unpacklo_epi8(*p2, zero); p1_16 = _mm_unpacklo_epi8(*p1, zero); p0_16 = _mm_unpacklo_epi8(*p0, zero); q0_16 = _mm_unpacklo_epi8(*q0, zero); q1_16 = _mm_unpacklo_epi8(*q1, zero); q2_16 = _mm_unpacklo_epi8(*q2, zero); p3_16 = _mm_unpacklo_epi8(*p3, zero); q3_16 = _mm_unpacklo_epi8(*q3, zero); // op2 workp_a = _mm_add_epi16(_mm_add_epi16(p3_16, p3_16), _mm_add_epi16(p2_16, p1_16)); workp_a = _mm_add_epi16(_mm_add_epi16(workp_a, four), p0_16); workp_b = _mm_add_epi16(_mm_add_epi16(q0_16, p2_16), p3_16); workp_shft2 = _mm_srli_epi16(_mm_add_epi16(workp_a, workp_b), 3); // op1 workp_b = _mm_add_epi16(_mm_add_epi16(q0_16, q1_16), p1_16); workp_shft0 = _mm_srli_epi16(_mm_add_epi16(workp_a, workp_b), 3); // op0 workp_a = _mm_add_epi16(_mm_sub_epi16(workp_a, p3_16), q2_16); workp_b = _mm_add_epi16(_mm_sub_epi16(workp_b, p1_16), p0_16); workp_shft1 = _mm_srli_epi16(_mm_add_epi16(workp_a, workp_b), 3); flat_p1p0 = _mm_packus_epi16(workp_shft1, workp_shft0); // oq0 workp_a = _mm_add_epi16(_mm_sub_epi16(workp_a, p3_16), q3_16); workp_b = _mm_add_epi16(_mm_sub_epi16(workp_b, p0_16), q0_16); workp_shft0 = _mm_srli_epi16(_mm_add_epi16(workp_a, workp_b), 3); // oq1 workp_a = _mm_add_epi16(_mm_sub_epi16(workp_a, p2_16), q3_16); workp_b = _mm_add_epi16(_mm_sub_epi16(workp_b, q0_16), q1_16); workp_shft1 = _mm_srli_epi16(_mm_add_epi16(workp_a, workp_b), 3); flat_q0q1 = _mm_packus_epi16(workp_shft0, workp_shft1); // oq2 workp_a = _mm_add_epi16(_mm_sub_epi16(workp_a, p1_16), q3_16); workp_b = _mm_add_epi16(_mm_sub_epi16(workp_b, q1_16), q2_16); workp_shft1 = _mm_srli_epi16(_mm_add_epi16(workp_a, workp_b), 3); opq2 = _mm_packus_epi16(workp_shft2, workp_shft1); work_pq = _mm_andnot_si128(flat, q2p2); pq2 = _mm_and_si128(flat, opq2); *p2 = _mm_or_si128(work_pq, pq2); *q2 = _mm_srli_si128(*p2, 8); qs1qs0 = _mm_andnot_si128(flat, *q1q0_out); q1q0 = _mm_and_si128(flat, flat_q0q1); *q1q0_out = _mm_or_si128(qs1qs0, q1q0); ps1ps0 = _mm_andnot_si128(flat, *p1p0_out); p1p0 = _mm_and_si128(flat, flat_p1p0); *p1p0_out = _mm_or_si128(ps1ps0, p1p0); } } void aom_lpf_horizontal_8_sse2(unsigned char *s, int p, const unsigned char *_blimit, const unsigned char *_limit, const unsigned char *_thresh) { __m128i p3, p2, p1, p0, q0, q1, q2, q3; __m128i q1q0, p1p0; __m128i blimit = _mm_load_si128((const __m128i *)_blimit); __m128i limit = _mm_load_si128((const __m128i *)_limit); __m128i thresh = _mm_load_si128((const __m128i *)_thresh); p3 = xx_loadl_32(s - 4 * p); p2 = xx_loadl_32(s - 3 * p); p1 = xx_loadl_32(s - 2 * p); p0 = xx_loadl_32(s - 1 * p); q0 = xx_loadl_32(s - 0 * p); q1 = xx_loadl_32(s + 1 * p); q2 = xx_loadl_32(s + 2 * p); q3 = xx_loadl_32(s + 3 * p); lpf_internal_8_sse2(&p3, &q3, &p2, &q2, &p1, &q1, &p0, &q0, &q1q0, &p1p0, &blimit, &limit, &thresh); xx_storel_32(s - 1 * p, p1p0); xx_storel_32(s - 2 * p, _mm_srli_si128(p1p0, 4)); xx_storel_32(s + 0 * p, q1q0); xx_storel_32(s + 1 * p, _mm_srli_si128(q1q0, 4)); xx_storel_32(s - 3 * p, p2); xx_storel_32(s + 2 * p, q2); } void aom_lpf_horizontal_14_dual_sse2(unsigned char *s, int p, const unsigned char *_blimit0, const unsigned char *_limit0, const unsigned char *_thresh0, const unsigned char *_blimit1, const unsigned char *_limit1, const unsigned char *_thresh1) { __m128i q6p6, q5p5, q4p4, q3p3, q2p2, q1p1, q0p0; __m128i blimit = _mm_unpacklo_epi32(_mm_load_si128((const __m128i *)_blimit0), _mm_load_si128((const __m128i *)_blimit1)); __m128i limit = _mm_unpacklo_epi32(_mm_load_si128((const __m128i *)_limit0), _mm_load_si128((const __m128i *)_limit1)); __m128i thresh = _mm_unpacklo_epi32(_mm_load_si128((const __m128i *)_thresh0), _mm_load_si128((const __m128i *)_thresh1)); q4p4 = _mm_unpacklo_epi64(_mm_loadl_epi64((__m128i *)(s - 5 * p)), _mm_loadl_epi64((__m128i *)(s + 4 * p))); q3p3 = _mm_unpacklo_epi64(_mm_loadl_epi64((__m128i *)(s - 4 * p)), _mm_loadl_epi64((__m128i *)(s + 3 * p))); q2p2 = _mm_unpacklo_epi64(_mm_loadl_epi64((__m128i *)(s - 3 * p)), _mm_loadl_epi64((__m128i *)(s + 2 * p))); q1p1 = _mm_unpacklo_epi64(_mm_loadl_epi64((__m128i *)(s - 2 * p)), _mm_loadl_epi64((__m128i *)(s + 1 * p))); q0p0 = _mm_unpacklo_epi64(_mm_loadl_epi64((__m128i *)(s - 1 * p)), _mm_loadl_epi64((__m128i *)(s - 0 * p))); q5p5 = _mm_unpacklo_epi64(_mm_loadl_epi64((__m128i *)(s - 6 * p)), _mm_loadl_epi64((__m128i *)(s + 5 * p))); q6p6 = _mm_unpacklo_epi64(_mm_loadl_epi64((__m128i *)(s - 7 * p)), _mm_loadl_epi64((__m128i *)(s + 6 * p))); lpf_internal_14_dual_sse2(&q6p6, &q5p5, &q4p4, &q3p3, &q2p2, &q1p1, &q0p0, &blimit, &limit, &thresh); _mm_storel_epi64((__m128i *)(s - 1 * p), q0p0); _mm_storel_epi64((__m128i *)(s + 0 * p), _mm_srli_si128(q0p0, 8)); _mm_storel_epi64((__m128i *)(s - 2 * p), q1p1); _mm_storel_epi64((__m128i *)(s + 1 * p), _mm_srli_si128(q1p1, 8)); _mm_storel_epi64((__m128i *)(s - 3 * p), q2p2); _mm_storel_epi64((__m128i *)(s + 2 * p), _mm_srli_si128(q2p2, 8)); _mm_storel_epi64((__m128i *)(s - 4 * p), q3p3); _mm_storel_epi64((__m128i *)(s + 3 * p), _mm_srli_si128(q3p3, 8)); _mm_storel_epi64((__m128i *)(s - 5 * p), q4p4); _mm_storel_epi64((__m128i *)(s + 4 * p), _mm_srli_si128(q4p4, 8)); _mm_storel_epi64((__m128i *)(s - 6 * p), q5p5); _mm_storel_epi64((__m128i *)(s + 5 * p), _mm_srli_si128(q5p5, 8)); } void aom_lpf_horizontal_8_dual_sse2(uint8_t *s, int p, const uint8_t *_blimit0, const uint8_t *_limit0, const uint8_t *_thresh0, const uint8_t *_blimit1, const uint8_t *_limit1, const uint8_t *_thresh1) { __m128i blimit = _mm_unpacklo_epi32(_mm_load_si128((__m128i *)_blimit0), _mm_load_si128((__m128i *)_blimit1)); __m128i limit = _mm_unpacklo_epi32(_mm_load_si128((__m128i *)_limit0), _mm_load_si128((__m128i *)_limit1)); __m128i thresh = _mm_unpacklo_epi32(_mm_load_si128((__m128i *)_thresh0), _mm_load_si128((__m128i *)_thresh1)); __m128i p2, p1, p0, q0, q1, q2, p3, q3; __m128i q1q0, p1p0; p3 = _mm_loadl_epi64((__m128i *)(s - 4 * p)); p2 = _mm_loadl_epi64((__m128i *)(s - 3 * p)); p1 = _mm_loadl_epi64((__m128i *)(s - 2 * p)); p0 = _mm_loadl_epi64((__m128i *)(s - 1 * p)); q0 = _mm_loadl_epi64((__m128i *)(s - 0 * p)); q1 = _mm_loadl_epi64((__m128i *)(s + 1 * p)); q2 = _mm_loadl_epi64((__m128i *)(s + 2 * p)); q3 = _mm_loadl_epi64((__m128i *)(s + 3 * p)); lpf_internal_8_dual_sse2(&p3, &q3, &p2, &q2, &p1, &q1, &p0, &q0, &q1q0, &p1p0, &blimit, &limit, &thresh); _mm_storel_epi64((__m128i *)(s - 1 * p), p1p0); _mm_storel_epi64((__m128i *)(s - 2 * p), _mm_srli_si128(p1p0, 8)); _mm_storel_epi64((__m128i *)(s + 0 * p), q1q0); _mm_storel_epi64((__m128i *)(s + 1 * p), _mm_srli_si128(q1q0, 8)); _mm_storel_epi64((__m128i *)(s - 3 * p), p2); _mm_storel_epi64((__m128i *)(s + 2 * p), q2); } void aom_lpf_horizontal_4_dual_sse2(unsigned char *s, int p, const unsigned char *_blimit0, const unsigned char *_limit0, const unsigned char *_thresh0, const unsigned char *_blimit1, const unsigned char *_limit1, const unsigned char *_thresh1) { __m128i p1, p0, q0, q1; __m128i qs1qs0, ps1ps0; p1 = _mm_loadl_epi64((__m128i *)(s - 2 * p)); p0 = _mm_loadl_epi64((__m128i *)(s - 1 * p)); q0 = _mm_loadl_epi64((__m128i *)(s - 0 * p)); q1 = _mm_loadl_epi64((__m128i *)(s + 1 * p)); const __m128i zero = _mm_setzero_si128(); const __m128i blimit = _mm_unpacklo_epi32(_mm_load_si128((const __m128i *)_blimit0), _mm_load_si128((const __m128i *)_blimit1)); const __m128i limit = _mm_unpacklo_epi32(_mm_load_si128((const __m128i *)_limit0), _mm_load_si128((const __m128i *)_limit1)); __m128i l = _mm_unpacklo_epi64(blimit, limit); __m128i thresh0 = _mm_unpacklo_epi8(_mm_loadl_epi64((const __m128i *)_thresh0), zero); __m128i thresh1 = _mm_unpacklo_epi8(_mm_loadl_epi64((const __m128i *)_thresh1), zero); __m128i t = _mm_unpacklo_epi64(thresh0, thresh1); lpf_internal_4_dual_sse2(&p1, &p0, &q0, &q1, &l, &t, &qs1qs0, &ps1ps0); _mm_storel_epi64((__m128i *)(s - 1 * p), ps1ps0); _mm_storel_epi64((__m128i *)(s - 2 * p), _mm_srli_si128(ps1ps0, 8)); _mm_storel_epi64((__m128i *)(s + 0 * p), qs1qs0); _mm_storel_epi64((__m128i *)(s + 1 * p), _mm_srli_si128(qs1qs0, 8)); } void aom_lpf_vertical_4_dual_sse2(uint8_t *s, int p, const uint8_t *_blimit0, const uint8_t *_limit0, const uint8_t *_thresh0, const uint8_t *_blimit1, const uint8_t *_limit1, const uint8_t *_thresh1) { __m128i p0, q0, q1, p1; __m128i x0, x1, x2, x3, x4, x5, x6, x7; __m128i d0, d1, d2, d3, d4, d5, d6, d7; __m128i qs1qs0, ps1ps0; const __m128i zero = _mm_setzero_si128(); const __m128i blimit = _mm_unpacklo_epi32(_mm_load_si128((const __m128i *)_blimit0), _mm_load_si128((const __m128i *)_blimit1)); const __m128i limit = _mm_unpacklo_epi32(_mm_load_si128((const __m128i *)_limit0), _mm_load_si128((const __m128i *)_limit1)); __m128i l = _mm_unpacklo_epi64(blimit, limit); __m128i thresh0 = _mm_unpacklo_epi8(_mm_loadl_epi64((const __m128i *)_thresh0), zero); __m128i thresh1 = _mm_unpacklo_epi8(_mm_loadl_epi64((const __m128i *)_thresh1), zero); __m128i t = _mm_unpacklo_epi64(thresh0, thresh1); x0 = _mm_loadl_epi64((__m128i *)((s - 2))); x1 = _mm_loadl_epi64((__m128i *)((s - 2) + p)); x2 = _mm_loadl_epi64((__m128i *)((s - 2) + 2 * p)); x3 = _mm_loadl_epi64((__m128i *)((s - 2) + 3 * p)); x4 = _mm_loadl_epi64((__m128i *)((s - 2) + 4 * p)); x5 = _mm_loadl_epi64((__m128i *)((s - 2) + 5 * p)); x6 = _mm_loadl_epi64((__m128i *)((s - 2) + 6 * p)); x7 = _mm_loadl_epi64((__m128i *)((s - 2) + 7 * p)); transpose8x8_low_sse2(&x0, &x1, &x2, &x3, &x4, &x5, &x6, &x7, &p1, &p0, &q0, &q1); lpf_internal_4_dual_sse2(&p1, &p0, &q0, &q1, &l, &t, &qs1qs0, &ps1ps0); p1 = _mm_srli_si128(ps1ps0, 8); q1 = _mm_srli_si128(qs1qs0, 8); transpose4x8_8x4_sse2(&p1, &ps1ps0, &qs1qs0, &q1, &d0, &d1, &d2, &d3, &d4, &d5, &d6, &d7); xx_storel_32((s - 2 + 0 * p), d0); xx_storel_32((s - 2 + 1 * p), d1); xx_storel_32((s - 2 + 2 * p), d2); xx_storel_32((s - 2 + 3 * p), d3); xx_storel_32((s - 2 + 4 * p), d4); xx_storel_32((s - 2 + 5 * p), d5); xx_storel_32((s - 2 + 6 * p), d6); xx_storel_32((s - 2 + 7 * p), d7); } void aom_lpf_vertical_6_sse2(unsigned char *s, int p, const unsigned char *_blimit, const unsigned char *_limit, const unsigned char *_thresh) { __m128i d0, d1, d2, d3, d4, d5, d6, d7; __m128i x2, x1, x0, x3; __m128i p0, q0; __m128i p1p0, q1q0; __m128i blimit = _mm_load_si128((__m128i *)_blimit); __m128i limit = _mm_load_si128((__m128i *)_limit); __m128i thresh = _mm_load_si128((__m128i *)_thresh); x3 = _mm_loadl_epi64((__m128i *)((s - 3) + 0 * p)); x2 = _mm_loadl_epi64((__m128i *)((s - 3) + 1 * p)); x1 = _mm_loadl_epi64((__m128i *)((s - 3) + 2 * p)); x0 = _mm_loadl_epi64((__m128i *)((s - 3) + 3 * p)); transpose4x8_8x4_sse2(&x3, &x2, &x1, &x0, &d0, &d1, &d2, &d3, &d4, &d5, &d6, &d7); lpf_internal_6_sse2(&d0, &d5, &d1, &d4, &d2, &d3, &q1q0, &p1p0, &blimit, &limit, &thresh); p0 = _mm_srli_si128(p1p0, 4); q0 = _mm_srli_si128(q1q0, 4); transpose4x8_8x4_low_sse2(&p0, &p1p0, &q1q0, &q0, &d0, &d1, &d2, &d3); xx_storel_32(s + 0 * p - 2, d0); xx_storel_32(s + 1 * p - 2, d1); xx_storel_32(s + 2 * p - 2, d2); xx_storel_32(s + 3 * p - 2, d3); } void aom_lpf_vertical_6_dual_sse2(uint8_t *s, int p, const uint8_t *_blimit0, const uint8_t *_limit0, const uint8_t *_thresh0, const uint8_t *_blimit1, const uint8_t *_limit1, const uint8_t *_thresh1) { __m128i blimit = _mm_unpacklo_epi32(_mm_load_si128((__m128i *)_blimit0), _mm_load_si128((__m128i *)_blimit1)); __m128i limit = _mm_unpacklo_epi32(_mm_load_si128((__m128i *)_limit0), _mm_load_si128((__m128i *)_limit1)); __m128i thresh = _mm_unpacklo_epi32(_mm_load_si128((__m128i *)_thresh0), _mm_load_si128((__m128i *)_thresh1)); __m128i d0, d1, d2, d3, d4, d5, d6, d7; __m128i x0, x1, x2, x3, x4, x5, x6, x7; __m128i p0, q0; __m128i p1p0, q1q0; __m128i d0d1, d2d3, d4d5, d6d7; x0 = _mm_loadl_epi64((__m128i *)((s - 3) + 0 * p)); x1 = _mm_loadl_epi64((__m128i *)((s - 3) + 1 * p)); x2 = _mm_loadl_epi64((__m128i *)((s - 3) + 2 * p)); x3 = _mm_loadl_epi64((__m128i *)((s - 3) + 3 * p)); x4 = _mm_loadl_epi64((__m128i *)((s - 3) + 4 * p)); x5 = _mm_loadl_epi64((__m128i *)((s - 3) + 5 * p)); x6 = _mm_loadl_epi64((__m128i *)((s - 3) + 6 * p)); x7 = _mm_loadl_epi64((__m128i *)((s - 3) + 7 * p)); transpose8x8_sse2(&x0, &x1, &x2, &x3, &x4, &x5, &x6, &x7, &d0d1, &d2d3, &d4d5, &d6d7); d1 = _mm_srli_si128(d0d1, 8); d3 = _mm_srli_si128(d2d3, 8); d5 = _mm_srli_si128(d4d5, 8); d7 = _mm_srli_si128(d6d7, 8); lpf_internal_6_dual_sse2(&d0d1, &d5, &d1, &d4d5, &d2d3, &d3, &q1q0, &p1p0, &blimit, &limit, &thresh); p0 = _mm_srli_si128(p1p0, 8); q0 = _mm_srli_si128(q1q0, 8); transpose4x8_8x4_sse2(&p0, &p1p0, &q1q0, &q0, &d0, &d1, &d2, &d3, &d4, &d5, &d6, &d7); xx_storel_32((s - 2 + 0 * p), d0); xx_storel_32((s - 2 + 1 * p), d1); xx_storel_32((s - 2 + 2 * p), d2); xx_storel_32((s - 2 + 3 * p), d3); xx_storel_32((s - 2 + 4 * p), d4); xx_storel_32((s - 2 + 5 * p), d5); xx_storel_32((s - 2 + 6 * p), d6); xx_storel_32((s - 2 + 7 * p), d7); } void aom_lpf_vertical_8_sse2(unsigned char *s, int p, const unsigned char *_blimit, const unsigned char *_limit, const unsigned char *_thresh) { __m128i d0, d1, d2, d3, d4, d5, d6, d7; __m128i p0, q0; __m128i x2, x1, x0, x3; __m128i q1q0, p1p0; __m128i blimit = _mm_load_si128((const __m128i *)_blimit); __m128i limit = _mm_load_si128((const __m128i *)_limit); __m128i thresh = _mm_load_si128((const __m128i *)_thresh); x3 = _mm_loadl_epi64((__m128i *)((s - 4) + 0 * p)); x2 = _mm_loadl_epi64((__m128i *)((s - 4) + 1 * p)); x1 = _mm_loadl_epi64((__m128i *)((s - 4) + 2 * p)); x0 = _mm_loadl_epi64((__m128i *)((s - 4) + 3 * p)); transpose4x8_8x4_sse2(&x3, &x2, &x1, &x0, &d0, &d1, &d2, &d3, &d4, &d5, &d6, &d7); // Loop filtering lpf_internal_8_sse2(&d0, &d7, &d1, &d6, &d2, &d5, &d3, &d4, &q1q0, &p1p0, &blimit, &limit, &thresh); p0 = _mm_srli_si128(p1p0, 4); q0 = _mm_srli_si128(q1q0, 4); transpose8x8_low_sse2(&d0, &d1, &p0, &p1p0, &q1q0, &q0, &d6, &d7, &d0, &d1, &d2, &d3); _mm_storel_epi64((__m128i *)(s - 4 + 0 * p), d0); _mm_storel_epi64((__m128i *)(s - 4 + 1 * p), d1); _mm_storel_epi64((__m128i *)(s - 4 + 2 * p), d2); _mm_storel_epi64((__m128i *)(s - 4 + 3 * p), d3); } void aom_lpf_vertical_8_dual_sse2(uint8_t *s, int p, const uint8_t *_blimit0, const uint8_t *_limit0, const uint8_t *_thresh0, const uint8_t *_blimit1, const uint8_t *_limit1, const uint8_t *_thresh1) { __m128i blimit = _mm_unpacklo_epi32(_mm_load_si128((__m128i *)_blimit0), _mm_load_si128((__m128i *)_blimit1)); __m128i limit = _mm_unpacklo_epi32(_mm_load_si128((__m128i *)_limit0), _mm_load_si128((__m128i *)_limit1)); __m128i thresh = _mm_unpacklo_epi32(_mm_load_si128((__m128i *)_thresh0), _mm_load_si128((__m128i *)_thresh1)); __m128i x0, x1, x2, x3, x4, x5, x6, x7; __m128i d1, d3, d5, d7; __m128i q1q0, p1p0; __m128i p1, q1; __m128i d0d1, d2d3, d4d5, d6d7; x0 = _mm_loadl_epi64((__m128i *)(s - 4 + 0 * p)); x1 = _mm_loadl_epi64((__m128i *)(s - 4 + 1 * p)); x2 = _mm_loadl_epi64((__m128i *)(s - 4 + 2 * p)); x3 = _mm_loadl_epi64((__m128i *)(s - 4 + 3 * p)); x4 = _mm_loadl_epi64((__m128i *)(s - 4 + 4 * p)); x5 = _mm_loadl_epi64((__m128i *)(s - 4 + 5 * p)); x6 = _mm_loadl_epi64((__m128i *)(s - 4 + 6 * p)); x7 = _mm_loadl_epi64((__m128i *)(s - 4 + 7 * p)); transpose8x8_sse2(&x0, &x1, &x2, &x3, &x4, &x5, &x6, &x7, &d0d1, &d2d3, &d4d5, &d6d7); d1 = _mm_srli_si128(d0d1, 8); d3 = _mm_srli_si128(d2d3, 8); d5 = _mm_srli_si128(d4d5, 8); d7 = _mm_srli_si128(d6d7, 8); lpf_internal_8_dual_sse2(&d0d1, &d7, &d1, &d6d7, &d2d3, &d5, &d3, &d4d5, &q1q0, &p1p0, &blimit, &limit, &thresh); p1 = _mm_srli_si128(p1p0, 8); q1 = _mm_srli_si128(q1q0, 8); transpose8x8_sse2(&d0d1, &d1, &p1, &p1p0, &q1q0, &q1, &d6d7, &d7, &d0d1, &d2d3, &d4d5, &d6d7); _mm_storel_epi64((__m128i *)(s - 4 + 0 * p), d0d1); _mm_storel_epi64((__m128i *)(s - 4 + 1 * p), _mm_srli_si128(d0d1, 8)); _mm_storel_epi64((__m128i *)(s - 4 + 2 * p), d2d3); _mm_storel_epi64((__m128i *)(s - 4 + 3 * p), _mm_srli_si128(d2d3, 8)); _mm_storel_epi64((__m128i *)(s - 4 + 4 * p), d4d5); _mm_storel_epi64((__m128i *)(s - 4 + 5 * p), _mm_srli_si128(d4d5, 8)); _mm_storel_epi64((__m128i *)(s - 4 + 6 * p), d6d7); _mm_storel_epi64((__m128i *)(s - 4 + 7 * p), _mm_srli_si128(d6d7, 8)); } void aom_lpf_vertical_14_sse2(unsigned char *s, int p, const unsigned char *_blimit, const unsigned char *_limit, const unsigned char *_thresh) { __m128i q7p7, q6p6, q5p5, q4p4, q3p3, q2p2, q1p1, q0p0; __m128i x6, x5, x4, x3; __m128i pq0, pq1, pq2, pq3; __m128i blimit = _mm_load_si128((__m128i *)_blimit); __m128i limit = _mm_load_si128((__m128i *)_limit); __m128i thresh = _mm_load_si128((__m128i *)_thresh); x6 = _mm_loadu_si128((__m128i *)((s - 8) + 0 * p)); x5 = _mm_loadu_si128((__m128i *)((s - 8) + 1 * p)); x4 = _mm_loadu_si128((__m128i *)((s - 8) + 2 * p)); x3 = _mm_loadu_si128((__m128i *)((s - 8) + 3 * p)); transpose_pq_14_sse2(&x6, &x5, &x4, &x3, &q0p0, &q1p1, &q2p2, &q3p3, &q4p4, &q5p5, &q6p6, &q7p7); lpf_internal_14_sse2(&q6p6, &q5p5, &q4p4, &q3p3, &q2p2, &q1p1, &q0p0, &blimit, &limit, &thresh); transpose_pq_14_inv_sse2(&q7p7, &q6p6, &q5p5, &q4p4, &q3p3, &q2p2, &q1p1, &q0p0, &pq0, &pq1, &pq2, &pq3); _mm_storeu_si128((__m128i *)(s - 8 + 0 * p), pq0); _mm_storeu_si128((__m128i *)(s - 8 + 1 * p), pq1); _mm_storeu_si128((__m128i *)(s - 8 + 2 * p), pq2); _mm_storeu_si128((__m128i *)(s - 8 + 3 * p), pq3); } void aom_lpf_vertical_14_dual_sse2( unsigned char *s, int p, const uint8_t *_blimit0, const uint8_t *_limit0, const uint8_t *_thresh0, const uint8_t *_blimit1, const uint8_t *_limit1, const uint8_t *_thresh1) { __m128i q6p6, q5p5, q4p4, q3p3, q2p2, q1p1, q0p0; __m128i x7, x6, x5, x4, x3, x2, x1, x0; __m128i d0d1, d2d3, d4d5, d6d7, d8d9, d10d11, d12d13, d14d15; __m128i q0, q1, q2, q3, q7; __m128i p0p1, p2p3, p4p5, p6p7; __m128i blimit = _mm_unpacklo_epi32(_mm_load_si128((const __m128i *)_blimit0), _mm_load_si128((const __m128i *)_blimit1)); __m128i limit = _mm_unpacklo_epi32(_mm_load_si128((const __m128i *)_limit0), _mm_load_si128((const __m128i *)_limit1)); __m128i thresh = _mm_unpacklo_epi32(_mm_load_si128((const __m128i *)_thresh0), _mm_load_si128((const __m128i *)_thresh1)); x7 = _mm_loadu_si128((__m128i *)((s - 8) + 0 * p)); x6 = _mm_loadu_si128((__m128i *)((s - 8) + 1 * p)); x5 = _mm_loadu_si128((__m128i *)((s - 8) + 2 * p)); x4 = _mm_loadu_si128((__m128i *)((s - 8) + 3 * p)); x3 = _mm_loadu_si128((__m128i *)((s - 8) + 4 * p)); x2 = _mm_loadu_si128((__m128i *)((s - 8) + 5 * p)); x1 = _mm_loadu_si128((__m128i *)((s - 8) + 6 * p)); x0 = _mm_loadu_si128((__m128i *)((s - 8) + 7 * p)); transpose8x16_16x8_sse2(&x7, &x6, &x5, &x4, &x3, &x2, &x1, &x0, &d0d1, &d2d3, &d4d5, &d6d7, &d8d9, &d10d11, &d12d13, &d14d15); q6p6 = _mm_unpacklo_epi64(d2d3, _mm_srli_si128(d12d13, 8)); q5p5 = _mm_unpacklo_epi64(d4d5, _mm_srli_si128(d10d11, 8)); q4p4 = _mm_unpacklo_epi64(d6d7, _mm_srli_si128(d8d9, 8)); q3p3 = _mm_unpacklo_epi64(d8d9, _mm_srli_si128(d6d7, 8)); q2p2 = _mm_unpacklo_epi64(d10d11, _mm_srli_si128(d4d5, 8)); q1p1 = _mm_unpacklo_epi64(d12d13, _mm_srli_si128(d2d3, 8)); q0p0 = _mm_unpacklo_epi64(d14d15, _mm_srli_si128(d0d1, 8)); q7 = _mm_srli_si128(d14d15, 8); lpf_internal_14_dual_sse2(&q6p6, &q5p5, &q4p4, &q3p3, &q2p2, &q1p1, &q0p0, &blimit, &limit, &thresh); x0 = _mm_srli_si128(q0p0, 8); x1 = _mm_srli_si128(q1p1, 8); x2 = _mm_srli_si128(q2p2, 8); x3 = _mm_srli_si128(q3p3, 8); x4 = _mm_srli_si128(q4p4, 8); x5 = _mm_srli_si128(q5p5, 8); x6 = _mm_srli_si128(q6p6, 8); transpose16x8_8x16_sse2(&d0d1, &q6p6, &q5p5, &q4p4, &q3p3, &q2p2, &q1p1, &q0p0, &x0, &x1, &x2, &x3, &x4, &x5, &x6, &q7, &p0p1, &p2p3, &p4p5, &p6p7, &q0, &q1, &q2, &q3); _mm_storeu_si128((__m128i *)(s - 8 + 0 * p), p0p1); _mm_storeu_si128((__m128i *)(s - 8 + 1 * p), p2p3); _mm_storeu_si128((__m128i *)(s - 8 + 2 * p), p4p5); _mm_storeu_si128((__m128i *)(s - 8 + 3 * p), p6p7); _mm_storeu_si128((__m128i *)(s - 8 + 4 * p), q0); _mm_storeu_si128((__m128i *)(s - 8 + 5 * p), q1); _mm_storeu_si128((__m128i *)(s - 8 + 6 * p), q2); _mm_storeu_si128((__m128i *)(s - 8 + 7 * p), q3); } static INLINE __m128i filter_add2_sub2(const __m128i *const total, const __m128i *const a1, const __m128i *const a2, const __m128i *const s1, const __m128i *const s2) { __m128i x = _mm_add_epi16(*a1, *total); x = _mm_add_epi16(_mm_sub_epi16(x, _mm_add_epi16(*s1, *s2)), *a2); return x; } static INLINE __m128i filter8_mask(const __m128i *const flat, const __m128i *const other_filt, const __m128i *const f8_lo, const __m128i *const f8_hi) { const __m128i f8 = _mm_packus_epi16(_mm_srli_epi16(*f8_lo, 3), _mm_srli_epi16(*f8_hi, 3)); const __m128i result = _mm_and_si128(*flat, f8); return _mm_or_si128(_mm_andnot_si128(*flat, *other_filt), result); } static INLINE __m128i filter16_mask(const __m128i *const flat, const __m128i *const other_filt, const __m128i *const f_lo, const __m128i *const f_hi) { const __m128i f = _mm_packus_epi16(_mm_srli_epi16(*f_lo, 4), _mm_srli_epi16(*f_hi, 4)); const __m128i result = _mm_and_si128(*flat, f); return _mm_or_si128(_mm_andnot_si128(*flat, *other_filt), result); } void aom_lpf_horizontal_14_quad_sse2(unsigned char *s, int p, const unsigned char *_blimit0, const unsigned char *_limit0, const unsigned char *_thresh0) { const __m128i zero = _mm_setzero_si128(); const __m128i one = _mm_set1_epi8(1); const __m128i blimit_v = _mm_load_si128((const __m128i *)_blimit0); const __m128i limit_v = _mm_load_si128((const __m128i *)_limit0); const __m128i thresh_v = _mm_load_si128((const __m128i *)_thresh0); __m128i mask, hev, flat, flat2; __m128i p6, p5; __m128i p4, p3, p2, p1, p0, q0, q1, q2, q3, q4; __m128i q6, q5; __m128i op2, op1, op0, oq0, oq1, oq2; __m128i max_abs_p1p0q1q0; p6 = _mm_loadu_si128((__m128i *)(s - 7 * p)); p5 = _mm_loadu_si128((__m128i *)(s - 6 * p)); p4 = _mm_loadu_si128((__m128i *)(s - 5 * p)); p3 = _mm_loadu_si128((__m128i *)(s - 4 * p)); p2 = _mm_loadu_si128((__m128i *)(s - 3 * p)); p1 = _mm_loadu_si128((__m128i *)(s - 2 * p)); p0 = _mm_loadu_si128((__m128i *)(s - 1 * p)); q0 = _mm_loadu_si128((__m128i *)(s - 0 * p)); q1 = _mm_loadu_si128((__m128i *)(s + 1 * p)); q2 = _mm_loadu_si128((__m128i *)(s + 2 * p)); q3 = _mm_loadu_si128((__m128i *)(s + 3 * p)); q4 = _mm_loadu_si128((__m128i *)(s + 4 * p)); q5 = _mm_loadu_si128((__m128i *)(s + 5 * p)); q6 = _mm_loadu_si128((__m128i *)(s + 6 * p)); { const __m128i abs_p1p0 = abs_diff(p1, p0); const __m128i abs_q1q0 = abs_diff(q1, q0); const __m128i fe = _mm_set1_epi8((int8_t)0xfe); const __m128i ff = _mm_cmpeq_epi8(zero, zero); __m128i abs_p0q0 = abs_diff(p0, q0); __m128i abs_p1q1 = abs_diff(p1, q1); __m128i work; max_abs_p1p0q1q0 = _mm_max_epu8(abs_p1p0, abs_q1q0); abs_p0q0 = _mm_adds_epu8(abs_p0q0, abs_p0q0); abs_p1q1 = _mm_srli_epi16(_mm_and_si128(abs_p1q1, fe), 1); mask = _mm_subs_epu8(_mm_adds_epu8(abs_p0q0, abs_p1q1), blimit_v); mask = _mm_xor_si128(_mm_cmpeq_epi8(mask, zero), ff); // mask |= (abs(p0 - q0) * 2 + abs(p1 - q1) / 2 > blimit) * -1; mask = _mm_max_epu8(max_abs_p1p0q1q0, mask); // mask |= (abs(p1 - p0) > limit) * -1; // mask |= (abs(q1 - q0) > limit) * -1; work = _mm_max_epu8(abs_diff(p2, p1), abs_diff(p3, p2)); mask = _mm_max_epu8(work, mask); work = _mm_max_epu8(abs_diff(q2, q1), abs_diff(q3, q2)); mask = _mm_max_epu8(work, mask); mask = _mm_subs_epu8(mask, limit_v); mask = _mm_cmpeq_epi8(mask, zero); } if (0xffff == _mm_movemask_epi8(_mm_cmpeq_epi8(mask, zero))) return; { __m128i work; work = _mm_max_epu8(abs_diff(p2, p0), abs_diff(q2, q0)); flat = _mm_max_epu8(work, max_abs_p1p0q1q0); work = _mm_max_epu8(abs_diff(p3, p0), abs_diff(q3, q0)); flat = _mm_max_epu8(work, flat); work = _mm_max_epu8(abs_diff(p4, p0), abs_diff(q4, q0)); flat = _mm_subs_epu8(flat, one); flat = _mm_cmpeq_epi8(flat, zero); flat = _mm_and_si128(flat, mask); flat2 = _mm_max_epu8(abs_diff(p5, p0), abs_diff(q5, q0)); flat2 = _mm_max_epu8(work, flat2); work = _mm_max_epu8(abs_diff(p6, p0), abs_diff(q6, q0)); flat2 = _mm_max_epu8(work, flat2); flat2 = _mm_subs_epu8(flat2, one); flat2 = _mm_cmpeq_epi8(flat2, zero); flat2 = _mm_and_si128(flat2, flat); // flat2 & flat & mask } // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ // filter4 { const __m128i t4 = _mm_set1_epi8(4); const __m128i t3 = _mm_set1_epi8(3); const __m128i t80 = _mm_set1_epi8((int8_t)0x80); const __m128i te0 = _mm_set1_epi8((int8_t)0xe0); const __m128i t1f = _mm_set1_epi8(0x1f); const __m128i t1 = _mm_set1_epi8(0x1); const __m128i t7f = _mm_set1_epi8(0x7f); const __m128i ff = _mm_cmpeq_epi8(t4, t4); __m128i filt; __m128i work_a; __m128i filter1, filter2; op1 = _mm_xor_si128(p1, t80); op0 = _mm_xor_si128(p0, t80); oq0 = _mm_xor_si128(q0, t80); oq1 = _mm_xor_si128(q1, t80); hev = _mm_subs_epu8(max_abs_p1p0q1q0, thresh_v); hev = _mm_xor_si128(_mm_cmpeq_epi8(hev, zero), ff); filt = _mm_and_si128(_mm_subs_epi8(op1, oq1), hev); work_a = _mm_subs_epi8(oq0, op0); filt = _mm_adds_epi8(filt, work_a); filt = _mm_adds_epi8(filt, work_a); filt = _mm_adds_epi8(filt, work_a); filt = _mm_and_si128(filt, mask); filter1 = _mm_adds_epi8(filt, t4); filter2 = _mm_adds_epi8(filt, t3); work_a = _mm_cmpgt_epi8(zero, filter1); filter1 = _mm_srli_epi16(filter1, 3); work_a = _mm_and_si128(work_a, te0); filter1 = _mm_and_si128(filter1, t1f); filter1 = _mm_or_si128(filter1, work_a); oq0 = _mm_xor_si128(_mm_subs_epi8(oq0, filter1), t80); work_a = _mm_cmpgt_epi8(zero, filter2); filter2 = _mm_srli_epi16(filter2, 3); work_a = _mm_and_si128(work_a, te0); filter2 = _mm_and_si128(filter2, t1f); filter2 = _mm_or_si128(filter2, work_a); op0 = _mm_xor_si128(_mm_adds_epi8(op0, filter2), t80); filt = _mm_adds_epi8(filter1, t1); work_a = _mm_cmpgt_epi8(zero, filt); filt = _mm_srli_epi16(filt, 1); work_a = _mm_and_si128(work_a, t80); filt = _mm_and_si128(filt, t7f); filt = _mm_or_si128(filt, work_a); filt = _mm_andnot_si128(hev, filt); op1 = _mm_xor_si128(_mm_adds_epi8(op1, filt), t80); oq1 = _mm_xor_si128(_mm_subs_epi8(oq1, filt), t80); // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ // filter8 if (0xffff != _mm_movemask_epi8(_mm_cmpeq_epi8(flat, zero))) { const __m128i four = _mm_set1_epi16(4); const __m128i p3_lo = _mm_unpacklo_epi8(p3, zero); const __m128i p2_lo = _mm_unpacklo_epi8(p2, zero); const __m128i p1_lo = _mm_unpacklo_epi8(p1, zero); const __m128i p0_lo = _mm_unpacklo_epi8(p0, zero); const __m128i q0_lo = _mm_unpacklo_epi8(q0, zero); const __m128i q1_lo = _mm_unpacklo_epi8(q1, zero); const __m128i q2_lo = _mm_unpacklo_epi8(q2, zero); const __m128i q3_lo = _mm_unpacklo_epi8(q3, zero); const __m128i p3_hi = _mm_unpackhi_epi8(p3, zero); const __m128i p2_hi = _mm_unpackhi_epi8(p2, zero); const __m128i p1_hi = _mm_unpackhi_epi8(p1, zero); const __m128i p0_hi = _mm_unpackhi_epi8(p0, zero); const __m128i q0_hi = _mm_unpackhi_epi8(q0, zero); const __m128i q1_hi = _mm_unpackhi_epi8(q1, zero); const __m128i q2_hi = _mm_unpackhi_epi8(q2, zero); const __m128i q3_hi = _mm_unpackhi_epi8(q3, zero); __m128i f8_lo, f8_hi; f8_lo = _mm_add_epi16(_mm_add_epi16(p3_lo, four), _mm_add_epi16(p3_lo, p2_lo)); f8_lo = _mm_add_epi16(_mm_add_epi16(p3_lo, f8_lo), _mm_add_epi16(p2_lo, p1_lo)); f8_lo = _mm_add_epi16(_mm_add_epi16(p0_lo, q0_lo), f8_lo); f8_hi = _mm_add_epi16(_mm_add_epi16(p3_hi, four), _mm_add_epi16(p3_hi, p2_hi)); f8_hi = _mm_add_epi16(_mm_add_epi16(p3_hi, f8_hi), _mm_add_epi16(p2_hi, p1_hi)); f8_hi = _mm_add_epi16(_mm_add_epi16(p0_hi, q0_hi), f8_hi); op2 = filter8_mask(&flat, &p2, &f8_lo, &f8_hi); f8_lo = filter_add2_sub2(&f8_lo, &q1_lo, &p1_lo, &p2_lo, &p3_lo); f8_hi = filter_add2_sub2(&f8_hi, &q1_hi, &p1_hi, &p2_hi, &p3_hi); op1 = filter8_mask(&flat, &op1, &f8_lo, &f8_hi); f8_lo = filter_add2_sub2(&f8_lo, &q2_lo, &p0_lo, &p1_lo, &p3_lo); f8_hi = filter_add2_sub2(&f8_hi, &q2_hi, &p0_hi, &p1_hi, &p3_hi); op0 = filter8_mask(&flat, &op0, &f8_lo, &f8_hi); f8_lo = filter_add2_sub2(&f8_lo, &q3_lo, &q0_lo, &p0_lo, &p3_lo); f8_hi = filter_add2_sub2(&f8_hi, &q3_hi, &q0_hi, &p0_hi, &p3_hi); oq0 = filter8_mask(&flat, &oq0, &f8_lo, &f8_hi); f8_lo = filter_add2_sub2(&f8_lo, &q3_lo, &q1_lo, &q0_lo, &p2_lo); f8_hi = filter_add2_sub2(&f8_hi, &q3_hi, &q1_hi, &q0_hi, &p2_hi); oq1 = filter8_mask(&flat, &oq1, &f8_lo, &f8_hi); f8_lo = filter_add2_sub2(&f8_lo, &q3_lo, &q2_lo, &q1_lo, &p1_lo); f8_hi = filter_add2_sub2(&f8_hi, &q3_hi, &q2_hi, &q1_hi, &p1_hi); oq2 = filter8_mask(&flat, &q2, &f8_lo, &f8_hi); // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ // wide flat calculations if (0xffff != _mm_movemask_epi8(_mm_cmpeq_epi8(flat2, zero))) { const __m128i eight = _mm_set1_epi16(8); const __m128i p6_lo = _mm_unpacklo_epi8(p6, zero); const __m128i p5_lo = _mm_unpacklo_epi8(p5, zero); const __m128i p4_lo = _mm_unpacklo_epi8(p4, zero); const __m128i q4_lo = _mm_unpacklo_epi8(q4, zero); const __m128i q5_lo = _mm_unpacklo_epi8(q5, zero); const __m128i q6_lo = _mm_unpacklo_epi8(q6, zero); const __m128i p6_hi = _mm_unpackhi_epi8(p6, zero); const __m128i p5_hi = _mm_unpackhi_epi8(p5, zero); const __m128i p4_hi = _mm_unpackhi_epi8(p4, zero); const __m128i q4_hi = _mm_unpackhi_epi8(q4, zero); const __m128i q5_hi = _mm_unpackhi_epi8(q5, zero); const __m128i q6_hi = _mm_unpackhi_epi8(q6, zero); __m128i f_lo; __m128i f_hi; f_lo = _mm_sub_epi16(_mm_slli_epi16(p6_lo, 3), p6_lo); f_lo = _mm_add_epi16(_mm_slli_epi16(p5_lo, 1), f_lo); f_lo = _mm_add_epi16(_mm_slli_epi16(p4_lo, 1), f_lo); f_lo = _mm_add_epi16(_mm_add_epi16(p3_lo, f_lo), _mm_add_epi16(p2_lo, p1_lo)); f_lo = _mm_add_epi16(_mm_add_epi16(p0_lo, q0_lo), f_lo); f_lo = _mm_add_epi16(f_lo, eight); f_hi = _mm_sub_epi16(_mm_slli_epi16(p6_hi, 3), p6_hi); f_hi = _mm_add_epi16(_mm_slli_epi16(p5_hi, 1), f_hi); f_hi = _mm_add_epi16(_mm_slli_epi16(p4_hi, 1), f_hi); f_hi = _mm_add_epi16(_mm_add_epi16(p3_hi, f_hi), _mm_add_epi16(p2_hi, p1_hi)); f_hi = _mm_add_epi16(_mm_add_epi16(p0_hi, q0_hi), f_hi); f_hi = _mm_add_epi16(f_hi, eight); p5 = filter16_mask(&flat2, &p5, &f_lo, &f_hi); _mm_storeu_si128((__m128i *)(s - 6 * p), p5); f_lo = filter_add2_sub2(&f_lo, &q1_lo, &p3_lo, &p6_lo, &p6_lo); f_hi = filter_add2_sub2(&f_hi, &q1_hi, &p3_hi, &p6_hi, &p6_hi); p4 = filter16_mask(&flat2, &p4, &f_lo, &f_hi); _mm_storeu_si128((__m128i *)(s - 5 * p), p4); f_lo = filter_add2_sub2(&f_lo, &q2_lo, &p2_lo, &p6_lo, &p5_lo); f_hi = filter_add2_sub2(&f_hi, &q2_hi, &p2_hi, &p6_hi, &p5_hi); p3 = filter16_mask(&flat2, &p3, &f_lo, &f_hi); _mm_storeu_si128((__m128i *)(s - 4 * p), p3); f_lo = filter_add2_sub2(&f_lo, &q3_lo, &p1_lo, &p6_lo, &p4_lo); f_hi = filter_add2_sub2(&f_hi, &q3_hi, &p1_hi, &p6_hi, &p4_hi); op2 = filter16_mask(&flat2, &op2, &f_lo, &f_hi); _mm_storeu_si128((__m128i *)(s - 3 * p), op2); f_lo = filter_add2_sub2(&f_lo, &q4_lo, &p0_lo, &p6_lo, &p3_lo); f_hi = filter_add2_sub2(&f_hi, &q4_hi, &p0_hi, &p6_hi, &p3_hi); op1 = filter16_mask(&flat2, &op1, &f_lo, &f_hi); _mm_storeu_si128((__m128i *)(s - 2 * p), op1); f_lo = filter_add2_sub2(&f_lo, &q5_lo, &q0_lo, &p6_lo, &p2_lo); f_hi = filter_add2_sub2(&f_hi, &q5_hi, &q0_hi, &p6_hi, &p2_hi); op0 = filter16_mask(&flat2, &op0, &f_lo, &f_hi); _mm_storeu_si128((__m128i *)(s - 1 * p), op0); f_lo = filter_add2_sub2(&f_lo, &q6_lo, &q1_lo, &p6_lo, &p1_lo); f_hi = filter_add2_sub2(&f_hi, &q6_hi, &q1_hi, &p6_hi, &p1_hi); oq0 = filter16_mask(&flat2, &oq0, &f_lo, &f_hi); _mm_storeu_si128((__m128i *)(s - 0 * p), oq0); f_lo = filter_add2_sub2(&f_lo, &q6_lo, &q2_lo, &p5_lo, &p0_lo); f_hi = filter_add2_sub2(&f_hi, &q6_hi, &q2_hi, &p5_hi, &p0_hi); oq1 = filter16_mask(&flat2, &oq1, &f_lo, &f_hi); _mm_storeu_si128((__m128i *)(s + 1 * p), oq1); f_lo = filter_add2_sub2(&f_lo, &q6_lo, &q3_lo, &p4_lo, &q0_lo); f_hi = filter_add2_sub2(&f_hi, &q6_hi, &q3_hi, &p4_hi, &q0_hi); oq2 = filter16_mask(&flat2, &oq2, &f_lo, &f_hi); _mm_storeu_si128((__m128i *)(s + 2 * p), oq2); f_lo = filter_add2_sub2(&f_lo, &q6_lo, &q4_lo, &p3_lo, &q1_lo); f_hi = filter_add2_sub2(&f_hi, &q6_hi, &q4_hi, &p3_hi, &q1_hi); q3 = filter16_mask(&flat2, &q3, &f_lo, &f_hi); _mm_storeu_si128((__m128i *)(s + 3 * p), q3); f_lo = filter_add2_sub2(&f_lo, &q6_lo, &q5_lo, &p2_lo, &q2_lo); f_hi = filter_add2_sub2(&f_hi, &q6_hi, &q5_hi, &p2_hi, &q2_hi); q4 = filter16_mask(&flat2, &q4, &f_lo, &f_hi); _mm_storeu_si128((__m128i *)(s + 4 * p), q4); f_lo = filter_add2_sub2(&f_lo, &q6_lo, &q6_lo, &p1_lo, &q3_lo); f_hi = filter_add2_sub2(&f_hi, &q6_hi, &q6_hi, &p1_hi, &q3_hi); q5 = filter16_mask(&flat2, &q5, &f_lo, &f_hi); _mm_storeu_si128((__m128i *)(s + 5 * p), q5); } else { _mm_storeu_si128((__m128i *)(s - 3 * p), op2); _mm_storeu_si128((__m128i *)(s - 2 * p), op1); _mm_storeu_si128((__m128i *)(s - 1 * p), op0); _mm_storeu_si128((__m128i *)(s - 0 * p), oq0); _mm_storeu_si128((__m128i *)(s + 1 * p), oq1); _mm_storeu_si128((__m128i *)(s + 2 * p), oq2); } } else { _mm_storeu_si128((__m128i *)(s - 2 * p), op1); _mm_storeu_si128((__m128i *)(s - 1 * p), op0); _mm_storeu_si128((__m128i *)(s - 0 * p), oq0); _mm_storeu_si128((__m128i *)(s + 1 * p), oq1); } } } void aom_lpf_horizontal_8_quad_sse2(unsigned char *s, int p, const unsigned char *_blimit0, const unsigned char *_limit0, const unsigned char *_thresh0) { const __m128i zero = _mm_setzero_si128(); const __m128i one = _mm_set1_epi8(1); const __m128i blimit_v = _mm_load_si128((const __m128i *)_blimit0); const __m128i limit_v = _mm_load_si128((const __m128i *)_limit0); const __m128i thresh_v = _mm_load_si128((const __m128i *)_thresh0); __m128i mask, hev, flat; __m128i p3, p2, p1, p0, q0, q1, q2, q3; __m128i op2, op1, op0, oq0, oq1, oq2; __m128i max_abs_p1p0q1q0; p3 = _mm_loadu_si128((__m128i *)(s - 4 * p)); p2 = _mm_loadu_si128((__m128i *)(s - 3 * p)); p1 = _mm_loadu_si128((__m128i *)(s - 2 * p)); p0 = _mm_loadu_si128((__m128i *)(s - 1 * p)); q0 = _mm_loadu_si128((__m128i *)(s - 0 * p)); q1 = _mm_loadu_si128((__m128i *)(s + 1 * p)); q2 = _mm_loadu_si128((__m128i *)(s + 2 * p)); q3 = _mm_loadu_si128((__m128i *)(s + 3 * p)); { const __m128i abs_p1p0 = abs_diff(p1, p0); const __m128i abs_q1q0 = abs_diff(q1, q0); const __m128i fe = _mm_set1_epi8((int8_t)0xfe); const __m128i ff = _mm_cmpeq_epi8(zero, zero); __m128i abs_p0q0 = abs_diff(p0, q0); __m128i abs_p1q1 = abs_diff(p1, q1); __m128i work; max_abs_p1p0q1q0 = _mm_max_epu8(abs_p1p0, abs_q1q0); abs_p0q0 = _mm_adds_epu8(abs_p0q0, abs_p0q0); abs_p1q1 = _mm_srli_epi16(_mm_and_si128(abs_p1q1, fe), 1); mask = _mm_subs_epu8(_mm_adds_epu8(abs_p0q0, abs_p1q1), blimit_v); mask = _mm_xor_si128(_mm_cmpeq_epi8(mask, zero), ff); // mask |= (abs(p0 - q0) * 2 + abs(p1 - q1) / 2 > blimit) * -1; mask = _mm_max_epu8(max_abs_p1p0q1q0, mask); // mask |= (abs(p1 - p0) > limit) * -1; // mask |= (abs(q1 - q0) > limit) * -1; work = _mm_max_epu8(abs_diff(p2, p1), abs_diff(p3, p2)); mask = _mm_max_epu8(work, mask); work = _mm_max_epu8(abs_diff(q2, q1), abs_diff(q3, q2)); mask = _mm_max_epu8(work, mask); mask = _mm_subs_epu8(mask, limit_v); mask = _mm_cmpeq_epi8(mask, zero); } if (0xffff == _mm_movemask_epi8(_mm_cmpeq_epi8(mask, zero))) return; { __m128i work; work = _mm_max_epu8(abs_diff(p2, p0), abs_diff(q2, q0)); flat = _mm_max_epu8(work, max_abs_p1p0q1q0); work = _mm_max_epu8(abs_diff(p3, p0), abs_diff(q3, q0)); flat = _mm_max_epu8(work, flat); flat = _mm_subs_epu8(flat, one); flat = _mm_cmpeq_epi8(flat, zero); flat = _mm_and_si128(flat, mask); } // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ // filter4 { const __m128i t4 = _mm_set1_epi8(4); const __m128i t3 = _mm_set1_epi8(3); const __m128i t80 = _mm_set1_epi8((int8_t)0x80); const __m128i te0 = _mm_set1_epi8((int8_t)0xe0); const __m128i t1f = _mm_set1_epi8(0x1f); const __m128i t1 = _mm_set1_epi8(0x1); const __m128i t7f = _mm_set1_epi8(0x7f); const __m128i ff = _mm_cmpeq_epi8(t4, t4); __m128i filt; __m128i work_a; __m128i filter1, filter2; op1 = _mm_xor_si128(p1, t80); op0 = _mm_xor_si128(p0, t80); oq0 = _mm_xor_si128(q0, t80); oq1 = _mm_xor_si128(q1, t80); hev = _mm_subs_epu8(max_abs_p1p0q1q0, thresh_v); hev = _mm_xor_si128(_mm_cmpeq_epi8(hev, zero), ff); filt = _mm_and_si128(_mm_subs_epi8(op1, oq1), hev); work_a = _mm_subs_epi8(oq0, op0); filt = _mm_adds_epi8(filt, work_a); filt = _mm_adds_epi8(filt, work_a); filt = _mm_adds_epi8(filt, work_a); filt = _mm_and_si128(filt, mask); filter1 = _mm_adds_epi8(filt, t4); filter2 = _mm_adds_epi8(filt, t3); work_a = _mm_cmpgt_epi8(zero, filter1); filter1 = _mm_srli_epi16(filter1, 3); work_a = _mm_and_si128(work_a, te0); filter1 = _mm_and_si128(filter1, t1f); filter1 = _mm_or_si128(filter1, work_a); oq0 = _mm_xor_si128(_mm_subs_epi8(oq0, filter1), t80); work_a = _mm_cmpgt_epi8(zero, filter2); filter2 = _mm_srli_epi16(filter2, 3); work_a = _mm_and_si128(work_a, te0); filter2 = _mm_and_si128(filter2, t1f); filter2 = _mm_or_si128(filter2, work_a); op0 = _mm_xor_si128(_mm_adds_epi8(op0, filter2), t80); filt = _mm_adds_epi8(filter1, t1); work_a = _mm_cmpgt_epi8(zero, filt); filt = _mm_srli_epi16(filt, 1); work_a = _mm_and_si128(work_a, t80); filt = _mm_and_si128(filt, t7f); filt = _mm_or_si128(filt, work_a); filt = _mm_andnot_si128(hev, filt); op1 = _mm_xor_si128(_mm_adds_epi8(op1, filt), t80); oq1 = _mm_xor_si128(_mm_subs_epi8(oq1, filt), t80); // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ // filter8 if (0xffff != _mm_movemask_epi8(_mm_cmpeq_epi8(flat, zero))) { const __m128i four = _mm_set1_epi16(4); const __m128i p3_lo = _mm_unpacklo_epi8(p3, zero); const __m128i p2_lo = _mm_unpacklo_epi8(p2, zero); const __m128i p1_lo = _mm_unpacklo_epi8(p1, zero); const __m128i p0_lo = _mm_unpacklo_epi8(p0, zero); const __m128i q0_lo = _mm_unpacklo_epi8(q0, zero); const __m128i q1_lo = _mm_unpacklo_epi8(q1, zero); const __m128i q2_lo = _mm_unpacklo_epi8(q2, zero); const __m128i q3_lo = _mm_unpacklo_epi8(q3, zero); const __m128i p3_hi = _mm_unpackhi_epi8(p3, zero); const __m128i p2_hi = _mm_unpackhi_epi8(p2, zero); const __m128i p1_hi = _mm_unpackhi_epi8(p1, zero); const __m128i p0_hi = _mm_unpackhi_epi8(p0, zero); const __m128i q0_hi = _mm_unpackhi_epi8(q0, zero); const __m128i q1_hi = _mm_unpackhi_epi8(q1, zero); const __m128i q2_hi = _mm_unpackhi_epi8(q2, zero); const __m128i q3_hi = _mm_unpackhi_epi8(q3, zero); __m128i f8_lo, f8_hi; f8_lo = _mm_add_epi16(_mm_add_epi16(p3_lo, four), _mm_add_epi16(p3_lo, p2_lo)); f8_lo = _mm_add_epi16(_mm_add_epi16(p3_lo, f8_lo), _mm_add_epi16(p2_lo, p1_lo)); f8_lo = _mm_add_epi16(_mm_add_epi16(p0_lo, q0_lo), f8_lo); f8_hi = _mm_add_epi16(_mm_add_epi16(p3_hi, four), _mm_add_epi16(p3_hi, p2_hi)); f8_hi = _mm_add_epi16(_mm_add_epi16(p3_hi, f8_hi), _mm_add_epi16(p2_hi, p1_hi)); f8_hi = _mm_add_epi16(_mm_add_epi16(p0_hi, q0_hi), f8_hi); op2 = filter8_mask(&flat, &p2, &f8_lo, &f8_hi); _mm_storeu_si128((__m128i *)(s - 3 * p), op2); f8_lo = filter_add2_sub2(&f8_lo, &q1_lo, &p1_lo, &p2_lo, &p3_lo); f8_hi = filter_add2_sub2(&f8_hi, &q1_hi, &p1_hi, &p2_hi, &p3_hi); op1 = filter8_mask(&flat, &op1, &f8_lo, &f8_hi); _mm_storeu_si128((__m128i *)(s - 2 * p), op1); f8_lo = filter_add2_sub2(&f8_lo, &q2_lo, &p0_lo, &p1_lo, &p3_lo); f8_hi = filter_add2_sub2(&f8_hi, &q2_hi, &p0_hi, &p1_hi, &p3_hi); op0 = filter8_mask(&flat, &op0, &f8_lo, &f8_hi); _mm_storeu_si128((__m128i *)(s - 1 * p), op0); f8_lo = filter_add2_sub2(&f8_lo, &q3_lo, &q0_lo, &p0_lo, &p3_lo); f8_hi = filter_add2_sub2(&f8_hi, &q3_hi, &q0_hi, &p0_hi, &p3_hi); oq0 = filter8_mask(&flat, &oq0, &f8_lo, &f8_hi); _mm_storeu_si128((__m128i *)(s - 0 * p), oq0); f8_lo = filter_add2_sub2(&f8_lo, &q3_lo, &q1_lo, &q0_lo, &p2_lo); f8_hi = filter_add2_sub2(&f8_hi, &q3_hi, &q1_hi, &q0_hi, &p2_hi); oq1 = filter8_mask(&flat, &oq1, &f8_lo, &f8_hi); _mm_storeu_si128((__m128i *)(s + 1 * p), oq1); f8_lo = filter_add2_sub2(&f8_lo, &q3_lo, &q2_lo, &q1_lo, &p1_lo); f8_hi = filter_add2_sub2(&f8_hi, &q3_hi, &q2_hi, &q1_hi, &p1_hi); oq2 = filter8_mask(&flat, &q2, &f8_lo, &f8_hi); _mm_storeu_si128((__m128i *)(s + 2 * p), oq2); } else { _mm_storeu_si128((__m128i *)(s - 2 * p), op1); _mm_storeu_si128((__m128i *)(s - 1 * p), op0); _mm_storeu_si128((__m128i *)(s - 0 * p), oq0); _mm_storeu_si128((__m128i *)(s + 1 * p), oq1); } } } void aom_lpf_horizontal_6_quad_sse2(unsigned char *s, int p, const unsigned char *_blimit0, const unsigned char *_limit0, const unsigned char *_thresh0) { const __m128i zero = _mm_setzero_si128(); const __m128i one = _mm_set1_epi8(1); const __m128i blimit_v = _mm_load_si128((const __m128i *)_blimit0); const __m128i limit_v = _mm_load_si128((const __m128i *)_limit0); const __m128i thresh_v = _mm_load_si128((const __m128i *)_thresh0); __m128i mask, hev, flat; __m128i p2, p1, p0, q0, q1, q2; __m128i op1, op0, oq0, oq1; __m128i max_abs_p1p0q1q0; p2 = _mm_loadu_si128((__m128i *)(s - 3 * p)); p1 = _mm_loadu_si128((__m128i *)(s - 2 * p)); p0 = _mm_loadu_si128((__m128i *)(s - 1 * p)); q0 = _mm_loadu_si128((__m128i *)(s - 0 * p)); q1 = _mm_loadu_si128((__m128i *)(s + 1 * p)); q2 = _mm_loadu_si128((__m128i *)(s + 2 * p)); { const __m128i abs_p1p0 = abs_diff(p1, p0); const __m128i abs_q1q0 = abs_diff(q1, q0); const __m128i fe = _mm_set1_epi8((int8_t)0xfe); const __m128i ff = _mm_cmpeq_epi8(zero, zero); __m128i abs_p0q0 = abs_diff(p0, q0); __m128i abs_p1q1 = abs_diff(p1, q1); __m128i work; max_abs_p1p0q1q0 = _mm_max_epu8(abs_p1p0, abs_q1q0); abs_p0q0 = _mm_adds_epu8(abs_p0q0, abs_p0q0); abs_p1q1 = _mm_srli_epi16(_mm_and_si128(abs_p1q1, fe), 1); mask = _mm_subs_epu8(_mm_adds_epu8(abs_p0q0, abs_p1q1), blimit_v); mask = _mm_xor_si128(_mm_cmpeq_epi8(mask, zero), ff); // mask |= (abs(p0 - q0) * 2 + abs(p1 - q1) / 2 > blimit) * -1; mask = _mm_max_epu8(max_abs_p1p0q1q0, mask); // mask |= (abs(p1 - p0) > limit) * -1; // mask |= (abs(q1 - q0) > limit) * -1; work = _mm_max_epu8(abs_diff(p2, p1), abs_diff(q2, q1)); mask = _mm_max_epu8(work, mask); mask = _mm_subs_epu8(mask, limit_v); mask = _mm_cmpeq_epi8(mask, zero); } if (0xffff == _mm_movemask_epi8(_mm_cmpeq_epi8(mask, zero))) return; { __m128i work; work = _mm_max_epu8(abs_diff(p2, p0), abs_diff(q2, q0)); flat = _mm_max_epu8(work, max_abs_p1p0q1q0); flat = _mm_subs_epu8(flat, one); flat = _mm_cmpeq_epi8(flat, zero); flat = _mm_and_si128(flat, mask); } // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ // filter4 { const __m128i t4 = _mm_set1_epi8(4); const __m128i t3 = _mm_set1_epi8(3); const __m128i t80 = _mm_set1_epi8((int8_t)0x80); const __m128i te0 = _mm_set1_epi8((int8_t)0xe0); const __m128i t1f = _mm_set1_epi8(0x1f); const __m128i t1 = _mm_set1_epi8(0x1); const __m128i t7f = _mm_set1_epi8(0x7f); const __m128i ff = _mm_cmpeq_epi8(t4, t4); __m128i filt; __m128i work_a; __m128i filter1, filter2; op1 = _mm_xor_si128(p1, t80); op0 = _mm_xor_si128(p0, t80); oq0 = _mm_xor_si128(q0, t80); oq1 = _mm_xor_si128(q1, t80); hev = _mm_subs_epu8(max_abs_p1p0q1q0, thresh_v); hev = _mm_xor_si128(_mm_cmpeq_epi8(hev, zero), ff); filt = _mm_and_si128(_mm_subs_epi8(op1, oq1), hev); work_a = _mm_subs_epi8(oq0, op0); filt = _mm_adds_epi8(filt, work_a); filt = _mm_adds_epi8(filt, work_a); filt = _mm_adds_epi8(filt, work_a); filt = _mm_and_si128(filt, mask); filter1 = _mm_adds_epi8(filt, t4); filter2 = _mm_adds_epi8(filt, t3); work_a = _mm_cmpgt_epi8(zero, filter1); filter1 = _mm_srli_epi16(filter1, 3); work_a = _mm_and_si128(work_a, te0); filter1 = _mm_and_si128(filter1, t1f); filter1 = _mm_or_si128(filter1, work_a); oq0 = _mm_xor_si128(_mm_subs_epi8(oq0, filter1), t80); work_a = _mm_cmpgt_epi8(zero, filter2); filter2 = _mm_srli_epi16(filter2, 3); work_a = _mm_and_si128(work_a, te0); filter2 = _mm_and_si128(filter2, t1f); filter2 = _mm_or_si128(filter2, work_a); op0 = _mm_xor_si128(_mm_adds_epi8(op0, filter2), t80); filt = _mm_adds_epi8(filter1, t1); work_a = _mm_cmpgt_epi8(zero, filt); filt = _mm_srli_epi16(filt, 1); work_a = _mm_and_si128(work_a, t80); filt = _mm_and_si128(filt, t7f); filt = _mm_or_si128(filt, work_a); filt = _mm_andnot_si128(hev, filt); op1 = _mm_xor_si128(_mm_adds_epi8(op1, filt), t80); oq1 = _mm_xor_si128(_mm_subs_epi8(oq1, filt), t80); // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ // filter6 if (0xffff != _mm_movemask_epi8(_mm_cmpeq_epi8(flat, zero))) { const __m128i four = _mm_set1_epi16(4); const __m128i p2_lo = _mm_unpacklo_epi8(p2, zero); const __m128i p1_lo = _mm_unpacklo_epi8(p1, zero); const __m128i p0_lo = _mm_unpacklo_epi8(p0, zero); const __m128i q0_lo = _mm_unpacklo_epi8(q0, zero); const __m128i q1_lo = _mm_unpacklo_epi8(q1, zero); const __m128i q2_lo = _mm_unpacklo_epi8(q2, zero); const __m128i p2_hi = _mm_unpackhi_epi8(p2, zero); const __m128i p1_hi = _mm_unpackhi_epi8(p1, zero); const __m128i p0_hi = _mm_unpackhi_epi8(p0, zero); const __m128i q0_hi = _mm_unpackhi_epi8(q0, zero); const __m128i q1_hi = _mm_unpackhi_epi8(q1, zero); const __m128i q2_hi = _mm_unpackhi_epi8(q2, zero); __m128i f8_lo, f8_hi; f8_lo = _mm_add_epi16(_mm_add_epi16(p2_lo, four), _mm_add_epi16(p2_lo, p2_lo)); f8_lo = _mm_add_epi16(_mm_add_epi16(p1_lo, f8_lo), _mm_add_epi16(p1_lo, p0_lo)); f8_lo = _mm_add_epi16(_mm_add_epi16(p0_lo, q0_lo), f8_lo); f8_hi = _mm_add_epi16(_mm_add_epi16(p2_hi, four), _mm_add_epi16(p2_hi, p2_hi)); f8_hi = _mm_add_epi16(_mm_add_epi16(p1_hi, f8_hi), _mm_add_epi16(p1_hi, p0_hi)); f8_hi = _mm_add_epi16(_mm_add_epi16(p0_hi, q0_hi), f8_hi); op1 = filter8_mask(&flat, &op1, &f8_lo, &f8_hi); _mm_storeu_si128((__m128i *)(s - 2 * p), op1); f8_lo = filter_add2_sub2(&f8_lo, &q0_lo, &q1_lo, &p2_lo, &p2_lo); f8_hi = filter_add2_sub2(&f8_hi, &q0_hi, &q1_hi, &p2_hi, &p2_hi); op0 = filter8_mask(&flat, &op0, &f8_lo, &f8_hi); _mm_storeu_si128((__m128i *)(s - 1 * p), op0); f8_lo = filter_add2_sub2(&f8_lo, &q1_lo, &q2_lo, &p1_lo, &p2_lo); f8_hi = filter_add2_sub2(&f8_hi, &q1_hi, &q2_hi, &p1_hi, &p2_hi); oq0 = filter8_mask(&flat, &oq0, &f8_lo, &f8_hi); _mm_storeu_si128((__m128i *)(s - 0 * p), oq0); f8_lo = filter_add2_sub2(&f8_lo, &q2_lo, &q2_lo, &p0_lo, &p1_lo); f8_hi = filter_add2_sub2(&f8_hi, &q2_hi, &q2_hi, &p0_hi, &p1_hi); oq1 = filter8_mask(&flat, &oq1, &f8_lo, &f8_hi); _mm_storeu_si128((__m128i *)(s + 1 * p), oq1); } else { _mm_storeu_si128((__m128i *)(s - 2 * p), op1); _mm_storeu_si128((__m128i *)(s - 1 * p), op0); _mm_storeu_si128((__m128i *)(s - 0 * p), oq0); _mm_storeu_si128((__m128i *)(s + 1 * p), oq1); } } } void aom_lpf_horizontal_4_quad_sse2(unsigned char *s, int p, const unsigned char *_blimit0, const unsigned char *_limit0, const unsigned char *_thresh0) { const __m128i zero = _mm_setzero_si128(); const __m128i blimit_v = _mm_load_si128((const __m128i *)_blimit0); const __m128i limit_v = _mm_load_si128((const __m128i *)_limit0); const __m128i thresh_v = _mm_load_si128((const __m128i *)_thresh0); __m128i mask, hev; __m128i p1, p0, q0, q1; __m128i op1, op0, oq0, oq1; __m128i max_abs_p1p0q1q0; p1 = _mm_loadu_si128((__m128i *)(s - 2 * p)); p0 = _mm_loadu_si128((__m128i *)(s - 1 * p)); q0 = _mm_loadu_si128((__m128i *)(s - 0 * p)); q1 = _mm_loadu_si128((__m128i *)(s + 1 * p)); { const __m128i abs_p1p0 = abs_diff(p1, p0); const __m128i abs_q1q0 = abs_diff(q1, q0); const __m128i fe = _mm_set1_epi8((int8_t)0xfe); const __m128i ff = _mm_cmpeq_epi8(zero, zero); __m128i abs_p0q0 = abs_diff(p0, q0); __m128i abs_p1q1 = abs_diff(p1, q1); max_abs_p1p0q1q0 = _mm_max_epu8(abs_p1p0, abs_q1q0); abs_p0q0 = _mm_adds_epu8(abs_p0q0, abs_p0q0); abs_p1q1 = _mm_srli_epi16(_mm_and_si128(abs_p1q1, fe), 1); mask = _mm_subs_epu8(_mm_adds_epu8(abs_p0q0, abs_p1q1), blimit_v); mask = _mm_xor_si128(_mm_cmpeq_epi8(mask, zero), ff); // mask |= (abs(p0 - q0) * 2 + abs(p1 - q1) / 2 > blimit) * -1; mask = _mm_max_epu8(max_abs_p1p0q1q0, mask); // mask |= (abs(p1 - p0) > limit) * -1; // mask |= (abs(q1 - q0) > limit) * -1; mask = _mm_subs_epu8(mask, limit_v); mask = _mm_cmpeq_epi8(mask, zero); } if (0xffff == _mm_movemask_epi8(_mm_cmpeq_epi8(mask, zero))) return; // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ // filter4 { const __m128i t4 = _mm_set1_epi8(4); const __m128i t3 = _mm_set1_epi8(3); const __m128i t80 = _mm_set1_epi8((int8_t)0x80); const __m128i te0 = _mm_set1_epi8((int8_t)0xe0); const __m128i t1f = _mm_set1_epi8(0x1f); const __m128i t1 = _mm_set1_epi8(0x1); const __m128i t7f = _mm_set1_epi8(0x7f); const __m128i ff = _mm_cmpeq_epi8(t4, t4); __m128i filt; __m128i work_a; __m128i filter1, filter2; op1 = _mm_xor_si128(p1, t80); op0 = _mm_xor_si128(p0, t80); oq0 = _mm_xor_si128(q0, t80); oq1 = _mm_xor_si128(q1, t80); hev = _mm_subs_epu8(max_abs_p1p0q1q0, thresh_v); hev = _mm_xor_si128(_mm_cmpeq_epi8(hev, zero), ff); filt = _mm_and_si128(_mm_subs_epi8(op1, oq1), hev); work_a = _mm_subs_epi8(oq0, op0); filt = _mm_adds_epi8(filt, work_a); filt = _mm_adds_epi8(filt, work_a); filt = _mm_adds_epi8(filt, work_a); filt = _mm_and_si128(filt, mask); filter1 = _mm_adds_epi8(filt, t4); filter2 = _mm_adds_epi8(filt, t3); work_a = _mm_cmpgt_epi8(zero, filter1); filter1 = _mm_srli_epi16(filter1, 3); work_a = _mm_and_si128(work_a, te0); filter1 = _mm_and_si128(filter1, t1f); filter1 = _mm_or_si128(filter1, work_a); oq0 = _mm_xor_si128(_mm_subs_epi8(oq0, filter1), t80); work_a = _mm_cmpgt_epi8(zero, filter2); filter2 = _mm_srli_epi16(filter2, 3); work_a = _mm_and_si128(work_a, te0); filter2 = _mm_and_si128(filter2, t1f); filter2 = _mm_or_si128(filter2, work_a); op0 = _mm_xor_si128(_mm_adds_epi8(op0, filter2), t80); filt = _mm_adds_epi8(filter1, t1); work_a = _mm_cmpgt_epi8(zero, filt); filt = _mm_srli_epi16(filt, 1); work_a = _mm_and_si128(work_a, t80); filt = _mm_and_si128(filt, t7f); filt = _mm_or_si128(filt, work_a); filt = _mm_andnot_si128(hev, filt); op1 = _mm_xor_si128(_mm_adds_epi8(op1, filt), t80); oq1 = _mm_xor_si128(_mm_subs_epi8(oq1, filt), t80); _mm_storeu_si128((__m128i *)(s - 2 * p), op1); _mm_storeu_si128((__m128i *)(s - 1 * p), op0); _mm_storeu_si128((__m128i *)(s - 0 * p), oq0); _mm_storeu_si128((__m128i *)(s + 1 * p), oq1); } } void aom_lpf_vertical_14_quad_sse2(unsigned char *s, int pitch, const uint8_t *_blimit0, const uint8_t *_limit0, const uint8_t *_thresh0) { DECLARE_ALIGNED(16, unsigned char, t_dst[256]); // Transpose 16x16 transpose_16x8(s - 8, s - 8 + 8 * pitch, pitch, t_dst, 16); transpose_16x8(s, s + 8 * pitch, pitch, t_dst + 8 * 16, 16); // Loop filtering aom_lpf_horizontal_14_quad(t_dst + 8 * 16, 16, _blimit0, _limit0, _thresh0); // Transpose back transpose_16x8(t_dst, t_dst + 8 * 16, 16, s - 8, pitch); transpose_16x8(t_dst + 8, t_dst + 8 + 8 * 16, 16, s - 8 + 8 * pitch, pitch); } void aom_lpf_vertical_8_quad_sse2(uint8_t *s, int pitch, const uint8_t *_blimit0, const uint8_t *_limit0, const uint8_t *_thresh0) { DECLARE_ALIGNED(16, unsigned char, t_dst[16 * 8]); // Transpose 16x8 transpose_16x8(s - 4, s - 4 + pitch * 8, pitch, t_dst, 16); // Loop filtering aom_lpf_horizontal_8_quad(t_dst + 4 * 16, 16, _blimit0, _limit0, _thresh0); // Transpose back transpose_16x8_to_8x16(t_dst, 16, s - 4, pitch); } void aom_lpf_vertical_6_quad_sse2(uint8_t *s, int pitch, const uint8_t *_blimit0, const uint8_t *_limit0, const uint8_t *_thresh0) { DECLARE_ALIGNED(16, unsigned char, t_dst[16 * 8]); // Transpose 16x8:: (wxh) 8x16 to 16x8 transpose_16x8(s - 4, s - 4 + pitch * 8, pitch, t_dst, 16); // Loop filtering aom_lpf_horizontal_6_quad(t_dst + 4 * 16, 16, _blimit0, _limit0, _thresh0); // Transpose back:: (wxh) 16x8 to 8x16 transpose_16x8_to_8x16(t_dst, 16, s - 4, pitch); } void aom_lpf_vertical_4_quad_sse2(uint8_t *s, int pitch, const uint8_t *_blimit0, const uint8_t *_limit0, const uint8_t *_thresh0) { DECLARE_ALIGNED(16, unsigned char, t_dst[16 * 8]); // Transpose 16x8 transpose_16x8(s - 4, s - 4 + pitch * 8, pitch, t_dst, 16); // Loop filtering aom_lpf_horizontal_4_quad_sse2(t_dst + 4 * 16, 16, _blimit0, _limit0, _thresh0); // Transpose back transpose_16x8_to_8x16(t_dst, 16, s - 4, pitch); }