bevy/crates/bevy_diagnostic/src/frame_time_diagnostics_plugin.rs

66 lines
2.3 KiB
Rust
Raw Normal View History

2020-05-04 20:21:14 +00:00
use crate::{Diagnostic, DiagnosticId, Diagnostics};
2020-07-17 01:47:51 +00:00
use bevy_app::prelude::*;
use bevy_core::Time;
2020-07-10 08:37:06 +00:00
use bevy_ecs::{IntoQuerySystem, Res, ResMut};
2020-05-04 20:21:14 +00:00
/// Adds "frame time" diagnostic to an App, specifically "frame time", "fps" and "frame count"
2020-05-04 20:21:14 +00:00
#[derive(Default)]
pub struct FrameTimeDiagnosticsPlugin;
pub struct FrameTimeDiagnosticsState {
frame_count: f64,
}
2020-08-08 03:22:17 +00:00
impl Plugin for FrameTimeDiagnosticsPlugin {
2020-05-04 20:21:14 +00:00
fn build(&self, app: &mut bevy_app::AppBuilder) {
app.add_startup_system(Self::setup_system.system())
.add_resource(FrameTimeDiagnosticsState { frame_count: 0.0 })
2020-05-04 20:21:14 +00:00
.add_system(Self::diagnostic_system.system());
}
}
impl FrameTimeDiagnosticsPlugin {
pub const FPS: DiagnosticId = DiagnosticId::from_u128(288146834822086093791974408528866909483);
pub const FRAME_COUNT: DiagnosticId =
DiagnosticId::from_u128(54021991829115352065418785002088010277);
2020-05-04 20:21:14 +00:00
pub const FRAME_TIME: DiagnosticId =
DiagnosticId::from_u128(73441630925388532774622109383099159699);
2020-05-04 20:21:14 +00:00
2020-05-14 00:52:47 +00:00
pub fn setup_system(mut diagnostics: ResMut<Diagnostics>) {
diagnostics.add(Diagnostic::new(Self::FRAME_TIME, "frame_time", 20));
diagnostics.add(Diagnostic::new(Self::FPS, "fps", 20));
diagnostics.add(Diagnostic::new(Self::FRAME_COUNT, "frame_count", 1));
2020-05-04 20:21:14 +00:00
}
pub fn diagnostic_system(
mut diagnostics: ResMut<Diagnostics>,
time: Res<Time>,
mut state: ResMut<FrameTimeDiagnosticsState>,
) {
state.frame_count += 1.0;
diagnostics.add_measurement(Self::FRAME_COUNT, state.frame_count);
2020-05-04 20:21:14 +00:00
if time.delta_seconds_f64 == 0.0 {
return;
}
diagnostics.add_measurement(Self::FRAME_TIME, time.delta_seconds_f64);
if let Some(fps) = diagnostics
.get(Self::FRAME_TIME)
.and_then(|frame_time_diagnostic| {
frame_time_diagnostic
.average()
.and_then(|frame_time_average| {
if frame_time_average > 0.0 {
Some(1.0 / frame_time_average)
} else {
None
}
})
})
{
diagnostics.add_measurement(Self::FPS, fps);
}
}
}