/*

 * Copyright (c) 2016, Alliance for Open Media. All rights reserved

 *

 * This source code is subject to the terms of the BSD 2 Clause License and

 * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License

 * was not distributed with this source code in the LICENSE file, you can

 * obtain it at www.aomedia.org/license/software. If the Alliance for Open

 * Media Patent License 1.0 was not distributed with this source code in the

 * PATENTS file, you can obtain it at www.aomedia.org/license/patent.

 */

#include <stdlib.h>

#include "av1/encoder/av1_fwd_txfm1d.h"

#include "av1/common/av1_txfm.h"

void av1_fdct4(const int32_t *input, int32_t *output, int8_t cos_bit,

               const int8_t *stage_range) {

  const int32_t size = 4;

  const int32_t *cospi;

  int32_t stage = 0;

  int32_t *bf0, *bf1;

  int32_t step[4];

  // stage 0;

  av1_range_check_buf(stage, input, input, size, stage_range[stage]);

  // stage 1;

  stage++;

  bf1 = output;

  bf1[0] = input[0] + input[3];

  bf1[1] = input[1] + input[2];

  bf1[2] = -input[2] + input[1];

  bf1[3] = -input[3] + input[0];

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

  // stage 2

  stage++;

  cospi = cospi_arr(cos_bit);

  bf0 = output;

  bf1 = step;

  bf1[0] = half_btf(cospi[32], bf0[0], cospi[32], bf0[1], cos_bit);

  bf1[1] = half_btf(-cospi[32], bf0[1], cospi[32], bf0[0], cos_bit);

  bf1[2] = half_btf(cospi[48], bf0[2], cospi[16], bf0[3], cos_bit);

  bf1[3] = half_btf(cospi[48], bf0[3], -cospi[16], bf0[2], cos_bit);

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

  // stage 3

  stage++;

  bf0 = step;

  bf1 = output;

  bf1[0] = bf0[0];

  bf1[1] = bf0[2];

  bf1[2] = bf0[1];

  bf1[3] = bf0[3];

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

}

void av1_fdct8(const int32_t *input, int32_t *output, int8_t cos_bit,

               const int8_t *stage_range) {

  const int32_t size = 8;

  const int32_t *cospi;

  int32_t stage = 0;

  int32_t *bf0, *bf1;

  int32_t step[8];

  // stage 0;

  av1_range_check_buf(stage, input, input, size, stage_range[stage]);

  // stage 1;

  stage++;

  bf1 = output;

  bf1[0] = input[0] + input[7];

  bf1[1] = input[1] + input[6];

  bf1[2] = input[2] + input[5];

  bf1[3] = input[3] + input[4];

  bf1[4] = -input[4] + input[3];

  bf1[5] = -input[5] + input[2];

  bf1[6] = -input[6] + input[1];

  bf1[7] = -input[7] + input[0];

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

  // stage 2

  stage++;

  cospi = cospi_arr(cos_bit);

  bf0 = output;

  bf1 = step;

  bf1[0] = bf0[0] + bf0[3];

  bf1[1] = bf0[1] + bf0[2];

  bf1[2] = -bf0[2] + bf0[1];

  bf1[3] = -bf0[3] + bf0[0];

  bf1[4] = bf0[4];

  bf1[5] = half_btf(-cospi[32], bf0[5], cospi[32], bf0[6], cos_bit);

  bf1[6] = half_btf(cospi[32], bf0[6], cospi[32], bf0[5], cos_bit);

  bf1[7] = bf0[7];

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

  // stage 3

  stage++;

  cospi = cospi_arr(cos_bit);

  bf0 = step;

  bf1 = output;

  bf1[0] = half_btf(cospi[32], bf0[0], cospi[32], bf0[1], cos_bit);

  bf1[1] = half_btf(-cospi[32], bf0[1], cospi[32], bf0[0], cos_bit);

  bf1[2] = half_btf(cospi[48], bf0[2], cospi[16], bf0[3], cos_bit);

  bf1[3] = half_btf(cospi[48], bf0[3], -cospi[16], bf0[2], cos_bit);

  bf1[4] = bf0[4] + bf0[5];

  bf1[5] = -bf0[5] + bf0[4];

  bf1[6] = -bf0[6] + bf0[7];

  bf1[7] = bf0[7] + bf0[6];

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

  // stage 4

  stage++;

  cospi = cospi_arr(cos_bit);

  bf0 = output;

  bf1 = step;

  bf1[0] = bf0[0];

  bf1[1] = bf0[1];

  bf1[2] = bf0[2];

  bf1[3] = bf0[3];

  bf1[4] = half_btf(cospi[56], bf0[4], cospi[8], bf0[7], cos_bit);

  bf1[5] = half_btf(cospi[24], bf0[5], cospi[40], bf0[6], cos_bit);

  bf1[6] = half_btf(cospi[24], bf0[6], -cospi[40], bf0[5], cos_bit);

  bf1[7] = half_btf(cospi[56], bf0[7], -cospi[8], bf0[4], cos_bit);

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

  // stage 5

  stage++;

  bf0 = step;

  bf1 = output;

  bf1[0] = bf0[0];

  bf1[1] = bf0[4];

  bf1[2] = bf0[2];

  bf1[3] = bf0[6];

  bf1[4] = bf0[1];

  bf1[5] = bf0[5];

  bf1[6] = bf0[3];

  bf1[7] = bf0[7];

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

}

