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

Add data_is_compressed property

Browse source
This commit is contained in:
Andrew 2025-03-08 14:29:38 +13:00
parent dd023f740b
commit 5a2e4387ce
2 changed files with 24 additions and 26 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
src/evaluate.rs
View file

@ -21,9 +21,9 @@ use crate::{atomicmin::AtomicMin, deflate, filters::RowFilter, png::PngImage, De
pub(crate) struct Candidate {
pub image: Arc<PngImage>,
pub idat_data: Vec<u8>,
pub data: Vec<u8>,
pub data_is_compressed: bool,
pub estimated_output_size: usize,
pub filtered: Vec<u8>,
pub filter: RowFilter,
// For determining tie-breaker
nth: usize,
@ -149,13 +149,12 @@ impl Evaluator {
let idat_data = deflater.deflate(&filtered, best_candidate_size.get());
if let Ok(idat_data) = idat_data {
let estimated_output_size = image.estimated_output_size(&idat_data);
// In the final round, we need the IDAT data but not the filtered data
// Otherwise, we want to keep the filtered data for the next round
// For the final round we need the IDAT data, otherwise the filtered data
let new = Candidate {
image: image.clone(),
idat_data: if final_round { idat_data } else { vec![] },
data: if final_round { idat_data } else { filtered },
data_is_compressed: final_round,
estimated_output_size,
filtered: if final_round { vec![] } else { filtered },
filter,
nth,
};

39
src/lib.rs
View file

@ -170,7 +170,7 @@ impl RawImage {
let mut png = PngData {
raw: result.image,
idat_data: result.idat_data,
idat_data: result.data,
aux_chunks,
frames: Vec::new(),
};
@ -359,7 +359,7 @@ fn optimize_png(
};
if let Some(result) = optimize_raw(raw.clone(), &opts, deadline.clone(), max_size) {
png.raw = result.image;
png.idat_data = result.idat_data;
png.idat_data = result.data;
recompress_frames(png, &opts, deadline, result.filter)?;
postprocess_chunks(&mut png.aux_chunks, &png.raw.ihdr, &raw.ihdr);
}
@ -470,7 +470,7 @@ fn optimize_raw(
(eval_result?, eval_deflater)
};
if !result.idat_data.is_empty()
if result.data_is_compressed
&& max_size.map_or(true, |max_size| result.estimated_output_size < max_size)
{
debug!("Found better result:");
@ -517,27 +517,26 @@ fn perform_trials(
eval_result = Some(result);
}
}
if opts.deflate == eval_deflater {
// No further compression required
return eval_result;
}
// We should have a result here - fail if not (e.g. deadline passed)
let mut result = eval_result?;
// Recompress with the main deflater
debug!("Trying filter {} with {}", result.filter, opts.deflate);
match opts.deflate.deflate(&result.filtered, max_size) {
Ok(idat_data) => {
result.estimated_output_size = result.image.estimated_output_size(&idat_data);
result.idat_data = idat_data;
trace!("{} bytes", result.estimated_output_size);
}
Err(PngError::DeflatedDataTooLong(bytes)) => {
trace!(">{bytes} bytes");
}
Err(_) => (),
};
if !result.data_is_compressed {
// Compress with the main deflater
debug!("Trying filter {} with {}", result.filter, opts.deflate);
match opts.deflate.deflate(&result.data, max_size) {
Ok(idat_data) => {
result.estimated_output_size = result.image.estimated_output_size(&idat_data);
result.data = idat_data;
result.data_is_compressed = true;
trace!("{} bytes", result.estimated_output_size);
}
Err(PngError::DeflatedDataTooLong(bytes)) => {
trace!(">{bytes} bytes");
}
Err(_) => (),
};
}
return Some(result);
}