summaryrefslogtreecommitdiff
path: root/src/SMAPI/Framework/Input/KeyboardStateBuilder.cs
blob: f95a28bf1373c34637e6a19bdeaa676cf5b42962 (plain)
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
70
71
72
73
74
75
76
77
78
using System.Collections.Generic;
using System.Linq;
using Microsoft.Xna.Framework.Input;

namespace StardewModdingAPI.Framework.Input
{
    /// <summary>Manages keyboard state.</summary>
    internal class KeyboardStateBuilder : IInputStateBuilder<KeyboardStateBuilder, KeyboardState>
    {
        /*********
        ** Fields
        *********/
        /// <summary>The underlying keyboard state.</summary>
        private KeyboardState? State;

        /// <summary>The pressed buttons.</summary>
        private readonly HashSet<Keys> PressedButtons = new HashSet<Keys>();


        /*********
        ** Public methods
        *********/
        /// <summary>Construct an instance.</summary>
        /// <param name="state">The initial state, or <c>null</c> to get the latest state.</param>
        public KeyboardStateBuilder(KeyboardState? state = null)
        {
            this.Reset(state);
        }

        /// <summary>Reset the tracked state.</summary>
        /// <param name="state">The state from which to reset, or <c>null</c> to get the latest state.</param>
        public KeyboardStateBuilder Reset(KeyboardState? state = null)
        {
            this.State = state ??= Keyboard.GetState();

            this.PressedButtons.Clear();
            foreach (var button in state.Value.GetPressedKeys())
                this.PressedButtons.Add(button);

            return this;
        }

        /// <summary>Override the states for a set of buttons.</summary>
        /// <param name="overrides">The button state overrides.</param>
        public KeyboardStateBuilder OverrideButtons(IDictionary<SButton, SButtonState> overrides)
        {
            foreach (var pair in overrides)
            {
                if (pair.Key.TryGetKeyboard(out Keys key))
                {
                    this.State = null;

                    if (pair.Value.IsDown())
                        this.PressedButtons.Add(key);
                    else
                        this.PressedButtons.Remove(key);
                }
            }

            return this;
        }

        /// <summary>Get the currently pressed buttons.</summary>
        public IEnumerable<SButton> GetPressedButtons()
        {
            foreach (Keys key in this.PressedButtons)
                yield return key.ToSButton();
        }

        /// <summary>Get the equivalent state.</summary>
        public KeyboardState GetState()
        {
            return
                this.State
                ?? (this.State = new KeyboardState(this.PressedButtons.ToArray())).Value;
        }
    }
}