/******************************************************************************
 *                                                                            *
 * 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 "ixheaace_adjust_threshold_data.h"
#include "iusace_bitbuffer.h"

/* DRC */
#include "impd_drc_common_enc.h"
#include "impd_drc_uni_drc.h"
#include "impd_drc_tables.h"
#include "impd_drc_api.h"
#include "impd_drc_uni_drc_eq.h"
#include "impd_drc_uni_drc_filter_bank.h"
#include "impd_drc_gain_enc.h"
#include "impd_drc_struct_def.h"

#include "iusace_cnst.h"
#include "iusace_tns_usac.h"
#include "iusace_psy_mod.h"
#include "iusace_config.h"
#include "iusace_arith_enc.h"
#include "iusace_block_switch_const.h"
#include "iusace_block_switch_struct_def.h"
#include "iusace_fd_qc_util.h"
#include "iusace_fd_quant.h"
#include "iusace_ms.h"
#include "iusace_signal_classifier.h"
#include "ixheaace_sbr_header.h"
#include "ixheaace_config.h"
#include "ixheaace_asc_write.h"
#include "iusace_main.h"
#include "iusace_func_prototypes.h"
#include "iusace_avq_enc.h"

static VOID iusace_lsf_weight_2st_flt(FLOAT32 *lsfq, FLOAT32 *w, WORD32 mode) {
  WORD32 i;
  FLOAT32 d[ORDER + 1];

  d[0] = lsfq[0];
  d[ORDER] = FREQ_MAX - lsfq[ORDER - 1];
  for (i = 1; i < ORDER; i++) {
    d[i] = lsfq[i] - lsfq[i - 1];
  }

  for (i = 0; i < ORDER; i++) {
    w[i] = (FLOAT32)(iusace_wlsf_factor_table[mode] / (FREQ_DIV / sqrt(d[i] * d[i + 1])));
  }
}

static VOID iusace_lsf_weight(FLOAT32 *lsf, FLOAT32 *ptr_w) {
  WORD32 i;
  FLOAT32 d[ORDER + 1];

  d[0] = lsf[0];
  d[ORDER] = FREQ_MAX - lsf[ORDER - 1];
  for (i = 1; i < ORDER; i++) {
    d[i] = lsf[i] - lsf[i - 1];
  }

  for (i = 0; i < ORDER; i++) {
    ptr_w[i] = (1.0f / d[i]) + (1.0f / d[i + 1]);
  }

  return;
}

static WORD32 iusace_avq_first_approx_abs(FLOAT32 *lsf, FLOAT32 *lsfq) {
  WORD32 i, j, index;
  FLOAT32 w[ORDER];
  FLOAT32 dist_min, dist, temp;
  const FLOAT32 *p_dico;

  iusace_lsf_weight(lsf, w);

  dist_min = 1.0e30f;
  p_dico = iusace_dico_lsf_abs_8b_flt;
  index = 0;

  for (i = 0; i < 256; i++) {
    dist = 0.0;
    for (j = 0; j < ORDER; j++) {
      temp = lsf[j] - *p_dico++;
      dist += w[j] * temp * temp;
    }
    if (dist < dist_min) {
      dist_min = dist;
      index = i;
    }
  }

  for (j = 0; j < ORDER; j++) {
    lsfq[j] = iusace_dico_lsf_abs_8b_flt[index * ORDER + j];
  }

  return index;
}

