devotee 0.2.0-beta.32

Visualization engine
Documentation 
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

use std::ops::RangeInclusive;

use crate::util::vector::Vector;
use crate::visual::image::ImageMut;

/// Pixel-related action strategy.
pub enum PixelStrategy<'a, T> {
    /// Overwrite destination pixels with the provided value.
    Overwrite(&'a T),

    /// Apply provided function to calculate the resulting pixel value.
    Modify(&'a mut dyn FnMut((i32, i32), T) -> T),
}

impl<'a, T> PixelStrategy<'a, T>
where
    T: Clone,
{
    pub(super) fn apply(&mut self, position: Vector<i32>, image: &mut dyn ImageMut<T>) {
        match self {
            PixelStrategy::Overwrite(value) => image.set_pixel(position, value),
            PixelStrategy::Modify(function) => image.modify_pixel(position, function),
        }
    }

    pub(super) fn apply_to_line(
        &mut self,
        x: RangeInclusive<i32>,
        y: i32,
        image: &mut dyn ImageMut<T>,
    ) {
        let start = *x.start();
        let end = *x.end();

        let x = if start < end {
            start..=end
        } else {
            end..=start
        };

        match self {
            PixelStrategy::Overwrite(value) => image.set_horizontal_line(x, y, value),
            PixelStrategy::Modify(function) => image.modify_horizontal_line(x, y, function),
        }
    }
}

impl<'a, T> From<&'a T> for PixelStrategy<'a, T> {
    fn from(value: &'a T) -> Self {
        Self::Overwrite(value)
    }
}

impl<'a, T, F> From<&'a mut F> for PixelStrategy<'a, T>
where
    F: FnMut((i32, i32), T) -> T,
{
    fn from(value: &'a mut F) -> Self {
        Self::Modify(value)
    }
}