dom/media/BitReader.h
author Johan Lorenzo <jlorenzo@mozilla.com>
Mon, 17 Feb 2020 10:07:08 +0000
changeset 524704 57352ccfac02e1a71512e414ba9af3fa35ab9168
parent 490034 0ceae9db9ec0be18daa1a279511ad305723185d4
permissions -rw-r--r--
Bug 1614287 - Remove Fennec Beta target_tasks so it does not mess up the Fenix migration r=rail a=jcristau Differential Revision: https://phabricator.services.mozilla.com/D62249

/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#ifndef BIT_READER_H_
#define BIT_READER_H_

#include "MediaData.h"

namespace mozilla {

class BitReader {
 public:
  explicit BitReader(const MediaByteBuffer* aBuffer);
  BitReader(const MediaByteBuffer* aBuffer, size_t aBits);
  BitReader(const uint8_t* aBuffer, size_t aBits);
  ~BitReader();
  uint32_t ReadBits(size_t aNum);
  bool ReadBit() { return ReadBits(1) != 0; }
  uint32_t ReadU32() { return ReadBits(32); }
  uint64_t ReadU64();

  // Read the UTF-8 sequence and convert it to its 64-bit UCS-4 encoded form.
  // Return 0xfffffffffffffff if sequence was invalid.
  uint64_t ReadUTF8();
  // Read unsigned integer Exp-Golomb-coded.
  uint32_t ReadUE();
  // Read signed integer Exp-Golomb-coded.
  int32_t ReadSE();

  // Return the number of bits parsed so far;
  size_t BitCount() const;
  // Return the number of bits left.
  size_t BitsLeft() const;

  // Return RBSP bit length.
  static uint32_t GetBitLength(const MediaByteBuffer* aNAL);

 private:
  void FillReservoir();
  const uint8_t* mData;
  const size_t mOriginalBitSize;
  size_t mTotalBitsLeft;
  size_t mSize;         // Size left in bytes
  uint32_t mReservoir;  // Left-aligned bits
  size_t mNumBitsLeft;  // Number of bits left in reservoir.
};

}  // namespace mozilla

#endif  // BIT_READER_H_