decoder/ixheaacd_mps_res_tns.c

/******************************************************************************
 *
 * 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 "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"

VOID ixheaacd_res_tns_parcor_2_lpc_32x16(WORD16 *parcor, WORD16 *lpc, WORD16 *scale, WORD order)

{
  WORD i, j, status;
  WORD32 z1;
  WORD16 z[MAX_ORDER + 1];
  WORD16 w[MAX_ORDER + 1];
  WORD32 accu1, accu2;

  status = 1;
  *scale = 0;
  while (status) {
    status = 0;

    for (i = MAX_ORDER; i >= 0; i--) {
      z[i] = 0;
      w[i] = 0;
    }

    accu1 = (0x7fffffff >> *scale);

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

      for (j = 0; j < order; j++) {
        w[j] = ixheaac_round16(accu1);

        accu1 = ixheaac_mac16x16in32_shl_sat(accu1, parcor[j], z[j]);
        if (ixheaac_abs32_sat(accu1) == 0x7fffffff) status = 1;
      }
      for (j = (order - 1); j >= 0; j--) {
        accu2 = ixheaac_deposit16h_in32(z[j]);
        accu2 = ixheaac_mac16x16in32_shl_sat(accu2, parcor[j], w[j]);
        z[j + 1] = ixheaac_round16(accu2);
        if (ixheaac_abs32_sat(accu2) == 0x7fffffff) status = 1;
      }

      z[0] = ixheaac_round16(z1);
      lpc[i] = ixheaac_round16(accu1);
      accu1 = 0;
    }
    accu1 = (status - 1);
    if (accu1 == 0) {
      *scale = *scale + 1;
    }
  }
}

VOID ixheaacd_res_tns_ar_filter_fixed_32x16(WORD32 *spectrum, WORD32 size, WORD32 inc,
                                            WORD16 *lpc, WORD32 order, WORD32 shift_value,
                                            WORD scale_spec) {
  WORD32 i, j;
  WORD32 y, state[MAX_ORDER + 1];

  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 = (*spectrum) << scale_spec;
    for (j = i; j > 0; j--) {
      y = ixheaac_sub32_sat(y, ixheaac_mult32x16in32_shl_sat(state[j - 1], lpc[j]));
      state[j] = state[j - 1];
    }

    state[0] = ixheaac_shl32_dir_sat_limit(y, shift_value);
    *spectrum = y >> scale_spec;
    spectrum += inc;
  }

  for (i = order; i < size; i++) {
    y = (*spectrum) << scale_spec;

    for (j = order; j > 0; j--) {
      y = ixheaac_sub32_sat(y, ixheaac_mult32x16in32_shl_sat(state[j - 1], lpc[j]));
      state[j] = state[j - 1];
    }

    state[0] = ixheaac_shl32_dir_sat_limit(y, shift_value);
    *spectrum = y >> scale_spec;
    spectrum += inc;
  }
}

WORD32 ixheaacd_res_calc_max_spectral_line(WORD32 *p_tmp, WORD32 size) {
  WORD32 max_spectral_line = 0, i;
  WORD count, remaining, temp_1, temp_2, temp3, temp4;

  count = size >> 3;
  for (i = count; i--;) {
    temp_1 = *p_tmp++;
    temp_2 = *p_tmp++;
    temp3 = *p_tmp++;
    temp4 = *p_tmp++;

    max_spectral_line = ixheaac_abs32_nrm(temp_1) | max_spectral_line;
    max_spectral_line = ixheaac_abs32_nrm(temp_2) | max_spectral_line;
    max_spectral_line = ixheaac_abs32_nrm(temp3) | max_spectral_line;
    max_spectral_line = ixheaac_abs32_nrm(temp4) | max_spectral_line;
    temp_1 = *p_tmp++;
    temp_2 = *p_tmp++;
    temp3 = *p_tmp++;
    temp4 = *p_tmp++;

    max_spectral_line = ixheaac_abs32_nrm(temp_1) | max_spectral_line;
    max_spectral_line = ixheaac_abs32_nrm(temp_2) | max_spectral_line;
    max_spectral_line = ixheaac_abs32_nrm(temp3) | max_spectral_line;
    max_spectral_line = ixheaac_abs32_nrm(temp4) | max_spectral_line;
  }

  remaining = size - (count << 3);
  if (remaining) {
    for (i = remaining; i--;) {
      max_spectral_line = ixheaac_abs32_nrm(*p_tmp) | max_spectral_line;
      p_tmp++;
    }
  }

  return ixheaac_norm32(max_spectral_line);
}