self.cache |= nw << (32 - self.bits);
}
- fn fill32le16(&mut self, src: &[u8], realbits: u8) {
- let mut nw = (((src[1] as u32) << 24) |
- ((src[0] as u32) << 16) |
- ((src[3] as u32) << 8) |
- ((src[2] as u32) << 0)) as u64;
- if realbits <= 16 { nw >>= 16; }
- self.cache |= nw << self.bits;
+ fn fill32le16(&mut self, src: &[u8]) {
+ let nw = (((src[1] as u32) << 24) |
+ ((src[0] as u32) << 16) |
+ ((src[3] as u32) << 8) |
+ ((src[2] as u32) << 0)) as u64;
+ self.cache |= nw << (32 - self.bits);
}
- fn fill32le32(&mut self, src: &[u8]) {
+ fn fill32le32(&mut self, src: &[u8], lsb: bool) {
let nw = (((src[3] as u32) << 24) |
((src[2] as u32) << 16) |
((src[1] as u32) << 8) |
((src[0] as u32) << 0)) as u64;
- self.cache |= nw << self.bits;
+ if lsb {
+ self.cache |= nw << self.bits;
+ } else {
+ self.cache |= nw << (32 - self.bits);
+ }
}
fn refill(&mut self) -> BitReaderResult<()> {
let buf = &self.src[self.pos..];
match self.mode {
BitReaderMode::BE => self.fill32be (buf),
- BitReaderMode::LE16MSB => self.fill32le16(buf, 32),
- BitReaderMode::LE => self.fill32le32(buf),
- BitReaderMode::LE32MSB => self.fill32le32(buf),
+ BitReaderMode::LE16MSB => self.fill32le16(buf),
+ BitReaderMode::LE => self.fill32le32(buf, true),
+ BitReaderMode::LE32MSB => self.fill32le32(buf, false),
}
self.pos += 4;
self.bits += 32;
if newbits == 0 { break; }
match self.mode {
BitReaderMode::BE => self.fill32be (&buf),
- BitReaderMode::LE16MSB => self.fill32le16(&buf, newbits),
- BitReaderMode::LE => self.fill32le32(&buf),
- BitReaderMode::LE32MSB => self.fill32le32(&buf),
+ BitReaderMode::LE16MSB => self.fill32le16(&buf),
+ BitReaderMode::LE => self.fill32le32(&buf, true),
+ BitReaderMode::LE32MSB => self.fill32le32(&buf, false),
}
self.bits += newbits;
}
Ok(res)
}
+ pub fn read_bool(&mut self) -> BitReaderResult<bool> {
+ if self.bits < 1 {
+ if let Err(err) = self.refill() { return Err(err) }
+ if self.bits < 1 { return Err(BitstreamEnd) }
+ }
+ let res = self.read_cache(1);
+ self.skip_cache(1);
+ Ok(res == 1)
+ }
+
pub fn peek(&mut self, nbits: u8) -> u32 {
if nbits > 32 { return 0 }
if self.bits < nbits { let _ = self.refill(); }