media/filters/vp9_raw_bits_reader.cc

   1 // Copyright 2015 The Chromium Authors
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "media/filters/vp9_raw_bits_reader.h"
   6
   7 #include <limits.h>
   8
   9 #include <memory>
  10
  11 #include "base/check_op.h"
  12 #include "media/base/bit_reader.h"
  13
  14 namespace media {
  15
  16 Vp9RawBitsReader::Vp9RawBitsReader() : valid_(true) {}
  17
  18 Vp9RawBitsReader::~Vp9RawBitsReader() = default;
  19
  20 void Vp9RawBitsReader::Initialize(const uint8_t* data, size_t size) {
  21   DCHECK(data);
  22   reader_ = std::make_unique<BitReader>(data, size);
  23   valid_ = true;
  24 }
  25
  26 bool Vp9RawBitsReader::ReadBool() {
  27   DCHECK(reader_);
  28   if (!valid_)
  29     return false;
  30
  31   int value = 0;
  32   valid_ = reader_->ReadBits(1, &value);
  33   return valid_ ? value == 1 : false;
  34 }
  35
  36 int Vp9RawBitsReader::ReadLiteral(int bits) {
  37   DCHECK(reader_);
  38   if (!valid_)
  39     return 0;
  40
  41   int value = 0;
  42   DCHECK_LT(static_cast<size_t>(bits), sizeof(value) * 8);
  43   valid_ = reader_->ReadBits(bits, &value);
  44   return valid_ ? value : 0;
  45 }
  46
  47 int Vp9RawBitsReader::ReadSignedLiteral(int bits) {
  48   int value = ReadLiteral(bits);
  49   return ReadBool() ? -value : value;
  50 }
  51
  52 size_t Vp9RawBitsReader::GetBytesRead() const {
  53   DCHECK(reader_);
  54   return (reader_->bits_read() + 7) / 8;
  55 }
  56
  57 bool Vp9RawBitsReader::ConsumeTrailingBits() {
  58   DCHECK(reader_);
  59   int bits_left = GetBytesRead() * 8 - reader_->bits_read();
  60   return ReadLiteral(bits_left) == 0;
  61 }
  62
  63 }  // namespace media