From be07227e9455e1549f1fee2f278ab2a266c5c3c2 Mon Sep 17 00:00:00 2001
From: rezural <rezural@protonmail.com>
Date: Sun, 20 Dec 2020 23:58:50 +1100
Subject: add HarnessState to callbacks, move HarnessPlugin to
 src_testbed/harness/plugin

---
 src_testbed/harness/mod.rs    | 26 +++++++++++++++++++++-----
 src_testbed/harness/plugin.rs | 10 ++++++++++
 2 files changed, 31 insertions(+), 5 deletions(-)
 create mode 100644 src_testbed/harness/plugin.rs

(limited to 'src_testbed/harness')

diff --git a/src_testbed/harness/mod.rs b/src_testbed/harness/mod.rs
index af9a45d..7d6a939 100644
--- a/src_testbed/harness/mod.rs
+++ b/src_testbed/harness/mod.rs
@@ -1,10 +1,13 @@
 use crate::physics::{PhysicsEvents, PhysicsState};
-use crate::HarnessPlugin;
 use rapier::dynamics::{IntegrationParameters, JointSet, RigidBodySet};
 use rapier::geometry::{BroadPhase, ColliderSet, NarrowPhase};
 use rapier::math::Vector;
 use rapier::pipeline::{ChannelEventCollector, PhysicsPipeline, QueryPipeline};
 
+pub mod plugin;
+
+use plugin::HarnessPlugin;
+
 // #[derive(PartialEq)]
 // pub enum RunState {
 //     Initialized,
@@ -32,7 +35,7 @@ pub struct Harness {
     pub state: HarnessState,
 }
 
-type Callbacks = Vec<Box<dyn FnMut(&mut PhysicsState, &PhysicsEvents, f32)>>;
+type Callbacks = Vec<Box<dyn FnMut(&mut PhysicsState, &PhysicsEvents, &HarnessState, f32)>>;
 
 #[allow(dead_code)]
 impl Harness {
@@ -141,7 +144,8 @@ impl Harness {
         self.plugins.push(Box::new(plugin));
     }
 
-    pub fn add_callback<F: FnMut(&mut PhysicsState, &PhysicsEvents, f32) + 'static>(
+    // type StepCallback = FnMut(&mut PhysicsState, &PhysicsEvents, f32);
+    pub fn add_callback<F: FnMut(&mut PhysicsState, &PhysicsEvents, &HarnessState, f32) + 'static>(
         &mut self,
         callback: F,
     ) {
@@ -191,10 +195,22 @@ impl Harness {
             plugin.step(&mut self.physics)
         }
 
-        //FIXME: not sure if this makes sense here, basically copied from Testbed
+        for f in &mut self.callbacks {
+            f(
+                &mut self.physics,
+                &self.events,
+                &self.state,
+                self.time,
+            )
+        }
+
         for plugin in &mut self.plugins {
-            plugin.run_callbacks(&mut self.physics, self.time)
+            plugin.run_callbacks(&mut self.physics, &self.events,&self.state, self.time)
         }
+
+        self.events.poll_all();
+
+        self.time += self.physics.integration_parameters.dt();
     }
 
     pub fn run(&mut self) {
diff --git a/src_testbed/harness/plugin.rs b/src_testbed/harness/plugin.rs
new file mode 100644
index 0000000..d5cf158
--- /dev/null
+++ b/src_testbed/harness/plugin.rs
@@ -0,0 +1,10 @@
+use crate::harness::HarnessState;
+use crate::PhysicsState;
+use crate::physics::PhysicsEvents;
+
+pub trait HarnessPlugin {
+    //FIXME: is run_callbacks needed?
+    fn run_callbacks(&mut self, physics: &mut PhysicsState, events: &PhysicsEvents, harness_state: &HarnessState, t: f32);
+    fn step(&mut self, physics: &mut PhysicsState);
+    fn profiling_string(&self) -> String;
+}
\ No newline at end of file
-- 
cgit