danvics/oxipng
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Allow user to specify the output buffer size as well

Browse source
This commit is contained in:
Chris Hennick 2023-06-18 13:41:17 -07:00
parent 986a27f047
commit 548f21d92f
No known key found for this signature in database
GPG key ID: 25653935CC8B6C74
1 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
src/deflate/mod.rs
View file

@ -59,13 +59,15 @@ impl Deflater for Deflaters {
#[derive(Copy, Clone, Debug)]
pub struct BufferedZopfliDeflater {
iterations: NonZeroU8,
buffer_size: usize
input_buffer_size: usize,
output_buffer_size: usize
}
impl BufferedZopfliDeflater {
pub const fn new(iterations: NonZeroU8,
buffer_size: usize) -> Self {
BufferedZopfliDeflater {iterations, buffer_size}
input_buffer_size: usize,
output_buffer_size: usize) -> Self {
BufferedZopfliDeflater {iterations, input_buffer_size, output_buffer_size }
}
}
@ -76,7 +78,7 @@ impl Deflater for BufferedZopfliDeflater {
iteration_count: self.iterations,
..Default::default()
};
let mut buffer = BufWriter::with_capacity(self.buffer_size,
let mut buffer = BufWriter::with_capacity(self.input_buffer_size,
DeflateEncoder::new(
options, Default::default(), Cursor::new(Vec::new())));
let result = (|| -> io::Result<Vec<u8>> {