added new is sim logic

Programming-TRIGON · Nov 11, 2024 · 2245cd8 · 2245cd8
1 parent 0841029
commit 2245cd8
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/src/main/java/org/trigon/hardware/RobotHardwareStats.java b/src/main/java/org/trigon/hardware/RobotHardwareStats.java
@@ -5,9 +5,9 @@ public class RobotHardwareStats {
     private static boolean IS_REPLAY = false;
     private static double PERIODIC_TIME_SECONDS = 0.02;
 
-    public static void setCurrentRobotStats(boolean isReal, boolean isSimulation, boolean isReplay) {
-        IS_SIMULATION = isSimulation && !isReal;
-        IS_REPLAY = isReplay && !isReal;
+    public static void setCurrentRobotStats(boolean isReal, ReplayType replayType) {
+        IS_SIMULATION = isReal || replayType.equals(ReplayType.NONE) ? !isReal : replayType.equals(ReplayType.SIMULATION_REPLAY);
+        IS_REPLAY = !isReal && !replayType.equals(ReplayType.NONE);
     }
 
     public static void setPeriodicTimeSeconds(double periodicTimeSeconds) {
@@ -25,4 +25,10 @@ public static boolean isReplay() {
     public static boolean isSimulation() {
         return IS_SIMULATION;
     }
-}
+
+    public enum ReplayType {
+        NONE,
+        SIMULATION_REPLAY,
+        REAL_REPLAY
+    }
+}