ori_core/views/
draw_handler.rs

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
use crate::{
    context::{BuildCx, DrawCx, EventCx, LayoutCx, RebuildCx},
    event::Event,
    layout::{Size, Space},
    view::View,
};

/// Create a new [`DrawHandler`].
pub fn on_draw<T, V>(
    content: V,
    on_draw: impl FnMut(&mut DrawCx, &mut T) + 'static,
) -> DrawHandler<T, V> {
    DrawHandler::new(content).on_draw(on_draw)
}

/// A view that hooks into the draw cycle.
pub struct DrawHandler<T, V> {
    /// The content.
    pub content: V,
    /// The draw callback.
    #[allow(clippy::type_complexity)]
    pub on_draw: Option<Box<dyn FnMut(&mut DrawCx, &mut T) + 'static>>,
}

impl<T, V> DrawHandler<T, V> {
    /// Create a new [`DrawHandler`].
    pub fn new(content: V) -> Self {
        Self {
            content,
            on_draw: Option::None,
        }
    }

    /// Set the draw callback.
    pub fn on_draw(mut self, on_draw: impl FnMut(&mut DrawCx, &mut T) + 'static) -> Self {
        self.on_draw = Some(Box::new(on_draw));
        self
    }
}

impl<T, V: View<T>> View<T> for DrawHandler<T, V> {
    type State = V::State;

    fn build(&mut self, cx: &mut BuildCx, data: &mut T) -> Self::State {
        self.content.build(cx, data)
    }

    fn rebuild(&mut self, state: &mut Self::State, cx: &mut RebuildCx, data: &mut T, old: &Self) {
        self.content.rebuild(state, cx, data, &old.content);
    }

    fn event(
        &mut self,
        state: &mut Self::State,
        cx: &mut EventCx,
        data: &mut T,
        event: &Event,
    ) -> bool {
        self.content.event(state, cx, data, event)
    }

    fn layout(
        &mut self,
        state: &mut Self::State,
        cx: &mut LayoutCx,
        data: &mut T,
        space: Space,
    ) -> Size {
        self.content.layout(state, cx, data, space)
    }

    fn draw(&mut self, state: &mut Self::State, cx: &mut DrawCx, data: &mut T) {
        self.content.draw(state, cx, data);

        if let Some(ref mut on_draw) = self.on_draw {
            on_draw(cx, data);
        }
    }
}