Added a method for swapping substate models

Author: PaulNonatomic

Runtime/StateGraph/StateMachineController.cs

@@ -36,7 +36,7 @@ public class StateMachineController : MonoBehaviour
 		/// Jumps to a state via a JumpIn state using the JumpId to identify the JumpIn node.
 		/// </summary>
 		/// <param name="id">The JumpIn id of the destination JumpIn node</param>
-		public void JumpTo(JumpId id)
+		public virtual void JumpTo(JumpId id)
 		{
 			_stateMachine?.JumpTo(id);
 		}
@@ -73,7 +73,7 @@ public virtual void Reset()
 		/// <summary>
 		/// Initializes the state machine when the script instance is being loaded.
 		/// </summary>
-		public void Awake()
+		public virtual void Awake()
 		{
 			CreateStateMachine();
 			_activated = _stateMachine != null;
@@ -83,7 +83,7 @@ public void Awake()
 		/// Starts the state machine if it has been activated.
 		/// This is called on the frame when a script is enabled just before any of the Update methods are called the first time.
 		/// </summary>
-		public void Start()
+		public virtual void Start()
 		{
 			if(!_activated) return;
 
@@ -96,7 +96,7 @@ public void Start()
 		/// Updates the state machine if it has been activated.
 		/// This is called every frame, if the MonoBehaviour is enabled.
 		/// </summary>
-		public void Update()
+		public virtual void Update()
 		{
 			if (!_activated) return;
 			_stateMachine?.Update();
@@ -106,7 +106,7 @@ public void Update()
 		/// Performs physics-based updates on the state machine if it has been activated.
 		/// This is called every fixed framerate frame, if the MonoBehaviour is enabled.
 		/// </summary>
-		public void FixedUpdate()
+		public virtual void FixedUpdate()
 		{
 			if (!_activated) return;
 			_stateMachine?.FixedUpdate();
@@ -115,7 +115,7 @@ public void FixedUpdate()
 		/// <summary>
 		/// Cleans up the state machine when the MonoBehaviour will be destroyed.
 		/// </summary>
-		public void OnDestroy()
+		public virtual void OnDestroy()
 		{
 			if (!_activated) return;
 			_stateMachine?.OnDestroy();

Runtime/StateGraph/States/BaseSubStateMachineState.cs

@@ -15,10 +15,12 @@ public abstract class BaseSubStateMachineState : State
 		public StateMachineModel Model
 		{
 			get => _model;
-			protected set => _model = value;
+			set => _model = value;
 		}
 
 		[SerializeField] private StateMachineModel _model;
+		private bool _started;
+		private bool _entered;
 
 		public override void OnAwakeState()
 		{
@@ -29,12 +31,14 @@ public override void OnAwakeState()
 		public override void OnStartState()
 		{
 			SubStateMachine?.Start();
+			_started = true;
 		}
 
 		public override void OnEnterState()
 		{
 			if(SubStateMachine == null) return;
-			
+
+			_entered = true;
 			SubStateMachine.Model.SetParent(SubStateMachine.Model);
 			SubStateMachine.OnComplete += OnSubStateComplete;
 			SubStateMachine.Enter();
@@ -56,6 +60,7 @@ public override void OnExitState()
 
 			SubStateMachine.OnComplete -= OnSubStateComplete;
 			SubStateMachine.Exit();
+			_entered = false;
 		}
 
 		public override void OnDestroyState()
@@ -64,6 +69,23 @@ public override void OnDestroyState()
 
 			SubStateMachine?.OnDestroy();
 		}
+		
+		protected virtual void SwitchModel(StateMachineModel value)
+		{
+			if (!value) return;
+			
+			SubStateMachine?.OnDestroy();
+			SubStateMachine = null;
+			
+			_model = value;
+			CreateStateMachine();
+
+			if (!GameObject.activeInHierarchy || !_started) return;
+			SubStateMachine?.Start();
+
+			if (!_entered) return;
+			SubStateMachine?.Enter();
+		}
 
 		protected virtual void CreateStateMachine()
 		{