void av1_fdct16(const int32_t *input, int32_t *output, int8_t cos_bit,

                const int8_t *stage_range) {

  const int32_t size = 16;

  const int32_t *cospi;

  int32_t stage = 0;

  int32_t *bf0, *bf1;

  int32_t step[16];

  // stage 0;

  av1_range_check_buf(stage, input, input, size, stage_range[stage]);

  // stage 1;

  stage++;

  bf1 = output;

  bf1[0] = input[0] + input[15];

  bf1[1] = input[1] + input[14];

  bf1[2] = input[2] + input[13];

  bf1[3] = input[3] + input[12];

  bf1[4] = input[4] + input[11];

  bf1[5] = input[5] + input[10];

  bf1[6] = input[6] + input[9];

  bf1[7] = input[7] + input[8];

  bf1[8] = -input[8] + input[7];

  bf1[9] = -input[9] + input[6];

  bf1[10] = -input[10] + input[5];

  bf1[11] = -input[11] + input[4];

  bf1[12] = -input[12] + input[3];

  bf1[13] = -input[13] + input[2];

  bf1[14] = -input[14] + input[1];

  bf1[15] = -input[15] + input[0];

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

  // stage 2

  stage++;

  cospi = cospi_arr(cos_bit);

  bf0 = output;

  bf1 = step;

  bf1[0] = bf0[0] + bf0[7];

  bf1[1] = bf0[1] + bf0[6];

  bf1[2] = bf0[2] + bf0[5];

  bf1[3] = bf0[3] + bf0[4];

  bf1[4] = -bf0[4] + bf0[3];

  bf1[5] = -bf0[5] + bf0[2];

  bf1[6] = -bf0[6] + bf0[1];

  bf1[7] = -bf0[7] + bf0[0];

  bf1[8] = bf0[8];

  bf1[9] = bf0[9];

  bf1[10] = half_btf(-cospi[32], bf0[10], cospi[32], bf0[13], cos_bit);

  bf1[11] = half_btf(-cospi[32], bf0[11], cospi[32], bf0[12], cos_bit);

  bf1[12] = half_btf(cospi[32], bf0[12], cospi[32], bf0[11], cos_bit);

  bf1[13] = half_btf(cospi[32], bf0[13], cospi[32], bf0[10], cos_bit);

  bf1[14] = bf0[14];

  bf1[15] = bf0[15];

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

  // stage 3

  stage++;

  cospi = cospi_arr(cos_bit);

  bf0 = step;

  bf1 = output;

  bf1[0] = bf0[0] + bf0[3];

  bf1[1] = bf0[1] + bf0[2];

  bf1[2] = -bf0[2] + bf0[1];

  bf1[3] = -bf0[3] + bf0[0];

  bf1[4] = bf0[4];

  bf1[5] = half_btf(-cospi[32], bf0[5], cospi[32], bf0[6], cos_bit);

  bf1[6] = half_btf(cospi[32], bf0[6], cospi[32], bf0[5], cos_bit);

  bf1[7] = bf0[7];

  bf1[8] = bf0[8] + bf0[11];

  bf1[9] = bf0[9] + bf0[10];

  bf1[10] = -bf0[10] + bf0[9];

  bf1[11] = -bf0[11] + bf0[8];

  bf1[12] = -bf0[12] + bf0[15];

  bf1[13] = -bf0[13] + bf0[14];

  bf1[14] = bf0[14] + bf0[13];

  bf1[15] = bf0[15] + bf0[12];

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

  // stage 4

  stage++;

  cospi = cospi_arr(cos_bit);

  bf0 = output;

  bf1 = step;

  bf1[0] = half_btf(cospi[32], bf0[0], cospi[32], bf0[1], cos_bit);

  bf1[1] = half_btf(-cospi[32], bf0[1], cospi[32], bf0[0], cos_bit);

  bf1[2] = half_btf(cospi[48], bf0[2], cospi[16], bf0[3], cos_bit);

  bf1[3] = half_btf(cospi[48], bf0[3], -cospi[16], bf0[2], cos_bit);

  bf1[4] = bf0[4] + bf0[5];

  bf1[5] = -bf0[5] + bf0[4];

  bf1[6] = -bf0[6] + bf0[7];

  bf1[7] = bf0[7] + bf0[6];

  bf1[8] = bf0[8];

  bf1[9] = half_btf(-cospi[16], bf0[9], cospi[48], bf0[14], cos_bit);

  bf1[10] = half_btf(-cospi[48], bf0[10], -cospi[16], bf0[13], cos_bit);

  bf1[11] = bf0[11];

  bf1[12] = bf0[12];

  bf1[13] = half_btf(cospi[48], bf0[13], -cospi[16], bf0[10], cos_bit);

  bf1[14] = half_btf(cospi[16], bf0[14], cospi[48], bf0[9], cos_bit);

  bf1[15] = bf0[15];

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

  // stage 5

  stage++;

  cospi = cospi_arr(cos_bit);

  bf0 = step;

  bf1 = output;

  bf1[0] = bf0[0];

  bf1[1] = bf0[1];

  bf1[2] = bf0[2];

  bf1[3] = bf0[3];

  bf1[4] = half_btf(cospi[56], bf0[4], cospi[8], bf0[7], cos_bit);

  bf1[5] = half_btf(cospi[24], bf0[5], cospi[40], bf0[6], cos_bit);

  bf1[6] = half_btf(cospi[24], bf0[6], -cospi[40], bf0[5], cos_bit);

  bf1[7] = half_btf(cospi[56], bf0[7], -cospi[8], bf0[4], cos_bit);

  bf1[8] = bf0[8] + bf0[9];

  bf1[9] = -bf0[9] + bf0[8];

  bf1[10] = -bf0[10] + bf0[11];

  bf1[11] = bf0[11] + bf0[10];

  bf1[12] = bf0[12] + bf0[13];

  bf1[13] = -bf0[13] + bf0[12];

  bf1[14] = -bf0[14] + bf0[15];

  bf1[15] = bf0[15] + bf0[14];

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

  // stage 6

  stage++;

  cospi = cospi_arr(cos_bit);

  bf0 = output;

  bf1 = step;

  bf1[0] = bf0[0];

  bf1[1] = bf0[1];

  bf1[2] = bf0[2];

  bf1[3] = bf0[3];

  bf1[4] = bf0[4];

  bf1[5] = bf0[5];

  bf1[6] = bf0[6];

  bf1[7] = bf0[7];

  bf1[8] = half_btf(cospi[60], bf0[8], cospi[4], bf0[15], cos_bit);

  bf1[9] = half_btf(cospi[28], bf0[9], cospi[36], bf0[14], cos_bit);

  bf1[10] = half_btf(cospi[44], bf0[10], cospi[20], bf0[13], cos_bit);

  bf1[11] = half_btf(cospi[12], bf0[11], cospi[52], bf0[12], cos_bit);

  bf1[12] = half_btf(cospi[12], bf0[12], -cospi[52], bf0[11], cos_bit);

  bf1[13] = half_btf(cospi[44], bf0[13], -cospi[20], bf0[10], cos_bit);

  bf1[14] = half_btf(cospi[28], bf0[14], -cospi[36], bf0[9], cos_bit);

  bf1[15] = half_btf(cospi[60], bf0[15], -cospi[4], bf0[8], cos_bit);

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

  // stage 7

  stage++;

  bf0 = step;

  bf1 = output;

  bf1[0] = bf0[0];

  bf1[1] = bf0[8];

  bf1[2] = bf0[4];

  bf1[3] = bf0[12];

  bf1[4] = bf0[2];

  bf1[5] = bf0[10];

  bf1[6] = bf0[6];

  bf1[7] = bf0[14];

  bf1[8] = bf0[1];

  bf1[9] = bf0[9];

  bf1[10] = bf0[5];

  bf1[11] = bf0[13];

  bf1[12] = bf0[3];

  bf1[13] = bf0[11];

  bf1[14] = bf0[7];

  bf1[15] = bf0[15];

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

}

