/*

* Copyright(c) 2019 Intel Corporation

* 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 https://www.aomedia.org/license/software-license. 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 https://www.aomedia.org/license/patent-license.

*/

#include <stdlib.h>

#include <string.h>

#include "transforms.h"

#include "common_utils.h"

#include "aom_dsp_rtcd.h"

const int8_t fwd_cos_bit_col[MAX_TXWH_IDX /*txw_idx*/][MAX_TXWH_IDX /*txh_idx*/] = {

    {13, 13, 13, 0, 0}, {13, 13, 13, 12, 0}, {13, 13, 13, 12, 13}, {0, 13, 13, 12, 13}, {0, 0, 13, 12, 13}};

const int8_t fwd_cos_bit_row[MAX_TXWH_IDX /*txw_idx*/][MAX_TXWH_IDX /*txh_idx*/] = {

    {13, 13, 12, 0, 0}, {13, 13, 13, 12, 0}, {13, 13, 12, 13, 12}, {0, 12, 13, 12, 11}, {0, 0, 12, 11, 10}};

const uint8_t tx_blocks_per_depth[BLOCK_SIZES_ALL][MAX_VARTX_DEPTH + 1] = {

    {1, 1, 1}, // BLOCK_4X4

    {1, 1, 1}, // BLOCK_4X8

    {1, 1, 1}, // BLOCK_8X4

    {1, 4, 4}, // BLOCK_8X8

    {1, 2, 8}, // BLOCK_8X16

    {1, 2, 8}, // BLOCK_16X8

    {1, 4, 16}, // BLOCK_16X16

    {1, 2, 8}, // BLOCK_16X32

    {1, 2, 8}, // BLOCK_32X16

    {1, 4, 16}, // BLOCK_32X32

    {1, 2, 8}, // BLOCK_32X64

    {1, 2, 8}, // BLOCK_64X32

    {1, 4, 16}, // BLOCK_64X64

    {2, 2, 2}, // BLOCK_64X128

    {2, 2, 2}, // BLOCK_128X64

    {4, 4, 4}, // BLOCK_128X128

    {1, 2, 4}, // BLOCK_4X16

    {1, 2, 4}, // BLOCK_16X4

    {1, 2, 4}, // BLOCK_8X32

    {1, 2, 4}, // BLOCK_32X8

    {1, 2, 4}, // BLOCK_16X64

    {1, 2, 4} // BLOCK_64X16

};

// origin is block - separate tables for INTRA (idx 0) and INTER (idx 1) needed b/c of tx depth 2

Position tx_org[BLOCK_SIZES_ALL][2 /*is_inter*/][MAX_VARTX_DEPTH + 1][MAX_TXB_COUNT]; // built at init

// Build the per-(bsize, is_inter, tx_depth) transform-block origins from tx_depth_to_tx_size[]. INTRA lists

// the txbs in raster order; INTER lists them in quad/Z order (raster within each depth-1 parent), which only

// differs from raster at tx_depth 2 of 2D-split blocks. Replaces a ~17 KB const table.

void svt_aom_build_tx_org(void) {

    memset(tx_org, 0, sizeof(tx_org));

    for (BlockSize bs = 0; bs < BLOCK_SIZES_ALL; bs++) {

        const int bw = block_size_wide[bs], bh = block_size_high[bs];

        for (int depth = 0; depth <= MAX_VARTX_DEPTH; depth++) {

            const TxSize txs = tx_depth_to_tx_size[depth][bs];

            const int    tw = tx_size_wide[txs], th = tx_size_high[txs];

            int          k = 0;

            for (int y = 0; y < bh; y += th) {

                for (int x = 0; x < bw; x += tw) {

                    tx_org[bs][0][depth][k++] = (Position){x, y};

                }

            }

            k = 0;

            if (depth == 0) {

                for (int y = 0; y < bh; y += th) {

                    for (int x = 0; x < bw; x += tw) {

                        tx_org[bs][1][depth][k++] = (Position){x, y};

                    }

                }

            } else {

                const TxSize ptxs = tx_depth_to_tx_size[depth - 1][bs];

                const int    ptw = tx_size_wide[ptxs], pth = tx_size_high[ptxs];

                for (int py = 0; py < bh; py += pth) {

                    for (int px = 0; px < bw; px += ptw) {

                        for (int y = 0; y < pth; y += th) {

                            for (int x = 0; x < ptw; x += tw) {

                                tx_org[bs][1][depth][k++] = (Position){px + x, py + y};

                            }

                        }

                    }

                }

            }

        }

    }

}

