Fix ActionSet

WorkingRobot · Nov 2, 2023 · 0c96ba7 · 0c96ba7
1 parent 74195b5
commit 0c96ba7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/Solver/ActionSet.cs b/Solver/ActionSet.cs
@@ -16,7 +16,7 @@ public struct ActionSet
     private static int FromAction(ActionType action)
     {
         var ret = Simulator.AcceptedActionsLUT[(byte)action];
-        if (ret == 0)
+        if (ret == -1)
             throw new ArgumentOutOfRangeException(nameof(action), action, $"Action {action} is unsupported in {nameof(ActionSet)}.");
         return ret;
     }

diff --git a/Solver/Simulator.cs b/Solver/Simulator.cs
@@ -63,6 +63,8 @@ public Simulator(SimulationState state, int maxStepCount) : base(state)
     static Simulator()
     {
         AcceptedActionsLUT = new int[Enum.GetValues<ActionType>().Length];
+        for (var i = 0; i < AcceptedActionsLUT.Length; i++)
+            AcceptedActionsLUT[i] = -1;
         for (var i = 0; i < AcceptedActions.Length; i++)
             AcceptedActionsLUT[(byte)AcceptedActions[i]] = i;
     }