Changeset 3407 for trunk/sources/HeuristicLab.Core.Views/3.3/OperatorTreeView.cs

Timestamp:

04/19/10 03:28:16 (14 years ago)

Author:

swagner

Message:

Restricted types of child operators in MultiOperator (#979)

File:

• trunk/sources/HeuristicLab.Core.Views/3.3/OperatorTreeView.cs (modified) (16 diffs)

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

@@ -24 +24 @@
 using System.ComponentModel;
 using System.Drawing;
+using System.Linq;
 using System.Windows.Forms;
 using HeuristicLab.Collections;
@@ -35 +36 @@
   [Content(typeof(IOperator), false)]
   public sealed partial class OperatorTreeView : ItemView {
-    private Dictionary<IValueParameter<IOperator>, List<TreeNode>> opParamNodeTable;
+    private Dictionary<IValueParameter, List<TreeNode>> opParamNodeTable;
     private Dictionary<IOperator, List<TreeNode>> operatorNodeTable;
     private Dictionary<IKeyedItemCollection<string, IParameter>, IOperator> parametersOperatorTable;
@@ -66 +67 @@
       InitializeComponent();
       graphTreeView.Sorted = true;
-      opParamNodeTable = new Dictionary<IValueParameter<IOperator>, List<TreeNode>>();
+      opParamNodeTable = new Dictionary<IValueParameter, List<TreeNode>>();
       operatorNodeTable = new Dictionary<IOperator, List<TreeNode>>();
       parametersOperatorTable = new Dictionary<IKeyedItemCollection<string, IParameter>, IOperator>();
@@ -116 +117 @@
 
     #region TreeNode Management
-    private TreeNode CreateTreeNode(IValueParameter<IOperator> opParam) {
+    private TreeNode CreateTreeNode(IValueParameter opParam) {
       TreeNode node = new TreeNode();
       node.Text = opParam.Name + ": ";
@@ -127 +128 @@
       opParamNodeTable[opParam].Add(node);
 
-      FillTreeNode(node, opParam.Value);
+      FillTreeNode(node, (IOperator)opParam.Value);
       return node;
     }
@@ -162 +163 @@
         else node.ForeColor = graphTreeView.ForeColor;
 
-        foreach (IParameter param in op.Parameters) {
-          if (param is IValueParameter<IOperator>)
+        foreach (IValueParameter param in op.Parameters.OfType<IValueParameter>()) {
+          if (typeof(IOperator).IsAssignableFrom(param.DataType))
             node.Nodes.Add(new TreeNode());
         }
@@ -199 +200 @@
       ClearTreeNode(node);
 
-      IValueParameter<IOperator> opParam = GetOperatorParameterTag(node);
+      IValueParameter opParam = GetOperatorParameterTag(node);
       if (opParam != null) {
         opParamNodeTable[opParam].Remove(node);
@@ -211 +212 @@
     }
     private void AddParameterNodes(IOperator op, IEnumerable<IParameter> parameters) {
-      foreach (IParameter param in parameters) {
-        IValueParameter<IOperator> opParam = param as IValueParameter<IOperator>;
-        if (opParam != null) {
+      foreach (IValueParameter param in parameters.OfType<IValueParameter>()) {
+        if (typeof(IOperator).IsAssignableFrom(param.DataType)) {
           foreach (TreeNode node in operatorNodeTable[op])
-            node.Nodes.Add(CreateTreeNode(opParam));
+            node.Nodes.Add(CreateTreeNode(param));
         }
       }
     }
     private void RemoveParameterNodes(IEnumerable<IParameter> parameters) {
-      foreach (IParameter param in parameters) {
-        IValueParameter<IOperator> opParam = param as IValueParameter<IOperator>;
-        if (opParam != null) {
-          while (opParamNodeTable.ContainsKey(opParam))
-            RemoveTreeNode(opParamNodeTable[opParam][0]);
+      foreach (IValueParameter param in parameters.OfType<IValueParameter>()) {
+        if (typeof(IOperator).IsAssignableFrom(param.DataType)) {
+          while (opParamNodeTable.ContainsKey(param))
+            RemoveTreeNode(opParamNodeTable[param][0]);
         }
       }
@@ -235 +234 @@
         Invoke(new EventHandler(opParam_ValueChanged), sender, e);
       else {
-        IValueParameter<IOperator> opParam = (IValueParameter<IOperator>)sender;
+        IValueParameter opParam = (IValueParameter)sender;
         foreach (TreeNode node in opParamNodeTable[opParam].ToArray())
           ClearTreeNode(node);
         foreach (TreeNode node in opParamNodeTable[opParam]) {
           node.Text = opParam.Name + ": ";
-          FillTreeNode(node, opParam.Value);
+          FillTreeNode(node, (IOperator)opParam.Value);
         }
       }
@@ -265 +264 @@
         IOperator op = (IOperator)sender;
         foreach (TreeNode node in operatorNodeTable[op]) {
-          IValueParameter<IOperator> opParam = GetOperatorParameterTag(node);
+          IValueParameter opParam = GetOperatorParameterTag(node);
           if (opParam == null)
             node.Text = op.Name + " (" + op.ItemName + ")";
@@ -330 +329 @@
         node.Nodes.Clear();
         IOperator op = GetOperatorTag(node);
-        foreach (IParameter param in op.Parameters) {
-          IValueParameter<IOperator> opParam = param as IValueParameter<IOperator>;
-          if (opParam != null) node.Nodes.Add(CreateTreeNode(opParam));
+        foreach (IValueParameter param in op.Parameters.OfType<IValueParameter>()) {
+          if (typeof(IOperator).IsAssignableFrom(param.DataType)) node.Nodes.Add(CreateTreeNode(param));
         }
       }
@@ -343 +341 @@
     private void graphTreeView_KeyDown(object sender, KeyEventArgs e) {
       if (!ReadOnly && (e.KeyCode == Keys.Delete) && (graphTreeView.SelectedNode != null)) {
-        IValueParameter<IOperator> opParam = GetOperatorParameterTag(graphTreeView.SelectedNode);
+        IValueParameter opParam = GetOperatorParameterTag(graphTreeView.SelectedNode);
         if (opParam != null) opParam.Value = null;
       }
@@ -371 +369 @@
     private void graphTreeView_ItemDrag(object sender, ItemDragEventArgs e) {
       TreeNode node = (TreeNode)e.Item;
-      IValueParameter<IOperator> opParam = GetOperatorParameterTag(node);
+      IValueParameter opParam = GetOperatorParameterTag(node);
       IOperator op = GetOperatorTag(node);
       DataObject data = new DataObject();
@@ -404 +402 @@
         if ((e.Effect & DragDropEffects.Copy) == DragDropEffects.Copy) op = (IOperator)op.Clone();
         TreeNode node = graphTreeView.GetNodeAt(graphTreeView.PointToClient(new Point(e.X, e.Y)));
-        IValueParameter<IOperator> opParam = GetOperatorParameterTag(node);
+        IValueParameter opParam = GetOperatorParameterTag(node);
         opParam.Value = op;
       }
@@ -436 +434 @@
     }
 
-    private IValueParameter<IOperator> GetOperatorParameterTag(TreeNode node) {
+    private IValueParameter GetOperatorParameterTag(TreeNode node) {
       if (node.Tag != null)
-        return ((Tuple<IValueParameter<IOperator>, IOperator>)node.Tag).Item1;
+        return ((Tuple<IValueParameter, IOperator>)node.Tag).Item1;
       else
         return null;
     }
-    private void SetOperatorParameterTag(TreeNode node, IValueParameter<IOperator> opParam) {
+    private void SetOperatorParameterTag(TreeNode node, IValueParameter opParam) {
       if (node.Tag == null)
-        node.Tag = new Tuple<IValueParameter<IOperator>, IOperator>(opParam, null);
+        node.Tag = new Tuple<IValueParameter, IOperator>(opParam, null);
       else
-        ((Tuple<IValueParameter<IOperator>, IOperator>)node.Tag).Item1 = opParam;
+        ((Tuple<IValueParameter, IOperator>)node.Tag).Item1 = opParam;
     }
     private IOperator GetOperatorTag(TreeNode node) {
       if (node.Tag != null)
-        return ((Tuple<IValueParameter<IOperator>, IOperator>)node.Tag).Item2;
+        return ((Tuple<IValueParameter, IOperator>)node.Tag).Item2;
       else
         return null;
@@ -456 +454 @@
     private void SetOperatorTag(TreeNode node, IOperator op) {
       if (node.Tag == null)
-        node.Tag = new Tuple<IValueParameter<IOperator>, IOperator>(null, op);
+        node.Tag = new Tuple<IValueParameter, IOperator>(null, op);
       else
-        ((Tuple<IValueParameter<IOperator>, IOperator>)node.Tag).Item2 = op;
+        ((Tuple<IValueParameter, IOperator>)node.Tag).Item2 = op;
     }