void av1_fdct32(const int32_t *input, int32_t *output, int8_t cos_bit,

                const int8_t *stage_range) {

  const int32_t size = 32;

  const int32_t *cospi;

  int32_t stage = 0;

  int32_t *bf0, *bf1;

  int32_t step[32];

  // stage 0;

  av1_range_check_buf(stage, input, input, size, stage_range[stage]);

  // stage 1;

  stage++;

  bf1 = output;

  bf1[0] = input[0] + input[31];

  bf1[1] = input[1] + input[30];

  bf1[2] = input[2] + input[29];

  bf1[3] = input[3] + input[28];

  bf1[4] = input[4] + input[27];

  bf1[5] = input[5] + input[26];

  bf1[6] = input[6] + input[25];

  bf1[7] = input[7] + input[24];

  bf1[8] = input[8] + input[23];

  bf1[9] = input[9] + input[22];

  bf1[10] = input[10] + input[21];

  bf1[11] = input[11] + input[20];

  bf1[12] = input[12] + input[19];

  bf1[13] = input[13] + input[18];

  bf1[14] = input[14] + input[17];

  bf1[15] = input[15] + input[16];

  bf1[16] = -input[16] + input[15];

  bf1[17] = -input[17] + input[14];

  bf1[18] = -input[18] + input[13];

  bf1[19] = -input[19] + input[12];

  bf1[20] = -input[20] + input[11];

  bf1[21] = -input[21] + input[10];

  bf1[22] = -input[22] + input[9];

  bf1[23] = -input[23] + input[8];

  bf1[24] = -input[24] + input[7];

  bf1[25] = -input[25] + input[6];

  bf1[26] = -input[26] + input[5];

  bf1[27] = -input[27] + input[4];

  bf1[28] = -input[28] + input[3];

  bf1[29] = -input[29] + input[2];

  bf1[30] = -input[30] + input[1];

  bf1[31] = -input[31] + input[0];

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

  // stage 2

  stage++;

  cospi = cospi_arr(cos_bit);

  bf0 = output;

  bf1 = step;

  bf1[0] = bf0[0] + bf0[15];

  bf1[1] = bf0[1] + bf0[14];

  bf1[2] = bf0[2] + bf0[13];

  bf1[3] = bf0[3] + bf0[12];

  bf1[4] = bf0[4] + bf0[11];

  bf1[5] = bf0[5] + bf0[10];

  bf1[6] = bf0[6] + bf0[9];

  bf1[7] = bf0[7] + bf0[8];

  bf1[8] = -bf0[8] + bf0[7];

  bf1[9] = -bf0[9] + bf0[6];

  bf1[10] = -bf0[10] + bf0[5];

  bf1[11] = -bf0[11] + bf0[4];

  bf1[12] = -bf0[12] + bf0[3];

  bf1[13] = -bf0[13] + bf0[2];

  bf1[14] = -bf0[14] + bf0[1];

  bf1[15] = -bf0[15] + bf0[0];

  bf1[16] = bf0[16];

  bf1[17] = bf0[17];

  bf1[18] = bf0[18];

  bf1[19] = bf0[19];

  bf1[20] = half_btf(-cospi[32], bf0[20], cospi[32], bf0[27], cos_bit);

  bf1[21] = half_btf(-cospi[32], bf0[21], cospi[32], bf0[26], cos_bit);

  bf1[22] = half_btf(-cospi[32], bf0[22], cospi[32], bf0[25], cos_bit);

  bf1[23] = half_btf(-cospi[32], bf0[23], cospi[32], bf0[24], cos_bit);

  bf1[24] = half_btf(cospi[32], bf0[24], cospi[32], bf0[23], cos_bit);

  bf1[25] = half_btf(cospi[32], bf0[25], cospi[32], bf0[22], cos_bit);

  bf1[26] = half_btf(cospi[32], bf0[26], cospi[32], bf0[21], cos_bit);

  bf1[27] = half_btf(cospi[32], bf0[27], cospi[32], bf0[20], cos_bit);

  bf1[28] = bf0[28];

  bf1[29] = bf0[29];

  bf1[30] = bf0[30];

  bf1[31] = bf0[31];

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

  // stage 3

  stage++;

  cospi = cospi_arr(cos_bit);

  bf0 = step;

  bf1 = output;

  bf1[0] = bf0[0] + bf0[7];

  bf1[1] = bf0[1] + bf0[6];

  bf1[2] = bf0[2] + bf0[5];

  bf1[3] = bf0[3] + bf0[4];

  bf1[4] = -bf0[4] + bf0[3];

  bf1[5] = -bf0[5] + bf0[2];

  bf1[6] = -bf0[6] + bf0[1];

  bf1[7] = -bf0[7] + bf0[0];

  bf1[8] = bf0[8];

  bf1[9] = bf0[9];

  bf1[10] = half_btf(-cospi[32], bf0[10], cospi[32], bf0[13], cos_bit);

  bf1[11] = half_btf(-cospi[32], bf0[11], cospi[32], bf0[12], cos_bit);

  bf1[12] = half_btf(cospi[32], bf0[12], cospi[32], bf0[11], cos_bit);

  bf1[13] = half_btf(cospi[32], bf0[13], cospi[32], bf0[10], cos_bit);

  bf1[14] = bf0[14];

  bf1[15] = bf0[15];

  bf1[16] = bf0[16] + bf0[23];

  bf1[17] = bf0[17] + bf0[22];

  bf1[18] = bf0[18] + bf0[21];

  bf1[19] = bf0[19] + bf0[20];

  bf1[20] = -bf0[20] + bf0[19];

  bf1[21] = -bf0[21] + bf0[18];

  bf1[22] = -bf0[22] + bf0[17];

  bf1[23] = -bf0[23] + bf0[16];

  bf1[24] = -bf0[24] + bf0[31];

  bf1[25] = -bf0[25] + bf0[30];

  bf1[26] = -bf0[26] + bf0[29];

  bf1[27] = -bf0[27] + bf0[28];

  bf1[28] = bf0[28] + bf0[27];

  bf1[29] = bf0[29] + bf0[26];

  bf1[30] = bf0[30] + bf0[25];

  bf1[31] = bf0[31] + bf0[24];

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

  // stage 4

  stage++;

  cospi = cospi_arr(cos_bit);

  bf0 = output;

  bf1 = step;

  bf1[0] = bf0[0] + bf0[3];

  bf1[1] = bf0[1] + bf0[2];

  bf1[2] = -bf0[2] + bf0[1];

  bf1[3] = -bf0[3] + bf0[0];

  bf1[4] = bf0[4];

  bf1[5] = half_btf(-cospi[32], bf0[5], cospi[32], bf0[6], cos_bit);

  bf1[6] = half_btf(cospi[32], bf0[6], cospi[32], bf0[5], cos_bit);

  bf1[7] = bf0[7];

  bf1[8] = bf0[8] + bf0[11];

  bf1[9] = bf0[9] + bf0[10];

  bf1[10] = -bf0[10] + bf0[9];

  bf1[11] = -bf0[11] + bf0[8];

  bf1[12] = -bf0[12] + bf0[15];

  bf1[13] = -bf0[13] + bf0[14];

  bf1[14] = bf0[14] + bf0[13];

  bf1[15] = bf0[15] + bf0[12];

  bf1[16] = bf0[16];

  bf1[17] = bf0[17];

  bf1[18] = half_btf(-cospi[16], bf0[18], cospi[48], bf0[29], cos_bit);

  bf1[19] = half_btf(-cospi[16], bf0[19], cospi[48], bf0[28], cos_bit);

  bf1[20] = half_btf(-cospi[48], bf0[20], -cospi[16], bf0[27], cos_bit);

  bf1[21] = half_btf(-cospi[48], bf0[21], -cospi[16], bf0[26], cos_bit);

  bf1[22] = bf0[22];

  bf1[23] = bf0[23];

  bf1[24] = bf0[24];

  bf1[25] = bf0[25];

  bf1[26] = half_btf(cospi[48], bf0[26], -cospi[16], bf0[21], cos_bit);

  bf1[27] = half_btf(cospi[48], bf0[27], -cospi[16], bf0[20], cos_bit);

  bf1[28] = half_btf(cospi[16], bf0[28], cospi[48], bf0[19], cos_bit);

  bf1[29] = half_btf(cospi[16], bf0[29], cospi[48], bf0[18], cos_bit);

  bf1[30] = bf0[30];

  bf1[31] = bf0[31];

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

  // stage 5

  stage++;

  cospi = cospi_arr(cos_bit);

  bf0 = step;

  bf1 = output;

  bf1[0] = half_btf(cospi[32], bf0[0], cospi[32], bf0[1], cos_bit);

  bf1[1] = half_btf(-cospi[32], bf0[1], cospi[32], bf0[0], cos_bit);

  bf1[2] = half_btf(cospi[48], bf0[2], cospi[16], bf0[3], cos_bit);

  bf1[3] = half_btf(cospi[48], bf0[3], -cospi[16], bf0[2], cos_bit);

  bf1[4] = bf0[4] + bf0[5];

  bf1[5] = -bf0[5] + bf0[4];

  bf1[6] = -bf0[6] + bf0[7];

  bf1[7] = bf0[7] + bf0[6];

  bf1[8] = bf0[8];

  bf1[9] = half_btf(-cospi[16], bf0[9], cospi[48], bf0[14], cos_bit);

  bf1[10] = half_btf(-cospi[48], bf0[10], -cospi[16], bf0[13], cos_bit);

  bf1[11] = bf0[11];

  bf1[12] = bf0[12];

  bf1[13] = half_btf(cospi[48], bf0[13], -cospi[16], bf0[10], cos_bit);

  bf1[14] = half_btf(cospi[16], bf0[14], cospi[48], bf0[9], cos_bit);

  bf1[15] = bf0[15];

  bf1[16] = bf0[16] + bf0[19];

  bf1[17] = bf0[17] + bf0[18];

  bf1[18] = -bf0[18] + bf0[17];

  bf1[19] = -bf0[19] + bf0[16];

  bf1[20] = -bf0[20] + bf0[23];

  bf1[21] = -bf0[21] + bf0[22];

  bf1[22] = bf0[22] + bf0[21];

  bf1[23] = bf0[23] + bf0[20];

  bf1[24] = bf0[24] + bf0[27];

  bf1[25] = bf0[25] + bf0[26];

  bf1[26] = -bf0[26] + bf0[25];

  bf1[27] = -bf0[27] + bf0[24];

  bf1[28] = -bf0[28] + bf0[31];

  bf1[29] = -bf0[29] + bf0[30];

  bf1[30] = bf0[30] + bf0[29];

  bf1[31] = bf0[31] + bf0[28];

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

  // stage 6

  stage++;

  cospi = cospi_arr(cos_bit);

  bf0 = output;

  bf1 = step;

  bf1[0] = bf0[0];

  bf1[1] = bf0[1];

  bf1[2] = bf0[2];

  bf1[3] = bf0[3];

  bf1[4] = half_btf(cospi[56], bf0[4], cospi[8], bf0[7], cos_bit);

  bf1[5] = half_btf(cospi[24], bf0[5], cospi[40], bf0[6], cos_bit);

  bf1[6] = half_btf(cospi[24], bf0[6], -cospi[40], bf0[5], cos_bit);

  bf1[7] = half_btf(cospi[56], bf0[7], -cospi[8], bf0[4], cos_bit);

  bf1[8] = bf0[8] + bf0[9];

  bf1[9] = -bf0[9] + bf0[8];

  bf1[10] = -bf0[10] + bf0[11];

  bf1[11] = bf0[11] + bf0[10];

  bf1[12] = bf0[12] + bf0[13];

  bf1[13] = -bf0[13] + bf0[12];

  bf1[14] = -bf0[14] + bf0[15];

  bf1[15] = bf0[15] + bf0[14];

  bf1[16] = bf0[16];

  bf1[17] = half_btf(-cospi[8], bf0[17], cospi[56], bf0[30], cos_bit);

  bf1[18] = half_btf(-cospi[56], bf0[18], -cospi[8], bf0[29], cos_bit);

  bf1[19] = bf0[19];

  bf1[20] = bf0[20];

  bf1[21] = half_btf(-cospi[40], bf0[21], cospi[24], bf0[26], cos_bit);

  bf1[22] = half_btf(-cospi[24], bf0[22], -cospi[40], bf0[25], cos_bit);

  bf1[23] = bf0[23];

  bf1[24] = bf0[24];

  bf1[25] = half_btf(cospi[24], bf0[25], -cospi[40], bf0[22], cos_bit);

  bf1[26] = half_btf(cospi[40], bf0[26], cospi[24], bf0[21], cos_bit);

  bf1[27] = bf0[27];

  bf1[28] = bf0[28];

  bf1[29] = half_btf(cospi[56], bf0[29], -cospi[8], bf0[18], cos_bit);

  bf1[30] = half_btf(cospi[8], bf0[30], cospi[56], bf0[17], cos_bit);

  bf1[31] = bf0[31];

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

  // stage 7

  stage++;

  cospi = cospi_arr(cos_bit);

  bf0 = step;

  bf1 = output;

  bf1[0] = bf0[0];

  bf1[1] = bf0[1];

  bf1[2] = bf0[2];

  bf1[3] = bf0[3];

  bf1[4] = bf0[4];

  bf1[5] = bf0[5];

  bf1[6] = bf0[6];

  bf1[7] = bf0[7];

  bf1[8] = half_btf(cospi[60], bf0[8], cospi[4], bf0[15], cos_bit);

  bf1[9] = half_btf(cospi[28], bf0[9], cospi[36], bf0[14], cos_bit);

  bf1[10] = half_btf(cospi[44], bf0[10], cospi[20], bf0[13], cos_bit);

  bf1[11] = half_btf(cospi[12], bf0[11], cospi[52], bf0[12], cos_bit);

  bf1[12] = half_btf(cospi[12], bf0[12], -cospi[52], bf0[11], cos_bit);

  bf1[13] = half_btf(cospi[44], bf0[13], -cospi[20], bf0[10], cos_bit);

  bf1[14] = half_btf(cospi[28], bf0[14], -cospi[36], bf0[9], cos_bit);

  bf1[15] = half_btf(cospi[60], bf0[15], -cospi[4], bf0[8], cos_bit);

  bf1[16] = bf0[16] + bf0[17];

  bf1[17] = -bf0[17] + bf0[16];

  bf1[18] = -bf0[18] + bf0[19];

  bf1[19] = bf0[19] + bf0[18];

  bf1[20] = bf0[20] + bf0[21];

  bf1[21] = -bf0[21] + bf0[20];

  bf1[22] = -bf0[22] + bf0[23];

  bf1[23] = bf0[23] + bf0[22];

  bf1[24] = bf0[24] + bf0[25];

  bf1[25] = -bf0[25] + bf0[24];

  bf1[26] = -bf0[26] + bf0[27];

  bf1[27] = bf0[27] + bf0[26];

  bf1[28] = bf0[28] + bf0[29];

  bf1[29] = -bf0[29] + bf0[28];

  bf1[30] = -bf0[30] + bf0[31];

  bf1[31] = bf0[31] + bf0[30];

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

  // stage 8

  stage++;

  cospi = cospi_arr(cos_bit);

  bf0 = output;

  bf1 = step;

  bf1[0] = bf0[0];

  bf1[1] = bf0[1];

  bf1[2] = bf0[2];

  bf1[3] = bf0[3];

  bf1[4] = bf0[4];

  bf1[5] = bf0[5];

  bf1[6] = bf0[6];

  bf1[7] = bf0[7];

  bf1[8] = bf0[8];

  bf1[9] = bf0[9];

  bf1[10] = bf0[10];

  bf1[11] = bf0[11];

  bf1[12] = bf0[12];

  bf1[13] = bf0[13];

  bf1[14] = bf0[14];

  bf1[15] = bf0[15];

  bf1[16] = half_btf(cospi[62], bf0[16], cospi[2], bf0[31], cos_bit);

  bf1[17] = half_btf(cospi[30], bf0[17], cospi[34], bf0[30], cos_bit);

  bf1[18] = half_btf(cospi[46], bf0[18], cospi[18], bf0[29], cos_bit);

  bf1[19] = half_btf(cospi[14], bf0[19], cospi[50], bf0[28], cos_bit);

  bf1[20] = half_btf(cospi[54], bf0[20], cospi[10], bf0[27], cos_bit);

  bf1[21] = half_btf(cospi[22], bf0[21], cospi[42], bf0[26], cos_bit);

  bf1[22] = half_btf(cospi[38], bf0[22], cospi[26], bf0[25], cos_bit);

  bf1[23] = half_btf(cospi[6], bf0[23], cospi[58], bf0[24], cos_bit);

  bf1[24] = half_btf(cospi[6], bf0[24], -cospi[58], bf0[23], cos_bit);

  bf1[25] = half_btf(cospi[38], bf0[25], -cospi[26], bf0[22], cos_bit);

  bf1[26] = half_btf(cospi[22], bf0[26], -cospi[42], bf0[21], cos_bit);

  bf1[27] = half_btf(cospi[54], bf0[27], -cospi[10], bf0[20], cos_bit);

  bf1[28] = half_btf(cospi[14], bf0[28], -cospi[50], bf0[19], cos_bit);

  bf1[29] = half_btf(cospi[46], bf0[29], -cospi[18], bf0[18], cos_bit);

  bf1[30] = half_btf(cospi[30], bf0[30], -cospi[34], bf0[17], cos_bit);

  bf1[31] = half_btf(cospi[62], bf0[31], -cospi[2], bf0[16], cos_bit);

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

  // stage 9

  stage++;

  bf0 = step;

  bf1 = output;

  bf1[0] = bf0[0];

  bf1[1] = bf0[16];

  bf1[2] = bf0[8];

  bf1[3] = bf0[24];

  bf1[4] = bf0[4];

  bf1[5] = bf0[20];

  bf1[6] = bf0[12];

  bf1[7] = bf0[28];

  bf1[8] = bf0[2];

  bf1[9] = bf0[18];

  bf1[10] = bf0[10];

  bf1[11] = bf0[26];

  bf1[12] = bf0[6];

  bf1[13] = bf0[22];

  bf1[14] = bf0[14];

  bf1[15] = bf0[30];

  bf1[16] = bf0[1];

  bf1[17] = bf0[17];

  bf1[18] = bf0[9];

  bf1[19] = bf0[25];

  bf1[20] = bf0[5];

  bf1[21] = bf0[21];

  bf1[22] = bf0[13];

  bf1[23] = bf0[29];

  bf1[24] = bf0[3];

  bf1[25] = bf0[19];

  bf1[26] = bf0[11];

  bf1[27] = bf0[27];

  bf1[28] = bf0[7];

  bf1[29] = bf0[23];

  bf1[30] = bf0[15];

  bf1[31] = bf0[31];

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

}

