CodeName.StateMachines/CapabilityMachineUtility.cs at main · TurnBasedTacticsGame/CodeName.StateMachines · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
using UnityEngine.Pool;

namespace CodeName.StateMachines
{
    public static class CapabilityMachineUtility
    {
        public static void Reset<TCapability>(this ICapabilityMachine<TCapability> capabilityMachine) where TCapability : ICapability
        {
            capabilityMachine.DisableAllCapabilities();
            capabilityMachine.EnableDefaultCapabilities();
        }

        public static void ResetOnStateChanged<TCapability, TState>(
            this ICapabilityMachine<TCapability> capabilityMachine,
            IStateMachine<TState> stateMachine)
            where TCapability : ICapability
            where TState : IState
        {
            stateMachine.StateChangeInProgress += (_, _) =>
            {
                capabilityMachine.Reset();
            };
        }

        public static void SetAllCapabilities<TCapability>(this ICapabilityMachine<TCapability> machine, bool isEnabled) where TCapability : ICapability
        {
            using var listHandle = ListPool<TCapability>.Get(out var capabilities);
            capabilities.AddRange(machine.EnabledCapabilities);

            foreach (var capability in capabilities)
            {
                machine.SetCapability(capability, isEnabled);
            }
        }

        public static void DisableAllCapabilities<TCapability>(this ICapabilityMachine<TCapability> machine) where TCapability : ICapability
        {
            machine.SetAllCapabilities(false);
        }

        public static void EnableAllCapabilities<TCapability>(this ICapabilityMachine<TCapability> machine) where TCapability : ICapability
        {
            machine.SetAllCapabilities(true);
        }

        public static void EnableDefaultCapabilities<TCapability>(this ICapabilityMachine<TCapability> machine) where TCapability : ICapability
        {
            using var listHandle = ListPool<TCapability>.Get(out var capabilities);
            capabilities.AddRange(machine.DefaultCapabilities);

            foreach (var capability in capabilities)
            {
                machine.EnableCapability(capability);
            }
        }

        public static void SetCapability<TCapability>(this ICapabilityMachine<TCapability> machine, TCapability capability, bool isEnabled) where TCapability : ICapability
        {
            if (isEnabled)
            {
                machine.EnableCapability(capability);
            }
            else
            {
                machine.DisableCapability(capability);
            }
        }
    }
}