Make it possible to scroll selectors in TeamTypes and AITriggerTypes windows

Author: Rampastring

src/TSMapEditor/UI/UIHelpers.cs

@@ -1,4 +1,6 @@
-using Rampastring.XNAUI.XNAControls;
+using Rampastring.XNAUI.Input;
+using Rampastring.XNAUI.XNAControls;
+using System.Collections.Generic;
 using TSMapEditor.UI.Controls;
 
 namespace TSMapEditor.UI
@@ -16,5 +18,38 @@ public static void AddSearchTipsBoxToControl(XNAControl control)
             var tooltip = new ToolTip(control.WindowManager, lblSearchTips);
             tooltip.Text = Translate("UIHelpers.AddSearchTipsBoxToControl.SearchTip", "Search Tips\r\n\r\nWith the text box activated:\r\n- Press ENTER to move to next match in list\r\n- Press ESC to clear search query");
         }
+
+        public static T GetScrollItem<T>(List<T> list, T current, Cursor cursor, bool allowSelectionIfNull)
+        {
+            if (current == null)
+            {
+                if (list.Count > 0 && allowSelectionIfNull)
+                {
+                    return current;
+                }
+
+                return default(T);
+            }
+
+            int index = list.IndexOf(current);
+
+            // Check for possible error condition
+            if (index < 0)
+            {
+                return default(T);
+            }
+
+            if (index > 0 && cursor.ScrollWheelValue > 0)
+            {
+                return list[index - 1];
+            }
+
+            if (index < list.Count - 1 && cursor.ScrollWheelValue < 0)
+            {
+                return list[index + 1];
+            }
+
+            return current;
+        }
     }
 }

src/TSMapEditor/UI/Windows/AITriggersWindow.cs

@@ -104,6 +104,9 @@ public override void Initialize()
             tbFilter = FindChild<EditorSuggestionTextBox>(nameof(tbFilter));
             tbFilter.TextChanged += TbFilter_TextChanged;
 
+            selPrimaryTeam.MouseScrolled += SelPrimaryTeam_MouseScrolled;
+            selSecondaryTeam.MouseScrolled += SelSecondaryTeam_MouseScrolled;
+
             FindChild<EditorButton>("btnNew").LeftClick += BtnNew_LeftClick;
             FindChild<EditorButton>("btnDelete").LeftClick += BtnDelete_LeftClick;
             FindChild<EditorButton>("btnClone").LeftClick += BtnClone_LeftClick;
@@ -140,6 +143,28 @@ public override void Initialize()
             lbAITriggers.SelectedIndexChanged += LbAITriggers_SelectedIndexChanged;
         }
 
+        private void SelSecondaryTeam_MouseScrolled(object sender, InputEventArgs e)
+        {
+            e.Handled = true;
+
+            if (editedAITrigger == null)
+                return;
+
+            editedAITrigger.SecondaryTeam = UIHelpers.GetScrollItem(map.TeamTypes, editedAITrigger.SecondaryTeam, Cursor, false);
+            EditAITrigger(editedAITrigger);
+        }
+
+        private void SelPrimaryTeam_MouseScrolled(object sender, InputEventArgs e)
+        {
+            e.Handled = true;
+
+            if (editedAITrigger == null)
+                return;
+
+            editedAITrigger.PrimaryTeam = UIHelpers.GetScrollItem(map.TeamTypes, editedAITrigger.PrimaryTeam, Cursor, false);
+            EditAITrigger(editedAITrigger);
+        }
+
         private void CloneForEasierDifficulties()
         {
             if (editedAITrigger == null)

src/TSMapEditor/UI/Windows/TeamTypesWindow.cs

@@ -141,6 +141,7 @@ public override void Initialize()
             var tagDarkeningPanel = DarkeningPanel.InitializeAndAddToParentControlWithChild(WindowManager, Parent, selectTagWindow);
             tagDarkeningPanel.Hidden += (s, e) => SelectionWindow_ApplyEffect(w => editedTeamType.Tag = w.SelectedObject, selectTagWindow);
 
+            selTaskForce.MouseScrolled += SelTaskForce_MouseScrolled;
             selTaskForce.LeftClick += (s, e) => 
             {
                 if (editedTeamType == null)
@@ -156,6 +157,7 @@ public override void Initialize()
                 }
             };
 
+            selScript.MouseScrolled += SelScript_MouseScrolled;
             selScript.LeftClick += (s, e) =>
             {
                 if (editedTeamType == null)
@@ -175,6 +177,7 @@ public override void Initialize()
             FindChild<EditorButton>("btnOpenScript").LeftClick += (s, e) => OpenScript();
             FindChild<EditorButton>("btnOpenTag").LeftClick += (s, e) => OpenTag();
 
+            selTag.MouseScrolled += SelTag_MouseScrolled;
             selTag.LeftClick += (s, e) => { if (editedTeamType != null) selectTagWindow.Open(editedTeamType.Tag); };
 
             var sortContextMenu = new EditorContextMenu(WindowManager);
@@ -205,6 +208,39 @@ public override void Initialize()
             map.TeamTypesChanged += Map_TeamTypesChanged;
         }
 
+        private void SelTag_MouseScrolled(object sender, InputEventArgs e)
+        {
+            e.Handled = true;
+
+            if (editedTeamType == null)
+                return;
+
+            editedTeamType.Tag = UIHelpers.GetScrollItem(map.Tags, editedTeamType.Tag, Cursor, false);
+            EditTeamType(editedTeamType);
+        }
+
+        private void SelScript_MouseScrolled(object sender, InputEventArgs e)
+        {
+            e.Handled = true;
+
+            if (editedTeamType == null)
+                return;
+
+            editedTeamType.Script = UIHelpers.GetScrollItem(map.Scripts, editedTeamType.Script, Cursor, true);
+            EditTeamType(editedTeamType);
+        }
+
+        private void SelTaskForce_MouseScrolled(object sender, InputEventArgs e)
+        {
+            e.Handled = true;
+
+            if (editedTeamType == null)
+                return;
+
+            editedTeamType.TaskForce = UIHelpers.GetScrollItem(map.TaskForces, editedTeamType.TaskForce, Cursor, true);
+            EditTeamType(editedTeamType);
+        }
+
         private void Map_TeamTypesChanged(object sender, EventArgs e)
         {
             if (Visible)