/*

 *  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 <emmintrin.h>  // SSE2

#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((int16_t)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 *const 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 *const in) {

  const __m128i k__sinpi_1_3 = pair_set_epi16(sinpi_1_9, sinpi_3_9);

  const __m128i k__sinpi_4_2 = pair_set_epi16(sinpi_4_9, sinpi_2_9);

  const __m128i k__sinpi_2_3 = pair_set_epi16(sinpi_2_9, sinpi_3_9);

  const __m128i k__sinpi_1_4 = pair_set_epi16(sinpi_1_9, sinpi_4_9);

  const __m128i k__sinpi_12_n3 =

      pair_set_epi16(sinpi_1_9 + sinpi_2_9, -sinpi_3_9);

  __m128i u[4], v[5];

  // 00 01 20 21  02 03 22 23

  // 10 11 30 31  12 13 32 33

  const __m128i tr0_0 = _mm_unpacklo_epi32(in[0], in[1]);

  const __m128i tr0_1 = _mm_unpackhi_epi32(in[0], in[1]);

  // 00 01 10 11  20 21 30 31

  // 02 03 12 13  22 23 32 33

  in[0] = _mm_unpacklo_epi32(tr0_0, tr0_1);

  in[1] = _mm_unpackhi_epi32(tr0_0, tr0_1);

  v[0] = _mm_madd_epi16(in[0], k__sinpi_1_3);    // s_1 * x0 + s_3 * x1

  v[1] = _mm_madd_epi16(in[1], k__sinpi_4_2);    // s_4 * x2 + s_2 * x3

  v[2] = _mm_madd_epi16(in[0], k__sinpi_2_3);    // s_2 * x0 + s_3 * x1

  v[3] = _mm_madd_epi16(in[1], k__sinpi_1_4);    // s_1 * x2 + s_4 * x3

  v[4] = _mm_madd_epi16(in[0], k__sinpi_12_n3);  // (s_1 + s_2) * x0 - s_3 * x1

  in[0] = _mm_sub_epi16(in[0], in[1]);           // x0 - x2

  in[1] = _mm_srli_epi32(in[1], 16);

  in[0] = _mm_add_epi16(in[0], in[1]);

  in[0] = _mm_slli_epi32(in[0], 16);  // x0 - x2 + x3

  u[0] = _mm_add_epi32(v[0], v[1]);

  u[1] = _mm_sub_epi32(v[2], v[3]);

  u[2] = _mm_madd_epi16(in[0], k__sinpi_1_3);

  u[3] = _mm_sub_epi32(v[1], v[3]);

  u[3] = _mm_add_epi32(u[3], v[4]);

  u[0] = dct_const_round_shift_sse2(u[0]);

  u[1] = dct_const_round_shift_sse2(u[1]);

  u[2] = dct_const_round_shift_sse2(u[2]);

  u[3] = dct_const_round_shift_sse2(u[3]);

  in[0] = _mm_packs_epi32(u[0], u[1]);

  in[1] = _mm_packs_epi32(u[2], u[3]);

}

static INLINE void load_buffer_8x8(const tran_low_t *const input,

                                   __m128i *const in) {

  in[0] = load_input_data8(input + 0 * 8);

  in[1] = load_input_data8(input + 1 * 8);

  in[2] = load_input_data8(input + 2 * 8);

  in[3] = load_input_data8(input + 3 * 8);

  in[4] = load_input_data8(input + 4 * 8);

  in[5] = load_input_data8(input + 5 * 8);

  in[6] = load_input_data8(input + 6 * 8);

  in[7] = load_input_data8(input + 7 * 8);

}

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++) {

    vpx_idct8_sse2(in);

  }

  write_buffer_8x8(in, dest, stride);

}

void vpx_idct8x8_12_add_sse2(const tran_low_t *input, uint8_t *dest,

                             int stride) {

  __m128i io[8];

  io[0] = load_input_data4(input + 0 * 8);

  io[1] = load_input_data4(input + 1 * 8);

  io[2] = load_input_data4(input + 2 * 8);

  io[3] = load_input_data4(input + 3 * 8);

  idct8x8_12_add_kernel_sse2(io);

  write_buffer_8x8(io, 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((int16_t)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((int16_t)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 vpx_idct8_sse2(__m128i *const 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 *const 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(cospi_16_64);

  const __m128i kZero = _mm_setzero_si128();

  __m128i s[8], u[16], v[8], w[16];

  // transpose

  transpose_16bit_8x8(in, in);

  // column transformation

  // stage 1

  // interleave and multiply/add into 32-bit integer

  s[0] = _mm_unpacklo_epi16(in[7], in[0]);

  s[1] = _mm_unpackhi_epi16(in[7], in[0]);

  s[2] = _mm_unpacklo_epi16(in[5], in[2]);

  s[3] = _mm_unpackhi_epi16(in[5], in[2]);

  s[4] = _mm_unpacklo_epi16(in[3], in[4]);

  s[5] = _mm_unpackhi_epi16(in[3], in[4]);

  s[6] = _mm_unpacklo_epi16(in[1], in[6]);

  s[7] = _mm_unpackhi_epi16(in[1], in[6]);

  u[0] = _mm_madd_epi16(s[0], k__cospi_p02_p30);

  u[1] = _mm_madd_epi16(s[1], k__cospi_p02_p30);

  u[2] = _mm_madd_epi16(s[0], k__cospi_p30_m02);

  u[3] = _mm_madd_epi16(s[1], k__cospi_p30_m02);

  u[4] = _mm_madd_epi16(s[2], k__cospi_p10_p22);

  u[5] = _mm_madd_epi16(s[3], k__cospi_p10_p22);

  u[6] = _mm_madd_epi16(s[2], k__cospi_p22_m10);

  u[7] = _mm_madd_epi16(s[3], k__cospi_p22_m10);

  u[8] = _mm_madd_epi16(s[4], k__cospi_p18_p14);

  u[9] = _mm_madd_epi16(s[5], k__cospi_p18_p14);

  u[10] = _mm_madd_epi16(s[4], k__cospi_p14_m18);

  u[11] = _mm_madd_epi16(s[5], k__cospi_p14_m18);

  u[12] = _mm_madd_epi16(s[6], k__cospi_p26_p06);

  u[13] = _mm_madd_epi16(s[7], k__cospi_p26_p06);

  u[14] = _mm_madd_epi16(s[6], k__cospi_p06_m26);

  u[15] = _mm_madd_epi16(s[7], k__cospi_p06_m26);

  // addition

  w[0] = _mm_add_epi32(u[0], u[8]);

  w[1] = _mm_add_epi32(u[1], u[9]);

  w[2] = _mm_add_epi32(u[2], u[10]);

  w[3] = _mm_add_epi32(u[3], u[11]);

  w[4] = _mm_add_epi32(u[4], u[12]);

  w[5] = _mm_add_epi32(u[5], u[13]);

  w[6] = _mm_add_epi32(u[6], u[14]);

  w[7] = _mm_add_epi32(u[7], u[15]);

  w[8] = _mm_sub_epi32(u[0], u[8]);

  w[9] = _mm_sub_epi32(u[1], u[9]);

  w[10] = _mm_sub_epi32(u[2], u[10]);

  w[11] = _mm_sub_epi32(u[3], u[11]);

  w[12] = _mm_sub_epi32(u[4], u[12]);

  w[13] = _mm_sub_epi32(u[5], u[13]);

  w[14] = _mm_sub_epi32(u[6], u[14]);

  w[15] = _mm_sub_epi32(u[7], u[15]);

  // shift and rounding

  u[0] = dct_const_round_shift_sse2(w[0]);

  u[1] = dct_const_round_shift_sse2(w[1]);

  u[2] = dct_const_round_shift_sse2(w[2]);

  u[3] = dct_const_round_shift_sse2(w[3]);

  u[4] = dct_const_round_shift_sse2(w[4]);

  u[5] = dct_const_round_shift_sse2(w[5]);

  u[6] = dct_const_round_shift_sse2(w[6]);

  u[7] = dct_const_round_shift_sse2(w[7]);

  u[8] = dct_const_round_shift_sse2(w[8]);

  u[9] = dct_const_round_shift_sse2(w[9]);

  u[10] = dct_const_round_shift_sse2(w[10]);

  u[11] = dct_const_round_shift_sse2(w[11]);

  u[12] = dct_const_round_shift_sse2(w[12]);

  u[13] = dct_const_round_shift_sse2(w[13]);

  u[14] = dct_const_round_shift_sse2(w[14]);

  u[15] = dct_const_round_shift_sse2(w[15]);

  // back to 16-bit and pack 8 integers into __m128i

  in[0] = _mm_packs_epi32(u[0], u[1]);

  in[1] = _mm_packs_epi32(u[2], u[3]);

  in[2] = _mm_packs_epi32(u[4], u[5]);

  in[3] = _mm_packs_epi32(u[6], u[7]);

  in[4] = _mm_packs_epi32(u[8], u[9]);

  in[5] = _mm_packs_epi32(u[10], u[11]);

  in[6] = _mm_packs_epi32(u[12], u[13]);

  in[7] = _mm_packs_epi32(u[14], u[15]);

  // stage 2

  s[0] = _mm_add_epi16(in[0], in[2]);

  s[1] = _mm_add_epi16(in[1], in[3]);

  s[2] = _mm_sub_epi16(in[0], in[2]);

  s[3] = _mm_sub_epi16(in[1], in[3]);

  u[0] = _mm_unpacklo_epi16(in[4], in[5]);

  u[1] = _mm_unpackhi_epi16(in[4], in[5]);

  u[2] = _mm_unpacklo_epi16(in[6], in[7]);

  u[3] = _mm_unpackhi_epi16(in[6], in[7]);

  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);

  w[0] = _mm_add_epi32(v[0], v[4]);

  w[1] = _mm_add_epi32(v[1], v[5]);

  w[2] = _mm_add_epi32(v[2], v[6]);

  w[3] = _mm_add_epi32(v[3], v[7]);

  w[4] = _mm_sub_epi32(v[0], v[4]);

  w[5] = _mm_sub_epi32(v[1], v[5]);

  w[6] = _mm_sub_epi32(v[2], v[6]);

  w[7] = _mm_sub_epi32(v[3], v[7]);

  u[0] = dct_const_round_shift_sse2(w[0]);

  u[1] = dct_const_round_shift_sse2(w[1]);

  u[2] = dct_const_round_shift_sse2(w[2]);

  u[3] = dct_const_round_shift_sse2(w[3]);

  u[4] = dct_const_round_shift_sse2(w[4]);

  u[5] = dct_const_round_shift_sse2(w[5]);

  u[6] = dct_const_round_shift_sse2(w[6]);

  u[7] = dct_const_round_shift_sse2(w[7]);

  // back to 16-bit intergers

  s[4] = _mm_packs_epi32(u[0], u[1]);

  s[5] = _mm_packs_epi32(u[2], u[3]);

  s[6] = _mm_packs_epi32(u[4], u[5]);

  s[7] = _mm_packs_epi32(u[6], u[7]);

  // stage 3

  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]);

  s[2] = idct_calc_wraplow_sse2(u[0], u[1], k__cospi_p16_p16);

  s[3] = idct_calc_wraplow_sse2(u[0], u[1], k__cospi_p16_m16);

  s[6] = idct_calc_wraplow_sse2(u[2], u[3], k__cospi_p16_p16);

  s[7] = idct_calc_wraplow_sse2(u[2], u[3], k__cospi_p16_m16);

  in[0] = s[0];

  in[1] = _mm_sub_epi16(kZero, s[4]);

  in[2] = s[6];

  in[3] = _mm_sub_epi16(kZero, s[2]);

  in[4] = s[3];

  in[5] = _mm_sub_epi16(kZero, s[7]);

  in[6] = s[5];

  in[7] = _mm_sub_epi16(kZero, s[1]);

}

static INLINE void idct16_load8x8(const tran_low_t *const input,

                                  __m128i *const in) {

  in[0] = load_input_data8(input + 0 * 16);

  in[1] = load_input_data8(input + 1 * 16);

  in[2] = load_input_data8(input + 2 * 16);

  in[3] = load_input_data8(input + 3 * 16);

  in[4] = load_input_data8(input + 4 * 16);

  in[5] = load_input_data8(input + 5 * 16);

  in[6] = load_input_data8(input + 6 * 16);

  in[7] = load_input_data8(input + 7 * 16);

}

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);

    in = r;

    input += 128;

  }

  for (i = 0; i < 16; i += 8) {

    int j;

    transpose_16bit_8x8(l + i, out);

    transpose_16bit_8x8(r + i, out + 8);

    idct16_8col(out, out);

    for (j = 0; j < 16; ++j) {

      write_buffer_8x1(dest + j * stride, out[j]);

    }

    dest += 8;

  }

}

void vpx_idct16x16_38_add_sse2(const tran_low_t *input, uint8_t *dest,

                               int stride) {

  __m128i in[16], temp[16], out[16];

  int i;

  idct16_load8x8(input, in);

  transpose_16bit_8x8(in, in);

  for (i = 8; i < 16; i++) {

    in[i] = _mm_setzero_si128();

  }

  idct16_8col(in, temp);

  for (i = 0; i < 16; i += 8) {

    int j;

    transpose_16bit_8x8(temp + i, in);

    idct16_8col(in, out);

    for (j = 0; j < 16; ++j) {

      write_buffer_8x1(dest + j * stride, out[j]);

    }

    dest += 8;

  }

}

void vpx_idct16x16_10_add_sse2(const tran_low_t *input, uint8_t *dest,

                               int stride) {

  __m128i in[16], l[16];

  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);

  idct16x16_10_pass1(in, l);

  // Second 1-D inverse transform, performed per 8x16 block

  for (i = 0; i < 16; i += 8) {

    int j;

    idct16x16_10_pass2(l + i, in);

    for (j = 0; j < 16; ++j) {

      write_buffer_8x1(dest + j * stride, in[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((int16_t)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((int16_t)a1);

  for (i = 0; i < 16; ++i) {

    recon_and_store_16(dest, dc_value);

    dest += stride;

  }

}

void vpx_iadst16_8col_sse2(__m128i *const 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(-cospi_16_64);

  const __m128i k__cospi_p16_p16 = _mm_set1_epi16(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 kZero = _mm_setzero_si128();

  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]);

  u[0] = dct_const_round_shift_sse2(u[0]);

  u[1] = dct_const_round_shift_sse2(u[1]);

  u[2] = dct_const_round_shift_sse2(u[2]);

  u[3] = dct_const_round_shift_sse2(u[3]);

  u[4] = dct_const_round_shift_sse2(u[4]);

  u[5] = dct_const_round_shift_sse2(u[5]);

  u[6] = dct_const_round_shift_sse2(u[6]);

  u[7] = dct_const_round_shift_sse2(u[7]);

  u[8] = dct_const_round_shift_sse2(u[8]);

  u[9] = dct_const_round_shift_sse2(u[9]);

  u[10] = dct_const_round_shift_sse2(u[10]);

  u[11] = dct_const_round_shift_sse2(u[11]);

  u[12] = dct_const_round_shift_sse2(u[12]);

  u[13] = dct_const_round_shift_sse2(u[13]);

  u[14] = dct_const_round_shift_sse2(u[14]);

  u[15] = dct_const_round_shift_sse2(u[15]);

  u[16] = dct_const_round_shift_sse2(u[16]);

  u[17] = dct_const_round_shift_sse2(u[17]);

  u[18] = dct_const_round_shift_sse2(u[18]);

  u[19] = dct_const_round_shift_sse2(u[19]);

  u[20] = dct_const_round_shift_sse2(u[20]);

  u[21] = dct_const_round_shift_sse2(u[21]);

  u[22] = dct_const_round_shift_sse2(u[22]);

  u[23] = dct_const_round_shift_sse2(u[23]);

  u[24] = dct_const_round_shift_sse2(u[24]);

  u[25] = dct_const_round_shift_sse2(u[25]);

  u[26] = dct_const_round_shift_sse2(u[26]);

  u[27] = dct_const_round_shift_sse2(u[27]);

  u[28] = dct_const_round_shift_sse2(u[28]);

  u[29] = dct_const_round_shift_sse2(u[29]);

  u[30] = dct_const_round_shift_sse2(u[30]);

  u[31] = dct_const_round_shift_sse2(u[31]);

  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]);

  u[0] = dct_const_round_shift_sse2(u[0]);

  u[1] = dct_const_round_shift_sse2(u[1]);

  u[2] = dct_const_round_shift_sse2(u[2]);

  u[3] = dct_const_round_shift_sse2(u[3]);

  u[4] = dct_const_round_shift_sse2(u[4]);

  u[5] = dct_const_round_shift_sse2(u[5]);

  u[6] = dct_const_round_shift_sse2(u[6]);

  u[7] = dct_const_round_shift_sse2(u[7]);

  u[8] = dct_const_round_shift_sse2(u[8]);

  u[9] = dct_const_round_shift_sse2(u[9]);

  u[10] = dct_const_round_shift_sse2(u[10]);

  u[11] = dct_const_round_shift_sse2(u[11]);

  u[12] = dct_const_round_shift_sse2(u[12]);

  u[13] = dct_const_round_shift_sse2(u[13]);

  u[14] = dct_const_round_shift_sse2(u[14]);

  u[15] = dct_const_round_shift_sse2(u[15]);

  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]);

  v[0] = dct_const_round_shift_sse2(u[0]);

  v[1] = dct_const_round_shift_sse2(u[1]);

  v[2] = dct_const_round_shift_sse2(u[2]);

  v[3] = dct_const_round_shift_sse2(u[3]);

  v[4] = dct_const_round_shift_sse2(u[4]);

  v[5] = dct_const_round_shift_sse2(u[5]);

  v[6] = dct_const_round_shift_sse2(u[6]);

  v[7] = dct_const_round_shift_sse2(u[7]);

  v[8] = dct_const_round_shift_sse2(u[8]);

  v[9] = dct_const_round_shift_sse2(u[9]);

  v[10] = dct_const_round_shift_sse2(u[10]);

  v[11] = dct_const_round_shift_sse2(u[11]);

  v[12] = dct_const_round_shift_sse2(u[12]);

  v[13] = dct_const_round_shift_sse2(u[13]);

  v[14] = dct_const_round_shift_sse2(u[14]);

  v[15] = dct_const_round_shift_sse2(u[15]);

  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 *const in0, __m128i *const in1) {

  transpose_16bit_16x16(in0, in1);

  idct16_8col(in0, in0);

  idct16_8col(in1, in1);

}

void iadst16_sse2(__m128i *const in0, __m128i *const in1) {

  transpose_16bit_16x16(in0, in1);

  vpx_iadst16_8col_sse2(in0);

  vpx_iadst16_8col_sse2(in1);

}

// Group the coefficient calculation into smaller functions to prevent stack

// spillover in 32x32 idct optimizations:

// quarter_1: 0-7

// quarter_2: 8-15

// quarter_3_4: 16-23, 24-31

// For each 8x32 block __m128i in[32],

// Input with index, 0, 4

// output pixels: 0-7 in __m128i out[32]

static INLINE void idct32_34_8x32_quarter_1(const __m128i *const in /*in[32]*/,

                                            __m128i *const out /*out[8]*/) {

  const __m128i zero = _mm_setzero_si128();

  __m128i step1[8], step2[8];

  // stage 3

  butterfly(in[4], zero, cospi_28_64, cospi_4_64, &step1[4], &step1[7]);

  // stage 4

  step2[0] = butterfly_cospi16(in[0]);

  step2[4] = step1[4];

  step2[5] = step1[4];

  step2[6] = step1[7];

  step2[7] = step1[7];

  // stage 5

  step1[0] = step2[0];

  step1[1] = step2[0];

  step1[2] = step2[0];

  step1[3] = step2[0];

  step1[4] = step2[4];

  butterfly(step2[6], step2[5], cospi_16_64, cospi_16_64, &step1[5], &step1[6]);

  step1[7] = step2[7];

  // stage 6

  out[0] = _mm_add_epi16(step1[0], step1[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], step1[4]);

  out[4] = _mm_sub_epi16(step1[3], step1[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], step1[7]);

}

