trezor-firmware/core/embed/rust/src/ui/demo/screen2.rs

use crate::ui::{
    component::{Component, Event, EventCtx, Never},
    display::Color,
    event::TouchEvent,
    geometry::{Offset, Point, Rect},
    layout::obj::{ComponentMsgObj, LayoutObj},
    model_tt::{component::Frame, theme},
    shape,
    shape::Renderer,
};

use crate::{
    error::Error,
    micropython::{gc::Gc, obj::Obj},
};
use pareen;

use super::anim_timer::AnimTimer;

pub struct Screen1 {
    jump_time: AnimTimer,
}

impl Screen1 {
    pub fn new() -> Self {
        Self {
            jump_time: {
                let mut timer = AnimTimer::new();
                timer.start();
                timer
            },
        }
    }
}

impl Screen1 {
    fn anim_radius(&self) -> (i16, i16, i16) {
        let r1_anim = pareen::circle()
            .cos()
            .scale_min_max(60f32, 64f32)
            .repeat(1.0f32);

        let r2_anim = pareen::circle()
            .cos()
            .scale_min_max(52f32, 58f32)
            .squeeze(0f32..=0.5f32)
            .repeat(0.5f32);

        let r3_anim = pareen::circle()
            .cos()
            .scale_min_max(40f32, 50f32)
            .squeeze(0f32..=2f32)
            .repeat(2f32);

        let t = self.jump_time.elapsed();

        (
            r1_anim.eval(t) as i16,
            r2_anim.eval(t) as i16,
            r3_anim.eval(t) as i16,
        )
    }
}

impl Component for Screen1 {
    type Msg = Never;

    fn place(&mut self, bounds: Rect) -> Rect {
        bounds
    }

    fn event(&mut self, _ctx: &mut EventCtx, event: Event) -> Option<Self::Msg> {
        match event {
            Event::Touch(TouchEvent::TouchStart(_)) => {
                self.jump_time.start();
            }
            _ => {}
        }

        None
    }

    fn paint(&mut self) {}

    fn render<'s>(&'s self, target: &mut impl Renderer<'s>) {
        let bg_color = Color::rgb(40, 20, 0);

        let r = Rect::new(Point::new(10, 30), Point::new(230, 230));
        shape::Bar::new(r).with_bg(bg_color).render(target);

        let center = Point::new(120, 120);

        let (r1, r2, r3) = self.anim_radius();

        shape::Circle::new(center, r1)
            .with_thickness(2)
            .with_bg(bg_color)
            .with_fg(Color::rgb(0, 120, 0))
            .render(target);

        shape::Circle::new(center, r2)
            .with_thickness(3)
            .with_bg(bg_color)
            .with_fg(Color::rgb(0, 120, 30))
            .render(target);

        shape::Circle::new(center, r3)
            .with_thickness(4)
            .with_bg(bg_color)
            .with_fg(Color::rgb(0, 120, 30))
            .render(target);
    }
}

#[cfg(feature = "ui_debug")]
impl crate::trace::Trace for Screen1 {
    fn trace(&self, t: &mut dyn crate::trace::Tracer) {
        t.component("Demo screen");
    }
}

impl ComponentMsgObj for Screen1 {
    fn msg_try_into_obj(&self, _msg: Self::Msg) -> Result<Obj, Error> {
        unreachable!();
    }
}

pub fn build_screen2() -> Result<Gc<LayoutObj>, Error> {
    LayoutObj::new(Frame::left_aligned(
        theme::label_title(),
        "Animation Demo 2".into(),
        Screen1::new(),
    ))
}