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

 *                                                                            *

 * 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 <stdio.h>

#include <string.h>

#include <math.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_sbr_common.h"

#include "ixheaac_error_standards.h"

#include <ixheaacd_cnst.h>

#include "ixheaac_constants.h"

#include "ixheaacd_intrinsics.h"

#include "ixheaacd_common_rom.h"

#include "ixheaacd_sbrdecsettings.h"

#include "ixheaacd_bitbuffer.h"

#include "ixheaacd_defines.h"

#include "ixheaacd_pns.h"

#include <ixheaacd_aac_rom.h>

#include "ixheaacd_pulsedata.h"

#include "ixheaacd_drc_data_struct.h"

#include "ixheaacd_interface.h"

#include "ixheaacd_info.h"

#include "ixheaacd_lt_predict.h"

#include "ixheaacd_ec_defines.h"

#include "ixheaacd_ec_rom.h"

#include "ixheaacd_ec_struct_def.h"

#include "ixheaacd_channelinfo.h"

#include "ixheaac_error_standards.h"

#include "ixheaacd_aac_rom.h"

static WORD32 ixheaacd_aac_ec_get_win_seq(WORD32 prev_win_seq) {

  WORD32 new_win_seq = ONLY_LONG_SEQUENCE;

  if (prev_win_seq == LONG_START_SEQUENCE || prev_win_seq == EIGHT_SHORT_SEQUENCE) {

    new_win_seq = LONG_STOP_SEQUENCE;

  }

  return new_win_seq;

}

static VOID ixheaacd_aac_ec_flip_spec_sign(WORD32 *ptr_spec_coeff, WORD32 num_samples) {

  WORD32 idx;

  WORD32 random_value;

  for (idx = 0; idx < num_samples; idx++) {

    random_value = ptr_spec_coeff[idx] ^ idx;

    if ((random_value & 1) == 0) {

      ptr_spec_coeff[idx] = ixheaac_negate32_sat(ptr_spec_coeff[idx]);

    }

  }

}

static VOID ixheaacd_aac_ec_store(ia_ec_state_str *pstr_ec_state,

                                  ia_aac_dec_channel_info_struct *pstr_aac_dec_channel_info,

                                  ia_ics_info_struct *pstr_ics_info) {

  WORD32 *ptr_spec_coeff = pstr_aac_dec_channel_info->ptr_spec_coeff;

  WORD16 *ptr_spec_scale = pstr_aac_dec_channel_info->ptr_scale_factor;

  UWORD8 win_shape = pstr_ec_state->win_shape;

  WORD32 win_seq = pstr_ec_state->win_seq;

  WORD16 q_spec_scale[MAX_SPEC_SCALE_LEN];

  WORD32 *ptr_temp_spec_coeff = &pstr_ec_state->str_ec_scratch.spec_coeff[0];

  memcpy(q_spec_scale, pstr_ec_state->q_spec_coeff, sizeof(q_spec_scale));

  pstr_ec_state->win_seq = pstr_ics_info->window_sequence;

  pstr_ec_state->win_shape = (UWORD8)pstr_ics_info->window_shape;

  pstr_ec_state->prev_win_group_len =

      *(pstr_ics_info->window_group_length + pstr_ics_info->num_window_groups - 1);

  memcpy(pstr_ec_state->q_spec_coeff, ptr_spec_scale, sizeof(pstr_ec_state->q_spec_coeff));

  memcpy(ptr_temp_spec_coeff, ptr_spec_coeff, LEN_SUPERFRAME * sizeof(ptr_temp_spec_coeff[0]));

  memcpy(ptr_spec_coeff, pstr_ec_state->spectral_coeff,

         LEN_SUPERFRAME * sizeof(ptr_spec_coeff[0]));

  memcpy(pstr_ec_state->spectral_coeff, ptr_temp_spec_coeff,

         sizeof(pstr_ec_state->spectral_coeff));

  pstr_ics_info->window_sequence = win_seq;

  pstr_ics_info->window_shape = win_shape;

  memcpy(ptr_spec_scale, q_spec_scale, MAX_SPEC_SCALE_LEN * sizeof(ptr_spec_scale[0]));

}

