From e09c667231e2f588b5bfd0bfa4ae4d76478df62c Mon Sep 17 00:00:00 2001
From: bendn <bend.n@outlook.com>
Date: Wed, 27 Sep 2023 08:19:28 +0700
Subject: [PATCH] add stroked box drawing

---
 src/drawing/box.rs | 70 ++++++++++++++++++++++++++++++++++++++++++----
 1 file changed, 65 insertions(+), 5 deletions(-)

diff --git a/src/drawing/box.rs b/src/drawing/box.rs
index ff85f0f..8cb3450 100644
--- a/src/drawing/box.rs
+++ b/src/drawing/box.rs
@@ -14,17 +14,20 @@ impl<T: DerefMut<Target = [u8]>, const CHANNELS: usize> Image<T, CHANNELS> {
     pub fn r#box(&mut self, (x1, y1): (u32, u32), width: u32, height: u32, c: [u8; CHANNELS]) {
         // skip sides, leave that to second loop
         for x in clamp(x1 + 1..width + x1, 0..self.width()) {
-            // top line
             // SAFETY: clamped to bounds
-            unsafe { self.set_pixel(x, x1, c) };
+            unsafe { self.set_pixel(x, y1, c) };
+        }
+        for x in clamp(x1 + 1..width + x1, 0..self.width()) {
             // SAFETY: clamped to bounds
-            unsafe { self.set_pixel(x, x1 + height, c) };
+            unsafe { self.set_pixel(x, (y1 + height).min(self.height() - 1), c) };
         }
         for y in clamp(y1..height + y1 + 1, 0..self.height()) {
             // SAFETY: clamped to bounds
-            unsafe { self.set_pixel(y1, y, c) };
+            unsafe { self.set_pixel(x1, y, c) };
+        }
+        for y in clamp(y1..height + y1 + 1, 0..self.height()) {
             // SAFETY: clamped to bounds
-            unsafe { self.set_pixel(y1 + width, y, c) };
+            unsafe { self.set_pixel((x1 + width).min(self.width() - 1), y, c) };
         }
     }
 
@@ -43,9 +46,66 @@ impl<T: DerefMut<Target = [u8]>, const CHANNELS: usize> Image<T, CHANNELS> {
             }
         }
     }
+
+    /// Draw a stroked box
+    /// ```
+    /// # use fimg::Image;
+    /// let mut b = Image::alloc(11, 11);
+    /// b.as_mut().stroked_box((2, 2), 6, 6, 2, [255]);
+    /// # assert_eq!(b.buffer(), b"\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xff\xff\xff\xff\xff\xff\xff\xff\xff\x00\x00\xff\xff\xff\xff\xff\xff\xff\xff\xff\x00\x00\xff\xff\x00\x00\x00\x00\x00\xff\xff\x00\x00\xff\xff\x00\x00\x00\x00\x00\xff\xff\x00\x00\xff\xff\x00\x00\x00\x00\x00\xff\xff\x00\x00\xff\xff\x00\x00\x00\x00\x00\xff\xff\x00\x00\xff\xff\x00\x00\x00\x00\x00\xff\xff\x00\x00\xff\xff\xff\xff\xff\xff\xff\xff\xff\x00\x00\xff\xff\xff\xff\xff\xff\xff\xff\xff\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00");
+    /// ```
+    pub fn stroked_box(
+        &mut self,
+        (x1, y1): (u32, u32),
+        width: u32,
+        height: u32,
+        stroke: u32,
+        c: [u8; CHANNELS],
+    ) {
+        let n = stroke / 2;
+        for n in 0..=n {
+            // TODO this is slightly stupid
+            // move it up and left, expand w, h
+            self.r#box((x1 - n, y1 - n), width + n + n, height + n + n, c)
+        }
+    }
 }
 
 /// clamp a range with another range
 fn clamp(r: Range<u32>, within: Range<u32>) -> Range<u32> {
     r.start.clamp(within.start, within.end)..r.end.clamp(within.start, within.end)
 }
+
+#[cfg(test)]
+mod tests {
+    use crate::Image;
+    #[test]
+    fn box_oob() {
+        let mut i = Image::alloc(5, 5);
+        i.r#box((7, 7), 5, 5, [255]);
+        assert_eq!(i.buffer(), &[0u8; 5 * 5]);
+    }
+
+    #[test]
+    fn partial_oob() {
+        let mut i = Image::alloc(5, 5);
+        i.r#box((2, 2), 2, 17, [255]);
+        assert_eq!(i.buffer(),b"\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xff\xff\xff\x00\x00\xff\x00\xff\x00\x00\xff\xff\xff");
+    }
+
+    #[test]
+    fn thick_box_oob() {
+        let mut i = Image::alloc(5, 5);
+        i.stroked_box((7, 7), 5, 5, 2, [255]);
+        assert_eq!(i.buffer(), &[0u8; 5 * 5]);
+    }
+
+    #[test]
+    fn thick_box_partial_oob() {
+        let mut i = Image::alloc(15, 15);
+        i.stroked_box((2, 2), 4, 17, 2, [255]);
+        // ideally the bottom would have a 2 stroke line, alas tis difficult.
+        assert_eq!(i.buffer(), b"\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xff\xff\xff\xff\xff\xff\xff\x00\x00\x00\x00\x00\x00\x00\x00\xff\xff\xff\xff\xff\xff\xff\x00\x00\x00\x00\x00\x00\x00\x00\xff\xff\x00\x00\x00\xff\xff\x00\x00\x00\x00\x00\x00\x00\x00\xff\xff\x00\x00\x00\xff\xff\x00\x00\x00\x00\x00\x00\x00\x00\xff\xff\x00\x00\x00\xff\xff\x00\x00\x00\x00\x00\x00\x00\x00\xff\xff\x00\x00\x00\xff\xff\x00\x00\x00\x00\x00\x00\x00\x00\xff\xff\x00\x00\x00\xff\xff\x00\x00\x00\x00\x00\x00\x00\x00\xff\xff\x00\x00\x00\xff\xff\x00\x00\x00\x00\x00\x00\x00\x00\xff\xff\x00\x00\x00\xff\xff\x00\x00\x00\x00\x00\x00\x00\x00\xff\xff\x00\x00\x00\xff\xff\x00\x00\x00\x00\x00\x00\x00\x00\xff\xff\x00\x00\x00\xff\xff\x00\x00\x00\x00\x00\x00\x00\x00\xff\xff\x00\x00\x00\xff\xff\x00\x00\x00\x00\x00\x00\x00\x00\xff\xff\x00\x00\x00\xff\xff\x00\x00\x00\x00\x00\x00\x00\x00\xff\xff\xff\xff\xff\xff\xff\x00\x00\x00\x00\x00\x00\x00");
+        panic!();
+    }
+}