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

added byte boundry case to push_bits

Browse source
This commit is contained in:
Nicolas FOURNIER 2020-03-20 08:44:25 +01:00
commit 57dcdbe9a0
1 changed files with 7 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

10
src/writestream.rs
View file

@ -98,15 +98,19 @@ where
debug_assert!(count < USIZE_BITS - 8); debug_assert!(count < USIZE_BITS - 8);
let bit_offset = self.bit_len & 7; let bit_offset = self.bit_len & 7;
let last_written_byte = self.bytes.pop().unwrap_or(0); let merge_byte = if bit_offset == 0 {
0
} else {
self.bytes.pop().unwrap_or(0)
};
let merged_byte_count = (count + bit_offset + 7) / 8; let merged_byte_count = (count + bit_offset + 7) / 8;
if E::is_le() { if E::is_le() {
let merged = last_written_byte as usize | bits << bit_offset; let merged = merge_byte as usize | bits << bit_offset;
self.bytes self.bytes
.extend_from_slice(&merged.to_le_bytes()[0..merged_byte_count]); .extend_from_slice(&merged.to_le_bytes()[0..merged_byte_count]);
} else { } else {
let merged = ((last_written_byte as usize) << (USIZE_BITS - 8)) let merged = ((merge_byte as usize) << (USIZE_BITS - 8))
| bits << (USIZE_BITS - bit_offset - count); | bits << (USIZE_BITS - bit_offset - count);
self.bytes self.bytes
.extend_from_slice(&merged.to_be_bytes()[0..merged_byte_count]); .extend_from_slice(&merged.to_be_bytes()[0..merged_byte_count]);