search in dropdowns unlocked

BHoM · Mar 23, 2023 · e3e2e86 · e3e2e86
1 parent 32942b7
commit e3e2e86
Show file tree

Hide file tree

Showing 2 changed files with 53 additions and 1 deletion.
diff --git a/BHoM_UI/CallerTemplates/MultiChoiceCaller.cs b/BHoM_UI/CallerTemplates/MultiChoiceCaller.cs
@@ -31,11 +31,20 @@
 using System.Windows.Forms;
 using BH.oM.Base;
 using System.Collections;
+using BH.UI.Base.Menus;
+using BH.Engine.Base;
 
 namespace BH.UI.Base
 {
     public abstract class MultiChoiceCaller : Caller
     {
+        /*************************************/
+        /**** Public Events               ****/
+        /*************************************/
+
+        public event EventHandler<int> ValueSelected;
+
+
         /*************************************/
         /**** Properties                  ****/
         /*************************************/
@@ -84,6 +93,48 @@ public override object Run(List<object> inputs)
         public abstract List<string> GetChoiceNames();
 
         /*************************************/
+
+        public override void AddToMenu(object menu)
+        {
+            base.AddToMenu(menu);
+
+            if (SelectedItem != null)
+            {
+                if (m_EnumSearchMenu == null)
+                    SetEnumSearchMenu();
+
+                m_EnumSearchMenu.FillMenu(menu);
+            }
+        }
+
+
+        /*************************************/
+        /**** Private Methods             ****/
+        /*************************************/
+
+        private void SetEnumSearchMenu()
+        {
+            List<SearchItem> items = Choices.Zip(GetChoiceNames(), (x, y) => new SearchItem { Item = x, Text = y }).ToList();
+            ItemSelectorMenu_WinForm enumSearchMenu = new ItemSelectorMenu_WinForm(items, null);
+            enumSearchMenu.ItemSelected += EnumValueSelected;
+            m_EnumSearchMenu = enumSearchMenu;
+        }
+
+        /*************************************/
+
+        private void EnumValueSelected(object sender, object e)
+        {
+            ValueSelected?.Invoke(this, Choices.IndexOf(e));
+        }
+
+
+        /*************************************/
+        /**** Private Fields              ****/
+        /*************************************/
+
+        private IItemSelectorMenu m_EnumSearchMenu = null;
+
+        /*************************************/
     }
 }
 

diff --git a/BHoM_UI/Menus/ItemSelectorMenu_WinForm.cs b/BHoM_UI/Menus/ItemSelectorMenu_WinForm.cs
@@ -49,7 +49,8 @@ public ItemSelectorMenu_WinForm(List<SearchItem> itemList, Tree<object> itemTree
 
         protected override void AddTree(ToolStripDropDown menu, Tree<object> itemTree)
         {
-            AppendMenuTree(itemTree, menu);
+            if (itemTree != null)
+                AppendMenuTree(itemTree, menu);
         }
 
         /*************************************/