您最多选择25个主题
主题必须以中文或者字母或数字开头，可以包含连字符 (-)，并且长度不得超过35个字符
69 行
1.8 KiB
69 行
1.8 KiB
using System.Collections;
|
|
using System.Collections.Generic;
|
|
using UnityEngine;
|
|
|
|
/// <summary>
|
|
/// Light weight state machine
|
|
/// </summary>
|
|
/// <typeparam name="T"></typeparam>
|
|
|
|
class StateMachine<T>
|
|
{
|
|
public delegate void StateFunc();
|
|
|
|
public void Add(T id, StateFunc enter, StateFunc update, StateFunc leave)
|
|
{
|
|
m_States.Add(id, new State(id, enter, update, leave));
|
|
}
|
|
|
|
public T CurrentState()
|
|
{
|
|
return m_CurrentState.Id;
|
|
}
|
|
|
|
public void Update()
|
|
{
|
|
m_CurrentState.Update();
|
|
}
|
|
|
|
public void Shutdown()
|
|
{
|
|
if (m_CurrentState != null && m_CurrentState.Leave != null)
|
|
m_CurrentState.Leave();
|
|
m_CurrentState = null;
|
|
}
|
|
|
|
public void SwitchTo(T state)
|
|
{
|
|
GameDebug.Assert(m_States.ContainsKey(state), "Trying to switch to unknown state " + state.ToString());
|
|
GameDebug.Assert(m_CurrentState == null || !m_CurrentState.Id.Equals(state), "Trying to switch to " + state.ToString() + " but that is already current state");
|
|
|
|
var newState = m_States[state];
|
|
GameDebug.Log("Switching state: " + (m_CurrentState != null ? m_CurrentState.Id.ToString() : "null") + " -> " + state.ToString());
|
|
|
|
if (m_CurrentState != null && m_CurrentState.Leave != null)
|
|
m_CurrentState.Leave();
|
|
if (newState.Enter != null)
|
|
newState.Enter();
|
|
m_CurrentState = newState;
|
|
|
|
}
|
|
|
|
class State
|
|
{
|
|
public State(T id, StateFunc enter, StateFunc update, StateFunc leave)
|
|
{
|
|
Id = id;
|
|
Enter = enter;
|
|
Update = update;
|
|
Leave = leave;
|
|
}
|
|
public T Id;
|
|
public StateFunc Enter;
|
|
public StateFunc Update;
|
|
public StateFunc Leave;
|
|
}
|
|
|
|
State m_CurrentState = null;
|
|
Dictionary<T, State> m_States = new Dictionary<T, State>();
|
|
}
|