void av1_fadst4(const int32_t *input, int32_t *output, int8_t cos_bit,

                const int8_t *stage_range) {

  int bit = cos_bit;

  const int32_t *sinpi = sinpi_arr(bit);

  int32_t x0, x1, x2, x3;

  int32_t s0, s1, s2, s3, s4, s5, s6, s7;

  // stage 0

  av1_range_check_buf(0, input, input, 4, stage_range[0]);

  x0 = input[0];

  x1 = input[1];

  x2 = input[2];

  x3 = input[3];

  if (!(x0 | x1 | x2 | x3)) {

    output[0] = output[1] = output[2] = output[3] = 0;

    return;

  }

  // stage 1

  s0 = range_check_value(sinpi[1] * x0, bit + stage_range[1]);

  s1 = range_check_value(sinpi[4] * x0, bit + stage_range[1]);

  s2 = range_check_value(sinpi[2] * x1, bit + stage_range[1]);

  s3 = range_check_value(sinpi[1] * x1, bit + stage_range[1]);

  s4 = range_check_value(sinpi[3] * x2, bit + stage_range[1]);

  s5 = range_check_value(sinpi[4] * x3, bit + stage_range[1]);

  s6 = range_check_value(sinpi[2] * x3, bit + stage_range[1]);

  s7 = range_check_value(x0 + x1, stage_range[1]);

  // stage 2

  s7 = range_check_value(s7 - x3, stage_range[2]);

  // stage 3

  x0 = range_check_value(s0 + s2, bit + stage_range[3]);

  x1 = range_check_value(sinpi[3] * s7, bit + stage_range[3]);

  x2 = range_check_value(s1 - s3, bit + stage_range[3]);

  x3 = range_check_value(s4, bit + stage_range[3]);

  // stage 4

  x0 = range_check_value(x0 + s5, bit + stage_range[4]);

  x2 = range_check_value(x2 + s6, bit + stage_range[4]);

  // stage 5

  s0 = range_check_value(x0 + x3, bit + stage_range[5]);

  s1 = range_check_value(x1, bit + stage_range[5]);

  s2 = range_check_value(x2 - x3, bit + stage_range[5]);

  s3 = range_check_value(x2 - x0, bit + stage_range[5]);

  // stage 6

  s3 = range_check_value(s3 + x3, bit + stage_range[6]);

  // 1-D transform scaling factor is sqrt(2).

  output[0] = round_shift(s0, bit);

  output[1] = round_shift(s1, bit);

  output[2] = round_shift(s2, bit);

  output[3] = round_shift(s3, bit);

  av1_range_check_buf(6, input, output, 4, stage_range[6]);

}

