מכונת מצבים

השיטה SetState של המגדלור תהיה זהה בכל אובייקט שיש לו מצבים: צריח, אויב, דלת. הקוד הזה יכול להיות במחלקה משלו, מכונת מצבים, שמחזיקה את המצב הנוכחי ומשנה אותו. כל אובייקט יוצר מכונה אחת וקורא לה, במקום לחזור על הקוד:

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

לקריאה הראשונה ל־ChangeState אין מצב לעזוב: Current הוא null, ו־?. מדלג על הקריאה ל־Exit. יחידה 1 הזהירה ש־?. לא מזהה את ה־null המדומה של Unity; זה קשור לאובייקטים של Unity שהושמדו, והמצבים הם אובייקטי C# רגילים, לכן ?. בטוח כאן. ל־Current יש private set, ולכן שום סקריפט לא יכול להחליף את המצב בלי לעבור דרך ChangeState וה־Exit וה־Enter שלה.

הסדר בתוך ChangeState חשוב בכל פעם ששני מצבים נוגעים באותו הדבר. Exit מבטל את מה שהמצב הישן שינה, ואז Enter מגדיר את המצב החדש, כך שלמצב החדש יש את המילה האחרונה. אם Enter של DarkState היה מכבה את המנורה ופועל לפני Exit של BlinkState, אותו Exit היה מדליק מחדש את המנורה במצב החשוך. גם StateMachine היא מחלקה רגילה: המגדלור, צריח ואויב יוצרים כל אחד מכונה משלו עם new וקוראים ל־Tick שלה מתוך Update שלהם.