OLD | NEW |
(Empty) | |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. |
| 4 |
| 5 #ifndef MEDIA_BASE_BIT_READER_H_ |
| 6 #define MEDIA_BASE_BIT_READER_H_ |
| 7 |
| 8 #include <sys/types.h> |
| 9 |
| 10 #include "base/basictypes.h" |
| 11 #include "base/logging.h" |
| 12 #include "media/base/media_export.h" |
| 13 |
| 14 namespace media { |
| 15 |
| 16 // A class to read bit streams. |
| 17 class MEDIA_EXPORT BitReader { |
| 18 public: |
| 19 // Initialize the reader to start reading at |data|, |size| being size |
| 20 // of |data| in bytes. |
| 21 BitReader(const uint8* data, off_t size); |
| 22 ~BitReader(); |
| 23 |
| 24 // Read |num_bits| next bits from stream and return in |*out|, first bit |
| 25 // from the stream starting at |num_bits| position in |*out|. |
| 26 // |num_bits| cannot be larger than the bits the type can hold. |
| 27 // Return false if the given number of bits cannot be read (not enough |
| 28 // bits in the stream), true otherwise. When return false, the stream will |
| 29 // enter a state where further ReadBits/SkipBits operations will always |
| 30 // return false unless |num_bits| is 0. The type |T| has to be a primitive |
| 31 // integer type. |
| 32 template<typename T> bool ReadBits(int num_bits, T *out) { |
| 33 DCHECK_LE(num_bits, static_cast<int>(sizeof(T) * 8)); |
| 34 uint64 temp; |
| 35 bool ret = ReadBitsInternal(num_bits, &temp); |
| 36 *out = static_cast<T>(temp); |
| 37 return ret; |
| 38 } |
| 39 |
| 40 private: |
| 41 // Help function used by ReadBits to avoid inlining the bit reading logic. |
| 42 bool ReadBitsInternal(int num_bits, uint64* out); |
| 43 |
| 44 // Advance to the next byte, loading it into curr_byte_. |
| 45 // If the num_remaining_bits_in_curr_byte_ is 0 after this function returns, |
| 46 // the stream has reached the end. |
| 47 void UpdateCurrByte(); |
| 48 |
| 49 // Pointer to the next unread (not in curr_byte_) byte in the stream. |
| 50 const uint8* data_; |
| 51 |
| 52 // Bytes left in the stream (without the curr_byte_). |
| 53 off_t bytes_left_; |
| 54 |
| 55 // Contents of the current byte; first unread bit starting at position |
| 56 // 8 - num_remaining_bits_in_curr_byte_ from MSB. |
| 57 uint8 curr_byte_; |
| 58 |
| 59 // Number of bits remaining in curr_byte_ |
| 60 int num_remaining_bits_in_curr_byte_; |
| 61 |
| 62 private: |
| 63 DISALLOW_COPY_AND_ASSIGN(BitReader); |
| 64 }; |
| 65 |
| 66 } // namespace media |
| 67 |
| 68 #endif // MEDIA_BASE_BIT_READER_H_ |
OLD | NEW |