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

 *                                                                            *

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

#include <assert.h>

#include "ixheaacd_sbr_common.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 "ixheaac_basic_op.h"

#include "ixheaacd_intrinsics.h"

#include "ixheaacd_bitbuffer.h"

#include "ixheaacd_adts_crc_check.h"

#include "ixheaacd_error_codes.h"

VOID ixheaacd_byte_align(ia_bit_buf_struct *it_bit_buff,

                         WORD32 *align_bits_cnt) {

  WORD alignment;

  alignment = (WORD)((*align_bits_cnt - it_bit_buff->cnt_bits) & 0x07);

  if (alignment) {

    ixheaacd_read_bits_buf(it_bit_buff, (8 - alignment));

  }

  *align_bits_cnt = it_bit_buff->cnt_bits;

}

WORD32 ixheaacd_skip_bits_buf(ia_bit_buf_struct *it_bit_buff, WORD no_of_bits) {

  UWORD8 *ptr_read_next = it_bit_buff->ptr_read_next;

  WORD bit_pos = it_bit_buff->bit_pos;

  if (it_bit_buff->cnt_bits < no_of_bits || it_bit_buff->cnt_bits < 0)

    longjmp(*(it_bit_buff->xaac_jmp_buf),

            IA_XHEAAC_DEC_EXE_NONFATAL_INSUFFICIENT_INPUT_BYTES);

  it_bit_buff->cnt_bits -= no_of_bits;

  ptr_read_next += no_of_bits / 8;

  bit_pos -= (no_of_bits % 8);

  if (bit_pos < 0) {

    bit_pos += 8;

    ptr_read_next++;

  }

  assert(bit_pos >= 0 && bit_pos <= 7);

  it_bit_buff->ptr_read_next = ptr_read_next;

  it_bit_buff->bit_pos = (WORD16)bit_pos;

  return no_of_bits;

}

WORD32 ixheaacd_show_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;

  if (no_of_bits == 0) {

    return 0;

  }

  if (it_bit_buff->cnt_bits < no_of_bits || it_bit_buff->cnt_bits < 0 ||

      no_of_bits > 25) {

    longjmp(*(it_bit_buff->xaac_jmp_buf),

            IA_XHEAAC_DEC_EXE_NONFATAL_INSUFFICIENT_INPUT_BYTES);

  }

  ret_val = (UWORD32)*ptr_read_next;

  bit_pos -= no_of_bits;

  while (bit_pos < -1) {

    bit_pos += 8;

    ptr_read_next++;

    ret_val <<= 8;

    ret_val |= (UWORD32)*ptr_read_next;

  }

  if (bit_pos == -1) {

    bit_pos += 8;

    ret_val <<= 8;

    ptr_read_next++;

  }

  ret_val = ret_val << ((31 - no_of_bits) - bit_pos) >> (32 - no_of_bits);

  return ret_val;

}

WORD32 ixheaacd_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;

  if (no_of_bits == 0) {

    return 0;

  }

  if (it_bit_buff->cnt_bits < no_of_bits || it_bit_buff->cnt_bits < 0 ||

      no_of_bits > 25) {

    longjmp(*(it_bit_buff->xaac_jmp_buf),

            IA_XHEAAC_DEC_EXE_NONFATAL_INSUFFICIENT_INPUT_BYTES);

  }

  it_bit_buff->cnt_bits -= no_of_bits;

  ret_val = (UWORD32)*ptr_read_next;

  bit_pos -= no_of_bits;

  while (bit_pos < -1) {

    bit_pos += 8;

    ptr_read_next++;

    ret_val <<= 8;

    ret_val |= (UWORD32)*ptr_read_next;

  }

  if (bit_pos == -1) {

    bit_pos += 8;

    ret_val <<= 8;

    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;

}

VOID ixheaacd_aac_read_byte(UWORD8 **ptr_read_next, WORD32 *bit_pos,

                            WORD32 *readword) {

  UWORD8 *v = *ptr_read_next;

  WORD32 bits_consumed = *bit_pos;

  if ((bits_consumed -= 8) >= 0) {

    *readword = (*readword << 8) | *v;

    v++;

  } else {

    bits_consumed += 8;

  }

  *bit_pos = bits_consumed;

  *ptr_read_next = v;

  return;

}

VOID ixheaacd_aac_read_byte_corr1(UWORD8 **ptr_read_next, WORD32 *ptr_bit_pos,

                                  WORD32 *readword, UWORD8 *p_bit_buf_end) {

  UWORD8 *v = *ptr_read_next;

  WORD32 bits_consumed = *ptr_bit_pos;

  WORD32 temp_bit_count = 0;

  while (bits_consumed >= 8) {

    bits_consumed -= 8;

    if ((p_bit_buf_end < v) && (p_bit_buf_end != 0))

      temp_bit_count += 8;

    else {

      *readword = (*readword << 8) | *v;

      v++;

    }

  }

  if (bits_consumed > (31 - temp_bit_count)) {

    if ((p_bit_buf_end != NULL) && (p_bit_buf_end < v)) {

      bits_consumed = 31 - temp_bit_count;

    }

  }

  *ptr_bit_pos = bits_consumed + temp_bit_count;

  *ptr_read_next = v;

  return;

}

