osu-lazer/osu.Game/Screens/Play/KeyCounter.cs

// Copyright (c) ppy Pty Ltd <contact@ppy.sh>. Licensed under the MIT Licence.
// See the LICENCE file in the repository root for full licence text.

using osu.Framework.Bindables;
using osu.Framework.Graphics;
using osu.Framework.Graphics.Containers;
using osu.Framework.Input.Events;

namespace osu.Game.Screens.Play
{
    public abstract partial class KeyCounter : Container
    {
        public readonly Trigger CounterTrigger;

        protected Bindable<bool> IsCountingBindable = new BindableBool(true);

        protected Bindable<int> PressesCount = new BindableInt
        {
            MinValue = 0
        };

        public bool IsCounting
        {
            get => IsCountingBindable.Value;
            set => IsCountingBindable.Value = value;
        }

        public int CountPresses
        {
            get => PressesCount.Value;
            private set => PressesCount.Value = value;
        }

        protected Bindable<bool> IsLit = new BindableBool();

        public void Increment()
        {
            if (!IsCounting)
                return;

            CountPresses++;
        }

        public void Decrement()
        {
            if (!IsCounting)
                return;

            CountPresses--;
        }

        protected override void LoadComplete()
        {
            Add(CounterTrigger);
            base.LoadComplete();
        }

        protected override bool Handle(UIEvent e) => CounterTrigger.TriggerEvent(e);

        protected KeyCounter(Trigger trigger)
        {
            CounterTrigger = trigger;
            trigger.Target = this;
            Name = trigger.Name;
        }

        public abstract partial class Trigger : Component
        {
            private KeyCounter? target;

            public KeyCounter Target
            {
                set => target = value;
            }

            protected Trigger(string name)
            {
                Name = name;
            }

            protected void Lit(bool increment = true)
            {
                if (target == null) return;

                target.IsLit.Value = true;
                if (increment)
                    target.Increment();
            }

            protected void Unlit(bool preserve = true)
            {
                if (target == null) return;

                target.IsLit.Value = false;
                if (!preserve)
                    target.Decrement();
            }
        }
    }
}