/******************************************************************************
 *                                                                            *
 * Copyright (C) 2023 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at:
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 *****************************************************************************
 * Originally developed and contributed by Ittiam Systems Pvt. Ltd, Bangalore
 */

#include <math.h>
#include "ixheaac_type_def.h"
#include "ixheaac_constants.h"
#include "ixheaace_aac_constants.h"
#include "ixheaace_psy_const.h"
#include "ixheaace_tns.h"
#include "ixheaace_tns_params.h"
#include "ixheaace_rom.h"
#include "ixheaace_common_rom.h"
#include "ixheaace_bitbuffer.h"

static VOID ia_enhaacplus_enc_ps_fft(complex *out, LOOPINDEX N, WORD32 nob,
                                     ixheaace_common_tables *comm_tab_ptr) {
  LOOPINDEX block_per_stage, stage_num, inner;

  WORD32 index_1, index_2, tab_modifier;
  WORD32 len, increment, i;

  FLOAT32 cos_val;
  FLOAT32 sin_val;

  WORD16 index1;
  FLOAT32 re_temp, re_temp2;
  FLOAT32 im_temp, im_temp2;
  FLOAT32 *out1_w32, *out2_w32;

  len = N >> 1;
  index_1 = 0;

  out1_w32 = (FLOAT32 *)&out[index_1];
  out2_w32 = (FLOAT32 *)&out[index_1 + 1];

  for (block_per_stage = len - 1; block_per_stage >= 0; block_per_stage--) {
    re_temp = out1_w32[0];
    im_temp = out1_w32[1];

    re_temp2 = out2_w32[0];
    im_temp2 = out2_w32[1];

    out1_w32[0] = re_temp + re_temp2;
    out1_w32[1] = im_temp + im_temp2;

    out2_w32[0] = re_temp - re_temp2;
    out2_w32[1] = im_temp - im_temp2;

    out1_w32 += 4;
    out2_w32 += 4;
  }

  if (nob == 3) {
    i = 2; /* used for dist calculation*/
    increment = 1;
    tab_modifier = 2;
    for (stage_num = 1; stage_num >= 0; stage_num--) {
      len = len >> 1;
      index_1 = 0;
      increment += 1;

      tab_modifier--;

      for (block_per_stage = len - 1; block_per_stage >= 0; block_per_stage--) {
        index_2 = index_1 + i;

        out1_w32 = (FLOAT32 *)&out[index_1];
        out2_w32 = (FLOAT32 *)&out[index_2];

        re_temp = out1_w32[0];
        im_temp = out1_w32[1];

        re_temp2 = out2_w32[0];
        im_temp2 = out2_w32[1];

        out1_w32[0] = re_temp + re_temp2;
        out1_w32[1] = im_temp + im_temp2;

        out2_w32[0] = re_temp - re_temp2;
        out2_w32[1] = im_temp - im_temp2;

        index1 = (WORD16)tab_modifier;

        out1_w32 += 2;
        out2_w32 += 2;

        for (inner = 0; inner < (2 * i - 2); inner += 2) {
          cos_val = comm_tab_ptr->cos_arr[index1];
          sin_val = comm_tab_ptr->sin_arr[index1];

          index1++;

          re_temp = (out2_w32[inner] * cos_val) + (out2_w32[inner + 1] * sin_val);
          im_temp = (out2_w32[inner + 1] * cos_val) - (out2_w32[inner] * sin_val);

          re_temp2 = out1_w32[inner];
          im_temp2 = out1_w32[inner + 1];

          out1_w32[inner] = re_temp2 + re_temp;
          out1_w32[inner + 1] = im_temp2 + im_temp;

          out2_w32[inner] = re_temp2 - re_temp;
          out2_w32[inner + 1] = im_temp2 - im_temp;
        }

        index_1 += (WORD32)pow(2, increment);
      }
      i <<= 1;
    }
  } else {
    out1_w32 = (FLOAT32 *)&out[0];
    out2_w32 = (FLOAT32 *)&out[2];

    re_temp = out1_w32[0];
    im_temp = out1_w32[1];

    re_temp2 = out2_w32[0];
    im_temp2 = out2_w32[1];

    out1_w32[0] = re_temp + re_temp2;
    out1_w32[1] = im_temp + im_temp2;

    out2_w32[0] = re_temp - re_temp2;
    out2_w32[1] = im_temp - im_temp2;

    out1_w32 += 2;
    out2_w32 += 2;

    sin_val = comm_tab_ptr->sin_arr[1];

    re_temp = out2_w32[1] * sin_val;
    im_temp = out2_w32[0] * sin_val;

    re_temp2 = out1_w32[0];
    im_temp2 = out1_w32[1];

    out1_w32[0] = re_temp2 + re_temp;
    out1_w32[1] = im_temp2 - im_temp;

    out2_w32[0] = re_temp2 - re_temp;
    out2_w32[1] = im_temp2 + im_temp;
  }
}

VOID ia_enhaacplus_enc_fft(complex *out, WORD32 N, ixheaace_common_tables *pstr_common_tab) {
  WORD32 nob;
  WORD32 len;

  /* time domain samples obtained with bit reversal*/
  len = N;

  if (len == 8) {
    nob = 3;
  } else {
    nob = 2;
  }
  ia_enhaacplus_enc_ps_fft(out, N, nob, pstr_common_tab);
}