/******************************************************************************
 *
 * 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 <stdio.h>
#include <stdlib.h>
#include <math.h>
#include <assert.h>
#include <string.h>

#include "impd_type_def.h"
#include "impd_drc_bitbuffer.h"
#include "impd_drc_extr_delta_coded_info.h"
#include "impd_drc_common.h"
#include "impd_drc_struct.h"
#include "impd_drc_parser.h"

WORD32 impd_read_bits_buf(ia_bit_buf_struct* it_bit_buff, WORD no_of_bits) {
  UWORD32 ret_val;
  UWORD8* ptr_read_next = it_bit_buff->ptr_read_next;
  WORD bit_pos = it_bit_buff->bit_pos;
  it_bit_buff->error = 0;

  if (it_bit_buff->cnt_bits <= 0) {
    it_bit_buff->error = 1;
    return -1;
  }

  if (no_of_bits == 0) {
    return 0;
  }

  it_bit_buff->cnt_bits -= no_of_bits;
  ret_val = (UWORD32)*ptr_read_next;

  bit_pos -= no_of_bits;
  while (bit_pos < 0) {
    bit_pos += 8;
    ptr_read_next++;

    if (ptr_read_next > it_bit_buff->ptr_bit_buf_end) {
      ptr_read_next = it_bit_buff->ptr_bit_buf_base;
    }

    ret_val <<= 8;

    ret_val |= (UWORD32)*ptr_read_next;
  }

  ret_val = ret_val << ((31 - no_of_bits) - bit_pos) >> (32 - no_of_bits);
  it_bit_buff->ptr_read_next = ptr_read_next;
  it_bit_buff->bit_pos = (WORD16)bit_pos;
  return ret_val;
}

ia_bit_buf_struct* impd_create_bit_buf(ia_bit_buf_struct* it_bit_buff,
                                       UWORD8* ptr_bit_buf_base,
                                       WORD32 bit_buf_size) {
  it_bit_buff->ptr_bit_buf_base = ptr_bit_buf_base;
  it_bit_buff->ptr_bit_buf_end = ptr_bit_buf_base + bit_buf_size - 1;

  it_bit_buff->ptr_read_next = ptr_bit_buf_base;
  it_bit_buff->bit_pos = 7;

  it_bit_buff->cnt_bits = 0;
  it_bit_buff->size = bit_buf_size << 3;

  return it_bit_buff;
}

ia_bit_buf_struct* impd_create_init_bit_buf(ia_bit_buf_struct* it_bit_buff,
                                            UWORD8* ptr_bit_buf_base,
                                            WORD32 bit_buf_size) {
  impd_create_bit_buf(it_bit_buff, ptr_bit_buf_base, bit_buf_size);
  it_bit_buff->cnt_bits = (bit_buf_size << 3);
  return (it_bit_buff);
}

WORD32 impd_init_drc_bitstream_dec(ia_drc_bits_dec_struct* p_drc_bs_dec_struct,
                                   WORD32 sample_rate, WORD32 frame_size,
                                   WORD32 delay_mode,
                                   WORD32 lfe_channel_map_count,
                                   WORD32* lfe_channel_map) {
  WORD32 i, err_code = 0;

  ia_drc_params_bs_dec_struct* ia_drc_params_struct =
      &p_drc_bs_dec_struct->ia_drc_params_struct;
  ia_drc_params_struct->drc_frame_size = frame_size;
  ia_drc_params_struct->delta_tmin_default = impd_get_delta_tmin(sample_rate);
  ia_drc_params_struct->num_gain_values_max_default =
      ia_drc_params_struct->drc_frame_size /
      ia_drc_params_struct->delta_tmin_default;
  ia_drc_params_struct->delay_mode = delay_mode;

  if ((frame_size < 1) || (frame_size > AUDIO_CODEC_FRAME_SIZE_MAX) ||
      (ia_drc_params_struct->drc_frame_size < 0.001f * sample_rate)) {
    return -1;
  }
  if (sample_rate < 1000) {
    return -1;
  }

  if (ia_drc_params_struct->delta_tmin_default >
      ia_drc_params_struct->drc_frame_size) {
    return -1;
  }

  if (lfe_channel_map_count >= 0) {
    if ((lfe_channel_map == NULL) ||
        (lfe_channel_map_count > MAX_CHANNEL_COUNT)) {
      return (-1);
    }

    ia_drc_params_struct->lfe_channel_map_count = lfe_channel_map_count;

    for (i = 0; i < lfe_channel_map_count; i++) {
      ia_drc_params_struct->lfe_channel_map[i] = lfe_channel_map[i];
    }
  } else {
    ia_drc_params_struct->lfe_channel_map_count = -1;

    for (i = 0; i < MAX_CHANNEL_COUNT; i++) {
      ia_drc_params_struct->lfe_channel_map[i] = 0;
    }
  }

  err_code = impd_init_tbls(ia_drc_params_struct->num_gain_values_max_default,
                            &p_drc_bs_dec_struct->tables_default);

  return err_code;
}

WORD32 impd_process_drc_bitstream_dec(
    ia_drc_bits_dec_struct* p_drc_bs_dec_struct, ia_bit_buf_struct* it_bit_buff,
    ia_drc_config* pstr_drc_config,
    ia_drc_loudness_info_set_struct* pstr_loudness_info,
    UWORD8* bitstream_config, WORD32 num_bytes, WORD32 num_bits_offset,
    WORD32* num_bits_read) {
  WORD32 err_code = 0;

  WORD32 loudness_info_set_present, drc_config_present, dummy;

  if (bitstream_config == NULL) {
    *num_bits_read = 0;
  } else {
    it_bit_buff =
        impd_create_init_bit_buf(it_bit_buff, bitstream_config, num_bytes);

    dummy = impd_read_bits_buf(it_bit_buff, num_bits_offset);
    if (it_bit_buff->error) return it_bit_buff->error;

    loudness_info_set_present = impd_read_bits_buf(it_bit_buff, 1);
    if (it_bit_buff->error) return it_bit_buff->error;

    if (loudness_info_set_present) {
      drc_config_present = impd_read_bits_buf(it_bit_buff, 1);
      if (it_bit_buff->error) return it_bit_buff->error;

      if (drc_config_present) {
        err_code = impd_parse_drc_config(
            it_bit_buff, &p_drc_bs_dec_struct->ia_drc_params_struct,
            pstr_drc_config);

        if (err_code) return (err_code);
      }

      err_code = impd_parse_loudness_info_set(it_bit_buff, pstr_loudness_info);

      if (err_code) return (err_code);
    }

    *num_bits_read = it_bit_buff->size - it_bit_buff->cnt_bits;
  }

  return err_code;
}

WORD32 impd_process_drc_bitstream_dec_config(
    ia_drc_bits_dec_struct* p_drc_bs_dec_struct, ia_bit_buf_struct* it_bit_buff,
    ia_drc_config* pstr_drc_config, UWORD8* bitstream_config,
    WORD32 num_bytes) {
  WORD32 err_code = 0;

  it_bit_buff =
      impd_create_init_bit_buf(it_bit_buff, bitstream_config, num_bytes);

  err_code = impd_parse_drc_config(
      it_bit_buff, &p_drc_bs_dec_struct->ia_drc_params_struct, pstr_drc_config);
  if (err_code) return (err_code);

  return err_code;
}

WORD32 impd_process_drc_bitstream_dec_gain(
    ia_drc_bits_dec_struct* p_drc_bs_dec_struct, ia_bit_buf_struct* it_bit_buff,
    ia_drc_config* pstr_drc_config, ia_drc_gain_struct* pstr_drc_gain,
    UWORD8* bitstream_gain, WORD32 num_bytes, WORD32 num_bits_offset,
    WORD32* num_bits_read) {
  WORD32 err_code = 0;

  WORD32 dummy;

  it_bit_buff =
      impd_create_init_bit_buf(it_bit_buff, bitstream_gain, num_bytes);

  dummy = impd_read_bits_buf(it_bit_buff, num_bits_offset);
  if (it_bit_buff->error) return it_bit_buff->error;

  err_code = impd_drc_uni_gain_read(it_bit_buff, p_drc_bs_dec_struct,
                                    pstr_drc_config, pstr_drc_gain);

  if (err_code > PROC_COMPLETE) return (err_code);

  *num_bits_read = (it_bit_buff->size) - it_bit_buff->cnt_bits;

  if (err_code == PROC_COMPLETE) {
    return err_code;
  }

  return 0;
}

WORD32 impd_process_drc_bitstream_dec_loudness_info_set(
    ia_bit_buf_struct* it_bit_buff,
    ia_drc_loudness_info_set_struct* pstr_loudness_info,
    UWORD8* bit_stream_loudness, WORD32 num_bytes_loudness) {
  WORD32 err_code = 0;

  it_bit_buff = impd_create_init_bit_buf(it_bit_buff, bit_stream_loudness,
                                         num_bytes_loudness);

  err_code = impd_parse_loudness_info_set(it_bit_buff, pstr_loudness_info);
  if (err_code) return (err_code);

  return 0;
}