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

Don't truncate bKGD on output

Browse source
This commit is contained in:
Andrew 2022-11-28 18:25:29 +13:00
parent c84186916e
commit 181c6dd96f
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/png/mod.rs
View file

@ -198,7 +198,11 @@ impl PngData {
// Palette // Palette
if let Some(ref palette) = self.raw.palette { if let Some(ref palette) = self.raw.palette {
let mut palette_data = Vec::with_capacity(palette.len() * 3); let mut palette_data = Vec::with_capacity(palette.len() * 3);
let max_palette_size = 1 << (self.raw.ihdr.bit_depth.as_u8() as usize); let mut max_palette_size = 1 << (self.raw.ihdr.bit_depth.as_u8() as usize);
// Ensure bKGD color doesn't get truncated from palette
if let Some(&idx) = self.raw.aux_headers.get(b"bKGD").and_then(|b| b.first()) {
max_palette_size = max_palette_size.max(idx as usize + 1);
}
for px in palette.iter().take(max_palette_size) { for px in palette.iter().take(max_palette_size) {
palette_data.extend_from_slice(px.rgb().as_slice()); palette_data.extend_from_slice(px.rgb().as_slice());
} }