| /* |
| * 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 "config/aom_config.h" |
| |
| #include "config/av1_rtcd.h" |
| |
| #include "av1/common/av1_inv_txfm1d_cfg.h" |
| #include "av1/common/x86/av1_txfm_sse2.h" |
| #include "av1/common/x86/av1_inv_txfm_avx2.h" |
| #include "av1/common/x86/av1_inv_txfm_ssse3.h" |
| |
| // TODO(venkatsanampudi@ittiam.com): move this to header file |
| |
| // Sqrt2, Sqrt2^2, Sqrt2^3, Sqrt2^4, Sqrt2^5 |
| static int32_t NewSqrt2list[TX_SIZES] = { 5793, 2 * 4096, 2 * 5793, 4 * 4096, |
| 4 * 5793 }; |
| |
| static inline void idct16_stage5_avx2(__m256i *x1, const int32_t *cospi, |
| const __m256i _r, int8_t cos_bit) { |
| const __m256i cospi_m32_p32 = pair_set_w16_epi16(-cospi[32], cospi[32]); |
| const __m256i cospi_p32_p32 = pair_set_w16_epi16(cospi[32], cospi[32]); |
| btf_16_adds_subs_avx2(&x1[0], &x1[3]); |
| btf_16_adds_subs_avx2(&x1[1], &x1[2]); |
| btf_16_w16_avx2(cospi_m32_p32, cospi_p32_p32, &x1[5], &x1[6], _r, cos_bit); |
| |
| btf_16_adds_subs_avx2(&x1[8], &x1[11]); |
| btf_16_adds_subs_avx2(&x1[9], &x1[10]); |
| btf_16_adds_subs_avx2(&x1[15], &x1[12]); |
| btf_16_adds_subs_avx2(&x1[14], &x1[13]); |
| } |
| |
| static inline void idct16_stage6_avx2(__m256i *x, const int32_t *cospi, |
| const __m256i _r, int8_t cos_bit) { |
| const __m256i cospi_m32_p32 = pair_set_w16_epi16(-cospi[32], cospi[32]); |
| const __m256i cospi_p32_p32 = pair_set_w16_epi16(cospi[32], cospi[32]); |
| btf_16_adds_subs_avx2(&x[0], &x[7]); |
| btf_16_adds_subs_avx2(&x[1], &x[6]); |
| btf_16_adds_subs_avx2(&x[2], &x[5]); |
| btf_16_adds_subs_avx2(&x[3], &x[4]); |
| btf_16_w16_avx2(cospi_m32_p32, cospi_p32_p32, &x[10], &x[13], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m32_p32, cospi_p32_p32, &x[11], &x[12], _r, cos_bit); |
| } |
| |
| static inline void idct16_stage7_avx2(__m256i *output, __m256i *x1) { |
| btf_16_adds_subs_out_avx2(&output[0], &output[15], x1[0], x1[15]); |
| btf_16_adds_subs_out_avx2(&output[1], &output[14], x1[1], x1[14]); |
| btf_16_adds_subs_out_avx2(&output[2], &output[13], x1[2], x1[13]); |
| btf_16_adds_subs_out_avx2(&output[3], &output[12], x1[3], x1[12]); |
| btf_16_adds_subs_out_avx2(&output[4], &output[11], x1[4], x1[11]); |
| btf_16_adds_subs_out_avx2(&output[5], &output[10], x1[5], x1[10]); |
| btf_16_adds_subs_out_avx2(&output[6], &output[9], x1[6], x1[9]); |
| btf_16_adds_subs_out_avx2(&output[7], &output[8], x1[7], x1[8]); |
| } |
| |
| static void idct16_avx2(const __m256i *input, __m256i *output) { |
| const int32_t *cospi = cospi_arr(INV_COS_BIT); |
| const __m256i _r = _mm256_set1_epi32(1 << (INV_COS_BIT - 1)); |
| |
| __m256i cospi_p60_m04 = pair_set_w16_epi16(cospi[60], -cospi[4]); |
| __m256i cospi_p04_p60 = pair_set_w16_epi16(cospi[4], cospi[60]); |
| __m256i cospi_p28_m36 = pair_set_w16_epi16(cospi[28], -cospi[36]); |
| __m256i cospi_p36_p28 = pair_set_w16_epi16(cospi[36], cospi[28]); |
| __m256i cospi_p44_m20 = pair_set_w16_epi16(cospi[44], -cospi[20]); |
| __m256i cospi_p20_p44 = pair_set_w16_epi16(cospi[20], cospi[44]); |
| __m256i cospi_p12_m52 = pair_set_w16_epi16(cospi[12], -cospi[52]); |
| __m256i cospi_p52_p12 = pair_set_w16_epi16(cospi[52], cospi[12]); |
| __m256i cospi_p56_m08 = pair_set_w16_epi16(cospi[56], -cospi[8]); |
| __m256i cospi_p08_p56 = pair_set_w16_epi16(cospi[8], cospi[56]); |
| __m256i cospi_p24_m40 = pair_set_w16_epi16(cospi[24], -cospi[40]); |
| __m256i cospi_p40_p24 = pair_set_w16_epi16(cospi[40], cospi[24]); |
| __m256i cospi_p32_p32 = pair_set_w16_epi16(cospi[32], cospi[32]); |
| __m256i cospi_p32_m32 = pair_set_w16_epi16(cospi[32], -cospi[32]); |
| __m256i cospi_p48_m16 = pair_set_w16_epi16(cospi[48], -cospi[16]); |
| __m256i cospi_p16_p48 = pair_set_w16_epi16(cospi[16], cospi[48]); |
| __m256i cospi_m16_p48 = pair_set_w16_epi16(-cospi[16], cospi[48]); |
| __m256i cospi_p48_p16 = pair_set_w16_epi16(cospi[48], cospi[16]); |
| __m256i cospi_m48_m16 = pair_set_w16_epi16(-cospi[48], -cospi[16]); |
| |
| // stage 1 |
| __m256i x1[16]; |
| x1[0] = input[0]; |
| x1[1] = input[8]; |
| x1[2] = input[4]; |
| x1[3] = input[12]; |
| x1[4] = input[2]; |
| x1[5] = input[10]; |
| x1[6] = input[6]; |
| x1[7] = input[14]; |
| x1[8] = input[1]; |
| x1[9] = input[9]; |
| x1[10] = input[5]; |
| x1[11] = input[13]; |
| x1[12] = input[3]; |
| x1[13] = input[11]; |
| x1[14] = input[7]; |
| x1[15] = input[15]; |
| |
| // stage 2 |
| btf_16_w16_avx2(cospi_p60_m04, cospi_p04_p60, &x1[8], &x1[15], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_p28_m36, cospi_p36_p28, &x1[9], &x1[14], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_p44_m20, cospi_p20_p44, &x1[10], &x1[13], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_p12_m52, cospi_p52_p12, &x1[11], &x1[12], _r, |
| INV_COS_BIT); |
| |
| // stage 3 |
| btf_16_w16_avx2(cospi_p56_m08, cospi_p08_p56, &x1[4], &x1[7], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_p24_m40, cospi_p40_p24, &x1[5], &x1[6], _r, |
| INV_COS_BIT); |
| btf_16_adds_subs_avx2(&x1[8], &x1[9]); |
| btf_16_adds_subs_avx2(&x1[11], &x1[10]); |
| btf_16_adds_subs_avx2(&x1[12], &x1[13]); |
| btf_16_adds_subs_avx2(&x1[15], &x1[14]); |
| |
| // stage 4 |
| btf_16_w16_avx2(cospi_p32_p32, cospi_p32_m32, &x1[0], &x1[1], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_p48_m16, cospi_p16_p48, &x1[2], &x1[3], _r, |
| INV_COS_BIT); |
| btf_16_adds_subs_avx2(&x1[4], &x1[5]); |
| btf_16_adds_subs_avx2(&x1[7], &x1[6]); |
| btf_16_w16_avx2(cospi_m16_p48, cospi_p48_p16, &x1[9], &x1[14], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_m48_m16, cospi_m16_p48, &x1[10], &x1[13], _r, |
| INV_COS_BIT); |
| |
| idct16_stage5_avx2(x1, cospi, _r, INV_COS_BIT); |
| idct16_stage6_avx2(x1, cospi, _r, INV_COS_BIT); |
| idct16_stage7_avx2(output, x1); |
| } |
| |
| static void idct16_low8_avx2(const __m256i *input, __m256i *output) { |
| const int32_t *cospi = cospi_arr(INV_COS_BIT); |
| const __m256i _r = _mm256_set1_epi32(1 << (INV_COS_BIT - 1)); |
| |
| const __m256i cospi_m16_p48 = pair_set_w16_epi16(-cospi[16], cospi[48]); |
| const __m256i cospi_p48_p16 = pair_set_w16_epi16(cospi[48], cospi[16]); |
| const __m256i cospi_m48_m16 = pair_set_w16_epi16(-cospi[48], -cospi[16]); |
| |
| // stage 1 |
| __m256i x1[16]; |
| x1[0] = input[0]; |
| x1[2] = input[4]; |
| x1[4] = input[2]; |
| x1[6] = input[6]; |
| x1[8] = input[1]; |
| x1[10] = input[5]; |
| x1[12] = input[3]; |
| x1[14] = input[7]; |
| |
| // stage 2 |
| btf_16_w16_0_avx2(cospi[60], cospi[4], x1[8], x1[8], x1[15]); |
| btf_16_w16_0_avx2(-cospi[36], cospi[28], x1[14], x1[9], x1[14]); |
| btf_16_w16_0_avx2(cospi[44], cospi[20], x1[10], x1[10], x1[13]); |
| btf_16_w16_0_avx2(-cospi[52], cospi[12], x1[12], x1[11], x1[12]); |
| |
| // stage 3 |
| btf_16_w16_0_avx2(cospi[56], cospi[8], x1[4], x1[4], x1[7]); |
| btf_16_w16_0_avx2(-cospi[40], cospi[24], x1[6], x1[5], x1[6]); |
| btf_16_adds_subs_avx2(&x1[8], &x1[9]); |
| btf_16_adds_subs_avx2(&x1[11], &x1[10]); |
| btf_16_adds_subs_avx2(&x1[12], &x1[13]); |
| btf_16_adds_subs_avx2(&x1[15], &x1[14]); |
| |
| // stage 4 |
| btf_16_w16_0_avx2(cospi[32], cospi[32], x1[0], x1[0], x1[1]); |
| btf_16_w16_0_avx2(cospi[48], cospi[16], x1[2], x1[2], x1[3]); |
| btf_16_adds_subs_avx2(&x1[4], &x1[5]); |
| btf_16_adds_subs_avx2(&x1[7], &x1[6]); |
| btf_16_w16_avx2(cospi_m16_p48, cospi_p48_p16, &x1[9], &x1[14], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_m48_m16, cospi_m16_p48, &x1[10], &x1[13], _r, |
| INV_COS_BIT); |
| |
| idct16_stage5_avx2(x1, cospi, _r, INV_COS_BIT); |
| idct16_stage6_avx2(x1, cospi, _r, INV_COS_BIT); |
| idct16_stage7_avx2(output, x1); |
| } |
| |
| static void idct16_low1_avx2(const __m256i *input, __m256i *output) { |
| const int32_t *cospi = cospi_arr(INV_COS_BIT); |
| |
| // stage 1 |
| __m256i x1[2]; |
| x1[0] = input[0]; |
| |
| // stage 2 |
| // stage 3 |
| // stage 4 |
| btf_16_w16_0_avx2(cospi[32], cospi[32], x1[0], x1[0], x1[1]); |
| |
| // stage 5 |
| // stage 6 |
| output[0] = x1[0]; |
| output[1] = x1[1]; |
| output[2] = x1[1]; |
| output[3] = x1[0]; |
| output[4] = x1[0]; |
| output[5] = x1[1]; |
| output[6] = x1[1]; |
| output[7] = x1[0]; |
| output[8] = x1[0]; |
| output[9] = x1[1]; |
| output[10] = x1[1]; |
| output[11] = x1[0]; |
| output[12] = x1[0]; |
| output[13] = x1[1]; |
| output[14] = x1[1]; |
| output[15] = x1[0]; |
| } |
| |
| static inline void iadst16_stage3_avx2(__m256i *x) { |
| btf_16_adds_subs_avx2(&x[0], &x[8]); |
| btf_16_adds_subs_avx2(&x[1], &x[9]); |
| btf_16_adds_subs_avx2(&x[2], &x[10]); |
| btf_16_adds_subs_avx2(&x[3], &x[11]); |
| btf_16_adds_subs_avx2(&x[4], &x[12]); |
| btf_16_adds_subs_avx2(&x[5], &x[13]); |
| btf_16_adds_subs_avx2(&x[6], &x[14]); |
| btf_16_adds_subs_avx2(&x[7], &x[15]); |
| } |
| |
| static inline void iadst16_stage4_avx2(__m256i *x, const int32_t *cospi, |
| const __m256i _r, int8_t cos_bit) { |
| const __m256i cospi_p08_p56 = pair_set_w16_epi16(cospi[8], cospi[56]); |
| const __m256i cospi_p56_m08 = pair_set_w16_epi16(cospi[56], -cospi[8]); |
| const __m256i cospi_p40_p24 = pair_set_w16_epi16(cospi[40], cospi[24]); |
| const __m256i cospi_p24_m40 = pair_set_w16_epi16(cospi[24], -cospi[40]); |
| const __m256i cospi_m56_p08 = pair_set_w16_epi16(-cospi[56], cospi[8]); |
| const __m256i cospi_m24_p40 = pair_set_w16_epi16(-cospi[24], cospi[40]); |
| btf_16_w16_avx2(cospi_p08_p56, cospi_p56_m08, &x[8], &x[9], _r, cos_bit); |
| btf_16_w16_avx2(cospi_p40_p24, cospi_p24_m40, &x[10], &x[11], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m56_p08, cospi_p08_p56, &x[12], &x[13], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m24_p40, cospi_p40_p24, &x[14], &x[15], _r, cos_bit); |
| } |
| |
| static inline void iadst16_stage5_avx2(__m256i *x) { |
| btf_16_adds_subs_avx2(&x[0], &x[4]); |
| btf_16_adds_subs_avx2(&x[1], &x[5]); |
| btf_16_adds_subs_avx2(&x[2], &x[6]); |
| btf_16_adds_subs_avx2(&x[3], &x[7]); |
| btf_16_adds_subs_avx2(&x[8], &x[12]); |
| btf_16_adds_subs_avx2(&x[9], &x[13]); |
| btf_16_adds_subs_avx2(&x[10], &x[14]); |
| btf_16_adds_subs_avx2(&x[11], &x[15]); |
| } |
| |
| static inline void iadst16_stage6_avx2(__m256i *x, const int32_t *cospi, |
| const __m256i _r, int8_t cos_bit) { |
| const __m256i cospi_p16_p48 = pair_set_w16_epi16(cospi[16], cospi[48]); |
| const __m256i cospi_p48_m16 = pair_set_w16_epi16(cospi[48], -cospi[16]); |
| const __m256i cospi_m48_p16 = pair_set_w16_epi16(-cospi[48], cospi[16]); |
| btf_16_w16_avx2(cospi_p16_p48, cospi_p48_m16, &x[4], &x[5], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m48_p16, cospi_p16_p48, &x[6], &x[7], _r, cos_bit); |
| btf_16_w16_avx2(cospi_p16_p48, cospi_p48_m16, &x[12], &x[13], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m48_p16, cospi_p16_p48, &x[14], &x[15], _r, cos_bit); |
| } |
| |
| static inline void iadst16_stage7_avx2(__m256i *x) { |
| btf_16_adds_subs_avx2(&x[0], &x[2]); |
| btf_16_adds_subs_avx2(&x[1], &x[3]); |
| btf_16_adds_subs_avx2(&x[4], &x[6]); |
| btf_16_adds_subs_avx2(&x[5], &x[7]); |
| btf_16_adds_subs_avx2(&x[8], &x[10]); |
| btf_16_adds_subs_avx2(&x[9], &x[11]); |
| btf_16_adds_subs_avx2(&x[12], &x[14]); |
| btf_16_adds_subs_avx2(&x[13], &x[15]); |
| } |
| |
| static inline void iadst16_stage8_avx2(__m256i *x1, const int32_t *cospi, |
| const __m256i _r, int8_t cos_bit) { |
| const __m256i cospi_p32_p32 = pair_set_w16_epi16(cospi[32], cospi[32]); |
| const __m256i cospi_p32_m32 = pair_set_w16_epi16(cospi[32], -cospi[32]); |
| btf_16_w16_avx2(cospi_p32_p32, cospi_p32_m32, &x1[2], &x1[3], _r, cos_bit); |
| btf_16_w16_avx2(cospi_p32_p32, cospi_p32_m32, &x1[6], &x1[7], _r, cos_bit); |
| btf_16_w16_avx2(cospi_p32_p32, cospi_p32_m32, &x1[10], &x1[11], _r, cos_bit); |
| btf_16_w16_avx2(cospi_p32_p32, cospi_p32_m32, &x1[14], &x1[15], _r, cos_bit); |
| } |
| |
| static inline void iadst16_stage9_avx2(__m256i *output, __m256i *x1) { |
| const __m256i __zero = _mm256_setzero_si256(); |
| output[0] = x1[0]; |
| output[1] = _mm256_subs_epi16(__zero, x1[8]); |
| output[2] = x1[12]; |
| output[3] = _mm256_subs_epi16(__zero, x1[4]); |
| output[4] = x1[6]; |
| output[5] = _mm256_subs_epi16(__zero, x1[14]); |
| output[6] = x1[10]; |
| output[7] = _mm256_subs_epi16(__zero, x1[2]); |
| output[8] = x1[3]; |
| output[9] = _mm256_subs_epi16(__zero, x1[11]); |
| output[10] = x1[15]; |
| output[11] = _mm256_subs_epi16(__zero, x1[7]); |
| output[12] = x1[5]; |
| output[13] = _mm256_subs_epi16(__zero, x1[13]); |
| output[14] = x1[9]; |
| output[15] = _mm256_subs_epi16(__zero, x1[1]); |
| } |
| |
| static void iadst16_avx2(const __m256i *input, __m256i *output) { |
| const int32_t *cospi = cospi_arr(INV_COS_BIT); |
| |
| const __m256i _r = _mm256_set1_epi32(1 << (INV_COS_BIT - 1)); |
| |
| __m256i cospi_p02_p62 = pair_set_w16_epi16(cospi[2], cospi[62]); |
| __m256i cospi_p62_m02 = pair_set_w16_epi16(cospi[62], -cospi[2]); |
| __m256i cospi_p10_p54 = pair_set_w16_epi16(cospi[10], cospi[54]); |
| __m256i cospi_p54_m10 = pair_set_w16_epi16(cospi[54], -cospi[10]); |
| __m256i cospi_p18_p46 = pair_set_w16_epi16(cospi[18], cospi[46]); |
| __m256i cospi_p46_m18 = pair_set_w16_epi16(cospi[46], -cospi[18]); |
| __m256i cospi_p26_p38 = pair_set_w16_epi16(cospi[26], cospi[38]); |
| __m256i cospi_p38_m26 = pair_set_w16_epi16(cospi[38], -cospi[26]); |
| __m256i cospi_p34_p30 = pair_set_w16_epi16(cospi[34], cospi[30]); |
| __m256i cospi_p30_m34 = pair_set_w16_epi16(cospi[30], -cospi[34]); |
| __m256i cospi_p42_p22 = pair_set_w16_epi16(cospi[42], cospi[22]); |
| __m256i cospi_p22_m42 = pair_set_w16_epi16(cospi[22], -cospi[42]); |
| __m256i cospi_p50_p14 = pair_set_w16_epi16(cospi[50], cospi[14]); |
| __m256i cospi_p14_m50 = pair_set_w16_epi16(cospi[14], -cospi[50]); |
| __m256i cospi_p58_p06 = pair_set_w16_epi16(cospi[58], cospi[6]); |
| __m256i cospi_p06_m58 = pair_set_w16_epi16(cospi[6], -cospi[58]); |
| |
| // stage 1 |
| __m256i x1[16]; |
| x1[0] = input[15]; |
| x1[1] = input[0]; |
| x1[2] = input[13]; |
| x1[3] = input[2]; |
| x1[4] = input[11]; |
| x1[5] = input[4]; |
| x1[6] = input[9]; |
| x1[7] = input[6]; |
| x1[8] = input[7]; |
| x1[9] = input[8]; |
| x1[10] = input[5]; |
| x1[11] = input[10]; |
| x1[12] = input[3]; |
| x1[13] = input[12]; |
| x1[14] = input[1]; |
| x1[15] = input[14]; |
| |
| // stage 2 |
| btf_16_w16_avx2(cospi_p02_p62, cospi_p62_m02, &x1[0], &x1[1], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_p10_p54, cospi_p54_m10, &x1[2], &x1[3], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_p18_p46, cospi_p46_m18, &x1[4], &x1[5], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_p26_p38, cospi_p38_m26, &x1[6], &x1[7], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_p34_p30, cospi_p30_m34, &x1[8], &x1[9], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_p42_p22, cospi_p22_m42, &x1[10], &x1[11], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_p50_p14, cospi_p14_m50, &x1[12], &x1[13], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_p58_p06, cospi_p06_m58, &x1[14], &x1[15], _r, |
| INV_COS_BIT); |
| |
| iadst16_stage3_avx2(x1); |
| iadst16_stage4_avx2(x1, cospi, _r, INV_COS_BIT); |
| iadst16_stage5_avx2(x1); |
| iadst16_stage6_avx2(x1, cospi, _r, INV_COS_BIT); |
| iadst16_stage7_avx2(x1); |
| iadst16_stage8_avx2(x1, cospi, _r, INV_COS_BIT); |
| iadst16_stage9_avx2(output, x1); |
| } |
| |
| static void iadst16_low8_avx2(const __m256i *input, __m256i *output) { |
| const int32_t *cospi = cospi_arr(INV_COS_BIT); |
| const __m256i _r = _mm256_set1_epi32(1 << (INV_COS_BIT - 1)); |
| |
| // stage 1 |
| __m256i x1[16]; |
| x1[1] = input[0]; |
| x1[3] = input[2]; |
| x1[5] = input[4]; |
| x1[7] = input[6]; |
| x1[8] = input[7]; |
| x1[10] = input[5]; |
| x1[12] = input[3]; |
| x1[14] = input[1]; |
| |
| // stage 2 |
| btf_16_w16_0_avx2(cospi[62], -cospi[2], x1[1], x1[0], x1[1]); |
| btf_16_w16_0_avx2(cospi[54], -cospi[10], x1[3], x1[2], x1[3]); |
| btf_16_w16_0_avx2(cospi[46], -cospi[18], x1[5], x1[4], x1[5]); |
| btf_16_w16_0_avx2(cospi[38], -cospi[26], x1[7], x1[6], x1[7]); |
| btf_16_w16_0_avx2(cospi[34], cospi[30], x1[8], x1[8], x1[9]); |
| btf_16_w16_0_avx2(cospi[42], cospi[22], x1[10], x1[10], x1[11]); |
| btf_16_w16_0_avx2(cospi[50], cospi[14], x1[12], x1[12], x1[13]); |
| btf_16_w16_0_avx2(cospi[58], cospi[06], x1[14], x1[14], x1[15]); |
| |
| iadst16_stage3_avx2(x1); |
| iadst16_stage4_avx2(x1, cospi, _r, INV_COS_BIT); |
| iadst16_stage5_avx2(x1); |
| iadst16_stage6_avx2(x1, cospi, _r, INV_COS_BIT); |
| iadst16_stage7_avx2(x1); |
| iadst16_stage8_avx2(x1, cospi, _r, INV_COS_BIT); |
| iadst16_stage9_avx2(output, x1); |
| } |
| |
| static void iadst16_low1_avx2(const __m256i *input, __m256i *output) { |
| const int32_t *cospi = cospi_arr(INV_COS_BIT); |
| const __m256i _r = _mm256_set1_epi32(1 << (INV_COS_BIT - 1)); |
| |
| const __m256i cospi_p08_p56 = pair_set_w16_epi16(cospi[8], cospi[56]); |
| const __m256i cospi_p56_m08 = pair_set_w16_epi16(cospi[56], -cospi[8]); |
| const __m256i cospi_p16_p48 = pair_set_w16_epi16(cospi[16], cospi[48]); |
| const __m256i cospi_p48_m16 = pair_set_w16_epi16(cospi[48], -cospi[16]); |
| |
| // stage 1 |
| __m256i x1[16]; |
| x1[1] = input[0]; |
| |
| // stage 2 |
| btf_16_w16_0_avx2(cospi[62], -cospi[2], x1[1], x1[0], x1[1]); |
| |
| // stage 3 |
| x1[8] = x1[0]; |
| x1[9] = x1[1]; |
| |
| // stage 4 |
| btf_16_w16_avx2(cospi_p08_p56, cospi_p56_m08, &x1[8], &x1[9], _r, |
| INV_COS_BIT); |
| |
| // stage 5 |
| x1[4] = x1[0]; |
| x1[5] = x1[1]; |
| |
| x1[12] = x1[8]; |
| x1[13] = x1[9]; |
| |
| // stage 6 |
| btf_16_w16_avx2(cospi_p16_p48, cospi_p48_m16, &x1[4], &x1[5], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_p16_p48, cospi_p48_m16, &x1[12], &x1[13], _r, |
| INV_COS_BIT); |
| |
| // stage 7 |
| x1[2] = x1[0]; |
| x1[3] = x1[1]; |
| x1[6] = x1[4]; |
| x1[7] = x1[5]; |
| x1[10] = x1[8]; |
| x1[11] = x1[9]; |
| x1[14] = x1[12]; |
| x1[15] = x1[13]; |
| |
| iadst16_stage8_avx2(x1, cospi, _r, INV_COS_BIT); |
| iadst16_stage9_avx2(output, x1); |
| } |
| |
| static inline void idct32_high16_stage3_avx2(__m256i *x) { |
| btf_16_adds_subs_avx2(&x[16], &x[17]); |
| btf_16_adds_subs_avx2(&x[19], &x[18]); |
| btf_16_adds_subs_avx2(&x[20], &x[21]); |
| btf_16_adds_subs_avx2(&x[23], &x[22]); |
| btf_16_adds_subs_avx2(&x[24], &x[25]); |
| btf_16_adds_subs_avx2(&x[27], &x[26]); |
| btf_16_adds_subs_avx2(&x[28], &x[29]); |
| btf_16_adds_subs_avx2(&x[31], &x[30]); |
| } |
| |
| static inline void idct32_high16_stage4_avx2(__m256i *x, const int32_t *cospi, |
| const __m256i _r, int8_t cos_bit) { |
| const __m256i cospi_m08_p56 = pair_set_w16_epi16(-cospi[8], cospi[56]); |
| const __m256i cospi_p56_p08 = pair_set_w16_epi16(cospi[56], cospi[8]); |
| const __m256i cospi_m56_m08 = pair_set_w16_epi16(-cospi[56], -cospi[8]); |
| const __m256i cospi_m40_p24 = pair_set_w16_epi16(-cospi[40], cospi[24]); |
| const __m256i cospi_p24_p40 = pair_set_w16_epi16(cospi[24], cospi[40]); |
| const __m256i cospi_m24_m40 = pair_set_w16_epi16(-cospi[24], -cospi[40]); |
| btf_16_w16_avx2(cospi_m08_p56, cospi_p56_p08, &x[17], &x[30], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m56_m08, cospi_m08_p56, &x[18], &x[29], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m40_p24, cospi_p24_p40, &x[21], &x[26], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m24_m40, cospi_m40_p24, &x[22], &x[25], _r, cos_bit); |
| } |
| |
| static inline void idct32_high24_stage5_avx2(__m256i *x, const int32_t *cospi, |
| const __m256i _r, int8_t cos_bit) { |
| const __m256i cospi_m16_p48 = pair_set_w16_epi16(-cospi[16], cospi[48]); |
| const __m256i cospi_p48_p16 = pair_set_w16_epi16(cospi[48], cospi[16]); |
| const __m256i cospi_m48_m16 = pair_set_w16_epi16(-cospi[48], -cospi[16]); |
| btf_16_w16_avx2(cospi_m16_p48, cospi_p48_p16, &x[9], &x[14], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m48_m16, cospi_m16_p48, &x[10], &x[13], _r, cos_bit); |
| btf_16_adds_subs_avx2(&x[16], &x[19]); |
| btf_16_adds_subs_avx2(&x[17], &x[18]); |
| btf_16_adds_subs_avx2(&x[23], &x[20]); |
| btf_16_adds_subs_avx2(&x[22], &x[21]); |
| btf_16_adds_subs_avx2(&x[24], &x[27]); |
| btf_16_adds_subs_avx2(&x[25], &x[26]); |
| btf_16_adds_subs_avx2(&x[31], &x[28]); |
| btf_16_adds_subs_avx2(&x[30], &x[29]); |
| } |
| |
| static inline void idct32_high28_stage6_avx2(__m256i *x, const int32_t *cospi, |
| const __m256i _r, int8_t cos_bit) { |
| const __m256i cospi_m32_p32 = pair_set_w16_epi16(-cospi[32], cospi[32]); |
| const __m256i cospi_p32_p32 = pair_set_w16_epi16(cospi[32], cospi[32]); |
| const __m256i cospi_m16_p48 = pair_set_w16_epi16(-cospi[16], cospi[48]); |
| const __m256i cospi_p48_p16 = pair_set_w16_epi16(cospi[48], cospi[16]); |
| const __m256i cospi_m48_m16 = pair_set_w16_epi16(-cospi[48], -cospi[16]); |
| btf_16_w16_avx2(cospi_m32_p32, cospi_p32_p32, &x[5], &x[6], _r, cos_bit); |
| btf_16_adds_subs_avx2(&x[8], &x[11]); |
| btf_16_adds_subs_avx2(&x[9], &x[10]); |
| btf_16_adds_subs_avx2(&x[15], &x[12]); |
| btf_16_adds_subs_avx2(&x[14], &x[13]); |
| btf_16_w16_avx2(cospi_m16_p48, cospi_p48_p16, &x[18], &x[29], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m16_p48, cospi_p48_p16, &x[19], &x[28], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m48_m16, cospi_m16_p48, &x[20], &x[27], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m48_m16, cospi_m16_p48, &x[21], &x[26], _r, cos_bit); |
| } |
| |
| static inline void idct32_stage7_avx2(__m256i *x, const int32_t *cospi, |
| const __m256i _r, int8_t cos_bit) { |
| const __m256i cospi_m32_p32 = pair_set_w16_epi16(-cospi[32], cospi[32]); |
| const __m256i cospi_p32_p32 = pair_set_w16_epi16(cospi[32], cospi[32]); |
| btf_16_adds_subs_avx2(&x[0], &x[7]); |
| btf_16_adds_subs_avx2(&x[1], &x[6]); |
| btf_16_adds_subs_avx2(&x[2], &x[5]); |
| btf_16_adds_subs_avx2(&x[3], &x[4]); |
| btf_16_w16_avx2(cospi_m32_p32, cospi_p32_p32, &x[10], &x[13], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m32_p32, cospi_p32_p32, &x[11], &x[12], _r, cos_bit); |
| btf_16_adds_subs_avx2(&x[16], &x[23]); |
| btf_16_adds_subs_avx2(&x[17], &x[22]); |
| btf_16_adds_subs_avx2(&x[18], &x[21]); |
| btf_16_adds_subs_avx2(&x[19], &x[20]); |
| btf_16_adds_subs_avx2(&x[31], &x[24]); |
| btf_16_adds_subs_avx2(&x[30], &x[25]); |
| btf_16_adds_subs_avx2(&x[29], &x[26]); |
| btf_16_adds_subs_avx2(&x[28], &x[27]); |
| } |
| |
| static inline void idct32_stage8_avx2(__m256i *x, const int32_t *cospi, |
| const __m256i _r, int8_t cos_bit) { |
| const __m256i cospi_m32_p32 = pair_set_w16_epi16(-cospi[32], cospi[32]); |
| const __m256i cospi_p32_p32 = pair_set_w16_epi16(cospi[32], cospi[32]); |
| btf_16_adds_subs_avx2(&x[0], &x[15]); |
| btf_16_adds_subs_avx2(&x[1], &x[14]); |
| btf_16_adds_subs_avx2(&x[2], &x[13]); |
| btf_16_adds_subs_avx2(&x[3], &x[12]); |
| btf_16_adds_subs_avx2(&x[4], &x[11]); |
| btf_16_adds_subs_avx2(&x[5], &x[10]); |
| btf_16_adds_subs_avx2(&x[6], &x[9]); |
| btf_16_adds_subs_avx2(&x[7], &x[8]); |
| btf_16_w16_avx2(cospi_m32_p32, cospi_p32_p32, &x[20], &x[27], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m32_p32, cospi_p32_p32, &x[21], &x[26], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m32_p32, cospi_p32_p32, &x[22], &x[25], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m32_p32, cospi_p32_p32, &x[23], &x[24], _r, cos_bit); |
| } |
| |
| static inline void idct32_stage9_avx2(__m256i *output, __m256i *x) { |
| btf_16_adds_subs_out_avx2(&output[0], &output[31], x[0], x[31]); |
| btf_16_adds_subs_out_avx2(&output[1], &output[30], x[1], x[30]); |
| btf_16_adds_subs_out_avx2(&output[2], &output[29], x[2], x[29]); |
| btf_16_adds_subs_out_avx2(&output[3], &output[28], x[3], x[28]); |
| btf_16_adds_subs_out_avx2(&output[4], &output[27], x[4], x[27]); |
| btf_16_adds_subs_out_avx2(&output[5], &output[26], x[5], x[26]); |
| btf_16_adds_subs_out_avx2(&output[6], &output[25], x[6], x[25]); |
| btf_16_adds_subs_out_avx2(&output[7], &output[24], x[7], x[24]); |
| btf_16_adds_subs_out_avx2(&output[8], &output[23], x[8], x[23]); |
| btf_16_adds_subs_out_avx2(&output[9], &output[22], x[9], x[22]); |
| btf_16_adds_subs_out_avx2(&output[10], &output[21], x[10], x[21]); |
| btf_16_adds_subs_out_avx2(&output[11], &output[20], x[11], x[20]); |
| btf_16_adds_subs_out_avx2(&output[12], &output[19], x[12], x[19]); |
| btf_16_adds_subs_out_avx2(&output[13], &output[18], x[13], x[18]); |
| btf_16_adds_subs_out_avx2(&output[14], &output[17], x[14], x[17]); |
| btf_16_adds_subs_out_avx2(&output[15], &output[16], x[15], x[16]); |
| } |
| |
| static void idct32_low1_avx2(const __m256i *input, __m256i *output) { |
| const int32_t *cospi = cospi_arr(INV_COS_BIT); |
| |
| // stage 1 |
| __m256i x[2]; |
| x[0] = input[0]; |
| |
| // stage 2 |
| // stage 3 |
| // stage 4 |
| // stage 5 |
| btf_16_w16_0_avx2(cospi[32], cospi[32], x[0], x[0], x[1]); |
| |
| // stage 6 |
| // stage 7 |
| // stage 8 |
| // stage 9 |
| output[0] = x[0]; |
| output[31] = x[0]; |
| output[1] = x[1]; |
| output[30] = x[1]; |
| output[2] = x[1]; |
| output[29] = x[1]; |
| output[3] = x[0]; |
| output[28] = x[0]; |
| output[4] = x[0]; |
| output[27] = x[0]; |
| output[5] = x[1]; |
| output[26] = x[1]; |
| output[6] = x[1]; |
| output[25] = x[1]; |
| output[7] = x[0]; |
| output[24] = x[0]; |
| output[8] = x[0]; |
| output[23] = x[0]; |
| output[9] = x[1]; |
| output[22] = x[1]; |
| output[10] = x[1]; |
| output[21] = x[1]; |
| output[11] = x[0]; |
| output[20] = x[0]; |
| output[12] = x[0]; |
| output[19] = x[0]; |
| output[13] = x[1]; |
| output[18] = x[1]; |
| output[14] = x[1]; |
| output[17] = x[1]; |
| output[15] = x[0]; |
| output[16] = x[0]; |
| } |
| |
| static void idct32_low8_avx2(const __m256i *input, __m256i *output) { |
| const int32_t *cospi = cospi_arr(INV_COS_BIT); |
| const __m256i _r = _mm256_set1_epi32(1 << (INV_COS_BIT - 1)); |
| |
| // stage 1 |
| __m256i x[32]; |
| x[0] = input[0]; |
| x[4] = input[4]; |
| x[8] = input[2]; |
| x[12] = input[6]; |
| x[16] = input[1]; |
| x[20] = input[5]; |
| x[24] = input[3]; |
| x[28] = input[7]; |
| |
| // stage 2 |
| btf_16_w16_0_avx2(cospi[62], cospi[2], x[16], x[16], x[31]); |
| btf_16_w16_0_avx2(-cospi[50], cospi[14], x[28], x[19], x[28]); |
| btf_16_w16_0_avx2(cospi[54], cospi[10], x[20], x[20], x[27]); |
| btf_16_w16_0_avx2(-cospi[58], cospi[6], x[24], x[23], x[24]); |
| |
| // stage 3 |
| btf_16_w16_0_avx2(cospi[60], cospi[4], x[8], x[8], x[15]); |
| btf_16_w16_0_avx2(-cospi[52], cospi[12], x[12], x[11], x[12]); |
| x[17] = x[16]; |
| x[18] = x[19]; |
| x[21] = x[20]; |
| x[22] = x[23]; |
| x[25] = x[24]; |
| x[26] = x[27]; |
| x[29] = x[28]; |
| x[30] = x[31]; |
| |
| // stage 4 |
| btf_16_w16_0_avx2(cospi[56], cospi[8], x[4], x[4], x[7]); |
| x[9] = x[8]; |
| x[10] = x[11]; |
| x[13] = x[12]; |
| x[14] = x[15]; |
| idct32_high16_stage4_avx2(x, cospi, _r, INV_COS_BIT); |
| |
| // stage 5 |
| btf_16_w16_0_avx2(cospi[32], cospi[32], x[0], x[0], x[1]); |
| x[5] = x[4]; |
| x[6] = x[7]; |
| idct32_high24_stage5_avx2(x, cospi, _r, INV_COS_BIT); |
| // stage 6 |
| x[3] = x[0]; |
| x[2] = x[1]; |
| idct32_high28_stage6_avx2(x, cospi, _r, INV_COS_BIT); |
| |
| idct32_stage7_avx2(x, cospi, _r, INV_COS_BIT); |
| idct32_stage8_avx2(x, cospi, _r, INV_COS_BIT); |
| idct32_stage9_avx2(output, x); |
| } |
| |
| static void idct32_low16_avx2(const __m256i *input, __m256i *output) { |
| const int32_t *cospi = cospi_arr(INV_COS_BIT); |
| const __m256i _r = _mm256_set1_epi32(1 << (INV_COS_BIT - 1)); |
| |
| // stage 1 |
| __m256i x[32]; |
| x[0] = input[0]; |
| x[2] = input[8]; |
| x[4] = input[4]; |
| x[6] = input[12]; |
| x[8] = input[2]; |
| x[10] = input[10]; |
| x[12] = input[6]; |
| x[14] = input[14]; |
| x[16] = input[1]; |
| x[18] = input[9]; |
| x[20] = input[5]; |
| x[22] = input[13]; |
| x[24] = input[3]; |
| x[26] = input[11]; |
| x[28] = input[7]; |
| x[30] = input[15]; |
| |
| // stage 2 |
| btf_16_w16_0_avx2(cospi[62], cospi[2], x[16], x[16], x[31]); |
| btf_16_w16_0_avx2(-cospi[34], cospi[30], x[30], x[17], x[30]); |
| btf_16_w16_0_avx2(cospi[46], cospi[18], x[18], x[18], x[29]); |
| btf_16_w16_0_avx2(-cospi[50], cospi[14], x[28], x[19], x[28]); |
| btf_16_w16_0_avx2(cospi[54], cospi[10], x[20], x[20], x[27]); |
| btf_16_w16_0_avx2(-cospi[42], cospi[22], x[26], x[21], x[26]); |
| btf_16_w16_0_avx2(cospi[38], cospi[26], x[22], x[22], x[25]); |
| btf_16_w16_0_avx2(-cospi[58], cospi[6], x[24], x[23], x[24]); |
| |
| // stage 3 |
| btf_16_w16_0_avx2(cospi[60], cospi[4], x[8], x[8], x[15]); |
| btf_16_w16_0_avx2(-cospi[36], cospi[28], x[14], x[9], x[14]); |
| btf_16_w16_0_avx2(cospi[44], cospi[20], x[10], x[10], x[13]); |
| btf_16_w16_0_avx2(-cospi[52], cospi[12], x[12], x[11], x[12]); |
| idct32_high16_stage3_avx2(x); |
| |
| // stage 4 |
| btf_16_w16_0_avx2(cospi[56], cospi[8], x[4], x[4], x[7]); |
| btf_16_w16_0_avx2(-cospi[40], cospi[24], x[6], x[5], x[6]); |
| btf_16_adds_subs_avx2(&x[8], &x[9]); |
| btf_16_adds_subs_avx2(&x[11], &x[10]); |
| btf_16_adds_subs_avx2(&x[12], &x[13]); |
| btf_16_adds_subs_avx2(&x[15], &x[14]); |
| idct32_high16_stage4_avx2(x, cospi, _r, INV_COS_BIT); |
| |
| // stage 5 |
| btf_16_w16_0_avx2(cospi[32], cospi[32], x[0], x[0], x[1]); |
| btf_16_w16_0_avx2(cospi[48], cospi[16], x[2], x[2], x[3]); |
| btf_16_adds_subs_avx2(&x[4], &x[5]); |
| btf_16_adds_subs_avx2(&x[7], &x[6]); |
| idct32_high24_stage5_avx2(x, cospi, _r, INV_COS_BIT); |
| |
| btf_16_adds_subs_avx2(&x[0], &x[3]); |
| btf_16_adds_subs_avx2(&x[1], &x[2]); |
| idct32_high28_stage6_avx2(x, cospi, _r, INV_COS_BIT); |
| |
| idct32_stage7_avx2(x, cospi, _r, INV_COS_BIT); |
| idct32_stage8_avx2(x, cospi, _r, INV_COS_BIT); |
| idct32_stage9_avx2(output, x); |
| } |
| |
| static void idct32_avx2(const __m256i *input, __m256i *output) { |
| const int32_t *cospi = cospi_arr(INV_COS_BIT); |
| const __m256i _r = _mm256_set1_epi32(1 << (INV_COS_BIT - 1)); |
| |
| __m256i cospi_p62_m02 = pair_set_w16_epi16(cospi[62], -cospi[2]); |
| __m256i cospi_p02_p62 = pair_set_w16_epi16(cospi[2], cospi[62]); |
| __m256i cospi_p30_m34 = pair_set_w16_epi16(cospi[30], -cospi[34]); |
| __m256i cospi_p34_p30 = pair_set_w16_epi16(cospi[34], cospi[30]); |
| __m256i cospi_p46_m18 = pair_set_w16_epi16(cospi[46], -cospi[18]); |
| __m256i cospi_p18_p46 = pair_set_w16_epi16(cospi[18], cospi[46]); |
| __m256i cospi_p14_m50 = pair_set_w16_epi16(cospi[14], -cospi[50]); |
| __m256i cospi_p50_p14 = pair_set_w16_epi16(cospi[50], cospi[14]); |
| __m256i cospi_p54_m10 = pair_set_w16_epi16(cospi[54], -cospi[10]); |
| __m256i cospi_p10_p54 = pair_set_w16_epi16(cospi[10], cospi[54]); |
| __m256i cospi_p22_m42 = pair_set_w16_epi16(cospi[22], -cospi[42]); |
| __m256i cospi_p42_p22 = pair_set_w16_epi16(cospi[42], cospi[22]); |
| __m256i cospi_p38_m26 = pair_set_w16_epi16(cospi[38], -cospi[26]); |
| __m256i cospi_p26_p38 = pair_set_w16_epi16(cospi[26], cospi[38]); |
| __m256i cospi_p06_m58 = pair_set_w16_epi16(cospi[6], -cospi[58]); |
| __m256i cospi_p58_p06 = pair_set_w16_epi16(cospi[58], cospi[6]); |
| __m256i cospi_p60_m04 = pair_set_w16_epi16(cospi[60], -cospi[4]); |
| __m256i cospi_p04_p60 = pair_set_w16_epi16(cospi[4], cospi[60]); |
| __m256i cospi_p28_m36 = pair_set_w16_epi16(cospi[28], -cospi[36]); |
| __m256i cospi_p36_p28 = pair_set_w16_epi16(cospi[36], cospi[28]); |
| __m256i cospi_p44_m20 = pair_set_w16_epi16(cospi[44], -cospi[20]); |
| __m256i cospi_p20_p44 = pair_set_w16_epi16(cospi[20], cospi[44]); |
| __m256i cospi_p12_m52 = pair_set_w16_epi16(cospi[12], -cospi[52]); |
| __m256i cospi_p52_p12 = pair_set_w16_epi16(cospi[52], cospi[12]); |
| __m256i cospi_p56_m08 = pair_set_w16_epi16(cospi[56], -cospi[8]); |
| __m256i cospi_p08_p56 = pair_set_w16_epi16(cospi[8], cospi[56]); |
| __m256i cospi_p24_m40 = pair_set_w16_epi16(cospi[24], -cospi[40]); |
| __m256i cospi_p40_p24 = pair_set_w16_epi16(cospi[40], cospi[24]); |
| __m256i cospi_p32_p32 = pair_set_w16_epi16(cospi[32], cospi[32]); |
| __m256i cospi_p32_m32 = pair_set_w16_epi16(cospi[32], -cospi[32]); |
| __m256i cospi_p48_m16 = pair_set_w16_epi16(cospi[48], -cospi[16]); |
| __m256i cospi_p16_p48 = pair_set_w16_epi16(cospi[16], cospi[48]); |
| |
| // stage 1 |
| __m256i x1[32]; |
| x1[0] = input[0]; |
| x1[1] = input[16]; |
| x1[2] = input[8]; |
| x1[3] = input[24]; |
| x1[4] = input[4]; |
| x1[5] = input[20]; |
| x1[6] = input[12]; |
| x1[7] = input[28]; |
| x1[8] = input[2]; |
| x1[9] = input[18]; |
| x1[10] = input[10]; |
| x1[11] = input[26]; |
| x1[12] = input[6]; |
| x1[13] = input[22]; |
| x1[14] = input[14]; |
| x1[15] = input[30]; |
| x1[16] = input[1]; |
| x1[17] = input[17]; |
| x1[18] = input[9]; |
| x1[19] = input[25]; |
| x1[20] = input[5]; |
| x1[21] = input[21]; |
| x1[22] = input[13]; |
| x1[23] = input[29]; |
| x1[24] = input[3]; |
| x1[25] = input[19]; |
| x1[26] = input[11]; |
| x1[27] = input[27]; |
| x1[28] = input[7]; |
| x1[29] = input[23]; |
| x1[30] = input[15]; |
| x1[31] = input[31]; |
| |
| // stage 2 |
| btf_16_w16_avx2(cospi_p62_m02, cospi_p02_p62, &x1[16], &x1[31], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_p30_m34, cospi_p34_p30, &x1[17], &x1[30], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_p46_m18, cospi_p18_p46, &x1[18], &x1[29], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_p14_m50, cospi_p50_p14, &x1[19], &x1[28], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_p54_m10, cospi_p10_p54, &x1[20], &x1[27], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_p22_m42, cospi_p42_p22, &x1[21], &x1[26], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_p38_m26, cospi_p26_p38, &x1[22], &x1[25], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_p06_m58, cospi_p58_p06, &x1[23], &x1[24], _r, |
| INV_COS_BIT); |
| |
| // stage 3 |
| btf_16_w16_avx2(cospi_p60_m04, cospi_p04_p60, &x1[8], &x1[15], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_p28_m36, cospi_p36_p28, &x1[9], &x1[14], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_p44_m20, cospi_p20_p44, &x1[10], &x1[13], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_p12_m52, cospi_p52_p12, &x1[11], &x1[12], _r, |
| INV_COS_BIT); |
| idct32_high16_stage3_avx2(x1); |
| |
| // stage 4 |
| btf_16_w16_avx2(cospi_p56_m08, cospi_p08_p56, &x1[4], &x1[7], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_p24_m40, cospi_p40_p24, &x1[5], &x1[6], _r, |
| INV_COS_BIT); |
| btf_16_adds_subs_avx2(&x1[8], &x1[9]); |
| btf_16_adds_subs_avx2(&x1[11], &x1[10]); |
| btf_16_adds_subs_avx2(&x1[12], &x1[13]); |
| btf_16_adds_subs_avx2(&x1[15], &x1[14]); |
| idct32_high16_stage4_avx2(x1, cospi, _r, INV_COS_BIT); |
| |
| // stage 5 |
| btf_16_w16_avx2(cospi_p32_p32, cospi_p32_m32, &x1[0], &x1[1], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_p48_m16, cospi_p16_p48, &x1[2], &x1[3], _r, |
| INV_COS_BIT); |
| btf_16_adds_subs_avx2(&x1[4], &x1[5]); |
| btf_16_adds_subs_avx2(&x1[7], &x1[6]); |
| idct32_high24_stage5_avx2(x1, cospi, _r, INV_COS_BIT); |
| |
| // stage 6 |
| btf_16_adds_subs_avx2(&x1[0], &x1[3]); |
| btf_16_adds_subs_avx2(&x1[1], &x1[2]); |
| idct32_high28_stage6_avx2(x1, cospi, _r, INV_COS_BIT); |
| |
| idct32_stage7_avx2(x1, cospi, _r, INV_COS_BIT); |
| idct32_stage8_avx2(x1, cospi, _r, INV_COS_BIT); |
| idct32_stage9_avx2(output, x1); |
| } |
| |
| static inline void idct64_stage4_high32_avx2(__m256i *x, const int32_t *cospi, |
| const __m256i _r, int8_t cos_bit) { |
| (void)cos_bit; |
| const __m256i cospi_m04_p60 = pair_set_w16_epi16(-cospi[4], cospi[60]); |
| const __m256i cospi_p60_p04 = pair_set_w16_epi16(cospi[60], cospi[4]); |
| const __m256i cospi_m60_m04 = pair_set_w16_epi16(-cospi[60], -cospi[4]); |
| const __m256i cospi_m36_p28 = pair_set_w16_epi16(-cospi[36], cospi[28]); |
| const __m256i cospi_p28_p36 = pair_set_w16_epi16(cospi[28], cospi[36]); |
| const __m256i cospi_m28_m36 = pair_set_w16_epi16(-cospi[28], -cospi[36]); |
| const __m256i cospi_m20_p44 = pair_set_w16_epi16(-cospi[20], cospi[44]); |
| const __m256i cospi_p44_p20 = pair_set_w16_epi16(cospi[44], cospi[20]); |
| const __m256i cospi_m44_m20 = pair_set_w16_epi16(-cospi[44], -cospi[20]); |
| const __m256i cospi_m52_p12 = pair_set_w16_epi16(-cospi[52], cospi[12]); |
| const __m256i cospi_p12_p52 = pair_set_w16_epi16(cospi[12], cospi[52]); |
| const __m256i cospi_m12_m52 = pair_set_w16_epi16(-cospi[12], -cospi[52]); |
| btf_16_w16_avx2(cospi_m04_p60, cospi_p60_p04, &x[33], &x[62], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m60_m04, cospi_m04_p60, &x[34], &x[61], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m36_p28, cospi_p28_p36, &x[37], &x[58], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m28_m36, cospi_m36_p28, &x[38], &x[57], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m20_p44, cospi_p44_p20, &x[41], &x[54], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m44_m20, cospi_m20_p44, &x[42], &x[53], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m52_p12, cospi_p12_p52, &x[45], &x[50], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m12_m52, cospi_m52_p12, &x[46], &x[49], _r, cos_bit); |
| } |
| |
| static inline void idct64_stage5_high48_avx2(__m256i *x, const int32_t *cospi, |
| const __m256i _r, int8_t cos_bit) { |
| (void)cos_bit; |
| const __m256i cospi_m08_p56 = pair_set_w16_epi16(-cospi[8], cospi[56]); |
| const __m256i cospi_p56_p08 = pair_set_w16_epi16(cospi[56], cospi[8]); |
| const __m256i cospi_m56_m08 = pair_set_w16_epi16(-cospi[56], -cospi[8]); |
| const __m256i cospi_m40_p24 = pair_set_w16_epi16(-cospi[40], cospi[24]); |
| const __m256i cospi_p24_p40 = pair_set_w16_epi16(cospi[24], cospi[40]); |
| const __m256i cospi_m24_m40 = pair_set_w16_epi16(-cospi[24], -cospi[40]); |
| btf_16_w16_avx2(cospi_m08_p56, cospi_p56_p08, &x[17], &x[30], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m56_m08, cospi_m08_p56, &x[18], &x[29], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m40_p24, cospi_p24_p40, &x[21], &x[26], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m24_m40, cospi_m40_p24, &x[22], &x[25], _r, cos_bit); |
| btf_16_adds_subs_avx2(&x[32], &x[35]); |
| btf_16_adds_subs_avx2(&x[33], &x[34]); |
| btf_16_adds_subs_avx2(&x[39], &x[36]); |
| btf_16_adds_subs_avx2(&x[38], &x[37]); |
| btf_16_adds_subs_avx2(&x[40], &x[43]); |
| btf_16_adds_subs_avx2(&x[41], &x[42]); |
| btf_16_adds_subs_avx2(&x[47], &x[44]); |
| btf_16_adds_subs_avx2(&x[46], &x[45]); |
| btf_16_adds_subs_avx2(&x[48], &x[51]); |
| btf_16_adds_subs_avx2(&x[49], &x[50]); |
| btf_16_adds_subs_avx2(&x[55], &x[52]); |
| btf_16_adds_subs_avx2(&x[54], &x[53]); |
| btf_16_adds_subs_avx2(&x[56], &x[59]); |
| btf_16_adds_subs_avx2(&x[57], &x[58]); |
| btf_16_adds_subs_avx2(&x[63], &x[60]); |
| btf_16_adds_subs_avx2(&x[62], &x[61]); |
| } |
| |
| static inline void idct64_stage6_high32_avx2(__m256i *x, const int32_t *cospi, |
| const __m256i _r, int8_t cos_bit) { |
| (void)cos_bit; |
| const __m256i cospi_m08_p56 = pair_set_w16_epi16(-cospi[8], cospi[56]); |
| const __m256i cospi_p56_p08 = pair_set_w16_epi16(cospi[56], cospi[8]); |
| const __m256i cospi_m56_m08 = pair_set_w16_epi16(-cospi[56], -cospi[8]); |
| const __m256i cospi_m40_p24 = pair_set_w16_epi16(-cospi[40], cospi[24]); |
| const __m256i cospi_p24_p40 = pair_set_w16_epi16(cospi[24], cospi[40]); |
| const __m256i cospi_m24_m40 = pair_set_w16_epi16(-cospi[24], -cospi[40]); |
| btf_16_w16_avx2(cospi_m08_p56, cospi_p56_p08, &x[34], &x[61], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m08_p56, cospi_p56_p08, &x[35], &x[60], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m56_m08, cospi_m08_p56, &x[36], &x[59], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m56_m08, cospi_m08_p56, &x[37], &x[58], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m40_p24, cospi_p24_p40, &x[42], &x[53], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m40_p24, cospi_p24_p40, &x[43], &x[52], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m24_m40, cospi_m40_p24, &x[44], &x[51], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m24_m40, cospi_m40_p24, &x[45], &x[50], _r, cos_bit); |
| } |
| |
| static inline void idct64_stage6_high48_avx2(__m256i *x, const int32_t *cospi, |
| const __m256i _r, int8_t cos_bit) { |
| btf_16_adds_subs_avx2(&x[16], &x[19]); |
| btf_16_adds_subs_avx2(&x[17], &x[18]); |
| btf_16_adds_subs_avx2(&x[23], &x[20]); |
| btf_16_adds_subs_avx2(&x[22], &x[21]); |
| btf_16_adds_subs_avx2(&x[24], &x[27]); |
| btf_16_adds_subs_avx2(&x[25], &x[26]); |
| btf_16_adds_subs_avx2(&x[31], &x[28]); |
| btf_16_adds_subs_avx2(&x[30], &x[29]); |
| idct64_stage6_high32_avx2(x, cospi, _r, cos_bit); |
| } |
| |
| static inline void idct64_stage7_high48_avx2(__m256i *x, const int32_t *cospi, |
| const __m256i _r, int8_t cos_bit) { |
| (void)cos_bit; |
| const __m256i cospi_m16_p48 = pair_set_w16_epi16(-cospi[16], cospi[48]); |
| const __m256i cospi_p48_p16 = pair_set_w16_epi16(cospi[48], cospi[16]); |
| const __m256i cospi_m48_m16 = pair_set_w16_epi16(-cospi[48], -cospi[16]); |
| btf_16_w16_avx2(cospi_m16_p48, cospi_p48_p16, &x[18], &x[29], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m16_p48, cospi_p48_p16, &x[19], &x[28], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m48_m16, cospi_m16_p48, &x[20], &x[27], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m48_m16, cospi_m16_p48, &x[21], &x[26], _r, cos_bit); |
| btf_16_adds_subs_avx2(&x[32], &x[39]); |
| btf_16_adds_subs_avx2(&x[33], &x[38]); |
| btf_16_adds_subs_avx2(&x[34], &x[37]); |
| btf_16_adds_subs_avx2(&x[35], &x[36]); |
| btf_16_adds_subs_avx2(&x[47], &x[40]); |
| btf_16_adds_subs_avx2(&x[46], &x[41]); |
| btf_16_adds_subs_avx2(&x[45], &x[42]); |
| btf_16_adds_subs_avx2(&x[44], &x[43]); |
| btf_16_adds_subs_avx2(&x[48], &x[55]); |
| btf_16_adds_subs_avx2(&x[49], &x[54]); |
| btf_16_adds_subs_avx2(&x[50], &x[53]); |
| btf_16_adds_subs_avx2(&x[51], &x[52]); |
| btf_16_adds_subs_avx2(&x[63], &x[56]); |
| btf_16_adds_subs_avx2(&x[62], &x[57]); |
| btf_16_adds_subs_avx2(&x[61], &x[58]); |
| btf_16_adds_subs_avx2(&x[60], &x[59]); |
| } |
| |
| static inline void idct64_stage8_high48_avx2(__m256i *x, const int32_t *cospi, |
| const __m256i _r, int8_t cos_bit) { |
| (void)cos_bit; |
| const __m256i cospi_m16_p48 = pair_set_w16_epi16(-cospi[16], cospi[48]); |
| const __m256i cospi_p48_p16 = pair_set_w16_epi16(cospi[48], cospi[16]); |
| const __m256i cospi_m48_m16 = pair_set_w16_epi16(-cospi[48], -cospi[16]); |
| btf_16_adds_subs_avx2(&x[16], &x[23]); |
| btf_16_adds_subs_avx2(&x[17], &x[22]); |
| btf_16_adds_subs_avx2(&x[18], &x[21]); |
| btf_16_adds_subs_avx2(&x[19], &x[20]); |
| btf_16_adds_subs_avx2(&x[31], &x[24]); |
| btf_16_adds_subs_avx2(&x[30], &x[25]); |
| btf_16_adds_subs_avx2(&x[29], &x[26]); |
| btf_16_adds_subs_avx2(&x[28], &x[27]); |
| btf_16_w16_avx2(cospi_m16_p48, cospi_p48_p16, &x[36], &x[59], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m16_p48, cospi_p48_p16, &x[37], &x[58], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m16_p48, cospi_p48_p16, &x[38], &x[57], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m16_p48, cospi_p48_p16, &x[39], &x[56], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m48_m16, cospi_m16_p48, &x[40], &x[55], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m48_m16, cospi_m16_p48, &x[41], &x[54], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m48_m16, cospi_m16_p48, &x[42], &x[53], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m48_m16, cospi_m16_p48, &x[43], &x[52], _r, cos_bit); |
| } |
| |
| static inline void idct64_stage9_avx2(__m256i *x, const int32_t *cospi, |
| const __m256i _r, int8_t cos_bit) { |
| (void)cos_bit; |
| const __m256i cospi_m32_p32 = pair_set_w16_epi16(-cospi[32], cospi[32]); |
| const __m256i cospi_p32_p32 = pair_set_w16_epi16(cospi[32], cospi[32]); |
| btf_16_adds_subs_avx2(&x[0], &x[15]); |
| btf_16_adds_subs_avx2(&x[1], &x[14]); |
| btf_16_adds_subs_avx2(&x[2], &x[13]); |
| btf_16_adds_subs_avx2(&x[3], &x[12]); |
| btf_16_adds_subs_avx2(&x[4], &x[11]); |
| btf_16_adds_subs_avx2(&x[5], &x[10]); |
| btf_16_adds_subs_avx2(&x[6], &x[9]); |
| btf_16_adds_subs_avx2(&x[7], &x[8]); |
| btf_16_w16_avx2(cospi_m32_p32, cospi_p32_p32, &x[20], &x[27], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m32_p32, cospi_p32_p32, &x[21], &x[26], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m32_p32, cospi_p32_p32, &x[22], &x[25], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m32_p32, cospi_p32_p32, &x[23], &x[24], _r, cos_bit); |
| btf_16_adds_subs_avx2(&x[32], &x[47]); |
| btf_16_adds_subs_avx2(&x[33], &x[46]); |
| btf_16_adds_subs_avx2(&x[34], &x[45]); |
| btf_16_adds_subs_avx2(&x[35], &x[44]); |
| btf_16_adds_subs_avx2(&x[36], &x[43]); |
| btf_16_adds_subs_avx2(&x[37], &x[42]); |
| btf_16_adds_subs_avx2(&x[38], &x[41]); |
| btf_16_adds_subs_avx2(&x[39], &x[40]); |
| btf_16_adds_subs_avx2(&x[63], &x[48]); |
| btf_16_adds_subs_avx2(&x[62], &x[49]); |
| btf_16_adds_subs_avx2(&x[61], &x[50]); |
| btf_16_adds_subs_avx2(&x[60], &x[51]); |
| btf_16_adds_subs_avx2(&x[59], &x[52]); |
| btf_16_adds_subs_avx2(&x[58], &x[53]); |
| btf_16_adds_subs_avx2(&x[57], &x[54]); |
| btf_16_adds_subs_avx2(&x[56], &x[55]); |
| } |
| |
| static inline void idct64_stage10_avx2(__m256i *x, const int32_t *cospi, |
| const __m256i _r, int8_t cos_bit) { |
| (void)cos_bit; |
| const __m256i cospi_m32_p32 = pair_set_w16_epi16(-cospi[32], cospi[32]); |
| const __m256i cospi_p32_p32 = pair_set_w16_epi16(cospi[32], cospi[32]); |
| btf_16_adds_subs_avx2(&x[0], &x[31]); |
| btf_16_adds_subs_avx2(&x[1], &x[30]); |
| btf_16_adds_subs_avx2(&x[2], &x[29]); |
| btf_16_adds_subs_avx2(&x[3], &x[28]); |
| btf_16_adds_subs_avx2(&x[4], &x[27]); |
| btf_16_adds_subs_avx2(&x[5], &x[26]); |
| btf_16_adds_subs_avx2(&x[6], &x[25]); |
| btf_16_adds_subs_avx2(&x[7], &x[24]); |
| btf_16_adds_subs_avx2(&x[8], &x[23]); |
| btf_16_adds_subs_avx2(&x[9], &x[22]); |
| btf_16_adds_subs_avx2(&x[10], &x[21]); |
| btf_16_adds_subs_avx2(&x[11], &x[20]); |
| btf_16_adds_subs_avx2(&x[12], &x[19]); |
| btf_16_adds_subs_avx2(&x[13], &x[18]); |
| btf_16_adds_subs_avx2(&x[14], &x[17]); |
| btf_16_adds_subs_avx2(&x[15], &x[16]); |
| btf_16_w16_avx2(cospi_m32_p32, cospi_p32_p32, &x[40], &x[55], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m32_p32, cospi_p32_p32, &x[41], &x[54], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m32_p32, cospi_p32_p32, &x[42], &x[53], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m32_p32, cospi_p32_p32, &x[43], &x[52], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m32_p32, cospi_p32_p32, &x[44], &x[51], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m32_p32, cospi_p32_p32, &x[45], &x[50], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m32_p32, cospi_p32_p32, &x[46], &x[49], _r, cos_bit); |
| btf_16_w16_avx2(cospi_m32_p32, cospi_p32_p32, &x[47], &x[48], _r, cos_bit); |
| } |
| |
| static inline void idct64_stage11_avx2(__m256i *output, __m256i *x) { |
| btf_16_adds_subs_out_avx2(&output[0], &output[63], x[0], x[63]); |
| btf_16_adds_subs_out_avx2(&output[1], &output[62], x[1], x[62]); |
| btf_16_adds_subs_out_avx2(&output[2], &output[61], x[2], x[61]); |
| btf_16_adds_subs_out_avx2(&output[3], &output[60], x[3], x[60]); |
| btf_16_adds_subs_out_avx2(&output[4], &output[59], x[4], x[59]); |
| btf_16_adds_subs_out_avx2(&output[5], &output[58], x[5], x[58]); |
| btf_16_adds_subs_out_avx2(&output[6], &output[57], x[6], x[57]); |
| btf_16_adds_subs_out_avx2(&output[7], &output[56], x[7], x[56]); |
| btf_16_adds_subs_out_avx2(&output[8], &output[55], x[8], x[55]); |
| btf_16_adds_subs_out_avx2(&output[9], &output[54], x[9], x[54]); |
| btf_16_adds_subs_out_avx2(&output[10], &output[53], x[10], x[53]); |
| btf_16_adds_subs_out_avx2(&output[11], &output[52], x[11], x[52]); |
| btf_16_adds_subs_out_avx2(&output[12], &output[51], x[12], x[51]); |
| btf_16_adds_subs_out_avx2(&output[13], &output[50], x[13], x[50]); |
| btf_16_adds_subs_out_avx2(&output[14], &output[49], x[14], x[49]); |
| btf_16_adds_subs_out_avx2(&output[15], &output[48], x[15], x[48]); |
| btf_16_adds_subs_out_avx2(&output[16], &output[47], x[16], x[47]); |
| btf_16_adds_subs_out_avx2(&output[17], &output[46], x[17], x[46]); |
| btf_16_adds_subs_out_avx2(&output[18], &output[45], x[18], x[45]); |
| btf_16_adds_subs_out_avx2(&output[19], &output[44], x[19], x[44]); |
| btf_16_adds_subs_out_avx2(&output[20], &output[43], x[20], x[43]); |
| btf_16_adds_subs_out_avx2(&output[21], &output[42], x[21], x[42]); |
| btf_16_adds_subs_out_avx2(&output[22], &output[41], x[22], x[41]); |
| btf_16_adds_subs_out_avx2(&output[23], &output[40], x[23], x[40]); |
| btf_16_adds_subs_out_avx2(&output[24], &output[39], x[24], x[39]); |
| btf_16_adds_subs_out_avx2(&output[25], &output[38], x[25], x[38]); |
| btf_16_adds_subs_out_avx2(&output[26], &output[37], x[26], x[37]); |
| btf_16_adds_subs_out_avx2(&output[27], &output[36], x[27], x[36]); |
| btf_16_adds_subs_out_avx2(&output[28], &output[35], x[28], x[35]); |
| btf_16_adds_subs_out_avx2(&output[29], &output[34], x[29], x[34]); |
| btf_16_adds_subs_out_avx2(&output[30], &output[33], x[30], x[33]); |
| btf_16_adds_subs_out_avx2(&output[31], &output[32], x[31], x[32]); |
| } |
| |
| static void idct64_low1_avx2(const __m256i *input, __m256i *output) { |
| const int32_t *cospi = cospi_arr(INV_COS_BIT); |
| |
| // stage 1 |
| __m256i x[32]; |
| x[0] = input[0]; |
| |
| // stage 2 |
| // stage 3 |
| // stage 4 |
| // stage 5 |
| // stage 6 |
| btf_16_w16_0_avx2(cospi[32], cospi[32], x[0], x[0], x[1]); |
| |
| // stage 7 |
| // stage 8 |
| // stage 9 |
| // stage 10 |
| // stage 11 |
| output[0] = x[0]; |
| output[63] = x[0]; |
| output[1] = x[1]; |
| output[62] = x[1]; |
| output[2] = x[1]; |
| output[61] = x[1]; |
| output[3] = x[0]; |
| output[60] = x[0]; |
| output[4] = x[0]; |
| output[59] = x[0]; |
| output[5] = x[1]; |
| output[58] = x[1]; |
| output[6] = x[1]; |
| output[57] = x[1]; |
| output[7] = x[0]; |
| output[56] = x[0]; |
| output[8] = x[0]; |
| output[55] = x[0]; |
| output[9] = x[1]; |
| output[54] = x[1]; |
| output[10] = x[1]; |
| output[53] = x[1]; |
| output[11] = x[0]; |
| output[52] = x[0]; |
| output[12] = x[0]; |
| output[51] = x[0]; |
| output[13] = x[1]; |
| output[50] = x[1]; |
| output[14] = x[1]; |
| output[49] = x[1]; |
| output[15] = x[0]; |
| output[48] = x[0]; |
| output[16] = x[0]; |
| output[47] = x[0]; |
| output[17] = x[1]; |
| output[46] = x[1]; |
| output[18] = x[1]; |
| output[45] = x[1]; |
| output[19] = x[0]; |
| output[44] = x[0]; |
| output[20] = x[0]; |
| output[43] = x[0]; |
| output[21] = x[1]; |
| output[42] = x[1]; |
| output[22] = x[1]; |
| output[41] = x[1]; |
| output[23] = x[0]; |
| output[40] = x[0]; |
| output[24] = x[0]; |
| output[39] = x[0]; |
| output[25] = x[1]; |
| output[38] = x[1]; |
| output[26] = x[1]; |
| output[37] = x[1]; |
| output[27] = x[0]; |
| output[36] = x[0]; |
| output[28] = x[0]; |
| output[35] = x[0]; |
| output[29] = x[1]; |
| output[34] = x[1]; |
| output[30] = x[1]; |
| output[33] = x[1]; |
| output[31] = x[0]; |
| output[32] = x[0]; |
| } |
| |
| static void idct64_low8_avx2(const __m256i *input, __m256i *output) { |
| const int32_t *cospi = cospi_arr(INV_COS_BIT); |
| const __m256i _r = _mm256_set1_epi32(1 << (INV_COS_BIT - 1)); |
| const __m256i cospi_m04_p60 = pair_set_w16_epi16(-cospi[4], cospi[60]); |
| const __m256i cospi_p60_p04 = pair_set_w16_epi16(cospi[60], cospi[4]); |
| const __m256i cospi_m36_p28 = pair_set_w16_epi16(-cospi[36], cospi[28]); |
| const __m256i cospi_m28_m36 = pair_set_w16_epi16(-cospi[28], -cospi[36]); |
| const __m256i cospi_m20_p44 = pair_set_w16_epi16(-cospi[20], cospi[44]); |
| const __m256i cospi_p44_p20 = pair_set_w16_epi16(cospi[44], cospi[20]); |
| const __m256i cospi_m52_p12 = pair_set_w16_epi16(-cospi[52], cospi[12]); |
| const __m256i cospi_m12_m52 = pair_set_w16_epi16(-cospi[12], -cospi[52]); |
| const __m256i cospi_m08_p56 = pair_set_w16_epi16(-cospi[8], cospi[56]); |
| const __m256i cospi_p56_p08 = pair_set_w16_epi16(cospi[56], cospi[8]); |
| const __m256i cospi_m40_p24 = pair_set_w16_epi16(-cospi[40], cospi[24]); |
| const __m256i cospi_m24_m40 = pair_set_w16_epi16(-cospi[24], -cospi[40]); |
| const __m256i cospi_p32_p32 = pair_set_w16_epi16(cospi[32], cospi[32]); |
| const __m256i cospi_m16_p48 = pair_set_w16_epi16(-cospi[16], cospi[48]); |
| const __m256i cospi_p48_p16 = pair_set_w16_epi16(cospi[48], cospi[16]); |
| const __m256i cospi_m32_p32 = pair_set_w16_epi16(-cospi[32], cospi[32]); |
| |
| // stage 1 |
| __m256i x[64]; |
| x[0] = input[0]; |
| x[8] = input[4]; |
| x[16] = input[2]; |
| x[24] = input[6]; |
| x[32] = input[1]; |
| x[40] = input[5]; |
| x[48] = input[3]; |
| x[56] = input[7]; |
| |
| // stage 2 |
| btf_16_w16_0_avx2(cospi[63], cospi[1], x[32], x[32], x[63]); |
| btf_16_w16_0_avx2(-cospi[57], cospi[7], x[56], x[39], x[56]); |
| btf_16_w16_0_avx2(cospi[59], cospi[5], x[40], x[40], x[55]); |
| btf_16_w16_0_avx2(-cospi[61], cospi[3], x[48], x[47], x[48]); |
| |
| // stage 3 |
| btf_16_w16_0_avx2(cospi[62], cospi[2], x[16], x[16], x[31]); |
| btf_16_w16_0_avx2(-cospi[58], cospi[6], x[24], x[23], x[24]); |
| x[33] = x[32]; |
| x[38] = x[39]; |
| x[41] = x[40]; |
| x[46] = x[47]; |
| x[49] = x[48]; |
| x[54] = x[55]; |
| x[57] = x[56]; |
| x[62] = x[63]; |
| |
| // stage 4 |
| btf_16_w16_0_avx2(cospi[60], cospi[4], x[8], x[8], x[15]); |
| x[17] = x[16]; |
| x[22] = x[23]; |
| x[25] = x[24]; |
| x[30] = x[31]; |
| btf_16_w16_avx2(cospi_m04_p60, cospi_p60_p04, &x[33], &x[62], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_m28_m36, cospi_m36_p28, &x[38], &x[57], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_m20_p44, cospi_p44_p20, &x[41], &x[54], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_m12_m52, cospi_m52_p12, &x[46], &x[49], _r, |
| INV_COS_BIT); |
| |
| // stage 5 |
| x[9] = x[8]; |
| x[14] = x[15]; |
| btf_16_w16_avx2(cospi_m08_p56, cospi_p56_p08, &x[17], &x[30], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_m24_m40, cospi_m40_p24, &x[22], &x[25], _r, |
| INV_COS_BIT); |
| x[35] = x[32]; |
| x[34] = x[33]; |
| x[36] = x[39]; |
| x[37] = x[38]; |
| x[43] = x[40]; |
| x[42] = x[41]; |
| x[44] = x[47]; |
| x[45] = x[46]; |
| x[51] = x[48]; |
| x[50] = x[49]; |
| x[52] = x[55]; |
| x[53] = x[54]; |
| x[59] = x[56]; |
| x[58] = x[57]; |
| x[60] = x[63]; |
| x[61] = x[62]; |
| |
| // stage 6 |
| btf_16_w16_0_avx2(cospi[32], cospi[32], x[0], x[0], x[1]); |
| btf_16_w16_avx2(cospi_m16_p48, cospi_p48_p16, &x[9], &x[14], _r, INV_COS_BIT); |
| x[19] = x[16]; |
| x[18] = x[17]; |
| x[20] = x[23]; |
| x[21] = x[22]; |
| x[27] = x[24]; |
| x[26] = x[25]; |
| x[28] = x[31]; |
| x[29] = x[30]; |
| idct64_stage6_high32_avx2(x, cospi, _r, INV_COS_BIT); |
| |
| // stage 7 |
| x[3] = x[0]; |
| x[2] = x[1]; |
| x[11] = x[8]; |
| x[10] = x[9]; |
| x[12] = x[15]; |
| x[13] = x[14]; |
| idct64_stage7_high48_avx2(x, cospi, _r, INV_COS_BIT); |
| |
| // stage 8 |
| x[7] = x[0]; |
| x[6] = x[1]; |
| x[5] = x[2]; |
| x[4] = x[3]; |
| btf_16_w16_avx2(cospi_m32_p32, cospi_p32_p32, &x[10], &x[13], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_m32_p32, cospi_p32_p32, &x[11], &x[12], _r, |
| INV_COS_BIT); |
| idct64_stage8_high48_avx2(x, cospi, _r, INV_COS_BIT); |
| |
| idct64_stage9_avx2(x, cospi, _r, INV_COS_BIT); |
| idct64_stage10_avx2(x, cospi, _r, INV_COS_BIT); |
| idct64_stage11_avx2(output, x); |
| } |
| |
| static void idct64_low16_avx2(const __m256i *input, __m256i *output) { |
| const int32_t *cospi = cospi_arr(INV_COS_BIT); |
| const __m256i _r = _mm256_set1_epi32(1 << (INV_COS_BIT - 1)); |
| |
| const __m256i cospi_p32_p32 = pair_set_w16_epi16(cospi[32], cospi[32]); |
| const __m256i cospi_m16_p48 = pair_set_w16_epi16(-cospi[16], cospi[48]); |
| const __m256i cospi_p48_p16 = pair_set_w16_epi16(cospi[48], cospi[16]); |
| const __m256i cospi_m48_m16 = pair_set_w16_epi16(-cospi[48], -cospi[16]); |
| const __m256i cospi_m32_p32 = pair_set_w16_epi16(-cospi[32], cospi[32]); |
| |
| // stage 1 |
| __m256i x[64]; |
| x[0] = input[0]; |
| x[4] = input[8]; |
| x[8] = input[4]; |
| x[12] = input[12]; |
| x[16] = input[2]; |
| x[20] = input[10]; |
| x[24] = input[6]; |
| x[28] = input[14]; |
| x[32] = input[1]; |
| x[36] = input[9]; |
| x[40] = input[5]; |
| x[44] = input[13]; |
| x[48] = input[3]; |
| x[52] = input[11]; |
| x[56] = input[7]; |
| x[60] = input[15]; |
| |
| // stage 2 |
| btf_16_w16_0_avx2(cospi[63], cospi[1], x[32], x[32], x[63]); |
| btf_16_w16_0_avx2(-cospi[49], cospi[15], x[60], x[35], x[60]); |
| btf_16_w16_0_avx2(cospi[55], cospi[9], x[36], x[36], x[59]); |
| btf_16_w16_0_avx2(-cospi[57], cospi[7], x[56], x[39], x[56]); |
| btf_16_w16_0_avx2(cospi[59], cospi[5], x[40], x[40], x[55]); |
| btf_16_w16_0_avx2(-cospi[53], cospi[11], x[52], x[43], x[52]); |
| btf_16_w16_0_avx2(cospi[51], cospi[13], x[44], x[44], x[51]); |
| btf_16_w16_0_avx2(-cospi[61], cospi[3], x[48], x[47], x[48]); |
| |
| // stage 3 |
| btf_16_w16_0_avx2(cospi[62], cospi[2], x[16], x[16], x[31]); |
| btf_16_w16_0_avx2(-cospi[50], cospi[14], x[28], x[19], x[28]); |
| btf_16_w16_0_avx2(cospi[54], cospi[10], x[20], x[20], x[27]); |
| btf_16_w16_0_avx2(-cospi[58], cospi[6], x[24], x[23], x[24]); |
| x[33] = x[32]; |
| x[34] = x[35]; |
| x[37] = x[36]; |
| x[38] = x[39]; |
| x[41] = x[40]; |
| x[42] = x[43]; |
| x[45] = x[44]; |
| x[46] = x[47]; |
| x[49] = x[48]; |
| x[50] = x[51]; |
| x[53] = x[52]; |
| x[54] = x[55]; |
| x[57] = x[56]; |
| x[58] = x[59]; |
| x[61] = x[60]; |
| x[62] = x[63]; |
| |
| // stage 4 |
| btf_16_w16_0_avx2(cospi[60], cospi[4], x[8], x[8], x[15]); |
| btf_16_w16_0_avx2(-cospi[52], cospi[12], x[12], x[11], x[12]); |
| x[17] = x[16]; |
| x[18] = x[19]; |
| x[21] = x[20]; |
| x[22] = x[23]; |
| x[25] = x[24]; |
| x[26] = x[27]; |
| x[29] = x[28]; |
| x[30] = x[31]; |
| idct64_stage4_high32_avx2(x, cospi, _r, INV_COS_BIT); |
| |
| // stage 5 |
| btf_16_w16_0_avx2(cospi[56], cospi[8], x[4], x[4], x[7]); |
| x[9] = x[8]; |
| x[10] = x[11]; |
| x[13] = x[12]; |
| x[14] = x[15]; |
| idct64_stage5_high48_avx2(x, cospi, _r, INV_COS_BIT); |
| |
| // stage 6 |
| btf_16_w16_0_avx2(cospi[32], cospi[32], x[0], x[0], x[1]); |
| x[5] = x[4]; |
| x[6] = x[7]; |
| btf_16_w16_avx2(cospi_m16_p48, cospi_p48_p16, &x[9], &x[14], _r, INV_COS_BIT); |
| btf_16_w16_avx2(cospi_m48_m16, cospi_m16_p48, &x[10], &x[13], _r, |
| INV_COS_BIT); |
| idct64_stage6_high48_avx2(x, cospi, _r, INV_COS_BIT); |
| |
| // stage 7 |
| x[3] = x[0]; |
| x[2] = x[1]; |
| btf_16_w16_avx2(cospi_m32_p32, cospi_p32_p32, &x[5], &x[6], _r, INV_COS_BIT); |
| btf_16_adds_subs_avx2(&x[8], &x[11]); |
| btf_16_adds_subs_avx2(&x[9], &x[10]); |
| btf_16_adds_subs_avx2(&x[15], &x[12]); |
| btf_16_adds_subs_avx2(&x[14], &x[13]); |
| idct64_stage7_high48_avx2(x, cospi, _r, INV_COS_BIT); |
| |
| // stage 8 |
| btf_16_adds_subs_avx2(&x[0], &x[7]); |
| btf_16_adds_subs_avx2(&x[1], &x[6]); |
| btf_16_adds_subs_avx2(&x[2], &x[5]); |
| btf_16_adds_subs_avx2(&x[3], &x[4]); |
| btf_16_w16_avx2(cospi_m32_p32, cospi_p32_p32, &x[10], &x[13], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_m32_p32, cospi_p32_p32, &x[11], &x[12], _r, |
| INV_COS_BIT); |
| idct64_stage8_high48_avx2(x, cospi, _r, INV_COS_BIT); |
| |
| idct64_stage9_avx2(x, cospi, _r, INV_COS_BIT); |
| idct64_stage10_avx2(x, cospi, _r, INV_COS_BIT); |
| idct64_stage11_avx2(output, x); |
| } |
| |
| static void idct64_low32_avx2(const __m256i *input, __m256i *output) { |
| const int32_t *cospi = cospi_arr(INV_COS_BIT); |
| const __m256i _r = _mm256_set1_epi32(1 << (INV_COS_BIT - 1)); |
| |
| const __m256i cospi_p32_p32 = pair_set_w16_epi16(cospi[32], cospi[32]); |
| const __m256i cospi_m16_p48 = pair_set_w16_epi16(-cospi[16], cospi[48]); |
| const __m256i cospi_p48_p16 = pair_set_w16_epi16(cospi[48], cospi[16]); |
| const __m256i cospi_m48_m16 = pair_set_w16_epi16(-cospi[48], -cospi[16]); |
| const __m256i cospi_m32_p32 = pair_set_w16_epi16(-cospi[32], cospi[32]); |
| |
| // stage 1 |
| __m256i x[64]; |
| x[0] = input[0]; |
| x[2] = input[16]; |
| x[4] = input[8]; |
| x[6] = input[24]; |
| x[8] = input[4]; |
| x[10] = input[20]; |
| x[12] = input[12]; |
| x[14] = input[28]; |
| x[16] = input[2]; |
| x[18] = input[18]; |
| x[20] = input[10]; |
| x[22] = input[26]; |
| x[24] = input[6]; |
| x[26] = input[22]; |
| x[28] = input[14]; |
| x[30] = input[30]; |
| x[32] = input[1]; |
| x[34] = input[17]; |
| x[36] = input[9]; |
| x[38] = input[25]; |
| x[40] = input[5]; |
| x[42] = input[21]; |
| x[44] = input[13]; |
| x[46] = input[29]; |
| x[48] = input[3]; |
| x[50] = input[19]; |
| x[52] = input[11]; |
| x[54] = input[27]; |
| x[56] = input[7]; |
| x[58] = input[23]; |
| x[60] = input[15]; |
| x[62] = input[31]; |
| |
| // stage 2 |
| btf_16_w16_0_avx2(cospi[63], cospi[1], x[32], x[32], x[63]); |
| btf_16_w16_0_avx2(-cospi[33], cospi[31], x[62], x[33], x[62]); |
| btf_16_w16_0_avx2(cospi[47], cospi[17], x[34], x[34], x[61]); |
| btf_16_w16_0_avx2(-cospi[49], cospi[15], x[60], x[35], x[60]); |
| btf_16_w16_0_avx2(cospi[55], cospi[9], x[36], x[36], x[59]); |
| btf_16_w16_0_avx2(-cospi[41], cospi[23], x[58], x[37], x[58]); |
| btf_16_w16_0_avx2(cospi[39], cospi[25], x[38], x[38], x[57]); |
| btf_16_w16_0_avx2(-cospi[57], cospi[7], x[56], x[39], x[56]); |
| btf_16_w16_0_avx2(cospi[59], cospi[5], x[40], x[40], x[55]); |
| btf_16_w16_0_avx2(-cospi[37], cospi[27], x[54], x[41], x[54]); |
| btf_16_w16_0_avx2(cospi[43], cospi[21], x[42], x[42], x[53]); |
| btf_16_w16_0_avx2(-cospi[53], cospi[11], x[52], x[43], x[52]); |
| btf_16_w16_0_avx2(cospi[51], cospi[13], x[44], x[44], x[51]); |
| btf_16_w16_0_avx2(-cospi[45], cospi[19], x[50], x[45], x[50]); |
| btf_16_w16_0_avx2(cospi[35], cospi[29], x[46], x[46], x[49]); |
| btf_16_w16_0_avx2(-cospi[61], cospi[3], x[48], x[47], x[48]); |
| |
| // stage 3 |
| btf_16_w16_0_avx2(cospi[62], cospi[2], x[16], x[16], x[31]); |
| btf_16_w16_0_avx2(-cospi[34], cospi[30], x[30], x[17], x[30]); |
| btf_16_w16_0_avx2(cospi[46], cospi[18], x[18], x[18], x[29]); |
| btf_16_w16_0_avx2(-cospi[50], cospi[14], x[28], x[19], x[28]); |
| btf_16_w16_0_avx2(cospi[54], cospi[10], x[20], x[20], x[27]); |
| btf_16_w16_0_avx2(-cospi[42], cospi[22], x[26], x[21], x[26]); |
| btf_16_w16_0_avx2(cospi[38], cospi[26], x[22], x[22], x[25]); |
| btf_16_w16_0_avx2(-cospi[58], cospi[6], x[24], x[23], x[24]); |
| btf_16_adds_subs_avx2(&x[32], &x[33]); |
| btf_16_adds_subs_avx2(&x[35], &x[34]); |
| btf_16_adds_subs_avx2(&x[36], &x[37]); |
| btf_16_adds_subs_avx2(&x[39], &x[38]); |
| btf_16_adds_subs_avx2(&x[40], &x[41]); |
| btf_16_adds_subs_avx2(&x[43], &x[42]); |
| btf_16_adds_subs_avx2(&x[44], &x[45]); |
| btf_16_adds_subs_avx2(&x[47], &x[46]); |
| btf_16_adds_subs_avx2(&x[48], &x[49]); |
| btf_16_adds_subs_avx2(&x[51], &x[50]); |
| btf_16_adds_subs_avx2(&x[52], &x[53]); |
| btf_16_adds_subs_avx2(&x[55], &x[54]); |
| btf_16_adds_subs_avx2(&x[56], &x[57]); |
| btf_16_adds_subs_avx2(&x[59], &x[58]); |
| btf_16_adds_subs_avx2(&x[60], &x[61]); |
| btf_16_adds_subs_avx2(&x[63], &x[62]); |
| |
| // stage 4 |
| btf_16_w16_0_avx2(cospi[60], cospi[4], x[8], x[8], x[15]); |
| btf_16_w16_0_avx2(-cospi[36], cospi[28], x[14], x[9], x[14]); |
| btf_16_w16_0_avx2(cospi[44], cospi[20], x[10], x[10], x[13]); |
| btf_16_w16_0_avx2(-cospi[52], cospi[12], x[12], x[11], x[12]); |
| btf_16_adds_subs_avx2(&x[16], &x[17]); |
| btf_16_adds_subs_avx2(&x[19], &x[18]); |
| btf_16_adds_subs_avx2(&x[20], &x[21]); |
| btf_16_adds_subs_avx2(&x[23], &x[22]); |
| btf_16_adds_subs_avx2(&x[24], &x[25]); |
| btf_16_adds_subs_avx2(&x[27], &x[26]); |
| btf_16_adds_subs_avx2(&x[28], &x[29]); |
| btf_16_adds_subs_avx2(&x[31], &x[30]); |
| idct64_stage4_high32_avx2(x, cospi, _r, INV_COS_BIT); |
| |
| // stage 5 |
| btf_16_w16_0_avx2(cospi[56], cospi[8], x[4], x[4], x[7]); |
| btf_16_w16_0_avx2(-cospi[40], cospi[24], x[6], x[5], x[6]); |
| btf_16_adds_subs_avx2(&x[8], &x[9]); |
| btf_16_adds_subs_avx2(&x[11], &x[10]); |
| btf_16_adds_subs_avx2(&x[12], &x[13]); |
| btf_16_adds_subs_avx2(&x[15], &x[14]); |
| idct64_stage5_high48_avx2(x, cospi, _r, INV_COS_BIT); |
| |
| // stage 6 |
| btf_16_w16_0_avx2(cospi[32], cospi[32], x[0], x[0], x[1]); |
| btf_16_w16_0_avx2(cospi[48], cospi[16], x[2], x[2], x[3]); |
| btf_16_adds_subs_avx2(&x[4], &x[5]); |
| btf_16_adds_subs_avx2(&x[7], &x[6]); |
| btf_16_w16_avx2(cospi_m16_p48, cospi_p48_p16, &x[9], &x[14], _r, INV_COS_BIT); |
| btf_16_w16_avx2(cospi_m48_m16, cospi_m16_p48, &x[10], &x[13], _r, |
| INV_COS_BIT); |
| idct64_stage6_high48_avx2(x, cospi, _r, INV_COS_BIT); |
| |
| // stage 7 |
| btf_16_adds_subs_avx2(&x[0], &x[3]); |
| btf_16_adds_subs_avx2(&x[1], &x[2]); |
| btf_16_w16_avx2(cospi_m32_p32, cospi_p32_p32, &x[5], &x[6], _r, INV_COS_BIT); |
| btf_16_adds_subs_avx2(&x[8], &x[11]); |
| btf_16_adds_subs_avx2(&x[9], &x[10]); |
| btf_16_adds_subs_avx2(&x[15], &x[12]); |
| btf_16_adds_subs_avx2(&x[14], &x[13]); |
| idct64_stage7_high48_avx2(x, cospi, _r, INV_COS_BIT); |
| |
| // stage 8 |
| btf_16_adds_subs_avx2(&x[0], &x[7]); |
| btf_16_adds_subs_avx2(&x[1], &x[6]); |
| btf_16_adds_subs_avx2(&x[2], &x[5]); |
| btf_16_adds_subs_avx2(&x[3], &x[4]); |
| btf_16_w16_avx2(cospi_m32_p32, cospi_p32_p32, &x[10], &x[13], _r, |
| INV_COS_BIT); |
| btf_16_w16_avx2(cospi_m32_p32, cospi_p32_p32, &x[11], &x[12], _r, |
| INV_COS_BIT); |
| idct64_stage8_high48_avx2(x, cospi, _r, INV_COS_BIT); |
| |
| // stage 9~11 |
| idct64_stage9_avx2(x, cospi, _r, INV_COS_BIT); |
| idct64_stage10_avx2(x, cospi, _r, INV_COS_BIT); |
| idct64_stage11_avx2(output, x); |
| } |
| |
| typedef void (*transform_1d_avx2)(const __m256i *input, __m256i *output); |
| |
| // 1D functions process 16 pixels at one time. |
| static const transform_1d_avx2 |
| lowbd_txfm_all_1d_zeros_w16_arr[TX_SIZES][ITX_TYPES_1D][4] = { |
| { |
| { NULL, NULL, NULL, NULL }, |
| { NULL, NULL, NULL, NULL }, |
| { NULL, NULL, NULL, NULL }, |
| }, |
| { { NULL, NULL, NULL, NULL }, |
| { NULL, NULL, NULL, NULL }, |
| { NULL, NULL, NULL, NULL } }, |
| { |
| { idct16_low1_avx2, idct16_low8_avx2, idct16_avx2, NULL }, |
| { iadst16_low1_avx2, iadst16_low8_avx2, iadst16_avx2, NULL }, |
| { NULL, NULL, NULL, NULL }, |
| }, |
| { { idct32_low1_avx2, idct32_low8_avx2, idct32_low16_avx2, idct32_avx2 }, |
| { NULL, NULL, NULL, NULL }, |
| { NULL, NULL, NULL, NULL } }, |
| { { idct64_low1_avx2, idct64_low8_avx2, idct64_low16_avx2, |
| idct64_low32_avx2 }, |
| { NULL, NULL, NULL, NULL }, |
| { NULL, NULL, NULL, NULL } } |
| }; |
| |
| // only process w >= 16 h >= 16 |
| static inline void lowbd_inv_txfm2d_add_no_identity_avx2( |
| const int32_t *input, uint8_t *output, int stride, TX_TYPE tx_type, |
| TX_SIZE tx_size, int eob) { |
| __m256i buf1[64 * 16]; |
| int eobx, eoby; |
| get_eobx_eoby_scan_default(&eobx, &eoby, tx_size, eob); |
| const int8_t *shift = av1_inv_txfm_shift_ls[tx_size]; |
| const int txw_idx = get_txw_idx(tx_size); |
| const int txh_idx = get_txh_idx(tx_size); |
| const int txfm_size_col = tx_size_wide[tx_size]; |
| const int txfm_size_row = tx_size_high[tx_size]; |
| const int buf_size_w_div16 = txfm_size_col >> 4; |
| const int buf_size_nonzero_w = ((eobx + 16) >> 4) << 4; |
| const int buf_size_nonzero_h_div16 = (eoby + 16) >> 4; |
| const int input_stride = AOMMIN(32, txfm_size_row); |
| const int rect_type = get_rect_tx_log_ratio(txfm_size_col, txfm_size_row); |
| |
| const int fun_idx_x = lowbd_txfm_all_1d_zeros_idx[eobx]; |
| const int fun_idx_y = lowbd_txfm_all_1d_zeros_idx[eoby]; |
| const transform_1d_avx2 row_txfm = |
| lowbd_txfm_all_1d_zeros_w16_arr[txw_idx][hitx_1d_tab[tx_type]][fun_idx_x]; |
| const transform_1d_avx2 col_txfm = |
| lowbd_txfm_all_1d_zeros_w16_arr[txh_idx][vitx_1d_tab[tx_type]][fun_idx_y]; |
| |
| assert(col_txfm != NULL); |
| assert(row_txfm != NULL); |
| int ud_flip, lr_flip; |
| get_flip_cfg(tx_type, &ud_flip, &lr_flip); |
| const __m256i scale0 = _mm256_set1_epi16(1 << (15 + shift[0])); |
| for (int i = 0; i < buf_size_nonzero_h_div16; i++) { |
| __m256i buf0[64]; |
| load_buffer_32bit_to_16bit_w16_avx2(input + 16 * i, input_stride, buf0, |
| buf_size_nonzero_w); |
| if (rect_type == 1 || rect_type == -1) { |
| round_shift_avx2(buf0, buf0, buf_size_nonzero_w); // rect special code |
| } |
| row_txfm(buf0, buf0); |
| for (int j = 0; j < txfm_size_col; ++j) { |
| buf0[j] = _mm256_mulhrs_epi16(buf0[j], scale0); |
| } |
| |
| __m256i *buf1_cur = buf1 + (i << 4); |
| if (lr_flip) { |
| for (int j = 0; j < buf_size_w_div16; ++j) { |
| __m256i temp[16]; |
| flip_buf_avx2(buf0 + 16 * j, temp, 16); |
| int offset = txfm_size_row * (buf_size_w_div16 - 1 - j); |
| transpose_16bit_16x16_avx2(temp, buf1_cur + offset); |
| } |
| } else { |
| for (int j = 0; j < buf_size_w_div16; ++j) { |
| transpose_16bit_16x16_avx2(buf0 + 16 * j, buf1_cur + txfm_size_row * j); |
| } |
| } |
| } |
| const __m256i scale1 = _mm256_set1_epi16(1 << (15 + shift[1])); |
| for (int i = 0; i < buf_size_w_div16; i++) { |
| __m256i *buf1_cur = buf1 + i * txfm_size_row; |
| col_txfm(buf1_cur, buf1_cur); |
| for (int j = 0; j < txfm_size_row; ++j) { |
| buf1_cur[j] = _mm256_mulhrs_epi16(buf1_cur[j], scale1); |
| } |
| } |
| for (int i = 0; i < buf_size_w_div16; i++) { |
| lowbd_write_buffer_16xn_avx2(buf1 + i * txfm_size_row, output + 16 * i, |
| stride, ud_flip, txfm_size_row); |
| } |
| } |
| |
| static inline void iidentity_row_16xn_avx2(__m256i *out, const int32_t *input, |
| int stride, int shift, int height, |
| int txw_idx, int rect_type) { |
| const int32_t *input_row = input; |
| const __m256i scale = _mm256_set1_epi16(NewSqrt2list[txw_idx]); |
| const __m256i _r = _mm256_set1_epi16((1 << (NewSqrt2Bits - 1)) + |
| (1 << (NewSqrt2Bits - shift - 1))); |
| const __m256i one = _mm256_set1_epi16(1); |
| const __m256i scale__r = _mm256_unpacklo_epi16(scale, _r); |
| if (rect_type != 1 && rect_type != -1) { |
| for (int i = 0; i < height; ++i) { |
| const __m256i src = load_32bit_to_16bit_w16_avx2(input_row); |
| input_row += stride; |
| __m256i lo = _mm256_unpacklo_epi16(src, one); |
| __m256i hi = _mm256_unpackhi_epi16(src, one); |
| lo = _mm256_madd_epi16(lo, scale__r); |
| hi = _mm256_madd_epi16(hi, scale__r); |
| lo = _mm256_srai_epi32(lo, NewSqrt2Bits - shift); |
| hi = _mm256_srai_epi32(hi, NewSqrt2Bits - shift); |
| out[i] = _mm256_packs_epi32(lo, hi); |
| } |
| } else { |
| const __m256i rect_scale = |
| _mm256_set1_epi16(NewInvSqrt2 << (15 - NewSqrt2Bits)); |
| for (int i = 0; i < height; ++i) { |
| __m256i src = load_32bit_to_16bit_w16_avx2(input_row); |
| src = _mm256_mulhrs_epi16(src, rect_scale); |
| input_row += stride; |
| __m256i lo = _mm256_unpacklo_epi16(src, one); |
| __m256i hi = _mm256_unpackhi_epi16(src, one); |
| lo = _mm256_madd_epi16(lo, scale__r); |
| hi = _mm256_madd_epi16(hi, scale__r); |
| lo = _mm256_srai_epi32(lo, NewSqrt2Bits - shift); |
| hi = _mm256_srai_epi32(hi, NewSqrt2Bits - shift); |
| out[i] = _mm256_packs_epi32(lo, hi); |
| } |
| } |
| } |
| |
| static inline void iidentity_col_16xn_avx2(uint8_t *output, int stride, |
| __m256i *buf, int shift, int height, |
| int txh_idx) { |
| const __m256i scale = _mm256_set1_epi16(NewSqrt2list[txh_idx]); |
| const __m256i scale__r = _mm256_set1_epi16(1 << (NewSqrt2Bits - 1)); |
| const __m256i shift__r = _mm256_set1_epi32(1 << (-shift - 1)); |
| const __m256i one = _mm256_set1_epi16(1); |
| const __m256i scale_coeff = _mm256_unpacklo_epi16(scale, scale__r); |
| for (int h = 0; h < height; ++h) { |
| __m256i lo = _mm256_unpacklo_epi16(buf[h], one); |
| __m256i hi = _mm256_unpackhi_epi16(buf[h], one); |
| lo = _mm256_madd_epi16(lo, scale_coeff); |
| hi = _mm256_madd_epi16(hi, scale_coeff); |
| lo = _mm256_srai_epi32(lo, NewSqrt2Bits); |
| hi = _mm256_srai_epi32(hi, NewSqrt2Bits); |
| lo = _mm256_add_epi32(lo, shift__r); |
| hi = _mm256_add_epi32(hi, shift__r); |
| lo = _mm256_srai_epi32(lo, -shift); |
| hi = _mm256_srai_epi32(hi, -shift); |
| const __m256i x = _mm256_packs_epi32(lo, hi); |
| write_recon_w16_avx2(x, output); |
| output += stride; |
| } |
| } |
| |
| static inline void lowbd_inv_txfm2d_add_idtx_avx2(const int32_t *input, |
| uint8_t *output, int stride, |
| TX_SIZE tx_size, |
| int32_t eob) { |
| (void)eob; |
| const int8_t *shift = av1_inv_txfm_shift_ls[tx_size]; |
| const int txw_idx = get_txw_idx(tx_size); |
| const int txh_idx = get_txh_idx(tx_size); |
| const int txfm_size_col = tx_size_wide[tx_size]; |
| const int txfm_size_row = tx_size_high[tx_size]; |
| const int col_max = AOMMIN(32, txfm_size_col); |
| const int row_max = AOMMIN(32, txfm_size_row); |
| const int input_stride = row_max; |
| const int rect_type = get_rect_tx_log_ratio(txfm_size_col, txfm_size_row); |
| __m256i buf[32]; |
| |
| for (int i = 0; i < (col_max >> 4); ++i) { |
| for (int j = 0; j < (row_max >> 4); j++) { |
| iidentity_row_16xn_avx2(buf, input + j * 16 + i * 16 * input_stride, |
| row_max, shift[0], 16, txw_idx, rect_type); |
| transpose_16bit_16x16_avx2(buf, buf); |
| iidentity_col_16xn_avx2(output + i * 16 + j * 16 * stride, stride, buf, |
| shift[1], 16, txh_idx); |
| } |
| } |
| } |
| |
| static inline void lowbd_inv_txfm2d_add_h_identity_avx2( |
| const int32_t *input, uint8_t *output, int stride, TX_TYPE tx_type, |
| TX_SIZE tx_size, int eob) { |
| int eobx, eoby; |
| get_eobx_eoby_scan_h_identity(&eobx, &eoby, tx_size, eob); |
| const int8_t *shift = av1_inv_txfm_shift_ls[tx_size]; |
| const int txw_idx = get_txw_idx(tx_size); |
| const int txh_idx = get_txh_idx(tx_size); |
| const int txfm_size_col = tx_size_wide[tx_size]; |
| const int txfm_size_row = tx_size_high[tx_size]; |
| const int txfm_size_row_notzero = AOMMIN(32, txfm_size_row); |
| const int input_stride = txfm_size_row_notzero; |
| const int buf_size_w_div16 = (eobx + 16) >> 4; |
| const int buf_size_h_div16 = (eoby + 16) >> 4; |
| const int rect_type = get_rect_tx_log_ratio(txfm_size_col, txfm_size_row); |
| |
| const int fun_idx_y = lowbd_txfm_all_1d_zeros_idx[eoby]; |
| const transform_1d_avx2 col_txfm = |
| lowbd_txfm_all_1d_zeros_w16_arr[txh_idx][vitx_1d_tab[tx_type]][fun_idx_y]; |
| |
| assert(col_txfm != NULL); |
| |
| int ud_flip, lr_flip; |
| get_flip_cfg(tx_type, &ud_flip, &lr_flip); |
| for (int i = 0; i < buf_size_w_div16; i++) { |
| __m256i buf0[64]; |
| for (int j = 0; j < buf_size_h_div16; j++) { |
| __m256i *buf0_cur = buf0 + j * 16; |
| const int32_t *input_cur = input + i * 16 * input_stride + j * 16; |
| iidentity_row_16xn_avx2(buf0_cur, input_cur, input_stride, shift[0], 16, |
| txw_idx, rect_type); |
| transpose_16bit_16x16_avx2(buf0_cur, buf0_cur); |
| } |
| col_txfm(buf0, buf0); |
| __m256i mshift = _mm256_set1_epi16(1 << (15 + shift[1])); |
| int k = ud_flip ? (txfm_size_row - 1) : 0; |
| const int step = ud_flip ? -1 : 1; |
| for (int j = 0; j < txfm_size_row; ++j, k += step) { |
| __m256i res = _mm256_mulhrs_epi16(buf0[k], mshift); |
| write_recon_w16_avx2(res, output + (i << 4) + j * stride); |
| } |
| } |
| } |
| |
| static inline void lowbd_inv_txfm2d_add_v_identity_avx2( |
| const int32_t *input, uint8_t *output, int stride, TX_TYPE tx_type, |
| TX_SIZE tx_size, int eob) { |
| __m256i buf1[64]; |
| int eobx, eoby; |
| get_eobx_eoby_scan_v_identity(&eobx, &eoby, tx_size, eob); |
| const int8_t *shift = av1_inv_txfm_shift_ls[tx_size]; |
| const int txw_idx = get_txw_idx(tx_size); |
| const int txh_idx = get_txh_idx(tx_size); |
| const int txfm_size_col = tx_size_wide[tx_size]; |
| const int txfm_size_row = tx_size_high[tx_size]; |
| const int buf_size_w_div16 = txfm_size_col >> 4; |
| const int buf_size_h_div16 = (eoby + 16) >> 4; |
| const int buf_size_nonzero_w = ((eobx + 8) >> 3) << 3; |
| const int input_stride = AOMMIN(32, txfm_size_row); |
| const int rect_type = get_rect_tx_log_ratio(txfm_size_col, txfm_size_row); |
| |
| const int fun_idx_x = lowbd_txfm_all_1d_zeros_idx[eobx]; |
| const transform_1d_avx2 row_txfm = |
| lowbd_txfm_all_1d_zeros_w16_arr[txw_idx][hitx_1d_tab[tx_type]][fun_idx_x]; |
| |
| assert(row_txfm != NULL); |
| |
| int ud_flip, lr_flip; |
| get_flip_cfg(tx_type, &ud_flip, &lr_flip); |
| for (int i = 0; i < buf_size_h_div16; i++) { |
| __m256i buf0[64]; |
| load_buffer_32bit_to_16bit_w16_avx2(input + i * 16, input_stride, buf0, |
| buf_size_nonzero_w); |
| if (rect_type == 1 || rect_type == -1) { |
| round_shift_avx2(buf0, buf0, buf_size_nonzero_w); // rect special code |
| } |
| row_txfm(buf0, buf0); |
| round_shift_16bit_w16_avx2(buf0, txfm_size_col, shift[0]); |
| __m256i *_buf1 = buf1; |
| if (lr_flip) { |
| for (int j = 0; j < buf_size_w_div16; ++j) { |
| __m256i temp[16]; |
| flip_buf_avx2(buf0 + 16 * j, temp, 16); |
| transpose_16bit_16x16_avx2(temp, |
| _buf1 + 16 * (buf_size_w_div16 - 1 - j)); |
| } |
| } else { |
| for (int j = 0; j < buf_size_w_div16; ++j) { |
| transpose_16bit_16x16_avx2(buf0 + 16 * j, _buf1 + 16 * j); |
| } |
| } |
| for (int j = 0; j < buf_size_w_div16; ++j) { |
| iidentity_col_16xn_avx2(output + i * 16 * stride + j * 16, stride, |
| buf1 + j * 16, shift[1], 16, txh_idx); |
| } |
| } |
| } |
| |
| static const transform_1d_ssse3 lowbd_txfm_all_1d_zeros_8x8_arr[2][2] = { |
| { av1_idct8_low1_ssse3, av1_idct8_sse2 }, |
| { av1_iadst8_low1_ssse3, av1_iadst8_sse2 } |
| }; |
| |
| static inline void load_buffer_avx2(const int32_t *in, int stride, |
| __m128i *out) { |
| const __m256i a = _mm256_load_si256((const __m256i *)in); |
| const __m256i b = _mm256_load_si256((const __m256i *)(in + stride * 1)); |
| const __m256i c = _mm256_load_si256((const __m256i *)(in + stride * 2)); |
| const __m256i d = _mm256_load_si256((const __m256i *)(in + stride * 3)); |
| const __m256i e = _mm256_load_si256((const __m256i *)(in + stride * 4)); |
| const __m256i f = _mm256_load_si256((const __m256i *)(in + stride * 5)); |
| const __m256i g = _mm256_load_si256((const __m256i *)(in + stride * 6)); |
| const __m256i h = _mm256_load_si256((const __m256i *)(in + stride * 7)); |
| |
| // a0 a1 a2 a3 b0 b1 b2 b3 a4 a5 a6 a7 b4 b5 b6 b7 |
| const __m256i ab_16bit = _mm256_packs_epi32(a, b); |
| // c0 c1 c2 c3 d0 d1 d2 d3 c4 c5 c6 c7 d4 d5 d6 d7 |
| const __m256i cd_16bit = _mm256_packs_epi32(c, d); |
| // e0 e1 e2 e3 f0 f1 f2 f3 e4 e5 e6 e7 f4 f5 f6 f7 |
| const __m256i ef_16bit = _mm256_packs_epi32(e, f); |
| // g0 g1 g2 g3 h0 h1 h2 h3 g4 g5 g6 g7 h4 h5 h6 h7 |
| const __m256i gh_16bit = _mm256_packs_epi32(g, h); |
| |
| // a0 a1 a2 a3 a4 a5 a6 a7 b0 b1 b2 b3 b4 b5 b6 b7 |
| const __m256i ab = _mm256_permute4x64_epi64(ab_16bit, 0xd8); |
| // c0 c1 c2 c3 c4 c5 c6 c7 d0 d1 d2 d3 d4 d5 d6 d7 |
| const __m256i cd = _mm256_permute4x64_epi64(cd_16bit, 0xd8); |
| // e0 e1 e2 e3 e4 e5 e6 e7 f0 f1 f2 f3 f4 f5 f6 f7 |
| const __m256i ef = _mm256_permute4x64_epi64(ef_16bit, 0xd8); |
| // g0 g1 g2 g3 g4 g5 g6 g7 h0 h1 h2 h3 h4 h5 h6 h7 |
| const __m256i gh = _mm256_permute4x64_epi64(gh_16bit, 0xd8); |
| |
| out[0] = _mm256_castsi256_si128(ab); |
| out[1] = _mm256_extractf128_si256(ab, 1); |
| out[2] = _mm256_castsi256_si128(cd); |
| out[3] = _mm256_extractf128_si256(cd, 1); |
| out[4] = _mm256_castsi256_si128(ef); |
| out[5] = _mm256_extractf128_si256(ef, 1); |
| out[6] = _mm256_castsi256_si128(gh); |
| out[7] = _mm256_extractf128_si256(gh, 1); |
| } |
| |
| static inline void round_and_transpose_avx2(const __m128i *const in, |
| __m128i *const out, int bit, |
| int *lr_flip) { |
| __m256i buf_temp[4]; |
| const __m256i scale = _mm256_set1_epi16(1 << (15 + bit)); |
| int j = *lr_flip ? 7 : 0; |
| const int step = *lr_flip ? -1 : 1; |
| |
| // 70 71 72 73 74 75 76 77 | 30 31 32 33 34 35 36 37 |
| buf_temp[0] = _mm256_inserti128_si256(_mm256_castsi128_si256(in[j]), |
| in[j + 4 * step], 1); |
| j += step; |
| // 60 61 62 63 64 65 66 67 | 20 21 22 23 24 25 26 27 |
| buf_temp[1] = _mm256_inserti128_si256(_mm256_castsi128_si256(in[j]), |
| in[j + 4 * step], 1); |
| j += step; |
| // 50 51 52 53 54 55 56 57 | 10 11 12 13 14 15 16 17 |
| buf_temp[2] = _mm256_inserti128_si256(_mm256_castsi128_si256(in[j]), |
| in[j + 4 * step], 1); |
| j += step; |
| // 40 41 42 43 44 45 46 47 | 00 01 02 03 04 05 06 07 |
| buf_temp[3] = _mm256_inserti128_si256(_mm256_castsi128_si256(in[j]), |
| in[j + 4 * step], 1); |
| |
| // 70 71 72 73 74 75 76 77 | 30 31 32 33 34 35 36 37 |
| buf_temp[0] = _mm256_mulhrs_epi16(buf_temp[0], scale); |
| // 60 61 62 63 64 65 66 67 | 20 21 22 23 24 25 26 27 |
| buf_temp[1] = _mm256_mulhrs_epi16(buf_temp[1], scale); |
| // 50 51 52 53 54 55 56 57 | 10 11 12 13 14 15 16 17 |
| buf_temp[2] = _mm256_mulhrs_epi16(buf_temp[2], scale); |
| // 40 41 42 43 44 45 46 47 | 00 01 02 03 04 05 06 07 |
| buf_temp[3] = _mm256_mulhrs_epi16(buf_temp[3], scale); |
| |
| // 70 60 71 61 72 62 73 63 | 30 20 31 21 32 22 33 23 |
| const __m256i unpcklo0 = _mm256_unpacklo_epi16(buf_temp[0], buf_temp[1]); |
| // 74 64 75 65 76 66 77 67 | 34 24 35 25 36 26 37 27 |
| const __m256i unpckhi0 = _mm256_unpackhi_epi16(buf_temp[0], buf_temp[1]); |
| // 50 40 51 41 52 42 53 43 | 10 00 11 01 12 02 13 03 |
| const __m256i unpcklo1 = _mm256_unpacklo_epi16(buf_temp[2], buf_temp[3]); |
| // 54 44 55 45 56 46 57 47 | 14 04 15 05 16 06 17 07 |
| const __m256i unpckhi1 = _mm256_unpackhi_epi16(buf_temp[2], buf_temp[3]); |
| |
| // 70 60 50 40 71 61 51 41 | 30 20 10 00 31 21 11 01 |
| const __m256i unpcklo00 = _mm256_unpacklo_epi32(unpcklo0, unpcklo1); |
| // 72 62 52 42 73 63 53 43 | 32 22 12 02 33 23 13 03 |
| const __m256i unpckhi00 = _mm256_unpackhi_epi32(unpcklo0, unpcklo1); |
| // 74 64 54 44 75 65 55 45 | 34 24 14 04 35 25 15 05 |
| const __m256i unpcklo01 = _mm256_unpacklo_epi32(unpckhi0, unpckhi1); |
| // 76 66 56 46 77 67 57 47 | 36 26 16 06 37 27 17 07 |
| const __m256i unpckhi01 = _mm256_unpackhi_epi32(unpckhi0, unpckhi1); |
| |
| // 70 60 50 40 30 20 10 00 | 71 61 51 41 31 21 11 01 |
| const __m256i reg_00 = _mm256_permute4x64_epi64(unpcklo00, 0xd8); |
| // 72 62 52 42 32 22 12 02 | 73 63 53 43 33 23 13 03 |
| const __m256i reg_01 = _mm256_permute4x64_epi64(unpckhi00, 0xd8); |
| // 74 64 54 44 34 24 14 04 | 75 65 55 45 35 25 15 05 |
| const __m256i reg_10 = _mm256_permute4x64_epi64(unpcklo01, 0xd8); |
| // 76 66 56 46 36 26 16 06 | 77 67 57 47 37 27 17 07 |
| const __m256i reg_11 = _mm256_permute4x64_epi64(unpckhi01, 0xd8); |
| |
| // 70 60 50 40 30 20 10 00 |
| out[0] = _mm256_castsi256_si128(reg_00); |
| // 71 61 51 41 31 21 11 01 |
| out[1] = _mm256_extracti128_si256(reg_00, 1); |
| // 72 62 52 42 32 22 12 02 |
| out[2] = _mm256_castsi256_si128(reg_01); |
| // 73 63 53 43 33 23 13 03 |
| out[3] = _mm256_extracti128_si256(reg_01, 1); |
| // 74 64 54 44 34 24 14 04 |
| out[4] = _mm256_castsi256_si128(reg_10); |
| // 75 65 55 45 35 25 15 05 |
| out[5] = _mm256_extracti128_si256(reg_10, 1); |
| // 76 66 56 46 36 26 16 06 |
| out[6] = _mm256_castsi256_si128(reg_11); |
| // 77 67 57 47 37 27 17 07 |
| out[7] = _mm256_extracti128_si256(reg_11, 1); |
| } |
| |
| static inline void round_shift_lowbd_write_buffer_avx2(__m128i *in, int bit, |
| uint8_t *output, |
| int stride, int flipud) { |
| __m256i in_256[4], v_256[4]; |
| int j = flipud ? 7 : 0; |
| const int step = flipud ? -1 : 1; |
| const __m256i scale = _mm256_set1_epi16(1 << (15 + bit)); |
| const __m256i zero = _mm256_setzero_si256(); |
| // in[0], in[1] |
| in_256[0] = |
| _mm256_inserti128_si256(_mm256_castsi128_si256(in[j]), in[j + step], 1); |
| j += 2 * step; |
| // in[2], in[3] |
| in_256[1] = |
| _mm256_inserti128_si256(_mm256_castsi128_si256(in[j]), in[j + step], 1); |
| j += 2 * step; |
| // in[4], in[5] |
| in_256[2] = |
| _mm256_inserti128_si256(_mm256_castsi128_si256(in[j]), in[j + step], 1); |
| j += 2 * step; |
| // in[6], in[7] |
| in_256[3] = |
| _mm256_inserti128_si256(_mm256_castsi128_si256(in[j]), in[j + step], 1); |
| |
| // i00 i01 i02 i03 i04 i05 i06 i07 i10 i11 i12 i13 i14 i15 i16 i17 |
| in_256[0] = _mm256_mulhrs_epi16(in_256[0], scale); |
| // i20 i21 i22 i23 i24 i25 i26 i27 i30 i31 i32 i33 i34 i35 i36 i37 |
| in_256[1] = _mm256_mulhrs_epi16(in_256[1], scale); |
| // i40 i41 i42 i43 i44 i45 i46 i47 i50 i51 i52 i53 i54 i55 i56 i57 |
| in_256[2] = _mm256_mulhrs_epi16(in_256[2], scale); |
| // i60 i61 i62 i63 i64 i65 i66 i67 i70 i71 i72 i73 i74 i75 i76 i77 |
| in_256[3] = _mm256_mulhrs_epi16(in_256[3], scale); |
| |
| const __m128i v0 = _mm_loadl_epi64((__m128i const *)(output)); |
| const __m128i v1 = _mm_loadl_epi64((__m128i const *)(output + stride)); |
| const __m128i v2 = _mm_loadl_epi64((__m128i const *)(output + 2 * stride)); |
| const __m128i v3 = _mm_loadl_epi64((__m128i const *)(output + 3 * stride)); |
| const __m128i v4 = _mm_loadl_epi64((__m128i const *)(output + 4 * stride)); |
| const __m128i v5 = _mm_loadl_epi64((__m128i const *)(output + 5 * stride)); |
| const __m128i v6 = _mm_loadl_epi64((__m128i const *)(output + 6 * stride)); |
| const __m128i v7 = _mm_loadl_epi64((__m128i const *)(output + 7 * stride)); |
| |
| v_256[0] = _mm256_inserti128_si256(_mm256_castsi128_si256(v0), v1, 1); |
| v_256[1] = _mm256_inserti128_si256(_mm256_castsi128_si256(v2), v3, 1); |
| v_256[2] = _mm256_inserti128_si256(_mm256_castsi128_si256(v4), v5, 1); |
| v_256[3] = _mm256_inserti128_si256(_mm256_castsi128_si256(v6), v7, 1); |
| |
| const __m256i unpcklo0 = _mm256_unpacklo_epi8(v_256[0], zero); |
| const __m256i unpcklo1 = _mm256_unpacklo_epi8(v_256[1], zero); |
| const __m256i unpcklo2 = _mm256_unpacklo_epi8(v_256[2], zero); |
| const __m256i unpcklo3 = _mm256_unpacklo_epi8(v_256[3], zero); |
| // 00 01 10 11 |
| const __m256i x0 = _mm256_adds_epi16(in_256[0], unpcklo0); |
| // 20 21 30 31 |
| const __m256i x1 = _mm256_adds_epi16(in_256[1], unpcklo1); |
| // 40 41 50 51 |
| const __m256i x2 = _mm256_adds_epi16(in_256[2], unpcklo2); |
| // 60 61 70 71 |
| const __m256i x3 = _mm256_adds_epi16(in_256[3], unpcklo3); |
| |
| // 00 01 20 21 10 11 30 31 |
| const __m256i res_0123 = _mm256_packus_epi16(x0, x1); |
| // 40 41 60 61 50 51 70 71 |
| const __m256i res_4567 = _mm256_packus_epi16(x2, x3); |
| |
| // 00 01 20 21 |
| const __m128i res_02 = _mm256_castsi256_si128(res_0123); |
| // 10 11 30 31 |
| const __m128i res_13 = _mm256_extracti128_si256(res_0123, 1); |
| // 40 41 60 61 |
| const __m128i res_46 = _mm256_castsi256_si128(res_4567); |
| // 50 51 70 71 |
| const __m128i res_57 = _mm256_extracti128_si256(res_4567, 1); |
| |
| // 00 01 |
| _mm_storel_epi64((__m128i *)(output), res_02); |
| // 10 11 |
| _mm_storel_epi64((__m128i *)(output + stride), res_13); |
| // 20 21 |
| _mm_storel_epi64((__m128i *)(output + 2 * stride), |
| _mm_unpackhi_epi64(res_02, res_02)); |
| // 30 31 |
| _mm_storel_epi64((__m128i *)(output + 3 * stride), |
| _mm_unpackhi_epi64(res_13, res_13)); |
| // 40 41 |
| _mm_storel_epi64((__m128i *)(output + 4 * stride), res_46); |
| // 50 51 |
| _mm_storel_epi64((__m128i *)(output + 5 * stride), res_57); |
| // 60 61 |
| _mm_storel_epi64((__m128i *)(output + 6 * stride), |
| _mm_unpackhi_epi64(res_46, res_46)); |
| // 70 71 |
| _mm_storel_epi64((__m128i *)(output + 7 * stride), |
| _mm_unpackhi_epi64(res_57, res_57)); |
| } |
| |
| // AVX2 implementation has the advantage when combined multiple operations |
| // together. |
| static inline void lowbd_inv_txfm2d_8x8_no_identity_avx2( |
| const int32_t *input, uint8_t *output, int stride, TX_TYPE tx_type, |
| TX_SIZE tx_size, int eob) { |
| __m128i buf1[8]; |
| const int input_stride = 8; |
| const int8_t *shift = av1_inv_txfm_shift_ls[tx_size]; |
| assert(hitx_1d_tab[tx_type] < 2); |
| assert(vitx_1d_tab[tx_type] < 2); |
| const transform_1d_ssse3 row_txfm = |
| lowbd_txfm_all_1d_zeros_8x8_arr[hitx_1d_tab[tx_type]][eob != 1]; |
| const transform_1d_ssse3 col_txfm = |
| lowbd_txfm_all_1d_zeros_8x8_arr[vitx_1d_tab[tx_type]][eob != 1]; |
| |
| assert(col_txfm != NULL); |
| assert(row_txfm != NULL); |
| int ud_flip, lr_flip; |
| get_flip_cfg(tx_type, &ud_flip, &lr_flip); |
| |
| __m128i buf0[8]; |
| __m128i *buf0_cur = buf0; |
| load_buffer_avx2(input, input_stride, buf0_cur); |
| row_txfm(buf0, buf0); |
| |
| assert(shift[0] < 0); |
| __m128i *_buf1 = buf1; |
| round_and_transpose_avx2(buf0, _buf1, shift[0], &lr_flip); |
| assert(shift[1] < 0); |
| col_txfm(buf1, buf1); |
| round_shift_lowbd_write_buffer_avx2(buf1, shift[1], output, stride, ud_flip); |
| } |
|