embedded_menu/selection_indicator/style/
animated_triangle.rs1use 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}