fix: prevent error or no previous state

readyplayerme · Jun 10, 2024 · 819e7c5 · 819e7c5
1 parent ac0d942
commit 819e7c5
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/Samples~/AvatarCreatorSamples/AvatarCreatorWizard/Scripts/UI/StateMachine/StateMachine.cs b/Samples~/AvatarCreatorSamples/AvatarCreatorWizard/Scripts/UI/StateMachine/StateMachine.cs
@@ -13,7 +13,7 @@ public abstract class StateMachine : MonoBehaviour
 
         protected Action<StateType, StateType> StateChanged;
         private StateType currentState;
-        
+
         protected void Initialize(List<State> states)
         {
             foreach (var state in states)
@@ -34,7 +34,7 @@ public void SetState(StateType stateType)
                     SetState(stateTypeMap[stateType].NextState);
                     return;
                 }
-                
+
                 previousStates.Push(previousState);
             }
 
@@ -56,7 +56,11 @@ public void GoToPreviousState()
             var previousState = currentState;
 
             DeactivateState(stateTypeMap[previousState]);
-
+
+            if (previousStates.Count == 0)
+            {
+                return;
+            }
             currentState = previousStates.Pop();
             if (currentState != StateType.None)
             {
@@ -70,7 +74,7 @@ private void ActivateState(State state)
             state.gameObject.SetActive(true);
             state.ActivateState();
         }
-        
+
         private void DeactivateState(State state)
         {
             state.gameObject.SetActive(false);