use std::{fs::read_dir, time::Duration};
use aarty::*;
use criterion::{criterion_group, criterion_main, BenchmarkId, Criterion};
fn to_text_bench(c: &mut Criterion) {
let mut images = Vec::new();
for entry in read_dir("samples").unwrap() {
let path = entry.unwrap().path();
images.push((path.clone(), image::open(path).unwrap()));
}
macro_rules! group {
($name: expr, $config: expr) => {{
let mut group = c.benchmark_group($name);
for (path, image) in &images {
group.bench_with_input(
BenchmarkId::from_parameter(path.to_str().unwrap()),
image,
|b, image| {
b.iter(|| {
let _ = image.to_text($config);
});
},
);
}
group.finish();
}};
}
group!(
"convert with no colors",
Config::new(vec![' ', '.', ',', '-', '~', '!', '*', '%', '$', '@', '#'].into())
);
group!(
"convert with foeground colors",
Config::new(vec![' ', '.', ',', '-', '~', '!', '*', '%', '$', '@', '#'].into())
.with_flags(COLORS)
);
group!(
"convert with foeground colors reversed",
Config::new(vec![' ', '.', ',', '-', '~', '!', '*', '%', '$', '@', '#'].into())
.with_flags(COLORS | REVERSE)
);
group!(
"convert with foeground colors reversed and empty set",
Config::new(Sympols::empty()).with_flags(COLORS | REVERSE)
);
}
criterion_group! {
name = benches;
config = Criterion::default().sample_size(10).with_plots().warm_up_time(Duration::from_millis(200));
targets = to_text_bench
}
criterion_main!(benches);