static VOID ixheaacd_aac_ec_calc_sfb_nrg(WORD32 *ptr_spec_coeff,

                                         const ia_usac_samp_rate_info *pstr_samp_rate_info,

                                         const WORD32 win_seq, WORD32 win_trans,

                                         WORD32 *ptr_sfb_energy) {

  const WORD16 *ptr_sfb_offset;

  WORD32 line = 0, sfb, total_scale_factor_bands = 0;

  switch (win_seq) {

    case EIGHT_SHORT_SEQUENCE:

      if (win_trans == NO_TRANSITION) {

        total_scale_factor_bands = pstr_samp_rate_info->num_sfb_128 - 1;

        ptr_sfb_offset = pstr_samp_rate_info->ptr_sfb_128;

        for (sfb = 0; sfb < total_scale_factor_bands; sfb++) {

          WORD32 accu = 1;

          WORD32 q_nrg;

          if (sfb == 0) {

            q_nrg = (sizeof(accu) << 3) - ixheaac_norm32(ptr_sfb_offset[sfb] - 0);

            for (; line < ptr_sfb_offset[sfb]; line++) {

              accu += ixheaac_mult32(ptr_spec_coeff[line], ptr_spec_coeff[line]) >> q_nrg;

            }

            ptr_sfb_energy[sfb] = ixheaac_norm32(accu);

          }

          q_nrg = (sizeof(accu) << 3) -

                  ixheaac_norm32(ptr_sfb_offset[sfb + 1] - ptr_sfb_offset[sfb]);

          for (; line < ptr_sfb_offset[sfb + 1]; line++) {

            accu += ixheaac_mult32(ptr_spec_coeff[line], ptr_spec_coeff[line]) >> q_nrg;

          }

          ptr_sfb_energy[sfb] = ixheaac_norm32(accu);

        }

      } else {

        total_scale_factor_bands = pstr_samp_rate_info->num_sfb_1024 - 1;

        ptr_sfb_offset = pstr_samp_rate_info->ptr_sfb_1024;

        for (sfb = 0; sfb < total_scale_factor_bands; sfb++) {

          WORD32 accu = 1;

          WORD32 q_nrg;

          if (sfb == 0) {

            q_nrg = (sizeof(accu) << 3) - ixheaac_norm32(ptr_sfb_offset[sfb] - 0);

            for (; line < ptr_sfb_offset[sfb]; line++) {

              accu +=

                  ixheaac_mult32(ptr_spec_coeff[line >> 3], ptr_spec_coeff[line >> 3]) >> q_nrg;

            }

            ptr_sfb_energy[sfb] = ixheaac_norm32(accu);

          }

          q_nrg = (sizeof(accu) << 3) -

                  ixheaac_norm32(ptr_sfb_offset[sfb + 1] - ptr_sfb_offset[sfb]);

          for (; line < ptr_sfb_offset[sfb + 1]; line++) {

            accu +=

                ixheaac_mult32(ptr_spec_coeff[line >> 3], ptr_spec_coeff[line >> 3]) >> q_nrg;

          }

          ptr_sfb_energy[sfb] = ixheaac_norm32(accu);

        }

      }

      break;

    case ONLY_LONG_SEQUENCE:

    case LONG_START_SEQUENCE:

    case LONG_STOP_SEQUENCE:

      if (win_trans == NO_TRANSITION) {

        total_scale_factor_bands = pstr_samp_rate_info->num_sfb_1024 - 1;

        ptr_sfb_offset = pstr_samp_rate_info->ptr_sfb_1024;

        for (sfb = 0; sfb < total_scale_factor_bands; sfb++) {

          WORD32 accu = 1;

          WORD32 q_nrg;

          if (sfb == 0) {

            q_nrg = (sizeof(accu) << 3) - ixheaac_norm32(ptr_sfb_offset[sfb] - 0);

            for (; line < ptr_sfb_offset[sfb]; line++) {

              accu += ixheaac_mult32(ptr_spec_coeff[line], ptr_spec_coeff[line]) >> q_nrg;

            }

            ptr_sfb_energy[sfb] = ixheaac_norm32(accu);

          }

          q_nrg = (sizeof(accu) << 3) -

                  ixheaac_norm32(ptr_sfb_offset[sfb + 1] - ptr_sfb_offset[sfb]);

          for (; line < ptr_sfb_offset[sfb + 1]; line++) {

            accu += ixheaac_mult32(ptr_spec_coeff[line], ptr_spec_coeff[line]) >> q_nrg;

          }

          ptr_sfb_energy[sfb] = ixheaac_norm32(accu);

        }

      } else {

        total_scale_factor_bands = pstr_samp_rate_info->num_sfb_128 - 1;

        ptr_sfb_offset = pstr_samp_rate_info->ptr_sfb_128;

        for (sfb = 0; sfb < total_scale_factor_bands; sfb++) {

          WORD32 accu = 1;

          WORD32 q_nrg;

          if (sfb == 0) {

            q_nrg = (sizeof(accu) << 3) - ixheaac_norm32(ptr_sfb_offset[sfb] - 0);

            for (; line < ptr_sfb_offset[sfb] << 3; line++) {

              accu += (accu +

                       (ixheaac_mult32(ptr_spec_coeff[line], ptr_spec_coeff[line]) >> q_nrg)) >>

                      3;

            }

            ptr_sfb_energy[sfb] = ixheaac_norm32(accu);

          }

          q_nrg = (sizeof(accu) << 3) -

                  ixheaac_norm32(ptr_sfb_offset[sfb + 1] - ptr_sfb_offset[sfb]);

          for (; line < ptr_sfb_offset[sfb + 1] << 3; line++) {

            accu +=

                (accu + (ixheaac_mult32(ptr_spec_coeff[line], ptr_spec_coeff[line]) >> q_nrg)) >>

                3;

          }

          ptr_sfb_energy[sfb] = ixheaac_norm32(accu);

        }

      }

      break;

  }

}