static const int8_t fdct4_range_mult2[4]    = {0, 2, 3, 3};

static const int8_t fdct8_range_mult2[6]    = {0, 2, 4, 5, 5, 5};

static const int8_t fdct16_range_mult2[8]   = {0, 2, 4, 6, 7, 7, 7, 7};

static const int8_t fdct32_range_mult2[10]  = {0, 2, 4, 6, 8, 9, 9, 9, 9, 9};

static const int8_t fdct64_range_mult2[12]  = {0, 2, 4, 6, 8, 10, 11, 11, 11, 11, 11, 11};

static const int8_t fadst4_range_mult2[7]   = {0, 2, 4, 3, 3, 3, 3};

static const int8_t fadst8_range_mult2[8]   = {0, 0, 1, 3, 3, 5, 5, 5};

static const int8_t fadst16_range_mult2[10] = {0, 0, 1, 3, 3, 5, 5, 7, 7, 7};

static const int8_t fadst32_range_mult2[12] = {0, 0, 1, 3, 3, 5, 5, 7, 7, 9, 9, 9};

static const int8_t fidtx4_range_mult2[1]   = {1};

static const int8_t fidtx8_range_mult2[1]   = {2};

static const int8_t fidtx16_range_mult2[1]  = {3};

static const int8_t fidtx32_range_mult2[1]  = {4};

static const int8_t fidtx64_range_mult2[1]  = {5};

static const int8_t* fwd_txfm_range_mult2_list[TXFM_TYPES] = {fdct4_range_mult2,

                                                              fdct8_range_mult2,

                                                              fdct16_range_mult2,

                                                              fdct32_range_mult2,

                                                              fdct64_range_mult2,

                                                              fadst4_range_mult2,

                                                              fadst8_range_mult2,

                                                              fadst16_range_mult2,

                                                              fadst32_range_mult2,

                                                              fidtx4_range_mult2,

                                                              fidtx8_range_mult2,

                                                              fidtx16_range_mult2,

                                                              fidtx32_range_mult2,

                                                              fidtx64_range_mult2};

static const int8_t fwd_shift_4x4[3]   = {2, 0, 0};

static const int8_t fwd_shift_8x8[3]   = {2, -1, 0};

static const int8_t fwd_shift_16x16[3] = {2, -2, 0};

static const int8_t fwd_shift_32x32[3] = {2, -4, 0};

static const int8_t fwd_shift_64x64[3] = {0, -2, -2};

static const int8_t fwd_shift_4x8[3]   = {2, -1, 0};

static const int8_t fwd_shift_8x4[3]   = {2, -1, 0};

static const int8_t fwd_shift_8x16[3]  = {2, -2, 0};

static const int8_t fwd_shift_16x8[3]  = {2, -2, 0};

static const int8_t fwd_shift_16x32[3] = {2, -4, 0};

static const int8_t fwd_shift_32x16[3] = {2, -4, 0};

static const int8_t fwd_shift_32x64[3] = {0, -2, -2};

static const int8_t fwd_shift_64x32[3] = {2, -4, -2};

static const int8_t fwd_shift_4x16[3]  = {2, -1, 0};

static const int8_t fwd_shift_16x4[3]  = {2, -1, 0};

static const int8_t fwd_shift_8x32[3]  = {2, -2, 0};

static const int8_t fwd_shift_32x8[3]  = {2, -2, 0};

static const int8_t fwd_shift_16x64[3] = {0, -2, 0};

static const int8_t fwd_shift_64x16[3] = {2, -4, 0};

const int8_t* fwd_txfm_shift_ls[TX_SIZES_ALL] = {

    fwd_shift_4x4,  fwd_shift_8x8,  fwd_shift_16x16, fwd_shift_32x32, fwd_shift_64x64, fwd_shift_4x8,   fwd_shift_8x4,

    fwd_shift_8x16, fwd_shift_16x8, fwd_shift_16x32, fwd_shift_32x16, fwd_shift_32x64, fwd_shift_64x32, fwd_shift_4x16,

    fwd_shift_16x4, fwd_shift_8x32, fwd_shift_32x8,  fwd_shift_16x64, fwd_shift_64x16,

};