// For each 8x32 block __m128i in[32],

// Input with index, 2, 6

// output pixels: 8-15 in __m128i out[32]

static INLINE void idct32_34_8x32_quarter_2(const __m128i *const in /*in[32]*/,

                                            __m128i *const out /*out[16]*/) {

  const __m128i zero = _mm_setzero_si128();

  __m128i step1[16], step2[16];

  // stage 2

  butterfly(in[2], zero, cospi_30_64, cospi_2_64, &step2[8], &step2[15]);

  butterfly(zero, in[6], cospi_6_64, cospi_26_64, &step2[11], &step2[12]);

  // stage 3

  step1[8] = step2[8];

  step1[9] = step2[8];

  step1[14] = step2[15];

  step1[15] = step2[15];

  step1[10] = step2[11];

  step1[11] = step2[11];

  step1[12] = step2[12];

  step1[13] = step2[12];

  idct32_8x32_quarter_2_stage_4_to_6(step1, out);

}

static INLINE void idct32_34_8x32_quarter_1_2(

    const __m128i *const in /*in[32]*/, __m128i *const out /*out[32]*/) {

  __m128i temp[16];

  idct32_34_8x32_quarter_1(in, temp);

  idct32_34_8x32_quarter_2(in, temp);

  // stage 7

  add_sub_butterfly(temp, out, 16);

}

