98 lines
2.5 KiB
Rust
98 lines
2.5 KiB
Rust
use std::time::Instant;
|
|
|
|
use color_eyre::Result;
|
|
use ratatui::{
|
|
layout::{Constraint, Layout, Rect},
|
|
style::{Style, Stylize},
|
|
text::Span,
|
|
widgets::Paragraph,
|
|
Frame,
|
|
};
|
|
|
|
use super::Component;
|
|
|
|
use crate::action::Action;
|
|
|
|
#[derive(Debug, Clone, PartialEq)]
|
|
pub struct FpsCounter {
|
|
last_tick_update: Instant,
|
|
tick_count: u32,
|
|
ticks_per_second: f64,
|
|
|
|
last_frame_update: Instant,
|
|
frame_count: u32,
|
|
frames_per_second: f64,
|
|
}
|
|
|
|
impl Default for FpsCounter {
|
|
fn default() -> Self {
|
|
Self::new()
|
|
}
|
|
}
|
|
|
|
impl FpsCounter {
|
|
pub fn new() -> Self {
|
|
Self {
|
|
last_tick_update: Instant::now(),
|
|
tick_count: 0,
|
|
ticks_per_second: 0.0,
|
|
last_frame_update: Instant::now(),
|
|
frame_count: 0,
|
|
frames_per_second: 0.0,
|
|
}
|
|
}
|
|
|
|
fn app_tick(&mut self) -> Result<()> {
|
|
self.tick_count += 1;
|
|
let now = Instant::now();
|
|
let elapsed = (now - self.last_tick_update).as_secs_f64();
|
|
if elapsed >= 1.0 {
|
|
self.ticks_per_second = self.tick_count as f64 / elapsed;
|
|
self.last_tick_update = now;
|
|
self.tick_count = 0;
|
|
}
|
|
Ok(())
|
|
}
|
|
|
|
fn render_tick(&mut self) -> Result<()> {
|
|
self.frame_count += 1;
|
|
let now = Instant::now();
|
|
let elapsed = (now - self.last_frame_update).as_secs_f64();
|
|
if elapsed >= 1.0 {
|
|
self.frames_per_second = self.frame_count as f64 / elapsed;
|
|
self.last_frame_update = now;
|
|
self.frame_count = 0;
|
|
}
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
impl Component for FpsCounter {
|
|
fn update(&mut self, action: Action) -> Result<Option<Action>> {
|
|
match action {
|
|
Action::Tick => self.app_tick()?,
|
|
Action::Render => self.render_tick()?,
|
|
_ => {}
|
|
};
|
|
Ok(None)
|
|
}
|
|
|
|
fn draw(&mut self, frame: &mut Frame, area: Rect) -> Result<()> {
|
|
let [column, _] =
|
|
Layout::horizontal([Constraint::Min(1), Constraint::Length(2)]).areas(area);
|
|
let [_, row, _] = Layout::vertical([
|
|
Constraint::Length(1),
|
|
Constraint::Length(1),
|
|
Constraint::Min(0),
|
|
])
|
|
.areas(column);
|
|
let message = format!(
|
|
"{:.2} ticks/sec, {:.2} FPS",
|
|
self.ticks_per_second, self.frames_per_second
|
|
);
|
|
let span = Span::styled(message, Style::new().dim());
|
|
let paragraph = Paragraph::new(span).right_aligned();
|
|
frame.render_widget(paragraph, row);
|
|
Ok(())
|
|
}
|
|
}
|