static WORD32 iusace_avq_first_approx_rel(FLOAT32 *ptr_lsf, FLOAT32 *ptr_lsfq, WORD32 *idx,
                                          WORD32 mode) {
  WORD32 i, num_bits;
  FLOAT32 w[ORDER], x[ORDER], temp;
  WORD32 nq, avq[ORDER];
  FLOAT32 lsf_min;

  iusace_lsf_weight_2st_flt(ptr_lsf, w, 1);

  temp = 0.0f;
  for (i = 0; i < ORDER; i++) {
    x[i] = (ptr_lsf[i] - ptr_lsfq[i]) / w[i];
    temp += x[i] * x[i];
  }

  if (temp < 8.0f) {
    idx[0] = 0;
    idx[1] = 0;
    if ((mode == 0) || (mode == 3)) {
      return (10);
    } else if (mode == 1) {
      return (2);
    } else {
      return (6);
    }
  }

  iusace_lsf_weight_2st_flt(ptr_lsfq, w, mode);

  for (i = 0; i < ORDER; i++) {
    x[i] = (ptr_lsf[i] - ptr_lsfq[i]) / w[i];
  }

  iusace_alg_vec_quant(x, avq, idx);

  for (i = 0; i < ORDER; i++) {
    ptr_lsfq[i] += (w[i] * (FLOAT32)avq[i]);
  }

  num_bits = 0;
  for (i = 0; i < 2; i++) {
    nq = idx[i];

    if ((mode == 0) || (mode == 3)) {
      num_bits += (2 + (nq * 4));
      if (nq > 6) {
        num_bits += nq - 3;
      } else if (nq > 4) {
        num_bits += nq - 4;
      } else if (nq == 0) {
        num_bits += 3;
      }
    } else if (mode == 1) {
      num_bits += nq * 5;
      if (nq == 0) {
        num_bits += 1;
      }
    } else {
      num_bits += (2 + (nq * 4));
      if (nq == 0) {
        num_bits += 1;
      } else if (nq > 4) {
        num_bits += nq - 3;
      }
    }
  }

  lsf_min = LSF_GAP;
  for (i = 0; i < ORDER; i++) {
    if (ptr_lsfq[i] < lsf_min) {
      ptr_lsfq[i] = lsf_min;
    }

    lsf_min = ptr_lsfq[i] + LSF_GAP;
  }

  lsf_min = FREQ_MAX - LSF_GAP;
  for (i = ORDER - 1; i >= 0; i--) {
    if (ptr_lsfq[i] > lsf_min) {
      ptr_lsfq[i] = lsf_min;
    }

    lsf_min = ptr_lsfq[i] - LSF_GAP;
  }

  return (num_bits);
}

