bevy/crates/bevy_diagnostic/src/lib.rs

mod diagnostic;
pub mod diagnostics;
pub use diagnostic::*;

use bevy_app::{AppBuilder, AppPlugin};
use diagnostics::{
    frame_time_diagnostic_system, print_diagnostics_system, setup_frame_time_diagnostic_system,
    PrintDiagnosticsState,
};
use legion::prelude::IntoSystem;
use std::time::Duration;

pub struct PrintDiagnostics {
    pub wait_duration: Duration,
    pub filter: Option<Vec<DiagnosticId>>,
}

pub struct DiagnosticsPlugin {
    pub print_diagnostics: Option<PrintDiagnostics>,
    pub add_defaults: bool,
}

impl Default for DiagnosticsPlugin {
    fn default() -> Self {
        DiagnosticsPlugin {
            print_diagnostics: Some(PrintDiagnostics {
                wait_duration: Duration::from_secs_f64(1.0),
                filter: None,
            }),
            add_defaults: true,
        }
    }
}

impl AppPlugin for DiagnosticsPlugin {
    fn build(&self, app: &mut AppBuilder) {
        app.init_resource::<Diagnostics>();
        if self.add_defaults {
            app.add_startup_system(setup_frame_time_diagnostic_system.system())
                .add_system(frame_time_diagnostic_system.system());
        }

        if let Some(ref print_diagnostics) = self.print_diagnostics {
            app.add_resource(PrintDiagnosticsState::new(print_diagnostics.wait_duration))
                .add_system(print_diagnostics_system.system());
        }
    }
}