From ee071242c63339e33943f3bcd2d6e2f45a84d1b2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?B=C3=A1lint=20Hodossy?= <bkh16@ic.ac.uk>
Date: Fri, 26 Jan 2024 14:35:03 +0000
Subject: [PATCH] Constrain MjScene EventHandlers as events, consistently use
 MjStepArgs

---
 unity/Runtime/Components/MjScene.cs | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/unity/Runtime/Components/MjScene.cs b/unity/Runtime/Components/MjScene.cs
index 6d80893e..c55c5cf5 100644
--- a/unity/Runtime/Components/MjScene.cs
+++ b/unity/Runtime/Components/MjScene.cs
@@ -78,11 +78,11 @@ public class MjScene : MonoBehaviour {
 
   private List<MjComponent> _orderedComponents;
 
-  public EventHandler<MjStepArgs> postInitEvent;
-  public EventHandler preUpdateEvent;
-  public EventHandler<MjStepArgs> ctrlCallback;
-  public EventHandler postUpdateEvent;
-  public EventHandler<MjStepArgs> preDestroyEvent;
+  public event EventHandler<MjStepArgs> postInitEvent;
+  public event EventHandler<MjStepArgs> preUpdateEvent;
+  public event EventHandler<MjStepArgs> ctrlCallback;
+  public event EventHandler<MjStepArgs> postUpdateEvent;
+  public event EventHandler<MjStepArgs> preDestroyEvent;
 
   protected unsafe void Start() {
     SceneRecreationAtLateUpdateRequested = false;
@@ -94,9 +94,9 @@ public class MjScene : MonoBehaviour {
   }
 
   protected unsafe void FixedUpdate() {
-    preUpdateEvent?.Invoke(this, EventArgs.Empty);
+    preUpdateEvent?.Invoke(this, new MjStepArgs(Model, Data));
     StepScene();
-    postUpdateEvent?.Invoke(this, EventArgs.Empty);
+    postUpdateEvent?.Invoke(this, new MjStepArgs(Model, Data));
   }
 
   public bool SceneRecreationAtLateUpdateRequested = false;