diff -r 000000000000 -r 6474c204b198 content/media/fmp4/demuxer/bit_reader.h --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/content/media/fmp4/demuxer/bit_reader.h Wed Dec 31 06:09:35 2014 +0100 @@ -0,0 +1,69 @@ +// Copyright (c) 2012 The Chromium Authors. All rights reserved. +// Use of this source code is governed by a BSD-style license that can be +// found in the LICENSE file. + +#ifndef MEDIA_BASE_BIT_READER_H_ +#define MEDIA_BASE_BIT_READER_H_ + +#include + +#include "mp4_demuxer/basictypes.h" + +namespace mp4_demuxer { + +// A class to read bit streams. +class BitReader { + public: + // Initialize the reader to start reading at |data|, |size| being size + // of |data| in bytes. + BitReader(const uint8_t* data, off_t size); + ~BitReader(); + + // Read |num_bits| next bits from stream and return in |*out|, first bit + // from the stream starting at |num_bits| position in |*out|. + // |num_bits| cannot be larger than the bits the type can hold. + // Return false if the given number of bits cannot be read (not enough + // bits in the stream), true otherwise. When return false, the stream will + // enter a state where further ReadBits/SkipBits operations will always + // return false unless |num_bits| is 0. The type |T| has to be a primitive + // integer type. + template bool ReadBits(int num_bits, T *out) { + DCHECK_LE(num_bits, static_cast(sizeof(T) * 8)); + uint64_t temp; + bool ret = ReadBitsInternal(num_bits, &temp); + *out = static_cast(temp); + return ret; + } + + // Returns the number of bits available for reading. + int bits_available() const; + + private: + // Help function used by ReadBits to avoid inlining the bit reading logic. + bool ReadBitsInternal(int num_bits, uint64_t* out); + + // Advance to the next byte, loading it into curr_byte_. + // If the num_remaining_bits_in_curr_byte_ is 0 after this function returns, + // the stream has reached the end. + void UpdateCurrByte(); + + // Pointer to the next unread (not in curr_byte_) byte in the stream. + const uint8_t* data_; + + // Bytes left in the stream (without the curr_byte_). + off_t bytes_left_; + + // Contents of the current byte; first unread bit starting at position + // 8 - num_remaining_bits_in_curr_byte_ from MSB. + uint8_t curr_byte_; + + // Number of bits remaining in curr_byte_ + int num_remaining_bits_in_curr_byte_; + + private: + DISALLOW_COPY_AND_ASSIGN(BitReader); +}; + +} // namespace mp4_demuxer + +#endif // MEDIA_BASE_BIT_READER_H_