mirror of
https://github.com/bevyengine/bevy
synced 2024-12-23 19:43:07 +00:00
111 lines
3.3 KiB
Rust
111 lines
3.3 KiB
Rust
|
use super::{Diagnostic, DiagnosticId, Diagnostics};
|
||
|
use bevy_app::{stage, AppPlugin};
|
||
|
use bevy_core::time::Time;
|
||
|
use legion::prelude::*;
|
||
|
use std::time::Duration;
|
||
|
|
||
|
pub struct PrintDiagnosticsPlugin {
|
||
|
pub debug: bool,
|
||
|
pub wait_duration: Duration,
|
||
|
pub filter: Option<Vec<DiagnosticId>>,
|
||
|
}
|
||
|
|
||
|
pub struct PrintDiagnosticsState {
|
||
|
elapsed: f64,
|
||
|
wait_seconds: f64,
|
||
|
filter: Option<Vec<DiagnosticId>>,
|
||
|
}
|
||
|
|
||
|
impl Default for PrintDiagnosticsPlugin {
|
||
|
fn default() -> Self {
|
||
|
PrintDiagnosticsPlugin {
|
||
|
debug: false,
|
||
|
wait_duration: Duration::from_secs(1),
|
||
|
filter: None,
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
impl AppPlugin for PrintDiagnosticsPlugin {
|
||
|
fn build(&self, app: &mut bevy_app::AppBuilder) {
|
||
|
app.add_resource(PrintDiagnosticsState {
|
||
|
elapsed: 0.0,
|
||
|
wait_seconds: self.wait_duration.as_secs_f64(),
|
||
|
filter: self.filter.clone(),
|
||
|
});
|
||
|
|
||
|
if self.debug {
|
||
|
app.add_system_to_stage(
|
||
|
stage::POST_UPDATE,
|
||
|
Self::print_diagnostics_debug_system.system(),
|
||
|
);
|
||
|
} else {
|
||
|
app.add_system_to_stage(stage::POST_UPDATE, Self::print_diagnostics_system.system());
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
impl PrintDiagnosticsPlugin {
|
||
|
pub fn filtered(filter: Vec<DiagnosticId>) -> Self {
|
||
|
PrintDiagnosticsPlugin {
|
||
|
filter: Some(filter),
|
||
|
..Default::default()
|
||
|
}
|
||
|
}
|
||
|
|
||
|
fn print_diagnostic(diagnostic: &Diagnostic) {
|
||
|
if let Some(value) = diagnostic.value() {
|
||
|
print!("{:<20}: {:<19.6}", diagnostic.name, value);
|
||
|
if let Some(average) = diagnostic.average() {
|
||
|
print!(" (avg {:.6})", average);
|
||
|
}
|
||
|
|
||
|
println!("\n");
|
||
|
}
|
||
|
}
|
||
|
|
||
|
pub fn print_diagnostics_system(
|
||
|
mut state: ResourceMut<PrintDiagnosticsState>,
|
||
|
time: Resource<Time>,
|
||
|
diagnostics: Resource<Diagnostics>,
|
||
|
) {
|
||
|
state.elapsed += time.delta_seconds_f64;
|
||
|
if state.elapsed >= state.wait_seconds {
|
||
|
state.elapsed = 0.0;
|
||
|
println!("Diagnostics:");
|
||
|
println!("{}", "-".repeat(50));
|
||
|
if let Some(ref filter) = state.filter {
|
||
|
for diagnostic in filter.iter().map(|id| diagnostics.get(*id).unwrap()) {
|
||
|
Self::print_diagnostic(diagnostic);
|
||
|
}
|
||
|
} else {
|
||
|
for diagnostic in diagnostics.iter() {
|
||
|
Self::print_diagnostic(diagnostic);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
pub fn print_diagnostics_debug_system(
|
||
|
mut state: ResourceMut<PrintDiagnosticsState>,
|
||
|
time: Resource<Time>,
|
||
|
diagnostics: Resource<Diagnostics>,
|
||
|
) {
|
||
|
state.elapsed += time.delta_seconds_f64;
|
||
|
if state.elapsed >= state.wait_seconds {
|
||
|
state.elapsed = 0.0;
|
||
|
println!("Diagnostics (Debug):");
|
||
|
println!("{}", "-".repeat(30));
|
||
|
if let Some(ref filter) = state.filter {
|
||
|
for diagnostic in filter.iter().map(|id| diagnostics.get(*id).unwrap()) {
|
||
|
println!("{:#?}\n", diagnostic);
|
||
|
}
|
||
|
} else {
|
||
|
for diagnostic in diagnostics.iter() {
|
||
|
println!("{:#?}\n", diagnostic);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|