static VOID ixheaacd_aac_ec_interpolate(WORD32 *ptr_spec_coeff, WORD16 *ptr_spec_scale_prev,

                                        WORD16 *ptr_spec_scale_act, WORD16 *ptr_spec_scale_out,

                                        WORD32 *nrg_prev, WORD32 *nrg_act, WORD32 num_sfb,

                                        const WORD16 *ptr_sfb_offset) {

  WORD32 sfb, line = 0;

  WORD32 fac_shift;

  WORD32 fac_mod;

  for (sfb = 0; sfb < num_sfb; sfb++) {

    fac_shift =

        nrg_prev[sfb] - nrg_act[sfb] + ((*ptr_spec_scale_act - *ptr_spec_scale_prev) << 1);

    fac_mod = fac_shift & 3;

    fac_shift = (fac_shift >> 2) + 1;

    fac_shift += *ptr_spec_scale_prev - max(*ptr_spec_scale_prev, *ptr_spec_scale_act);

    fac_shift = max(min(fac_shift, INT_BITS - 1), -(INT_BITS - 1));

    for (; line < ptr_sfb_offset[sfb]; line++) {

      WORD32 accu =

          ixheaac_mult32x16in32_shl(ptr_spec_coeff[line], ia_ec_interpolation_fac[fac_mod]);

      ptr_spec_coeff[line] = ixheaac_shl32_dir_sat(accu, fac_shift);

    }

  }

  *ptr_spec_scale_out = max(*ptr_spec_scale_prev, *ptr_spec_scale_act);

}

static VOID ixheaacd_aac_ec_state(ia_ec_state_str *pstr_ec_state, WORD32 frame_status) {

  WORD32 ec_state_val = (pstr_ec_state->prev_frame_ok[0] << 2) +

                        (pstr_ec_state->prev_frame_ok[1] << 1) + (frame_status);

  switch (ec_state_val) {

    case 0:

    case 4:

      if (pstr_ec_state->fade_idx < MAX_FADE_FRAMES) {

        pstr_ec_state->fade_idx++;

      }

      pstr_ec_state->conceal_state = FRAME_CONCEAL_SINGLE;

      break;

    case 1:

    case 2:

      if (pstr_ec_state->fade_idx > 0) {

        pstr_ec_state->fade_idx--;

      }

      pstr_ec_state->conceal_state = FRAME_FADE;

      break;

    case 5:

      if (pstr_ec_state->fade_idx > 0) {

        pstr_ec_state->fade_idx--;

      }

      pstr_ec_state->conceal_state = FRAME_OKAY;

      break;

      break;

    case 3:

    case 6:

    case 7:

      if (pstr_ec_state->fade_idx > 0) {

        pstr_ec_state->fade_idx--;

      }

      pstr_ec_state->conceal_state = FRAME_OKAY;

      break;

    default:

      pstr_ec_state->conceal_state = FRAME_OKAY;

  }

  if (pstr_ec_state->fade_idx > MAX_FADE_FRAMES) {

    pstr_ec_state->fade_idx = MAX_FADE_FRAMES;

  }

  if (pstr_ec_state->fade_idx == MAX_FADE_FRAMES) {

    pstr_ec_state->conceal_state = FRAME_MUTE;

  }

  if (pstr_ec_state->fade_idx < 0) {

    pstr_ec_state->fade_idx = 0;

  }

}

