Refactor feedback sample playback logic

Author: nekodex

osu.Game/Graphics/UserInterface/OsuTextBox.cs

@@ -1,6 +1,8 @@
 // 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 System;
+using System.Collections.Generic;
 using System.Linq;
 using osu.Framework.Allocation;
 using osu.Framework.Audio;
@@ -40,30 +42,26 @@ public class OsuTextBox : BasicTextBox
             Margin = new MarginPadding { Left = 2 },
         };
 
-        private readonly Sample?[] textAddedSamples = new Sample[4];
-        private Sample? capsTextAddedSample;
-        private Sample? textRemovedSample;
-        private Sample? textCommittedSample;
-        private Sample? caretMovedSample;
-
-        private Sample? selectCharSample;
-        private Sample? selectWordSample;
-        private Sample? selectAllSample;
-        private Sample? deselectSample;
-
         private OsuCaret? caret;
 
         private bool selectionStarted;
         private double sampleLastPlaybackTime;
 
-        private enum SelectionSampleType
+        private enum FeedbackSampleType
         {
-            Character,
-            Word,
-            All,
+            TextAdd,
+            TextAddCaps,
+            TextRemove,
+            TextConfirm,
+            CaretMove,
+            SelectCharacter,
+            SelectWord,
+            SelectAll,
             Deselect
         }
 
+        private Dictionary<FeedbackSampleType, Sample?[]> sampleMap = new Dictionary<FeedbackSampleType, Sample?[]>();
+
         public OsuTextBox()
         {
             Height = 40;
@@ -87,18 +85,22 @@ private void load(OverlayColourProvider? colourProvider, OsuColour colour, Audio
 
             Placeholder.Colour = colourProvider?.Foreground1 ?? new Color4(180, 180, 180, 255);
 
+            var textAddedSamples = new Sample?[4];
             for (int i = 0; i < textAddedSamples.Length; i++)
                 textAddedSamples[i] = audio.Samples.Get($@"Keyboard/key-press-{1 + i}");
 
-            capsTextAddedSample = audio.Samples.Get(@"Keyboard/key-caps");
-            textRemovedSample = audio.Samples.Get(@"Keyboard/key-delete");
-            textCommittedSample = audio.Samples.Get(@"Keyboard/key-confirm");
-            caretMovedSample = audio.Samples.Get(@"Keyboard/key-movement");
-
-            selectCharSample = audio.Samples.Get(@"Keyboard/select-char");
-            selectWordSample = audio.Samples.Get(@"Keyboard/select-word");
-            selectAllSample = audio.Samples.Get(@"Keyboard/select-all");
-            deselectSample = audio.Samples.Get(@"Keyboard/deselect");
+            sampleMap = new Dictionary<FeedbackSampleType, Sample?[]>
+            {
+                { FeedbackSampleType.TextAdd, textAddedSamples },
+                { FeedbackSampleType.TextAddCaps, new[] { audio.Samples.Get(@"Keyboard/key-caps") } },
+                { FeedbackSampleType.TextRemove, new[] { audio.Samples.Get(@"Keyboard/key-delete") } },
+                { FeedbackSampleType.TextConfirm, new[] { audio.Samples.Get(@"Keyboard/key-confirm") } },
+                { FeedbackSampleType.CaretMove, new[] { audio.Samples.Get(@"Keyboard/key-movement") } },
+                { FeedbackSampleType.SelectCharacter, new[] { audio.Samples.Get(@"Keyboard/select-char") } },
+                { FeedbackSampleType.SelectWord, new[] { audio.Samples.Get(@"Keyboard/select-word") } },
+                { FeedbackSampleType.SelectAll, new[] { audio.Samples.Get(@"Keyboard/select-all") } },
+                { FeedbackSampleType.Deselect, new[] { audio.Samples.Get(@"Keyboard/deselect") } }
+            };
         }
 
         private Color4 selectionColour;
@@ -110,31 +112,31 @@ protected override void OnUserTextAdded(string added)
             base.OnUserTextAdded(added);
 
             if (added.Any(char.IsUpper) && AllowUniqueCharacterSamples)
-                capsTextAddedSample?.Play();
+                playSample(FeedbackSampleType.TextAddCaps);
             else
-                playTextAddedSample();
+                playSample(FeedbackSampleType.TextAdd);
         }
 
         protected override void OnUserTextRemoved(string removed)
         {
             base.OnUserTextRemoved(removed);
 
-            textRemovedSample?.Play();
+            playSample(FeedbackSampleType.TextRemove);
         }
 
         protected override void OnTextCommitted(bool textChanged)
         {
             base.OnTextCommitted(textChanged);
 
-            textCommittedSample?.Play();
+            playSample(FeedbackSampleType.TextConfirm);
         }
 
         protected override void OnCaretMoved(bool selecting)
         {
             base.OnCaretMoved(selecting);
 
             if (!selecting)
-                caretMovedSample?.Play();
+                playSample(FeedbackSampleType.CaretMove);
         }
 
         protected override void OnTextSelectionChanged(TextSelectionType selectionType)
