Update StackOverflowModule locking, version 0.1.13

techno-dwarf-works · May 21, 2024 · c212a89 · c212a89
1 parent 1b5b0df
commit c212a89
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 10 deletions.
diff --git a/Assets/BetterStateMachine/Runtime/Modules/StackOverflowModule.cs b/Assets/BetterStateMachine/Runtime/Modules/StackOverflowModule.cs
@@ -38,7 +38,17 @@ public void Unlock()
 
         protected internal override bool AllowChangeState(IStateMachine<TState> stateMachine, TState state)
         {
-            return base.AllowChangeState(stateMachine, state) && !IsLocked;
+            if (!base.AllowChangeState(stateMachine, state))
+            {
+                return false;
+            }
+
+            if (Depth >= OverflowDepth)
+            {
+                Lock();
+            }
+
+            return !IsLocked;
         }
 
         protected internal override void OnStatePreChanged(IStateMachine<TState> stateMachine, TState state)
@@ -47,11 +57,6 @@ protected internal override void OnStatePreChanged(IStateMachine<TState> stateMa
 
             Depth++;
             MaxDepth = Mathf.Max(MaxDepth, Depth);
-
-            if (Depth >= OverflowDepth)
-            {
-                Lock();
-            }
         }
 
         protected internal override void OnStateChanged(IStateMachine<TState> stateMachine, TState state)
@@ -65,9 +70,7 @@ protected internal override void OnMachineStopped(IStateMachine<TState> stateMac
         {
             base.OnMachineStopped(stateMachine);
 
-            Depth = 0;
-            MaxDepth = 0;
-            IsLocked = false;
+            Unlock();
         }
     }
 }
diff --git a/Assets/BetterStateMachine/package.json b/Assets/BetterStateMachine/package.json
@@ -1,7 +1,7 @@
 {
   "name": "com.tdw.better.statemachine",
   "displayName": "Better State Machine",
-  "version": "0.1.12",
+  "version": "0.1.13",
   "unity": "2021.3",
   "description": " ",
   "dependencies": {