softbuffer/benches/buffer_mut.rs

use criterion::{criterion_group, criterion_main, Criterion};

fn buffer_mut(c: &mut Criterion) {
    #[cfg(any(target_arch = "wasm32", target_arch = "wasm64"))]
    {
        // Do nothing.
        let _ = c;
    }

    #[cfg(not(any(target_arch = "wasm32", target_arch = "wasm64")))]
    {
        use criterion::black_box;
        use softbuffer::{Context, Surface};
        use std::num::NonZeroU32;
        use winit::event_loop::ControlFlow;
        use winit::platform::run_on_demand::EventLoopExtRunOnDemand;

        let mut evl = winit::event_loop::EventLoop::new().unwrap();
        let window = winit::window::WindowBuilder::new()
            .with_visible(false)
            .build(&evl)
            .unwrap();

        evl.run_on_demand(move |ev, elwt| {
            elwt.set_control_flow(ControlFlow::Poll);

            if let winit::event::Event::AboutToWait = ev {
                elwt.exit();

                let mut surface = {
                    let context = Context::new(elwt).unwrap();
                    Surface::new(&context, &window).unwrap()
                };

                let size = window.inner_size();
                surface
                    .resize(
                        NonZeroU32::new(size.width).unwrap(),
                        NonZeroU32::new(size.height).unwrap(),
                    )
                    .unwrap();

                c.bench_function("buffer_mut()", |b| {
                    b.iter(|| {
                        for _ in 0..500 {
                            black_box(surface.buffer_mut().unwrap());
                        }
                    });
                });

                c.bench_function("pixels_mut()", |b| {
                    let mut buffer = surface.buffer_mut().unwrap();
                    b.iter(|| {
                        for _ in 0..500 {
                            let x: &mut [u32] = &mut buffer;
                            black_box(x);
                        }
                    });
                });
            }
        })
        .unwrap();
    }
}

criterion_group!(benches, buffer_mut);
criterion_main!(benches);