| /* |
| * 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 <emmintrin.h> // SSE2 |
| |
| #include "./aom_dsp_rtcd.h" |
| #include "aom_dsp/x86/lpf_common_sse2.h" |
| #include "aom_ports/emmintrin_compat.h" |
| #include "aom_ports/mem.h" |
| |
| static INLINE void pixel_clamp(const __m128i *min, const __m128i *max, |
| __m128i *pixel) { |
| __m128i clamped, mask; |
| |
| mask = _mm_cmpgt_epi16(*pixel, *max); |
| clamped = _mm_andnot_si128(mask, *pixel); |
| mask = _mm_and_si128(mask, *max); |
| clamped = _mm_or_si128(mask, clamped); |
| |
| mask = _mm_cmpgt_epi16(clamped, *min); |
| clamped = _mm_and_si128(mask, clamped); |
| mask = _mm_andnot_si128(mask, *min); |
| *pixel = _mm_or_si128(clamped, mask); |
| } |
| |
| // TODO(debargha, peter): Break up large functions into smaller ones |
| // in this file. |
| void aom_highbd_lpf_horizontal_edge_8_sse2(uint16_t *s, int p, |
| const uint8_t *_blimit, |
| const uint8_t *_limit, |
| const uint8_t *_thresh, int bd) { |
| const __m128i zero = _mm_set1_epi16(0); |
| const __m128i one = _mm_set1_epi16(1); |
| __m128i blimit, limit, thresh; |
| __m128i q7, p7, q6, p6, q5, p5, q4, p4, q3, p3, q2, p2, q1, p1, q0, p0; |
| __m128i mask, hev, flat, flat2, abs_p1p0, abs_q1q0; |
| __m128i ps1, qs1, ps0, qs0; |
| __m128i abs_p0q0, abs_p1q1, ffff, work; |
| __m128i filt, work_a, filter1, filter2; |
| __m128i flat2_q6, flat2_p6, flat2_q5, flat2_p5, flat2_q4, flat2_p4; |
| __m128i flat2_q3, flat2_p3, flat2_q2, flat2_p2, flat2_q1, flat2_p1; |
| __m128i flat2_q0, flat2_p0; |
| __m128i flat_q2, flat_p2, flat_q1, flat_p1, flat_q0, flat_p0; |
| __m128i pixelFilter_p, pixelFilter_q; |
| __m128i pixetFilter_p2p1p0, pixetFilter_q2q1q0; |
| __m128i sum_p7, sum_q7, sum_p3, sum_q3; |
| __m128i t4, t3, t80, t1; |
| __m128i eight, four; |
| |
| if (bd == 8) { |
| blimit = _mm_unpacklo_epi8(_mm_load_si128((const __m128i *)_blimit), zero); |
| limit = _mm_unpacklo_epi8(_mm_load_si128((const __m128i *)_limit), zero); |
| thresh = _mm_unpacklo_epi8(_mm_load_si128((const __m128i *)_thresh), zero); |
| } else if (bd == 10) { |
| blimit = _mm_slli_epi16( |
| _mm_unpacklo_epi8(_mm_load_si128((const __m128i *)_blimit), zero), 2); |
| limit = _mm_slli_epi16( |
| _mm_unpacklo_epi8(_mm_load_si128((const __m128i *)_limit), zero), 2); |
| thresh = _mm_slli_epi16( |
| _mm_unpacklo_epi8(_mm_load_si128((const __m128i *)_thresh), zero), 2); |
| } else { // bd == 12 |
| blimit = _mm_slli_epi16( |
| _mm_unpacklo_epi8(_mm_load_si128((const __m128i *)_blimit), zero), 4); |
| limit = _mm_slli_epi16( |
| _mm_unpacklo_epi8(_mm_load_si128((const __m128i *)_limit), zero), 4); |
| thresh = _mm_slli_epi16( |
| _mm_unpacklo_epi8(_mm_load_si128((const __m128i *)_thresh), zero), 4); |
| } |
| |
| q4 = _mm_load_si128((__m128i *)(s + 4 * p)); |
| p4 = _mm_load_si128((__m128i *)(s - 5 * p)); |
| q3 = _mm_load_si128((__m128i *)(s + 3 * p)); |
| p3 = _mm_load_si128((__m128i *)(s - 4 * p)); |
| q2 = _mm_load_si128((__m128i *)(s + 2 * p)); |
| p2 = _mm_load_si128((__m128i *)(s - 3 * p)); |
| q1 = _mm_load_si128((__m128i *)(s + 1 * p)); |
| p1 = _mm_load_si128((__m128i *)(s - 2 * p)); |
| q0 = _mm_load_si128((__m128i *)(s + 0 * p)); |
| p0 = _mm_load_si128((__m128i *)(s - 1 * p)); |
| |
| // highbd_filter_mask |
| abs_p1p0 = _mm_or_si128(_mm_subs_epu16(p1, p0), _mm_subs_epu16(p0, p1)); |
| abs_q1q0 = _mm_or_si128(_mm_subs_epu16(q1, q0), _mm_subs_epu16(q0, q1)); |
| |
| ffff = _mm_cmpeq_epi16(abs_p1p0, abs_p1p0); |
| |
| abs_p0q0 = _mm_or_si128(_mm_subs_epu16(p0, q0), _mm_subs_epu16(q0, p0)); |
| abs_p1q1 = _mm_or_si128(_mm_subs_epu16(p1, q1), _mm_subs_epu16(q1, p1)); |
| |
| // highbd_hev_mask (in C code this is actually called from highbd_filter4) |
| flat = _mm_max_epi16(abs_p1p0, abs_q1q0); |
| hev = _mm_subs_epu16(flat, thresh); |
| hev = _mm_xor_si128(_mm_cmpeq_epi16(hev, zero), ffff); |
| |
| abs_p0q0 = _mm_adds_epu16(abs_p0q0, abs_p0q0); // abs(p0 - q0) * 2 |
| abs_p1q1 = _mm_srli_epi16(abs_p1q1, 1); // abs(p1 - q1) / 2 |
| mask = _mm_subs_epu16(_mm_adds_epu16(abs_p0q0, abs_p1q1), blimit); |
| mask = _mm_xor_si128(_mm_cmpeq_epi16(mask, zero), ffff); |
| mask = _mm_and_si128(mask, _mm_adds_epu16(limit, one)); |
| work = _mm_max_epi16( |
| _mm_or_si128(_mm_subs_epu16(p1, p0), _mm_subs_epu16(p0, p1)), |
| _mm_or_si128(_mm_subs_epu16(q1, q0), _mm_subs_epu16(q0, q1))); |
| mask = _mm_max_epi16(work, mask); |
| work = _mm_max_epi16( |
| _mm_or_si128(_mm_subs_epu16(p2, p1), _mm_subs_epu16(p1, p2)), |
| _mm_or_si128(_mm_subs_epu16(q2, q1), _mm_subs_epu16(q1, q2))); |
| mask = _mm_max_epi16(work, mask); |
| work = _mm_max_epi16( |
| _mm_or_si128(_mm_subs_epu16(p3, p2), _mm_subs_epu16(p2, p3)), |
| _mm_or_si128(_mm_subs_epu16(q3, q2), _mm_subs_epu16(q2, q3))); |
| mask = _mm_max_epi16(work, mask); |
| |
| mask = _mm_subs_epu16(mask, limit); |
| mask = _mm_cmpeq_epi16(mask, zero); // return ~mask |
| |
| // lp filter |
| // highbd_filter4 |
| t4 = _mm_set1_epi16(4); |
| t3 = _mm_set1_epi16(3); |
| if (bd == 8) |
| t80 = _mm_set1_epi16(0x80); |
| else if (bd == 10) |
| t80 = _mm_set1_epi16(0x200); |
| else // bd == 12 |
| t80 = _mm_set1_epi16(0x800); |
| |
| t1 = _mm_set1_epi16(0x1); |
| |
| ps1 = _mm_subs_epi16(p1, t80); |
| qs1 = _mm_subs_epi16(q1, t80); |
| ps0 = _mm_subs_epi16(p0, t80); |
| qs0 = _mm_subs_epi16(q0, t80); |
| |
| const __m128i pmax = |
| _mm_subs_epi16(_mm_subs_epi16(_mm_slli_epi16(one, bd), one), t80); |
| const __m128i pmin = _mm_subs_epi16(zero, t80); |
| |
| filt = _mm_subs_epi16(ps1, qs1); |
| pixel_clamp(&pmin, &pmax, &filt); |
| filt = _mm_and_si128(filt, hev); |
| |
| work_a = _mm_subs_epi16(qs0, ps0); |
| filt = _mm_adds_epi16(filt, work_a); |
| filt = _mm_adds_epi16(filt, work_a); |
| |
| filt = _mm_adds_epi16(filt, work_a); |
| pixel_clamp(&pmin, &pmax, &filt); |
| |
| filt = _mm_and_si128(filt, mask); |
| |
| filter1 = _mm_adds_epi16(filt, t4); |
| pixel_clamp(&pmin, &pmax, &filter1); |
| filter2 = _mm_adds_epi16(filt, t3); |
| pixel_clamp(&pmin, &pmax, &filter2); |
| |
| // Filter1 >> 3 |
| filter1 = _mm_srai_epi16(filter1, 0x3); |
| filter2 = _mm_srai_epi16(filter2, 0x3); |
| |
| qs0 = _mm_subs_epi16(qs0, filter1); |
| pixel_clamp(&pmin, &pmax, &qs0); |
| qs0 = _mm_adds_epi16(qs0, t80); |
| |
| ps0 = _mm_adds_epi16(ps0, filter2); |
| pixel_clamp(&pmin, &pmax, &ps0); |
| ps0 = _mm_adds_epi16(ps0, t80); |
| |
| filt = _mm_adds_epi16(filter1, t1); |
| filt = _mm_srai_epi16(filt, 1); |
| filt = _mm_andnot_si128(hev, filt); |
| |
| qs1 = _mm_subs_epi16(qs1, filt); |
| pixel_clamp(&pmin, &pmax, &qs1); |
| qs1 = _mm_adds_epi16(qs1, t80); |
| |
| ps1 = _mm_adds_epi16(ps1, filt); |
| pixel_clamp(&pmin, &pmax, &ps1); |
| ps1 = _mm_adds_epi16(ps1, t80); |
| |
| // end highbd_filter4 |
| // loopfilter done |
| |
| // highbd_flat_mask4 |
| flat = _mm_max_epi16( |
| _mm_or_si128(_mm_subs_epu16(p2, p0), _mm_subs_epu16(p0, p2)), |
| _mm_or_si128(_mm_subs_epu16(p3, p0), _mm_subs_epu16(p0, p3))); |
| work = _mm_max_epi16( |
| _mm_or_si128(_mm_subs_epu16(q2, q0), _mm_subs_epu16(q0, q2)), |
| _mm_or_si128(_mm_subs_epu16(q3, q0), _mm_subs_epu16(q0, q3))); |
| flat = _mm_max_epi16(work, flat); |
| work = _mm_max_epi16(abs_p1p0, abs_q1q0); |
| flat = _mm_max_epi16(work, flat); |
| |
| if (bd == 8) |
| flat = _mm_subs_epu16(flat, one); |
| else if (bd == 10) |
| flat = _mm_subs_epu16(flat, _mm_slli_epi16(one, 2)); |
| else // bd == 12 |
| flat = _mm_subs_epu16(flat, _mm_slli_epi16(one, 4)); |
| |
| flat = _mm_cmpeq_epi16(flat, zero); |
| // end flat_mask4 |
| |
| // flat & mask = flat && mask (as used in filter8) |
| // (because, in both vars, each block of 16 either all 1s or all 0s) |
| flat = _mm_and_si128(flat, mask); |
| |
| p5 = _mm_load_si128((__m128i *)(s - 6 * p)); |
| q5 = _mm_load_si128((__m128i *)(s + 5 * p)); |
| p6 = _mm_load_si128((__m128i *)(s - 7 * p)); |
| q6 = _mm_load_si128((__m128i *)(s + 6 * p)); |
| p7 = _mm_load_si128((__m128i *)(s - 8 * p)); |
| q7 = _mm_load_si128((__m128i *)(s + 7 * p)); |
| |
| // highbd_flat_mask5 (arguments passed in are p0, q0, p4-p7, q4-q7 |
| // but referred to as p0-p4 & q0-q4 in fn) |
| flat2 = _mm_max_epi16( |
| _mm_or_si128(_mm_subs_epu16(p4, p0), _mm_subs_epu16(p0, p4)), |
| _mm_or_si128(_mm_subs_epu16(q4, q0), _mm_subs_epu16(q0, q4))); |
| |
| work = _mm_max_epi16( |
| _mm_or_si128(_mm_subs_epu16(p5, p0), _mm_subs_epu16(p0, p5)), |
| _mm_or_si128(_mm_subs_epu16(q5, q0), _mm_subs_epu16(q0, q5))); |
| flat2 = _mm_max_epi16(work, flat2); |
| |
| work = _mm_max_epi16( |
| _mm_or_si128(_mm_subs_epu16(p6, p0), _mm_subs_epu16(p0, p6)), |
| _mm_or_si128(_mm_subs_epu16(q6, q0), _mm_subs_epu16(q0, q6))); |
| flat2 = _mm_max_epi16(work, flat2); |
| |
| work = _mm_max_epi16( |
| _mm_or_si128(_mm_subs_epu16(p7, p0), _mm_subs_epu16(p0, p7)), |
| _mm_or_si128(_mm_subs_epu16(q7, q0), _mm_subs_epu16(q0, q7))); |
| flat2 = _mm_max_epi16(work, flat2); |
| |
| if (bd == 8) |
| flat2 = _mm_subs_epu16(flat2, one); |
| else if (bd == 10) |
| flat2 = _mm_subs_epu16(flat2, _mm_slli_epi16(one, 2)); |
| else // bd == 12 |
| flat2 = _mm_subs_epu16(flat2, _mm_slli_epi16(one, 4)); |
| |
| flat2 = _mm_cmpeq_epi16(flat2, zero); |
| flat2 = _mm_and_si128(flat2, flat); // flat2 & flat & mask |
| // end highbd_flat_mask5 |
| |
| // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ |
| // flat and wide flat calculations |
| eight = _mm_set1_epi16(8); |
| four = _mm_set1_epi16(4); |
| |
| pixelFilter_p = _mm_add_epi16(_mm_add_epi16(p6, p5), _mm_add_epi16(p4, p3)); |
| pixelFilter_q = _mm_add_epi16(_mm_add_epi16(q6, q5), _mm_add_epi16(q4, q3)); |
| |
| pixetFilter_p2p1p0 = _mm_add_epi16(p0, _mm_add_epi16(p2, p1)); |
| pixelFilter_p = _mm_add_epi16(pixelFilter_p, pixetFilter_p2p1p0); |
| |
| pixetFilter_q2q1q0 = _mm_add_epi16(q0, _mm_add_epi16(q2, q1)); |
| 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)); |
| flat2_p0 = |
| _mm_srli_epi16(_mm_add_epi16(pixelFilter_p, _mm_add_epi16(p7, p0)), 4); |
| flat2_q0 = |
| _mm_srli_epi16(_mm_add_epi16(pixelFilter_p, _mm_add_epi16(q7, q0)), 4); |
| flat_p0 = _mm_srli_epi16( |
| _mm_add_epi16(pixetFilter_p2p1p0, _mm_add_epi16(p3, p0)), 3); |
| flat_q0 = _mm_srli_epi16( |
| _mm_add_epi16(pixetFilter_p2p1p0, _mm_add_epi16(q3, q0)), 3); |
| |
| sum_p7 = _mm_add_epi16(p7, p7); |
| sum_q7 = _mm_add_epi16(q7, q7); |
| sum_p3 = _mm_add_epi16(p3, p3); |
| sum_q3 = _mm_add_epi16(q3, q3); |
| |
| pixelFilter_q = _mm_sub_epi16(pixelFilter_p, p6); |
| pixelFilter_p = _mm_sub_epi16(pixelFilter_p, q6); |
| flat2_p1 = _mm_srli_epi16( |
| _mm_add_epi16(pixelFilter_p, _mm_add_epi16(sum_p7, p1)), 4); |
| flat2_q1 = _mm_srli_epi16( |
| _mm_add_epi16(pixelFilter_q, _mm_add_epi16(sum_q7, q1)), 4); |
| |
| pixetFilter_q2q1q0 = _mm_sub_epi16(pixetFilter_p2p1p0, p2); |
| pixetFilter_p2p1p0 = _mm_sub_epi16(pixetFilter_p2p1p0, q2); |
| flat_p1 = _mm_srli_epi16( |
| _mm_add_epi16(pixetFilter_p2p1p0, _mm_add_epi16(sum_p3, p1)), 3); |
| flat_q1 = _mm_srli_epi16( |
| _mm_add_epi16(pixetFilter_q2q1q0, _mm_add_epi16(sum_q3, q1)), 3); |
| |
| sum_p7 = _mm_add_epi16(sum_p7, p7); |
| sum_q7 = _mm_add_epi16(sum_q7, q7); |
| sum_p3 = _mm_add_epi16(sum_p3, p3); |
| sum_q3 = _mm_add_epi16(sum_q3, q3); |
| |
| pixelFilter_p = _mm_sub_epi16(pixelFilter_p, q5); |
| pixelFilter_q = _mm_sub_epi16(pixelFilter_q, p5); |
| flat2_p2 = _mm_srli_epi16( |
| _mm_add_epi16(pixelFilter_p, _mm_add_epi16(sum_p7, p2)), 4); |
| flat2_q2 = _mm_srli_epi16( |
| _mm_add_epi16(pixelFilter_q, _mm_add_epi16(sum_q7, q2)), 4); |
| |
| pixetFilter_p2p1p0 = _mm_sub_epi16(pixetFilter_p2p1p0, q1); |
| pixetFilter_q2q1q0 = _mm_sub_epi16(pixetFilter_q2q1q0, p1); |
| flat_p2 = _mm_srli_epi16( |
| _mm_add_epi16(pixetFilter_p2p1p0, _mm_add_epi16(sum_p3, p2)), 3); |
| flat_q2 = _mm_srli_epi16( |
| _mm_add_epi16(pixetFilter_q2q1q0, _mm_add_epi16(sum_q3, q2)), 3); |
| |
| sum_p7 = _mm_add_epi16(sum_p7, p7); |
| sum_q7 = _mm_add_epi16(sum_q7, q7); |
| pixelFilter_p = _mm_sub_epi16(pixelFilter_p, q4); |
| pixelFilter_q = _mm_sub_epi16(pixelFilter_q, p4); |
| flat2_p3 = _mm_srli_epi16( |
| _mm_add_epi16(pixelFilter_p, _mm_add_epi16(sum_p7, p3)), 4); |
| flat2_q3 = _mm_srli_epi16( |
| _mm_add_epi16(pixelFilter_q, _mm_add_epi16(sum_q7, q3)), 4); |
| |
| sum_p7 = _mm_add_epi16(sum_p7, p7); |
| sum_q7 = _mm_add_epi16(sum_q7, q7); |
| pixelFilter_p = _mm_sub_epi16(pixelFilter_p, q3); |
| pixelFilter_q = _mm_sub_epi16(pixelFilter_q, p3); |
| flat2_p4 = _mm_srli_epi16( |
| _mm_add_epi16(pixelFilter_p, _mm_add_epi16(sum_p7, p4)), 4); |
| flat2_q4 = _mm_srli_epi16( |
| _mm_add_epi16(pixelFilter_q, _mm_add_epi16(sum_q7, q4)), 4); |
| |
| sum_p7 = _mm_add_epi16(sum_p7, p7); |
| sum_q7 = _mm_add_epi16(sum_q7, q7); |
| pixelFilter_p = _mm_sub_epi16(pixelFilter_p, q2); |
| pixelFilter_q = _mm_sub_epi16(pixelFilter_q, p2); |
| flat2_p5 = _mm_srli_epi16( |
| _mm_add_epi16(pixelFilter_p, _mm_add_epi16(sum_p7, p5)), 4); |
| flat2_q5 = _mm_srli_epi16( |
| _mm_add_epi16(pixelFilter_q, _mm_add_epi16(sum_q7, q5)), 4); |
| |
| sum_p7 = _mm_add_epi16(sum_p7, p7); |
| sum_q7 = _mm_add_epi16(sum_q7, q7); |
| pixelFilter_p = _mm_sub_epi16(pixelFilter_p, q1); |
| pixelFilter_q = _mm_sub_epi16(pixelFilter_q, p1); |
| flat2_p6 = _mm_srli_epi16( |
| _mm_add_epi16(pixelFilter_p, _mm_add_epi16(sum_p7, p6)), 4); |
| flat2_q6 = _mm_srli_epi16( |
| _mm_add_epi16(pixelFilter_q, _mm_add_epi16(sum_q7, q6)), 4); |
| |
| // wide flat |
| // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ |
| |
| // highbd_filter8 |
| p2 = _mm_andnot_si128(flat, p2); |
| // p2 remains unchanged if !(flat && mask) |
| flat_p2 = _mm_and_si128(flat, flat_p2); |
| // when (flat && mask) |
| p2 = _mm_or_si128(p2, flat_p2); // full list of p2 values |
| q2 = _mm_andnot_si128(flat, q2); |
| flat_q2 = _mm_and_si128(flat, flat_q2); |
| q2 = _mm_or_si128(q2, flat_q2); // full list of q2 values |
| |
| ps1 = _mm_andnot_si128(flat, ps1); |
| // p1 takes the value assigned to in in filter4 if !(flat && mask) |
| flat_p1 = _mm_and_si128(flat, flat_p1); |
| // when (flat && mask) |
| p1 = _mm_or_si128(ps1, flat_p1); // full list of p1 values |
| qs1 = _mm_andnot_si128(flat, qs1); |
| flat_q1 = _mm_and_si128(flat, flat_q1); |
| q1 = _mm_or_si128(qs1, flat_q1); // full list of q1 values |
| |
| ps0 = _mm_andnot_si128(flat, ps0); |
| // p0 takes the value assigned to in in filter4 if !(flat && mask) |
| flat_p0 = _mm_and_si128(flat, flat_p0); |
| // when (flat && mask) |
| p0 = _mm_or_si128(ps0, flat_p0); // full list of p0 values |
| qs0 = _mm_andnot_si128(flat, qs0); |
| flat_q0 = _mm_and_si128(flat, flat_q0); |
| q0 = _mm_or_si128(qs0, flat_q0); // full list of q0 values |
| // end highbd_filter8 |
| |
| // highbd_filter16 |
| p6 = _mm_andnot_si128(flat2, p6); |
| // p6 remains unchanged if !(flat2 && flat && mask) |
| flat2_p6 = _mm_and_si128(flat2, flat2_p6); |
| // get values for when (flat2 && flat && mask) |
| p6 = _mm_or_si128(p6, flat2_p6); // full list of p6 values |
| q6 = _mm_andnot_si128(flat2, q6); |
| // q6 remains unchanged if !(flat2 && flat && mask) |
| flat2_q6 = _mm_and_si128(flat2, flat2_q6); |
| // get values for when (flat2 && flat && mask) |
| q6 = _mm_or_si128(q6, flat2_q6); // full list of q6 values |
| _mm_store_si128((__m128i *)(s - 7 * p), p6); |
| _mm_store_si128((__m128i *)(s + 6 * p), q6); |
| |
| p5 = _mm_andnot_si128(flat2, p5); |
| // p5 remains unchanged if !(flat2 && flat && mask) |
| flat2_p5 = _mm_and_si128(flat2, flat2_p5); |
| // get values for when (flat2 && flat && mask) |
| p5 = _mm_or_si128(p5, flat2_p5); |
| // full list of p5 values |
| q5 = _mm_andnot_si128(flat2, q5); |
| // q5 remains unchanged if !(flat2 && flat && mask) |
| flat2_q5 = _mm_and_si128(flat2, flat2_q5); |
| // get values for when (flat2 && flat && mask) |
| q5 = _mm_or_si128(q5, flat2_q5); |
| // full list of q5 values |
| _mm_store_si128((__m128i *)(s - 6 * p), p5); |
| _mm_store_si128((__m128i *)(s + 5 * p), q5); |
| |
| p4 = _mm_andnot_si128(flat2, p4); |
| // p4 remains unchanged if !(flat2 && flat && mask) |
| flat2_p4 = _mm_and_si128(flat2, flat2_p4); |
| // get values for when (flat2 && flat && mask) |
| p4 = _mm_or_si128(p4, flat2_p4); // full list of p4 values |
| q4 = _mm_andnot_si128(flat2, q4); |
| // q4 remains unchanged if !(flat2 && flat && mask) |
| flat2_q4 = _mm_and_si128(flat2, flat2_q4); |
| // get values for when (flat2 && flat && mask) |
| q4 = _mm_or_si128(q4, flat2_q4); // full list of q4 values |
| _mm_store_si128((__m128i *)(s - 5 * p), p4); |
| _mm_store_si128((__m128i *)(s + 4 * p), q4); |
| |
| p3 = _mm_andnot_si128(flat2, p3); |
| // p3 takes value from highbd_filter8 if !(flat2 && flat && mask) |
| flat2_p3 = _mm_and_si128(flat2, flat2_p3); |
| // get values for when (flat2 && flat && mask) |
| p3 = _mm_or_si128(p3, flat2_p3); // full list of p3 values |
| q3 = _mm_andnot_si128(flat2, q3); |
| // q3 takes value from highbd_filter8 if !(flat2 && flat && mask) |
| flat2_q3 = _mm_and_si128(flat2, flat2_q3); |
| // get values for when (flat2 && flat && mask) |
| q3 = _mm_or_si128(q3, flat2_q3); // full list of q3 values |
| _mm_store_si128((__m128i *)(s - 4 * p), p3); |
| _mm_store_si128((__m128i *)(s + 3 * p), q3); |
| |
| p2 = _mm_andnot_si128(flat2, p2); |
| // p2 takes value from highbd_filter8 if !(flat2 && flat && mask) |
| flat2_p2 = _mm_and_si128(flat2, flat2_p2); |
| // get values for when (flat2 && flat && mask) |
| p2 = _mm_or_si128(p2, flat2_p2); |
| // full list of p2 values |
| q2 = _mm_andnot_si128(flat2, q2); |
| // q2 takes value from highbd_filter8 if !(flat2 && flat && mask) |
| flat2_q2 = _mm_and_si128(flat2, flat2_q2); |
| // get values for when (flat2 && flat && mask) |
| q2 = _mm_or_si128(q2, flat2_q2); // full list of q2 values |
| _mm_store_si128((__m128i *)(s - 3 * p), p2); |
| _mm_store_si128((__m128i *)(s + 2 * p), q2); |
| |
| p1 = _mm_andnot_si128(flat2, p1); |
| // p1 takes value from highbd_filter8 if !(flat2 && flat && mask) |
| flat2_p1 = _mm_and_si128(flat2, flat2_p1); |
| // get values for when (flat2 && flat && mask) |
| p1 = _mm_or_si128(p1, flat2_p1); // full list of p1 values |
| q1 = _mm_andnot_si128(flat2, q1); |
| // q1 takes value from highbd_filter8 if !(flat2 && flat && mask) |
| flat2_q1 = _mm_and_si128(flat2, flat2_q1); |
| // get values for when (flat2 && flat && mask) |
| q1 = _mm_or_si128(q1, flat2_q1); // full list of q1 values |
| _mm_store_si128((__m128i *)(s - 2 * p), p1); |
| _mm_store_si128((__m128i *)(s + 1 * p), q1); |
| |
| p0 = _mm_andnot_si128(flat2, p0); |
| // p0 takes value from highbd_filter8 if !(flat2 && flat && mask) |
| flat2_p0 = _mm_and_si128(flat2, flat2_p0); |
| // get values for when (flat2 && flat && mask) |
| p0 = _mm_or_si128(p0, flat2_p0); // full list of p0 values |
| q0 = _mm_andnot_si128(flat2, q0); |
| // q0 takes value from highbd_filter8 if !(flat2 && flat && mask) |
| flat2_q0 = _mm_and_si128(flat2, flat2_q0); |
| // get values for when (flat2 && flat && mask) |
| q0 = _mm_or_si128(q0, flat2_q0); // full list of q0 values |
| _mm_store_si128((__m128i *)(s - 1 * p), p0); |
| _mm_store_si128((__m128i *)(s - 0 * p), q0); |
| } |
| |
| void aom_highbd_lpf_horizontal_edge_16_sse2(uint16_t *s, int p, |
| const uint8_t *_blimit, |
| const uint8_t *_limit, |
| const uint8_t *_thresh, int bd) { |
| aom_highbd_lpf_horizontal_edge_8_sse2(s, p, _blimit, _limit, _thresh, bd); |
| aom_highbd_lpf_horizontal_edge_8_sse2(s + 8, p, _blimit, _limit, _thresh, bd); |
| } |
| |
| void aom_highbd_lpf_horizontal_8_sse2(uint16_t *s, int p, |
| const uint8_t *_blimit, |
| const uint8_t *_limit, |
| const uint8_t *_thresh, int bd) { |
| DECLARE_ALIGNED(16, uint16_t, flat_op2[16]); |
| DECLARE_ALIGNED(16, uint16_t, flat_op1[16]); |
| DECLARE_ALIGNED(16, uint16_t, flat_op0[16]); |
| DECLARE_ALIGNED(16, uint16_t, flat_oq2[16]); |
| DECLARE_ALIGNED(16, uint16_t, flat_oq1[16]); |
| DECLARE_ALIGNED(16, uint16_t, flat_oq0[16]); |
| const __m128i zero = _mm_set1_epi16(0); |
| __m128i blimit, limit, thresh; |
| __m128i mask, hev, flat; |
| __m128i p3 = _mm_load_si128((__m128i *)(s - 4 * p)); |
| __m128i q3 = _mm_load_si128((__m128i *)(s + 3 * p)); |
| __m128i p2 = _mm_load_si128((__m128i *)(s - 3 * p)); |
| __m128i q2 = _mm_load_si128((__m128i *)(s + 2 * p)); |
| __m128i p1 = _mm_load_si128((__m128i *)(s - 2 * p)); |
| __m128i q1 = _mm_load_si128((__m128i *)(s + 1 * p)); |
| __m128i p0 = _mm_load_si128((__m128i *)(s - 1 * p)); |
| __m128i q0 = _mm_load_si128((__m128i *)(s + 0 * p)); |
| const __m128i one = _mm_set1_epi16(1); |
| const __m128i ffff = _mm_cmpeq_epi16(one, one); |
| __m128i abs_p1q1, abs_p0q0, abs_q1q0, abs_p1p0, work; |
| const __m128i four = _mm_set1_epi16(4); |
| __m128i workp_a, workp_b, workp_shft; |
| |
| const __m128i t4 = _mm_set1_epi16(4); |
| const __m128i t3 = _mm_set1_epi16(3); |
| __m128i t80; |
| const __m128i t1 = _mm_set1_epi16(0x1); |
| __m128i ps1, ps0, qs0, qs1; |
| __m128i filt; |
| __m128i work_a; |
| __m128i filter1, filter2; |
| |
| if (bd == 8) { |
| blimit = _mm_unpacklo_epi8(_mm_load_si128((const __m128i *)_blimit), zero); |
| limit = _mm_unpacklo_epi8(_mm_load_si128((const __m128i *)_limit), zero); |
| thresh = _mm_unpacklo_epi8(_mm_load_si128((const __m128i *)_thresh), zero); |
| t80 = _mm_set1_epi16(0x80); |
| } else if (bd == 10) { |
| blimit = _mm_slli_epi16( |
| _mm_unpacklo_epi8(_mm_load_si128((const __m128i *)_blimit), zero), 2); |
| limit = _mm_slli_epi16( |
| _mm_unpacklo_epi8(_mm_load_si128((const __m128i *)_limit), zero), 2); |
| thresh = _mm_slli_epi16( |
| _mm_unpacklo_epi8(_mm_load_si128((const __m128i *)_thresh), zero), 2); |
| t80 = _mm_set1_epi16(0x200); |
| } else { // bd == 12 |
| blimit = _mm_slli_epi16( |
| _mm_unpacklo_epi8(_mm_load_si128((const __m128i *)_blimit), zero), 4); |
| limit = _mm_slli_epi16( |
| _mm_unpacklo_epi8(_mm_load_si128((const __m128i *)_limit), zero), 4); |
| thresh = _mm_slli_epi16( |
| _mm_unpacklo_epi8(_mm_load_si128((const __m128i *)_thresh), zero), 4); |
| t80 = _mm_set1_epi16(0x800); |
| } |
| |
| ps1 = _mm_subs_epi16(p1, t80); |
| ps0 = _mm_subs_epi16(p0, t80); |
| qs0 = _mm_subs_epi16(q0, t80); |
| qs1 = _mm_subs_epi16(q1, t80); |
| |
| // filter_mask and hev_mask |
| abs_p1p0 = _mm_or_si128(_mm_subs_epu16(p1, p0), _mm_subs_epu16(p0, p1)); |
| abs_q1q0 = _mm_or_si128(_mm_subs_epu16(q1, q0), _mm_subs_epu16(q0, q1)); |
| |
| abs_p0q0 = _mm_or_si128(_mm_subs_epu16(p0, q0), _mm_subs_epu16(q0, p0)); |
| abs_p1q1 = _mm_or_si128(_mm_subs_epu16(p1, q1), _mm_subs_epu16(q1, p1)); |
| flat = _mm_max_epi16(abs_p1p0, abs_q1q0); |
| hev = _mm_subs_epu16(flat, thresh); |
| hev = _mm_xor_si128(_mm_cmpeq_epi16(hev, zero), ffff); |
| |
| abs_p0q0 = _mm_adds_epu16(abs_p0q0, abs_p0q0); |
| abs_p1q1 = _mm_srli_epi16(abs_p1q1, 1); |
| mask = _mm_subs_epu16(_mm_adds_epu16(abs_p0q0, abs_p1q1), blimit); |
| mask = _mm_xor_si128(_mm_cmpeq_epi16(mask, zero), ffff); |
| // mask |= (abs(p0 - q0) * 2 + abs(p1 - q1) / 2 > blimit) * -1; |
| // So taking maximums continues to work: |
| mask = _mm_and_si128(mask, _mm_adds_epu16(limit, one)); |
| mask = _mm_max_epi16(abs_p1p0, mask); |
| // mask |= (abs(p1 - p0) > limit) * -1; |
| mask = _mm_max_epi16(abs_q1q0, mask); |
| // mask |= (abs(q1 - q0) > limit) * -1; |
| |
| work = _mm_max_epi16( |
| _mm_or_si128(_mm_subs_epu16(p2, p1), _mm_subs_epu16(p1, p2)), |
| _mm_or_si128(_mm_subs_epu16(q2, q1), _mm_subs_epu16(q1, q2))); |
| mask = _mm_max_epi16(work, mask); |
| work = _mm_max_epi16( |
| _mm_or_si128(_mm_subs_epu16(p3, p2), _mm_subs_epu16(p2, p3)), |
| _mm_or_si128(_mm_subs_epu16(q3, q2), _mm_subs_epu16(q2, q3))); |
| mask = _mm_max_epi16(work, mask); |
| mask = _mm_subs_epu16(mask, limit); |
| mask = _mm_cmpeq_epi16(mask, zero); |
| |
| // flat_mask4 |
| flat = _mm_max_epi16( |
| _mm_or_si128(_mm_subs_epu16(p2, p0), _mm_subs_epu16(p0, p2)), |
| _mm_or_si128(_mm_subs_epu16(q2, q0), _mm_subs_epu16(q0, q2))); |
| work = _mm_max_epi16( |
| _mm_or_si128(_mm_subs_epu16(p3, p0), _mm_subs_epu16(p0, p3)), |
| _mm_or_si128(_mm_subs_epu16(q3, q0), _mm_subs_epu16(q0, q3))); |
| flat = _mm_max_epi16(work, flat); |
| flat = _mm_max_epi16(abs_p1p0, flat); |
| flat = _mm_max_epi16(abs_q1q0, flat); |
| |
| if (bd == 8) |
| flat = _mm_subs_epu16(flat, one); |
| else if (bd == 10) |
| flat = _mm_subs_epu16(flat, _mm_slli_epi16(one, 2)); |
| else // bd == 12 |
| flat = _mm_subs_epu16(flat, _mm_slli_epi16(one, 4)); |
| |
| flat = _mm_cmpeq_epi16(flat, zero); |
| flat = _mm_and_si128(flat, mask); // flat & mask |
| |
| // Added before shift for rounding part of ROUND_POWER_OF_TWO |
| |
| workp_a = _mm_add_epi16(_mm_add_epi16(p3, p3), _mm_add_epi16(p2, p1)); |
| workp_a = _mm_add_epi16(_mm_add_epi16(workp_a, four), p0); |
| workp_b = _mm_add_epi16(_mm_add_epi16(q0, p2), p3); |
| workp_shft = _mm_srli_epi16(_mm_add_epi16(workp_a, workp_b), 3); |
| _mm_store_si128((__m128i *)&flat_op2[0], workp_shft); |
| |
| workp_b = _mm_add_epi16(_mm_add_epi16(q0, q1), p1); |
| workp_shft = _mm_srli_epi16(_mm_add_epi16(workp_a, workp_b), 3); |
| _mm_store_si128((__m128i *)&flat_op1[0], workp_shft); |
| |
| workp_a = _mm_add_epi16(_mm_sub_epi16(workp_a, p3), q2); |
| workp_b = _mm_add_epi16(_mm_sub_epi16(workp_b, p1), p0); |
| workp_shft = _mm_srli_epi16(_mm_add_epi16(workp_a, workp_b), 3); |
| _mm_store_si128((__m128i *)&flat_op0[0], workp_shft); |
| |
| workp_a = _mm_add_epi16(_mm_sub_epi16(workp_a, p3), q3); |
| workp_b = _mm_add_epi16(_mm_sub_epi16(workp_b, p0), q0); |
| workp_shft = _mm_srli_epi16(_mm_add_epi16(workp_a, workp_b), 3); |
| _mm_store_si128((__m128i *)&flat_oq0[0], workp_shft); |
| |
| workp_a = _mm_add_epi16(_mm_sub_epi16(workp_a, p2), q3); |
| workp_b = _mm_add_epi16(_mm_sub_epi16(workp_b, q0), q1); |
| workp_shft = _mm_srli_epi16(_mm_add_epi16(workp_a, workp_b), 3); |
| _mm_store_si128((__m128i *)&flat_oq1[0], workp_shft); |
| |
| workp_a = _mm_add_epi16(_mm_sub_epi16(workp_a, p1), q3); |
| workp_b = _mm_add_epi16(_mm_sub_epi16(workp_b, q1), q2); |
| workp_shft = _mm_srli_epi16(_mm_add_epi16(workp_a, workp_b), 3); |
| _mm_store_si128((__m128i *)&flat_oq2[0], workp_shft); |
| |
| // lp filter |
| const __m128i pmax = |
| _mm_subs_epi16(_mm_subs_epi16(_mm_slli_epi16(one, bd), one), t80); |
| const __m128i pmin = _mm_subs_epi16(zero, t80); |
| |
| filt = _mm_subs_epi16(ps1, qs1); |
| pixel_clamp(&pmin, &pmax, &filt); |
| |
| filt = _mm_and_si128(filt, hev); |
| work_a = _mm_subs_epi16(qs0, ps0); |
| filt = _mm_adds_epi16(filt, work_a); |
| filt = _mm_adds_epi16(filt, work_a); |
| filt = _mm_adds_epi16(filt, work_a); |
| // (aom_filter + 3 * (qs0 - ps0)) & mask |
| pixel_clamp(&pmin, &pmax, &filt); |
| filt = _mm_and_si128(filt, mask); |
| |
| filter1 = _mm_adds_epi16(filt, t4); |
| filter2 = _mm_adds_epi16(filt, t3); |
| |
| // Filter1 >> 3 |
| pixel_clamp(&pmin, &pmax, &filter1); |
| filter1 = _mm_srai_epi16(filter1, 3); |
| |
| // Filter2 >> 3 |
| pixel_clamp(&pmin, &pmax, &filter2); |
| filter2 = _mm_srai_epi16(filter2, 3); |
| |
| // filt >> 1 |
| filt = _mm_adds_epi16(filter1, t1); |
| filt = _mm_srai_epi16(filt, 1); |
| filt = _mm_andnot_si128(hev, filt); |
| |
| work_a = _mm_subs_epi16(qs0, filter1); |
| pixel_clamp(&pmin, &pmax, &work_a); |
| work_a = _mm_adds_epi16(work_a, t80); |
| q0 = _mm_load_si128((__m128i *)flat_oq0); |
| work_a = _mm_andnot_si128(flat, work_a); |
| q0 = _mm_and_si128(flat, q0); |
| q0 = _mm_or_si128(work_a, q0); |
| |
| work_a = _mm_subs_epi16(qs1, filt); |
| pixel_clamp(&pmin, &pmax, &work_a); |
| work_a = _mm_adds_epi16(work_a, t80); |
| q1 = _mm_load_si128((__m128i *)flat_oq1); |
| work_a = _mm_andnot_si128(flat, work_a); |
| q1 = _mm_and_si128(flat, q1); |
| q1 = _mm_or_si128(work_a, q1); |
| |
| work_a = _mm_loadu_si128((__m128i *)(s + 2 * p)); |
| q2 = _mm_load_si128((__m128i *)flat_oq2); |
| work_a = _mm_andnot_si128(flat, work_a); |
| q2 = _mm_and_si128(flat, q2); |
| q2 = _mm_or_si128(work_a, q2); |
| |
| work_a = _mm_adds_epi16(ps0, filter2); |
| pixel_clamp(&pmin, &pmax, &work_a); |
| work_a = _mm_adds_epi16(work_a, t80); |
| p0 = _mm_load_si128((__m128i *)flat_op0); |
| work_a = _mm_andnot_si128(flat, work_a); |
| p0 = _mm_and_si128(flat, p0); |
| p0 = _mm_or_si128(work_a, p0); |
| |
| work_a = _mm_adds_epi16(ps1, filt); |
| pixel_clamp(&pmin, &pmax, &work_a); |
| work_a = _mm_adds_epi16(work_a, t80); |
| p1 = _mm_load_si128((__m128i *)flat_op1); |
| work_a = _mm_andnot_si128(flat, work_a); |
| p1 = _mm_and_si128(flat, p1); |
| p1 = _mm_or_si128(work_a, p1); |
| |
| work_a = _mm_loadu_si128((__m128i *)(s - 3 * p)); |
| p2 = _mm_load_si128((__m128i *)flat_op2); |
| work_a = _mm_andnot_si128(flat, work_a); |
| p2 = _mm_and_si128(flat, p2); |
| p2 = _mm_or_si128(work_a, p2); |
| |
| _mm_store_si128((__m128i *)(s - 3 * p), p2); |
| _mm_store_si128((__m128i *)(s - 2 * p), p1); |
| _mm_store_si128((__m128i *)(s - 1 * p), p0); |
| _mm_store_si128((__m128i *)(s + 0 * p), q0); |
| _mm_store_si128((__m128i *)(s + 1 * p), q1); |
| _mm_store_si128((__m128i *)(s + 2 * p), q2); |
| } |
| |
| void aom_highbd_lpf_horizontal_8_dual_sse2( |
| uint16_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, int bd) { |
| aom_highbd_lpf_horizontal_8_sse2(s, p, _blimit0, _limit0, _thresh0, bd); |
| aom_highbd_lpf_horizontal_8_sse2(s + 8, p, _blimit1, _limit1, _thresh1, bd); |
| } |
| |
| void aom_highbd_lpf_horizontal_4_sse2(uint16_t *s, int p, |
| const uint8_t *_blimit, |
| const uint8_t *_limit, |
| const uint8_t *_thresh, int bd) { |
| const __m128i zero = _mm_set1_epi16(0); |
| __m128i blimit, limit, thresh; |
| __m128i mask, hev, flat; |
| __m128i p3 = _mm_loadu_si128((__m128i *)(s - 4 * p)); |
| __m128i p2 = _mm_loadu_si128((__m128i *)(s - 3 * p)); |
| __m128i p1 = _mm_loadu_si128((__m128i *)(s - 2 * p)); |
| __m128i p0 = _mm_loadu_si128((__m128i *)(s - 1 * p)); |
| __m128i q0 = _mm_loadu_si128((__m128i *)(s - 0 * p)); |
| __m128i q1 = _mm_loadu_si128((__m128i *)(s + 1 * p)); |
| __m128i q2 = _mm_loadu_si128((__m128i *)(s + 2 * p)); |
| __m128i q3 = _mm_loadu_si128((__m128i *)(s + 3 * p)); |
| const __m128i abs_p1p0 = |
| _mm_or_si128(_mm_subs_epu16(p1, p0), _mm_subs_epu16(p0, p1)); |
| const __m128i abs_q1q0 = |
| _mm_or_si128(_mm_subs_epu16(q1, q0), _mm_subs_epu16(q0, q1)); |
| const __m128i ffff = _mm_cmpeq_epi16(abs_p1p0, abs_p1p0); |
| const __m128i one = _mm_set1_epi16(1); |
| __m128i abs_p0q0 = |
| _mm_or_si128(_mm_subs_epu16(p0, q0), _mm_subs_epu16(q0, p0)); |
| __m128i abs_p1q1 = |
| _mm_or_si128(_mm_subs_epu16(p1, q1), _mm_subs_epu16(q1, p1)); |
| __m128i work; |
| const __m128i t4 = _mm_set1_epi16(4); |
| const __m128i t3 = _mm_set1_epi16(3); |
| __m128i t80; |
| __m128i tff80; |
| __m128i tffe0; |
| __m128i t1f; |
| // equivalent to shifting 0x1f left by bitdepth - 8 |
| // and setting new bits to 1 |
| const __m128i t1 = _mm_set1_epi16(0x1); |
| __m128i t7f; |
| // equivalent to shifting 0x7f left by bitdepth - 8 |
| // and setting new bits to 1 |
| __m128i ps1, ps0, qs0, qs1; |
| __m128i filt; |
| __m128i work_a; |
| __m128i filter1, filter2; |
| |
| if (bd == 8) { |
| blimit = _mm_unpacklo_epi8(_mm_load_si128((const __m128i *)_blimit), zero); |
| limit = _mm_unpacklo_epi8(_mm_load_si128((const __m128i *)_limit), zero); |
| thresh = _mm_unpacklo_epi8(_mm_load_si128((const __m128i *)_thresh), zero); |
| t80 = _mm_set1_epi16(0x80); |
| tff80 = _mm_set1_epi16(0xff80); |
| tffe0 = _mm_set1_epi16(0xffe0); |
| t1f = _mm_srli_epi16(_mm_set1_epi16(0x1fff), 8); |
| t7f = _mm_srli_epi16(_mm_set1_epi16(0x7fff), 8); |
| } else if (bd == 10) { |
| blimit = _mm_slli_epi16( |
| _mm_unpacklo_epi8(_mm_load_si128((const __m128i *)_blimit), zero), 2); |
| limit = _mm_slli_epi16( |
| _mm_unpacklo_epi8(_mm_load_si128((const __m128i *)_limit), zero), 2); |
| thresh = _mm_slli_epi16( |
| _mm_unpacklo_epi8(_mm_load_si128((const __m128i *)_thresh), zero), 2); |
| t80 = _mm_slli_epi16(_mm_set1_epi16(0x80), 2); |
| tff80 = _mm_slli_epi16(_mm_set1_epi16(0xff80), 2); |
| tffe0 = _mm_slli_epi16(_mm_set1_epi16(0xffe0), 2); |
| t1f = _mm_srli_epi16(_mm_set1_epi16(0x1fff), 6); |
| t7f = _mm_srli_epi16(_mm_set1_epi16(0x7fff), 6); |
| } else { // bd == 12 |
| blimit = _mm_slli_epi16( |
| _mm_unpacklo_epi8(_mm_load_si128((const __m128i *)_blimit), zero), 4); |
| limit = _mm_slli_epi16( |
| _mm_unpacklo_epi8(_mm_load_si128((const __m128i *)_limit), zero), 4); |
| thresh = _mm_slli_epi16( |
| _mm_unpacklo_epi8(_mm_load_si128((const __m128i *)_thresh), zero), 4); |
| t80 = _mm_slli_epi16(_mm_set1_epi16(0x80), 4); |
| tff80 = _mm_slli_epi16(_mm_set1_epi16(0xff80), 4); |
| tffe0 = _mm_slli_epi16(_mm_set1_epi16(0xffe0), 4); |
| t1f = _mm_srli_epi16(_mm_set1_epi16(0x1fff), 4); |
| t7f = _mm_srli_epi16(_mm_set1_epi16(0x7fff), 4); |
| } |
| |
| ps1 = _mm_subs_epi16(_mm_loadu_si128((__m128i *)(s - 2 * p)), t80); |
| ps0 = _mm_subs_epi16(_mm_loadu_si128((__m128i *)(s - 1 * p)), t80); |
| qs0 = _mm_subs_epi16(_mm_loadu_si128((__m128i *)(s + 0 * p)), t80); |
| qs1 = _mm_subs_epi16(_mm_loadu_si128((__m128i *)(s + 1 * p)), t80); |
| |
| // filter_mask and hev_mask |
| flat = _mm_max_epi16(abs_p1p0, abs_q1q0); |
| hev = _mm_subs_epu16(flat, thresh); |
| hev = _mm_xor_si128(_mm_cmpeq_epi16(hev, zero), ffff); |
| |
| abs_p0q0 = _mm_adds_epu16(abs_p0q0, abs_p0q0); |
| abs_p1q1 = _mm_srli_epi16(abs_p1q1, 1); |
| mask = _mm_subs_epu16(_mm_adds_epu16(abs_p0q0, abs_p1q1), blimit); |
| mask = _mm_xor_si128(_mm_cmpeq_epi16(mask, zero), ffff); |
| // mask |= (abs(p0 - q0) * 2 + abs(p1 - q1) / 2 > blimit) * -1; |
| // So taking maximums continues to work: |
| mask = _mm_and_si128(mask, _mm_adds_epu16(limit, one)); |
| mask = _mm_max_epi16(flat, mask); |
| // mask |= (abs(p1 - p0) > limit) * -1; |
| // mask |= (abs(q1 - q0) > limit) * -1; |
| work = _mm_max_epi16( |
| _mm_or_si128(_mm_subs_epu16(p2, p1), _mm_subs_epu16(p1, p2)), |
| _mm_or_si128(_mm_subs_epu16(p3, p2), _mm_subs_epu16(p2, p3))); |
| mask = _mm_max_epi16(work, mask); |
| work = _mm_max_epi16( |
| _mm_or_si128(_mm_subs_epu16(q2, q1), _mm_subs_epu16(q1, q2)), |
| _mm_or_si128(_mm_subs_epu16(q3, q2), _mm_subs_epu16(q2, q3))); |
| mask = _mm_max_epi16(work, mask); |
| mask = _mm_subs_epu16(mask, limit); |
| mask = _mm_cmpeq_epi16(mask, zero); |
| |
| // filter4 |
| const __m128i pmax = |
| _mm_subs_epi16(_mm_subs_epi16(_mm_slli_epi16(one, bd), one), t80); |
| const __m128i pmin = _mm_subs_epi16(zero, t80); |
| |
| filt = _mm_subs_epi16(ps1, qs1); |
| pixel_clamp(&pmin, &pmax, &filt); |
| filt = _mm_and_si128(filt, hev); |
| work_a = _mm_subs_epi16(qs0, ps0); |
| filt = _mm_adds_epi16(filt, work_a); |
| filt = _mm_adds_epi16(filt, work_a); |
| filt = _mm_adds_epi16(filt, work_a); |
| pixel_clamp(&pmin, &pmax, &filt); |
| |
| // (aom_filter + 3 * (qs0 - ps0)) & mask |
| filt = _mm_and_si128(filt, mask); |
| |
| filter1 = _mm_adds_epi16(filt, t4); |
| pixel_clamp(&pmin, &pmax, &filter1); |
| |
| filter2 = _mm_adds_epi16(filt, t3); |
| pixel_clamp(&pmin, &pmax, &filter2); |
| |
| // Filter1 >> 3 |
| work_a = _mm_cmpgt_epi16(zero, filter1); // get the values that are <0 |
| filter1 = _mm_srli_epi16(filter1, 3); |
| work_a = _mm_and_si128(work_a, tffe0); // sign bits for the values < 0 |
| filter1 = _mm_and_si128(filter1, t1f); // clamp the range |
| filter1 = _mm_or_si128(filter1, work_a); // reinsert the sign bits |
| |
| // Filter2 >> 3 |
| work_a = _mm_cmpgt_epi16(zero, filter2); |
| filter2 = _mm_srli_epi16(filter2, 3); |
| work_a = _mm_and_si128(work_a, tffe0); |
| filter2 = _mm_and_si128(filter2, t1f); |
| filter2 = _mm_or_si128(filter2, work_a); |
| |
| // filt >> 1 |
| filt = _mm_adds_epi16(filter1, t1); |
| work_a = _mm_cmpgt_epi16(zero, filt); |
| filt = _mm_srli_epi16(filt, 1); |
| work_a = _mm_and_si128(work_a, tff80); |
| filt = _mm_and_si128(filt, t7f); |
| filt = _mm_or_si128(filt, work_a); |
| |
| filt = _mm_andnot_si128(hev, filt); |
| |
| q0 = _mm_subs_epi16(qs0, filter1); |
| pixel_clamp(&pmin, &pmax, &q0); |
| q0 = _mm_adds_epi16(q0, t80); |
| |
| q1 = _mm_subs_epi16(qs1, filt); |
| pixel_clamp(&pmin, &pmax, &q1); |
| q1 = _mm_adds_epi16(q1, t80); |
| |
| p0 = _mm_adds_epi16(ps0, filter2); |
| pixel_clamp(&pmin, &pmax, &p0); |
| p0 = _mm_adds_epi16(p0, t80); |
| |
| p1 = _mm_adds_epi16(ps1, filt); |
| pixel_clamp(&pmin, &pmax, &p1); |
| p1 = _mm_adds_epi16(p1, t80); |
| |
| _mm_storeu_si128((__m128i *)(s - 2 * p), p1); |
| _mm_storeu_si128((__m128i *)(s - 1 * p), p0); |
| _mm_storeu_si128((__m128i *)(s + 0 * p), q0); |
| _mm_storeu_si128((__m128i *)(s + 1 * p), q1); |
| } |
| |
| void aom_highbd_lpf_horizontal_4_dual_sse2( |
| uint16_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, int bd) { |
| aom_highbd_lpf_horizontal_4_sse2(s, p, _blimit0, _limit0, _thresh0, bd); |
| aom_highbd_lpf_horizontal_4_sse2(s + 8, p, _blimit1, _limit1, _thresh1, bd); |
| } |
| |
| void aom_highbd_lpf_vertical_4_sse2(uint16_t *s, int p, const uint8_t *blimit, |
| const uint8_t *limit, const uint8_t *thresh, |
| int bd) { |
| DECLARE_ALIGNED(16, uint16_t, t_dst[8 * 8]); |
| uint16_t *src[1]; |
| uint16_t *dst[1]; |
| |
| // Transpose 8x8 |
| src[0] = s - 4; |
| dst[0] = t_dst; |
| |
| highbd_transpose(src, p, dst, 8, 1); |
| |
| // Loop filtering |
| aom_highbd_lpf_horizontal_4_sse2(t_dst + 4 * 8, 8, blimit, limit, thresh, bd); |
| |
| src[0] = t_dst; |
| dst[0] = s - 4; |
| |
| // Transpose back |
| highbd_transpose(src, 8, dst, p, 1); |
| } |
| |
| void aom_highbd_lpf_vertical_4_dual_sse2( |
| uint16_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, int bd) { |
| DECLARE_ALIGNED(16, uint16_t, t_dst[16 * 8]); |
| uint16_t *src[2]; |
| uint16_t *dst[2]; |
| |
| // Transpose 8x16 |
| highbd_transpose8x16(s - 4, s - 4 + p * 8, p, t_dst, 16); |
| |
| // Loop filtering |
| aom_highbd_lpf_horizontal_4_dual_sse2(t_dst + 4 * 16, 16, blimit0, limit0, |
| thresh0, blimit1, limit1, thresh1, bd); |
| src[0] = t_dst; |
| src[1] = t_dst + 8; |
| dst[0] = s - 4; |
| dst[1] = s - 4 + p * 8; |
| |
| // Transpose back |
| highbd_transpose(src, 16, dst, p, 2); |
| } |
| |
| void aom_highbd_lpf_vertical_8_sse2(uint16_t *s, int p, const uint8_t *blimit, |
| const uint8_t *limit, const uint8_t *thresh, |
| int bd) { |
| DECLARE_ALIGNED(16, uint16_t, t_dst[8 * 8]); |
| uint16_t *src[1]; |
| uint16_t *dst[1]; |
| |
| // Transpose 8x8 |
| src[0] = s - 4; |
| dst[0] = t_dst; |
| |
| highbd_transpose(src, p, dst, 8, 1); |
| |
| // Loop filtering |
| aom_highbd_lpf_horizontal_8_sse2(t_dst + 4 * 8, 8, blimit, limit, thresh, bd); |
| |
| src[0] = t_dst; |
| dst[0] = s - 4; |
| |
| // Transpose back |
| highbd_transpose(src, 8, dst, p, 1); |
| } |
| |
| void aom_highbd_lpf_vertical_8_dual_sse2( |
| uint16_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, int bd) { |
| DECLARE_ALIGNED(16, uint16_t, t_dst[16 * 8]); |
| uint16_t *src[2]; |
| uint16_t *dst[2]; |
| |
| // Transpose 8x16 |
| highbd_transpose8x16(s - 4, s - 4 + p * 8, p, t_dst, 16); |
| |
| // Loop filtering |
| aom_highbd_lpf_horizontal_8_dual_sse2(t_dst + 4 * 16, 16, blimit0, limit0, |
| thresh0, blimit1, limit1, thresh1, bd); |
| src[0] = t_dst; |
| src[1] = t_dst + 8; |
| |
| dst[0] = s - 4; |
| dst[1] = s - 4 + p * 8; |
| |
| // Transpose back |
| highbd_transpose(src, 16, dst, p, 2); |
| } |
| |
| void aom_highbd_lpf_vertical_16_sse2(uint16_t *s, int p, const uint8_t *blimit, |
| const uint8_t *limit, |
| const uint8_t *thresh, int bd) { |
| DECLARE_ALIGNED(16, uint16_t, t_dst[8 * 16]); |
| uint16_t *src[2]; |
| uint16_t *dst[2]; |
| |
| src[0] = s - 8; |
| src[1] = s; |
| dst[0] = t_dst; |
| dst[1] = t_dst + 8 * 8; |
| |
| // Transpose 16x8 |
| highbd_transpose(src, p, dst, 8, 2); |
| |
| // Loop filtering |
| aom_highbd_lpf_horizontal_edge_8_sse2(t_dst + 8 * 8, 8, blimit, limit, thresh, |
| bd); |
| src[0] = t_dst; |
| src[1] = t_dst + 8 * 8; |
| dst[0] = s - 8; |
| dst[1] = s; |
| |
| // Transpose back |
| highbd_transpose(src, 8, dst, p, 2); |
| } |
| |
| void aom_highbd_lpf_vertical_16_dual_sse2(uint16_t *s, int p, |
| const uint8_t *blimit, |
| const uint8_t *limit, |
| const uint8_t *thresh, int bd) { |
| DECLARE_ALIGNED(16, uint16_t, t_dst[256]); |
| |
| // Transpose 16x16 |
| highbd_transpose8x16(s - 8, s - 8 + 8 * p, p, t_dst, 16); |
| highbd_transpose8x16(s, s + 8 * p, p, t_dst + 8 * 16, 16); |
| |
| // Loop filtering |
| aom_highbd_lpf_horizontal_edge_16_sse2(t_dst + 8 * 16, 16, blimit, limit, |
| thresh, bd); |
| |
| // Transpose back |
| highbd_transpose8x16(t_dst, t_dst + 8 * 16, 16, s - 8, p); |
| highbd_transpose8x16(t_dst + 8, t_dst + 8 + 8 * 16, 16, s - 8 + 8 * p, p); |
| } |