denyhosts/clamav/libclamav_rust/.cargo/vendor/png/benches/decoder.rs

41 lines
1.2 KiB
Rust

use std::fs;
use criterion::{criterion_group, criterion_main, Criterion, Throughput};
use png::Decoder;
fn load_all(c: &mut Criterion) {
for file in fs::read_dir("tests/benches/").unwrap() {
if let Ok(entry) = file {
match entry.path().extension() {
Some(st) if st == "png" => {}
_ => continue,
}
let data = fs::read(entry.path()).unwrap();
bench_file(c, data, entry.file_name().into_string().unwrap());
}
}
}
criterion_group!(benches, load_all);
criterion_main!(benches);
fn bench_file(c: &mut Criterion, data: Vec<u8>, name: String) {
let mut group = c.benchmark_group("decode");
group.sample_size(20);
let decoder = Decoder::new(&*data);
let mut reader = decoder.read_info().unwrap();
let mut image = vec![0; reader.output_buffer_size()];
let info = reader.next_frame(&mut image).unwrap();
group.throughput(Throughput::Bytes(info.buffer_size() as u64));
group.bench_with_input(name, &data, |b, data| {
b.iter(|| {
let decoder = Decoder::new(data.as_slice());
let mut decoder = decoder.read_info().unwrap();
decoder.next_frame(&mut image).unwrap();
})
});
}