מכונת מצבים

שיטת 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 שלהם.