@@ -144,15 +146,15 @@ protected override void OnTextSelectionChanged(TextSelectionType selectionType)
             switch (selectionType)
             {
                 case TextSelectionType.Character:
-                    playSelectSample(SelectionSampleType.Character);
+                    playSample(FeedbackSampleType.SelectCharacter);
                     break;
 
                 case TextSelectionType.Word:
-                    playSelectSample(selectionStarted ? SelectionSampleType.Character : SelectionSampleType.Word);
+                    playSample(selectionStarted ? FeedbackSampleType.SelectCharacter : FeedbackSampleType.SelectWord);
                     break;
 
                 case TextSelectionType.All:
-                    playSelectSample(SelectionSampleType.All);
+                    playSample(FeedbackSampleType.SelectAll);
                     break;
             }
 
@@ -165,7 +167,7 @@ protected override void OnTextDeselected()
 
             if (!selectionStarted) return;
 
-            playSelectSample(SelectionSampleType.Deselect);
+            playSample(FeedbackSampleType.Deselect);
 
             selectionStarted = false;
         }
@@ -184,36 +186,36 @@ protected override void OnImeComposition(string newComposition, int removedTextL
 
                     case 1:
                         // composition probably ended by pressing backspace, or was cancelled.
-                        textRemovedSample?.Play();
+                        playSample(FeedbackSampleType.TextRemove);
                         return;
 
                     default:
                         // longer text removed, composition ended because it was cancelled.
                         // could be a different sample if desired.
-                        textRemovedSample?.Play();
+                        playSample(FeedbackSampleType.TextRemove);
                         return;
                 }
             }
 
             if (addedTextLength > 0)
             {
                 // some text was added, probably due to typing new text or by changing the candidate.
-                playTextAddedSample();
+                playSample(FeedbackSampleType.TextAdd);
                 return;
             }
 
             if (removedTextLength > 0)
             {
                 // text was probably removed by backspacing.
                 // it's also possible that a candidate that only removed text was changed to.
-                textRemovedSample?.Play();
+                playSample(FeedbackSampleType.TextRemove);
                 return;
             }
 
             if (caretMoved)
             {
                 // only the caret/selection was moved.
-                caretMovedSample?.Play();
+                playSample(FeedbackSampleType.CaretMove);
             }
         }
 
@@ -224,13 +226,13 @@ protected override void OnImeResult(string result, bool successful)
             if (successful)
             {
                 // composition was successfully completed, usually by pressing the enter key.
-                textCommittedSample?.Play();
+                playSample(FeedbackSampleType.TextConfirm);
             }
             else
             {
                 // composition was prematurely ended, eg. by clicking inside the textbox.
                 // could be a different sample if desired.
-                textCommittedSample?.Play();
+                playSample(FeedbackSampleType.TextConfirm);
             }
         }
 
@@ -259,43 +261,35 @@ protected override void OnFocusLost(FocusLostEvent e)
             SelectionColour = SelectionColour,
         };
 
-        private void playSelectSample(SelectionSampleType selectionType)
+        private SampleChannel? getSampleChannel(FeedbackSampleType feedbackSampleType)
         {
-            if (Time.Current < sampleLastPlaybackTime + 15) return;
+            var samples = sampleMap[feedbackSampleType];
 
-            SampleChannel? channel;
-            double pitch = 0.98 + RNG.NextDouble(0.04);
-
-            switch (selectionType)
-            {
-                case SelectionSampleType.All:
-                    channel = selectAllSample?.GetChannel();
-                    break;
+            if (samples == null || samples.Length == 0)
+                return null;
 
-                case SelectionSampleType.Word:
-                    channel = selectWordSample?.GetChannel();
-                    break;
+            return samples[RNG.Next(0, samples.Length)]?.GetChannel();
+        }
 
-                case SelectionSampleType.Deselect:
-                    channel = deselectSample?.GetChannel();
-                    break;
+        private void playSample(FeedbackSampleType feedbackSample)
+        {
+            if (Time.Current < sampleLastPlaybackTime + 15) return;
 
-                default:
-                    channel = selectCharSample?.GetChannel();
-                    pitch += (SelectedText.Length / (double)Text.Length) * 0.15f;
-                    break;
-            }
+            SampleChannel? channel = getSampleChannel(feedbackSample);
 
             if (channel == null) return;
 
+            double pitch = 0.98 + RNG.NextDouble(0.04);
+
+            if (feedbackSample == FeedbackSampleType.SelectCharacter)
+                pitch += ((double)SelectedText.Length / Math.Max(1, Text.Length)) * 0.15f;
+
             channel.Frequency.Value = pitch;
             channel.Play();
 
             sampleLastPlaybackTime = Time.Current;
         }
 
-        private void playTextAddedSample() => textAddedSamples[RNG.Next(0, textAddedSamples.Length)]?.Play();
-
         private class OsuCaret : Caret
         {
             private const float caret_move_time = 60;