void av1_fadst8(const int32_t *input, int32_t *output, int8_t cos_bit,

                const int8_t *stage_range) {

  const int32_t size = 8;

  const int32_t *cospi;

  int32_t stage = 0;

  int32_t *bf0, *bf1;

  int32_t step[8];

  // stage 0;

  av1_range_check_buf(stage, input, input, size, stage_range[stage]);

  // stage 1;

  stage++;

  assert(output != input);

  bf1 = output;

  bf1[0] = input[0];

  bf1[1] = -input[7];

  bf1[2] = -input[3];

  bf1[3] = input[4];

  bf1[4] = -input[1];

  bf1[5] = input[6];

  bf1[6] = input[2];

  bf1[7] = -input[5];

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

  // stage 2

  stage++;

  cospi = cospi_arr(cos_bit);

  bf0 = output;

  bf1 = step;

  bf1[0] = bf0[0];

  bf1[1] = bf0[1];

  bf1[2] = half_btf(cospi[32], bf0[2], cospi[32], bf0[3], cos_bit);

  bf1[3] = half_btf(cospi[32], bf0[2], -cospi[32], bf0[3], cos_bit);

  bf1[4] = bf0[4];

  bf1[5] = bf0[5];

  bf1[6] = half_btf(cospi[32], bf0[6], cospi[32], bf0[7], cos_bit);

  bf1[7] = half_btf(cospi[32], bf0[6], -cospi[32], bf0[7], cos_bit);

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

  // stage 3

  stage++;

  bf0 = step;

  bf1 = output;

  bf1[0] = bf0[0] + bf0[2];

  bf1[1] = bf0[1] + bf0[3];

  bf1[2] = bf0[0] - bf0[2];

  bf1[3] = bf0[1] - bf0[3];

  bf1[4] = bf0[4] + bf0[6];

  bf1[5] = bf0[5] + bf0[7];

  bf1[6] = bf0[4] - bf0[6];

  bf1[7] = bf0[5] - bf0[7];

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

  // stage 4

  stage++;

  cospi = cospi_arr(cos_bit);

  bf0 = output;

  bf1 = step;

  bf1[0] = bf0[0];

  bf1[1] = bf0[1];

  bf1[2] = bf0[2];

  bf1[3] = bf0[3];

  bf1[4] = half_btf(cospi[16], bf0[4], cospi[48], bf0[5], cos_bit);

  bf1[5] = half_btf(cospi[48], bf0[4], -cospi[16], bf0[5], cos_bit);

  bf1[6] = half_btf(-cospi[48], bf0[6], cospi[16], bf0[7], cos_bit);

  bf1[7] = half_btf(cospi[16], bf0[6], cospi[48], bf0[7], cos_bit);

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

  // stage 5

  stage++;

  bf0 = step;

  bf1 = output;

  bf1[0] = bf0[0] + bf0[4];

  bf1[1] = bf0[1] + bf0[5];

  bf1[2] = bf0[2] + bf0[6];

  bf1[3] = bf0[3] + bf0[7];

  bf1[4] = bf0[0] - bf0[4];

  bf1[5] = bf0[1] - bf0[5];

  bf1[6] = bf0[2] - bf0[6];

  bf1[7] = bf0[3] - bf0[7];

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

  // stage 6

  stage++;

  cospi = cospi_arr(cos_bit);

  bf0 = output;

  bf1 = step;

  bf1[0] = half_btf(cospi[4], bf0[0], cospi[60], bf0[1], cos_bit);

  bf1[1] = half_btf(cospi[60], bf0[0], -cospi[4], bf0[1], cos_bit);

  bf1[2] = half_btf(cospi[20], bf0[2], cospi[44], bf0[3], cos_bit);

  bf1[3] = half_btf(cospi[44], bf0[2], -cospi[20], bf0[3], cos_bit);

  bf1[4] = half_btf(cospi[36], bf0[4], cospi[28], bf0[5], cos_bit);

  bf1[5] = half_btf(cospi[28], bf0[4], -cospi[36], bf0[5], cos_bit);

  bf1[6] = half_btf(cospi[52], bf0[6], cospi[12], bf0[7], cos_bit);

  bf1[7] = half_btf(cospi[12], bf0[6], -cospi[52], bf0[7], cos_bit);

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

  // stage 7

  stage++;

  bf0 = step;

  bf1 = output;

  bf1[0] = bf0[1];

  bf1[1] = bf0[6];

  bf1[2] = bf0[3];

  bf1[3] = bf0[4];

  bf1[4] = bf0[5];

  bf1[5] = bf0[2];

  bf1[6] = bf0[7];

  bf1[7] = bf0[0];

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

}

