1 files changed, 11 insertions, 1 deletions
diff --git a/src_testbed/plugin.rs b/src_testbed/plugin.rs
index 24a91dd..e7f2a84 100644
--- a/src_testbed/plugin.rs
+++ b/src_testbed/plugin.rs
@@ -2,6 +2,7 @@ use crate::harness::Harness;
 use crate::physics::PhysicsState;
 use crate::GraphicsManager;
 use bevy::prelude::{Assets, Commands, Mesh, Query, StandardMaterial, Transform};
+use bevy_egui::EguiContext;
 use na::Point3;
 
 pub trait TestbedPlugin {
@@ -13,7 +14,6 @@ pub trait TestbedPlugin {
         materials: &mut Assets<StandardMaterial>,
         components: &mut Query<(&mut Transform,)>,
         harness: &mut Harness,
-
         gen_color: &mut dyn FnMut() -> Point3<f32>,
     );
     fn clear_graphics(&mut self, graphics: &mut GraphicsManager, commands: &mut Commands);
@@ -28,5 +28,15 @@ pub trait TestbedPlugin {
         components: &mut Query<(&mut Transform,)>,
         harness: &mut Harness,
     );
+    fn update_ui(
+        &mut self,
+        ui_context: &EguiContext,
+        harness: &mut Harness,
+        graphics: &mut GraphicsManager,
+        commands: &mut Commands,
+        meshes: &mut Assets<Mesh>,
+        materials: &mut Assets<StandardMaterial>,
+        components: &mut Query<(&mut Transform,)>,
+    );
     fn profiling_string(&self) -> String;
 }