source: trunk/sources/HeuristicLab.Optimization.Views/3.3/ExperimentTreeView.cs @ 6517

Last change on this file since 6517 was 6517, checked in by mkommend, 8 years ago

#1555: Implemented context menu to collapse & expand nodes.

File size: 33.6 KB
Line 
1#region License Information
2/* HeuristicLab
3 * Copyright (C) 2002-2011 Heuristic and Evolutionary Algorithms Laboratory (HEAL)
4 *
5 * This file is part of HeuristicLab.
6 *
7 * HeuristicLab is free software: you can redistribute it and/or modify
8 * it under the terms of the GNU General Public License as published by
9 * the Free Software Foundation, either version 3 of the License, or
10 * (at your option) any later version.
11 *
12 * HeuristicLab is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15 * GNU General Public License for more details.
16 *
17 * You should have received a copy of the GNU General Public License
18 * along with HeuristicLab. If not, see <http://www.gnu.org/licenses/>.
19 */
20#endregion
21
22using System;
23using System.Collections;
24using System.Collections.Generic;
25using System.Drawing;
26using System.Linq;
27using System.Windows.Forms;
28using HeuristicLab.Collections;
29using HeuristicLab.Common;
30using HeuristicLab.Core.Views;
31using HeuristicLab.MainForm;
32using HeuristicLab.PluginInfrastructure;
33
34namespace HeuristicLab.Optimization.Views {
35  public sealed partial class ExperimentTreeView : ItemView {
36    private TypeSelectorDialog typeSelectorDialog;
37    private Dictionary<IOptimizer, List<TreeNode>> optimizerTreeViewMapping;
38
39    public ExperimentTreeView() {
40      InitializeComponent();
41      optimizerTreeViewMapping = new Dictionary<IOptimizer, List<TreeNode>>();
42    }
43
44    protected override void Dispose(bool disposing) {
45      if (disposing) {
46        if (typeSelectorDialog != null) typeSelectorDialog.Dispose();
47        if (components != null) components.Dispose();
48      }
49      base.Dispose(disposing);
50    }
51
52    #region necessary code to handle dock correctly regarding the expanded nodes
53    bool[] expandendedState;
54    protected override void OnHandleCreated(EventArgs e) {
55      base.OnHandleCreated(e);
56      if (expandendedState == null) return;
57      var nodes = IterateTreeNodes().ToList();
58      for (int i = 0; i < nodes.Count; i++)
59        if (expandendedState[i]) nodes[i].Expand();
60    }
61    protected override void OnHandleDestroyed(EventArgs e) {
62      base.OnHandleDestroyed(e);
63      var nodes = IterateTreeNodes().ToList();
64      expandendedState = new bool[nodes.Count];
65      for (int i = 0; i < nodes.Count; i++)
66        expandendedState[i] = nodes[i].IsExpanded;
67    }
68    #endregion
69
70    public new Experiment Content {
71      get { return (Experiment)base.Content; }
72      set { base.Content = value; }
73    }
74
75    #region events registration
76    protected override void RegisterContentEvents() {
77      base.RegisterContentEvents();
78      Content.ExecutionStateChanged += new EventHandler(Content_ExecutionStateChanged);
79      Content.Optimizers.ItemsAdded += new CollectionItemsChangedEventHandler<IndexedItem<IOptimizer>>(Optimizers_ItemsAdded);
80      Content.Optimizers.ItemsMoved += new CollectionItemsChangedEventHandler<IndexedItem<IOptimizer>>(Optimizers_ItemsMoved);
81      Content.Optimizers.ItemsRemoved += new CollectionItemsChangedEventHandler<IndexedItem<IOptimizer>>(Optimizers_ItemsRemoved);
82      Content.Optimizers.ItemsReplaced += new CollectionItemsChangedEventHandler<IndexedItem<IOptimizer>>(Optimizers_ItemsReplaced);
83      Content.Optimizers.CollectionReset += new CollectionItemsChangedEventHandler<IndexedItem<IOptimizer>>(Optimizers_CollectionReset);
84    }
85
86    protected override void DeregisterContentEvents() {
87      Content.ExecutionStateChanged -= new EventHandler(Content_ExecutionStateChanged);
88      Content.Optimizers.ItemsAdded -= new CollectionItemsChangedEventHandler<IndexedItem<IOptimizer>>(Optimizers_ItemsAdded);
89      Content.Optimizers.ItemsMoved -= new CollectionItemsChangedEventHandler<IndexedItem<IOptimizer>>(Optimizers_ItemsMoved);
90      Content.Optimizers.ItemsRemoved -= new CollectionItemsChangedEventHandler<IndexedItem<IOptimizer>>(Optimizers_ItemsRemoved);
91      Content.Optimizers.ItemsReplaced -= new CollectionItemsChangedEventHandler<IndexedItem<IOptimizer>>(Optimizers_ItemsReplaced);
92      Content.Optimizers.CollectionReset -= new CollectionItemsChangedEventHandler<IndexedItem<IOptimizer>>(Optimizers_CollectionReset);
93      foreach (var optimizer in optimizerTreeViewMapping.Keys)
94        DeregisterOptimizerEvents(optimizer);
95      base.DeregisterContentEvents();
96    }
97
98    private void RegisterOptimizerEvents(IOptimizer optimizer) {
99      optimizer.ToStringChanged += new EventHandler(optimizer_ToStringChanged);
100      optimizer.ExecutionStateChanged += new EventHandler(optimizer_ExecutionStateChanged);
101
102      var batchRun = optimizer as BatchRun;
103      var experiment = optimizer as Experiment;
104      if (batchRun != null) {
105        batchRun.OptimizerChanged += new EventHandler(batchRun_OptimizerChanged);
106      }
107      if (experiment != null) {
108        experiment.Optimizers.ItemsAdded += new CollectionItemsChangedEventHandler<IndexedItem<IOptimizer>>(Optimizers_ItemsAdded);
109        experiment.Optimizers.ItemsMoved += new CollectionItemsChangedEventHandler<IndexedItem<IOptimizer>>(Optimizers_ItemsMoved);
110        experiment.Optimizers.ItemsRemoved += new CollectionItemsChangedEventHandler<IndexedItem<IOptimizer>>(Optimizers_ItemsRemoved);
111        experiment.Optimizers.ItemsReplaced += new CollectionItemsChangedEventHandler<IndexedItem<IOptimizer>>(Optimizers_ItemsReplaced);
112        experiment.Optimizers.CollectionReset += new CollectionItemsChangedEventHandler<IndexedItem<IOptimizer>>(Optimizers_CollectionReset);
113      }
114    }
115
116    private void DeregisterOptimizerEvents(IOptimizer optimizer) {
117      optimizer.ToStringChanged -= new EventHandler(optimizer_ToStringChanged);
118      optimizer.ExecutionStateChanged -= new EventHandler(optimizer_ExecutionStateChanged);
119
120      var batchRun = optimizer as BatchRun;
121      var experiment = optimizer as Experiment;
122      if (batchRun != null) {
123        batchRun.OptimizerChanged -= new EventHandler(batchRun_OptimizerChanged);
124      }
125      if (experiment != null) {
126        experiment.Optimizers.ItemsAdded -= new CollectionItemsChangedEventHandler<IndexedItem<IOptimizer>>(Optimizers_ItemsAdded);
127        experiment.Optimizers.ItemsMoved -= new CollectionItemsChangedEventHandler<IndexedItem<IOptimizer>>(Optimizers_ItemsMoved);
128        experiment.Optimizers.ItemsRemoved -= new CollectionItemsChangedEventHandler<IndexedItem<IOptimizer>>(Optimizers_ItemsRemoved);
129        experiment.Optimizers.ItemsReplaced -= new CollectionItemsChangedEventHandler<IndexedItem<IOptimizer>>(Optimizers_ItemsReplaced);
130        experiment.Optimizers.CollectionReset -= new CollectionItemsChangedEventHandler<IndexedItem<IOptimizer>>(Optimizers_CollectionReset);
131      }
132    }
133    #endregion
134
135    protected override void OnContentChanged() {
136      base.OnContentChanged();
137      if (Content == null) {
138        optimizerTreeView.Nodes.Clear();
139      } else {
140        UpdateOptimizerTreeView();
141        optimizerTreeView.ExpandAll();
142      }
143    }
144
145    #region content events
146    private void Content_ExecutionStateChanged(object sender, EventArgs e) {
147      if (InvokeRequired) {
148        Invoke((Action<object, EventArgs>)Content_ExecutionStateChanged, sender, e);
149        return;
150      }
151      RebuildImageList();
152      SetEnabledStateOfControls();
153    }
154
155    private void optimizer_ExecutionStateChanged(object sender, EventArgs e) {
156      if (InvokeRequired) {
157        Invoke((Action<object, EventArgs>)optimizer_ExecutionStateChanged, sender, e);
158        return;
159      }
160      RebuildImageList();
161    }
162
163    private void batchRun_OptimizerChanged(object sender, EventArgs e) {
164      if (InvokeRequired) {
165        Invoke((Action<object, EventArgs>)batchRun_OptimizerChanged, sender, e);
166        return;
167      }
168      var batchRun = (BatchRun)sender;
169      foreach (TreeNode node in optimizerTreeViewMapping[batchRun]) {
170        foreach (TreeNode childNode in node.Nodes) {
171          DisposeTreeNode(childNode);
172          childNode.Remove();
173        }
174
175        if (batchRun.Optimizer != null) {
176          TreeNode childNode = CreateTreeNode(batchRun.Optimizer);
177          UpdateChildTreeNodes(childNode.Nodes, batchRun.Optimizer);
178          node.Nodes.Add(childNode);
179          node.Expand();
180        }
181      }
182      RebuildImageList();
183      UpdateDetailsViewHost();
184    }
185
186    private void Optimizers_ItemsAdded(object sender, CollectionItemsChangedEventArgs<IndexedItem<IOptimizer>> e) {
187      if (InvokeRequired) {
188        Invoke((Action<object, CollectionItemsChangedEventArgs<IndexedItem<IOptimizer>>>)Optimizers_ItemsAdded, sender, e);
189        return;
190      }
191
192      var optimizerList = (OptimizerList)sender;
193      IEnumerable<TreeNodeCollection> parentNodes;
194      if (optimizerList == Content.Optimizers) parentNodes = new List<TreeNodeCollection>() { optimizerTreeView.Nodes };
195      else {
196        Experiment experiment = optimizerTreeViewMapping.Keys.OfType<Experiment>().Where(exp => exp.Optimizers == optimizerList).First();
197        parentNodes = optimizerTreeViewMapping[experiment].Select(node => node.Nodes);
198      }
199
200      foreach (TreeNodeCollection parentNode in parentNodes) {
201        foreach (var childOptimizer in e.Items) {
202          TreeNode childNode = CreateTreeNode(childOptimizer.Value);
203          UpdateChildTreeNodes(childNode.Nodes, childOptimizer.Value);
204          parentNode.Insert(childOptimizer.Index, childNode);
205          childNode.ExpandAll();
206          if (childNode.Parent != null) childNode.Parent.ExpandAll();
207        }
208      }
209      RebuildImageList();
210    }
211    private void Optimizers_ItemsMoved(object sender, CollectionItemsChangedEventArgs<IndexedItem<IOptimizer>> e) {
212      if (InvokeRequired) {
213        Invoke((Action<object, CollectionItemsChangedEventArgs<IndexedItem<IOptimizer>>>)Optimizers_ItemsMoved, sender, e);
214        return;
215      }
216
217      var optimizerList = (OptimizerList)sender;
218      IEnumerable<TreeNodeCollection> parentNodes;
219      if (optimizerList == Content.Optimizers) parentNodes = new List<TreeNodeCollection>() { optimizerTreeView.Nodes };
220      else {
221        Experiment experiment = optimizerTreeViewMapping.Keys.OfType<Experiment>().Where(exp => exp.Optimizers == optimizerList).First();
222        parentNodes = optimizerTreeViewMapping[experiment].Select(node => node.Nodes);
223      }
224
225      foreach (TreeNodeCollection parentNode in parentNodes) {
226        //get all effected child nodes
227        foreach (TreeNode childNode in parentNode.OfType<TreeNode>()
228          .Where(n => e.OldItems.Select(x => x.Value).Contains((IOptimizer)n.Tag)).ToList()) {
229          DisposeTreeNode(childNode);
230          childNode.Remove();
231        }
232
233        foreach (var childOptimizer in e.Items) {
234          TreeNode childNode = CreateTreeNode(childOptimizer.Value);
235          UpdateChildTreeNodes(childNode.Nodes, childOptimizer.Value);
236          childNode.ExpandAll();
237          parentNode.Insert(childOptimizer.Index, childNode);
238        }
239      }
240      RebuildImageList();
241      UpdateDetailsViewHost();
242    }
243    private void Optimizers_ItemsRemoved(object sender, CollectionItemsChangedEventArgs<IndexedItem<IOptimizer>> e) {
244      if (InvokeRequired) {
245        Invoke((Action<object, CollectionItemsChangedEventArgs<IndexedItem<IOptimizer>>>)Optimizers_ItemsRemoved, sender, e);
246        return;
247      }
248
249      var optimizerList = (OptimizerList)sender;
250      IEnumerable<TreeNodeCollection> parentNodes;
251      if (optimizerList == Content.Optimizers) parentNodes = new List<TreeNodeCollection>() { optimizerTreeView.Nodes };
252      else {
253        Experiment experiment = optimizerTreeViewMapping.Keys.OfType<Experiment>().Where(exp => exp.Optimizers == optimizerList).First();
254        parentNodes = optimizerTreeViewMapping[experiment].Select(node => node.Nodes);
255      }
256
257      foreach (TreeNodeCollection parentNode in parentNodes) {
258        foreach (var childOptimizer in e.Items) {
259          TreeNode childNode = parentNode[childOptimizer.Index];
260          DisposeTreeNode(childNode);
261          childNode.Remove();
262        }
263      }
264      RebuildImageList();
265      UpdateDetailsViewHost();
266    }
267    private void Optimizers_ItemsReplaced(object sender, CollectionItemsChangedEventArgs<IndexedItem<IOptimizer>> e) {
268      if (InvokeRequired) {
269        Invoke((Action<object, CollectionItemsChangedEventArgs<IndexedItem<IOptimizer>>>)Optimizers_ItemsReplaced, sender, e);
270        return;
271      }
272
273      var optimizerList = (OptimizerList)sender;
274      IEnumerable<TreeNodeCollection> parentNodes;
275      if (optimizerList == Content.Optimizers) parentNodes = new List<TreeNodeCollection>() { optimizerTreeView.Nodes };
276      else {
277        Experiment experiment = optimizerTreeViewMapping.Keys.OfType<Experiment>().Where(exp => exp.Optimizers == optimizerList).First();
278        parentNodes = optimizerTreeViewMapping[experiment].Select(node => node.Nodes);
279      }
280
281      foreach (TreeNodeCollection parentNode in parentNodes) {
282        foreach (var childOptimizer in e.OldItems) {
283          TreeNode childNode = parentNode.Cast<TreeNode>().Where(n => n.Tag == childOptimizer.Value && n.Index == childOptimizer.Index).First();
284          DisposeTreeNode(childNode);
285          childNode.Remove();
286        }
287        foreach (var childOptimizer in e.Items) {
288          TreeNode childNode = CreateTreeNode(childOptimizer.Value);
289          UpdateChildTreeNodes(childNode.Nodes, childOptimizer.Value);
290          parentNode.Insert(childOptimizer.Index, childNode);
291        }
292      }
293      RebuildImageList();
294      UpdateDetailsViewHost();
295    }
296    private void Optimizers_CollectionReset(object sender, CollectionItemsChangedEventArgs<IndexedItem<IOptimizer>> e) {
297      if (InvokeRequired) {
298        Invoke((Action<object, CollectionItemsChangedEventArgs<IndexedItem<IOptimizer>>>)Optimizers_CollectionReset, sender, e);
299        return;
300      }
301
302      var optimizerList = (OptimizerList)sender;
303      IEnumerable<TreeNodeCollection> parentNodes;
304      if (optimizerList == Content.Optimizers) parentNodes = new List<TreeNodeCollection>() { optimizerTreeView.Nodes };
305      else {
306        Experiment experiment = optimizerTreeViewMapping.Keys.OfType<Experiment>().Where(exp => exp.Optimizers == optimizerList).First();
307        parentNodes = optimizerTreeViewMapping[experiment].Select(node => node.Nodes);
308      }
309
310      foreach (TreeNodeCollection parentNode in parentNodes) {
311        foreach (var childOptimizer in e.OldItems) {
312          TreeNode childNode = parentNode.Cast<TreeNode>().Where(n => n.Tag == childOptimizer.Value && n.Index == childOptimizer.Index).First();
313          DisposeTreeNode(childNode);
314          childNode.Remove();
315        }
316        foreach (var childOptimizer in e.Items) {
317          TreeNode childNode = CreateTreeNode(childOptimizer.Value);
318          UpdateChildTreeNodes(childNode.Nodes, childOptimizer.Value);
319          parentNode.Insert(childOptimizer.Index, childNode);
320        }
321      }
322      RebuildImageList();
323      UpdateDetailsViewHost();
324    }
325
326    private void optimizer_ToStringChanged(object sender, EventArgs e) {
327      if (InvokeRequired) {
328        Invoke((Action<object, EventArgs>)optimizer_ToStringChanged, sender, e);
329        return;
330      }
331      var optimizer = (IOptimizer)sender;
332      foreach (TreeNode node in optimizerTreeViewMapping[optimizer])
333        node.Text = optimizer.ToString();
334    }
335    #endregion
336
337    protected override void SetEnabledStateOfControls() {
338      base.SetEnabledStateOfControls();
339      BatchRun batchRun = null;
340      BatchRun parentBatchRun = null;
341      Experiment experiment = null;
342      if (optimizerTreeView.SelectedNode != null) {
343        batchRun = optimizerTreeView.SelectedNode.Tag as BatchRun;
344        experiment = optimizerTreeView.SelectedNode.Tag as Experiment;
345        if (optimizerTreeView.SelectedNode.Parent != null)
346          parentBatchRun = optimizerTreeView.SelectedNode.Parent.Tag as BatchRun;
347      }
348
349      optimizerTreeView.Enabled = Content != null;
350      detailsViewHost.Enabled = Content != null && optimizerTreeView.SelectedNode != null;
351
352      addButton.Enabled = Content != null && !Locked && !ReadOnly &&
353        (optimizerTreeView.SelectedNode == null || experiment != null || (batchRun != null && batchRun.Optimizer == null));
354      moveUpButton.Enabled = Content != null && !Locked && !ReadOnly &&
355        optimizerTreeView.SelectedNode != null && optimizerTreeView.SelectedNode.PrevNode != null && parentBatchRun == null;
356      moveDownButton.Enabled = Content != null && !Locked && !ReadOnly &&
357        optimizerTreeView.SelectedNode != null && optimizerTreeView.SelectedNode.NextNode != null && parentBatchRun == null;
358      removeButton.Enabled = Content != null && !Locked && !ReadOnly &&
359        optimizerTreeView.SelectedNode != null;
360    }
361
362    private void UpdateOptimizerTreeView() {
363      optimizerTreeView.Nodes.Clear();
364      UpdateChildTreeNodes(optimizerTreeView.Nodes, Content);
365      RebuildImageList();
366    }
367
368
369    private void UpdateChildTreeNodes(TreeNodeCollection collection, IOptimizer optimizer) {
370      var batchRun = optimizer as BatchRun;
371      var experiment = optimizer as Experiment;
372      if (experiment != null) UpdateChildTreeNodes(collection, experiment.Optimizers);
373      else if (batchRun != null && batchRun.Optimizer != null) UpdateChildTreeNodes(collection, new List<IOptimizer>() { batchRun.Optimizer });
374    }
375    private void UpdateChildTreeNodes(TreeNodeCollection collection, IEnumerable<IOptimizer> optimizers) {
376      foreach (IOptimizer optimizer in optimizers) {
377        var node = CreateTreeNode(optimizer);
378        collection.Add(node);
379        UpdateChildTreeNodes(node.Nodes, optimizer);
380      }
381    }
382
383
384    #region drag & drop
385    private void optimizerTreeView_ItemDrag(object sender, ItemDragEventArgs e) {
386      if (Locked) return;
387
388      TreeNode selectedNode = (TreeNode)e.Item;
389      var optimizer = (IOptimizer)selectedNode.Tag;
390      DataObject data = new DataObject();
391      data.SetData(HeuristicLab.Common.Constants.DragDropDataFormat, optimizer);
392      validDragOperation = true;
393
394      if (ReadOnly) {
395        DoDragDrop(data, DragDropEffects.Copy | DragDropEffects.Link);
396      } else {
397        DragDropEffects result = DoDragDrop(data, DragDropEffects.Copy | DragDropEffects.Link | DragDropEffects.Move);
398        if ((result & DragDropEffects.Move) == DragDropEffects.Move) {
399          if (selectedNode.Parent == null) Content.Optimizers.Remove(optimizer);
400          else {
401            var parentOptimizer = (IOptimizer)selectedNode.Parent.Tag;
402            var parentBatchRun = parentOptimizer as BatchRun;
403            var parentExperiment = parentOptimizer as Experiment;
404            if (parentBatchRun != null) parentBatchRun.Optimizer = null;
405            else if (parentExperiment != null) parentExperiment.Optimizers.Remove(optimizer);
406            else throw new NotSupportedException("Handling for specific type not implemented" + parentOptimizer.GetType());
407          }
408          SetEnabledStateOfControls();
409          UpdateDetailsViewHost();
410          RebuildImageList();
411        }
412      }
413    }
414
415    private bool validDragOperation = false;
416    private void optimizerTreeView_DragEnter(object sender, DragEventArgs e) {
417      validDragOperation = false;
418      if (!ReadOnly) {
419        if ((e.Data.GetData(HeuristicLab.Common.Constants.DragDropDataFormat) is IOptimizer)) validDragOperation = true;
420        else if (e.Data.GetData(HeuristicLab.Common.Constants.DragDropDataFormat) is IEnumerable) {
421          validDragOperation = true;
422          IEnumerable items = (IEnumerable)e.Data.GetData(HeuristicLab.Common.Constants.DragDropDataFormat);
423          foreach (object item in items)
424            validDragOperation = validDragOperation && (item is IOptimizer);
425        }
426      }
427    }
428    private void optimizerTreeView_DragOver(object sender, DragEventArgs e) {
429      e.Effect = DragDropEffects.None;
430      if (validDragOperation) {
431        Point coordinates = optimizerTreeView.PointToClient(new Point(e.X, e.Y));
432        TreeNode node = optimizerTreeView.GetNodeAt(coordinates);
433        Experiment experiment = null;
434        BatchRun batchRun = null;
435
436        if (node == null) experiment = Content;
437        else {
438          experiment = node.Tag as Experiment;
439          batchRun = node.Tag as BatchRun;
440        }
441
442        if (batchRun == null && experiment == null) return;
443        if (batchRun != null) {
444          var optimizer = e.Data.GetData(HeuristicLab.Common.Constants.DragDropDataFormat) as IOptimizer;
445          if (optimizer == null) return;
446          if (batchRun.Optimizer != null) return;
447          if (optimizer.GetObjectGraphObjects().OfType<IOptimizer>().Contains(batchRun)) return;
448        }
449
450        //do not allow recursive nesting of contents
451        if (experiment != null) {
452          var optimizer = e.Data.GetData(HeuristicLab.Common.Constants.DragDropDataFormat) as IOptimizer;
453          IEnumerable<IOptimizer> optimizers = null;
454          var enumerable = e.Data.GetData(HeuristicLab.Common.Constants.DragDropDataFormat) as IEnumerable;
455          if (enumerable != null) optimizers = enumerable.Cast<IOptimizer>();
456
457          if (optimizer != null && optimizer.GetObjectGraphObjects().OfType<IOptimizer>().Contains(experiment)) return;
458          if (optimizers != null && optimizers.GetObjectGraphObjects().OfType<IOptimizer>().Contains(experiment)) return;
459        }
460
461        if ((e.KeyState & 32) == 32) e.Effect = DragDropEffects.Link;  // ALT key
462        else if ((e.KeyState & 4) == 4) e.Effect = DragDropEffects.Move;  // SHIFT key
463        else if (e.AllowedEffect.HasFlag(DragDropEffects.Copy)) e.Effect = DragDropEffects.Copy;
464        else if (e.AllowedEffect.HasFlag(DragDropEffects.Move)) e.Effect = DragDropEffects.Move;
465        else if (e.AllowedEffect.HasFlag(DragDropEffects.Link)) e.Effect = DragDropEffects.Link;
466      }
467    }
468    private void optimizerTreeView_DragDrop(object sender, DragEventArgs e) {
469      Point coordinates = optimizerTreeView.PointToClient(new Point(e.X, e.Y));
470      TreeNode node = optimizerTreeView.GetNodeAt(coordinates);
471      Experiment experiment = null;
472      BatchRun batchRun = null;
473
474      if (node == null) experiment = Content;
475      else {
476        experiment = node.Tag as Experiment;
477        batchRun = node.Tag as BatchRun;
478      }
479
480      if (e.Data.GetData(HeuristicLab.Common.Constants.DragDropDataFormat) is IOptimizer) {
481        IOptimizer optimizer = (IOptimizer)e.Data.GetData(HeuristicLab.Common.Constants.DragDropDataFormat);
482        if (e.Effect.HasFlag(DragDropEffects.Copy)) optimizer = (IOptimizer)optimizer.Clone();
483        if (batchRun != null) batchRun.Optimizer = optimizer;
484        else if (experiment != null) experiment.Optimizers.Add(optimizer);
485        else throw new NotSupportedException("Handling for specific type not implemented" + node.Tag.GetType());
486      } else if (e.Data.GetData(HeuristicLab.Common.Constants.DragDropDataFormat) is IEnumerable) {
487        IEnumerable<IOptimizer> optimizers = ((IEnumerable)e.Data.GetData(HeuristicLab.Common.Constants.DragDropDataFormat)).Cast<IOptimizer>();
488        if (e.Effect.HasFlag(DragDropEffects.Copy)) {
489          Cloner cloner = new Cloner();
490          optimizers = optimizers.Select(o => (IOptimizer)o.Clone(cloner));
491        }
492        if (experiment != null) experiment.Optimizers.AddRange(optimizers);
493        else throw new NotSupportedException("Handling for specific type not implemented" + node.Tag.GetType());
494      }
495    }
496    #endregion
497
498    #region control events
499    private void optimizerTreeView_NodeMouseDoubleClick(object sender, TreeNodeMouseClickEventArgs e) {
500      if (rightClickOccured) return;
501      if (e.X < e.Node.Bounds.Left || e.X > e.Node.Bounds.Right) return;
502      e.Node.Toggle();
503      IOptimizer optimizer = (IOptimizer)e.Node.Tag;
504      MainFormManager.MainForm.ShowContent(optimizer);
505    }
506    private void optimizerTreeview_NodeMouseClick(object sender, TreeNodeMouseClickEventArgs e) {
507      if (e.X < e.Node.Bounds.Left || e.X > e.Node.Bounds.Right) return;
508      optimizerTreeView.SelectedNode = e.Node;
509      detailsViewHost.Content = (IOptimizer)e.Node.Tag;
510      SetEnabledStateOfControls();
511    }
512    private void optimizerTreeView_MouseDown(object sender, MouseEventArgs e) {
513      // enables deselection of treeNodes
514      if (e.Button != MouseButtons.Right) rightClickOccured = false;
515      if (optimizerTreeView.SelectedNode == null) return;
516      Point coordinates = new Point(e.X, e.Y);
517      TreeNode node = optimizerTreeView.GetNodeAt(coordinates);
518      if (node == null || coordinates.X < node.Bounds.Left || coordinates.X > node.Bounds.Right) {
519        optimizerTreeView.SelectedNode = null;
520        detailsViewHost.Content = null;
521        SetEnabledStateOfControls();
522      }
523    }
524
525
526    private void optimizerTreeView_KeyDown(object sender, KeyEventArgs e) {
527      if (ReadOnly) return;
528      if (optimizerTreeView.SelectedNode == null) return;
529      if (e.KeyCode != Keys.Delete) return;
530
531      var treeNode = optimizerTreeView.SelectedNode;
532      var optimizer = (IOptimizer)treeNode.Tag;
533
534      if (treeNode.Parent == null)
535        Content.Optimizers.Remove(optimizer);
536      else {
537        var batchRun = treeNode.Parent.Tag as BatchRun;
538        var experiment = treeNode.Parent.Tag as Experiment;
539        if (batchRun != null) batchRun.Optimizer = null;
540        else if (experiment != null) experiment.Optimizers.Remove(optimizer);
541        else throw new NotSupportedException("Handling for specific type not implemented" + optimizerTreeView.SelectedNode.Tag.GetType());
542      }
543      SetEnabledStateOfControls();
544      UpdateDetailsViewHost();
545      RebuildImageList();
546    }
547
548    private bool rightClickOccured = true;
549    private TreeNode toolStripMenuNode = null;
550    private void optimizerTreeView_RightClick(object sender, EventArgs e) {
551      rightClickOccured = true;
552      Point coordinates = optimizerTreeView.PointToClient(Cursor.Position);
553      toolStripMenuNode = optimizerTreeView.GetNodeAt(coordinates);
554
555      if (toolStripMenuNode != null && coordinates.X >= toolStripMenuNode.Bounds.Left && coordinates.X <= toolStripMenuNode.Bounds.Right) {
556        optimizerTreeView.SelectedNode = toolStripMenuNode;
557        detailsViewHost.Content = (IContent)toolStripMenuNode.Tag;
558        SetEnabledStateOfControls();
559
560        ExpandToolStripMenuItem.Enabled = !toolStripMenuNode.IsExpanded && toolStripMenuNode.Nodes.Count > 0;
561        ExpandToolStripMenuItem.Visible = !toolStripMenuNode.IsExpanded && toolStripMenuNode.Nodes.Count > 0;
562        CollapseToolStripMenuItem.Enabled = toolStripMenuNode.IsExpanded;
563        CollapseToolStripMenuItem.Visible = toolStripMenuNode.IsExpanded;
564        if (contextMenuStrip.Items.Cast<ToolStripMenuItem>().Any(item => item.Enabled))
565          contextMenuStrip.Show(Cursor.Position);
566      }
567    }
568
569    private void ExpandToolStripMenuItem_Click(object sender, EventArgs e) {
570      if (toolStripMenuNode != null) toolStripMenuNode.ExpandAll();
571    }
572    private void CollapseToolStripMenuItem_Click(object sender, EventArgs e) {
573      if (toolStripMenuNode != null) toolStripMenuNode.Collapse();
574    }
575
576    private void addButton_Click(object sender, System.EventArgs e) {
577      if (typeSelectorDialog == null) {
578        typeSelectorDialog = new TypeSelectorDialog();
579        typeSelectorDialog.Caption = "Select Optimizer";
580        typeSelectorDialog.TypeSelector.Caption = "Available Optimizers";
581        typeSelectorDialog.TypeSelector.Configure(typeof(IOptimizer), false, true);
582      }
583
584      if (typeSelectorDialog.ShowDialog(this) == DialogResult.OK) {
585        try {
586          IOptimizer optimizer = (IOptimizer)typeSelectorDialog.TypeSelector.CreateInstanceOfSelectedType();
587          if (optimizerTreeView.SelectedNode == null) Content.Optimizers.Add(optimizer);
588          else {
589            var batchRun = optimizerTreeView.SelectedNode.Tag as BatchRun;
590            var experiment = optimizerTreeView.SelectedNode.Tag as Experiment;
591            if (batchRun != null) batchRun.Optimizer = optimizer;
592            else if (experiment != null) experiment.Optimizers.Add(optimizer);
593            else throw new NotSupportedException("Handling for specific type not implemented" + optimizerTreeView.SelectedNode.Tag.GetType());
594          }
595        }
596        catch (Exception ex) {
597          ErrorHandling.ShowErrorDialog(this, ex);
598        }
599      }
600    }
601
602    private void moveUpButton_Click(object sender, EventArgs e) {
603      var optimizer = optimizerTreeView.SelectedNode.Tag as IOptimizer;
604      Experiment experiment = null;
605      if (optimizerTreeView.SelectedNode.Parent == null) experiment = Content;
606      else experiment = (Experiment)optimizerTreeView.SelectedNode.Parent.Tag;
607
608      int index = optimizerTreeView.SelectedNode.Index;
609      experiment.Optimizers.Reverse(index - 1, 2);
610      optimizerTreeView.SelectedNode = optimizerTreeViewMapping[optimizer].First();
611      SetEnabledStateOfControls();
612      UpdateDetailsViewHost();
613      RebuildImageList();
614    }
615    private void moveDownButton_Click(object sender, EventArgs e) {
616      var optimizer = optimizerTreeView.SelectedNode.Tag as IOptimizer;
617      Experiment experiment = null;
618      if (optimizerTreeView.SelectedNode.Parent == null) experiment = Content;
619      else experiment = (Experiment)optimizerTreeView.SelectedNode.Parent.Tag;
620
621      int index = optimizerTreeView.SelectedNode.Index;
622      experiment.Optimizers.Reverse(index, 2);
623      optimizerTreeView.SelectedNode = optimizerTreeViewMapping[optimizer].First();
624      SetEnabledStateOfControls();
625      UpdateDetailsViewHost();
626      RebuildImageList();
627    }
628
629    private void removeButton_Click(object sender, EventArgs e) {
630      var treeNode = optimizerTreeView.SelectedNode;
631      var optimizer = (IOptimizer)treeNode.Tag;
632
633      if (treeNode.Parent == null)
634        Content.Optimizers.Remove(optimizer);
635      else {
636        var batchRun = treeNode.Parent.Tag as BatchRun;
637        var experiment = treeNode.Parent.Tag as Experiment;
638        if (batchRun != null) batchRun.Optimizer = null;
639        else if (experiment != null) experiment.Optimizers.Remove(optimizer);
640        else throw new NotSupportedException("Handling for specific type not implemented" + optimizerTreeView.SelectedNode.Tag.GetType());
641      }
642      SetEnabledStateOfControls();
643      UpdateDetailsViewHost();
644      RebuildImageList();
645    }
646
647    private void showDetailsCheckBox_CheckedChanged(object sender, System.EventArgs e) {
648      if (showDetailsCheckBox.Checked) {
649        splitContainer.Panel2Collapsed = false;
650        detailsGroupBox.Enabled = optimizerTreeView.SelectedNode != null;
651        detailsViewHost.Content = optimizerTreeView.SelectedNode != null ? (IOptimizer)optimizerTreeView.SelectedNode.Tag : null;
652      } else {
653        splitContainer.Panel2Collapsed = true;
654        detailsViewHost.Content = null;
655      }
656    }
657    #endregion
658
659    #region helpers
660    private void UpdateDetailsViewHost() {
661      if (optimizerTreeView.SelectedNode != null)
662        detailsViewHost.Content = (IOptimizer)optimizerTreeView.SelectedNode.Tag;
663      else
664        detailsViewHost.Content = null;
665    }
666
667    private TreeNode CreateTreeNode(IOptimizer optimizer) {
668      TreeNode node = new TreeNode(optimizer.ToString());
669      node.Tag = optimizer;
670      List<TreeNode> nodes;
671      if (!optimizerTreeViewMapping.TryGetValue(optimizer, out nodes)) {
672        nodes = new List<TreeNode>();
673        optimizerTreeViewMapping.Add(optimizer, nodes);
674        RegisterOptimizerEvents(optimizer);
675      }
676      nodes.Add(node);
677      return node;
678    }
679
680    private void DisposeTreeNode(TreeNode node) {
681      var optimizer = (IOptimizer)node.Tag;
682      List<TreeNode> nodes;
683      if (!optimizerTreeViewMapping.TryGetValue(optimizer, out nodes))
684        throw new ArgumentException();
685      nodes.Remove(node);
686      if (nodes.Count == 0) {
687        optimizerTreeViewMapping.Remove(optimizer);
688        DeregisterOptimizerEvents(optimizer);
689      }
690    }
691
692    private IEnumerable<TreeNode> IterateTreeNodes(TreeNode node = null) {
693      TreeNodeCollection nodes;
694      if (node == null)
695        nodes = optimizerTreeView.Nodes;
696      else {
697        nodes = node.Nodes;
698        yield return node;
699      }
700
701      foreach (var childNode in nodes.OfType<TreeNode>())
702        foreach (var n in IterateTreeNodes(childNode))
703          yield return n;
704    }
705
706    private void RebuildImageList() {
707      if (InvokeRequired) {
708        Invoke((Action)RebuildImageList);
709        return;
710      }
711
712      optimizerTreeView.ImageList.Images.Clear();
713      foreach (TreeNode treeNode in IterateTreeNodes()) {
714        var optimizer = (IOptimizer)treeNode.Tag;
715        optimizerTreeView.ImageList.Images.Add(optimizer == null ? HeuristicLab.Common.Resources.VSImageLibrary.Nothing : optimizer.ItemImage);
716        treeNode.ImageIndex = optimizerTreeView.ImageList.Images.Count - 1;
717        treeNode.SelectedImageIndex = treeNode.ImageIndex;
718      }
719    }
720    #endregion
721
722
723    public sealed class CustomTreeView : System.Windows.Forms.TreeView {
724      protected override void WndProc(ref System.Windows.Forms.Message m) {
725        const int WM_RBUTTONDOWN = 0x204;
726        if (m.Msg == WM_RBUTTONDOWN) {
727          //Raise your custom event right click event to prevent node highlighting
728          OnRightClick();
729          return;
730        }
731        base.WndProc(ref m);
732      }
733
734      public event EventHandler RightClick;
735      private void OnRightClick() {
736        var handler = RightClick;
737        if (handler != null) RightClick(this, EventArgs.Empty);
738      }
739    }
740  }
741}
Note: See TracBrowser for help on using the repository browser.