aboutsummaryrefslogtreecommitdiff
path: root/vendor/gif/benches/rgb_frame.rs
diff options
context:
space:
mode:
authorValentin Popov <valentin@popov.link>2024-01-08 00:21:28 +0300
committerValentin Popov <valentin@popov.link>2024-01-08 00:21:28 +0300
commit1b6a04ca5504955c571d1c97504fb45ea0befee4 (patch)
tree7579f518b23313e8a9748a88ab6173d5e030b227 /vendor/gif/benches/rgb_frame.rs
parent5ecd8cf2cba827454317368b68571df0d13d7842 (diff)
downloadfparkan-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.rs73
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();
+}