43 lines
1.0 KiB
C++
43 lines
1.0 KiB
C++
// Copyright 2017 Google Inc. All rights reserved.
|
|
//
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file or at
|
|
// https://developers.google.com/open-source/licenses/bsd
|
|
|
|
#include "packager/media/base/bit_writer.h"
|
|
|
|
namespace shaka {
|
|
namespace media {
|
|
|
|
BitWriter::BitWriter(std::vector<uint8_t>* storage)
|
|
: storage_(storage), initial_storage_size_(storage_->size()) {}
|
|
|
|
void BitWriter::WriteBits(uint32_t bits, size_t number_of_bits) {
|
|
DCHECK_NE(number_of_bits, 0u);
|
|
DCHECK_LE(number_of_bits, 32u);
|
|
DCHECK_LT(bits, 1ULL << number_of_bits);
|
|
|
|
num_bits_ += number_of_bits;
|
|
DCHECK_LE(num_bits_, 64);
|
|
bits_ |= static_cast<uint64_t>(bits) << (64 - num_bits_);
|
|
|
|
while (num_bits_ >= 8) {
|
|
storage_->push_back(bits_ >> 56);
|
|
bits_ <<= 8;
|
|
num_bits_ -= 8;
|
|
}
|
|
}
|
|
|
|
void BitWriter::Flush() {
|
|
while (num_bits_ > 0) {
|
|
storage_->push_back(bits_ >> 56);
|
|
bits_ <<= 8;
|
|
num_bits_ -= 8;
|
|
}
|
|
bits_ = 0;
|
|
num_bits_ = 0;
|
|
}
|
|
|
|
} // namespace media
|
|
} // namespace shaka
|