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

Remove special case for empty last line

Browse source
This commit is contained in:
Kornel 2021-02-04 13:48:28 +00:00
parent c318ac78e1
commit 7ebdb8bb1a
2 changed files with 17 additions and 39 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

55
src/filters.rs
View file

@ -65,6 +65,7 @@ pub fn filter_line(filter: u8, bpp: usize, data: &[u8], last_line: &[u8], buf: &
pub fn unfilter_line(filter: u8, bpp: usize, data: &[u8], last_line: &[u8], buf: &mut Vec<u8>) {
buf.clear();
buf.reserve(data.len());
assert_eq!(data.len(), last_line.len());
match filter {
0 => {
buf.extend_from_slice(data);
@ -81,55 +82,31 @@ pub fn unfilter_line(filter: u8, bpp: usize, data: &[u8], last_line: &[u8], buf:
}
}
2 => {
if last_line.is_empty() {
buf.extend_from_slice(data);
} else {
buf.extend(
data.iter()
.zip(last_line.iter())
.map(|(cur, last)| cur.wrapping_add(*last)),
);
};
buf.extend(
data.iter()
.zip(last_line.iter())
.map(|(cur, last)| cur.wrapping_add(*last)),
);
}
3 => {
for (i, byte) in data.iter().enumerate() {
buf.push(if last_line.is_empty() {
match i.checked_sub(bpp) {
Some(x) => {
let b = buf[x];
byte.wrapping_add(b >> 1)
}
None => *byte,
}
} else {
match i.checked_sub(bpp) {
Some(x) => {
let b = buf[x];
byte.wrapping_add(((u16::from(b) + u16::from(last_line[i])) >> 1) as u8)
}
None => byte.wrapping_add(last_line[i] >> 1),
buf.push(match i.checked_sub(bpp) {
Some(x) => {
let b = buf[x];
byte.wrapping_add(((u16::from(b) + u16::from(last_line[i])) >> 1) as u8)
}
None => byte.wrapping_add(last_line[i] >> 1),
});
}
}
4 => {
for (i, byte) in data.iter().enumerate() {
buf.push(if last_line.is_empty() {
match i.checked_sub(bpp) {
Some(x) => {
let b = buf[x];
byte.wrapping_add(b)
}
None => *byte,
}
} else {
match i.checked_sub(bpp) {
Some(x) => {
let b = buf[x];
byte.wrapping_add(paeth_predictor(b, last_line[i], last_line[x]))
}
None => byte.wrapping_add(last_line[i]),
buf.push(match i.checked_sub(bpp) {
Some(x) => {
let b = buf[x];
byte.wrapping_add(paeth_predictor(b, last_line[i], last_line[x]))
}
None => byte.wrapping_add(last_line[i]),
});
}
}

1
src/png/mod.rs
View file

@ -294,6 +294,7 @@ impl PngImage {
last_pass = pass;
}
}
last_line.resize(line.data.len(), 0);
unfilter_line(
line.filter,
bpp,