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

use crossterm::style::{ContentStyle, StyledContent};

#[derive(Debug, Clone, Default)]
pub struct StyledChar {
    pub style: ContentStyle,
    pub content: char,
}

impl PartialEq for StyledChar {
    fn eq(&self, other: &Self) -> bool {
        self.content == other.content
            && self.style.foreground_color == other.style.foreground_color
            && self.style.background_color == other.style.background_color
            && self.style.attributes == other.style.attributes
    }
}

impl StyledChar {
    pub(crate) fn from_char(c: char) -> StyledChar {
        StyledChar { style: Default::default(), content: c}
    }
    pub fn to_styled_content(&self) -> StyledContent<char> {
        StyledContent::new(self.style.clone(), self.content)
    }
}

#[derive(Clone)]
pub struct SizedBuffer {
    size: (u16, u16), // (width, height)
    buffer: Box<[StyledChar]>,
}

pub struct Pixel {
    pub position: (u16, u16),
    pub value: StyledChar,
}

impl SizedBuffer {
    pub fn set_pixel(&mut self, val: &StyledChar, x: u16, y: u16) -> bool {
        if x>=self.width()|| y>=self.height() {
            return false;
        }
        self.buffer[(y * self.width() + x) as usize] = val.clone();
        return true;
    }

    pub fn get_pixel(&self, x: u16, y: u16) -> StyledChar {
        self.buffer[(y * self.width() + x) as usize].clone()
    }

    pub fn draw_rect(&mut self, val: &StyledChar, x1: u16, y1: u16, x2:u16, y2: u16) {
        for y in y1..=y2 {
            self.draw_hline(val, y, x1, x2);
        }
    }

    pub fn draw_vline(&mut self, val: &StyledChar, x: u16, y1: u16, y2: u16) {
        for y in y1..=y2 {
            self.set_pixel(val, x, y);
        }
    }

    pub fn draw_hline(&mut self, val: &StyledChar, y: u16, x1: u16, x2: u16) {
        for x in x1..=x2 {
            self.set_pixel(val, x, y);
        }
    }

    pub fn width(&self) -> u16 { self.size.0 }

    pub fn height(&self) -> u16 { self.size.1 }

    pub fn compare(&self, other: &SizedBuffer) -> Vec<Pixel> {
        let mut ret: Vec<Pixel> = vec![];
        for x in 0..self.size.0{
            for y in 0..self.size.1{
                if self.get_pixel(x, y) != other.get_pixel(x,y) {
                    ret.push(Pixel { position: (x, y), value: self.get_pixel(x, y) })
                }
            }
        }
        ret
    }

    pub fn new(w: u16, h: u16) -> SizedBuffer {
        SizedBuffer{ size: (w, h), buffer: vec![Default::default(); (w*h) as usize].into_boxed_slice() }
    }
}