/******************************************************************************

 *                                                                            *

 * Copyright (C) 2018 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 "ixheaacd_sbr_common.h"

#include "ixheaac_type_def.h"

#include "ixheaac_constants.h"

#include "ixheaac_basic_ops32.h"

#include "ixheaac_basic_ops16.h"

#include "ixheaac_basic_ops40.h"

#include "ixheaac_basic_ops.h"

#include "ixheaacd_defines.h"

#include "ixheaacd_bitbuffer.h"

#include "ixheaacd_error_codes.h"

#include "ixheaacd_aac_rom.h"

#include "ixheaacd_pulsedata.h"

#include "ixheaacd_pns.h"

#include "ixheaacd_drc_data_struct.h"

#include "ixheaacd_lt_predict.h"

#include "ixheaacd_cnst.h"

#include "ixheaacd_ec_defines.h"

#include "ixheaacd_ec_struct_def.h"

#include "ixheaacd_channelinfo.h"

#include "ixheaacd_drc_dec.h"

#include "ixheaacd_sbrdecoder.h"

#include "ixheaacd_tns.h"

#include "ixheaacd_intrinsics.h"

#include "ixheaacd_common_rom.h"

#include "ixheaacd_block.h"

#include "ixheaacd_channel.h"

#include "ixheaacd_audioobjtypes.h"

#include "ixheaacd_latmdemux.h"

#include "ixheaacd_aacdec.h"

static PLATFORM_INLINE WORD32 ixheaacd_mac32_tns_sat(WORD32 a, WORD32 b,

                                                     WORD32 c) {

  WORD32 result;

  WORD64 temp_result;

  temp_result = (WORD64)a * (WORD64)b;

  result = (WORD32)(temp_result >> 32);

  result = ixheaac_add32_sat(c, result);

  return (result);

}

VOID ixheaacd_tns_decode_coefficients(

    const ia_filter_info_struct *filter, WORD32 *a,

    ia_aac_dec_tables_struct *ptr_aac_tables) {

  WORD32 i;

  WORD32 tmp;

  WORD32 *aptr = a;

  WORD32 *tns_coeff_ptr;

  WORD8 ixheaacd_drc_offset;

  tmp = filter->resolution;

  if (tmp == 0) {

    tns_coeff_ptr = ptr_aac_tables->pstr_block_tables->tns_coeff3;

    ixheaacd_drc_offset = 4;

  } else {

    tns_coeff_ptr = ptr_aac_tables->pstr_block_tables->tns_coeff4;

    ixheaacd_drc_offset = 8;

  }

  for (i = 0; i < filter->order; i++) {

    *aptr++ = tns_coeff_ptr[filter->coef[i] + ixheaacd_drc_offset];

  }

}

VOID ixheaacd_tns_parcor_to_lpc(WORD32 *parcor, WORD32 *lpc, WORD16 *scale,

                                WORD32 order)

{

  WORD i, j, status;

  WORD32 z1;

  WORD32 z[MAX_ORDER + 1];

  WORD32 w[MAX_ORDER + 1];

  WORD32 accu1, accu2;

  status = 1;

  *scale = 1;

  while (status) {

    status = 0;

    for (i = MAX_ORDER; i >= 0; i--) {

      z[i] = 0;

      w[i] = 0;

    }

    accu1 = (0x40000000 >> (*scale - 1));

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

      z1 = accu1;

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

        w[j] = (accu1);

        accu1 = ixheaac_add32_sat(accu1,

                                   ixheaac_mult32_shl_sat(parcor[j], (z[j])));

        if (ixheaac_abs32_sat(accu1) == 0x7fffffff) status = 1;

      }

      for (j = (order - 1); j >= 0; j--) {

        accu2 = (z[j]);

        accu2 = ixheaac_add32_sat(accu2,

                                   ixheaac_mult32_shl_sat(parcor[j], (w[j])));

        z[j + 1] = (accu2);

        if (ixheaac_abs32_sat(accu2) == 0x7fffffff) status = 1;

      }

      z[0] = (z1);

      lpc[i] = (accu1);

      accu1 = 0;

    }

    accu1 = (status - 1);

    if (accu1 == 0) {

      *scale = *scale + 1;

    }

  }

}

VOID ixheaacd_tns_parcor_lpc_convert_dec(WORD16 *parcor, WORD16 *lpc,

                                         WORD16 *scale, WORD order)

{

  WORD i, j, status;

  WORD32 accu;

  WORD16 temp_buf1[MAX_ORDER + 1];

  WORD16 temp_buf2[MAX_ORDER + 1];

  WORD32 accu1, accu2;

  status = 1;

  *scale = 0;

  while (status) {

    status = 0;

    for (i = MAX_ORDER; i >= 0; i--) {

      temp_buf1[i] = 0;

      temp_buf2[i] = 0;

    }

    accu1 = (0x7fffffff >> *scale);

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

      accu = accu1;

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

        temp_buf2[j] = ixheaac_round16(accu1);

        accu1 = ixheaac_mac16x16in32_shl_sat(accu1, parcor[j], temp_buf1[j]);

        if (ixheaac_abs32_sat(accu1) == 0x7fffffff) {

          status = 1;

        }

      }

      for (j = (order - 1); j >= 0; j--) {

        accu2 = ixheaac_deposit16h_in32(temp_buf1[j]);

        accu2 = ixheaac_mac16x16in32_shl_sat(accu2, parcor[j], temp_buf2[j]);

        temp_buf1[j + 1] = ixheaac_round16(accu2);

        if (ixheaac_abs32_sat(accu2) == 0x7fffffff) {

          status = 1;

        }

      }

      temp_buf1[0] = ixheaac_round16(accu);

      lpc[i] = ixheaac_round16(accu1);

      accu1 = 0;

    }

    accu1 = (status - 1);

    if (accu1 == 0) {

      *scale = *scale + 1;

    }

  }

}

VOID ixheaacd_tns_ar_filter_fixed_dec(WORD32 *spectrum, WORD32 size, WORD32 inc,

                                      WORD32 *lpc, WORD32 order,

                                      WORD32 shift_value, WORD scale_spec)

{

  WORD32 i, j;

  WORD32 y, state[MAX_ORDER + 1];

  WORD32 acc;

  if ((order & 3) != 0) {

    for (i = order + 1; i < ((WORD32)(order & 0xfffffffc) + 4); i++) {

      lpc[i] = 0;

    }

    lpc[i] = 0;

    order = ((order & 0xfffffffc) + 4);

    order = order & 31;

  }

  {

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

      y = ixheaac_shl32_sat((*spectrum), scale_spec);

      acc = 0;

      for (j = i; j > 0; j--) {

        acc = ixheaacd_mac32_tns_sat(state[j - 1], lpc[j], acc);

        state[j] = state[j - 1];

      }

      y = ixheaac_sub32_sat(y, ixheaac_shl32_sat(acc, 1));

      state[0] = ixheaac_shl32_sat(y, shift_value);

      *spectrum = y >> scale_spec;

      spectrum += inc;

    }

    for (i = order; i < size; i++) {

      y = ixheaac_shl32_sat((*spectrum), scale_spec);

      acc = 0;

      for (j = order; j > 0; j--) {

        acc = ixheaacd_mac32_tns_sat(state[j - 1], lpc[j], acc);

        state[j] = state[j - 1];

      }

      y = ixheaac_sub32_sat(y, ixheaac_shl32_sat(acc, 1));

      state[0] = ixheaac_shl32_sat(y, shift_value);

      *spectrum = y >> scale_spec;

      spectrum += inc;

    }

  }

}

VOID ixheaacd_tns_ar_filter_fixed_non_neon_armv7(WORD32 *spectrum, WORD32 size,

                                                 WORD32 inc, WORD32 *lpc,

                                                 WORD32 order,

                                                 WORD32 shift_value,

                                                 WORD scale_spec) {

  WORD32 i, j;

  WORD32 y, state[MAX_ORDER + 1];

  WORD32 acc;

  if ((order & 3) != 0) {

    for (i = order + 1; i < ((WORD32)(order & 0xfffffffc) + 4); i++) {

      lpc[i] = 0;

    }

    lpc[i] = 0;

    order = ((order & 0xfffffffc) + 4);

  }

  {

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

      y = ixheaac_shl32_sat((*spectrum), scale_spec);

      acc = 0;

      for (j = i; j > 0; j--) {

        acc = ixheaacd_mac32_tns_sat(state[j - 1], lpc[j], acc);

        state[j] = state[j - 1];

      }

      y = ixheaac_sub32_sat(y, ixheaac_shl32_sat(acc, 1));

      state[0] = ixheaac_shl32_sat(y, shift_value);

      *spectrum = y >> scale_spec;

      spectrum += inc;

    }

    for (i = order; i < size; i++) {

      WORD64 acc = 0;

      WORD32 acc1;

      y = ixheaac_shl32_sat((*spectrum), scale_spec);

      for (j = order; j > 0; j--) {

        acc = ixheaac_mac32x32in64_dual(state[j - 1], lpc[j], acc);

        state[j] = state[j - 1];

      }

      acc1 = (WORD32)(acc >> 32);

      y = ixheaac_sub32_sat(y, ixheaac_shl32_sat(acc1, 1));

      state[0] = ixheaac_shl32_sat(y, shift_value);

      *spectrum = y >> scale_spec;

      spectrum += inc;

    }

  }

}

VOID ixheaacd_tns_ar_filter_fixed_armv8(WORD32 *spectrum, WORD32 size,

                                        WORD32 inc, WORD32 *lpc, WORD32 order,

                                        WORD32 shift_value, WORD scale_spec) {

  WORD32 i, j;

  WORD32 y, state[MAX_ORDER + 1];

  WORD32 acc;

  if ((order & 3) != 0) {

    for (i = order + 1; i < ((WORD32)(order & 0xfffffffc) + 4); i++) {

      lpc[i] = 0;

    }

    lpc[i] = 0;

    order = ((order & 0xfffffffc) + 4);

  }

  {

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

      y = ixheaac_shl32_sat((*spectrum), scale_spec);

      acc = 0;

      for (j = i; j > 0; j--) {

        acc = ixheaacd_mac32_tns_sat(state[j - 1], lpc[j], acc);

        state[j] = state[j - 1];

      }

      y = ixheaac_sub32_sat(y, ixheaac_shl32_sat(acc, 1));

      state[0] = ixheaac_shl32_sat(y, shift_value);

      *spectrum = y >> scale_spec;

      spectrum += inc;

    }

    for (i = order; i < size; i++) {

      WORD64 acc = 0;

      WORD32 acc1;

      y = ixheaac_shl32_sat((*spectrum), scale_spec);

      for (j = order; j > 0; j--) {

        acc = ixheaac_mac32x32in64_dual(state[j - 1], lpc[j], acc);

        state[j] = state[j - 1];

      }

      acc1 = (WORD32)(acc >> 32);

      y = ixheaac_sub32_sat(y, ixheaac_shl32_sat(acc1, 1));

      state[0] = ixheaac_shl32_sat(y, shift_value);

      *spectrum = y >> scale_spec;

      spectrum += inc;

    }

  }

}

void ixheaacd_tns_ma_filter_fixed_ld(WORD32 *spectrum, WORD32 size, WORD32 inc,

                                     WORD32 *lpc, WORD32 order,

                                     WORD16 shift_value) {

  WORD32 i, j;

  WORD32 y, state[MAX_ORDER];

  for (i = 0; i < order; i++) state[i] = 0;

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

    y = *spectrum;

    for (j = 0; j < order; j++) y += ixheaac_mult32_shl(state[j], lpc[j + 1]);

    for (j = order - 1; j > 0; j--) state[j] = state[j - 1];

    state[0] = ixheaac_shl32_dir_sat(*spectrum, shift_value);

    *spectrum = y;

    spectrum += inc;

  }

}

VOID ixheaacd_tns_ar_filter_dec(WORD32 *spectrum, WORD32 size, WORD32 inc,

                                WORD16 *lpc, WORD32 order, WORD32 shift_value,

                                WORD scale_spec, WORD32 *ptr_filter_state) {

  WORD32 i, j;

  WORD32 y;

  WORD32 acc;

  if ((order & 3) != 0) {

    for (i = order + 1; i < ((WORD32)(order & (~3)) + 4); i++) {

      lpc[i] = 0;

    }

    if (i < (MAX_ORDER + 1)) {

      lpc[i] = 0;

      order = ((order & (~3)) + 4);

    } else {

      order = MAX_ORDER;

    }

  }

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

    y = ixheaac_shl32_sat((*spectrum), scale_spec);

    acc = 0;

    for (j = i; j > 0; j--) {

      acc = ixheaac_add32_sat(

          acc, ixheaac_mult32x16in32(ptr_filter_state[j - 1], lpc[j]));

      ptr_filter_state[j] = ptr_filter_state[j - 1];

    }

    y = ixheaac_sub32_sat(y, ixheaac_shl32_sat(acc, 1));

    ptr_filter_state[0] = ixheaac_shl32_sat(y, shift_value);

    *spectrum = y >> scale_spec;

    spectrum += inc;

  }

  for (i = order; i < size; i++) {

    y = ixheaac_shl32_sat((*spectrum), scale_spec);

    acc = 0;

    for (j = order; j > 0; j--) {

      acc = ixheaac_add32_sat(

          acc, ixheaac_mult32x16in32(ptr_filter_state[j - 1], lpc[j]));

      ptr_filter_state[j] = ptr_filter_state[j - 1];

    }

    y = ixheaac_sub32_sat(y, ixheaac_shl32_sat(acc, 1));

    ptr_filter_state[0] = ixheaac_shl32_sat(y, shift_value);

    *spectrum = y >> scale_spec;

    spectrum += inc;

  }

}

WORD32 ixheaacd_calc_max_spectral_line_dec(WORD32 *ptr_tmp, WORD32 size) {

  WORD32 max_spec_line = 0, i;

  WORD unroll_cnt, rem;

  unroll_cnt = size >> 3;

  for (i = unroll_cnt; i--;) {

    max_spec_line = ixheaac_abs32_nrm(*ptr_tmp++) | max_spec_line;

    max_spec_line = ixheaac_abs32_nrm(*ptr_tmp++) | max_spec_line;

    max_spec_line = ixheaac_abs32_nrm(*ptr_tmp++) | max_spec_line;

    max_spec_line = ixheaac_abs32_nrm(*ptr_tmp++) | max_spec_line;

    max_spec_line = ixheaac_abs32_nrm(*ptr_tmp++) | max_spec_line;

    max_spec_line = ixheaac_abs32_nrm(*ptr_tmp++) | max_spec_line;

    max_spec_line = ixheaac_abs32_nrm(*ptr_tmp++) | max_spec_line;

    max_spec_line = ixheaac_abs32_nrm(*ptr_tmp++) | max_spec_line;

  }

  rem = size - (unroll_cnt << 3);

  if (rem) {

    for (i = rem; i--;) {

      max_spec_line = ixheaac_abs32_nrm(*ptr_tmp++) | max_spec_line;

    }

  }

  return ixheaac_norm32(max_spec_line);

}