Changeset 2952 for trunk/sources/HeuristicLab.Core.Views/3.3/TypeSelector.cs

Timestamp:

03/07/10 03:10:38 (14 years ago)

Author:

swagner

Message:

Implemented searching in TypeSelector (#894)

File:

• trunk/sources/HeuristicLab.Core.Views/3.3/TypeSelector.cs (modified) (7 diffs)

trunk/sources/HeuristicLab.Core.Views/3.3/TypeSelector.cs

@@ -26 +26 @@
 using HeuristicLab.Common;
 using HeuristicLab.PluginInfrastructure;
+using System.Collections.Generic;
 
 namespace HeuristicLab.Core.Views { 
   public partial class TypeSelector : UserControl {
+    protected List<TreeNode> treeNodes;
+    protected string currentSearchString;
+
     protected Type baseType;
     public Type BaseType {
@@ -66 +70 @@
     public TypeSelector() {
       InitializeComponent();
+      treeNodes = new List<TreeNode>();
+      currentSearchString = string.Empty;
       selectedType = null;
     }
@@ -78 +84 @@
         this.showGenericTypes = showGenericTypes;
 
+        TreeNode selectedNode = typesTreeView.SelectedNode;
         typesTreeView.Nodes.Clear();
+        treeNodes.Clear();
         imageList.Images.Clear();
         imageList.Images.Add(HeuristicLab.Common.Resources.VS2008ImageLibrary.Class);      // default icon
@@ -86 +94 @@
         imageList.Images.Add(HeuristicLab.Common.Resources.VS2008ImageLibrary.Template);   // generic types
 
-        TreeNode selectedNode = null;
         var plugins = from p in ApplicationManager.Manager.Plugins
                       orderby p.Name, p.Version ascending
@@ -126 +133 @@
               typeNode.SelectedImageIndex = typeNode.ImageIndex;
               typeNode.Tag = type;
-              if (type == SelectedType) selectedNode = typeNode;
               pluginNode.Nodes.Add(typeNode);
             }
           }
           if (pluginNode.Nodes.Count > 0)
-            typesTreeView.Nodes.Add(pluginNode);
-        }
+            treeNodes.Add(pluginNode);
+        }
+        foreach (TreeNode node in treeNodes)
+          typesTreeView.Nodes.Add((TreeNode)node.Clone());
+        RestoreSelectedNode(selectedNode);
+        Filter(searchTextBox.Text);
+      }
+    }
+
+    public virtual void Filter(string searchString) {
+      if (InvokeRequired)
+        Invoke(new Action<string>(Filter), searchString);
+      else {
+        searchString = searchString.ToLower();
+
+        typesTreeView.BeginUpdate();
+        if (!searchString.Contains(currentSearchString)) {
+          // expand search -> restore all tree nodes
+          TreeNode selectedNode = typesTreeView.SelectedNode;
+          typesTreeView.Nodes.Clear();
+          foreach (TreeNode node in treeNodes)
+            typesTreeView.Nodes.Add((TreeNode)node.Clone());
+          RestoreSelectedNode(selectedNode);
+        }
+
+        // remove nodes
+        int i = 0;
+        while (i < typesTreeView.Nodes.Count) {
+          int j = 0;
+          while (j < typesTreeView.Nodes[i].Nodes.Count) {
+            if (!typesTreeView.Nodes[i].Nodes[j].Text.ToLower().Contains(searchString))
+              typesTreeView.Nodes[i].Nodes[j].Remove();
+            else
+              j++;
+          }
+          if (typesTreeView.Nodes[i].Nodes.Count == 0)
+            typesTreeView.Nodes[i].Remove();
+          else
+            i++;
+        }
+        currentSearchString = searchString;
+
+        // if there is just one type node left, select by default
+        if (typesTreeView.Nodes.Count == 1) {
+          if (typesTreeView.Nodes[0].Nodes.Count == 1) {
+            typesTreeView.SelectedNode = typesTreeView.Nodes[0].Nodes[0];
+          }
+        }
+
         if (typesTreeView.Nodes.Count == 0) {
-          typesTreeView.Nodes.Add("No types of base type \"" + BaseType.GetPrettyName() + "\" found");
+          SelectedType = null;
           typesTreeView.Enabled = false;
-        }
-        if (selectedNode != null) {
-          typesTreeView.SelectedNode = selectedNode;
-          selectedNode.EnsureVisible();
         } else {
-          SelectedType = null;
-        }
+          SetTreeNodeVisibility();
+          typesTreeView.Enabled = true;
+        }
+        UpdateDescription();
+        typesTreeView.EndUpdate();
       }
     }
@@ -178 +230 @@
             descriptionTextBox.Text = description;
         }
-      }
+      } else if (typesTreeView.Nodes.Count == 0) {
+        descriptionTextBox.Text = "No types found";
+      }
+    }
+
+    protected virtual void searchTextBox_TextChanged(object sender, System.EventArgs e) {
+      Filter(searchTextBox.Text);
     }
 
     protected virtual void typesTreeView_AfterSelect(object sender, TreeViewEventArgs e) {
-      if (typesTreeView.SelectedNode != null) {
-        SelectedType = typesTreeView.SelectedNode.Tag as Type;
-      }
+      if (typesTreeView.SelectedNode == null) SelectedType = null;
+      else SelectedType = typesTreeView.SelectedNode.Tag as Type;
       UpdateDescription();
     }
@@ -202 +259 @@
       }
     }
+
+    protected virtual void typesTreeView_VisibleChanged(object sender, EventArgs e) {
+      if (Visible) SetTreeNodeVisibility();
+    }
+
+    #region Helpers
+    private void RestoreSelectedNode(TreeNode selectedNode) {
+      if (selectedNode != null) {
+        foreach (TreeNode node in typesTreeView.Nodes) {
+          if (node.Text.Equals(selectedNode.Text)) typesTreeView.SelectedNode = node;
+          foreach (TreeNode child in node.Nodes) {
+            if ((child.Text.Equals(selectedNode.Text)) && (child.Tag == selectedNode.Tag))
+              typesTreeView.SelectedNode = child;
+          }
+        }
+        if (typesTreeView.SelectedNode == null) SelectedType = null;
+      }
+    }
+    private void SetTreeNodeVisibility() {
+      TreeNode selectedNode = typesTreeView.SelectedNode;
+      if (string.IsNullOrEmpty(currentSearchString)) {
+        typesTreeView.CollapseAll();
+        if (selectedNode != null) typesTreeView.SelectedNode = selectedNode;
+      } else {
+        typesTreeView.ExpandAll();
+      }
+      if (selectedNode != null) selectedNode.EnsureVisible();
+    }
+    #endregion
   }
 }