pub struct BitSourceBuilder {
output: Vec<u8>,
nextByte: u32,
bitsLeftInNextByte: u32,
}
impl BitSourceBuilder {
pub fn new() -> Self {
Self {
output: Vec::new(),
nextByte: 0,
bitsLeftInNextByte: 8,
}
}
pub fn write(&mut self, value: u32, numBits: u32) {
if numBits <= self.bitsLeftInNextByte {
self.nextByte <<= numBits;
self.nextByte |= value;
self.bitsLeftInNextByte -= numBits;
if self.bitsLeftInNextByte == 0 {
self.output.push(self.nextByte as u8);
self.nextByte = 0;
self.bitsLeftInNextByte = 8;
}
} else {
let bitsToWriteNow = self.bitsLeftInNextByte;
let numRestOfBits = numBits - bitsToWriteNow;
let mask = 0xFF >> (8 - bitsToWriteNow);
let valueToWriteNow = (value >> numRestOfBits) & mask;
self.write(valueToWriteNow, bitsToWriteNow);
self.write(value, numRestOfBits);
}
}
pub fn toByteArray(&mut self) -> &Vec<u8> {
if self.bitsLeftInNextByte < 8 {
self.write(0, self.bitsLeftInNextByte);
}
&self.output
}
}
impl Default for BitSourceBuilder {
fn default() -> Self {
Self::new()
}
}