void svt_av1_gen_fwd_stage_range(int8_t* stage_range_col, int8_t* stage_range_row, const Txfm2dFlipCfg* cfg,

                                 int32_t bd) {

    // Take the shift from the larger dimension in the rectangular case.

    const int8_t* shift = cfg->shift;

    // i < MAX_TXFM_STAGE_NUM will mute above array bounds warning

    for (int32_t i = 0; i < cfg->stage_num_col && i < MAX_TXFM_STAGE_NUM; ++i) {

        stage_range_col[i] = (int8_t)(cfg->stage_range_col[i] + shift[0] + bd + 1);

    }

    // i < MAX_TXFM_STAGE_NUM will mute above array bounds warning

    for (int32_t i = 0; i < cfg->stage_num_row && i < MAX_TXFM_STAGE_NUM; ++i) {

        stage_range_row[i] = (int8_t)(cfg->stage_range_row[i] + shift[0] + shift[1] + bd + 1);

    }

}

void svt_av1_fdct4_new(const int32_t* input, int32_t* output, int8_t cos_bit, const int8_t* stage_range) {

    (void)stage_range;

    const int32_t* cospi;

    int32_t *bf0, *bf1;

    int32_t  step[4];

    // stage 0;

    // stage 1;

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

    // stage 2

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

    // stage 3

    bf0    = step;

    bf1    = output;

    bf1[0] = bf0[0];

    bf1[1] = bf0[2];

    bf1[2] = bf0[1];

    bf1[3] = bf0[3];

}

void svt_av1_fdct8_new(const int32_t* input, int32_t* output, int8_t cos_bit, const int8_t* stage_range) {

    (void)stage_range;

    const int32_t* cospi;

    int32_t *bf0, *bf1;

    int32_t  step[8];

    // stage 0;

    // stage 1;

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

    // stage 2

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

    // stage 3

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

    // stage 4

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

    // stage 5

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

}

void svt_av1_fdct16_new(const int32_t* input, int32_t* output, int8_t cos_bit, const int8_t* stage_range) {

    (void)stage_range;

    const int32_t* cospi;

    int32_t *bf0, *bf1;

    int32_t  step[16];

    // stage 0;

    // stage 1;

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

    // stage 2

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

    // stage 3

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

    // stage 4

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

    // stage 5

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

    // stage 6

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

    // stage 7

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

}

void svt_av1_fdct32_new(const int32_t* input, int32_t* output, int8_t cos_bit, const int8_t* stage_range) {

    (void)stage_range;

    const int32_t* cospi;

    int32_t *bf0, *bf1;

    int32_t  step[32];

    // stage 0;

    // stage 1;

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

    // stage 2

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

    // stage 3

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

    // stage 4

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

    // stage 5

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

    // stage 6

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

    // stage 7

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

    // stage 8

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

    // stage 9

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

}

