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

use std::marker::PhantomData;

use crate::{
    core::{CoreStateImplementation, DrawingContextEnum},
    state::CoreStateData,
    util::PhantomUnsend,
    Color, RectRegion,
};

pub struct DrawingContext<'a> {
    context: DrawingContextEnum,
    data: *mut CoreStateData,
    _unsend: PhantomUnsend,
    _phantom_data: PhantomData<&'a ()>,
}

impl<'a> DrawingContext<'a> {
    pub fn draw_line(&mut self, x1: u16, y1: u16, x2: u16, y2: u16) {
        let context = self.context;

        self.get_data_mut()
            .core_state
            .draw_line(context, x1, y1, x2, y2)
            .unwrap()
    }

    pub fn draw_rectangle(&mut self, rectangle: RectRegion) {
        let context = self.context;

        self.get_data_mut()
            .core_state
            .draw_rectangle(context, rectangle)
            .unwrap()
    }

    pub fn set_draw_color(&mut self, color: Color) {
        let context = self.context;

        self.get_data_mut()
            .core_state
            .set_draw_color(context, color)
            .unwrap()
    }
}

impl<'a> DrawingContext<'a> {
    pub(crate) fn from_parts(context: DrawingContextEnum, data: *mut CoreStateData) -> Self {
        Self {
            context,
            data,
            _unsend: Default::default(),
            _phantom_data: PhantomData,
        }
    }

    fn get_data_mut(&mut self) -> &mut CoreStateData {
        unsafe { &mut *self.data }
    }
}