1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-22 23:48:12 +00:00

feat(core): introduce a stopwatch

[no changelog]
This commit is contained in:
cepetr 2024-04-26 08:28:33 +02:00 committed by TychoVrahe
parent 212391fc37
commit 1f01150e7c

View File

@ -67,6 +67,18 @@ impl Div<Duration> for Duration {
}
}
impl From<f32> for Duration {
fn from(value: f32) -> Self {
Self::from_millis((value * 1000.0) as u32)
}
}
impl From<Duration> for f32 {
fn from(value: Duration) -> Self {
value.to_millis() as f32 / 1000.0
}
}
/* Instants can wrap around and we want them to be comparable even after
* wrapping around. This works by setting a maximum allowable difference
* between two Instants to half the range. In checked_add and checked_sub, we
@ -148,6 +160,75 @@ impl Ord for Instant {
}
}
/// A stopwatch is a utility designed for measuring the amount of time
/// that elapses between its start and stop points. It can be used in various
/// situations - animation timing, event timing, testing and debugging.
pub enum Stopwatch {
Stopped(Duration),
Running(Instant),
}
impl Default for Stopwatch {
/// Returns a new sopteed stopwatch by default.
fn default() -> Self {
Self::new_stopped()
}
}
impl Stopwatch {
/// Creates a new stopped stopwatch with duration of zero
pub fn new_stopped() -> Self {
Self::Stopped(Duration::ZERO)
}
/// Creates a new started stopwatch that starts
/// at the current instant.
pub fn new_started() -> Self {
Self::Running(Instant::now())
}
/// Starts or restarts the stopwatch.
///
/// If the stopwatch is already running, it restarts, setting
/// the elapsed time to zero.
pub fn start(&mut self) {
*self = Self::Running(Instant::now());
}
/// Stops the stopwatch.
///
/// When stopped, the `elapsed()` method will return the total
/// duration for which the stopwatch was running.
pub fn stop(&mut self) {
*self = Self::Stopped(self.elapsed());
}
/// Returns the elapsed duration since the stopwatch was last started.
///
/// If the stopwatch is running, it calculates the time from the last
/// start instant to the current instant.
pub fn elapsed(&self) -> Duration {
match *self {
Self::Stopped(duration) => duration,
Self::Running(time) => Instant::now().checked_duration_since(time).unwrap(),
}
}
/// Returns `true` if the stopwatch is currenly running.
pub fn is_running(&self) -> bool {
matches!(*self, Self::Running(_))
}
/// Checks if the stopwatch is running and whether the elapsed
/// time since the last start is less than or equal to a specified limit.
pub fn is_running_within(&self, limit: Duration) -> bool {
match *self {
Self::Stopped(_) => false,
Self::Running(_) => self.elapsed() <= limit,
}
}
}
#[cfg(test)]
mod tests {
use super::*;
@ -160,4 +241,54 @@ mod tests {
assert_eq!(later, Instant { millis: 0 });
assert!(earlier < later);
}
#[test]
fn stopwatch_builds_correctly() {
let sw = Stopwatch::new_started();
assert!(sw.is_running());
let sw = Stopwatch::new_stopped();
assert!(!sw.is_running());
let sw: Stopwatch = Default::default();
assert!(!sw.is_running());
}
fn wait(duration: Duration) {
let origin = Instant::now();
while Instant::now().checked_duration_since(origin).unwrap() < duration {}
}
#[test]
fn stopwatch_starts_correcly() {
let mut sw = Stopwatch::new_stopped();
assert!(!sw.is_running());
sw.start();
assert!(sw.is_running());
wait(Duration::from_millis(10));
assert!(sw.elapsed() >= Duration::from_millis(10));
assert!(!sw.is_running_within(Duration::from_millis(5)));
assert!(sw.is_running_within(Duration::from_millis(10000)));
}
#[test]
fn stopwatch_stops_correctly() {
let mut sw = Stopwatch::new_started();
assert!(sw.is_running());
wait(Duration::from_millis(10));
sw.stop();
assert!(!sw.is_running());
let elapsed = sw.elapsed();
assert!(elapsed >= Duration::from_millis(10));
wait(Duration::from_millis(10));
assert!(sw.elapsed() == elapsed);
assert!(!sw.is_running_within(Duration::from_millis(5)));
assert!(!sw.is_running_within(Duration::from_millis(10000)));
}
}