static VOID ixheaacd_aac_ec_interpolate_frame(

    ia_ec_state_str *pstr_ec_state, ia_aac_dec_channel_info_struct *pstr_aac_dec_channel_info,

    const ia_usac_samp_rate_info *pstr_samp_rate_info, const WORD32 num_samples,

    const WORD32 frame_status, ia_ics_info_struct *pstr_ics_info) {

  WORD32 *ptr_spec_coeff = pstr_aac_dec_channel_info->ptr_spec_coeff;

  WORD16 *ptr_spec_scale = pstr_aac_dec_channel_info->ptr_scale_factor;

  WORD32 sfb_nrg_prev[WIN_LEN_64];

  WORD32 sfb_nrg_act[WIN_LEN_64];

  WORD32 idx;

  memset(sfb_nrg_prev, 0, sizeof(sfb_nrg_prev));

  memset(sfb_nrg_act, 0, sizeof(sfb_nrg_act));

  if (!frame_status) {

    pstr_ics_info->window_shape = pstr_ec_state->win_shape;

    pstr_ics_info->window_sequence = pstr_ec_state->win_seq;

    for (idx = 0; idx < num_samples; idx++) {

      ptr_spec_coeff[idx] = pstr_ec_state->spectral_coeff[idx];

    }

    memcpy(ptr_spec_scale, pstr_ec_state->q_spec_coeff, 8 * sizeof(ptr_spec_scale[0]));

  }

  if (!pstr_ec_state->prev_frame_ok[1]) {

    if (frame_status && pstr_ec_state->prev_frame_ok[0]) {

      if (pstr_ics_info->window_sequence == EIGHT_SHORT_SEQUENCE) {

        WORD32 window;

        if (pstr_ec_state->win_seq == EIGHT_SHORT_SEQUENCE) {

          WORD32 total_scale_factor_bands = pstr_samp_rate_info->num_sfb_128 - 1;

          const WORD16 *ptr_sfb_offset = pstr_samp_rate_info->ptr_sfb_128;

          pstr_ics_info->window_shape = 1;

          pstr_ics_info->window_sequence = EIGHT_SHORT_SEQUENCE;

          for (window = 0; window < 8; window++) {

            ixheaacd_aac_ec_calc_sfb_nrg(&ptr_spec_coeff[window * (num_samples / 8)],

                                         pstr_samp_rate_info, EIGHT_SHORT_SEQUENCE, NO_TRANSITION,

                                         sfb_nrg_prev);

            ixheaacd_aac_ec_calc_sfb_nrg(

                &pstr_ec_state->spectral_coeff[window * (num_samples / 8)], pstr_samp_rate_info,

                EIGHT_SHORT_SEQUENCE, NO_TRANSITION, sfb_nrg_act);

            ixheaacd_aac_ec_interpolate(

                &ptr_spec_coeff[window * (num_samples / 8)], &ptr_spec_scale[window],

                &pstr_ec_state->q_spec_coeff[window], &ptr_spec_scale[window], sfb_nrg_prev,

                sfb_nrg_act, total_scale_factor_bands, ptr_sfb_offset);

          }

        } else {

          WORD32 total_scale_factor_bands = pstr_samp_rate_info->num_sfb_1024 - 1;

          const WORD16 *ptr_sfb_offset = pstr_samp_rate_info->ptr_sfb_1024;

          WORD16 spec_scale_out;

          ixheaacd_aac_ec_calc_sfb_nrg(&ptr_spec_coeff[num_samples - (num_samples / 8)],

                                       pstr_samp_rate_info, EIGHT_SHORT_SEQUENCE,

                                       TRANS_SHORT_LONG, sfb_nrg_act);

          ixheaacd_aac_ec_calc_sfb_nrg(pstr_ec_state->spectral_coeff, pstr_samp_rate_info,

                                       ONLY_LONG_SEQUENCE, NO_TRANSITION, sfb_nrg_prev);

          pstr_ics_info->window_shape = 0;

          pstr_ics_info->window_sequence = LONG_STOP_SEQUENCE;

          for (idx = 0; idx < num_samples; idx++) {

            ptr_spec_coeff[idx] = pstr_ec_state->spectral_coeff[idx];

          }

          for (idx = 0; idx < 8; idx++) {

            if (ptr_spec_scale[idx] > ptr_spec_scale[0]) {

              ptr_spec_scale[0] = ptr_spec_scale[idx];

            }

          }

          ixheaacd_aac_ec_interpolate(ptr_spec_coeff, &pstr_ec_state->q_spec_coeff[0],

                                      &ptr_spec_scale[0], &spec_scale_out, sfb_nrg_prev,

                                      sfb_nrg_act, total_scale_factor_bands, ptr_sfb_offset);

          ptr_spec_scale[0] = spec_scale_out;

        }

      } else {

        WORD32 total_scale_factor_bands = pstr_samp_rate_info->num_sfb_1024 - 1;

        const WORD16 *ptr_sfb_offset = pstr_samp_rate_info->ptr_sfb_1024;

        WORD16 spec_scale_act = pstr_ec_state->q_spec_coeff[0];

        ixheaacd_aac_ec_calc_sfb_nrg(ptr_spec_coeff, pstr_samp_rate_info, ONLY_LONG_SEQUENCE,

                                     NO_TRANSITION, sfb_nrg_prev);

        if (pstr_ec_state->win_seq == EIGHT_SHORT_SEQUENCE) {

          pstr_ics_info->window_shape = 1;

          pstr_ics_info->window_sequence = LONG_START_SEQUENCE;

          for (idx = 1; idx < 8; idx++) {

            if (pstr_ec_state->q_spec_coeff[idx] > spec_scale_act) {

              spec_scale_act = pstr_ec_state->q_spec_coeff[idx];

            }

          }

          ixheaacd_aac_ec_calc_sfb_nrg(pstr_ec_state->spectral_coeff, pstr_samp_rate_info,

                                       EIGHT_SHORT_SEQUENCE, TRANS_SHORT_LONG, sfb_nrg_act);

        } else {

          pstr_ics_info->window_shape = 0;

          pstr_ics_info->window_sequence = ONLY_LONG_SEQUENCE;

          ixheaacd_aac_ec_calc_sfb_nrg(pstr_ec_state->spectral_coeff, pstr_samp_rate_info,

                                       ONLY_LONG_SEQUENCE, NO_TRANSITION, sfb_nrg_act);

        }

        ixheaacd_aac_ec_interpolate(ptr_spec_coeff, &ptr_spec_scale[0], &spec_scale_act,

                                    &ptr_spec_scale[0], sfb_nrg_prev, sfb_nrg_act,

                                    total_scale_factor_bands, ptr_sfb_offset);

      }

    }

    ixheaacd_aac_ec_flip_spec_sign(ptr_spec_coeff, num_samples);

  }

  if (FRAME_MUTE == pstr_ec_state->conceal_state) {

    pstr_ics_info->window_shape = pstr_ec_state->win_shape;

    pstr_ics_info->window_sequence = ixheaacd_aac_ec_get_win_seq(pstr_ec_state->win_seq);

    pstr_ec_state->win_seq = pstr_ics_info->window_sequence;

    memset(ptr_spec_coeff, 0, num_samples * sizeof(ptr_spec_coeff[0]));

  }

}

