/******************************************************************************
 *                                                                            *
 * 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 <stdlib.h>

#include "ixheaac_type_def.h"
#include "ixheaac_constants.h"
#include "ixheaace_aac_constants.h"
#include "ixheaace_bitbuffer.h"

static VOID ia_enhaacplus_enc_update_bitbuf_word_ptr(ixheaace_bit_buf_handle pstr_bit_buf,
                                                     UWORD8 **p_bitbuf_word, WORD32 cnt) {
  *p_bitbuf_word += cnt;

  if (*p_bitbuf_word > pstr_bit_buf->ptr_bit_buf_end) {
    *p_bitbuf_word -= (pstr_bit_buf->ptr_bit_buf_end - pstr_bit_buf->ptr_bit_buf_base + 1);
  }

  if (*p_bitbuf_word < pstr_bit_buf->ptr_bit_buf_base) {
    *p_bitbuf_word += (pstr_bit_buf->ptr_bit_buf_end - pstr_bit_buf->ptr_bit_buf_base + 1);
  }
}

ixheaace_bit_buf_handle ia_enhaacplus_enc_create_bitbuffer(ixheaace_bit_buf_handle pstr_bit_buf,
                                                           UWORD8 *ptr_bit_buf_base,
                                                           UWORD32 bitbuf_size) {
  pstr_bit_buf->ptr_bit_buf_base = ptr_bit_buf_base;

  pstr_bit_buf->ptr_bit_buf_end = ptr_bit_buf_base + bitbuf_size - 1;

  pstr_bit_buf->ptr_read_next = ptr_bit_buf_base;
  pstr_bit_buf->ptr_write_next = ptr_bit_buf_base;

  pstr_bit_buf->write_position = 7;
  pstr_bit_buf->read_position = 7;

  pstr_bit_buf->cnt_bits = 0;

  pstr_bit_buf->size = bitbuf_size * 8;

  return pstr_bit_buf;
}

VOID ia_enhaacplus_enc_delete_bitbuffer(ixheaace_bit_buf_handle pstr_bit_buf) {
  pstr_bit_buf = NULL;
}

VOID ixheaace_reset_bitbuf(ixheaace_bit_buf_handle pstr_bit_buf, UWORD8 *ptr_bit_buf_base,
                           UWORD32 bitbuf_size) {
  pstr_bit_buf->ptr_bit_buf_base = ptr_bit_buf_base;

  pstr_bit_buf->ptr_bit_buf_end = ptr_bit_buf_base + bitbuf_size - 1;

  pstr_bit_buf->ptr_read_next = ptr_bit_buf_base;
  pstr_bit_buf->ptr_write_next = ptr_bit_buf_base;

  pstr_bit_buf->read_position = 7;
  pstr_bit_buf->write_position = 7;

  pstr_bit_buf->cnt_bits = 0;
}
VOID ixheaace_copy_bitbuf_to_and_fro(ixheaace_bit_buf_handle h_bitbuf_src,
                                     ixheaace_bit_buf_handle h_bitbuf_dst)

{
  WORD32 i;
  WORD32 bytes_to_go_src =
      (WORD32)(h_bitbuf_src->ptr_bit_buf_end - h_bitbuf_src->ptr_bit_buf_base);

  UWORD8 *dst = &h_bitbuf_dst->ptr_bit_buf_base[0];
  UWORD8 *src = &h_bitbuf_src->ptr_bit_buf_base[0];
  WORD32 temp321, temp322;
  UWORD8 temp1, temp2, temp3;
  WORD32 count;
  WORD32 remaining;

  count = bytes_to_go_src >> 1;
  remaining = bytes_to_go_src - (count << 1);

  for (i = count - 1; i >= 0; i--) {
    temp1 = *src;
    temp2 = *dst;
    temp3 = *(src + 1);
    *dst++ = temp1;
    temp1 = *dst;
    *src++ = temp2;
    *dst++ = temp3;
    *src++ = temp1;
  }
  if (remaining)
    for (i = remaining - 1; i >= 0; i--) {
      temp1 = *src;
      temp2 = *dst;
      *dst++ = temp1;
      *src++ = temp2;
    }

  src = h_bitbuf_src->ptr_read_next;
  dst = h_bitbuf_dst->ptr_read_next;

  h_bitbuf_dst->ptr_read_next = src;
  h_bitbuf_src->ptr_read_next = dst;

  src = h_bitbuf_src->ptr_write_next;
  dst = h_bitbuf_dst->ptr_write_next;

  h_bitbuf_dst->ptr_write_next = src;
  h_bitbuf_src->ptr_write_next = dst;

  temp321 = h_bitbuf_dst->read_position;
  temp322 = h_bitbuf_src->read_position;

  h_bitbuf_dst->read_position = temp322;
  h_bitbuf_src->read_position = temp321;

  temp321 = h_bitbuf_dst->write_position;
  temp322 = h_bitbuf_src->write_position;

  h_bitbuf_dst->write_position = temp322;
  h_bitbuf_src->write_position = temp321;

  temp321 = h_bitbuf_dst->cnt_bits;
  temp322 = h_bitbuf_src->cnt_bits;

  h_bitbuf_dst->cnt_bits = temp322;
  h_bitbuf_src->cnt_bits = temp321;
}

VOID ia_enhaacplus_enc_copy_bitbuf(ixheaace_bit_buf_handle h_bitbuf_src,
                                   ixheaace_bit_buf_handle h_bitbuf_dst) {
  WORD32 i;
  WORD32 bytes_to_go_src =
      (WORD32)(h_bitbuf_src->ptr_bit_buf_end - h_bitbuf_src->ptr_bit_buf_base);

  for (i = 0; i < bytes_to_go_src; i++) {
    h_bitbuf_dst->ptr_bit_buf_base[i] = h_bitbuf_src->ptr_bit_buf_base[i];
  }

  h_bitbuf_dst->ptr_read_next = h_bitbuf_src->ptr_read_next;
  h_bitbuf_dst->ptr_write_next = h_bitbuf_src->ptr_write_next;

  h_bitbuf_dst->read_position = h_bitbuf_src->read_position;
  h_bitbuf_dst->write_position = h_bitbuf_src->write_position;

  h_bitbuf_dst->cnt_bits = h_bitbuf_src->cnt_bits;
}

WORD32 ia_enhaacplus_enc_get_bits_available(ixheaace_bit_buf_handle pstr_bit_buf_handle) {
  return pstr_bit_buf_handle->cnt_bits;
}
UWORD32
ixheaace_readbits(ixheaace_bit_buf_handle pstr_bit_buf, UWORD8 no_bits_to_read) {
  UWORD32 return_value;

  if (no_bits_to_read >= 25) {
    return 0;
  }

  pstr_bit_buf->cnt_bits -= no_bits_to_read;
  pstr_bit_buf->read_position -= no_bits_to_read;

  return_value = (UWORD32)*pstr_bit_buf->ptr_read_next;

  while (pstr_bit_buf->read_position < 0) {
    pstr_bit_buf->read_position += 8;
    pstr_bit_buf->ptr_read_next++;

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

    return_value <<= 8;

    return_value |= (UWORD32)*pstr_bit_buf->ptr_read_next;
  }
  return_value = return_value << (31 - no_bits_to_read - pstr_bit_buf->read_position) >>
                 (32 - no_bits_to_read);

  return (return_value);
}

VOID ia_enhaacplus_enc_wind_bitbuffer_bidirectional(ixheaace_bit_buf_handle pstr_bit_buf,
                                                    WORD32 offset) {
  if (offset != 0) {
    WORD32 buff_offset;
    pstr_bit_buf->read_position -= offset;
    buff_offset = (pstr_bit_buf->read_position) >> 3;
    pstr_bit_buf->read_position -= buff_offset << 3;

    if (buff_offset) {
      ia_enhaacplus_enc_update_bitbuf_word_ptr(pstr_bit_buf, &pstr_bit_buf->ptr_read_next,
                                               -buff_offset);
    }

    pstr_bit_buf->cnt_bits -= offset;
  }
}