41 lines
1.2 KiB
Rust
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();
|
|
})
|
|
});
|
|
}
|