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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106

//! text raster

use crate::{
    pixels::{float, Wam},
    Image, Pack,
};
use fontdue::{layout::TextStyle, Font};
use umath::{generic_float::Constructors, FF32};

/// note: `N` may != channels
pub trait Text<const N: usize> {
    fn text(&mut self, x: u32, y: u32, size: f32, font: &Font, text: &str, color: [u8; N]);
}

impl<T: AsMut<[u32]> + AsRef<[u32]>> Text<4> for Image<T, 1> {
    fn text(&mut self, x: u32, y: u32, size: f32, font: &Font, text: &str, color: [u8; 4]) {
        let mut lay =
            fontdue::layout::Layout::new(fontdue::layout::CoordinateSystem::PositiveYDown);
        lay.append(&[font], &TextStyle::new(text, size, 0));
        for glpyh in lay.glyphs() {
            let (metrics, bitmap) = font.rasterize(glpyh.parent, size);
            for i in 0..metrics.width {
                for j in 0..metrics.height {
                    let x = x + i as u32 + glpyh.x as u32;
                    if x >= self.width() {
                        continue;
                    }
                    let y = y + j as u32 + glpyh.y as u32;
                    if y >= self.height() {
                        continue;
                    }

                    // SAFETY: the rasterizer kinda promises that metrics width and height are in bounds
                    let fill = unsafe { float(*bitmap.get_unchecked(j * metrics.width + i)) };
                    // SAFETY: we clampin
                    let bg: [u8; 4] = unsafe { Pack::unpack(self.pixel(x, y)[0]) };
                    // SAFETY: see above
                    *(&mut unsafe { self.pixel_mut(x, y) }[0]) =
                        // SAFETY: fill is 0..=1
                        Pack::pack(unsafe { &bg.wam(color, FF32::one() - fill, fill) });
                }
            }
        }
    }
}

impl<const N: usize, T: AsMut<[u8]> + AsRef<[u8]>> Text<N> for Image<T, N> {
    fn text(&mut self, x: u32, y: u32, size: f32, font: &Font, text: &str, color: [u8; N]) {
        let mut lay =
            fontdue::layout::Layout::new(fontdue::layout::CoordinateSystem::PositiveYDown);
        lay.append(&[font], &TextStyle::new(text, size, 0));
        for glpyh in lay.glyphs() {
            let (metrics, bitmap) = font.rasterize(glpyh.parent, size);
            for i in 0..metrics.width {
                for j in 0..metrics.height {
                    let x = x + i as u32 + glpyh.x as u32;
                    if x >= self.width() {
                        continue;
                    }
                    let y = y + j as u32 + glpyh.y as u32;
                    if y >= self.height() {
                        continue;
                    }

                    // SAFETY: the rasterizer kinda promises that metrics width and height are in bounds
                    let fill = unsafe { float(*bitmap.get_unchecked(j * metrics.width + i)) };
                    // SAFETY: we clampin
                    let bg = unsafe { &mut *(self.pixel_mut(x, y).as_mut_ptr() as *mut [u8; N]) };
                    // SAFETY: fill is 0..=1
                    *bg = unsafe { bg.wam(color, FF32::one() - fill, fill) };
                }
            }
        }
    }
}

impl<const N: usize, T> Image<T, N> {
    /// Draw text.
    ///
    /// ```
    /// # use fimg::Image;
    /// let font = fontdue::Font::from_bytes(
    ///     &include_bytes!("../../data/CascadiaCode.ttf")[..],
    ///     fontdue::FontSettings {
    ///         scale: 200.0,
    ///         ..Default::default()
    ///     },
    /// ).unwrap();
    /// let mut i: Image<_, 4> = Image::alloc(750, 250).boxed();
    /// i.text(50, 10, 200.0, &font, "hello", [0, 0, 0, 255]);
    /// # assert_eq!(&**i.buffer(), include_bytes!("../../tdata/text.imgbuf"));
    /// ```
    pub fn text<const P: usize>(
        &mut self,
        x: u32,
        y: u32,
        size: f32,
        font: &Font,
        text: &str,
        color: [u8; P],
    ) where
        Image<T, N>: Text<P>,
    {
        Text::text(self, x, y, size, font, text, color)
    }
}