VOID iusace_quantize_lpc_avq(FLOAT32 *ptr_lsf, FLOAT32 *ptr_lsfq, WORD32 lpc0,
                             WORD32 *ptr_lpc_idx, WORD32 *nb_indices, WORD32 *nbbits) {
  WORD32 i;
  FLOAT32 lsfq[ORDER];
  WORD32 *ptr_index, indxt[100], num_bits, nbt, nit;

  ptr_index = &ptr_lpc_idx[0];
  *nb_indices = 0;
  *nbbits = 0;

  ptr_index[0] = iusace_avq_first_approx_abs(&ptr_lsf[3 * ORDER], &ptr_lsfq[3 * ORDER]);

  nbt = iusace_avq_first_approx_rel(&ptr_lsf[3 * ORDER], &ptr_lsfq[3 * ORDER], &ptr_index[1], 0);
  nit = 1 + iusace_get_num_params(&ptr_lpc_idx[1]);

  ptr_index += nit;
  *nb_indices += nit;
  *nbbits += 8 + nbt;

  if (lpc0) {
    *ptr_index = 0;
    ptr_index++;
    *nb_indices += 1;
    *nbbits += 1;

    ptr_index[0] = iusace_avq_first_approx_abs(&ptr_lsf[-ORDER], &ptr_lsfq[-ORDER]);

    num_bits = iusace_avq_first_approx_rel(&ptr_lsf[-ORDER], &ptr_lsfq[-ORDER], &ptr_index[1], 0);
    nbt = 8 + num_bits;
    nit = 1 + iusace_get_num_params(&ptr_index[1]);

    for (i = 0; i < ORDER; i++) lsfq[i] = ptr_lsfq[3 * ORDER + i];

    num_bits = iusace_avq_first_approx_rel(&ptr_lsf[-ORDER], &lsfq[0], indxt, 3);

    if (num_bits < nbt) {
      nbt = num_bits;
      nit = iusace_get_num_params(&indxt[0]);
      ptr_index[-1] = 1;
      for (i = 0; i < ORDER; i++) ptr_lsfq[-ORDER + i] = lsfq[i];
      for (i = 0; i < nit; i++) ptr_index[i] = indxt[i];
    }

    ptr_index += nit;
    *nb_indices += nit;
    *nbbits += nbt;
  }

  *ptr_index = 0;
  ptr_index++;
  *nb_indices += 1;
  *nbbits += 1;

  ptr_index[0] = iusace_avq_first_approx_abs(&ptr_lsf[ORDER], &ptr_lsfq[ORDER]);

  num_bits = iusace_avq_first_approx_rel(&ptr_lsf[ORDER], &ptr_lsfq[ORDER], &ptr_index[1], 0);
  nbt = 8 + num_bits;
  nit = 1 + iusace_get_num_params(&ptr_index[1]);

  for (i = 0; i < ORDER; i++) lsfq[i] = ptr_lsfq[3 * ORDER + i];

  num_bits = iusace_avq_first_approx_rel(&ptr_lsf[ORDER], &lsfq[0], indxt, 3);

  if (num_bits < nbt) {
    nbt = num_bits;
    nit = iusace_get_num_params(&indxt[0]);
    ptr_index[-1] = 1;
    for (i = 0; i < ORDER; i++) ptr_lsfq[ORDER + i] = lsfq[i];
    for (i = 0; i < nit; i++) ptr_index[i] = indxt[i];
  }

  ptr_index += nit;
  *nb_indices += nit;
  *nbbits += nbt;

  *ptr_index = 0;
  ptr_index++;
  *nb_indices += 1;

  ptr_index[0] = iusace_avq_first_approx_abs(&ptr_lsf[0], &ptr_lsfq[0]);

  num_bits = iusace_avq_first_approx_rel(&ptr_lsf[0], &ptr_lsfq[0], &ptr_index[1], 0);
  nbt = 2 + 8 + num_bits;
  nit = 1 + iusace_get_num_params(&ptr_index[1]);

  for (i = 0; i < ORDER; i++) lsfq[i] = 0.5f * (ptr_lsfq[-ORDER + i] + ptr_lsfq[ORDER + i]);

  num_bits = iusace_avq_first_approx_rel(&ptr_lsf[0], lsfq, indxt, 1);

  if (num_bits < 10) {
    nbt = 2;
    nit = 0;
    ptr_index[-1] = 1;
    for (i = 0; i < ORDER; i++) ptr_lsfq[i] = lsfq[i];
  }

  for (i = 0; i < ORDER; i++) lsfq[i] = ptr_lsfq[ORDER + i];

  num_bits = iusace_avq_first_approx_rel(&ptr_lsf[0], lsfq, indxt, 2);
  num_bits += 1;

  if (num_bits < nbt) {
    nbt = num_bits;
    nit = iusace_get_num_params(&indxt[0]);
    ptr_index[-1] = 2;
    for (i = 0; i < ORDER; i++) ptr_lsfq[i] = lsfq[i];
    for (i = 0; i < nit; i++) ptr_index[i] = indxt[i];
  }

  ptr_index += nit;
  *nb_indices += nit;
  *nbbits += nbt;

  *ptr_index = 0;
  ptr_index++;
  *nb_indices += 1;

  ptr_index[0] = iusace_avq_first_approx_abs(&ptr_lsf[2 * ORDER], &ptr_lsfq[2 * ORDER]);

  num_bits =
      iusace_avq_first_approx_rel(&ptr_lsf[2 * ORDER], &ptr_lsfq[2 * ORDER], &ptr_index[1], 0);
  nbt = 2 + 8 + num_bits;
  nit = 1 + iusace_get_num_params(&ptr_index[1]);

  for (i = 0; i < ORDER; i++) lsfq[i] = 0.5f * (ptr_lsfq[ORDER + i] + ptr_lsfq[3 * ORDER + i]);

  num_bits = iusace_avq_first_approx_rel(&ptr_lsf[2 * ORDER], lsfq, indxt, 1);
  num_bits += 1;

  if (num_bits < nbt) {
    nbt = num_bits;
    nit = iusace_get_num_params(&indxt[0]);
    ptr_index[-1] = 1;
    for (i = 0; i < ORDER; i++) ptr_lsfq[2 * ORDER + i] = lsfq[i];
    for (i = 0; i < nit; i++) ptr_index[i] = indxt[i];
  }

  for (i = 0; i < ORDER; i++) lsfq[i] = ptr_lsfq[ORDER + i];

  num_bits = iusace_avq_first_approx_rel(&ptr_lsf[2 * ORDER], lsfq, indxt, 2);
  num_bits += 3;

  if (num_bits < nbt) {
    nbt = num_bits;
    nit = iusace_get_num_params(&indxt[0]);
    ptr_index[-1] = 2;
    for (i = 0; i < ORDER; i++) ptr_lsfq[2 * ORDER + i] = lsfq[i];
    for (i = 0; i < nit; i++) ptr_index[i] = indxt[i];
  }

  for (i = 0; i < ORDER; i++) lsfq[i] = ptr_lsfq[3 * ORDER + i];

  num_bits = iusace_avq_first_approx_rel(&ptr_lsf[2 * ORDER], lsfq, indxt, 2);
  num_bits += 3;

  if (num_bits < nbt) {
    nbt = num_bits;
    nit = iusace_get_num_params(&indxt[0]);
    ptr_index[-1] = 3;
    for (i = 0; i < ORDER; i++) ptr_lsfq[2 * ORDER + i] = lsfq[i];
    for (i = 0; i < nit; i++) ptr_index[i] = indxt[i];
  }

  *nb_indices += nit;
  *nbbits += nbt;

  return;
}