plugin-library/Assets/00.StaryEvo/Runtime/Tool/FSM/FSMISystemAsync.cs

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

namespace Stary.Evo
{
    public interface IFsmSystemAsync
    {
        FSMIStateAsync CurState { get; set; }
        void AddState(FSMIStateAsync state);
        void RemoveState(FSMIStateAsync 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);
        FSMIStateAsync GetStateWithName(string name);
        HashMap<string, FSMIStateAsync> States { get; set; }
    }

    public class FsmSystemAsync : IFsmSystemAsync
    {
        public FSMIStateAsync CurState { get; set; }
        public HashMap<string, FSMIStateAsync> States { get; set; }

        public FsmSystemAsync()
        {
            States = new HashMap<string, FSMIStateAsync>();
            AddState(new DefaultStateAsync(this));
        }

        public void AddState(FSMIStateAsync 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(FSMIStateAsync 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();
            FSMIStateAsync 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();
            }

            FSMIStateAsync 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();
            }

            FSMIStateAsync 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();
            }

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

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

            return null;
        }
    }
}