VOID ixheaacd_aac_read_byte_corr(UWORD8 **ptr_read_next, WORD32 *ptr_bit_pos,

                                 WORD32 *readword, UWORD8 *p_bit_buf_end) {

  UWORD8 *v = *ptr_read_next;

  WORD32 bits_consumed = *ptr_bit_pos;

  if ((bits_consumed -= 8) >= 0) {

    if (p_bit_buf_end < v)

      bits_consumed += 8;

    else {

      *readword = (*readword << 8) | *v;

      v++;

    }

  } else {

    bits_consumed += 8;

  }

  if (bits_consumed > 31) {

    if (p_bit_buf_end < v) {

      bits_consumed = 31;

    }

  }

  *ptr_bit_pos = bits_consumed;

  *ptr_read_next = v;

  return;

}

WORD32 ixheaacd_aac_read_bit(ia_bit_buf_struct *it_bit_buff) {

  UWORD8 ret_val;

  UWORD8 *ptr_read_next = it_bit_buff->ptr_read_next;

  WORD bit_pos = it_bit_buff->bit_pos;

  UWORD32 temp;

  WORD no_of_bits = 1;

  if (bit_pos < 0) {

    bit_pos = 7;

    ptr_read_next--;

  }

  if (ptr_read_next < it_bit_buff->ptr_bit_buf_base) {

    longjmp(*(it_bit_buff->xaac_jmp_buf),

            IA_XHEAAC_DEC_EXE_NONFATAL_INSUFFICIENT_INPUT_BYTES);

  }

  it_bit_buff->cnt_bits += no_of_bits;

  ret_val = *ptr_read_next;

  bit_pos -= no_of_bits;

  temp = (ret_val << 24) << (bit_pos + no_of_bits);

  it_bit_buff->ptr_read_next = ptr_read_next;

  it_bit_buff->bit_pos = (WORD16)bit_pos;

  return temp >> (32 - no_of_bits);

}

WORD32 ixheaacd_aac_read_bit_rev(ia_bit_buf_struct *it_bit_buff) {

  UWORD8 ret_val;

  UWORD8 *ptr_read_next = it_bit_buff->ptr_read_next;

  WORD bit_pos = it_bit_buff->bit_pos;

  UWORD32 temp;

  WORD no_of_bits = 1;

  if (it_bit_buff->cnt_bits < no_of_bits || it_bit_buff->cnt_bits < 0) {

    longjmp(*(it_bit_buff->xaac_jmp_buf),

            IA_XHEAAC_DEC_EXE_NONFATAL_INSUFFICIENT_INPUT_BYTES);

  }

  if (bit_pos >= 8) {

    bit_pos -= 8;

    ptr_read_next++;

  }

  it_bit_buff->cnt_bits -= no_of_bits;

  ret_val = *ptr_read_next;

  bit_pos += no_of_bits;

  temp = (ret_val << 24) << (bit_pos - no_of_bits);

  it_bit_buff->ptr_read_next = ptr_read_next;

  it_bit_buff->bit_pos = (WORD16)bit_pos;

  return temp >> (32 - no_of_bits);

}

VOID ixheaacd_write_bit(ia_bit_buf_struct *it_bit_buff, WORD32 value,

                        WORD32 no_of_bits)

{

  WORD32 mask;

  if (no_of_bits == 0) return;

  mask = 0x1;

  mask <<= no_of_bits - 1;

  it_bit_buff->bit_count += no_of_bits;

  while (no_of_bits > 0) {

    while (no_of_bits > 0 && it_bit_buff->valid_bits < 8) {

      it_bit_buff->byte <<= 1;

      if (value & mask) it_bit_buff->byte |= 0x1;

      value <<= 1;

      no_of_bits--;

      it_bit_buff->valid_bits++;

    }

    if (it_bit_buff->valid_bits == 8) {

      *it_bit_buff->byte_ptr++ = it_bit_buff->byte;

      it_bit_buff->byte = 0;

      it_bit_buff->valid_bits = 0;

    }

  }

}

WORD32 ixheaacd_read_bit(ia_bit_buf_struct *it_bit_buff, WORD32 no_of_bits) {

  UWORD32 ret_val;

  UWORD8 *ptr_read_next = it_bit_buff->byte_ptr;

  if (no_of_bits == 0) {

    return 0;

  }

  ret_val =

      ixheaacd_aac_showbits_32(ptr_read_next, it_bit_buff->bit_count, NULL);

  it_bit_buff->byte_ptr += (no_of_bits >> 3);

  if (it_bit_buff->valid_bits != 8) {

    UWORD8 *v = it_bit_buff->byte_ptr;

    ret_val = (ret_val << (8 - it_bit_buff->valid_bits)) |

              (*v >> it_bit_buff->valid_bits);

  }

  it_bit_buff->valid_bits -= (no_of_bits % 8);

  ret_val = ret_val >> (32 - no_of_bits);

  return ret_val;

}