VOID ixheaacd_aac_ec_init(ia_ec_state_str *pstr_ec_state) {

  pstr_ec_state->win_shape = CONCEAL_NOT_DEFINED;

  pstr_ec_state->win_seq = ONLY_LONG_SEQUENCE;

  pstr_ec_state->prev_win_group_len = 1;

  pstr_ec_state->conceal_state = FRAME_OKAY;

  memset(pstr_ec_state->spectral_coeff, 0, sizeof(pstr_ec_state->spectral_coeff));

  memset(pstr_ec_state->q_spec_coeff, 0, sizeof(pstr_ec_state->q_spec_coeff));

  pstr_ec_state->prev_frame_ok[0] = 1;

  pstr_ec_state->prev_frame_ok[1] = 1;

  pstr_ec_state->fade_idx = 0;

}

VOID ixheaacd_aac_apply_ec(ia_ec_state_str *pstr_ec_state,

                           ia_aac_dec_channel_info_struct *pstr_aac_dec_channel_info,

                           const ia_usac_samp_rate_info *pstr_samp_rate_info,

                           const WORD32 num_samples, ia_ics_info_struct *pstr_ics_info,

                           const WORD32 frame_status) {

  if (pstr_ec_state->win_shape == CONCEAL_NOT_DEFINED) {

    pstr_ec_state->win_shape = (UWORD8)pstr_ics_info->window_shape;

  }

  if (frame_status && pstr_ec_state->prev_frame_ok[1]) {

    ixheaacd_aac_ec_store(pstr_ec_state, pstr_aac_dec_channel_info, pstr_ics_info);

  }

  ixheaacd_aac_ec_state(pstr_ec_state, frame_status);

  ixheaacd_aac_ec_interpolate_frame(pstr_ec_state, pstr_aac_dec_channel_info, pstr_samp_rate_info,

                                    num_samples, frame_status, pstr_ics_info);

  pstr_ec_state->prev_frame_ok[0] = pstr_ec_state->prev_frame_ok[1];

  pstr_ec_state->prev_frame_ok[1] = frame_status;

}