From b014bfea3e96f5195ef0822db094866d0d7aedc0 Mon Sep 17 00:00:00 2001
From: Salman Alshamrani <frenzibyte@gmail.com>
Date: Sun, 17 Nov 2024 09:56:19 -0500
Subject: [PATCH] Fix key counter not updating activation state on initial load

---
 osu.Game/Screens/Play/HUD/InputTrigger.cs | 7 +++++++
 osu.Game/Screens/Play/HUD/KeyCounter.cs   | 8 ++++++++
 2 files changed, 15 insertions(+)

diff --git a/osu.Game/Screens/Play/HUD/InputTrigger.cs b/osu.Game/Screens/Play/HUD/InputTrigger.cs
index edc61ec142..72dade25f6 100644
--- a/osu.Game/Screens/Play/HUD/InputTrigger.cs
+++ b/osu.Game/Screens/Play/HUD/InputTrigger.cs
@@ -34,6 +34,11 @@ public abstract partial class InputTrigger : Component
         /// </summary>
         public IBindable<int> ActivationCount => activationCount;
 
+        /// <summary>
+        /// Whether this <see cref="InputTrigger"/> is currently active.
+        /// </summary>
+        public bool IsActive { get; private set; }
+
         /// <summary>
         /// Whether any activation or deactivation of this <see cref="InputTrigger"/> impacts its <see cref="ActivationCount"/>
         /// </summary>
@@ -49,6 +54,7 @@ protected void Activate(bool forwardPlayback = true)
             if (forwardPlayback && isCounting.Value)
                 activationCount.Value++;
 
+            IsActive = true;
             OnActivate?.Invoke(forwardPlayback);
         }
 
@@ -57,6 +63,7 @@ protected void Deactivate(bool forwardPlayback = true)
             if (!forwardPlayback && isCounting.Value)
                 activationCount.Value--;
 
+            IsActive = false;
             OnDeactivate?.Invoke(forwardPlayback);
         }
     }
diff --git a/osu.Game/Screens/Play/HUD/KeyCounter.cs b/osu.Game/Screens/Play/HUD/KeyCounter.cs
index 66f9dfd6f2..b506694044 100644
--- a/osu.Game/Screens/Play/HUD/KeyCounter.cs
+++ b/osu.Game/Screens/Play/HUD/KeyCounter.cs
@@ -36,6 +36,14 @@ protected KeyCounter(InputTrigger trigger)
             Trigger.OnDeactivate += Deactivate;
         }
 
+        protected override void LoadComplete()
+        {
+            base.LoadComplete();
+
+            if (Trigger.IsActive)
+                Activate();
+        }
+
         protected virtual void Activate(bool forwardPlayback = true)
         {
             isActive.Value = true;