/******************************************************************************
 *
 * 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 "ixheaacd_type_def.h"
#include "ixheaacd_mps_struct_def.h"
#include "ixheaacd_error_codes.h"
#include "ixheaacd_mps_res_rom.h"
#include "ixheaacd_mps_aac_struct.h"
#include "ixheaacd_constants.h"
#include "ixheaacd_basic_ops32.h"
#include "ixheaacd_basic_ops40.h"
#include "ixheaacd_bitbuffer.h"
#include "ixheaacd_common_rom.h"
#include "ixheaacd_sbrdecsettings.h"
#include "ixheaacd_sbr_scale.h"
#include "ixheaacd_env_extr_part.h"
#include "ixheaacd_sbr_rom.h"
#include "ixheaacd_hybrid.h"
#include "ixheaacd_ps_dec.h"
#include "ixheaacd_error_standards.h"
#include "ixheaacd_mps_polyphase.h"
#include "ixheaacd_config.h"
#include "ixheaacd_qmf_dec.h"
#include "ixheaacd_mps_dec.h"
#include "ixheaacd_mps_macro_def.h"
#include "ixheaacd_mps_bitdec.h"
#include "ixheaacd_mps_calc_m1m2_tree_config.h"

VOID ixheaacd_buffer_m1(ia_heaac_mps_state_struct *pstr_mps_state) {
  ia_mps_dec_m1_param_struct *m1_param = pstr_mps_state->array_struct->m1_param;
  WORD32 pb, row, col;

  ia_mps_persistent_mem *persistent_mem = &pstr_mps_state->mps_persistent_mem;
  WORD32 *m1_param_real_prev = persistent_mem->m1_param_real_prev;
  WORD32 *m1_param_imag_prev = persistent_mem->m1_param_imag_prev;

  WORD32 *m1_param_real, *m1_param_imag;

  WORD32 *p_m1_param_real, *p_m1_param_re;

  WORD32 num_parameter_bands = pstr_mps_state->num_parameter_bands;
  WORD32 num_v_channels = pstr_mps_state->num_v_channels;
  WORD32 num_x_channels = pstr_mps_state->num_x_channels;
  WORD32 m1_param_imag_present = pstr_mps_state->m1_param_imag_present;

  WORD32 num_parameter_sets_prev;

  pstr_mps_state->num_parameter_sets_prev = pstr_mps_state->num_parameter_sets;
  num_parameter_sets_prev = pstr_mps_state->num_parameter_sets_prev;

  if (m1_param_imag_present) {
    WORD32 *p_m1_param_imag = &m1_param->m1_param_imag[0][0][0][0];
    p_m1_param_real = &m1_param->m1_param_real[0][0][0][0];
    for (row = 0; row < num_v_channels; row++) {
      WORD32 *p_m1_param_im = p_m1_param_imag;
      p_m1_param_re = p_m1_param_real;

      for (col = 0; col < num_x_channels; col++) {
        m1_param_real = p_m1_param_re + (num_parameter_sets_prev - 1) * MAX_PARAMETER_BANDS;
        m1_param_imag = p_m1_param_im + (num_parameter_sets_prev - 1) * MAX_PARAMETER_BANDS;

        for (pb = 0; pb < num_parameter_bands; pb++) {
          *m1_param_real_prev++ = *m1_param_real++;
          *m1_param_imag_prev++ = *m1_param_imag++;
        }
        p_m1_param_re += PBXPS;
        p_m1_param_im += PBXPS;
      }
      p_m1_param_real += MAX_INPUT_CHANNELS_MPS * PBXPS;
      p_m1_param_imag += MAX_INPUT_CHANNELS_MPS * PBXPS;
    }
  } else {
    p_m1_param_real = &m1_param->m1_param_real[0][0][0][0];
    for (row = 0; row < num_v_channels; row++) {
      p_m1_param_re = p_m1_param_real;
      for (col = 0; col < num_x_channels; col++) {
        m1_param_real = p_m1_param_re + (num_parameter_sets_prev - 1) * MAX_PARAMETER_BANDS;
        for (pb = 0; pb < num_parameter_bands; pb++) *m1_param_real_prev++ = *m1_param_real++;

        p_m1_param_re += PBXPS;
      }
      p_m1_param_real += MAX_INPUT_CHANNELS_MPS * PBXPS;
    }
  }
}

VOID ixheaacd_buffer_m2(ia_heaac_mps_state_struct *pstr_mps_state) {
  WORD32 num_direct_signals = pstr_mps_state->num_direct_signals;
  WORD32 pb, row, col, col_counter = num_direct_signals + pstr_mps_state->num_decor_signals;

  ia_mps_persistent_mem *persistent_mem = &pstr_mps_state->mps_persistent_mem;
  ia_mps_dec_m2_param_struct *m2_param = pstr_mps_state->aux_struct->m2_param;

  WORD32 *m2_decor_real_prev = persistent_mem->m2_decor_real_prev;
  WORD32 *m2_decor_imag_prev = persistent_mem->m2_decor_imag_prev;

  WORD32 *m2_resid_real_prev = persistent_mem->m2_resid_real_prev;
  WORD32 *m2_resid_imag_prev = persistent_mem->m2_resid_imag_prev;

  WORD32 *m2_decor_real, *m2_decor_imag, *m2_resid_real, *m2_resid_imag;
  WORD32 idx = -1;

  WORD32 resid_col_counter;

  WORD32 num_parameter_bands = pstr_mps_state->num_parameter_bands;
  WORD32 num_output_channels = pstr_mps_state->num_output_channels;
  WORD32 num_parameter_sets_prev = pstr_mps_state->num_parameter_sets_prev;

  if (pstr_mps_state->residual_coding)
    resid_col_counter = col_counter;
  else
    resid_col_counter = num_direct_signals;

  if (pstr_mps_state->m2_param_imag_present) {
    for (row = 0; row < num_output_channels; row++) {
      for (col = num_direct_signals; col < col_counter; col++) {
        if (pstr_mps_state->m2_param_present[row][col] & 1) {
          idx++;
          m2_decor_real = &m2_param->m2_decor_real[idx][num_parameter_sets_prev - 1][0];

          for (pb = 0; pb < num_parameter_bands; pb++) *m2_decor_real_prev++ = *m2_decor_real++;

          m2_decor_imag = &m2_param->m2_decor_imag[idx][num_parameter_sets_prev - 1][0];

          for (pb = 0; pb < num_parameter_bands; pb++) *m2_decor_imag_prev++ = *m2_decor_imag++;
        }
      }
    }

    idx = -1;
    for (row = 0; row < num_output_channels; row++) {
      for (col = 0; col < resid_col_counter; col++) {
        if (pstr_mps_state->m2_param_present[row][col] & 2) {
          idx++;
          m2_resid_real = &m2_param->m2_resid_real[idx][num_parameter_sets_prev - 1][0];

          for (pb = 0; pb < num_parameter_bands; pb++) *m2_resid_real_prev++ = *m2_resid_real++;
          m2_resid_imag = &m2_param->m2_resid_imag[idx][num_parameter_sets_prev - 1][0];
          for (pb = 0; pb < num_parameter_bands; pb++) *m2_resid_imag_prev++ = *m2_resid_imag++;
        }
      }
    }
  } else {
    for (row = 0; row < num_output_channels; row++) {
      for (col = num_direct_signals; col < col_counter; col++) {
        if (pstr_mps_state->m2_param_present[row][col] & 1) {
          idx++;
          m2_decor_real = &m2_param->m2_decor_real[idx][num_parameter_sets_prev - 1][0];

          for (pb = 0; pb < num_parameter_bands; pb++) *m2_decor_real_prev++ = *m2_decor_real++;
        }
      }
    }

    idx = -1;
    for (row = 0; row < num_output_channels; row++) {
      for (col = 0; col < resid_col_counter; col++) {
        if (pstr_mps_state->m2_param_present[row][col] & 2) {
          idx++;
          m2_resid_real = &m2_param->m2_resid_real[idx][num_parameter_sets_prev - 1][0];

          for (pb = 0; pb < num_parameter_bands; pb++) *m2_resid_real_prev++ = *m2_resid_real++;
        }
      }
    }
  }
}

static VOID ixheaacd_update_alpha(ia_heaac_mps_state_struct *pstr_mps_state) {
  WORD32 alpha;

  WORD32 *arbdmx_alpha_prev = pstr_mps_state->mps_persistent_mem.arbdmx_alpha_prev;
  ia_mps_dec_auxilary_struct *p_aux_struct = pstr_mps_state->aux_struct;
  WORD32 *arbdmx_alpha = p_aux_struct->arbdmx_alpha;

  WORD32 n_ch_in = pstr_mps_state->num_input_channels;
  WORD32 ch;

  for (ch = 0; ch < n_ch_in; ch++) {
    *arbdmx_alpha_prev++ = arbdmx_alpha[ch];

    if (pstr_mps_state->arbitrary_downmix == 2) {
      alpha = arbdmx_alpha[ch];

      if (p_aux_struct->arbdmx_residual_abs[ch]) {
        alpha -= POINT_THREE_THREE_Q15;

        if (alpha < 0) alpha = 0;
      } else {
        alpha += POINT_THREE_THREE_Q15;

        if (alpha > ONE_IN_Q15) alpha = ONE_IN_Q15;
      }
    } else {
      alpha = ONE_IN_Q15;
    }

    arbdmx_alpha[ch] = alpha;
  }
}

VOID ixheaacd_calc_m1m2(ia_heaac_mps_state_struct *pstr_mps_state) {
  WORD32 up_mix_type = pstr_mps_state->up_mix_type;
  WORD32 binaural_quality = pstr_mps_state->binaural_quality;

  if (pstr_mps_state->arbitrary_downmix != 0) {
    ixheaacd_update_alpha(pstr_mps_state);
  }

  switch (pstr_mps_state->tree_config) {
    case TREE_5151: {
      if (up_mix_type == 3) {
        ixheaacd_calc_m1m2_51s1(pstr_mps_state);
      } else {
        ixheaacd_calc_m1m2_5151(pstr_mps_state);
      }
    } break;
    case TREE_5152: {
      if (up_mix_type == 3) {
        ixheaacd_calc_m1m2_51s2(pstr_mps_state);
      } else {
        ixheaacd_calc_m1m2_5152(pstr_mps_state);
      }
    } break;
    case TREE_525:
      if (up_mix_type == 1) {
        ixheaacd_calc_m1m2_emm(pstr_mps_state);
      } else if (up_mix_type == 2) {
        if (binaural_quality == 1) {
          ixheaacd_calc_m1m2_5227(pstr_mps_state);
        }
      } else {
        ixheaacd_calc_m1m2_5251(pstr_mps_state);
      }
      break;
    case TREE_7271:
      if (up_mix_type == 0) {
        ixheaacd_calc_m1m2_7271(pstr_mps_state);
      } else if (up_mix_type == 2) {
        if (binaural_quality == 1) {
          ixheaacd_calc_m1m2_5227(pstr_mps_state);
        }
      }
      break;
    case TREE_7272:
      if (up_mix_type == 0) {
        ixheaacd_calc_m1m2_7272(pstr_mps_state);
      } else if (up_mix_type == 2) {
        if (binaural_quality == 1) {
          ixheaacd_calc_m1m2_5227(pstr_mps_state);
        }
      }
      break;
    case TREE_7571:
      ixheaacd_calc_m1m2_7571(pstr_mps_state);
      break;
    case TREE_7572:
      ixheaacd_calc_m1m2_7572(pstr_mps_state);
      break;
    default:
      break;
  };

  return;
}