plugin-library/Assets/00.StaryEvo/Runtime/Utility/FSM/FSMISystem.cs

using System.Collections.Generic;
using Cysharp.Threading.Tasks;
using UnityEngine;

namespace Stary.Evo
{
    public interface IFsmSystem
    {
        FSMIState CurState { get; set; }
        void AddState(FSMIState state);
        void RemoveState(FSMIState state);
        UniTask SetCurState(string name);
        
        UniTask SetCurState(string name, object param);
        UniTask SetCurState<T>(string name, T param);
        UniTask SetCurState<T1, T2>(string name, T1 param1, T2 param2);
        FSMIState GetStateWithName(string name);
        HashMap<string, FSMIState> States { get; set; }
    }

    public class FsmSystem : MonoBehaviour, IFsmSystem
    {
        public FSMIState CurState { get; set; }
        public HashMap<string, FSMIState> States { get; set; }

        public FsmSystem()
        {
            States = new HashMap<string, FSMIState>();
            AddState(new DefaultState(this));
        }

        public void AddState(FSMIState state)
        {
            Debug.Log(state.Name);
            if (!States.ContainsKey(state.Name))
            {
                States.Add(state.Name, state);
            }
            else
            {
                Debug.LogErrorFormat("States状态机容器里已存在名字为--【{0}】--的状态", state.Name.ToString());
            }
        }

        public void RemoveState(FSMIState state)
        {
            if (States.ContainsKey(state.Name))
            {
                States.Remove(state.Name);
            }
            else
            {
                Debug.LogErrorFormat("States状态机容器里不存在名字为--【{0}】--的状态", state.Name.ToString());
            }
        }

        /// <summary>
        /// 普通状态切换，适用于无参
        /// </summary>
        /// <param name="name"></param>
        public async UniTask SetCurState(string name)
        {
            if (CurState != null)
               await CurState.OnExitAsync();
            FSMIState state = GetStateWithName(name);
            CurState = state;
            CurState.OnEnterAsync();
        }
        /// <summary>
        /// 异步状态切换，适用于有参
        /// </summary>
        /// <param name="name"></param>
        /// <param name="param"></param>
        /// <typeparam name="T"></typeparam>
        public async UniTask SetCurState(string name, object param)
        {
            if (CurState != null)
            {
                await CurState.OnExitAsync();
            }

            FSMIState state = GetStateWithName(name);
            CurState = state;
            await CurState.OnEnterAsync(param);
        }
        /// <summary>
        /// 异步状态切换，适用于有参
        /// </summary>
        /// <param name="name"></param>
        /// <param name="param"></param>
        /// <typeparam name="T"></typeparam>
        public async UniTask SetCurState<T>(string name, T param)
        {
            if (CurState != null)
            {
                await CurState.OnExitAsync();
            }

            FSMIState state = GetStateWithName(name);
            CurState = state;
            await CurState.OnEnterAsync<T>(param);
        }

        /// <summary>
        /// 异步状态切换，适用于有参
        /// </summary>
        /// <param name="name"></param>
        /// <param name="param1"></param>
        /// <param name="param2"></param>
        /// <typeparam name="T1"></typeparam>
        /// <typeparam name="T2"></typeparam>
        public async UniTask SetCurState<T1, T2>(string name, T1 param1, T2 param2)
        {
            if (CurState != null)
            {
                await CurState.OnExitAsync();
            }

            FSMIState state = GetStateWithName(name);
            CurState = state;
            await CurState.OnEnterAsync(param1, param2);
        }

        public FSMIState GetStateWithName(string name)
        {
            if (States.ContainsKey(name))
            {
                return States[name];
            }

            return null;
        }
    }
}