qoi/benches/bench.rs

59 lines
1.9 KiB
Rust
Raw Normal View History

//! To run benchmarks, also pass RUSTFLAGS="--cfg bench" until cargo does this automatically.
use std::fs::File;
use criterion::{black_box, criterion_group, criterion_main, Criterion};
use qoi_fast::*;
pub fn criterion_benchmark(c: &mut Criterion) {
// let three_raw = include_bytes!("../assets/three.raw");
// let four_raw = include_bytes!("../assets/four.raw");
// let three_qoi = qoi_encode::<3>(three_raw, 572, 354).unwrap();
// c.bench_function("encode 3", |b| {
// b.iter(|| black_box(qoi_encode::<3>(three_raw, 572, 354).unwrap()))
// });
// c.bench_function("decode 3", |b| {
// b.iter(|| black_box(qoi_decode::<3>(&three_qoi).unwrap()))
// });
// c.bench_function("encode 4", |b| {
// b.iter(|| black_box(qoi_encode::<4>(four_raw, 572, 354).unwrap()))
// });
let decoder = png::Decoder::new(
File::open("/Users/ivansmirnov/projects/rust/qoi-other/images/kodak/kodim11.png").unwrap(),
);
let mut reader = decoder.read_info().unwrap();
let mut buf = vec![0; reader.output_buffer_size()];
let info = reader.next_frame(&mut buf).unwrap();
assert_eq!(info.color_type.samples(), 3);
let png_bytes = &buf[..info.buffer_size()];
let qoi_bytes = qoi_encode_to_vec(
png_bytes,
info.width as _,
info.height as _,
info.color_type.samples() as _,
)
.unwrap();
c.bench_function("kodim11.png (encode-3)", |b| {
b.iter(|| {
black_box(qoi_encode_to_vec(
png_bytes,
info.width as _,
info.height as _,
info.color_type.samples() as _,
))
.unwrap()
})
});
c.bench_function("kodim11.png (decode-3)", |b| {
b.iter(|| black_box(qoi_decode_to_vec(&qoi_bytes, 3)).unwrap())
});
}
criterion_group!(benches, criterion_benchmark);
criterion_main!(benches);