Add Buffer::width and Buffer::height (#269)

These return `NonZeroU32`.

Author: madsmtm

CHANGELOG.md

@@ -3,6 +3,7 @@
 - Update to `objc2` 0.6.0.
 - Bump MSRV to Rust 1.71.
 - Make `Context` cloneable.
+- Added `Buffer::width()` and `Buffer::height()` getters.
 
 # 0.4.6
 

README.md

@@ -101,21 +101,18 @@ fn main() {
                     eprintln!("RedrawRequested fired before Resumed or after Suspended");
                     return;
                 };
-                let (width, height) = {
-                    let size = window.inner_size();
-                    (size.width, size.height)
-                };
+                let size = window.inner_size();
                 surface
                     .resize(
-                        NonZeroU32::new(width).unwrap(),
-                        NonZeroU32::new(height).unwrap(),
+                        NonZeroU32::new(size.width).unwrap(),
+                        NonZeroU32::new(size.height).unwrap(),
                     )
                     .unwrap();
 
                 let mut buffer = surface.buffer_mut().unwrap();
-                for index in 0..(width * height) {
-                    let y = index / width;
-                    let x = index % width;
+                for index in 0..(buffer.width().get() * buffer.height().get()) {
+                    let y = index / buffer.width().get();
+                    let x = index % buffer.width().get();
                     let red = x % 255;
                     let green = y % 255;
                     let blue = (x * y) % 255;

examples/animation.rs

@@ -57,23 +57,20 @@ fn main() {
                     return;
                 };
 
-                let size = window.inner_size();
-                if let (Some(width), Some(height)) =
-                    (NonZeroU32::new(size.width), NonZeroU32::new(size.height))
-                {
-                    let elapsed = start.elapsed().as_secs_f64() % 1.0;
+                let elapsed = start.elapsed().as_secs_f64() % 1.0;
 
-                    if (width.get(), height.get()) != *old_size {
-                        *old_size = (width.get(), height.get());
-                        *frames = pre_render_frames(width.get() as usize, height.get() as usize);
-                    };
+                let mut buffer = surface.buffer_mut().unwrap();
 
-                    let frame = &frames[((elapsed * 60.0).round() as usize).clamp(0, 59)];
-
-                    let mut buffer = surface.buffer_mut().unwrap();
-                    buffer.copy_from_slice(frame);
-                    buffer.present().unwrap();
+                let size = (buffer.width().get(), buffer.height().get());
+                if size != *old_size {
+                    *old_size = size;
+                    *frames = pre_render_frames(size.0, size.1);
                 }
+
+                let frame = &frames[((elapsed * 60.0).round() as usize).clamp(0, 59)];
+
+                buffer.copy_from_slice(frame);
+                buffer.present().unwrap();
             }
             WindowEvent::CloseRequested
             | WindowEvent::KeyboardInput {
@@ -97,7 +94,7 @@ fn main() {
     winit_app::run_app(event_loop, app);
 }
 
-fn pre_render_frames(width: usize, height: usize) -> Vec<Vec<u32>> {
+fn pre_render_frames(width: u32, height: u32) -> Vec<Vec<u32>> {
     let render = |frame_id| {
         let elapsed = ((frame_id as f64) / (60.0)) * 2.0 * PI;
 

examples/fruit.rs

@@ -50,14 +50,14 @@ fn main() {
                 };
 
                 let mut buffer = surface.buffer_mut().unwrap();
-                let width = fruit.width() as usize;
+                let width = fruit.width();
                 for (x, y, pixel) in fruit.pixels() {
                     let red = pixel.0[0] as u32;
                     let green = pixel.0[1] as u32;
                     let blue = pixel.0[2] as u32;
 
                     let color = blue | (green << 8) | (red << 16);
-                    buffer[y as usize * width + x as usize] = color;
+                    buffer[(y * width + x) as usize] = color;
                 }
 
                 buffer.present().unwrap();

examples/rectangle.rs

@@ -1,3 +1,5 @@
+use raw_window_handle::{HasDisplayHandle, HasWindowHandle};
+use softbuffer::Buffer;
 use std::num::NonZeroU32;
 use winit::event::{ElementState, KeyEvent, WindowEvent};
 use winit::event_loop::{ControlFlow, EventLoop};
@@ -6,7 +8,9 @@ use winit::keyboard::{Key, NamedKey};
 #[path = "utils/winit_app.rs"]
 mod winit_app;
 
-fn redraw(buffer: &mut [u32], width: usize, height: usize, flag: bool) {
+fn redraw(buffer: &mut Buffer<'_, impl HasDisplayHandle, impl HasWindowHandle>, flag: bool) {
+    let width = buffer.width().get();
+    let height = buffer.height().get();
     for y in 0..height {
         for x in 0..width {
             let value = if flag && x >= 100 && x < width - 100 && y >= 100 && y < height - 100 {
@@ -15,9 +19,9 @@ fn redraw(buffer: &mut [u32], width: usize, height: usize, flag: bool) {
                 let red = (x & 0xff) ^ (y & 0xff);
                 let green = (x & 0x7f) ^ (y & 0x7f);
                 let blue = (x & 0x3f) ^ (y & 0x3f);
-                (blue | (green << 8) | (red << 16)) as u32
+                blue | (green << 8) | (red << 16)
             };
-            buffer[y * width + x] = value;
+            buffer[(y * width + x) as usize] = value;
         }
     }
 }
@@ -66,21 +70,10 @@ fn main() {
                     eprintln!("RedrawRequested fired before Resumed or after Suspended");
                     return;
                 };
-                // Grab the window's client area dimensions, and ensure they're valid
-                let size = window.inner_size();
-                if let (Some(width), Some(height)) =
-                    (NonZeroU32::new(size.width), NonZeroU32::new(size.height))
-                {
-                    // Draw something in the window
-                    let mut buffer = surface.buffer_mut().unwrap();
-                    redraw(
-                        &mut buffer,
-                        width.get() as usize,
-                        height.get() as usize,
-                        *flag,
-                    );
-                    buffer.present().unwrap();
-                }
+                // Draw something in the window
+                let mut buffer = surface.buffer_mut().unwrap();
+                redraw(&mut buffer, *flag);
+                buffer.present().unwrap();
             }
 
             WindowEvent::CloseRequested

examples/winit.rs

@@ -43,23 +43,19 @@ pub(crate) fn entry(event_loop: EventLoop<()>) {
                     eprintln!("RedrawRequested fired before Resumed or after Suspended");
                     return;
                 };
-                let size = window.inner_size();
-                if let (Some(width), Some(height)) =
-                    (NonZeroU32::new(size.width), NonZeroU32::new(size.height))
-                {
-                    let mut buffer = surface.buffer_mut().unwrap();
-                    for y in 0..height.get() {
-                        for x in 0..width.get() {
-                            let red = x % 255;
-                            let green = y % 255;
-                            let blue = (x * y) % 255;
-                            let index = y as usize * width.get() as usize + x as usize;
-                            buffer[index] = blue | (green << 8) | (red << 16);
-                        }
-                    }
 
-                    buffer.present().unwrap();
+                let mut buffer = surface.buffer_mut().unwrap();
+                for y in 0..buffer.height().get() {
+                    for x in 0..buffer.width().get() {
+                        let red = x % 255;
+                        let green = y % 255;
+                        let blue = (x * y) % 255;
+                        let index = y * buffer.width().get() + x;
+                        buffer[index as usize] = blue | (green << 8) | (red << 16);
+                    }
                 }
+
+                buffer.present().unwrap();
             }
             WindowEvent::CloseRequested
             | WindowEvent::KeyboardInput {

examples/winit_multithread.rs

@@ -40,13 +40,13 @@ pub mod ex {
                 surface.resize(width, height).unwrap();
 
                 let mut buffer = surface.buffer_mut().unwrap();
-                for y in 0..height.get() {
-                    for x in 0..width.get() {
+                for y in 0..buffer.height().get() {
+                    for x in 0..buffer.width().get() {
                         let red = x % 255;
                         let green = y % 255;
                         let blue = (x * y) % 255;
-                        let index = y as usize * width.get() as usize + x as usize;
-                        buffer[index] = blue | (green << 8) | (red << 16);
+                        let index = y * buffer.width().get() + x;
+                        buffer[index as usize] = blue | (green << 8) | (red << 16);
                     }
                 }
 

examples/winit_wrong_sized_buffer.rs

@@ -6,9 +6,6 @@ use winit::keyboard::{Key, NamedKey};
 #[path = "utils/winit_app.rs"]
 mod winit_app;
 
-const BUFFER_WIDTH: usize = 256;
-const BUFFER_HEIGHT: usize = 128;
-
 fn main() {
     let event_loop = EventLoop::new().unwrap();
     let context = softbuffer::Context::new(event_loop.owned_display_handle()).unwrap();
@@ -19,10 +16,7 @@ fn main() {
             let mut surface = softbuffer::Surface::new(&context, window.clone()).unwrap();
             // Intentionally set the size of the surface to something different than the size of the window.
             surface
-                .resize(
-                    NonZeroU32::new(BUFFER_WIDTH as u32).unwrap(),
-                    NonZeroU32::new(BUFFER_HEIGHT as u32).unwrap(),
-                )
+                .resize(NonZeroU32::new(256).unwrap(), NonZeroU32::new(128).unwrap())
                 .unwrap();
             surface
         },
@@ -42,14 +36,15 @@ fn main() {
                 };
 
                 let mut buffer = surface.buffer_mut().unwrap();
-                for y in 0..BUFFER_HEIGHT {
-                    for x in 0..BUFFER_WIDTH {
-                        let red = x as u32 % 255;
-                        let green = y as u32 % 255;
-                        let blue = (x as u32 * y as u32) % 255;
+                let width = buffer.width().get();
+                for y in 0..buffer.height().get() {
+                    for x in 0..width {
+                        let red = x % 255;
+                        let green = y % 255;
+                        let blue = (x * y) % 255;
 
                         let color = blue | (green << 8) | (red << 16);
-                        buffer[y * BUFFER_WIDTH + x] = color;
+                        buffer[(y * width + x) as usize] = color;
                     }
                 }
                 buffer.present().unwrap();

src/backend_dispatch.rs

@@ -125,6 +125,26 @@ macro_rules! make_dispatch {
         }
 
         impl<'a, D: HasDisplayHandle, W: HasWindowHandle> BufferInterface for BufferDispatch<'a, D, W> {
+            #[inline]
+            fn width(&self) -> NonZeroU32 {
+                match self {
+                    $(
+                        $(#[$attr])*
+                        Self::$name(inner) => inner.width(),
+                    )*
+                }
+            }
+
+            #[inline]
+            fn height(&self) -> NonZeroU32 {
+                match self {
+                    $(
+                        $(#[$attr])*
+                        Self::$name(inner) => inner.height(),
+                    )*
+                }
+            }
+
             #[inline]
             fn pixels(&self) -> &[u32] {
                 match self {

src/backend_interface.rs

@@ -35,6 +35,8 @@ pub(crate) trait SurfaceInterface<D: HasDisplayHandle + ?Sized, W: HasWindowHand
 }
 
 pub(crate) trait BufferInterface {
+    fn width(&self) -> NonZeroU32;
+    fn height(&self) -> NonZeroU32;
     fn pixels(&self) -> &[u32];
     fn pixels_mut(&mut self) -> &mut [u32];
     fn age(&self) -> u8;