Maszyna stanów

Metoda SetState obiektu Beacon byłaby taka sama dla każdego obiektu ze stanami: wieżyczki, przeciwnika, drzwi. Ten kod może znajdować się we własnej klasie — maszynie stanów, która przechowuje bieżący stan i go zmienia. Każdy obiekt tworzy jedną maszynę i ją wywołuje, zamiast powtarzać kod:

public class StateMachine
{
    public IState Current { get; private set; }

    public void ChangeState(IState next)
    {
        Current?.Exit();
        Current = next;
        Current.Enter();
    }

    public void Tick()
    {
        Current?.Tick();
    }
}

public class Beacon : MonoBehaviour
{
    public Renderer lamp;
    private StateMachine machine = new StateMachine();
    private IState blink;
    private IState steady;

    void Start()
    {
        blink = new BlinkState(lamp);
        steady = new SteadyState();
        machine.ChangeState(steady);
    }

    void Update()
    {
        if (Keyboard.current.spaceKey.wasPressedThisFrame) machine.ChangeState(machine.Current == steady ? blink : steady);
        machine.Tick();
    }
}

Pierwsze wywołanie ChangeState nie ma stanu, z którego mogłoby wyjść: Current ma wartość null, a ?. pomija wywołanie Exit. W sekcji 1 było ostrzeżenie, że ?. nie wykrywa fałszywego null w Unity; dotyczy to zniszczonych obiektów Unity, a stany są zwykłymi obiektami C#, więc ?. jest tutaj bezpieczne. Właściwość Current ma private set, więc żaden skrypt nie może zmienić stanu bez przejścia przez ChangeState oraz jego Exit i Enter.

Kolejność wewnątrz ChangeState ma znaczenie zawsze, gdy dwa stany wpływają na ten sam element. Exit cofa zmiany wprowadzone przez stary stan, a następnie Enter przygotowuje nowy, więc ostatnie słowo należy do nowego stanu. Gdyby Enter stanu DarkState wyłączało lampę i uruchamiało się przed Exit stanu BlinkState, to Exit ponownie włączyłoby lampę w ciemnym stanie. StateMachine również jest zwykłą klasą: latarnia, wieżyczka i przeciwnik tworzą własne maszyny za pomocą new i wywołują ich Tick z poziomu Update.