embedded_menu/selection_indicator/style/
animated_triangle.rs

1use embedded_graphics::{
2    prelude::{DrawTarget, Point},
3    primitives::Rectangle,
4    transform::Transform,
5};
6
7use crate::{
8    interaction::InputState,
9    selection_indicator::{
10        style::{interpolate, triangle::Arrow, IndicatorStyle},
11        Insets,
12    },
13    theme::Theme,
14};
15
16#[derive(Clone, Copy)]
17pub struct AnimatedTriangle {
18    period: i32,
19}
20
21impl AnimatedTriangle {
22    pub const fn new(period: i32) -> Self {
23        Self { period }
24    }
25}
26
27#[derive(Default, Clone, Copy)]
28pub struct State {
29    current: i32,
30}
31
32impl IndicatorStyle for AnimatedTriangle {
33    type Shape = Arrow;
34    type State = State;
35
36    fn on_target_changed(&self, state: &mut Self::State) {
37        state.current = 0;
38    }
39
40    fn update(&self, state: &mut Self::State, input_state: InputState) {
41        state.current = if input_state == InputState::Idle {
42            (state.current + 1) % self.period
43        } else {
44            0
45        };
46    }
47
48    fn padding(&self, _state: &Self::State, height: i32) -> Insets {
49        Insets {
50            left: height / 2 + 1,
51            top: 0,
52            right: 0,
53            bottom: 0,
54        }
55    }
56
57    fn shape(&self, state: &Self::State, bounds: Rectangle, fill_width: u32) -> Self::Shape {
58        let max_offset = Self::Shape::tip_width(bounds);
59
60        let half_move = self.period / 5;
61        let rest = 3 * half_move;
62
63        let offset = if state.current < rest {
64            0
65        } else if state.current < rest + half_move {
66            state.current - rest
67        } else {
68            self.period - state.current
69        };
70
71        let offset = offset * max_offset / half_move;
72
73        Arrow::new(bounds, fill_width).translate(Point::new(-offset, 0))
74    }
75
76    fn draw<T, D>(
77        &self,
78        state: &Self::State,
79        input_state: InputState,
80        theme: &T,
81        display: &mut D,
82    ) -> Result<Self::Shape, D::Error>
83    where
84        T: Theme,
85        D: DrawTarget<Color = T::Color>,
86    {
87        let display_area = display.bounding_box();
88
89        let fill_width = if let InputState::InProgress(progress) = input_state {
90            interpolate(progress as u32, 0, 255, 0, display_area.size.width)
91        } else {
92            0
93        };
94
95        let shape = self.shape(state, display_area, fill_width);
96
97        shape.draw(theme.selection_color(), display)?;
98
99        Ok(shape)
100    }
101}