// For each 8x32 block __m128i in[32],

// Input with odd index, 1, 3, 5, 7

// output pixels: 16-23, 24-31 in __m128i out[32]

static INLINE void idct32_34_8x32_quarter_3_4(

    const __m128i *const in /*in[32]*/, __m128i *const out /*out[32]*/) {

  const __m128i zero = _mm_setzero_si128();

  __m128i step1[32];

  // stage 1

  butterfly(in[1], zero, cospi_31_64, cospi_1_64, &step1[16], &step1[31]);

  butterfly(zero, in[7], cospi_7_64, cospi_25_64, &step1[19], &step1[28]);

  butterfly(in[5], zero, cospi_27_64, cospi_5_64, &step1[20], &step1[27]);

  butterfly(zero, in[3], cospi_3_64, cospi_29_64, &step1[23], &step1[24]);

  // stage 3

  butterfly(step1[31], step1[16], cospi_28_64, cospi_4_64, &step1[17],

            &step1[30]);

  butterfly(step1[28], step1[19], -cospi_4_64, cospi_28_64, &step1[18],

            &step1[29]);

  butterfly(step1[27], step1[20], cospi_12_64, cospi_20_64, &step1[21],

            &step1[26]);

  butterfly(step1[24], step1[23], -cospi_20_64, cospi_12_64, &step1[22],

            &step1[25]);

  idct32_8x32_quarter_3_4_stage_4_to_7(step1, out);

}

void idct32_34_8x32_sse2(const __m128i *const in /*in[32]*/,

                         __m128i *const out /*out[32]*/) {

  __m128i temp[32];

  idct32_34_8x32_quarter_1_2(in, temp);

  idct32_34_8x32_quarter_3_4(in, temp);

  // final stage

  add_sub_butterfly(temp, out, 32);

}

// Only upper-left 8x8 has non-zero coeff

void vpx_idct32x32_34_add_sse2(const tran_low_t *input, uint8_t *dest,

                               int stride) {

  __m128i io[32], col[32];

  int i;

  // Load input data. Only need to load the top left 8x8 block.

  load_transpose_16bit_8x8(input, 32, io);

  idct32_34_8x32_sse2(io, col);