void svt_av1_fdct64_new(const int32_t* input, int32_t* output, int8_t cos_bit, const int8_t* stage_range) {

    (void)stage_range;

    const int32_t* cospi;

    int32_t *bf0, *bf1;

    int32_t  step[64];

    // stage 0;

    // stage 1;

    bf1     = output;

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

    // stage 2

    cospi   = cospi_arr(cos_bit);

    bf0     = output;

    bf1     = step;

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

    bf1[32] = bf0[32];

    bf1[33] = bf0[33];

    bf1[34] = bf0[34];

    bf1[35] = bf0[35];

    bf1[36] = bf0[36];

    bf1[37] = bf0[37];

    bf1[38] = bf0[38];

    bf1[39] = bf0[39];

    bf1[40] = half_btf(-cospi[32], bf0[40], cospi[32], bf0[55], cos_bit);

    bf1[41] = half_btf(-cospi[32], bf0[41], cospi[32], bf0[54], cos_bit);

    bf1[42] = half_btf(-cospi[32], bf0[42], cospi[32], bf0[53], cos_bit);

    bf1[43] = half_btf(-cospi[32], bf0[43], cospi[32], bf0[52], cos_bit);

    bf1[44] = half_btf(-cospi[32], bf0[44], cospi[32], bf0[51], cos_bit);

    bf1[45] = half_btf(-cospi[32], bf0[45], cospi[32], bf0[50], cos_bit);

    bf1[46] = half_btf(-cospi[32], bf0[46], cospi[32], bf0[49], cos_bit);

    bf1[47] = half_btf(-cospi[32], bf0[47], cospi[32], bf0[48], cos_bit);

    bf1[48] = half_btf(cospi[32], bf0[48], cospi[32], bf0[47], cos_bit);

    bf1[49] = half_btf(cospi[32], bf0[49], cospi[32], bf0[46], cos_bit);

    bf1[50] = half_btf(cospi[32], bf0[50], cospi[32], bf0[45], cos_bit);

    bf1[51] = half_btf(cospi[32], bf0[51], cospi[32], bf0[44], cos_bit);

    bf1[52] = half_btf(cospi[32], bf0[52], cospi[32], bf0[43], cos_bit);

    bf1[53] = half_btf(cospi[32], bf0[53], cospi[32], bf0[42], cos_bit);

    bf1[54] = half_btf(cospi[32], bf0[54], cospi[32], bf0[41], cos_bit);

    bf1[55] = half_btf(cospi[32], bf0[55], cospi[32], bf0[40], cos_bit);

    bf1[56] = bf0[56];

    bf1[57] = bf0[57];

    bf1[58] = bf0[58];

    bf1[59] = bf0[59];

    bf1[60] = bf0[60];

    bf1[61] = bf0[61];

    bf1[62] = bf0[62];

    bf1[63] = bf0[63];

    // stage 3

    cospi   = cospi_arr(cos_bit);

    bf0     = step;

    bf1     = output;

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

    bf1[32] = bf0[32] + bf0[47];

    bf1[33] = bf0[33] + bf0[46];

    bf1[34] = bf0[34] + bf0[45];

    bf1[35] = bf0[35] + bf0[44];

    bf1[36] = bf0[36] + bf0[43];

    bf1[37] = bf0[37] + bf0[42];

    bf1[38] = bf0[38] + bf0[41];

    bf1[39] = bf0[39] + bf0[40];

    bf1[40] = -bf0[40] + bf0[39];

    bf1[41] = -bf0[41] + bf0[38];

    bf1[42] = -bf0[42] + bf0[37];

    bf1[43] = -bf0[43] + bf0[36];

    bf1[44] = -bf0[44] + bf0[35];

    bf1[45] = -bf0[45] + bf0[34];

    bf1[46] = -bf0[46] + bf0[33];

    bf1[47] = -bf0[47] + bf0[32];

    bf1[48] = -bf0[48] + bf0[63];

    bf1[49] = -bf0[49] + bf0[62];

    bf1[50] = -bf0[50] + bf0[61];

    bf1[51] = -bf0[51] + bf0[60];

    bf1[52] = -bf0[52] + bf0[59];

    bf1[53] = -bf0[53] + bf0[58];

    bf1[54] = -bf0[54] + bf0[57];

    bf1[55] = -bf0[55] + bf0[56];

    bf1[56] = bf0[56] + bf0[55];

    bf1[57] = bf0[57] + bf0[54];

    bf1[58] = bf0[58] + bf0[53];

    bf1[59] = bf0[59] + bf0[52];

    bf1[60] = bf0[60] + bf0[51];

    bf1[61] = bf0[61] + bf0[50];

    bf1[62] = bf0[62] + bf0[49];

    bf1[63] = bf0[63] + bf0[48];

    // stage 4

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

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

    bf1[32] = bf0[32];

    bf1[33] = bf0[33];

    bf1[34] = bf0[34];

    bf1[35] = bf0[35];

    bf1[36] = half_btf(-cospi[16], bf0[36], cospi[48], bf0[59], cos_bit);

    bf1[37] = half_btf(-cospi[16], bf0[37], cospi[48], bf0[58], cos_bit);

    bf1[38] = half_btf(-cospi[16], bf0[38], cospi[48], bf0[57], cos_bit);

    bf1[39] = half_btf(-cospi[16], bf0[39], cospi[48], bf0[56], cos_bit);

    bf1[40] = half_btf(-cospi[48], bf0[40], -cospi[16], bf0[55], cos_bit);

    bf1[41] = half_btf(-cospi[48], bf0[41], -cospi[16], bf0[54], cos_bit);