void av1_fadst16(const int32_t *input, int32_t *output, int8_t cos_bit,

                 const int8_t *stage_range) {

  const int32_t size = 16;

  const int32_t *cospi;

  int32_t stage = 0;

  int32_t *bf0, *bf1;

  int32_t step[16];

  // stage 0;

  av1_range_check_buf(stage, input, input, size, stage_range[stage]);

  // stage 1;

  stage++;

  assert(output != input);

  bf1 = output;

  bf1[0] = input[0];

  bf1[1] = -input[15];

  bf1[2] = -input[7];

  bf1[3] = input[8];

  bf1[4] = -input[3];

  bf1[5] = input[12];

  bf1[6] = input[4];

  bf1[7] = -input[11];

  bf1[8] = -input[1];

  bf1[9] = input[14];

  bf1[10] = input[6];

  bf1[11] = -input[9];

  bf1[12] = input[2];

  bf1[13] = -input[13];

  bf1[14] = -input[5];

  bf1[15] = input[10];

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

  // stage 2

  stage++;

  cospi = cospi_arr(cos_bit);

  bf0 = output;

  bf1 = step;

  bf1[0] = bf0[0];

  bf1[1] = bf0[1];

  bf1[2] = half_btf(cospi[32], bf0[2], cospi[32], bf0[3], cos_bit);

  bf1[3] = half_btf(cospi[32], bf0[2], -cospi[32], bf0[3], cos_bit);

  bf1[4] = bf0[4];

  bf1[5] = bf0[5];

  bf1[6] = half_btf(cospi[32], bf0[6], cospi[32], bf0[7], cos_bit);

  bf1[7] = half_btf(cospi[32], bf0[6], -cospi[32], bf0[7], cos_bit);

  bf1[8] = bf0[8];

  bf1[9] = bf0[9];

  bf1[10] = half_btf(cospi[32], bf0[10], cospi[32], bf0[11], cos_bit);

  bf1[11] = half_btf(cospi[32], bf0[10], -cospi[32], bf0[11], cos_bit);

  bf1[12] = bf0[12];

  bf1[13] = bf0[13];

  bf1[14] = half_btf(cospi[32], bf0[14], cospi[32], bf0[15], cos_bit);

  bf1[15] = half_btf(cospi[32], bf0[14], -cospi[32], bf0[15], cos_bit);

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

  // stage 3

  stage++;

  bf0 = step;

  bf1 = output;

  bf1[0] = bf0[0] + bf0[2];

  bf1[1] = bf0[1] + bf0[3];

  bf1[2] = bf0[0] - bf0[2];

  bf1[3] = bf0[1] - bf0[3];

  bf1[4] = bf0[4] + bf0[6];

  bf1[5] = bf0[5] + bf0[7];

  bf1[6] = bf0[4] - bf0[6];

  bf1[7] = bf0[5] - bf0[7];

  bf1[8] = bf0[8] + bf0[10];

  bf1[9] = bf0[9] + bf0[11];

  bf1[10] = bf0[8] - bf0[10];

  bf1[11] = bf0[9] - bf0[11];

  bf1[12] = bf0[12] + bf0[14];

  bf1[13] = bf0[13] + bf0[15];

  bf1[14] = bf0[12] - bf0[14];

  bf1[15] = bf0[13] - bf0[15];

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

  // stage 4

  stage++;

  cospi = cospi_arr(cos_bit);

  bf0 = output;

  bf1 = step;

  bf1[0] = bf0[0];

  bf1[1] = bf0[1];

  bf1[2] = bf0[2];

  bf1[3] = bf0[3];

  bf1[4] = half_btf(cospi[16], bf0[4], cospi[48], bf0[5], cos_bit);

  bf1[5] = half_btf(cospi[48], bf0[4], -cospi[16], bf0[5], cos_bit);

  bf1[6] = half_btf(-cospi[48], bf0[6], cospi[16], bf0[7], cos_bit);

  bf1[7] = half_btf(cospi[16], bf0[6], cospi[48], bf0[7], cos_bit);

  bf1[8] = bf0[8];

  bf1[9] = bf0[9];

  bf1[10] = bf0[10];

  bf1[11] = bf0[11];

  bf1[12] = half_btf(cospi[16], bf0[12], cospi[48], bf0[13], cos_bit);

  bf1[13] = half_btf(cospi[48], bf0[12], -cospi[16], bf0[13], cos_bit);

  bf1[14] = half_btf(-cospi[48], bf0[14], cospi[16], bf0[15], cos_bit);

  bf1[15] = half_btf(cospi[16], bf0[14], cospi[48], bf0[15], cos_bit);

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

  // stage 5

  stage++;

  bf0 = step;

  bf1 = output;

  bf1[0] = bf0[0] + bf0[4];

  bf1[1] = bf0[1] + bf0[5];

  bf1[2] = bf0[2] + bf0[6];

  bf1[3] = bf0[3] + bf0[7];

  bf1[4] = bf0[0] - bf0[4];

  bf1[5] = bf0[1] - bf0[5];

  bf1[6] = bf0[2] - bf0[6];

  bf1[7] = bf0[3] - bf0[7];

  bf1[8] = bf0[8] + bf0[12];

  bf1[9] = bf0[9] + bf0[13];

  bf1[10] = bf0[10] + bf0[14];

  bf1[11] = bf0[11] + bf0[15];

  bf1[12] = bf0[8] - bf0[12];

  bf1[13] = bf0[9] - bf0[13];

  bf1[14] = bf0[10] - bf0[14];

  bf1[15] = bf0[11] - bf0[15];

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

  // stage 6

  stage++;

  cospi = cospi_arr(cos_bit);

  bf0 = output;

  bf1 = step;

  bf1[0] = bf0[0];

  bf1[1] = bf0[1];

  bf1[2] = bf0[2];

  bf1[3] = bf0[3];

  bf1[4] = bf0[4];

  bf1[5] = bf0[5];

  bf1[6] = bf0[6];

  bf1[7] = bf0[7];

  bf1[8] = half_btf(cospi[8], bf0[8], cospi[56], bf0[9], cos_bit);

  bf1[9] = half_btf(cospi[56], bf0[8], -cospi[8], bf0[9], cos_bit);

  bf1[10] = half_btf(cospi[40], bf0[10], cospi[24], bf0[11], cos_bit);

  bf1[11] = half_btf(cospi[24], bf0[10], -cospi[40], bf0[11], cos_bit);

  bf1[12] = half_btf(-cospi[56], bf0[12], cospi[8], bf0[13], cos_bit);

  bf1[13] = half_btf(cospi[8], bf0[12], cospi[56], bf0[13], cos_bit);

  bf1[14] = half_btf(-cospi[24], bf0[14], cospi[40], bf0[15], cos_bit);

  bf1[15] = half_btf(cospi[40], bf0[14], cospi[24], bf0[15], cos_bit);

  av1_range_check_buf(stage, input, bf1, size, stage_range[stage]);

  // stage 7

  stage++;

  bf0 = step;

  bf1 = output;

  bf1[0] = bf0[0] + bf0[8];

  bf1[1] = bf0[1] + bf0[9];

  bf1[2] = bf0[2] + bf0[10];

  bf1[3] = bf0[3] + bf0[11];

  bf1[4] = bf0[4] + bf0[12];

  bf1[5] = bf0[5] + bf0[13];

  bf1[6] = bf0[6] + bf0[14];

  bf1[7] = bf0[7] + bf0[15];

  bf1[8] = bf0[0] - bf0[8];

  bf1[9] = bf0[1] - bf0[9];

  bf1[10] = bf0[2] - bf0[10];

  bf1[11] = bf0[3] - bf0[11];

  bf1[12] = bf0[4] - bf0[12];

  bf1[13] = bf0[5] - bf0[13];

  bf1[14] = bf0[6] - bf0[14];

  bf1[15] = bf0[7] - bf0[15];