ref: dcdb013b55b557e215de8e72391f9876620bca25
dir: /vpx_dsp/x86/inv_txfm_sse2.c/
/* * Copyright (c) 2015 The WebM project authors. All Rights Reserved. * * Use of this source code is governed by a BSD-style license * that can be found in the LICENSE file in the root of the source * tree. An additional intellectual property rights grant can be found * in the file PATENTS. All contributing project authors may * be found in the AUTHORS file in the root of the source tree. */ #include "./vpx_dsp_rtcd.h" #include "vpx_dsp/x86/inv_txfm_sse2.h" #include "vpx_dsp/x86/transpose_sse2.h" #include "vpx_dsp/x86/txfm_common_sse2.h" static INLINE void transpose_16bit_4(__m128i *res) { const __m128i tr0_0 = _mm_unpacklo_epi16(res[0], res[1]); const __m128i tr0_1 = _mm_unpackhi_epi16(res[0], res[1]); res[0] = _mm_unpacklo_epi16(tr0_0, tr0_1); res[1] = _mm_unpackhi_epi16(tr0_0, tr0_1); } void vpx_idct4x4_16_add_sse2(const tran_low_t *input, uint8_t *dest, int stride) { const __m128i eight = _mm_set1_epi16(8); __m128i in[2]; // Rows in[0] = load_input_data8(input); in[1] = load_input_data8(input + 8); idct4_sse2(in); // Columns idct4_sse2(in); // Final round and shift in[0] = _mm_add_epi16(in[0], eight); in[1] = _mm_add_epi16(in[1], eight); in[0] = _mm_srai_epi16(in[0], 4); in[1] = _mm_srai_epi16(in[1], 4); recon_and_store4x4_sse2(in, dest, stride); } void vpx_idct4x4_1_add_sse2(const tran_low_t *input, uint8_t *dest, int stride) { const __m128i zero = _mm_setzero_si128(); int a; __m128i dc_value, d[2]; a = (int)dct_const_round_shift(input[0] * cospi_16_64); a = (int)dct_const_round_shift(a * cospi_16_64); a = ROUND_POWER_OF_TWO(a, 4); dc_value = _mm_set1_epi16(a); // Reconstruction and Store d[0] = _mm_cvtsi32_si128(*(const int *)(dest)); d[1] = _mm_cvtsi32_si128(*(const int *)(dest + stride * 3)); d[0] = _mm_unpacklo_epi32(d[0], _mm_cvtsi32_si128(*(const int *)(dest + stride))); d[1] = _mm_unpacklo_epi32( _mm_cvtsi32_si128(*(const int *)(dest + stride * 2)), d[1]); d[0] = _mm_unpacklo_epi8(d[0], zero); d[1] = _mm_unpacklo_epi8(d[1], zero); d[0] = _mm_add_epi16(d[0], dc_value); d[1] = _mm_add_epi16(d[1], dc_value); d[0] = _mm_packus_epi16(d[0], d[1]); *(int *)dest = _mm_cvtsi128_si32(d[0]); d[0] = _mm_srli_si128(d[0], 4); *(int *)(dest + stride) = _mm_cvtsi128_si32(d[0]); d[0] = _mm_srli_si128(d[0], 4); *(int *)(dest + stride * 2) = _mm_cvtsi128_si32(d[0]); d[0] = _mm_srli_si128(d[0], 4); *(int *)(dest + stride * 3) = _mm_cvtsi128_si32(d[0]); } void idct4_sse2(__m128i *in) { const __m128i k__cospi_p16_p16 = pair_set_epi16(cospi_16_64, cospi_16_64); const __m128i k__cospi_p16_m16 = pair_set_epi16(cospi_16_64, -cospi_16_64); const __m128i k__cospi_p24_m08 = pair_set_epi16(cospi_24_64, -cospi_8_64); const __m128i k__cospi_p08_p24 = pair_set_epi16(cospi_8_64, cospi_24_64); __m128i u[2]; transpose_16bit_4(in); // stage 1 u[0] = _mm_unpacklo_epi16(in[0], in[1]); u[1] = _mm_unpackhi_epi16(in[0], in[1]); u[0] = idct_calc_wraplow_sse2(k__cospi_p16_p16, k__cospi_p16_m16, u[0]); u[1] = idct_calc_wraplow_sse2(k__cospi_p08_p24, k__cospi_p24_m08, u[1]); // stage 2 in[0] = _mm_add_epi16(u[0], u[1]); in[1] = _mm_sub_epi16(u[0], u[1]); in[1] = _mm_shuffle_epi32(in[1], 0x4E); } void iadst4_sse2(__m128i *in) { const __m128i k__sinpi_p01_p04 = pair_set_epi16(sinpi_1_9, sinpi_4_9); const __m128i k__sinpi_p03_p02 = pair_set_epi16(sinpi_3_9, sinpi_2_9); const __m128i k__sinpi_p02_m01 = pair_set_epi16(sinpi_2_9, -sinpi_1_9); const __m128i k__sinpi_p03_m04 = pair_set_epi16(sinpi_3_9, -sinpi_4_9); const __m128i k__sinpi_p03_p03 = _mm_set1_epi16((int16_t)sinpi_3_9); const __m128i kZero = _mm_set1_epi16(0); const __m128i k__DCT_CONST_ROUNDING = _mm_set1_epi32(DCT_CONST_ROUNDING); __m128i u[8], v[8], in7; transpose_16bit_4(in); in7 = _mm_srli_si128(in[1], 8); in7 = _mm_add_epi16(in7, in[0]); in7 = _mm_sub_epi16(in7, in[1]); u[0] = _mm_unpacklo_epi16(in[0], in[1]); u[1] = _mm_unpackhi_epi16(in[0], in[1]); u[2] = _mm_unpacklo_epi16(in7, kZero); u[3] = _mm_unpackhi_epi16(in[0], kZero); v[0] = _mm_madd_epi16(u[0], k__sinpi_p01_p04); // s0 + s3 v[1] = _mm_madd_epi16(u[1], k__sinpi_p03_p02); // s2 + s5 v[2] = _mm_madd_epi16(u[2], k__sinpi_p03_p03); // x2 v[3] = _mm_madd_epi16(u[0], k__sinpi_p02_m01); // s1 - s4 v[4] = _mm_madd_epi16(u[1], k__sinpi_p03_m04); // s2 - s6 v[5] = _mm_madd_epi16(u[3], k__sinpi_p03_p03); // s2 u[0] = _mm_add_epi32(v[0], v[1]); u[1] = _mm_add_epi32(v[3], v[4]); u[2] = v[2]; u[3] = _mm_add_epi32(u[0], u[1]); u[4] = _mm_slli_epi32(v[5], 2); u[5] = _mm_add_epi32(u[3], v[5]); u[6] = _mm_sub_epi32(u[5], u[4]); v[0] = _mm_add_epi32(u[0], k__DCT_CONST_ROUNDING); v[1] = _mm_add_epi32(u[1], k__DCT_CONST_ROUNDING); v[2] = _mm_add_epi32(u[2], k__DCT_CONST_ROUNDING); v[3] = _mm_add_epi32(u[6], k__DCT_CONST_ROUNDING); u[0] = _mm_srai_epi32(v[0], DCT_CONST_BITS); u[1] = _mm_srai_epi32(v[1], DCT_CONST_BITS); u[2] = _mm_srai_epi32(v[2], DCT_CONST_BITS); u[3] = _mm_srai_epi32(v[3], DCT_CONST_BITS); in[0] = _mm_packs_epi32(u[0], u[1]); in[1] = _mm_packs_epi32(u[2], u[3]); } // Multiply elements by constants and add them together. static INLINE void multiplication_and_add( const __m128i *const in0, const __m128i *const in1, const __m128i *const in2, const __m128i *const in3, const __m128i *const cst0, const __m128i *const cst1, const __m128i *const cst2, const __m128i *const cst3, __m128i *const res0, __m128i *const res1, __m128i *const res2, __m128i *const res3) { const __m128i lo_0 = _mm_unpacklo_epi16(*in0, *in1); const __m128i hi_0 = _mm_unpackhi_epi16(*in0, *in1); const __m128i lo_1 = _mm_unpacklo_epi16(*in2, *in3); const __m128i hi_1 = _mm_unpackhi_epi16(*in2, *in3); *res0 = idct_calc_wraplow_sse2(lo_0, hi_0, *cst0); *res1 = idct_calc_wraplow_sse2(lo_0, hi_0, *cst1); *res2 = idct_calc_wraplow_sse2(lo_1, hi_1, *cst2); *res3 = idct_calc_wraplow_sse2(lo_1, hi_1, *cst3); } static INLINE void idct8(const __m128i *const in, __m128i *const out) { const __m128i cp_16_16 = pair_set_epi16(cospi_16_64, cospi_16_64); const __m128i cp_16_n16 = pair_set_epi16(cospi_16_64, -cospi_16_64); __m128i step1[8], step2[8]; // stage 1 { const __m128i cp_28_n4 = pair_set_epi16(cospi_28_64, -cospi_4_64); const __m128i cp_4_28 = pair_set_epi16(cospi_4_64, cospi_28_64); const __m128i cp_n20_12 = pair_set_epi16(-cospi_20_64, cospi_12_64); const __m128i cp_12_20 = pair_set_epi16(cospi_12_64, cospi_20_64); multiplication_and_add(&in[1], &in[7], &in[3], &in[5], &cp_28_n4, &cp_4_28, &cp_n20_12, &cp_12_20, &step1[4], &step1[7], &step1[5], &step1[6]); } // stage 2 { const __m128i cp_24_n8 = pair_set_epi16(cospi_24_64, -cospi_8_64); const __m128i cp_8_24 = pair_set_epi16(cospi_8_64, cospi_24_64); multiplication_and_add(&in[0], &in[4], &in[2], &in[6], &cp_16_16, &cp_16_n16, &cp_24_n8, &cp_8_24, &step2[0], &step2[1], &step2[2], &step2[3]); } step2[4] = _mm_add_epi16(step1[4], step1[5]); step2[5] = _mm_sub_epi16(step1[4], step1[5]); step2[6] = _mm_sub_epi16(step1[7], step1[6]); step2[7] = _mm_add_epi16(step1[7], step1[6]); // stage 3 step1[0] = _mm_add_epi16(step2[0], step2[3]); step1[1] = _mm_add_epi16(step2[1], step2[2]); step1[2] = _mm_sub_epi16(step2[1], step2[2]); step1[3] = _mm_sub_epi16(step2[0], step2[3]); multiplication_and_add_2(&step2[6], &step2[5], &cp_16_n16, &cp_16_16, &step1[5], &step1[6]); // stage 4 out[0] = _mm_add_epi16(step1[0], step2[7]); out[1] = _mm_add_epi16(step1[1], step1[6]); out[2] = _mm_add_epi16(step1[2], step1[5]); out[3] = _mm_add_epi16(step1[3], step2[4]); out[4] = _mm_sub_epi16(step1[3], step2[4]); out[5] = _mm_sub_epi16(step1[2], step1[5]); out[6] = _mm_sub_epi16(step1[1], step1[6]); out[7] = _mm_sub_epi16(step1[0], step2[7]); } void vpx_idct8x8_64_add_sse2(const tran_low_t *input, uint8_t *dest, int stride) { __m128i in[8]; int i; // Load input data. load_buffer_8x8(input, in); // 2-D for (i = 0; i < 2; i++) { idct8_sse2(in); } write_buffer_8x8(in, dest, stride); } static INLINE void recon_and_store_8_dual(uint8_t *const dest, const __m128i in_x, const int stride) { const __m128i zero = _mm_setzero_si128(); __m128i d0, d1; d0 = _mm_loadl_epi64((__m128i *)(dest + 0 * stride)); d1 = _mm_loadl_epi64((__m128i *)(dest + 1 * stride)); d0 = _mm_unpacklo_epi8(d0, zero); d1 = _mm_unpacklo_epi8(d1, zero); d0 = _mm_add_epi16(in_x, d0); d1 = _mm_add_epi16(in_x, d1); d0 = _mm_packus_epi16(d0, d1); _mm_storel_epi64((__m128i *)(dest + 0 * stride), d0); _mm_storeh_pi((__m64 *)(dest + 1 * stride), _mm_castsi128_ps(d0)); } void vpx_idct8x8_1_add_sse2(const tran_low_t *input, uint8_t *dest, int stride) { __m128i dc_value; tran_high_t a1; tran_low_t out = WRAPLOW(dct_const_round_shift(input[0] * cospi_16_64)); out = WRAPLOW(dct_const_round_shift(out * cospi_16_64)); a1 = ROUND_POWER_OF_TWO(out, 5); dc_value = _mm_set1_epi16(a1); recon_and_store_8_dual(dest, dc_value, stride); dest += 2 * stride; recon_and_store_8_dual(dest, dc_value, stride); dest += 2 * stride; recon_and_store_8_dual(dest, dc_value, stride); dest += 2 * stride; recon_and_store_8_dual(dest, dc_value, stride); } void idct8_sse2(__m128i *in) { // 8x8 Transpose is copied from vpx_fdct8x8_sse2() transpose_16bit_8x8(in, in); // 4-stage 1D idct8x8 idct8(in, in); } void iadst8_sse2(__m128i *in) { const __m128i k__cospi_p02_p30 = pair_set_epi16(cospi_2_64, cospi_30_64); const __m128i k__cospi_p30_m02 = pair_set_epi16(cospi_30_64, -cospi_2_64); const __m128i k__cospi_p10_p22 = pair_set_epi16(cospi_10_64, cospi_22_64); const __m128i k__cospi_p22_m10 = pair_set_epi16(cospi_22_64, -cospi_10_64); const __m128i k__cospi_p18_p14 = pair_set_epi16(cospi_18_64, cospi_14_64); const __m128i k__cospi_p14_m18 = pair_set_epi16(cospi_14_64, -cospi_18_64); const __m128i k__cospi_p26_p06 = pair_set_epi16(cospi_26_64, cospi_6_64); const __m128i k__cospi_p06_m26 = pair_set_epi16(cospi_6_64, -cospi_26_64); const __m128i k__cospi_p08_p24 = pair_set_epi16(cospi_8_64, cospi_24_64); const __m128i k__cospi_p24_m08 = pair_set_epi16(cospi_24_64, -cospi_8_64); const __m128i k__cospi_m24_p08 = pair_set_epi16(-cospi_24_64, cospi_8_64); const __m128i k__cospi_p16_m16 = pair_set_epi16(cospi_16_64, -cospi_16_64); const __m128i k__cospi_p16_p16 = _mm_set1_epi16((int16_t)cospi_16_64); const __m128i k__const_0 = _mm_set1_epi16(0); const __m128i k__DCT_CONST_ROUNDING = _mm_set1_epi32(DCT_CONST_ROUNDING); __m128i u0, u1, u2, u3, u4, u5, u6, u7, u8, u9, u10, u11, u12, u13, u14, u15; __m128i v0, v1, v2, v3, v4, v5, v6, v7, v8, v9, v10, v11, v12, v13, v14, v15; __m128i w0, w1, w2, w3, w4, w5, w6, w7, w8, w9, w10, w11, w12, w13, w14, w15; __m128i s0, s1, s2, s3, s4, s5, s6, s7; __m128i in0, in1, in2, in3, in4, in5, in6, in7; // transpose transpose_16bit_8x8(in, in); // properly aligned for butterfly input in0 = in[7]; in1 = in[0]; in2 = in[5]; in3 = in[2]; in4 = in[3]; in5 = in[4]; in6 = in[1]; in7 = in[6]; // column transformation // stage 1 // interleave and multiply/add into 32-bit integer s0 = _mm_unpacklo_epi16(in0, in1); s1 = _mm_unpackhi_epi16(in0, in1); s2 = _mm_unpacklo_epi16(in2, in3); s3 = _mm_unpackhi_epi16(in2, in3); s4 = _mm_unpacklo_epi16(in4, in5); s5 = _mm_unpackhi_epi16(in4, in5); s6 = _mm_unpacklo_epi16(in6, in7); s7 = _mm_unpackhi_epi16(in6, in7); u0 = _mm_madd_epi16(s0, k__cospi_p02_p30); u1 = _mm_madd_epi16(s1, k__cospi_p02_p30); u2 = _mm_madd_epi16(s0, k__cospi_p30_m02); u3 = _mm_madd_epi16(s1, k__cospi_p30_m02); u4 = _mm_madd_epi16(s2, k__cospi_p10_p22); u5 = _mm_madd_epi16(s3, k__cospi_p10_p22); u6 = _mm_madd_epi16(s2, k__cospi_p22_m10); u7 = _mm_madd_epi16(s3, k__cospi_p22_m10); u8 = _mm_madd_epi16(s4, k__cospi_p18_p14); u9 = _mm_madd_epi16(s5, k__cospi_p18_p14); u10 = _mm_madd_epi16(s4, k__cospi_p14_m18); u11 = _mm_madd_epi16(s5, k__cospi_p14_m18); u12 = _mm_madd_epi16(s6, k__cospi_p26_p06); u13 = _mm_madd_epi16(s7, k__cospi_p26_p06); u14 = _mm_madd_epi16(s6, k__cospi_p06_m26); u15 = _mm_madd_epi16(s7, k__cospi_p06_m26); // addition w0 = _mm_add_epi32(u0, u8); w1 = _mm_add_epi32(u1, u9); w2 = _mm_add_epi32(u2, u10); w3 = _mm_add_epi32(u3, u11); w4 = _mm_add_epi32(u4, u12); w5 = _mm_add_epi32(u5, u13); w6 = _mm_add_epi32(u6, u14); w7 = _mm_add_epi32(u7, u15); w8 = _mm_sub_epi32(u0, u8); w9 = _mm_sub_epi32(u1, u9); w10 = _mm_sub_epi32(u2, u10); w11 = _mm_sub_epi32(u3, u11); w12 = _mm_sub_epi32(u4, u12); w13 = _mm_sub_epi32(u5, u13); w14 = _mm_sub_epi32(u6, u14); w15 = _mm_sub_epi32(u7, u15); // shift and rounding v0 = _mm_add_epi32(w0, k__DCT_CONST_ROUNDING); v1 = _mm_add_epi32(w1, k__DCT_CONST_ROUNDING); v2 = _mm_add_epi32(w2, k__DCT_CONST_ROUNDING); v3 = _mm_add_epi32(w3, k__DCT_CONST_ROUNDING); v4 = _mm_add_epi32(w4, k__DCT_CONST_ROUNDING); v5 = _mm_add_epi32(w5, k__DCT_CONST_ROUNDING); v6 = _mm_add_epi32(w6, k__DCT_CONST_ROUNDING); v7 = _mm_add_epi32(w7, k__DCT_CONST_ROUNDING); v8 = _mm_add_epi32(w8, k__DCT_CONST_ROUNDING); v9 = _mm_add_epi32(w9, k__DCT_CONST_ROUNDING); v10 = _mm_add_epi32(w10, k__DCT_CONST_ROUNDING); v11 = _mm_add_epi32(w11, k__DCT_CONST_ROUNDING); v12 = _mm_add_epi32(w12, k__DCT_CONST_ROUNDING); v13 = _mm_add_epi32(w13, k__DCT_CONST_ROUNDING); v14 = _mm_add_epi32(w14, k__DCT_CONST_ROUNDING); v15 = _mm_add_epi32(w15, k__DCT_CONST_ROUNDING); u0 = _mm_srai_epi32(v0, DCT_CONST_BITS); u1 = _mm_srai_epi32(v1, DCT_CONST_BITS); u2 = _mm_srai_epi32(v2, DCT_CONST_BITS); u3 = _mm_srai_epi32(v3, DCT_CONST_BITS); u4 = _mm_srai_epi32(v4, DCT_CONST_BITS); u5 = _mm_srai_epi32(v5, DCT_CONST_BITS); u6 = _mm_srai_epi32(v6, DCT_CONST_BITS); u7 = _mm_srai_epi32(v7, DCT_CONST_BITS); u8 = _mm_srai_epi32(v8, DCT_CONST_BITS); u9 = _mm_srai_epi32(v9, DCT_CONST_BITS); u10 = _mm_srai_epi32(v10, DCT_CONST_BITS); u11 = _mm_srai_epi32(v11, DCT_CONST_BITS); u12 = _mm_srai_epi32(v12, DCT_CONST_BITS); u13 = _mm_srai_epi32(v13, DCT_CONST_BITS); u14 = _mm_srai_epi32(v14, DCT_CONST_BITS); u15 = _mm_srai_epi32(v15, DCT_CONST_BITS); // back to 16-bit and pack 8 integers into __m128i in[0] = _mm_packs_epi32(u0, u1); in[1] = _mm_packs_epi32(u2, u3); in[2] = _mm_packs_epi32(u4, u5); in[3] = _mm_packs_epi32(u6, u7); in[4] = _mm_packs_epi32(u8, u9); in[5] = _mm_packs_epi32(u10, u11); in[6] = _mm_packs_epi32(u12, u13); in[7] = _mm_packs_epi32(u14, u15); // stage 2 s0 = _mm_add_epi16(in[0], in[2]); s1 = _mm_add_epi16(in[1], in[3]); s2 = _mm_sub_epi16(in[0], in[2]); s3 = _mm_sub_epi16(in[1], in[3]); u0 = _mm_unpacklo_epi16(in[4], in[5]); u1 = _mm_unpackhi_epi16(in[4], in[5]); u2 = _mm_unpacklo_epi16(in[6], in[7]); u3 = _mm_unpackhi_epi16(in[6], in[7]); v0 = _mm_madd_epi16(u0, k__cospi_p08_p24); v1 = _mm_madd_epi16(u1, k__cospi_p08_p24); v2 = _mm_madd_epi16(u0, k__cospi_p24_m08); v3 = _mm_madd_epi16(u1, k__cospi_p24_m08); v4 = _mm_madd_epi16(u2, k__cospi_m24_p08); v5 = _mm_madd_epi16(u3, k__cospi_m24_p08); v6 = _mm_madd_epi16(u2, k__cospi_p08_p24); v7 = _mm_madd_epi16(u3, k__cospi_p08_p24); w0 = _mm_add_epi32(v0, v4); w1 = _mm_add_epi32(v1, v5); w2 = _mm_add_epi32(v2, v6); w3 = _mm_add_epi32(v3, v7); w4 = _mm_sub_epi32(v0, v4); w5 = _mm_sub_epi32(v1, v5); w6 = _mm_sub_epi32(v2, v6); w7 = _mm_sub_epi32(v3, v7); v0 = _mm_add_epi32(w0, k__DCT_CONST_ROUNDING); v1 = _mm_add_epi32(w1, k__DCT_CONST_ROUNDING); v2 = _mm_add_epi32(w2, k__DCT_CONST_ROUNDING); v3 = _mm_add_epi32(w3, k__DCT_CONST_ROUNDING); v4 = _mm_add_epi32(w4, k__DCT_CONST_ROUNDING); v5 = _mm_add_epi32(w5, k__DCT_CONST_ROUNDING); v6 = _mm_add_epi32(w6, k__DCT_CONST_ROUNDING); v7 = _mm_add_epi32(w7, k__DCT_CONST_ROUNDING); u0 = _mm_srai_epi32(v0, DCT_CONST_BITS); u1 = _mm_srai_epi32(v1, DCT_CONST_BITS); u2 = _mm_srai_epi32(v2, DCT_CONST_BITS); u3 = _mm_srai_epi32(v3, DCT_CONST_BITS); u4 = _mm_srai_epi32(v4, DCT_CONST_BITS); u5 = _mm_srai_epi32(v5, DCT_CONST_BITS); u6 = _mm_srai_epi32(v6, DCT_CONST_BITS); u7 = _mm_srai_epi32(v7, DCT_CONST_BITS); // back to 16-bit intergers s4 = _mm_packs_epi32(u0, u1); s5 = _mm_packs_epi32(u2, u3); s6 = _mm_packs_epi32(u4, u5); s7 = _mm_packs_epi32(u6, u7); // stage 3 u0 = _mm_unpacklo_epi16(s2, s3); u1 = _mm_unpackhi_epi16(s2, s3); u2 = _mm_unpacklo_epi16(s6, s7); u3 = _mm_unpackhi_epi16(s6, s7); s2 = idct_calc_wraplow_sse2(u0, u1, k__cospi_p16_p16); s3 = idct_calc_wraplow_sse2(u0, u1, k__cospi_p16_m16); s6 = idct_calc_wraplow_sse2(u2, u3, k__cospi_p16_p16); s7 = idct_calc_wraplow_sse2(u2, u3, k__cospi_p16_m16); in[0] = s0; in[1] = _mm_sub_epi16(k__const_0, s4); in[2] = s6; in[3] = _mm_sub_epi16(k__const_0, s2); in[4] = s3; in[5] = _mm_sub_epi16(k__const_0, s7); in[6] = s5; in[7] = _mm_sub_epi16(k__const_0, s1); } void vpx_idct8x8_12_add_sse2(const tran_low_t *input, uint8_t *dest, int stride) { const __m128i zero = _mm_setzero_si128(); const __m128i cp_16_16 = pair_set_epi16(cospi_16_64, cospi_16_64); const __m128i cp_16_n16 = pair_set_epi16(cospi_16_64, -cospi_16_64); __m128i in[8], step1[8], step2[8], tmp[4]; in[0] = load_input_data4(input + 0 * 8); in[1] = load_input_data4(input + 1 * 8); in[2] = load_input_data4(input + 2 * 8); in[3] = load_input_data4(input + 3 * 8); transpose_16bit_4x4(in, in); // in[0]: 00 10 20 30 01 11 21 31 // in[1]: 02 12 22 32 03 13 23 33 // stage 1 { const __m128i cp_28_n4 = pair_set_epi16(cospi_28_64, -cospi_4_64); const __m128i cp_4_28 = pair_set_epi16(cospi_4_64, cospi_28_64); const __m128i cp_n20_12 = pair_set_epi16(-cospi_20_64, cospi_12_64); const __m128i cp_12_20 = pair_set_epi16(cospi_12_64, cospi_20_64); const __m128i lo_1 = _mm_unpackhi_epi16(in[0], zero); const __m128i lo_3 = _mm_unpackhi_epi16(in[1], zero); step1[4] = idct_calc_wraplow_sse2(cp_28_n4, cp_4_28, lo_1); // step1 4&7 step1[5] = idct_calc_wraplow_sse2(cp_n20_12, cp_12_20, lo_3); // step1 5&6 } // stage 2 { const __m128i cp_24_n8 = pair_set_epi16(cospi_24_64, -cospi_8_64); const __m128i cp_8_24 = pair_set_epi16(cospi_8_64, cospi_24_64); const __m128i lo_0 = _mm_unpacklo_epi16(in[0], zero); const __m128i lo_2 = _mm_unpacklo_epi16(in[1], zero); step2[0] = idct_calc_wraplow_sse2(cp_16_16, cp_16_n16, lo_0); // step2 0&1 step2[2] = idct_calc_wraplow_sse2(cp_8_24, cp_24_n8, lo_2); // step2 3&2 step2[4] = _mm_add_epi16(step1[4], step1[5]); // step2 4&7 step2[5] = _mm_sub_epi16(step1[4], step1[5]); // step2 5&6 step2[6] = _mm_unpackhi_epi64(step2[5], zero); // step2 6 } // stage 3 { const __m128i lo_65 = _mm_unpacklo_epi16(step2[6], step2[5]); tmp[0] = _mm_add_epi16(step2[0], step2[2]); // step1 0&1 tmp[1] = _mm_sub_epi16(step2[0], step2[2]); // step1 3&2 step1[2] = _mm_unpackhi_epi64(tmp[1], tmp[0]); // step1 2&1 step1[3] = _mm_unpacklo_epi64(tmp[1], tmp[0]); // step1 3&0 step1[5] = idct_calc_wraplow_sse2(cp_16_n16, cp_16_16, lo_65); // step1 5&6 } // stage 4 tmp[0] = _mm_add_epi16(step1[3], step2[4]); // output 3&0 tmp[1] = _mm_add_epi16(step1[2], step1[5]); // output 2&1 tmp[2] = _mm_sub_epi16(step1[3], step2[4]); // output 4&7 tmp[3] = _mm_sub_epi16(step1[2], step1[5]); // output 5&6 idct8x8_12_transpose_16bit_4x8(tmp, in); in[4] = in[5] = in[6] = in[7] = zero; idct8(in, in); write_buffer_8x8(in, dest, stride); } #define IDCT16_10 \ /* Stage2 */ \ multiplication_and_add(&in[1], &zero, &zero, &in[3], &stg2_0, &stg2_1, \ &stg2_6, &stg2_7, &stp1_8_0, &stp1_15, &stp1_11, \ &stp1_12_0); \ \ /* Stage3 */ \ multiplication_and_add_2(&in[2], &zero, &stg3_0, &stg3_1, &stp2_4, &stp2_7); \ \ stp1_9 = stp1_8_0; \ stp1_10 = stp1_11; \ stp1_13 = stp1_12_0; \ stp1_14 = stp1_15; \ \ /* Stage4 */ \ multiplication_and_add_2(&in[0], &zero, &stg4_0, &stg4_1, &stp1_0, &stp1_1); \ stp2_5 = stp2_4; \ stp2_6 = stp2_7; \ \ multiplication_and_add(&stp1_9, &stp1_14, &stp1_10, &stp1_13, &stg4_4, \ &stg4_5, &stg4_6, &stg4_7, &stp2_9, &stp2_14, \ &stp2_10, &stp2_13); \ \ /* Stage5 */ \ stp1_2 = stp1_1; \ stp1_3 = stp1_0; \ multiplication_and_add_2(&stp2_6, &stp2_5, &stg4_1, &stg4_0, &stp1_5, \ &stp1_6); \ \ stp1_8 = _mm_add_epi16(stp1_8_0, stp1_11); \ stp1_9 = _mm_add_epi16(stp2_9, stp2_10); \ stp1_10 = _mm_sub_epi16(stp2_9, stp2_10); \ stp1_11 = _mm_sub_epi16(stp1_8_0, stp1_11); \ \ stp1_12 = _mm_sub_epi16(stp1_15, stp1_12_0); \ stp1_13 = _mm_sub_epi16(stp2_14, stp2_13); \ stp1_14 = _mm_add_epi16(stp2_14, stp2_13); \ stp1_15 = _mm_add_epi16(stp1_15, stp1_12_0); \ \ /* Stage6 */ \ stp2_0 = _mm_add_epi16(stp1_0, stp2_7); \ stp2_1 = _mm_add_epi16(stp1_1, stp1_6); \ stp2_2 = _mm_add_epi16(stp1_2, stp1_5); \ stp2_3 = _mm_add_epi16(stp1_3, stp2_4); \ stp2_4 = _mm_sub_epi16(stp1_3, stp2_4); \ stp2_5 = _mm_sub_epi16(stp1_2, stp1_5); \ stp2_6 = _mm_sub_epi16(stp1_1, stp1_6); \ stp2_7 = _mm_sub_epi16(stp1_0, stp2_7); \ \ multiplication_and_add(&stp1_10, &stp1_13, &stp1_11, &stp1_12, &stg6_0, \ &stg4_0, &stg6_0, &stg4_0, &stp2_10, &stp2_13, \ &stp2_11, &stp2_12); static INLINE void idct16_8col(__m128i *const in) { const __m128i k__cospi_p16_p16 = _mm_set1_epi16((int16_t)cospi_16_64); const __m128i k__cospi_m16_p16 = pair_set_epi16(-cospi_16_64, cospi_16_64); __m128i s[16], t[16]; // stage 2 { const __m128i k__cospi_p30_m02 = pair_set_epi16(cospi_30_64, -cospi_2_64); const __m128i k__cospi_p02_p30 = pair_set_epi16(cospi_2_64, cospi_30_64); const __m128i k__cospi_p14_m18 = pair_set_epi16(cospi_14_64, -cospi_18_64); const __m128i k__cospi_p18_p14 = pair_set_epi16(cospi_18_64, cospi_14_64); multiplication_and_add(&in[1], &in[15], &in[9], &in[7], &k__cospi_p30_m02, &k__cospi_p02_p30, &k__cospi_p14_m18, &k__cospi_p18_p14, &s[8], &s[15], &s[9], &s[14]); } { const __m128i k__cospi_p22_m10 = pair_set_epi16(cospi_22_64, -cospi_10_64); const __m128i k__cospi_p10_p22 = pair_set_epi16(cospi_10_64, cospi_22_64); const __m128i k__cospi_p06_m26 = pair_set_epi16(cospi_6_64, -cospi_26_64); const __m128i k__cospi_p26_p06 = pair_set_epi16(cospi_26_64, cospi_6_64); multiplication_and_add(&in[5], &in[11], &in[13], &in[3], &k__cospi_p22_m10, &k__cospi_p10_p22, &k__cospi_p06_m26, &k__cospi_p26_p06, &s[10], &s[13], &s[11], &s[12]); } // stage 3 { const __m128i k__cospi_p28_m04 = pair_set_epi16(cospi_28_64, -cospi_4_64); const __m128i k__cospi_p04_p28 = pair_set_epi16(cospi_4_64, cospi_28_64); const __m128i k__cospi_p12_m20 = pair_set_epi16(cospi_12_64, -cospi_20_64); const __m128i k__cospi_p20_p12 = pair_set_epi16(cospi_20_64, cospi_12_64); multiplication_and_add(&in[2], &in[14], &in[10], &in[6], &k__cospi_p28_m04, &k__cospi_p04_p28, &k__cospi_p12_m20, &k__cospi_p20_p12, &t[4], &t[7], &t[5], &t[6]); } t[8] = _mm_add_epi16(s[8], s[9]); t[9] = _mm_sub_epi16(s[8], s[9]); t[10] = _mm_sub_epi16(s[11], s[10]); t[11] = _mm_add_epi16(s[10], s[11]); t[12] = _mm_add_epi16(s[12], s[13]); t[13] = _mm_sub_epi16(s[12], s[13]); t[14] = _mm_sub_epi16(s[15], s[14]); t[15] = _mm_add_epi16(s[14], s[15]); // stage 4 { const __m128i k__cospi_p24_m08 = pair_set_epi16(cospi_24_64, -cospi_8_64); const __m128i k__cospi_p16_m16 = pair_set_epi16(cospi_16_64, -cospi_16_64); const __m128i k__cospi_p08_p24 = pair_set_epi16(cospi_8_64, cospi_24_64); multiplication_and_add(&in[0], &in[8], &in[4], &in[12], &k__cospi_p16_p16, &k__cospi_p16_m16, &k__cospi_p24_m08, &k__cospi_p08_p24, &s[0], &s[1], &s[2], &s[3]); } s[5] = _mm_sub_epi16(t[4], t[5]); t[4] = _mm_add_epi16(t[4], t[5]); s[6] = _mm_sub_epi16(t[7], t[6]); t[7] = _mm_add_epi16(t[6], t[7]); s[8] = t[8]; { const __m128i k__cospi_m08_p24 = pair_set_epi16(-cospi_8_64, cospi_24_64); const __m128i k__cospi_p24_p08 = pair_set_epi16(cospi_24_64, cospi_8_64); const __m128i k__cospi_m24_m08 = pair_set_epi16(-cospi_24_64, -cospi_8_64); multiplication_and_add(&t[9], &t[14], &t[10], &t[13], &k__cospi_m08_p24, &k__cospi_p24_p08, &k__cospi_m24_m08, &k__cospi_m08_p24, &s[9], &s[14], &s[10], &s[13]); } s[11] = t[11]; s[12] = t[12]; s[15] = t[15]; // stage 5 t[0] = _mm_add_epi16(s[0], s[3]); t[1] = _mm_add_epi16(s[1], s[2]); t[2] = _mm_sub_epi16(s[1], s[2]); t[3] = _mm_sub_epi16(s[0], s[3]); multiplication_and_add_2(&s[5], &s[6], &k__cospi_m16_p16, &k__cospi_p16_p16, &t[5], &t[6]); t[8] = _mm_add_epi16(s[8], s[11]); t[9] = _mm_add_epi16(s[9], s[10]); t[10] = _mm_sub_epi16(s[9], s[10]); t[11] = _mm_sub_epi16(s[8], s[11]); t[12] = _mm_sub_epi16(s[15], s[12]); t[13] = _mm_sub_epi16(s[14], s[13]); t[14] = _mm_add_epi16(s[13], s[14]); t[15] = _mm_add_epi16(s[12], s[15]); // stage 6 s[0] = _mm_add_epi16(t[0], t[7]); s[1] = _mm_add_epi16(t[1], t[6]); s[2] = _mm_add_epi16(t[2], t[5]); s[3] = _mm_add_epi16(t[3], t[4]); s[4] = _mm_sub_epi16(t[3], t[4]); s[5] = _mm_sub_epi16(t[2], t[5]); s[6] = _mm_sub_epi16(t[1], t[6]); s[7] = _mm_sub_epi16(t[0], t[7]); multiplication_and_add(&t[10], &t[13], &t[11], &t[12], &k__cospi_m16_p16, &k__cospi_p16_p16, &k__cospi_m16_p16, &k__cospi_p16_p16, &s[10], &s[13], &s[11], &s[12]); // stage 7 in[0] = _mm_add_epi16(s[0], t[15]); in[1] = _mm_add_epi16(s[1], t[14]); in[2] = _mm_add_epi16(s[2], s[13]); in[3] = _mm_add_epi16(s[3], s[12]); in[4] = _mm_add_epi16(s[4], s[11]); in[5] = _mm_add_epi16(s[5], s[10]); in[6] = _mm_add_epi16(s[6], t[9]); in[7] = _mm_add_epi16(s[7], t[8]); in[8] = _mm_sub_epi16(s[7], t[8]); in[9] = _mm_sub_epi16(s[6], t[9]); in[10] = _mm_sub_epi16(s[5], s[10]); in[11] = _mm_sub_epi16(s[4], s[11]); in[12] = _mm_sub_epi16(s[3], s[12]); in[13] = _mm_sub_epi16(s[2], s[13]); in[14] = _mm_sub_epi16(s[1], t[14]); in[15] = _mm_sub_epi16(s[0], t[15]); } static INLINE void idct16_load8x8(const tran_low_t *const input, __m128i *const in) { in[0] = load_input_data8(input); in[1] = load_input_data8(input + 8 * 2); in[2] = load_input_data8(input + 8 * 4); in[3] = load_input_data8(input + 8 * 6); in[4] = load_input_data8(input + 8 * 8); in[5] = load_input_data8(input + 8 * 10); in[6] = load_input_data8(input + 8 * 12); in[7] = load_input_data8(input + 8 * 14); } void vpx_idct16x16_256_add_sse2(const tran_low_t *input, uint8_t *dest, int stride) { __m128i l[16], r[16], out[16], *in; int i; in = l; for (i = 0; i < 2; i++) { idct16_load8x8(input, in); transpose_16bit_8x8(in, in); idct16_load8x8(input + 8, in + 8); transpose_16bit_8x8(in + 8, in + 8); idct16_8col(in); in = r; input += 128; } for (i = 0; i < 2; i++) { int j; transpose_16bit_8x8(l + i * 8, out); transpose_16bit_8x8(r + i * 8, out + 8); idct16_8col(out); // Final rounding and shift for (j = 0; j < 16; ++j) { const __m128i final_rounding = _mm_set1_epi16(1 << 5); out[j] = _mm_adds_epi16(out[j], final_rounding); out[j] = _mm_srai_epi16(out[j], 6); recon_and_store(dest + j * stride, out[j]); } dest += 8; } } static INLINE void recon_and_store_16(uint8_t *const dest, const __m128i in_x) { const __m128i zero = _mm_setzero_si128(); __m128i d0, d1; d0 = _mm_load_si128((__m128i *)(dest)); d1 = _mm_unpackhi_epi8(d0, zero); d0 = _mm_unpacklo_epi8(d0, zero); d0 = _mm_add_epi16(in_x, d0); d1 = _mm_add_epi16(in_x, d1); d0 = _mm_packus_epi16(d0, d1); _mm_store_si128((__m128i *)(dest), d0); } void vpx_idct16x16_1_add_sse2(const tran_low_t *input, uint8_t *dest, int stride) { __m128i dc_value; int i; tran_high_t a1; tran_low_t out = WRAPLOW(dct_const_round_shift(input[0] * cospi_16_64)); out = WRAPLOW(dct_const_round_shift(out * cospi_16_64)); a1 = ROUND_POWER_OF_TWO(out, 6); dc_value = _mm_set1_epi16(a1); for (i = 0; i < 16; ++i) { recon_and_store_16(dest, dc_value); dest += stride; } } static void iadst16_8col(__m128i *in) { // perform 16x16 1-D ADST for 8 columns __m128i s[16], x[16], u[32], v[32]; const __m128i k__cospi_p01_p31 = pair_set_epi16(cospi_1_64, cospi_31_64); const __m128i k__cospi_p31_m01 = pair_set_epi16(cospi_31_64, -cospi_1_64); const __m128i k__cospi_p05_p27 = pair_set_epi16(cospi_5_64, cospi_27_64); const __m128i k__cospi_p27_m05 = pair_set_epi16(cospi_27_64, -cospi_5_64); const __m128i k__cospi_p09_p23 = pair_set_epi16(cospi_9_64, cospi_23_64); const __m128i k__cospi_p23_m09 = pair_set_epi16(cospi_23_64, -cospi_9_64); const __m128i k__cospi_p13_p19 = pair_set_epi16(cospi_13_64, cospi_19_64); const __m128i k__cospi_p19_m13 = pair_set_epi16(cospi_19_64, -cospi_13_64); const __m128i k__cospi_p17_p15 = pair_set_epi16(cospi_17_64, cospi_15_64); const __m128i k__cospi_p15_m17 = pair_set_epi16(cospi_15_64, -cospi_17_64); const __m128i k__cospi_p21_p11 = pair_set_epi16(cospi_21_64, cospi_11_64); const __m128i k__cospi_p11_m21 = pair_set_epi16(cospi_11_64, -cospi_21_64); const __m128i k__cospi_p25_p07 = pair_set_epi16(cospi_25_64, cospi_7_64); const __m128i k__cospi_p07_m25 = pair_set_epi16(cospi_7_64, -cospi_25_64); const __m128i k__cospi_p29_p03 = pair_set_epi16(cospi_29_64, cospi_3_64); const __m128i k__cospi_p03_m29 = pair_set_epi16(cospi_3_64, -cospi_29_64); const __m128i k__cospi_p04_p28 = pair_set_epi16(cospi_4_64, cospi_28_64); const __m128i k__cospi_p28_m04 = pair_set_epi16(cospi_28_64, -cospi_4_64); const __m128i k__cospi_p20_p12 = pair_set_epi16(cospi_20_64, cospi_12_64); const __m128i k__cospi_p12_m20 = pair_set_epi16(cospi_12_64, -cospi_20_64); const __m128i k__cospi_m28_p04 = pair_set_epi16(-cospi_28_64, cospi_4_64); const __m128i k__cospi_m12_p20 = pair_set_epi16(-cospi_12_64, cospi_20_64); const __m128i k__cospi_p08_p24 = pair_set_epi16(cospi_8_64, cospi_24_64); const __m128i k__cospi_p24_m08 = pair_set_epi16(cospi_24_64, -cospi_8_64); const __m128i k__cospi_m24_p08 = pair_set_epi16(-cospi_24_64, cospi_8_64); const __m128i k__cospi_m16_m16 = _mm_set1_epi16((int16_t)-cospi_16_64); const __m128i k__cospi_p16_p16 = _mm_set1_epi16((int16_t)cospi_16_64); const __m128i k__cospi_p16_m16 = pair_set_epi16(cospi_16_64, -cospi_16_64); const __m128i k__cospi_m16_p16 = pair_set_epi16(-cospi_16_64, cospi_16_64); const __m128i k__DCT_CONST_ROUNDING = _mm_set1_epi32(DCT_CONST_ROUNDING); const __m128i kZero = _mm_set1_epi16(0); u[0] = _mm_unpacklo_epi16(in[15], in[0]); u[1] = _mm_unpackhi_epi16(in[15], in[0]); u[2] = _mm_unpacklo_epi16(in[13], in[2]); u[3] = _mm_unpackhi_epi16(in[13], in[2]); u[4] = _mm_unpacklo_epi16(in[11], in[4]); u[5] = _mm_unpackhi_epi16(in[11], in[4]); u[6] = _mm_unpacklo_epi16(in[9], in[6]); u[7] = _mm_unpackhi_epi16(in[9], in[6]); u[8] = _mm_unpacklo_epi16(in[7], in[8]); u[9] = _mm_unpackhi_epi16(in[7], in[8]); u[10] = _mm_unpacklo_epi16(in[5], in[10]); u[11] = _mm_unpackhi_epi16(in[5], in[10]); u[12] = _mm_unpacklo_epi16(in[3], in[12]); u[13] = _mm_unpackhi_epi16(in[3], in[12]); u[14] = _mm_unpacklo_epi16(in[1], in[14]); u[15] = _mm_unpackhi_epi16(in[1], in[14]); v[0] = _mm_madd_epi16(u[0], k__cospi_p01_p31); v[1] = _mm_madd_epi16(u[1], k__cospi_p01_p31); v[2] = _mm_madd_epi16(u[0], k__cospi_p31_m01); v[3] = _mm_madd_epi16(u[1], k__cospi_p31_m01); v[4] = _mm_madd_epi16(u[2], k__cospi_p05_p27); v[5] = _mm_madd_epi16(u[3], k__cospi_p05_p27); v[6] = _mm_madd_epi16(u[2], k__cospi_p27_m05); v[7] = _mm_madd_epi16(u[3], k__cospi_p27_m05); v[8] = _mm_madd_epi16(u[4], k__cospi_p09_p23); v[9] = _mm_madd_epi16(u[5], k__cospi_p09_p23); v[10] = _mm_madd_epi16(u[4], k__cospi_p23_m09); v[11] = _mm_madd_epi16(u[5], k__cospi_p23_m09); v[12] = _mm_madd_epi16(u[6], k__cospi_p13_p19); v[13] = _mm_madd_epi16(u[7], k__cospi_p13_p19); v[14] = _mm_madd_epi16(u[6], k__cospi_p19_m13); v[15] = _mm_madd_epi16(u[7], k__cospi_p19_m13); v[16] = _mm_madd_epi16(u[8], k__cospi_p17_p15); v[17] = _mm_madd_epi16(u[9], k__cospi_p17_p15); v[18] = _mm_madd_epi16(u[8], k__cospi_p15_m17); v[19] = _mm_madd_epi16(u[9], k__cospi_p15_m17); v[20] = _mm_madd_epi16(u[10], k__cospi_p21_p11); v[21] = _mm_madd_epi16(u[11], k__cospi_p21_p11); v[22] = _mm_madd_epi16(u[10], k__cospi_p11_m21); v[23] = _mm_madd_epi16(u[11], k__cospi_p11_m21); v[24] = _mm_madd_epi16(u[12], k__cospi_p25_p07); v[25] = _mm_madd_epi16(u[13], k__cospi_p25_p07); v[26] = _mm_madd_epi16(u[12], k__cospi_p07_m25); v[27] = _mm_madd_epi16(u[13], k__cospi_p07_m25); v[28] = _mm_madd_epi16(u[14], k__cospi_p29_p03); v[29] = _mm_madd_epi16(u[15], k__cospi_p29_p03); v[30] = _mm_madd_epi16(u[14], k__cospi_p03_m29); v[31] = _mm_madd_epi16(u[15], k__cospi_p03_m29); u[0] = _mm_add_epi32(v[0], v[16]); u[1] = _mm_add_epi32(v[1], v[17]); u[2] = _mm_add_epi32(v[2], v[18]); u[3] = _mm_add_epi32(v[3], v[19]); u[4] = _mm_add_epi32(v[4], v[20]); u[5] = _mm_add_epi32(v[5], v[21]); u[6] = _mm_add_epi32(v[6], v[22]); u[7] = _mm_add_epi32(v[7], v[23]); u[8] = _mm_add_epi32(v[8], v[24]); u[9] = _mm_add_epi32(v[9], v[25]); u[10] = _mm_add_epi32(v[10], v[26]); u[11] = _mm_add_epi32(v[11], v[27]); u[12] = _mm_add_epi32(v[12], v[28]); u[13] = _mm_add_epi32(v[13], v[29]); u[14] = _mm_add_epi32(v[14], v[30]); u[15] = _mm_add_epi32(v[15], v[31]); u[16] = _mm_sub_epi32(v[0], v[16]); u[17] = _mm_sub_epi32(v[1], v[17]); u[18] = _mm_sub_epi32(v[2], v[18]); u[19] = _mm_sub_epi32(v[3], v[19]); u[20] = _mm_sub_epi32(v[4], v[20]); u[21] = _mm_sub_epi32(v[5], v[21]); u[22] = _mm_sub_epi32(v[6], v[22]); u[23] = _mm_sub_epi32(v[7], v[23]); u[24] = _mm_sub_epi32(v[8], v[24]); u[25] = _mm_sub_epi32(v[9], v[25]); u[26] = _mm_sub_epi32(v[10], v[26]); u[27] = _mm_sub_epi32(v[11], v[27]); u[28] = _mm_sub_epi32(v[12], v[28]); u[29] = _mm_sub_epi32(v[13], v[29]); u[30] = _mm_sub_epi32(v[14], v[30]); u[31] = _mm_sub_epi32(v[15], v[31]); v[0] = _mm_add_epi32(u[0], k__DCT_CONST_ROUNDING); v[1] = _mm_add_epi32(u[1], k__DCT_CONST_ROUNDING); v[2] = _mm_add_epi32(u[2], k__DCT_CONST_ROUNDING); v[3] = _mm_add_epi32(u[3], k__DCT_CONST_ROUNDING); v[4] = _mm_add_epi32(u[4], k__DCT_CONST_ROUNDING); v[5] = _mm_add_epi32(u[5], k__DCT_CONST_ROUNDING); v[6] = _mm_add_epi32(u[6], k__DCT_CONST_ROUNDING); v[7] = _mm_add_epi32(u[7], k__DCT_CONST_ROUNDING); v[8] = _mm_add_epi32(u[8], k__DCT_CONST_ROUNDING); v[9] = _mm_add_epi32(u[9], k__DCT_CONST_ROUNDING); v[10] = _mm_add_epi32(u[10], k__DCT_CONST_ROUNDING); v[11] = _mm_add_epi32(u[11], k__DCT_CONST_ROUNDING); v[12] = _mm_add_epi32(u[12], k__DCT_CONST_ROUNDING); v[13] = _mm_add_epi32(u[13], k__DCT_CONST_ROUNDING); v[14] = _mm_add_epi32(u[14], k__DCT_CONST_ROUNDING); v[15] = _mm_add_epi32(u[15], k__DCT_CONST_ROUNDING); v[16] = _mm_add_epi32(u[16], k__DCT_CONST_ROUNDING); v[17] = _mm_add_epi32(u[17], k__DCT_CONST_ROUNDING); v[18] = _mm_add_epi32(u[18], k__DCT_CONST_ROUNDING); v[19] = _mm_add_epi32(u[19], k__DCT_CONST_ROUNDING); v[20] = _mm_add_epi32(u[20], k__DCT_CONST_ROUNDING); v[21] = _mm_add_epi32(u[21], k__DCT_CONST_ROUNDING); v[22] = _mm_add_epi32(u[22], k__DCT_CONST_ROUNDING); v[23] = _mm_add_epi32(u[23], k__DCT_CONST_ROUNDING); v[24] = _mm_add_epi32(u[24], k__DCT_CONST_ROUNDING); v[25] = _mm_add_epi32(u[25], k__DCT_CONST_ROUNDING); v[26] = _mm_add_epi32(u[26], k__DCT_CONST_ROUNDING); v[27] = _mm_add_epi32(u[27], k__DCT_CONST_ROUNDING); v[28] = _mm_add_epi32(u[28], k__DCT_CONST_ROUNDING); v[29] = _mm_add_epi32(u[29], k__DCT_CONST_ROUNDING); v[30] = _mm_add_epi32(u[30], k__DCT_CONST_ROUNDING); v[31] = _mm_add_epi32(u[31], k__DCT_CONST_ROUNDING); u[0] = _mm_srai_epi32(v[0], DCT_CONST_BITS); u[1] = _mm_srai_epi32(v[1], DCT_CONST_BITS); u[2] = _mm_srai_epi32(v[2], DCT_CONST_BITS); u[3] = _mm_srai_epi32(v[3], DCT_CONST_BITS); u[4] = _mm_srai_epi32(v[4], DCT_CONST_BITS); u[5] = _mm_srai_epi32(v[5], DCT_CONST_BITS); u[6] = _mm_srai_epi32(v[6], DCT_CONST_BITS); u[7] = _mm_srai_epi32(v[7], DCT_CONST_BITS); u[8] = _mm_srai_epi32(v[8], DCT_CONST_BITS); u[9] = _mm_srai_epi32(v[9], DCT_CONST_BITS); u[10] = _mm_srai_epi32(v[10], DCT_CONST_BITS); u[11] = _mm_srai_epi32(v[11], DCT_CONST_BITS); u[12] = _mm_srai_epi32(v[12], DCT_CONST_BITS); u[13] = _mm_srai_epi32(v[13], DCT_CONST_BITS); u[14] = _mm_srai_epi32(v[14], DCT_CONST_BITS); u[15] = _mm_srai_epi32(v[15], DCT_CONST_BITS); u[16] = _mm_srai_epi32(v[16], DCT_CONST_BITS); u[17] = _mm_srai_epi32(v[17], DCT_CONST_BITS); u[18] = _mm_srai_epi32(v[18], DCT_CONST_BITS); u[19] = _mm_srai_epi32(v[19], DCT_CONST_BITS); u[20] = _mm_srai_epi32(v[20], DCT_CONST_BITS); u[21] = _mm_srai_epi32(v[21], DCT_CONST_BITS); u[22] = _mm_srai_epi32(v[22], DCT_CONST_BITS); u[23] = _mm_srai_epi32(v[23], DCT_CONST_BITS); u[24] = _mm_srai_epi32(v[24], DCT_CONST_BITS); u[25] = _mm_srai_epi32(v[25], DCT_CONST_BITS); u[26] = _mm_srai_epi32(v[26], DCT_CONST_BITS); u[27] = _mm_srai_epi32(v[27], DCT_CONST_BITS); u[28] = _mm_srai_epi32(v[28], DCT_CONST_BITS); u[29] = _mm_srai_epi32(v[29], DCT_CONST_BITS); u[30] = _mm_srai_epi32(v[30], DCT_CONST_BITS); u[31] = _mm_srai_epi32(v[31], DCT_CONST_BITS); s[0] = _mm_packs_epi32(u[0], u[1]); s[1] = _mm_packs_epi32(u[2], u[3]); s[2] = _mm_packs_epi32(u[4], u[5]); s[3] = _mm_packs_epi32(u[6], u[7]); s[4] = _mm_packs_epi32(u[8], u[9]); s[5] = _mm_packs_epi32(u[10], u[11]); s[6] = _mm_packs_epi32(u[12], u[13]); s[7] = _mm_packs_epi32(u[14], u[15]); s[8] = _mm_packs_epi32(u[16], u[17]); s[9] = _mm_packs_epi32(u[18], u[19]); s[10] = _mm_packs_epi32(u[20], u[21]); s[11] = _mm_packs_epi32(u[22], u[23]); s[12] = _mm_packs_epi32(u[24], u[25]); s[13] = _mm_packs_epi32(u[26], u[27]); s[14] = _mm_packs_epi32(u[28], u[29]); s[15] = _mm_packs_epi32(u[30], u[31]); // stage 2 u[0] = _mm_unpacklo_epi16(s[8], s[9]); u[1] = _mm_unpackhi_epi16(s[8], s[9]); u[2] = _mm_unpacklo_epi16(s[10], s[11]); u[3] = _mm_unpackhi_epi16(s[10], s[11]); u[4] = _mm_unpacklo_epi16(s[12], s[13]); u[5] = _mm_unpackhi_epi16(s[12], s[13]); u[6] = _mm_unpacklo_epi16(s[14], s[15]); u[7] = _mm_unpackhi_epi16(s[14], s[15]); v[0] = _mm_madd_epi16(u[0], k__cospi_p04_p28); v[1] = _mm_madd_epi16(u[1], k__cospi_p04_p28); v[2] = _mm_madd_epi16(u[0], k__cospi_p28_m04); v[3] = _mm_madd_epi16(u[1], k__cospi_p28_m04); v[4] = _mm_madd_epi16(u[2], k__cospi_p20_p12); v[5] = _mm_madd_epi16(u[3], k__cospi_p20_p12); v[6] = _mm_madd_epi16(u[2], k__cospi_p12_m20); v[7] = _mm_madd_epi16(u[3], k__cospi_p12_m20); v[8] = _mm_madd_epi16(u[4], k__cospi_m28_p04); v[9] = _mm_madd_epi16(u[5], k__cospi_m28_p04); v[10] = _mm_madd_epi16(u[4], k__cospi_p04_p28); v[11] = _mm_madd_epi16(u[5], k__cospi_p04_p28); v[12] = _mm_madd_epi16(u[6], k__cospi_m12_p20); v[13] = _mm_madd_epi16(u[7], k__cospi_m12_p20); v[14] = _mm_madd_epi16(u[6], k__cospi_p20_p12); v[15] = _mm_madd_epi16(u[7], k__cospi_p20_p12); u[0] = _mm_add_epi32(v[0], v[8]); u[1] = _mm_add_epi32(v[1], v[9]); u[2] = _mm_add_epi32(v[2], v[10]); u[3] = _mm_add_epi32(v[3], v[11]); u[4] = _mm_add_epi32(v[4], v[12]); u[5] = _mm_add_epi32(v[5], v[13]); u[6] = _mm_add_epi32(v[6], v[14]); u[7] = _mm_add_epi32(v[7], v[15]); u[8] = _mm_sub_epi32(v[0], v[8]); u[9] = _mm_sub_epi32(v[1], v[9]); u[10] = _mm_sub_epi32(v[2], v[10]); u[11] = _mm_sub_epi32(v[3], v[11]); u[12] = _mm_sub_epi32(v[4], v[12]); u[13] = _mm_sub_epi32(v[5], v[13]); u[14] = _mm_sub_epi32(v[6], v[14]); u[15] = _mm_sub_epi32(v[7], v[15]); v[0] = _mm_add_epi32(u[0], k__DCT_CONST_ROUNDING); v[1] = _mm_add_epi32(u[1], k__DCT_CONST_ROUNDING); v[2] = _mm_add_epi32(u[2], k__DCT_CONST_ROUNDING); v[3] = _mm_add_epi32(u[3], k__DCT_CONST_ROUNDING); v[4] = _mm_add_epi32(u[4], k__DCT_CONST_ROUNDING); v[5] = _mm_add_epi32(u[5], k__DCT_CONST_ROUNDING); v[6] = _mm_add_epi32(u[6], k__DCT_CONST_ROUNDING); v[7] = _mm_add_epi32(u[7], k__DCT_CONST_ROUNDING); v[8] = _mm_add_epi32(u[8], k__DCT_CONST_ROUNDING); v[9] = _mm_add_epi32(u[9], k__DCT_CONST_ROUNDING); v[10] = _mm_add_epi32(u[10], k__DCT_CONST_ROUNDING); v[11] = _mm_add_epi32(u[11], k__DCT_CONST_ROUNDING); v[12] = _mm_add_epi32(u[12], k__DCT_CONST_ROUNDING); v[13] = _mm_add_epi32(u[13], k__DCT_CONST_ROUNDING); v[14] = _mm_add_epi32(u[14], k__DCT_CONST_ROUNDING); v[15] = _mm_add_epi32(u[15], k__DCT_CONST_ROUNDING); u[0] = _mm_srai_epi32(v[0], DCT_CONST_BITS); u[1] = _mm_srai_epi32(v[1], DCT_CONST_BITS); u[2] = _mm_srai_epi32(v[2], DCT_CONST_BITS); u[3] = _mm_srai_epi32(v[3], DCT_CONST_BITS); u[4] = _mm_srai_epi32(v[4], DCT_CONST_BITS); u[5] = _mm_srai_epi32(v[5], DCT_CONST_BITS); u[6] = _mm_srai_epi32(v[6], DCT_CONST_BITS); u[7] = _mm_srai_epi32(v[7], DCT_CONST_BITS); u[8] = _mm_srai_epi32(v[8], DCT_CONST_BITS); u[9] = _mm_srai_epi32(v[9], DCT_CONST_BITS); u[10] = _mm_srai_epi32(v[10], DCT_CONST_BITS); u[11] = _mm_srai_epi32(v[11], DCT_CONST_BITS); u[12] = _mm_srai_epi32(v[12], DCT_CONST_BITS); u[13] = _mm_srai_epi32(v[13], DCT_CONST_BITS); u[14] = _mm_srai_epi32(v[14], DCT_CONST_BITS); u[15] = _mm_srai_epi32(v[15], DCT_CONST_BITS); x[0] = _mm_add_epi16(s[0], s[4]); x[1] = _mm_add_epi16(s[1], s[5]); x[2] = _mm_add_epi16(s[2], s[6]); x[3] = _mm_add_epi16(s[3], s[7]); x[4] = _mm_sub_epi16(s[0], s[4]); x[5] = _mm_sub_epi16(s[1], s[5]); x[6] = _mm_sub_epi16(s[2], s[6]); x[7] = _mm_sub_epi16(s[3], s[7]); x[8] = _mm_packs_epi32(u[0], u[1]); x[9] = _mm_packs_epi32(u[2], u[3]); x[10] = _mm_packs_epi32(u[4], u[5]); x[11] = _mm_packs_epi32(u[6], u[7]); x[12] = _mm_packs_epi32(u[8], u[9]); x[13] = _mm_packs_epi32(u[10], u[11]); x[14] = _mm_packs_epi32(u[12], u[13]); x[15] = _mm_packs_epi32(u[14], u[15]); // stage 3 u[0] = _mm_unpacklo_epi16(x[4], x[5]); u[1] = _mm_unpackhi_epi16(x[4], x[5]); u[2] = _mm_unpacklo_epi16(x[6], x[7]); u[3] = _mm_unpackhi_epi16(x[6], x[7]); u[4] = _mm_unpacklo_epi16(x[12], x[13]); u[5] = _mm_unpackhi_epi16(x[12], x[13]); u[6] = _mm_unpacklo_epi16(x[14], x[15]); u[7] = _mm_unpackhi_epi16(x[14], x[15]); v[0] = _mm_madd_epi16(u[0], k__cospi_p08_p24); v[1] = _mm_madd_epi16(u[1], k__cospi_p08_p24); v[2] = _mm_madd_epi16(u[0], k__cospi_p24_m08); v[3] = _mm_madd_epi16(u[1], k__cospi_p24_m08); v[4] = _mm_madd_epi16(u[2], k__cospi_m24_p08); v[5] = _mm_madd_epi16(u[3], k__cospi_m24_p08); v[6] = _mm_madd_epi16(u[2], k__cospi_p08_p24); v[7] = _mm_madd_epi16(u[3], k__cospi_p08_p24); v[8] = _mm_madd_epi16(u[4], k__cospi_p08_p24); v[9] = _mm_madd_epi16(u[5], k__cospi_p08_p24); v[10] = _mm_madd_epi16(u[4], k__cospi_p24_m08); v[11] = _mm_madd_epi16(u[5], k__cospi_p24_m08); v[12] = _mm_madd_epi16(u[6], k__cospi_m24_p08); v[13] = _mm_madd_epi16(u[7], k__cospi_m24_p08); v[14] = _mm_madd_epi16(u[6], k__cospi_p08_p24); v[15] = _mm_madd_epi16(u[7], k__cospi_p08_p24); u[0] = _mm_add_epi32(v[0], v[4]); u[1] = _mm_add_epi32(v[1], v[5]); u[2] = _mm_add_epi32(v[2], v[6]); u[3] = _mm_add_epi32(v[3], v[7]); u[4] = _mm_sub_epi32(v[0], v[4]); u[5] = _mm_sub_epi32(v[1], v[5]); u[6] = _mm_sub_epi32(v[2], v[6]); u[7] = _mm_sub_epi32(v[3], v[7]); u[8] = _mm_add_epi32(v[8], v[12]); u[9] = _mm_add_epi32(v[9], v[13]); u[10] = _mm_add_epi32(v[10], v[14]); u[11] = _mm_add_epi32(v[11], v[15]); u[12] = _mm_sub_epi32(v[8], v[12]); u[13] = _mm_sub_epi32(v[9], v[13]); u[14] = _mm_sub_epi32(v[10], v[14]); u[15] = _mm_sub_epi32(v[11], v[15]); u[0] = _mm_add_epi32(u[0], k__DCT_CONST_ROUNDING); u[1] = _mm_add_epi32(u[1], k__DCT_CONST_ROUNDING); u[2] = _mm_add_epi32(u[2], k__DCT_CONST_ROUNDING); u[3] = _mm_add_epi32(u[3], k__DCT_CONST_ROUNDING); u[4] = _mm_add_epi32(u[4], k__DCT_CONST_ROUNDING); u[5] = _mm_add_epi32(u[5], k__DCT_CONST_ROUNDING); u[6] = _mm_add_epi32(u[6], k__DCT_CONST_ROUNDING); u[7] = _mm_add_epi32(u[7], k__DCT_CONST_ROUNDING); u[8] = _mm_add_epi32(u[8], k__DCT_CONST_ROUNDING); u[9] = _mm_add_epi32(u[9], k__DCT_CONST_ROUNDING); u[10] = _mm_add_epi32(u[10], k__DCT_CONST_ROUNDING); u[11] = _mm_add_epi32(u[11], k__DCT_CONST_ROUNDING); u[12] = _mm_add_epi32(u[12], k__DCT_CONST_ROUNDING); u[13] = _mm_add_epi32(u[13], k__DCT_CONST_ROUNDING); u[14] = _mm_add_epi32(u[14], k__DCT_CONST_ROUNDING); u[15] = _mm_add_epi32(u[15], k__DCT_CONST_ROUNDING); v[0] = _mm_srai_epi32(u[0], DCT_CONST_BITS); v[1] = _mm_srai_epi32(u[1], DCT_CONST_BITS); v[2] = _mm_srai_epi32(u[2], DCT_CONST_BITS); v[3] = _mm_srai_epi32(u[3], DCT_CONST_BITS); v[4] = _mm_srai_epi32(u[4], DCT_CONST_BITS); v[5] = _mm_srai_epi32(u[5], DCT_CONST_BITS); v[6] = _mm_srai_epi32(u[6], DCT_CONST_BITS); v[7] = _mm_srai_epi32(u[7], DCT_CONST_BITS); v[8] = _mm_srai_epi32(u[8], DCT_CONST_BITS); v[9] = _mm_srai_epi32(u[9], DCT_CONST_BITS); v[10] = _mm_srai_epi32(u[10], DCT_CONST_BITS); v[11] = _mm_srai_epi32(u[11], DCT_CONST_BITS); v[12] = _mm_srai_epi32(u[12], DCT_CONST_BITS); v[13] = _mm_srai_epi32(u[13], DCT_CONST_BITS); v[14] = _mm_srai_epi32(u[14], DCT_CONST_BITS); v[15] = _mm_srai_epi32(u[15], DCT_CONST_BITS); s[0] = _mm_add_epi16(x[0], x[2]); s[1] = _mm_add_epi16(x[1], x[3]); s[2] = _mm_sub_epi16(x[0], x[2]); s[3] = _mm_sub_epi16(x[1], x[3]); s[4] = _mm_packs_epi32(v[0], v[1]); s[5] = _mm_packs_epi32(v[2], v[3]); s[6] = _mm_packs_epi32(v[4], v[5]); s[7] = _mm_packs_epi32(v[6], v[7]); s[8] = _mm_add_epi16(x[8], x[10]); s[9] = _mm_add_epi16(x[9], x[11]); s[10] = _mm_sub_epi16(x[8], x[10]); s[11] = _mm_sub_epi16(x[9], x[11]); s[12] = _mm_packs_epi32(v[8], v[9]); s[13] = _mm_packs_epi32(v[10], v[11]); s[14] = _mm_packs_epi32(v[12], v[13]); s[15] = _mm_packs_epi32(v[14], v[15]); // stage 4 u[0] = _mm_unpacklo_epi16(s[2], s[3]); u[1] = _mm_unpackhi_epi16(s[2], s[3]); u[2] = _mm_unpacklo_epi16(s[6], s[7]); u[3] = _mm_unpackhi_epi16(s[6], s[7]); u[4] = _mm_unpacklo_epi16(s[10], s[11]); u[5] = _mm_unpackhi_epi16(s[10], s[11]); u[6] = _mm_unpacklo_epi16(s[14], s[15]); u[7] = _mm_unpackhi_epi16(s[14], s[15]); in[7] = idct_calc_wraplow_sse2(u[0], u[1], k__cospi_m16_m16); in[8] = idct_calc_wraplow_sse2(u[0], u[1], k__cospi_p16_m16); in[4] = idct_calc_wraplow_sse2(u[2], u[3], k__cospi_p16_p16); in[11] = idct_calc_wraplow_sse2(u[2], u[3], k__cospi_m16_p16); in[6] = idct_calc_wraplow_sse2(u[4], u[5], k__cospi_p16_p16); in[9] = idct_calc_wraplow_sse2(u[4], u[5], k__cospi_m16_p16); in[5] = idct_calc_wraplow_sse2(u[6], u[7], k__cospi_m16_m16); in[10] = idct_calc_wraplow_sse2(u[6], u[7], k__cospi_p16_m16); in[0] = s[0]; in[1] = _mm_sub_epi16(kZero, s[8]); in[2] = s[12]; in[3] = _mm_sub_epi16(kZero, s[4]); in[12] = s[5]; in[13] = _mm_sub_epi16(kZero, s[13]); in[14] = s[9]; in[15] = _mm_sub_epi16(kZero, s[1]); } void idct16_sse2(__m128i *in0, __m128i *in1) { transpose_16bit_16x16(in0, in1); idct16_8col(in0); idct16_8col(in1); } void iadst16_sse2(__m128i *in0, __m128i *in1) { transpose_16bit_16x16(in0, in1); iadst16_8col(in0); iadst16_8col(in1); } void vpx_idct16x16_10_add_sse2(const tran_low_t *input, uint8_t *dest, int stride) { const __m128i final_rounding = _mm_set1_epi16(1 << 5); const __m128i zero = _mm_setzero_si128(); const __m128i stg2_0 = pair_set_epi16(cospi_30_64, -cospi_2_64); const __m128i stg2_1 = pair_set_epi16(cospi_2_64, cospi_30_64); const __m128i stg2_6 = pair_set_epi16(cospi_6_64, -cospi_26_64); const __m128i stg2_7 = pair_set_epi16(cospi_26_64, cospi_6_64); const __m128i stg3_0 = pair_set_epi16(cospi_28_64, -cospi_4_64); const __m128i stg3_1 = pair_set_epi16(cospi_4_64, cospi_28_64); const __m128i stg4_0 = pair_set_epi16(cospi_16_64, cospi_16_64); const __m128i stg4_1 = pair_set_epi16(cospi_16_64, -cospi_16_64); const __m128i stg4_4 = pair_set_epi16(-cospi_8_64, cospi_24_64); const __m128i stg4_5 = pair_set_epi16(cospi_24_64, cospi_8_64); const __m128i stg4_6 = pair_set_epi16(-cospi_24_64, -cospi_8_64); const __m128i stg4_7 = pair_set_epi16(-cospi_8_64, cospi_24_64); const __m128i stg6_0 = pair_set_epi16(-cospi_16_64, cospi_16_64); __m128i in[16], l[16]; __m128i stp1_0, stp1_1, stp1_2, stp1_3, stp1_4, stp1_5, stp1_6, stp1_8, stp1_9, stp1_10, stp1_11, stp1_12, stp1_13, stp1_14, stp1_15, stp1_8_0, stp1_12_0; __m128i stp2_0, stp2_1, stp2_2, stp2_3, stp2_4, stp2_5, stp2_6, stp2_7, stp2_8, stp2_9, stp2_10, stp2_11, stp2_12, stp2_13, stp2_14; __m128i tmp0, tmp1, tmp2, tmp3; int i; // First 1-D inverse DCT // Load input data. in[0] = load_input_data4(input + 0 * 16); in[1] = load_input_data4(input + 1 * 16); in[2] = load_input_data4(input + 2 * 16); in[3] = load_input_data4(input + 3 * 16); transpose_16bit_4x4(in, in); // Stage2 { const __m128i lo_1_15 = _mm_unpackhi_epi16(in[0], zero); const __m128i lo_13_3 = _mm_unpackhi_epi16(zero, in[1]); stp2_8 = idct_calc_wraplow_sse2(stg2_0, stg2_1, lo_1_15); stp2_11 = idct_calc_wraplow_sse2(stg2_6, stg2_7, lo_13_3); } // Stage3 { const __m128i lo_2_14 = _mm_unpacklo_epi16(in[1], zero); stp1_4 = idct_calc_wraplow_sse2(stg3_0, stg3_1, lo_2_14); stp1_13 = _mm_unpackhi_epi64(stp2_11, zero); stp1_14 = _mm_unpackhi_epi64(stp2_8, zero); } // Stage4 { const __m128i lo_0_8 = _mm_unpacklo_epi16(in[0], zero); const __m128i lo_9_14 = _mm_unpacklo_epi16(stp2_8, stp1_14); const __m128i lo_10_13 = _mm_unpacklo_epi16(stp2_11, stp1_13); tmp0 = idct_madd_round_shift_sse2(lo_0_8, stg4_0); tmp1 = idct_madd_round_shift_sse2(lo_0_8, stg4_1); stp1_0 = _mm_packs_epi32(tmp0, tmp0); stp1_1 = _mm_packs_epi32(tmp1, tmp1); stp2_9 = idct_calc_wraplow_sse2(stg4_4, stg4_5, lo_9_14); stp2_10 = idct_calc_wraplow_sse2(stg4_6, stg4_7, lo_10_13); stp2_6 = _mm_unpackhi_epi64(stp1_4, zero); } // Stage5 and Stage6 { tmp0 = _mm_add_epi16(stp2_8, stp2_11); tmp1 = _mm_sub_epi16(stp2_8, stp2_11); tmp2 = _mm_add_epi16(stp2_9, stp2_10); tmp3 = _mm_sub_epi16(stp2_9, stp2_10); stp1_9 = _mm_unpacklo_epi64(tmp2, zero); stp1_10 = _mm_unpacklo_epi64(tmp3, zero); stp1_8 = _mm_unpacklo_epi64(tmp0, zero); stp1_11 = _mm_unpacklo_epi64(tmp1, zero); stp1_13 = _mm_unpackhi_epi64(tmp3, zero); stp1_14 = _mm_unpackhi_epi64(tmp2, zero); stp1_12 = _mm_unpackhi_epi64(tmp1, zero); stp1_15 = _mm_unpackhi_epi64(tmp0, zero); } // Stage6 { const __m128i lo_6_5 = _mm_unpacklo_epi16(stp2_6, stp1_4); const __m128i lo_10_13 = _mm_unpacklo_epi16(stp1_10, stp1_13); const __m128i lo_11_12 = _mm_unpacklo_epi16(stp1_11, stp1_12); stp1_6 = idct_calc_wraplow_sse2(stg4_0, stg4_1, lo_6_5); tmp0 = idct_madd_round_shift_sse2(lo_10_13, stg6_0); tmp1 = idct_madd_round_shift_sse2(lo_10_13, stg4_0); tmp2 = idct_madd_round_shift_sse2(lo_11_12, stg6_0); tmp3 = idct_madd_round_shift_sse2(lo_11_12, stg4_0); stp2_10 = _mm_packs_epi32(tmp0, zero); stp2_13 = _mm_packs_epi32(tmp1, zero); stp2_11 = _mm_packs_epi32(tmp2, zero); stp2_12 = _mm_packs_epi32(tmp3, zero); tmp0 = _mm_add_epi16(stp1_0, stp1_4); tmp1 = _mm_sub_epi16(stp1_0, stp1_4); tmp2 = _mm_add_epi16(stp1_1, stp1_6); tmp3 = _mm_sub_epi16(stp1_1, stp1_6); stp2_0 = _mm_unpackhi_epi64(tmp0, zero); stp2_1 = _mm_unpacklo_epi64(tmp2, zero); stp2_2 = _mm_unpackhi_epi64(tmp2, zero); stp2_3 = _mm_unpacklo_epi64(tmp0, zero); stp2_4 = _mm_unpacklo_epi64(tmp1, zero); stp2_5 = _mm_unpackhi_epi64(tmp3, zero); stp2_6 = _mm_unpacklo_epi64(tmp3, zero); stp2_7 = _mm_unpackhi_epi64(tmp1, zero); } // Stage7. Left 8x16 only. l[0] = _mm_add_epi16(stp2_0, stp1_15); l[1] = _mm_add_epi16(stp2_1, stp1_14); l[2] = _mm_add_epi16(stp2_2, stp2_13); l[3] = _mm_add_epi16(stp2_3, stp2_12); l[4] = _mm_add_epi16(stp2_4, stp2_11); l[5] = _mm_add_epi16(stp2_5, stp2_10); l[6] = _mm_add_epi16(stp2_6, stp1_9); l[7] = _mm_add_epi16(stp2_7, stp1_8); l[8] = _mm_sub_epi16(stp2_7, stp1_8); l[9] = _mm_sub_epi16(stp2_6, stp1_9); l[10] = _mm_sub_epi16(stp2_5, stp2_10); l[11] = _mm_sub_epi16(stp2_4, stp2_11); l[12] = _mm_sub_epi16(stp2_3, stp2_12); l[13] = _mm_sub_epi16(stp2_2, stp2_13); l[14] = _mm_sub_epi16(stp2_1, stp1_14); l[15] = _mm_sub_epi16(stp2_0, stp1_15); // Second 1-D inverse transform, performed per 8x16 block for (i = 0; i < 2; i++) { int j; transpose_16bit_4x8(l + 8 * i, in); IDCT16_10 // Stage7 in[0] = _mm_add_epi16(stp2_0, stp1_15); in[1] = _mm_add_epi16(stp2_1, stp1_14); in[2] = _mm_add_epi16(stp2_2, stp2_13); in[3] = _mm_add_epi16(stp2_3, stp2_12); in[4] = _mm_add_epi16(stp2_4, stp2_11); in[5] = _mm_add_epi16(stp2_5, stp2_10); in[6] = _mm_add_epi16(stp2_6, stp1_9); in[7] = _mm_add_epi16(stp2_7, stp1_8); in[8] = _mm_sub_epi16(stp2_7, stp1_8); in[9] = _mm_sub_epi16(stp2_6, stp1_9); in[10] = _mm_sub_epi16(stp2_5, stp2_10); in[11] = _mm_sub_epi16(stp2_4, stp2_11); in[12] = _mm_sub_epi16(stp2_3, stp2_12); in[13] = _mm_sub_epi16(stp2_2, stp2_13); in[14] = _mm_sub_epi16(stp2_1, stp1_14); in[15] = _mm_sub_epi16(stp2_0, stp1_15); for (j = 0; j < 16; ++j) { // Final rounding and shift in[j] = _mm_adds_epi16(in[j], final_rounding); in[j] = _mm_srai_epi16(in[j], 6); recon_and_store(dest + j * stride, in[j]); } dest += 8; } } #define IDCT32_34 \ /* Stage1 */ \ multiplication_and_add_2(&in[1], &zero, &stg1_0, &stg1_1, &stp1_16, \ &stp1_31); \ multiplication_and_add_2(&zero, &in[7], &stg1_6, &stg1_7, &stp1_19, \ &stp1_28); \ multiplication_and_add_2(&in[5], &zero, &stg1_8, &stg1_9, &stp1_20, \ &stp1_27); \ multiplication_and_add_2(&zero, &in[3], &stg1_14, &stg1_15, &stp1_23, \ &stp1_24); \ \ /* Stage2 */ \ multiplication_and_add_2(&in[2], &zero, &stg2_0, &stg2_1, &stp2_8, \ &stp2_15); \ multiplication_and_add_2(&zero, &in[6], &stg2_6, &stg2_7, &stp2_11, \ &stp2_12); \ \ stp2_16 = stp1_16; \ stp2_19 = stp1_19; \ \ stp2_20 = stp1_20; \ stp2_23 = stp1_23; \ \ stp2_24 = stp1_24; \ stp2_27 = stp1_27; \ \ stp2_28 = stp1_28; \ stp2_31 = stp1_31; \ \ /* Stage3 */ \ multiplication_and_add_2(&in[4], &zero, &stg3_0, &stg3_1, &stp1_4, &stp1_7); \ \ stp1_8 = stp2_8; \ stp1_11 = stp2_11; \ stp1_12 = stp2_12; \ stp1_15 = stp2_15; \ \ multiplication_and_add(&stp1_16, &stp1_31, &stp1_19, &stp1_28, &stg3_4, \ &stg3_5, &stg3_6, &stg3_4, &stp1_17, &stp1_30, \ &stp1_18, &stp1_29); \ \ multiplication_and_add(&stp1_20, &stp1_27, &stp1_23, &stp1_24, &stg3_8, \ &stg3_9, &stg3_10, &stg3_8, &stp1_21, &stp1_26, \ &stp1_22, &stp1_25); \ \ stp1_16 = stp2_16; \ stp1_31 = stp2_31; \ stp1_19 = stp2_19; \ stp1_20 = stp2_20; \ stp1_23 = stp2_23; \ stp1_24 = stp2_24; \ stp1_27 = stp2_27; \ stp1_28 = stp2_28; \ \ /* Stage4 */ \ multiplication_and_add_2(&in[0], &zero, &stg4_0, &stg4_1, &stp2_0, &stp2_1); \ \ stp2_4 = stp1_4; \ stp2_5 = stp1_4; \ stp2_6 = stp1_7; \ stp2_7 = stp1_7; \ \ multiplication_and_add(&stp2_8, &stp2_15, &stp2_11, &stp2_12, &stg4_4, \ &stg4_5, &stg4_6, &stg4_4, &stp2_9, &stp2_14, \ &stp2_10, &stp2_13); \ \ stp2_8 = stp1_8; \ stp2_15 = stp1_15; \ stp2_11 = stp1_11; \ stp2_12 = stp1_12; \ \ stp2_16 = _mm_add_epi16(stp1_16, stp1_19); \ stp2_17 = _mm_add_epi16(stp1_17, stp1_18); \ stp2_18 = _mm_sub_epi16(stp1_17, stp1_18); \ stp2_19 = _mm_sub_epi16(stp1_16, stp1_19); \ stp2_20 = _mm_sub_epi16(stp1_23, stp1_20); \ stp2_21 = _mm_sub_epi16(stp1_22, stp1_21); \ stp2_22 = _mm_add_epi16(stp1_22, stp1_21); \ stp2_23 = _mm_add_epi16(stp1_23, stp1_20); \ \ stp2_24 = _mm_add_epi16(stp1_24, stp1_27); \ stp2_25 = _mm_add_epi16(stp1_25, stp1_26); \ stp2_26 = _mm_sub_epi16(stp1_25, stp1_26); \ stp2_27 = _mm_sub_epi16(stp1_24, stp1_27); \ stp2_28 = _mm_sub_epi16(stp1_31, stp1_28); \ stp2_29 = _mm_sub_epi16(stp1_30, stp1_29); \ stp2_30 = _mm_add_epi16(stp1_29, stp1_30); \ stp2_31 = _mm_add_epi16(stp1_28, stp1_31); \ \ /* Stage5 */ \ stp1_0 = stp2_0; \ stp1_1 = stp2_1; \ stp1_2 = stp2_1; \ stp1_3 = stp2_0; \ multiplication_and_add_2(&stp2_6, &stp2_5, &stg4_1, &stg4_0, &stp1_5, \ &stp1_6); \ \ stp1_4 = stp2_4; \ stp1_7 = stp2_7; \ \ stp1_8 = _mm_add_epi16(stp2_8, stp2_11); \ stp1_9 = _mm_add_epi16(stp2_9, stp2_10); \ stp1_10 = _mm_sub_epi16(stp2_9, stp2_10); \ stp1_11 = _mm_sub_epi16(stp2_8, stp2_11); \ stp1_12 = _mm_sub_epi16(stp2_15, stp2_12); \ stp1_13 = _mm_sub_epi16(stp2_14, stp2_13); \ stp1_14 = _mm_add_epi16(stp2_14, stp2_13); \ stp1_15 = _mm_add_epi16(stp2_15, stp2_12); \ \ stp1_16 = stp2_16; \ stp1_17 = stp2_17; \ \ multiplication_and_add(&stp2_18, &stp2_29, &stp2_19, &stp2_28, &stg4_4, \ &stg4_5, &stg4_4, &stg4_5, &stp1_18, &stp1_29, \ &stp1_19, &stp1_28); \ multiplication_and_add(&stp2_20, &stp2_27, &stp2_21, &stp2_26, &stg4_6, \ &stg4_4, &stg4_6, &stg4_4, &stp1_20, &stp1_27, \ &stp1_21, &stp1_26); \ \ stp1_22 = stp2_22; \ stp1_23 = stp2_23; \ stp1_24 = stp2_24; \ stp1_25 = stp2_25; \ stp1_30 = stp2_30; \ stp1_31 = stp2_31; \ \ /* Stage6 */ \ stp2_0 = _mm_add_epi16(stp1_0, stp1_7); \ stp2_1 = _mm_add_epi16(stp1_1, stp1_6); \ stp2_2 = _mm_add_epi16(stp1_2, stp1_5); \ stp2_3 = _mm_add_epi16(stp1_3, stp1_4); \ stp2_4 = _mm_sub_epi16(stp1_3, stp1_4); \ stp2_5 = _mm_sub_epi16(stp1_2, stp1_5); \ stp2_6 = _mm_sub_epi16(stp1_1, stp1_6); \ stp2_7 = _mm_sub_epi16(stp1_0, stp1_7); \ \ stp2_8 = stp1_8; \ stp2_9 = stp1_9; \ stp2_14 = stp1_14; \ stp2_15 = stp1_15; \ \ multiplication_and_add(&stp1_10, &stp1_13, &stp1_11, &stp1_12, &stg6_0, \ &stg4_0, &stg6_0, &stg4_0, &stp2_10, &stp2_13, \ &stp2_11, &stp2_12); \ \ stp2_16 = _mm_add_epi16(stp1_16, stp1_23); \ stp2_17 = _mm_add_epi16(stp1_17, stp1_22); \ stp2_18 = _mm_add_epi16(stp1_18, stp1_21); \ stp2_19 = _mm_add_epi16(stp1_19, stp1_20); \ stp2_20 = _mm_sub_epi16(stp1_19, stp1_20); \ stp2_21 = _mm_sub_epi16(stp1_18, stp1_21); \ stp2_22 = _mm_sub_epi16(stp1_17, stp1_22); \ stp2_23 = _mm_sub_epi16(stp1_16, stp1_23); \ \ stp2_24 = _mm_sub_epi16(stp1_31, stp1_24); \ stp2_25 = _mm_sub_epi16(stp1_30, stp1_25); \ stp2_26 = _mm_sub_epi16(stp1_29, stp1_26); \ stp2_27 = _mm_sub_epi16(stp1_28, stp1_27); \ stp2_28 = _mm_add_epi16(stp1_27, stp1_28); \ stp2_29 = _mm_add_epi16(stp1_26, stp1_29); \ stp2_30 = _mm_add_epi16(stp1_25, stp1_30); \ stp2_31 = _mm_add_epi16(stp1_24, stp1_31); \ \ /* Stage7 */ \ stp1_0 = _mm_add_epi16(stp2_0, stp2_15); \ stp1_1 = _mm_add_epi16(stp2_1, stp2_14); \ stp1_2 = _mm_add_epi16(stp2_2, stp2_13); \ stp1_3 = _mm_add_epi16(stp2_3, stp2_12); \ stp1_4 = _mm_add_epi16(stp2_4, stp2_11); \ stp1_5 = _mm_add_epi16(stp2_5, stp2_10); \ stp1_6 = _mm_add_epi16(stp2_6, stp2_9); \ stp1_7 = _mm_add_epi16(stp2_7, stp2_8); \ stp1_8 = _mm_sub_epi16(stp2_7, stp2_8); \ stp1_9 = _mm_sub_epi16(stp2_6, stp2_9); \ stp1_10 = _mm_sub_epi16(stp2_5, stp2_10); \ stp1_11 = _mm_sub_epi16(stp2_4, stp2_11); \ stp1_12 = _mm_sub_epi16(stp2_3, stp2_12); \ stp1_13 = _mm_sub_epi16(stp2_2, stp2_13); \ stp1_14 = _mm_sub_epi16(stp2_1, stp2_14); \ stp1_15 = _mm_sub_epi16(stp2_0, stp2_15); \ \ stp1_16 = stp2_16; \ stp1_17 = stp2_17; \ stp1_18 = stp2_18; \ stp1_19 = stp2_19; \ \ multiplication_and_add(&stp2_20, &stp2_27, &stp2_21, &stp2_26, &stg6_0, \ &stg4_0, &stg6_0, &stg4_0, &stp1_20, &stp1_27, \ &stp1_21, &stp1_26); \ multiplication_and_add(&stp2_22, &stp2_25, &stp2_23, &stp2_24, &stg6_0, \ &stg4_0, &stg6_0, &stg4_0, &stp1_22, &stp1_25, \ &stp1_23, &stp1_24); \ \ stp1_28 = stp2_28; \ stp1_29 = stp2_29; \ stp1_30 = stp2_30; \ stp1_31 = stp2_31; // Only upper-left 8x8 has non-zero coeff void vpx_idct32x32_34_add_sse2(const tran_low_t *input, uint8_t *dest, int stride) { const __m128i zero = _mm_setzero_si128(); const __m128i final_rounding = _mm_set1_epi16(1 << 5); // idct constants for each stage const __m128i stg1_0 = pair_set_epi16(cospi_31_64, -cospi_1_64); const __m128i stg1_1 = pair_set_epi16(cospi_1_64, cospi_31_64); const __m128i stg1_6 = pair_set_epi16(cospi_7_64, -cospi_25_64); const __m128i stg1_7 = pair_set_epi16(cospi_25_64, cospi_7_64); const __m128i stg1_8 = pair_set_epi16(cospi_27_64, -cospi_5_64); const __m128i stg1_9 = pair_set_epi16(cospi_5_64, cospi_27_64); const __m128i stg1_14 = pair_set_epi16(cospi_3_64, -cospi_29_64); const __m128i stg1_15 = pair_set_epi16(cospi_29_64, cospi_3_64); const __m128i stg2_0 = pair_set_epi16(cospi_30_64, -cospi_2_64); const __m128i stg2_1 = pair_set_epi16(cospi_2_64, cospi_30_64); const __m128i stg2_6 = pair_set_epi16(cospi_6_64, -cospi_26_64); const __m128i stg2_7 = pair_set_epi16(cospi_26_64, cospi_6_64); const __m128i stg3_0 = pair_set_epi16(cospi_28_64, -cospi_4_64); const __m128i stg3_1 = pair_set_epi16(cospi_4_64, cospi_28_64); const __m128i stg3_4 = pair_set_epi16(-cospi_4_64, cospi_28_64); const __m128i stg3_5 = pair_set_epi16(cospi_28_64, cospi_4_64); const __m128i stg3_6 = pair_set_epi16(-cospi_28_64, -cospi_4_64); const __m128i stg3_8 = pair_set_epi16(-cospi_20_64, cospi_12_64); const __m128i stg3_9 = pair_set_epi16(cospi_12_64, cospi_20_64); const __m128i stg3_10 = pair_set_epi16(-cospi_12_64, -cospi_20_64); const __m128i stg4_0 = pair_set_epi16(cospi_16_64, cospi_16_64); const __m128i stg4_1 = pair_set_epi16(cospi_16_64, -cospi_16_64); const __m128i stg4_4 = pair_set_epi16(-cospi_8_64, cospi_24_64); const __m128i stg4_5 = pair_set_epi16(cospi_24_64, cospi_8_64); const __m128i stg4_6 = pair_set_epi16(-cospi_24_64, -cospi_8_64); const __m128i stg6_0 = pair_set_epi16(-cospi_16_64, cospi_16_64); __m128i in[32], col[32]; __m128i stp1_0, stp1_1, stp1_2, stp1_3, stp1_4, stp1_5, stp1_6, stp1_7, stp1_8, stp1_9, stp1_10, stp1_11, stp1_12, stp1_13, stp1_14, stp1_15, stp1_16, stp1_17, stp1_18, stp1_19, stp1_20, stp1_21, stp1_22, stp1_23, stp1_24, stp1_25, stp1_26, stp1_27, stp1_28, stp1_29, stp1_30, stp1_31; __m128i stp2_0, stp2_1, stp2_2, stp2_3, stp2_4, stp2_5, stp2_6, stp2_7, stp2_8, stp2_9, stp2_10, stp2_11, stp2_12, stp2_13, stp2_14, stp2_15, stp2_16, stp2_17, stp2_18, stp2_19, stp2_20, stp2_21, stp2_22, stp2_23, stp2_24, stp2_25, stp2_26, stp2_27, stp2_28, stp2_29, stp2_30, stp2_31; int i; // Load input data. Only need to load the top left 8x8 block. in[0] = load_input_data8(input + 0 * 32); in[1] = load_input_data8(input + 1 * 32); in[2] = load_input_data8(input + 2 * 32); in[3] = load_input_data8(input + 3 * 32); in[4] = load_input_data8(input + 4 * 32); in[5] = load_input_data8(input + 5 * 32); in[6] = load_input_data8(input + 6 * 32); in[7] = load_input_data8(input + 7 * 32); transpose_16bit_8x8(in, in); IDCT32_34 // 1_D: Store 32 intermediate results for each 8x32 block. col[0] = _mm_add_epi16(stp1_0, stp1_31); col[1] = _mm_add_epi16(stp1_1, stp1_30); col[2] = _mm_add_epi16(stp1_2, stp1_29); col[3] = _mm_add_epi16(stp1_3, stp1_28); col[4] = _mm_add_epi16(stp1_4, stp1_27); col[5] = _mm_add_epi16(stp1_5, stp1_26); col[6] = _mm_add_epi16(stp1_6, stp1_25); col[7] = _mm_add_epi16(stp1_7, stp1_24); col[8] = _mm_add_epi16(stp1_8, stp1_23); col[9] = _mm_add_epi16(stp1_9, stp1_22); col[10] = _mm_add_epi16(stp1_10, stp1_21); col[11] = _mm_add_epi16(stp1_11, stp1_20); col[12] = _mm_add_epi16(stp1_12, stp1_19); col[13] = _mm_add_epi16(stp1_13, stp1_18); col[14] = _mm_add_epi16(stp1_14, stp1_17); col[15] = _mm_add_epi16(stp1_15, stp1_16); col[16] = _mm_sub_epi16(stp1_15, stp1_16); col[17] = _mm_sub_epi16(stp1_14, stp1_17); col[18] = _mm_sub_epi16(stp1_13, stp1_18); col[19] = _mm_sub_epi16(stp1_12, stp1_19); col[20] = _mm_sub_epi16(stp1_11, stp1_20); col[21] = _mm_sub_epi16(stp1_10, stp1_21); col[22] = _mm_sub_epi16(stp1_9, stp1_22); col[23] = _mm_sub_epi16(stp1_8, stp1_23); col[24] = _mm_sub_epi16(stp1_7, stp1_24); col[25] = _mm_sub_epi16(stp1_6, stp1_25); col[26] = _mm_sub_epi16(stp1_5, stp1_26); col[27] = _mm_sub_epi16(stp1_4, stp1_27); col[28] = _mm_sub_epi16(stp1_3, stp1_28); col[29] = _mm_sub_epi16(stp1_2, stp1_29); col[30] = _mm_sub_epi16(stp1_1, stp1_30); col[31] = _mm_sub_epi16(stp1_0, stp1_31); for (i = 0; i < 4; i++) { int j; // Transpose 32x8 block to 8x32 block transpose_16bit_8x8(col + i * 8, in); IDCT32_34 // 2_D: Calculate the results and store them to destination. in[0] = _mm_add_epi16(stp1_0, stp1_31); in[1] = _mm_add_epi16(stp1_1, stp1_30); in[2] = _mm_add_epi16(stp1_2, stp1_29); in[3] = _mm_add_epi16(stp1_3, stp1_28); in[4] = _mm_add_epi16(stp1_4, stp1_27); in[5] = _mm_add_epi16(stp1_5, stp1_26); in[6] = _mm_add_epi16(stp1_6, stp1_25); in[7] = _mm_add_epi16(stp1_7, stp1_24); in[8] = _mm_add_epi16(stp1_8, stp1_23); in[9] = _mm_add_epi16(stp1_9, stp1_22); in[10] = _mm_add_epi16(stp1_10, stp1_21); in[11] = _mm_add_epi16(stp1_11, stp1_20); in[12] = _mm_add_epi16(stp1_12, stp1_19); in[13] = _mm_add_epi16(stp1_13, stp1_18); in[14] = _mm_add_epi16(stp1_14, stp1_17); in[15] = _mm_add_epi16(stp1_15, stp1_16); in[16] = _mm_sub_epi16(stp1_15, stp1_16); in[17] = _mm_sub_epi16(stp1_14, stp1_17); in[18] = _mm_sub_epi16(stp1_13, stp1_18); in[19] = _mm_sub_epi16(stp1_12, stp1_19); in[20] = _mm_sub_epi16(stp1_11, stp1_20); in[21] = _mm_sub_epi16(stp1_10, stp1_21); in[22] = _mm_sub_epi16(stp1_9, stp1_22); in[23] = _mm_sub_epi16(stp1_8, stp1_23); in[24] = _mm_sub_epi16(stp1_7, stp1_24); in[25] = _mm_sub_epi16(stp1_6, stp1_25); in[26] = _mm_sub_epi16(stp1_5, stp1_26); in[27] = _mm_sub_epi16(stp1_4, stp1_27); in[28] = _mm_sub_epi16(stp1_3, stp1_28); in[29] = _mm_sub_epi16(stp1_2, stp1_29); in[30] = _mm_sub_epi16(stp1_1, stp1_30); in[31] = _mm_sub_epi16(stp1_0, stp1_31); for (j = 0; j < 32; ++j) { // Final rounding and shift in[j] = _mm_adds_epi16(in[j], final_rounding); in[j] = _mm_srai_epi16(in[j], 6); recon_and_store(dest + j * stride, in[j]); } dest += 8; } } // For each 8x32 block __m128i in[32], // Input with index, 0, 4, 8, 12, 16, 20, 24, 28 // output pixels: 0-7 in __m128i in[32] static void idct32_full_8x32_quarter_1(const __m128i *in /*in[32]*/, __m128i *out /*out[8]*/) { __m128i u0, u1, u2, u3, u4, u5, u6, u7; // stp1_ __m128i v0, v1, v2, v3, v4, v5, v6, v7; // stp2_ { const __m128i stg3_0 = pair_set_epi16(cospi_28_64, -cospi_4_64); const __m128i stg3_1 = pair_set_epi16(cospi_4_64, cospi_28_64); const __m128i stg3_2 = pair_set_epi16(cospi_12_64, -cospi_20_64); const __m128i stg3_3 = pair_set_epi16(cospi_20_64, cospi_12_64); butterfly(&in[4], &in[28], &stg3_0, &stg3_1, &u4, &u7); butterfly(&in[20], &in[12], &stg3_2, &stg3_3, &u5, &u6); } v4 = _mm_add_epi16(u4, u5); v5 = _mm_sub_epi16(u4, u5); v6 = _mm_sub_epi16(u7, u6); v7 = _mm_add_epi16(u7, u6); { const __m128i stg4_0 = pair_set_epi16(cospi_16_64, cospi_16_64); const __m128i stg4_1 = pair_set_epi16(cospi_16_64, -cospi_16_64); const __m128i stg4_2 = pair_set_epi16(cospi_24_64, -cospi_8_64); const __m128i stg4_3 = pair_set_epi16(cospi_8_64, cospi_24_64); butterfly(&v6, &v5, &stg4_1, &stg4_0, &v5, &v6); butterfly(&in[0], &in[16], &stg4_0, &stg4_1, &u0, &u1); butterfly(&in[8], &in[24], &stg4_2, &stg4_3, &u2, &u3); } v0 = _mm_add_epi16(u0, u3); v1 = _mm_add_epi16(u1, u2); v2 = _mm_sub_epi16(u1, u2); v3 = _mm_sub_epi16(u0, u3); out[0] = _mm_add_epi16(v0, v7); out[1] = _mm_add_epi16(v1, v6); out[2] = _mm_add_epi16(v2, v5); out[3] = _mm_add_epi16(v3, v4); out[4] = _mm_sub_epi16(v3, v4); out[5] = _mm_sub_epi16(v2, v5); out[6] = _mm_sub_epi16(v1, v6); out[7] = _mm_sub_epi16(v0, v7); } // For each 8x32 block __m128i in[32], // Input with index, 2, 6, 10, 14, 18, 22, 26, 30 // output pixels: 8-15 in __m128i in[32] static void idct32_full_8x32_quarter_2(const __m128i *in /*in[32]*/, __m128i *out /*out[16]*/) { __m128i u8, u9, u10, u11, u12, u13, u14, u15; // stp2_ __m128i v8, v9, v10, v11, v12, v13, v14, v15; // stp1_ { const __m128i stg2_0 = pair_set_epi16(cospi_30_64, -cospi_2_64); const __m128i stg2_1 = pair_set_epi16(cospi_2_64, cospi_30_64); const __m128i stg2_2 = pair_set_epi16(cospi_14_64, -cospi_18_64); const __m128i stg2_3 = pair_set_epi16(cospi_18_64, cospi_14_64); butterfly(&in[2], &in[30], &stg2_0, &stg2_1, &u8, &u15); butterfly(&in[18], &in[14], &stg2_2, &stg2_3, &u9, &u14); } v8 = _mm_add_epi16(u8, u9); v9 = _mm_sub_epi16(u8, u9); v14 = _mm_sub_epi16(u15, u14); v15 = _mm_add_epi16(u15, u14); { const __m128i stg2_4 = pair_set_epi16(cospi_22_64, -cospi_10_64); const __m128i stg2_5 = pair_set_epi16(cospi_10_64, cospi_22_64); const __m128i stg2_6 = pair_set_epi16(cospi_6_64, -cospi_26_64); const __m128i stg2_7 = pair_set_epi16(cospi_26_64, cospi_6_64); butterfly(&in[10], &in[22], &stg2_4, &stg2_5, &u10, &u13); butterfly(&in[26], &in[6], &stg2_6, &stg2_7, &u11, &u12); } v10 = _mm_sub_epi16(u11, u10); v11 = _mm_add_epi16(u11, u10); v12 = _mm_add_epi16(u12, u13); v13 = _mm_sub_epi16(u12, u13); { const __m128i stg4_4 = pair_set_epi16(-cospi_8_64, cospi_24_64); const __m128i stg4_5 = pair_set_epi16(cospi_24_64, cospi_8_64); const __m128i stg4_6 = pair_set_epi16(-cospi_24_64, -cospi_8_64); butterfly_self(&v9, &v14, &stg4_4, &stg4_5); butterfly_self(&v10, &v13, &stg4_6, &stg4_4); } out[0] = _mm_add_epi16(v8, v11); out[1] = _mm_add_epi16(v9, v10); out[6] = _mm_add_epi16(v14, v13); out[7] = _mm_add_epi16(v15, v12); out[2] = _mm_sub_epi16(v9, v10); out[3] = _mm_sub_epi16(v8, v11); out[4] = _mm_sub_epi16(v15, v12); out[5] = _mm_sub_epi16(v14, v13); { const __m128i stg4_0 = pair_set_epi16(cospi_16_64, cospi_16_64); const __m128i stg6_0 = pair_set_epi16(-cospi_16_64, cospi_16_64); butterfly_self(&out[2], &out[5], &stg6_0, &stg4_0); butterfly_self(&out[3], &out[4], &stg6_0, &stg4_0); } } // For each 8x32 block __m128i in[32], // Input with odd index, // 1, 3, 5, 7, 9, 11, 13, 15, 17, 19, 21, 23, 25, 27, 29, 31 // output pixels: 16-23, 24-31 in __m128i in[32] // We avoid hide an offset, 16, inside this function. So we output 0-15 into // array out[16] static void idct32_full_8x32_quarter_3_4(const __m128i *in /*in[32]*/, __m128i *out /*out[16]*/) { __m128i v16, v17, v18, v19, v20, v21, v22, v23; __m128i v24, v25, v26, v27, v28, v29, v30, v31; __m128i u16, u17, u18, u19, u20, u21, u22, u23; __m128i u24, u25, u26, u27, u28, u29, u30, u31; { const __m128i stg1_0 = pair_set_epi16(cospi_31_64, -cospi_1_64); const __m128i stg1_1 = pair_set_epi16(cospi_1_64, cospi_31_64); const __m128i stg1_2 = pair_set_epi16(cospi_15_64, -cospi_17_64); const __m128i stg1_3 = pair_set_epi16(cospi_17_64, cospi_15_64); const __m128i stg1_4 = pair_set_epi16(cospi_23_64, -cospi_9_64); const __m128i stg1_5 = pair_set_epi16(cospi_9_64, cospi_23_64); const __m128i stg1_6 = pair_set_epi16(cospi_7_64, -cospi_25_64); const __m128i stg1_7 = pair_set_epi16(cospi_25_64, cospi_7_64); const __m128i stg1_8 = pair_set_epi16(cospi_27_64, -cospi_5_64); const __m128i stg1_9 = pair_set_epi16(cospi_5_64, cospi_27_64); const __m128i stg1_10 = pair_set_epi16(cospi_11_64, -cospi_21_64); const __m128i stg1_11 = pair_set_epi16(cospi_21_64, cospi_11_64); const __m128i stg1_12 = pair_set_epi16(cospi_19_64, -cospi_13_64); const __m128i stg1_13 = pair_set_epi16(cospi_13_64, cospi_19_64); const __m128i stg1_14 = pair_set_epi16(cospi_3_64, -cospi_29_64); const __m128i stg1_15 = pair_set_epi16(cospi_29_64, cospi_3_64); butterfly(&in[1], &in[31], &stg1_0, &stg1_1, &u16, &u31); butterfly(&in[17], &in[15], &stg1_2, &stg1_3, &u17, &u30); butterfly(&in[9], &in[23], &stg1_4, &stg1_5, &u18, &u29); butterfly(&in[25], &in[7], &stg1_6, &stg1_7, &u19, &u28); butterfly(&in[5], &in[27], &stg1_8, &stg1_9, &u20, &u27); butterfly(&in[21], &in[11], &stg1_10, &stg1_11, &u21, &u26); butterfly(&in[13], &in[19], &stg1_12, &stg1_13, &u22, &u25); butterfly(&in[29], &in[3], &stg1_14, &stg1_15, &u23, &u24); } v16 = _mm_add_epi16(u16, u17); v17 = _mm_sub_epi16(u16, u17); v18 = _mm_sub_epi16(u19, u18); v19 = _mm_add_epi16(u19, u18); v20 = _mm_add_epi16(u20, u21); v21 = _mm_sub_epi16(u20, u21); v22 = _mm_sub_epi16(u23, u22); v23 = _mm_add_epi16(u23, u22); v24 = _mm_add_epi16(u24, u25); v25 = _mm_sub_epi16(u24, u25); v26 = _mm_sub_epi16(u27, u26); v27 = _mm_add_epi16(u27, u26); v28 = _mm_add_epi16(u28, u29); v29 = _mm_sub_epi16(u28, u29); v30 = _mm_sub_epi16(u31, u30); v31 = _mm_add_epi16(u31, u30); { const __m128i stg3_4 = pair_set_epi16(-cospi_4_64, cospi_28_64); const __m128i stg3_5 = pair_set_epi16(cospi_28_64, cospi_4_64); const __m128i stg3_6 = pair_set_epi16(-cospi_28_64, -cospi_4_64); const __m128i stg3_8 = pair_set_epi16(-cospi_20_64, cospi_12_64); const __m128i stg3_9 = pair_set_epi16(cospi_12_64, cospi_20_64); const __m128i stg3_10 = pair_set_epi16(-cospi_12_64, -cospi_20_64); butterfly_self(&v17, &v30, &stg3_4, &stg3_5); butterfly_self(&v18, &v29, &stg3_6, &stg3_4); butterfly_self(&v21, &v26, &stg3_8, &stg3_9); butterfly_self(&v22, &v25, &stg3_10, &stg3_8); } u16 = _mm_add_epi16(v16, v19); u17 = _mm_add_epi16(v17, v18); u18 = _mm_sub_epi16(v17, v18); u19 = _mm_sub_epi16(v16, v19); u20 = _mm_sub_epi16(v23, v20); u21 = _mm_sub_epi16(v22, v21); u22 = _mm_add_epi16(v22, v21); u23 = _mm_add_epi16(v23, v20); u24 = _mm_add_epi16(v24, v27); u25 = _mm_add_epi16(v25, v26); u26 = _mm_sub_epi16(v25, v26); u27 = _mm_sub_epi16(v24, v27); u28 = _mm_sub_epi16(v31, v28); u29 = _mm_sub_epi16(v30, v29); u30 = _mm_add_epi16(v29, v30); u31 = _mm_add_epi16(v28, v31); { const __m128i stg4_4 = pair_set_epi16(-cospi_8_64, cospi_24_64); const __m128i stg4_5 = pair_set_epi16(cospi_24_64, cospi_8_64); const __m128i stg4_6 = pair_set_epi16(-cospi_24_64, -cospi_8_64); butterfly_self(&u18, &u29, &stg4_4, &stg4_5); butterfly_self(&u19, &u28, &stg4_4, &stg4_5); butterfly_self(&u20, &u27, &stg4_6, &stg4_4); butterfly_self(&u21, &u26, &stg4_6, &stg4_4); } out[0] = _mm_add_epi16(u16, u23); out[1] = _mm_add_epi16(u17, u22); out[2] = _mm_add_epi16(u18, u21); out[3] = _mm_add_epi16(u19, u20); out[4] = _mm_sub_epi16(u19, u20); out[5] = _mm_sub_epi16(u18, u21); out[6] = _mm_sub_epi16(u17, u22); out[7] = _mm_sub_epi16(u16, u23); out[8] = _mm_sub_epi16(u31, u24); out[9] = _mm_sub_epi16(u30, u25); out[10] = _mm_sub_epi16(u29, u26); out[11] = _mm_sub_epi16(u28, u27); out[12] = _mm_add_epi16(u27, u28); out[13] = _mm_add_epi16(u26, u29); out[14] = _mm_add_epi16(u25, u30); out[15] = _mm_add_epi16(u24, u31); { const __m128i stg4_0 = pair_set_epi16(cospi_16_64, cospi_16_64); const __m128i stg6_0 = pair_set_epi16(-cospi_16_64, cospi_16_64); butterfly_self(&out[4], &out[11], &stg6_0, &stg4_0); butterfly_self(&out[5], &out[10], &stg6_0, &stg4_0); butterfly_self(&out[6], &out[9], &stg6_0, &stg4_0); butterfly_self(&out[7], &out[8], &stg6_0, &stg4_0); } } static void idct32_full_8x32_quarter_1_2(const __m128i *in /*in[32]*/, __m128i *out /*out[32]*/) { __m128i temp[16]; idct32_full_8x32_quarter_1(in, temp); idct32_full_8x32_quarter_2(in, &temp[8]); add_sub_butterfly(temp, out, 16); } static void idct32_full_8x32(const __m128i *in /*in[32]*/, __m128i *out /*out[32]*/) { __m128i temp[32]; idct32_full_8x32_quarter_1_2(in, temp); idct32_full_8x32_quarter_3_4(in, &temp[16]); add_sub_butterfly(temp, out, 32); } static void load_buffer_8x32(const tran_low_t *input, __m128i *in) { int i; for (i = 0; i < 8; ++i) { in[i] = load_input_data8(input); in[i + 8] = load_input_data8(input + 8); in[i + 16] = load_input_data8(input + 16); in[i + 24] = load_input_data8(input + 24); input += 32; } } void vpx_idct32x32_1024_add_sse2(const tran_low_t *input, uint8_t *dest, int stride) { __m128i col[128], in[32]; int i, j; // rows for (i = 0; i < 4; ++i) { load_buffer_8x32(input, in); input += 32 << 3; // Transpose 32x8 block to 8x32 block transpose_16bit_8x8(in, in); transpose_16bit_8x8(in + 8, in + 8); transpose_16bit_8x8(in + 16, in + 16); transpose_16bit_8x8(in + 24, in + 24); idct32_full_8x32(in, col + (i << 5)); } // columns for (i = 0; i < 4; ++i) { j = i << 3; // Transpose 32x8 block to 8x32 block transpose_16bit_8x8(col + j, in); transpose_16bit_8x8(col + j + 32, in + 8); transpose_16bit_8x8(col + j + 64, in + 16); transpose_16bit_8x8(col + j + 96, in + 24); idct32_full_8x32(in, in); store_buffer_8x32(in, dest, stride); dest += 8; } } void vpx_idct32x32_1_add_sse2(const tran_low_t *input, uint8_t *dest, int stride) { __m128i dc_value; int j; tran_high_t a1; tran_low_t out = WRAPLOW(dct_const_round_shift(input[0] * cospi_16_64)); out = WRAPLOW(dct_const_round_shift(out * cospi_16_64)); a1 = ROUND_POWER_OF_TWO(out, 6); dc_value = _mm_set1_epi16(a1); for (j = 0; j < 32; ++j) { recon_and_store_16(dest + j * stride + 0, dc_value); recon_and_store_16(dest + j * stride + 16, dc_value); } }