| /* |
| * Copyright (c) 2018, 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 <immintrin.h> |
| |
| #include "config/av1_rtcd.h" |
| |
| #include "aom/aom_integer.h" |
| #include "aom_dsp/blend.h" |
| #include "aom_dsp/x86/synonyms.h" |
| #include "aom_dsp/x86/synonyms_avx2.h" |
| #include "av1/common/blockd.h" |
| |
| static INLINE __m256i calc_mask_avx2(const __m256i mask_base, const __m256i s0, |
| const __m256i s1) { |
| const __m256i diff = _mm256_abs_epi16(_mm256_sub_epi16(s0, s1)); |
| return _mm256_abs_epi16( |
| _mm256_add_epi16(mask_base, _mm256_srli_epi16(diff, 4))); |
| // clamp(diff, 0, 64) can be skiped for diff is always in the range ( 38, 54) |
| } |
| void av1_build_compound_diffwtd_mask_avx2(uint8_t *mask, |
| DIFFWTD_MASK_TYPE mask_type, |
| const uint8_t *src0, int src0_stride, |
| const uint8_t *src1, int src1_stride, |
| int h, int w) { |
| const int mb = (mask_type == DIFFWTD_38_INV) ? AOM_BLEND_A64_MAX_ALPHA : 0; |
| const __m256i y_mask_base = _mm256_set1_epi16(38 - mb); |
| int i = 0; |
| if (4 == w) { |
| do { |
| const __m128i s0A = xx_loadl_32(src0); |
| const __m128i s0B = xx_loadl_32(src0 + src0_stride); |
| const __m128i s0C = xx_loadl_32(src0 + src0_stride * 2); |
| const __m128i s0D = xx_loadl_32(src0 + src0_stride * 3); |
| const __m128i s0AB = _mm_unpacklo_epi32(s0A, s0B); |
| const __m128i s0CD = _mm_unpacklo_epi32(s0C, s0D); |
| const __m128i s0ABCD = _mm_unpacklo_epi64(s0AB, s0CD); |
| const __m256i s0ABCD_w = _mm256_cvtepu8_epi16(s0ABCD); |
| |
| const __m128i s1A = xx_loadl_32(src1); |
| const __m128i s1B = xx_loadl_32(src1 + src1_stride); |
| const __m128i s1C = xx_loadl_32(src1 + src1_stride * 2); |
| const __m128i s1D = xx_loadl_32(src1 + src1_stride * 3); |
| const __m128i s1AB = _mm_unpacklo_epi32(s1A, s1B); |
| const __m128i s1CD = _mm_unpacklo_epi32(s1C, s1D); |
| const __m128i s1ABCD = _mm_unpacklo_epi64(s1AB, s1CD); |
| const __m256i s1ABCD_w = _mm256_cvtepu8_epi16(s1ABCD); |
| const __m256i m16 = calc_mask_avx2(y_mask_base, s0ABCD_w, s1ABCD_w); |
| const __m256i m8 = _mm256_packus_epi16(m16, _mm256_setzero_si256()); |
| const __m128i x_m8 = |
| _mm256_castsi256_si128(_mm256_permute4x64_epi64(m8, 0xd8)); |
| xx_storeu_128(mask, x_m8); |
| src0 += (src0_stride << 2); |
| src1 += (src1_stride << 2); |
| mask += 16; |
| i += 4; |
| } while (i < h); |
| } else if (8 == w) { |
| do { |
| const __m128i s0A = xx_loadl_64(src0); |
| const __m128i s0B = xx_loadl_64(src0 + src0_stride); |
| const __m128i s0C = xx_loadl_64(src0 + src0_stride * 2); |
| const __m128i s0D = xx_loadl_64(src0 + src0_stride * 3); |
| const __m256i s0AC_w = _mm256_cvtepu8_epi16(_mm_unpacklo_epi64(s0A, s0C)); |
| const __m256i s0BD_w = _mm256_cvtepu8_epi16(_mm_unpacklo_epi64(s0B, s0D)); |
| const __m128i s1A = xx_loadl_64(src1); |
| const __m128i s1B = xx_loadl_64(src1 + src1_stride); |
| const __m128i s1C = xx_loadl_64(src1 + src1_stride * 2); |
| const __m128i s1D = xx_loadl_64(src1 + src1_stride * 3); |
| const __m256i s1AB_w = _mm256_cvtepu8_epi16(_mm_unpacklo_epi64(s1A, s1C)); |
| const __m256i s1CD_w = _mm256_cvtepu8_epi16(_mm_unpacklo_epi64(s1B, s1D)); |
| const __m256i m16AC = calc_mask_avx2(y_mask_base, s0AC_w, s1AB_w); |
| const __m256i m16BD = calc_mask_avx2(y_mask_base, s0BD_w, s1CD_w); |
| const __m256i m8 = _mm256_packus_epi16(m16AC, m16BD); |
| yy_storeu_256(mask, m8); |
| src0 += src0_stride << 2; |
| src1 += src1_stride << 2; |
| mask += 32; |
| i += 4; |
| } while (i < h); |
| } else if (16 == w) { |
| do { |
| const __m128i s0A = xx_load_128(src0); |
| const __m128i s0B = xx_load_128(src0 + src0_stride); |
| const __m128i s1A = xx_load_128(src1); |
| const __m128i s1B = xx_load_128(src1 + src1_stride); |
| const __m256i s0AL = _mm256_cvtepu8_epi16(s0A); |
| const __m256i s0BL = _mm256_cvtepu8_epi16(s0B); |
| const __m256i s1AL = _mm256_cvtepu8_epi16(s1A); |
| const __m256i s1BL = _mm256_cvtepu8_epi16(s1B); |
| |
| const __m256i m16AL = calc_mask_avx2(y_mask_base, s0AL, s1AL); |
| const __m256i m16BL = calc_mask_avx2(y_mask_base, s0BL, s1BL); |
| |
| const __m256i m8 = |
| _mm256_permute4x64_epi64(_mm256_packus_epi16(m16AL, m16BL), 0xd8); |
| yy_storeu_256(mask, m8); |
| src0 += src0_stride << 1; |
| src1 += src1_stride << 1; |
| mask += 32; |
| i += 2; |
| } while (i < h); |
| } else { |
| do { |
| int j = 0; |
| do { |
| const __m256i s0 = yy_loadu_256(src0 + j); |
| const __m256i s1 = yy_loadu_256(src1 + j); |
| const __m256i s0L = _mm256_cvtepu8_epi16(_mm256_castsi256_si128(s0)); |
| const __m256i s1L = _mm256_cvtepu8_epi16(_mm256_castsi256_si128(s1)); |
| const __m256i s0H = |
| _mm256_cvtepu8_epi16(_mm256_extracti128_si256(s0, 1)); |
| const __m256i s1H = |
| _mm256_cvtepu8_epi16(_mm256_extracti128_si256(s1, 1)); |
| const __m256i m16L = calc_mask_avx2(y_mask_base, s0L, s1L); |
| const __m256i m16H = calc_mask_avx2(y_mask_base, s0H, s1H); |
| const __m256i m8 = |
| _mm256_permute4x64_epi64(_mm256_packus_epi16(m16L, m16H), 0xd8); |
| yy_storeu_256(mask + j, m8); |
| j += 32; |
| } while (j < w); |
| src0 += src0_stride; |
| src1 += src1_stride; |
| mask += w; |
| i += 1; |
| } while (i < h); |
| } |
| } |
| |
| static INLINE __m256i calc_mask_d16_avx2(const __m256i *data_src0, |
| const __m256i *data_src1, |
| const __m256i *round_const, |
| const __m256i *mask_base_16, |
| const __m256i *clip_diff, int round) { |
| const __m256i diffa = _mm256_subs_epu16(*data_src0, *data_src1); |
| const __m256i diffb = _mm256_subs_epu16(*data_src1, *data_src0); |
| const __m256i diff = _mm256_max_epu16(diffa, diffb); |
| const __m256i diff_round = |
| _mm256_srli_epi16(_mm256_adds_epu16(diff, *round_const), round); |
| const __m256i diff_factor = _mm256_srli_epi16(diff_round, DIFF_FACTOR_LOG2); |
| const __m256i diff_mask = _mm256_adds_epi16(diff_factor, *mask_base_16); |
| const __m256i diff_clamp = _mm256_min_epi16(diff_mask, *clip_diff); |
| return diff_clamp; |
| } |
| |
| static INLINE __m256i calc_mask_d16_inv_avx2(const __m256i *data_src0, |
| const __m256i *data_src1, |
| const __m256i *round_const, |
| const __m256i *mask_base_16, |
| const __m256i *clip_diff, |
| int round) { |
| const __m256i diffa = _mm256_subs_epu16(*data_src0, *data_src1); |
| const __m256i diffb = _mm256_subs_epu16(*data_src1, *data_src0); |
| const __m256i diff = _mm256_max_epu16(diffa, diffb); |
| const __m256i diff_round = |
| _mm256_srli_epi16(_mm256_adds_epu16(diff, *round_const), round); |
| const __m256i diff_factor = _mm256_srli_epi16(diff_round, DIFF_FACTOR_LOG2); |
| const __m256i diff_mask = _mm256_adds_epi16(diff_factor, *mask_base_16); |
| const __m256i diff_clamp = _mm256_min_epi16(diff_mask, *clip_diff); |
| const __m256i diff_const_16 = _mm256_sub_epi16(*clip_diff, diff_clamp); |
| return diff_const_16; |
| } |
| |
| static INLINE void build_compound_diffwtd_mask_d16_avx2( |
| uint8_t *mask, const CONV_BUF_TYPE *src0, int src0_stride, |
| const CONV_BUF_TYPE *src1, int src1_stride, int h, int w, int shift) { |
| const int mask_base = 38; |
| const __m256i _r = _mm256_set1_epi16((1 << shift) >> 1); |
| const __m256i y38 = _mm256_set1_epi16(mask_base); |
| const __m256i y64 = _mm256_set1_epi16(AOM_BLEND_A64_MAX_ALPHA); |
| int i = 0; |
| if (w == 4) { |
| do { |
| const __m128i s0A = xx_loadl_64(src0); |
| const __m128i s0B = xx_loadl_64(src0 + src0_stride); |
| const __m128i s0C = xx_loadl_64(src0 + src0_stride * 2); |
| const __m128i s0D = xx_loadl_64(src0 + src0_stride * 3); |
| const __m128i s1A = xx_loadl_64(src1); |
| const __m128i s1B = xx_loadl_64(src1 + src1_stride); |
| const __m128i s1C = xx_loadl_64(src1 + src1_stride * 2); |
| const __m128i s1D = xx_loadl_64(src1 + src1_stride * 3); |
| const __m256i s0 = yy_set_m128i(_mm_unpacklo_epi64(s0C, s0D), |
| _mm_unpacklo_epi64(s0A, s0B)); |
| const __m256i s1 = yy_set_m128i(_mm_unpacklo_epi64(s1C, s1D), |
| _mm_unpacklo_epi64(s1A, s1B)); |
| const __m256i m16 = calc_mask_d16_avx2(&s0, &s1, &_r, &y38, &y64, shift); |
| const __m256i m8 = _mm256_packus_epi16(m16, _mm256_setzero_si256()); |
| xx_storeu_128(mask, |
| _mm256_castsi256_si128(_mm256_permute4x64_epi64(m8, 0xd8))); |
| src0 += src0_stride << 2; |
| src1 += src1_stride << 2; |
| mask += 16; |
| i += 4; |
| } while (i < h); |
| } else if (w == 8) { |
| do { |
| const __m256i s0AB = yy_loadu2_128(src0 + src0_stride, src0); |
| const __m256i s0CD = |
| yy_loadu2_128(src0 + src0_stride * 3, src0 + src0_stride * 2); |
| const __m256i s1AB = yy_loadu2_128(src1 + src1_stride, src1); |
| const __m256i s1CD = |
| yy_loadu2_128(src1 + src1_stride * 3, src1 + src1_stride * 2); |
| const __m256i m16AB = |
| calc_mask_d16_avx2(&s0AB, &s1AB, &_r, &y38, &y64, shift); |
| const __m256i m16CD = |
| calc_mask_d16_avx2(&s0CD, &s1CD, &_r, &y38, &y64, shift); |
| const __m256i m8 = _mm256_packus_epi16(m16AB, m16CD); |
| yy_storeu_256(mask, _mm256_permute4x64_epi64(m8, 0xd8)); |
| src0 += src0_stride << 2; |
| src1 += src1_stride << 2; |
| mask += 32; |
| i += 4; |
| } while (i < h); |
| } else if (w == 16) { |
| do { |
| const __m256i s0A = yy_loadu_256(src0); |
| const __m256i s0B = yy_loadu_256(src0 + src0_stride); |
| const __m256i s1A = yy_loadu_256(src1); |
| const __m256i s1B = yy_loadu_256(src1 + src1_stride); |
| const __m256i m16A = |
| calc_mask_d16_avx2(&s0A, &s1A, &_r, &y38, &y64, shift); |
| const __m256i m16B = |
| calc_mask_d16_avx2(&s0B, &s1B, &_r, &y38, &y64, shift); |
| const __m256i m8 = _mm256_packus_epi16(m16A, m16B); |
| yy_storeu_256(mask, _mm256_permute4x64_epi64(m8, 0xd8)); |
| src0 += src0_stride << 1; |
| src1 += src1_stride << 1; |
| mask += 32; |
| i += 2; |
| } while (i < h); |
| } else if (w == 32) { |
| do { |
| const __m256i s0A = yy_loadu_256(src0); |
| const __m256i s0B = yy_loadu_256(src0 + 16); |
| const __m256i s1A = yy_loadu_256(src1); |
| const __m256i s1B = yy_loadu_256(src1 + 16); |
| const __m256i m16A = |
| calc_mask_d16_avx2(&s0A, &s1A, &_r, &y38, &y64, shift); |
| const __m256i m16B = |
| calc_mask_d16_avx2(&s0B, &s1B, &_r, &y38, &y64, shift); |
| const __m256i m8 = _mm256_packus_epi16(m16A, m16B); |
| yy_storeu_256(mask, _mm256_permute4x64_epi64(m8, 0xd8)); |
| src0 += src0_stride; |
| src1 += src1_stride; |
| mask += 32; |
| i += 1; |
| } while (i < h); |
| } else if (w == 64) { |
| do { |
| const __m256i s0A = yy_loadu_256(src0); |
| const __m256i s0B = yy_loadu_256(src0 + 16); |
| const __m256i s0C = yy_loadu_256(src0 + 32); |
| const __m256i s0D = yy_loadu_256(src0 + 48); |
| const __m256i s1A = yy_loadu_256(src1); |
| const __m256i s1B = yy_loadu_256(src1 + 16); |
| const __m256i s1C = yy_loadu_256(src1 + 32); |
| const __m256i s1D = yy_loadu_256(src1 + 48); |
| const __m256i m16A = |
| calc_mask_d16_avx2(&s0A, &s1A, &_r, &y38, &y64, shift); |
| const __m256i m16B = |
| calc_mask_d16_avx2(&s0B, &s1B, &_r, &y38, &y64, shift); |
| const __m256i m16C = |
| calc_mask_d16_avx2(&s0C, &s1C, &_r, &y38, &y64, shift); |
| const __m256i m16D = |
| calc_mask_d16_avx2(&s0D, &s1D, &_r, &y38, &y64, shift); |
| const __m256i m8AB = _mm256_packus_epi16(m16A, m16B); |
| const __m256i m8CD = _mm256_packus_epi16(m16C, m16D); |
| yy_storeu_256(mask, _mm256_permute4x64_epi64(m8AB, 0xd8)); |
| yy_storeu_256(mask + 32, _mm256_permute4x64_epi64(m8CD, 0xd8)); |
| src0 += src0_stride; |
| src1 += src1_stride; |
| mask += 64; |
| i += 1; |
| } while (i < h); |
| } else { |
| do { |
| const __m256i s0A = yy_loadu_256(src0); |
| const __m256i s0B = yy_loadu_256(src0 + 16); |
| const __m256i s0C = yy_loadu_256(src0 + 32); |
| const __m256i s0D = yy_loadu_256(src0 + 48); |
| const __m256i s0E = yy_loadu_256(src0 + 64); |
| const __m256i s0F = yy_loadu_256(src0 + 80); |
| const __m256i s0G = yy_loadu_256(src0 + 96); |
| const __m256i s0H = yy_loadu_256(src0 + 112); |
| const __m256i s1A = yy_loadu_256(src1); |
| const __m256i s1B = yy_loadu_256(src1 + 16); |
| const __m256i s1C = yy_loadu_256(src1 + 32); |
| const __m256i s1D = yy_loadu_256(src1 + 48); |
| const __m256i s1E = yy_loadu_256(src1 + 64); |
| const __m256i s1F = yy_loadu_256(src1 + 80); |
| const __m256i s1G = yy_loadu_256(src1 + 96); |
| const __m256i s1H = yy_loadu_256(src1 + 112); |
| const __m256i m16A = |
| calc_mask_d16_avx2(&s0A, &s1A, &_r, &y38, &y64, shift); |
| const __m256i m16B = |
| calc_mask_d16_avx2(&s0B, &s1B, &_r, &y38, &y64, shift); |
| const __m256i m16C = |
| calc_mask_d16_avx2(&s0C, &s1C, &_r, &y38, &y64, shift); |
| const __m256i m16D = |
| calc_mask_d16_avx2(&s0D, &s1D, &_r, &y38, &y64, shift); |
| const __m256i m16E = |
| calc_mask_d16_avx2(&s0E, &s1E, &_r, &y38, &y64, shift); |
| const __m256i m16F = |
| calc_mask_d16_avx2(&s0F, &s1F, &_r, &y38, &y64, shift); |
| const __m256i m16G = |
| calc_mask_d16_avx2(&s0G, &s1G, &_r, &y38, &y64, shift); |
| const __m256i m16H = |
| calc_mask_d16_avx2(&s0H, &s1H, &_r, &y38, &y64, shift); |
| const __m256i m8AB = _mm256_packus_epi16(m16A, m16B); |
| const __m256i m8CD = _mm256_packus_epi16(m16C, m16D); |
| const __m256i m8EF = _mm256_packus_epi16(m16E, m16F); |
| const __m256i m8GH = _mm256_packus_epi16(m16G, m16H); |
| yy_storeu_256(mask, _mm256_permute4x64_epi64(m8AB, 0xd8)); |
| yy_storeu_256(mask + 32, _mm256_permute4x64_epi64(m8CD, 0xd8)); |
| yy_storeu_256(mask + 64, _mm256_permute4x64_epi64(m8EF, 0xd8)); |
| yy_storeu_256(mask + 96, _mm256_permute4x64_epi64(m8GH, 0xd8)); |
| src0 += src0_stride; |
| src1 += src1_stride; |
| mask += 128; |
| i += 1; |
| } while (i < h); |
| } |
| } |
| |
| static INLINE void build_compound_diffwtd_mask_d16_inv_avx2( |
| uint8_t *mask, const CONV_BUF_TYPE *src0, int src0_stride, |
| const CONV_BUF_TYPE *src1, int src1_stride, int h, int w, int shift) { |
| const int mask_base = 38; |
| const __m256i _r = _mm256_set1_epi16((1 << shift) >> 1); |
| const __m256i y38 = _mm256_set1_epi16(mask_base); |
| const __m256i y64 = _mm256_set1_epi16(AOM_BLEND_A64_MAX_ALPHA); |
| int i = 0; |
| if (w == 4) { |
| do { |
| const __m128i s0A = xx_loadl_64(src0); |
| const __m128i s0B = xx_loadl_64(src0 + src0_stride); |
| const __m128i s0C = xx_loadl_64(src0 + src0_stride * 2); |
| const __m128i s0D = xx_loadl_64(src0 + src0_stride * 3); |
| const __m128i s1A = xx_loadl_64(src1); |
| const __m128i s1B = xx_loadl_64(src1 + src1_stride); |
| const __m128i s1C = xx_loadl_64(src1 + src1_stride * 2); |
| const __m128i s1D = xx_loadl_64(src1 + src1_stride * 3); |
| const __m256i s0 = yy_set_m128i(_mm_unpacklo_epi64(s0C, s0D), |
| _mm_unpacklo_epi64(s0A, s0B)); |
| const __m256i s1 = yy_set_m128i(_mm_unpacklo_epi64(s1C, s1D), |
| _mm_unpacklo_epi64(s1A, s1B)); |
| const __m256i m16 = |
| calc_mask_d16_inv_avx2(&s0, &s1, &_r, &y38, &y64, shift); |
| const __m256i m8 = _mm256_packus_epi16(m16, _mm256_setzero_si256()); |
| xx_storeu_128(mask, |
| _mm256_castsi256_si128(_mm256_permute4x64_epi64(m8, 0xd8))); |
| src0 += src0_stride << 2; |
| src1 += src1_stride << 2; |
| mask += 16; |
| i += 4; |
| } while (i < h); |
| } else if (w == 8) { |
| do { |
| const __m256i s0AB = yy_loadu2_128(src0 + src0_stride, src0); |
| const __m256i s0CD = |
| yy_loadu2_128(src0 + src0_stride * 3, src0 + src0_stride * 2); |
| const __m256i s1AB = yy_loadu2_128(src1 + src1_stride, src1); |
| const __m256i s1CD = |
| yy_loadu2_128(src1 + src1_stride * 3, src1 + src1_stride * 2); |
| const __m256i m16AB = |
| calc_mask_d16_inv_avx2(&s0AB, &s1AB, &_r, &y38, &y64, shift); |
| const __m256i m16CD = |
| calc_mask_d16_inv_avx2(&s0CD, &s1CD, &_r, &y38, &y64, shift); |
| const __m256i m8 = _mm256_packus_epi16(m16AB, m16CD); |
| yy_storeu_256(mask, _mm256_permute4x64_epi64(m8, 0xd8)); |
| src0 += src0_stride << 2; |
| src1 += src1_stride << 2; |
| mask += 32; |
| i += 4; |
| } while (i < h); |
| } else if (w == 16) { |
| do { |
| const __m256i s0A = yy_loadu_256(src0); |
| const __m256i s0B = yy_loadu_256(src0 + src0_stride); |
| const __m256i s1A = yy_loadu_256(src1); |
| const __m256i s1B = yy_loadu_256(src1 + src1_stride); |
| const __m256i m16A = |
| calc_mask_d16_inv_avx2(&s0A, &s1A, &_r, &y38, &y64, shift); |
| const __m256i m16B = |
| calc_mask_d16_inv_avx2(&s0B, &s1B, &_r, &y38, &y64, shift); |
| const __m256i m8 = _mm256_packus_epi16(m16A, m16B); |
| yy_storeu_256(mask, _mm256_permute4x64_epi64(m8, 0xd8)); |
| src0 += src0_stride << 1; |
| src1 += src1_stride << 1; |
| mask += 32; |
| i += 2; |
| } while (i < h); |
| } else if (w == 32) { |
| do { |
| const __m256i s0A = yy_loadu_256(src0); |
| const __m256i s0B = yy_loadu_256(src0 + 16); |
| const __m256i s1A = yy_loadu_256(src1); |
| const __m256i s1B = yy_loadu_256(src1 + 16); |
| const __m256i m16A = |
| calc_mask_d16_inv_avx2(&s0A, &s1A, &_r, &y38, &y64, shift); |
| const __m256i m16B = |
| calc_mask_d16_inv_avx2(&s0B, &s1B, &_r, &y38, &y64, shift); |
| const __m256i m8 = _mm256_packus_epi16(m16A, m16B); |
| yy_storeu_256(mask, _mm256_permute4x64_epi64(m8, 0xd8)); |
| src0 += src0_stride; |
| src1 += src1_stride; |
| mask += 32; |
| i += 1; |
| } while (i < h); |
| } else if (w == 64) { |
| do { |
| const __m256i s0A = yy_loadu_256(src0); |
| const __m256i s0B = yy_loadu_256(src0 + 16); |
| const __m256i s0C = yy_loadu_256(src0 + 32); |
| const __m256i s0D = yy_loadu_256(src0 + 48); |
| const __m256i s1A = yy_loadu_256(src1); |
| const __m256i s1B = yy_loadu_256(src1 + 16); |
| const __m256i s1C = yy_loadu_256(src1 + 32); |
| const __m256i s1D = yy_loadu_256(src1 + 48); |
| const __m256i m16A = |
| calc_mask_d16_inv_avx2(&s0A, &s1A, &_r, &y38, &y64, shift); |
| const __m256i m16B = |
| calc_mask_d16_inv_avx2(&s0B, &s1B, &_r, &y38, &y64, shift); |
| const __m256i m16C = |
| calc_mask_d16_inv_avx2(&s0C, &s1C, &_r, &y38, &y64, shift); |
| const __m256i m16D = |
| calc_mask_d16_inv_avx2(&s0D, &s1D, &_r, &y38, &y64, shift); |
| const __m256i m8AB = _mm256_packus_epi16(m16A, m16B); |
| const __m256i m8CD = _mm256_packus_epi16(m16C, m16D); |
| yy_storeu_256(mask, _mm256_permute4x64_epi64(m8AB, 0xd8)); |
| yy_storeu_256(mask + 32, _mm256_permute4x64_epi64(m8CD, 0xd8)); |
| src0 += src0_stride; |
| src1 += src1_stride; |
| mask += 64; |
| i += 1; |
| } while (i < h); |
| } else { |
| do { |
| const __m256i s0A = yy_loadu_256(src0); |
| const __m256i s0B = yy_loadu_256(src0 + 16); |
| const __m256i s0C = yy_loadu_256(src0 + 32); |
| const __m256i s0D = yy_loadu_256(src0 + 48); |
| const __m256i s0E = yy_loadu_256(src0 + 64); |
| const __m256i s0F = yy_loadu_256(src0 + 80); |
| const __m256i s0G = yy_loadu_256(src0 + 96); |
| const __m256i s0H = yy_loadu_256(src0 + 112); |
| const __m256i s1A = yy_loadu_256(src1); |
| const __m256i s1B = yy_loadu_256(src1 + 16); |
| const __m256i s1C = yy_loadu_256(src1 + 32); |
| const __m256i s1D = yy_loadu_256(src1 + 48); |
| const __m256i s1E = yy_loadu_256(src1 + 64); |
| const __m256i s1F = yy_loadu_256(src1 + 80); |
| const __m256i s1G = yy_loadu_256(src1 + 96); |
| const __m256i s1H = yy_loadu_256(src1 + 112); |
| const __m256i m16A = |
| calc_mask_d16_inv_avx2(&s0A, &s1A, &_r, &y38, &y64, shift); |
| const __m256i m16B = |
| calc_mask_d16_inv_avx2(&s0B, &s1B, &_r, &y38, &y64, shift); |
| const __m256i m16C = |
| calc_mask_d16_inv_avx2(&s0C, &s1C, &_r, &y38, &y64, shift); |
| const __m256i m16D = |
| calc_mask_d16_inv_avx2(&s0D, &s1D, &_r, &y38, &y64, shift); |
| const __m256i m16E = |
| calc_mask_d16_inv_avx2(&s0E, &s1E, &_r, &y38, &y64, shift); |
| const __m256i m16F = |
| calc_mask_d16_inv_avx2(&s0F, &s1F, &_r, &y38, &y64, shift); |
| const __m256i m16G = |
| calc_mask_d16_inv_avx2(&s0G, &s1G, &_r, &y38, &y64, shift); |
| const __m256i m16H = |
| calc_mask_d16_inv_avx2(&s0H, &s1H, &_r, &y38, &y64, shift); |
| const __m256i m8AB = _mm256_packus_epi16(m16A, m16B); |
| const __m256i m8CD = _mm256_packus_epi16(m16C, m16D); |
| const __m256i m8EF = _mm256_packus_epi16(m16E, m16F); |
| const __m256i m8GH = _mm256_packus_epi16(m16G, m16H); |
| yy_storeu_256(mask, _mm256_permute4x64_epi64(m8AB, 0xd8)); |
| yy_storeu_256(mask + 32, _mm256_permute4x64_epi64(m8CD, 0xd8)); |
| yy_storeu_256(mask + 64, _mm256_permute4x64_epi64(m8EF, 0xd8)); |
| yy_storeu_256(mask + 96, _mm256_permute4x64_epi64(m8GH, 0xd8)); |
| src0 += src0_stride; |
| src1 += src1_stride; |
| mask += 128; |
| i += 1; |
| } while (i < h); |
| } |
| } |
| |
| void av1_build_compound_diffwtd_mask_d16_avx2( |
| uint8_t *mask, DIFFWTD_MASK_TYPE mask_type, const CONV_BUF_TYPE *src0, |
| int src0_stride, const CONV_BUF_TYPE *src1, int src1_stride, int h, int w, |
| ConvolveParams *conv_params, int bd) { |
| const int shift = |
| 2 * FILTER_BITS - conv_params->round_0 - conv_params->round_1 + (bd - 8); |
| // When rounding constant is added, there is a possibility of overflow. |
| // However that much precision is not required. Code should very well work for |
| // other values of DIFF_FACTOR_LOG2 and AOM_BLEND_A64_MAX_ALPHA as well. But |
| // there is a possibility of corner case bugs. |
| assert(DIFF_FACTOR_LOG2 == 4); |
| assert(AOM_BLEND_A64_MAX_ALPHA == 64); |
| |
| if (mask_type == DIFFWTD_38) { |
| build_compound_diffwtd_mask_d16_avx2(mask, src0, src0_stride, src1, |
| src1_stride, h, w, shift); |
| } else { |
| build_compound_diffwtd_mask_d16_inv_avx2(mask, src0, src0_stride, src1, |
| src1_stride, h, w, shift); |
| } |
| } |
| |
| void av1_build_compound_diffwtd_mask_highbd_avx2( |
| uint8_t *mask, DIFFWTD_MASK_TYPE mask_type, const uint8_t *src0, |
| int src0_stride, const uint8_t *src1, int src1_stride, int h, int w, |
| int bd) { |
| if (w < 16) { |
| av1_build_compound_diffwtd_mask_highbd_ssse3( |
| mask, mask_type, src0, src0_stride, src1, src1_stride, h, w, bd); |
| } else { |
| assert(mask_type == DIFFWTD_38 || mask_type == DIFFWTD_38_INV); |
| assert(bd >= 8); |
| assert((w % 16) == 0); |
| const __m256i y0 = _mm256_setzero_si256(); |
| const __m256i yAOM_BLEND_A64_MAX_ALPHA = |
| _mm256_set1_epi16(AOM_BLEND_A64_MAX_ALPHA); |
| const int mask_base = 38; |
| const __m256i ymask_base = _mm256_set1_epi16(mask_base); |
| const uint16_t *ssrc0 = CONVERT_TO_SHORTPTR(src0); |
| const uint16_t *ssrc1 = CONVERT_TO_SHORTPTR(src1); |
| if (bd == 8) { |
| if (mask_type == DIFFWTD_38_INV) { |
| for (int i = 0; i < h; ++i) { |
| for (int j = 0; j < w; j += 16) { |
| __m256i s0 = _mm256_loadu_si256((const __m256i *)&ssrc0[j]); |
| __m256i s1 = _mm256_loadu_si256((const __m256i *)&ssrc1[j]); |
| __m256i diff = _mm256_srai_epi16( |
| _mm256_abs_epi16(_mm256_sub_epi16(s0, s1)), DIFF_FACTOR_LOG2); |
| __m256i m = _mm256_min_epi16( |
| _mm256_max_epi16(y0, _mm256_add_epi16(diff, ymask_base)), |
| yAOM_BLEND_A64_MAX_ALPHA); |
| m = _mm256_sub_epi16(yAOM_BLEND_A64_MAX_ALPHA, m); |
| m = _mm256_packus_epi16(m, m); |
| m = _mm256_permute4x64_epi64(m, _MM_SHUFFLE(0, 0, 2, 0)); |
| __m128i m0 = _mm256_castsi256_si128(m); |
| _mm_storeu_si128((__m128i *)&mask[j], m0); |
| } |
| ssrc0 += src0_stride; |
| ssrc1 += src1_stride; |
| mask += w; |
| } |
| } else { |
| for (int i = 0; i < h; ++i) { |
| for (int j = 0; j < w; j += 16) { |
| __m256i s0 = _mm256_loadu_si256((const __m256i *)&ssrc0[j]); |
| __m256i s1 = _mm256_loadu_si256((const __m256i *)&ssrc1[j]); |
| __m256i diff = _mm256_srai_epi16( |
| _mm256_abs_epi16(_mm256_sub_epi16(s0, s1)), DIFF_FACTOR_LOG2); |
| __m256i m = _mm256_min_epi16( |
| _mm256_max_epi16(y0, _mm256_add_epi16(diff, ymask_base)), |
| yAOM_BLEND_A64_MAX_ALPHA); |
| m = _mm256_packus_epi16(m, m); |
| m = _mm256_permute4x64_epi64(m, _MM_SHUFFLE(0, 0, 2, 0)); |
| __m128i m0 = _mm256_castsi256_si128(m); |
| _mm_storeu_si128((__m128i *)&mask[j], m0); |
| } |
| ssrc0 += src0_stride; |
| ssrc1 += src1_stride; |
| mask += w; |
| } |
| } |
| } else { |
| const __m128i xshift = xx_set1_64_from_32i(bd - 8 + DIFF_FACTOR_LOG2); |
| if (mask_type == DIFFWTD_38_INV) { |
| for (int i = 0; i < h; ++i) { |
| for (int j = 0; j < w; j += 16) { |
| __m256i s0 = _mm256_loadu_si256((const __m256i *)&ssrc0[j]); |
| __m256i s1 = _mm256_loadu_si256((const __m256i *)&ssrc1[j]); |
| __m256i diff = _mm256_sra_epi16( |
| _mm256_abs_epi16(_mm256_sub_epi16(s0, s1)), xshift); |
| __m256i m = _mm256_min_epi16( |
| _mm256_max_epi16(y0, _mm256_add_epi16(diff, ymask_base)), |
| yAOM_BLEND_A64_MAX_ALPHA); |
| m = _mm256_sub_epi16(yAOM_BLEND_A64_MAX_ALPHA, m); |
| m = _mm256_packus_epi16(m, m); |
| m = _mm256_permute4x64_epi64(m, _MM_SHUFFLE(0, 0, 2, 0)); |
| __m128i m0 = _mm256_castsi256_si128(m); |
| _mm_storeu_si128((__m128i *)&mask[j], m0); |
| } |
| ssrc0 += src0_stride; |
| ssrc1 += src1_stride; |
| mask += w; |
| } |
| } else { |
| for (int i = 0; i < h; ++i) { |
| for (int j = 0; j < w; j += 16) { |
| __m256i s0 = _mm256_loadu_si256((const __m256i *)&ssrc0[j]); |
| __m256i s1 = _mm256_loadu_si256((const __m256i *)&ssrc1[j]); |
| __m256i diff = _mm256_sra_epi16( |
| _mm256_abs_epi16(_mm256_sub_epi16(s0, s1)), xshift); |
| __m256i m = _mm256_min_epi16( |
| _mm256_max_epi16(y0, _mm256_add_epi16(diff, ymask_base)), |
| yAOM_BLEND_A64_MAX_ALPHA); |
| m = _mm256_packus_epi16(m, m); |
| m = _mm256_permute4x64_epi64(m, _MM_SHUFFLE(0, 0, 2, 0)); |
| __m128i m0 = _mm256_castsi256_si128(m); |
| _mm_storeu_si128((__m128i *)&mask[j], m0); |
| } |
| ssrc0 += src0_stride; |
| ssrc1 += src1_stride; |
| mask += w; |
| } |
| } |
| } |
| } |
| } |
