您最多选择25个主题
主题必须以中文或者字母或数字开头，可以包含连字符 (-)，并且长度不得超过35个字符
99 行
2.2 KiB
99 行
2.2 KiB
using System;
|
|
using System.Collections.Generic;
|
|
using UnityEngine;
|
|
|
|
namespace UOP1.StateMachine
|
|
{
|
|
public class StateMachine : MonoBehaviour
|
|
{
|
|
[Tooltip("Set the initial state of this StateMachine")]
|
|
[SerializeField] private ScriptableObjects.TransitionTableSO _transitionTableSO = default;
|
|
|
|
#if UNITY_EDITOR
|
|
[Space]
|
|
[SerializeField]
|
|
internal Debugging.StateMachineDebugger _debugger = default;
|
|
#endif
|
|
|
|
private readonly Dictionary<Type, Component> _cachedComponents = new Dictionary<Type, Component>();
|
|
internal State _currentState;
|
|
|
|
private void Awake()
|
|
{
|
|
_currentState = _transitionTableSO.GetInitialState(this);
|
|
#if UNITY_EDITOR
|
|
_debugger.Awake(this);
|
|
#endif
|
|
}
|
|
|
|
#if UNITY_EDITOR
|
|
private void OnEnable()
|
|
{
|
|
UnityEditor.AssemblyReloadEvents.afterAssemblyReload += OnAfterAssemblyReload;
|
|
}
|
|
|
|
private void OnAfterAssemblyReload()
|
|
{
|
|
_currentState = _transitionTableSO.GetInitialState(this);
|
|
_debugger.Awake(this);
|
|
}
|
|
|
|
private void OnDisable()
|
|
{
|
|
UnityEditor.AssemblyReloadEvents.afterAssemblyReload -= OnAfterAssemblyReload;
|
|
}
|
|
#endif
|
|
|
|
private void Start()
|
|
{
|
|
_currentState.OnStateEnter();
|
|
}
|
|
|
|
public new bool TryGetComponent<T>(out T component) where T : Component
|
|
{
|
|
var type = typeof(T);
|
|
if (!_cachedComponents.TryGetValue(type, out var value))
|
|
{
|
|
if (base.TryGetComponent<T>(out component))
|
|
_cachedComponents.Add(type, component);
|
|
|
|
return component != null;
|
|
}
|
|
|
|
component = (T)value;
|
|
return true;
|
|
}
|
|
|
|
public T GetOrAddComponent<T>() where T : Component
|
|
{
|
|
if (!TryGetComponent<T>(out var component))
|
|
{
|
|
component = gameObject.AddComponent<T>();
|
|
_cachedComponents.Add(typeof(T), component);
|
|
}
|
|
|
|
return component;
|
|
}
|
|
|
|
public new T GetComponent<T>() where T : Component
|
|
{
|
|
return TryGetComponent(out T component)
|
|
? component : throw new InvalidOperationException($"{typeof(T).Name} not found in {name}.");
|
|
}
|
|
|
|
private void Update()
|
|
{
|
|
if (_currentState.TryGetTransition(out var transitionState))
|
|
Transition(transitionState);
|
|
|
|
_currentState.OnUpdate();
|
|
}
|
|
|
|
private void Transition(State transitionState)
|
|
{
|
|
_currentState.OnStateExit();
|
|
_currentState = transitionState;
|
|
_currentState.OnStateEnter();
|
|
}
|
|
}
|
|
}
|