Fix ActionSet

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

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/Craftimizer.Test/Solver/ActionSet.cs b/Craftimizer.Test/Solver/ActionSet.cs
@@ -15,7 +15,7 @@ public void TestAcceptedActions()
         foreach (var i in Enum.GetValues<ActionType>())
         {
             var idx = lut[(byte)i];
-            if (idx != 0)
+            if (idx != -1)
                 Assert.AreEqual(i, actions[idx]);
         }
     }

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;
     }