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

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

#1555: Corrected tree node stated (expanded / collapsed) after double click.

File size: 31.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 (e.X < e.Node.Bounds.Left || e.X > e.Node.Bounds.Right) return;
501      IOptimizer optimizer = (IOptimizer)e.Node.Tag;
502      MainFormManager.MainForm.ShowContent(optimizer);
503    }
504    private void optimizerTreeview_NodeMouseClick(object sender, TreeNodeMouseClickEventArgs e) {
505      if (e.X < e.Node.Bounds.Left || e.X > e.Node.Bounds.Right) return;
506      optimizerTreeView.SelectedNode = e.Node;
507      optimizerTreeView.SelectedNode.Toggle();
508      detailsViewHost.Content = (IOptimizer)e.Node.Tag;
509      SetEnabledStateOfControls();
510    }
511    private void optimizerTreeView_MouseDown(object sender, MouseEventArgs e) {
512      // enables deselection of treeNodes
513      if (e.Button != System.Windows.Forms.MouseButtons.Left) return;
514      if (optimizerTreeView.SelectedNode == null) return;
515      Point coordinates = new Point(e.X, e.Y);
516      TreeNode node = optimizerTreeView.GetNodeAt(coordinates);
517      if (node == null || coordinates.X < node.Bounds.Left || coordinates.X > node.Bounds.Right) {
518        optimizerTreeView.SelectedNode = null;
519        detailsViewHost.Content = null;
520        SetEnabledStateOfControls();
521      }
522    }
523
524    private void optimizerTreeView_KeyDown(object sender, KeyEventArgs e) {
525      if (ReadOnly) return;
526      if (optimizerTreeView.SelectedNode == null) return;
527      if (e.KeyCode != Keys.Delete) return;
528
529      var treeNode = optimizerTreeView.SelectedNode;
530      var optimizer = (IOptimizer)treeNode.Tag;
531
532      if (treeNode.Parent == null)
533        Content.Optimizers.Remove(optimizer);
534      else {
535        var batchRun = treeNode.Parent.Tag as BatchRun;
536        var experiment = treeNode.Parent.Tag as Experiment;
537        if (batchRun != null) batchRun.Optimizer = null;
538        else if (experiment != null) experiment.Optimizers.Remove(optimizer);
539        else throw new NotSupportedException("Handling for specific type not implemented" + optimizerTreeView.SelectedNode.Tag.GetType());
540      }
541      SetEnabledStateOfControls();
542      UpdateDetailsViewHost();
543      RebuildImageList();
544    }
545
546    private void addButton_Click(object sender, System.EventArgs e) {
547      if (typeSelectorDialog == null) {
548        typeSelectorDialog = new TypeSelectorDialog();
549        typeSelectorDialog.Caption = "Select Optimizer";
550        typeSelectorDialog.TypeSelector.Caption = "Available Optimizers";
551        typeSelectorDialog.TypeSelector.Configure(typeof(IOptimizer), false, true);
552      }
553
554      if (typeSelectorDialog.ShowDialog(this) == DialogResult.OK) {
555        try {
556          IOptimizer optimizer = (IOptimizer)typeSelectorDialog.TypeSelector.CreateInstanceOfSelectedType();
557          if (optimizerTreeView.SelectedNode == null) Content.Optimizers.Add(optimizer);
558          else {
559            var batchRun = optimizerTreeView.SelectedNode.Tag as BatchRun;
560            var experiment = optimizerTreeView.SelectedNode.Tag as Experiment;
561            if (batchRun != null) batchRun.Optimizer = optimizer;
562            else if (experiment != null) experiment.Optimizers.Add(optimizer);
563            else throw new NotSupportedException("Handling for specific type not implemented" + optimizerTreeView.SelectedNode.Tag.GetType());
564          }
565        }
566        catch (Exception ex) {
567          ErrorHandling.ShowErrorDialog(this, ex);
568        }
569      }
570    }
571
572    private void moveUpButton_Click(object sender, EventArgs e) {
573      var optimizer = optimizerTreeView.SelectedNode.Tag as IOptimizer;
574      Experiment experiment = null;
575      if (optimizerTreeView.SelectedNode.Parent == null) experiment = Content;
576      else experiment = (Experiment)optimizerTreeView.SelectedNode.Parent.Tag;
577
578      int index = optimizerTreeView.SelectedNode.Index;
579      experiment.Optimizers.Reverse(index - 1, 2);
580      optimizerTreeView.SelectedNode = optimizerTreeViewMapping[optimizer].First();
581      SetEnabledStateOfControls();
582      UpdateDetailsViewHost();
583      RebuildImageList();
584    }
585    private void moveDownButton_Click(object sender, EventArgs e) {
586      var optimizer = optimizerTreeView.SelectedNode.Tag as IOptimizer;
587      Experiment experiment = null;
588      if (optimizerTreeView.SelectedNode.Parent == null) experiment = Content;
589      else experiment = (Experiment)optimizerTreeView.SelectedNode.Parent.Tag;
590
591      int index = optimizerTreeView.SelectedNode.Index;
592      experiment.Optimizers.Reverse(index, 2);
593      optimizerTreeView.SelectedNode = optimizerTreeViewMapping[optimizer].First();
594      SetEnabledStateOfControls();
595      UpdateDetailsViewHost();
596      RebuildImageList();
597    }
598
599    private void removeButton_Click(object sender, EventArgs e) {
600      var treeNode = optimizerTreeView.SelectedNode;
601      var optimizer = (IOptimizer)treeNode.Tag;
602
603      if (treeNode.Parent == null)
604        Content.Optimizers.Remove(optimizer);
605      else {
606        var batchRun = treeNode.Parent.Tag as BatchRun;
607        var experiment = treeNode.Parent.Tag as Experiment;
608        if (batchRun != null) batchRun.Optimizer = null;
609        else if (experiment != null) experiment.Optimizers.Remove(optimizer);
610        else throw new NotSupportedException("Handling for specific type not implemented" + optimizerTreeView.SelectedNode.Tag.GetType());
611      }
612      SetEnabledStateOfControls();
613      UpdateDetailsViewHost();
614      RebuildImageList();
615    }
616
617    private void showDetailsCheckBox_CheckedChanged(object sender, System.EventArgs e) {
618      if (showDetailsCheckBox.Checked) {
619        splitContainer.Panel2Collapsed = false;
620        detailsGroupBox.Enabled = optimizerTreeView.SelectedNode != null;
621        detailsViewHost.Content = optimizerTreeView.SelectedNode != null ? (IOptimizer)optimizerTreeView.SelectedNode.Tag : null;
622      } else {
623        splitContainer.Panel2Collapsed = true;
624        detailsViewHost.Content = null;
625      }
626    }
627    #endregion
628
629    #region helpers
630    private void UpdateDetailsViewHost() {
631      if (optimizerTreeView.SelectedNode != null)
632        detailsViewHost.Content = (IOptimizer)optimizerTreeView.SelectedNode.Tag;
633      else
634        detailsViewHost.Content = null;
635    }
636
637    private TreeNode CreateTreeNode(IOptimizer optimizer) {
638      TreeNode node = new TreeNode(optimizer.ToString());
639      node.Tag = optimizer;
640      List<TreeNode> nodes;
641      if (!optimizerTreeViewMapping.TryGetValue(optimizer, out nodes)) {
642        nodes = new List<TreeNode>();
643        optimizerTreeViewMapping.Add(optimizer, nodes);
644        RegisterOptimizerEvents(optimizer);
645      }
646      nodes.Add(node);
647      return node;
648    }
649
650    private void DisposeTreeNode(TreeNode node) {
651      var optimizer = (IOptimizer)node.Tag;
652      List<TreeNode> nodes;
653      if (!optimizerTreeViewMapping.TryGetValue(optimizer, out nodes))
654        throw new ArgumentException();
655      nodes.Remove(node);
656      if (nodes.Count == 0) {
657        optimizerTreeViewMapping.Remove(optimizer);
658        DeregisterOptimizerEvents(optimizer);
659      }
660    }
661
662    private IEnumerable<TreeNode> IterateTreeNodes(TreeNode node = null) {
663      TreeNodeCollection nodes;
664      if (node == null)
665        nodes = optimizerTreeView.Nodes;
666      else {
667        nodes = node.Nodes;
668        yield return node;
669      }
670
671      foreach (var childNode in nodes.OfType<TreeNode>())
672        foreach (var n in IterateTreeNodes(childNode))
673          yield return n;
674    }
675
676    private void RebuildImageList() {
677      if (InvokeRequired) {
678        Invoke((Action)RebuildImageList);
679        return;
680      }
681
682      optimizerTreeView.ImageList.Images.Clear();
683      foreach (TreeNode treeNode in IterateTreeNodes()) {
684        var optimizer = (IOptimizer)treeNode.Tag;
685        optimizerTreeView.ImageList.Images.Add(optimizer == null ? HeuristicLab.Common.Resources.VSImageLibrary.Nothing : optimizer.ItemImage);
686        treeNode.ImageIndex = optimizerTreeView.ImageList.Images.Count - 1;
687        treeNode.SelectedImageIndex = treeNode.ImageIndex;
688      }
689    }
690    #endregion
691  }
692}
Note: See TracBrowser for help on using the repository browser.