fparkan/vendor/gif/benches/decode.rs
Valentin Popov 1b6a04ca55
Initial vendor packages
Signed-off-by: Valentin Popov <valentin@popov.link>
2024-01-08 01:21:28 +04:00

82 lines
2.2 KiB
Rust

use criterion::{black_box, BenchmarkId, BenchmarkGroup, Criterion, Throughput, measurement::Measurement};
use gif::Decoder;
fn read_image(image: &[u8]) -> Option<Vec<u8>> {
let decoder = Decoder::new(black_box(image));
//decoder.set_param(gif::ColorOutput::RGBA);
let mut reader = decoder.unwrap();
while let Some(_) = reader.next_frame_info().unwrap() {
let mut v = vec![0; reader.buffer_size()];
reader.fill_buffer(&mut v).unwrap();
return Some(v);
}
None
}
fn read_metadata(image: &[u8]) {
let decoder = Decoder::new(black_box(image));
decoder.unwrap();
}
fn main() {
struct BenchDef {
data: &'static [u8],
id: &'static str,
sample_size: usize,
}
fn run_bench_def<M: Measurement>(group: &mut BenchmarkGroup<M>, def: BenchDef) {
group
.sample_size(def.sample_size)
.throughput(Throughput::Bytes(def.data.len() as u64))
.bench_with_input(
BenchmarkId::new(def.id, def.data.len()),
def.data,
|b, input| {
b.iter(|| read_image(input))
}
);
}
let mut c = Criterion::default().configure_from_args();
let mut group = c.benchmark_group("gif");
run_bench_def(&mut group, BenchDef {
data: include_bytes!("note.gif"),
id: "note.gif",
sample_size: 100,
});
run_bench_def(&mut group, BenchDef {
data: include_bytes!("photo.gif"),
id: "photo.gif",
sample_size: 20,
});
run_bench_def(&mut group, BenchDef {
data: include_bytes!(concat!(env!("CARGO_MANIFEST_DIR"), "/tests/samples/sample_1.gif")),
id: "sample_1.gif",
sample_size: 100,
});
run_bench_def(&mut group, BenchDef {
data: include_bytes!(concat!(env!("CARGO_MANIFEST_DIR"), "/tests/samples/sample_big.gif")),
id: "sample_big.gif",
sample_size: 20,
});
group
.bench_with_input(
"extract-metadata-note",
include_bytes!("note.gif"),
|b, input| {
b.iter(|| read_metadata(input))
}
);
group.finish();
c.final_summary();
}