diff options
author | Valentin Popov <valentin@popov.link> | 2024-01-08 00:21:28 +0300 |
---|---|---|
committer | Valentin Popov <valentin@popov.link> | 2024-01-08 00:21:28 +0300 |
commit | 1b6a04ca5504955c571d1c97504fb45ea0befee4 (patch) | |
tree | 7579f518b23313e8a9748a88ab6173d5e030b227 /vendor/gif/benches/rgb_frame.rs | |
parent | 5ecd8cf2cba827454317368b68571df0d13d7842 (diff) | |
download | fparkan-1b6a04ca5504955c571d1c97504fb45ea0befee4.tar.xz fparkan-1b6a04ca5504955c571d1c97504fb45ea0befee4.zip |
Initial vendor packages
Signed-off-by: Valentin Popov <valentin@popov.link>
Diffstat (limited to 'vendor/gif/benches/rgb_frame.rs')
-rw-r--r-- | vendor/gif/benches/rgb_frame.rs | 73 |
1 files changed, 73 insertions, 0 deletions
diff --git a/vendor/gif/benches/rgb_frame.rs b/vendor/gif/benches/rgb_frame.rs new file mode 100644 index 0000000..a0d1934 --- /dev/null +++ b/vendor/gif/benches/rgb_frame.rs @@ -0,0 +1,73 @@ +use std::fs; + +use criterion::{Criterion, Throughput}; +use gif::{Encoder, Frame, Repeat}; +use png; + +const DIR: &str = "benches/samples"; + +fn main() +{ + let mut c = Criterion::default().configure_from_args(); + let mut group = c.benchmark_group("rgb_frame"); + + let dir = fs::read_dir(DIR).expect("Cant'r read dir:\n{}"); + + for path in dir { + let path = path.expect("Can't read path:\n{}").path(); + if path.extension().unwrap() != "png" { + continue; + } + + let mut reader = { + let input = fs::File::open(&path).unwrap(); + let decoder = png::Decoder::new(input); + decoder.read_info().unwrap() + }; + + let mut buf = vec![0; reader.output_buffer_size()]; + let info = reader.next_frame(&mut buf).unwrap(); + + let (w, h, size) = { + // could use try_into().unwrap() but probably no need + (info.width as u16, info.height as u16, info.buffer_size()) + }; + + //size might have to be adjusted for large images + group + .sample_size(50) + .throughput(Throughput::Bytes(size as u64)) + .bench_function(path.file_name().unwrap().to_str().unwrap(), + |b| { + match info.color_type { + png::ColorType::Rgb => b.iter(|| { + Frame::from_rgb_speed(w, h, &mut buf[..size], 30) + }), + png::ColorType::Rgba => b.iter(|| { + Frame::from_rgba_speed(w, h, &mut buf[..size], 30) + }), + c => { + println!("Image has wrong color type: {:?}", c); + } + } + }); + + // actually write the image as a singe frame gif... while MSE can be used + // for quality check, it might not be as good as visual inspection + let mut encoder = { + let output = fs::File::create(path.with_extension("gif")).unwrap(); + Encoder::new(output, w, h, &[]).unwrap() + }; + encoder.set_repeat(Repeat::Finite(0)).unwrap(); + + let frame = match info.color_type { + png::ColorType::Rgb => Frame::from_rgb(w, h, &mut buf[..size]), + png::ColorType::Rgba => Frame::from_rgba(w, h, &mut buf[..size]), + _ => continue, + }; + + encoder.write_frame(&frame).unwrap(); + } + group.finish(); + c.final_summary(); +} |