robin/bitbuffer
1
0
Fork 0
mirror of https://codeberg.org/icewind/bitbuffer.git synced 2026-06-03 08:34:07 +02:00
Code Issues Releases Wiki Activity

fix writing small bit streams

Browse source
This commit is contained in:
Robin Appelman 2021-07-19 17:57:36 +02:00
commit 6042fd5426
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/writestream.rs
View file

@ -1,4 +1,5 @@
use num_traits::{Float, PrimInt}; use num_traits::{Float, PrimInt};
use std::cmp::min;
use std::mem::size_of; use std::mem::size_of;
use std::ops::{BitOrAssign, BitXor}; use std::ops::{BitOrAssign, BitXor};
@ -215,8 +216,9 @@ where
let mut bits = bits.clone(); let mut bits = bits.clone();
let bit_offset = self.bit_len() % 8; let bit_offset = self.bit_len() % 8;
if bit_offset > 0 { if bit_offset > 0 {
let start = bits.read_int::<u8>(8 - bit_offset)?; let bit_count = min(8 - bit_offset, bits.bits_left());
self.push_bits(start as usize, 8 - bit_offset); let start = bits.read_int::<u8>(bit_count)?;
self.push_bits(start as usize, bit_count);
} }
while bits.bits_left() > 32 { while bits.bits_left() > 32 {