fimg - fast image operations


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

//! `Box<cat>`
use std::ops::Range;

use crate::Image;

impl<const CHANNELS: usize> Image<&mut [u8], CHANNELS> {
    /// Draw a bordered box
    /// ```
    /// # use fimg::Image;
    /// let mut b = Image::alloc(10, 9);
    /// b.as_mut().r#box((1, 1), 7, 6, [255]);
    /// # assert_eq!(b.buffer(), b"\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xff\xff\xff\xff\xff\xff\xff\xff\x00\x00\xff\x00\x00\x00\x00\x00\x00\xff\x00\x00\xff\x00\x00\x00\x00\x00\x00\xff\x00\x00\xff\x00\x00\x00\x00\x00\x00\xff\x00\x00\xff\x00\x00\x00\x00\x00\x00\xff\x00\x00\xff\x00\x00\x00\x00\x00\x00\xff\x00\x00\xff\xff\xff\xff\xff\xff\xff\xff\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00")
    /// ```
    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) };
            // SAFETY: clamped to bounds
            unsafe { self.set_pixel(x, x1 + height, c) };
        }
        for y in clamp(y1..height + y1 + 1, 0..self.height()) {
            // SAFETY: clamped to bounds
            unsafe { self.set_pixel(y1, y, c) };
            // SAFETY: clamped to bounds
            unsafe { self.set_pixel(y1 + width, y, c) };
        }
    }

    /// Draw a *filled* box.
    /// ```
    /// # use fimg::Image;
    /// let mut b = Image::alloc(10, 9);
    /// b.as_mut().filled_box((1, 1), 7, 6, [255]);
    /// # assert_eq!(b.buffer(), b"\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xff\xff\xff\xff\xff\xff\xff\xff\x00\x00\xff\xff\xff\xff\xff\xff\xff\xff\x00\x00\xff\xff\xff\xff\xff\xff\xff\xff\x00\x00\xff\xff\xff\xff\xff\xff\xff\xff\x00\x00\xff\xff\xff\xff\xff\xff\xff\xff\x00\x00\xff\xff\xff\xff\xff\xff\xff\xff\x00\x00\xff\xff\xff\xff\xff\xff\xff\xff\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00")
    /// ```
    pub fn filled_box(&mut self, (x1, y1): (u32, u32), width: u32, height: u32, c: [u8; CHANNELS]) {
        for x in clamp(x1..1 + width + x1, 0..self.width()) {
            for y in clamp(y1..1 + height + y1, 0..self.height()) {
                // SAFETY: clamped to bounds
                unsafe { self.set_pixel(x, y, 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)
}