| /// A wrapper for the output slice used when decompressing. |
| /// |
| /// Using this rather than `Cursor` lets us implement the writing methods directly on |
| /// the buffer and lets us use a usize rather than u64 for the position which helps with |
| /// performance on 32-bit systems. |
| pub struct OutputBuffer<'a> { |
| slice: &'a mut [u8], |
| position: usize, |
| } |
| |
| impl<'a> OutputBuffer<'a> { |
| #[inline] |
| pub fn from_slice_and_pos(slice: &'a mut [u8], position: usize) -> OutputBuffer<'a> { |
| OutputBuffer { slice, position } |
| } |
| |
| #[inline] |
| pub fn position(&self) -> usize { |
| self.position |
| } |
| |
| #[inline] |
| pub fn set_position(&mut self, position: usize) { |
| self.position = position; |
| } |
| |
| /// Write a byte to the current position and increment |
| /// |
| /// Assumes that there is space. |
| #[inline] |
| pub fn write_byte(&mut self, byte: u8) { |
| self.slice[self.position] = byte; |
| self.position += 1; |
| } |
| |
| /// Write a slice to the current position and increment |
| /// |
| /// Assumes that there is space. |
| #[inline] |
| pub fn write_slice(&mut self, data: &[u8]) { |
| let len = data.len(); |
| self.slice[self.position..self.position + len].copy_from_slice(data); |
| self.position += data.len(); |
| } |
| |
| #[inline] |
| pub fn bytes_left(&self) -> usize { |
| self.slice.len() - self.position |
| } |
| |
| #[inline] |
| pub fn get_ref(&self) -> &[u8] { |
| self.slice |
| } |
| |
| #[inline] |
| pub fn get_mut(&mut self) -> &mut [u8] { |
| self.slice |
| } |
| } |