#region License Information /* HeuristicLab * Copyright (C) 2002-2011 Heuristic and Evolutionary Algorithms Laboratory (HEAL) * * This file is part of HeuristicLab. * * HeuristicLab is free software: you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. * * HeuristicLab is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with HeuristicLab. If not, see . */ #endregion using System; using System.Collections; using System.Collections.Generic; using System.Drawing; using System.Linq; using System.Windows.Forms; using HeuristicLab.Collections; using HeuristicLab.Common; using HeuristicLab.Core; using HeuristicLab.Core.Views; using HeuristicLab.MainForm; using HeuristicLab.PluginInfrastructure; namespace HeuristicLab.Optimization.Views { public sealed partial class ExperimentTreeView : ItemView { private TypeSelectorDialog typeSelectorDialog; private Dictionary> treeNodeTagMapping; public ExperimentTreeView() { InitializeComponent(); treeNodeTagMapping = new Dictionary>(); } protected override void Dispose(bool disposing) { if (disposing) { if (typeSelectorDialog != null) typeSelectorDialog.Dispose(); if (components != null) components.Dispose(); } base.Dispose(disposing); } #region necessary code to handle dock correctly regarding the expanded nodes bool[] expandendedState; protected override void OnHandleCreated(EventArgs e) { base.OnHandleCreated(e); if (expandendedState == null) return; var nodes = IterateTreeNodes().ToList(); for (int i = 0; i < nodes.Count; i++) if (expandendedState[i]) nodes[i].Expand(); } protected override void OnHandleDestroyed(EventArgs e) { base.OnHandleDestroyed(e); var nodes = IterateTreeNodes().ToList(); expandendedState = new bool[nodes.Count]; for (int i = 0; i < nodes.Count; i++) expandendedState[i] = nodes[i].IsExpanded; } #endregion public new Experiment Content { get { return (Experiment)base.Content; } set { base.Content = value; } } #region events registration protected override void RegisterContentEvents() { base.RegisterContentEvents(); Content.ExecutionStateChanged += new EventHandler(Content_ExecutionStateChanged); Content.Optimizers.ItemsAdded += new CollectionItemsChangedEventHandler>(Optimizers_ItemsAdded); Content.Optimizers.ItemsMoved += new CollectionItemsChangedEventHandler>(Optimizers_ItemsMoved); Content.Optimizers.ItemsRemoved += new CollectionItemsChangedEventHandler>(Optimizers_ItemsRemoved); Content.Optimizers.ItemsReplaced += new CollectionItemsChangedEventHandler>(Optimizers_ItemsReplaced); Content.Optimizers.CollectionReset += new CollectionItemsChangedEventHandler>(Optimizers_CollectionReset); } protected override void DeregisterContentEvents() { Content.ExecutionStateChanged -= new EventHandler(Content_ExecutionStateChanged); Content.Optimizers.ItemsAdded -= new CollectionItemsChangedEventHandler>(Optimizers_ItemsAdded); Content.Optimizers.ItemsMoved -= new CollectionItemsChangedEventHandler>(Optimizers_ItemsMoved); Content.Optimizers.ItemsRemoved -= new CollectionItemsChangedEventHandler>(Optimizers_ItemsRemoved); Content.Optimizers.ItemsReplaced -= new CollectionItemsChangedEventHandler>(Optimizers_ItemsReplaced); Content.Optimizers.CollectionReset -= new CollectionItemsChangedEventHandler>(Optimizers_CollectionReset); foreach (var optimizer in treeNodeTagMapping.Keys) DeregisterNamedItemEvents(optimizer); base.DeregisterContentEvents(); } private void RegisterNamedItemEvents(INamedItem namedItem) { namedItem.ToStringChanged += new EventHandler(namedItem_ToStringChanged); namedItem.ItemImageChanged += new EventHandler(namedItem_ItemImageChanged); var algorithm = namedItem as IAlgorithm; var batchRun = namedItem as BatchRun; var experiment = namedItem as Experiment; if (algorithm != null) { algorithm.ProblemChanged += new EventHandler(algorithm_ProblemChanged); } else if (batchRun != null) { batchRun.OptimizerChanged += new EventHandler(batchRun_OptimizerChanged); } else if (experiment != null) { experiment.Optimizers.ItemsAdded += new CollectionItemsChangedEventHandler>(Optimizers_ItemsAdded); experiment.Optimizers.ItemsMoved += new CollectionItemsChangedEventHandler>(Optimizers_ItemsMoved); experiment.Optimizers.ItemsRemoved += new CollectionItemsChangedEventHandler>(Optimizers_ItemsRemoved); experiment.Optimizers.ItemsReplaced += new CollectionItemsChangedEventHandler>(Optimizers_ItemsReplaced); experiment.Optimizers.CollectionReset += new CollectionItemsChangedEventHandler>(Optimizers_CollectionReset); } } private void DeregisterNamedItemEvents(INamedItem namedItem) { namedItem.ToStringChanged -= new EventHandler(namedItem_ToStringChanged); namedItem.ItemImageChanged -= new EventHandler(namedItem_ItemImageChanged); var algorithm = namedItem as IAlgorithm; var batchRun = namedItem as BatchRun; var experiment = namedItem as Experiment; if (algorithm != null) { algorithm.ProblemChanged -= new EventHandler(algorithm_ProblemChanged); } else if (batchRun != null) { batchRun.OptimizerChanged -= new EventHandler(batchRun_OptimizerChanged); } else if (experiment != null) { experiment.Optimizers.ItemsAdded -= new CollectionItemsChangedEventHandler>(Optimizers_ItemsAdded); experiment.Optimizers.ItemsMoved -= new CollectionItemsChangedEventHandler>(Optimizers_ItemsMoved); experiment.Optimizers.ItemsRemoved -= new CollectionItemsChangedEventHandler>(Optimizers_ItemsRemoved); experiment.Optimizers.ItemsReplaced -= new CollectionItemsChangedEventHandler>(Optimizers_ItemsReplaced); experiment.Optimizers.CollectionReset -= new CollectionItemsChangedEventHandler>(Optimizers_CollectionReset); } } #endregion protected override void OnContentChanged() { base.OnContentChanged(); if (Content == null) { treeView.Nodes.Clear(); } else { UpdateOptimizerTreeView(); treeView.ExpandAll(); } } #region content events private void Content_ExecutionStateChanged(object sender, EventArgs e) { if (InvokeRequired) { Invoke((Action)Content_ExecutionStateChanged, sender, e); return; } SetEnabledStateOfControls(); } private void algorithm_ProblemChanged(object sender, EventArgs e) { if (InvokeRequired) { Invoke((Action)algorithm_ProblemChanged, sender, e); return; } var algorithm = (IAlgorithm)sender; foreach (TreeNode node in treeNodeTagMapping[algorithm]) { foreach (TreeNode childNode in node.Nodes.OfType().ToList()) { DisposeTreeNode(childNode); childNode.Remove(); } foreach (TreeNode childNode in CreateAlgorithmChildNodes(algorithm)) node.Nodes.Add(childNode); node.Expand(); } RebuildImageList(); UpdateDetailsViewHost(); } private void batchRun_OptimizerChanged(object sender, EventArgs e) { if (InvokeRequired) { Invoke((Action)batchRun_OptimizerChanged, sender, e); return; } var batchRun = (BatchRun)sender; foreach (TreeNode node in treeNodeTagMapping[batchRun]) { foreach (TreeNode childNode in node.Nodes.OfType().ToList()) { DisposeTreeNode(childNode); childNode.Remove(); } if (batchRun.Optimizer != null) { UpdateChildTreeNodes(node.Nodes, batchRun); node.Expand(); } } RebuildImageList(); UpdateDetailsViewHost(); } private void Optimizers_ItemsAdded(object sender, CollectionItemsChangedEventArgs> e) { if (InvokeRequired) { Invoke((Action>>)Optimizers_ItemsAdded, sender, e); return; } var optimizerList = (OptimizerList)sender; IEnumerable parentNodes; if (optimizerList == Content.Optimizers) parentNodes = new List() { treeView.Nodes }; else { Experiment experiment = treeNodeTagMapping.Keys.OfType().Where(exp => exp.Optimizers == optimizerList).First(); parentNodes = treeNodeTagMapping[experiment].Select(node => node.Nodes); } foreach (TreeNodeCollection parentNode in parentNodes) { foreach (var childOptimizer in e.Items) { TreeNode childNode = CreateTreeNode(childOptimizer.Value); UpdateChildTreeNodes(childNode.Nodes, childOptimizer.Value); parentNode.Insert(childOptimizer.Index, childNode); childNode.ExpandAll(); if (childNode.Parent != null) childNode.Parent.ExpandAll(); } } RebuildImageList(); } private void Optimizers_ItemsMoved(object sender, CollectionItemsChangedEventArgs> e) { if (InvokeRequired) { Invoke((Action>>)Optimizers_ItemsMoved, sender, e); return; } var optimizerList = (OptimizerList)sender; IEnumerable parentNodes; if (optimizerList == Content.Optimizers) parentNodes = new List() { treeView.Nodes }; else { Experiment experiment = treeNodeTagMapping.Keys.OfType().Where(exp => exp.Optimizers == optimizerList).First(); parentNodes = treeNodeTagMapping[experiment].Select(node => node.Nodes); } foreach (TreeNodeCollection parentNode in parentNodes) { //get all effected child nodes foreach (TreeNode childNode in parentNode.OfType() .Where(n => e.OldItems.Select(x => x.Value).Contains((IOptimizer)n.Tag)).ToList()) { DisposeTreeNode(childNode); childNode.Remove(); } foreach (var childOptimizer in e.Items) { TreeNode childNode = CreateTreeNode(childOptimizer.Value); UpdateChildTreeNodes(childNode.Nodes, childOptimizer.Value); childNode.ExpandAll(); parentNode.Insert(childOptimizer.Index, childNode); } } RebuildImageList(); UpdateDetailsViewHost(); } private void Optimizers_ItemsRemoved(object sender, CollectionItemsChangedEventArgs> e) { if (InvokeRequired) { Invoke((Action>>)Optimizers_ItemsRemoved, sender, e); return; } var optimizerList = (OptimizerList)sender; IEnumerable parentNodes; if (optimizerList == Content.Optimizers) parentNodes = new List() { treeView.Nodes }; else { Experiment experiment = treeNodeTagMapping.Keys.OfType().Where(exp => exp.Optimizers == optimizerList).First(); parentNodes = treeNodeTagMapping[experiment].Select(node => node.Nodes); } foreach (TreeNodeCollection parentNode in parentNodes) { foreach (var childOptimizer in e.Items) { TreeNode childNode = parentNode[childOptimizer.Index]; DisposeTreeNode(childNode); childNode.Remove(); } } RebuildImageList(); UpdateDetailsViewHost(); } private void Optimizers_ItemsReplaced(object sender, CollectionItemsChangedEventArgs> e) { if (InvokeRequired) { Invoke((Action>>)Optimizers_ItemsReplaced, sender, e); return; } var optimizerList = (OptimizerList)sender; IEnumerable parentNodes; if (optimizerList == Content.Optimizers) parentNodes = new List() { treeView.Nodes }; else { Experiment experiment = treeNodeTagMapping.Keys.OfType().Where(exp => exp.Optimizers == optimizerList).First(); parentNodes = treeNodeTagMapping[experiment].Select(node => node.Nodes); } foreach (TreeNodeCollection parentNode in parentNodes) { foreach (var childOptimizer in e.OldItems) { TreeNode childNode = parentNode.Cast().Where(n => n.Tag == childOptimizer.Value && n.Index == childOptimizer.Index).First(); DisposeTreeNode(childNode); childNode.Remove(); } foreach (var childOptimizer in e.Items) { TreeNode childNode = CreateTreeNode(childOptimizer.Value); UpdateChildTreeNodes(childNode.Nodes, childOptimizer.Value); parentNode.Insert(childOptimizer.Index, childNode); } } RebuildImageList(); UpdateDetailsViewHost(); } private void Optimizers_CollectionReset(object sender, CollectionItemsChangedEventArgs> e) { if (InvokeRequired) { Invoke((Action>>)Optimizers_CollectionReset, sender, e); return; } var optimizerList = (OptimizerList)sender; IEnumerable parentNodes; if (optimizerList == Content.Optimizers) parentNodes = new List() { treeView.Nodes }; else { Experiment experiment = treeNodeTagMapping.Keys.OfType().Where(exp => exp.Optimizers == optimizerList).First(); parentNodes = treeNodeTagMapping[experiment].Select(node => node.Nodes); } foreach (TreeNodeCollection parentNode in parentNodes) { foreach (var childOptimizer in e.OldItems) { TreeNode childNode = parentNode.Cast().Where(n => n.Tag == childOptimizer.Value && n.Index == childOptimizer.Index).First(); DisposeTreeNode(childNode); childNode.Remove(); } foreach (var childOptimizer in e.Items) { TreeNode childNode = CreateTreeNode(childOptimizer.Value); UpdateChildTreeNodes(childNode.Nodes, childOptimizer.Value); parentNode.Insert(childOptimizer.Index, childNode); } } RebuildImageList(); UpdateDetailsViewHost(); } private void namedItem_ToStringChanged(object sender, EventArgs e) { if (InvokeRequired) { Invoke((Action)namedItem_ToStringChanged, sender, e); return; } var namedItem = (INamedItem)sender; foreach (TreeNode node in treeNodeTagMapping[namedItem]) node.Text = namedItem.ToString(); } private void namedItem_ItemImageChanged(object sender, EventArgs e) { if (InvokeRequired) { Invoke((Action)namedItem_ItemImageChanged, sender, e); return; } INamedItem namedItem = (INamedItem)sender; foreach (TreeNode node in treeNodeTagMapping[namedItem]) { treeView.ImageList.Images[node.ImageIndex] = namedItem.ItemImage; node.ImageIndex = node.ImageIndex; } SetEnabledStateOfControls(); } #endregion protected override void PropagateStateChanges(Control control, Type type, System.Reflection.PropertyInfo propertyInfo) { return; } protected override void SetEnabledStateOfControls() { base.SetEnabledStateOfControls(); IOptimizer optimizer = null; IAlgorithm algorithm = null; BatchRun batchRun = null; Experiment experiment = null; IOptimizer parentOptimizer = null; Experiment parentExperiment = null; if (treeView.SelectedNode != null) { optimizer = treeView.SelectedNode.Tag as IOptimizer; algorithm = optimizer as IAlgorithm; batchRun = optimizer as BatchRun; experiment = optimizer as Experiment; if (treeView.SelectedNode.Parent != null) parentOptimizer = treeView.SelectedNode.Parent.Tag as IOptimizer; else parentOptimizer = Content; parentExperiment = parentOptimizer as Experiment; } treeView.Enabled = Content != null; if (parentOptimizer != null) { detailsViewHost.ReadOnly = parentOptimizer.ExecutionState == ExecutionState.Started; detailsViewHost.Locked = parentOptimizer.ExecutionState == ExecutionState.Started; } addButton.Enabled = Content != null && !Locked && !ReadOnly && (treeView.SelectedNode == null || experiment != null || batchRun != null || algorithm != null); moveUpButton.Enabled = Content != null && !Locked && !ReadOnly && treeView.SelectedNode != null && treeView.SelectedNode.PrevNode != null && parentExperiment != null; moveDownButton.Enabled = Content != null && !Locked && !ReadOnly && treeView.SelectedNode != null && treeView.SelectedNode.NextNode != null && parentExperiment != null; removeButton.Enabled = Content != null && !Locked && !ReadOnly && optimizer != null; } private void UpdateOptimizerTreeView() { treeView.Nodes.Clear(); UpdateChildTreeNodes(treeView.Nodes, Content); RebuildImageList(); } private void UpdateChildTreeNodes(TreeNodeCollection collection, IOptimizer optimizer) { var batchRun = optimizer as BatchRun; var experiment = optimizer as Experiment; if (batchRun != null && batchRun.Optimizer != null) UpdateChildTreeNodes(collection, new List() { batchRun.Optimizer }); else if (experiment != null) UpdateChildTreeNodes(collection, experiment.Optimizers); } private void UpdateChildTreeNodes(TreeNodeCollection collection, IEnumerable optimizers) { foreach (IOptimizer optimizer in optimizers) { var node = CreateTreeNode(optimizer); collection.Add(node); UpdateChildTreeNodes(node.Nodes, optimizer); } } #region drag & drop private void optimizerTreeView_ItemDrag(object sender, ItemDragEventArgs e) { if (Locked) return; TreeNode selectedNode = (TreeNode)e.Item; var item = (IItem)selectedNode.Tag; if (item == null) return; DataObject data = new DataObject(); data.SetData(HeuristicLab.Common.Constants.DragDropDataFormat, item); validDragOperation = true; if (ReadOnly || !(item is IOptimizer)) { DoDragDrop(data, DragDropEffects.Copy | DragDropEffects.Link); } else { DragDropEffects result = DoDragDrop(data, DragDropEffects.Copy | DragDropEffects.Link | DragDropEffects.Move); if ((result & DragDropEffects.Move) == DragDropEffects.Move) { var optimizer = (IOptimizer)item; if (selectedNode.Parent == null) Content.Optimizers.Remove(optimizer); else { var parentOptimizer = (IOptimizer)selectedNode.Parent.Tag; var parentBatchRun = parentOptimizer as BatchRun; var parentExperiment = parentOptimizer as Experiment; if (parentBatchRun != null) parentBatchRun.Optimizer = null; else if (parentExperiment != null) parentExperiment.Optimizers.Remove(optimizer); else throw new NotSupportedException("Handling for specific type not implemented" + parentOptimizer.GetType()); } } } } private bool validDragOperation = false; private void optimizerTreeView_DragEnter(object sender, DragEventArgs e) { validDragOperation = false; if (!ReadOnly) { var data = e.Data.GetData(HeuristicLab.Common.Constants.DragDropDataFormat); if (data is IOptimizer) validDragOperation = true; else if (data is IProblem) validDragOperation = true; else if (data is IEnumerable) { IEnumerable items = (IEnumerable)data; validDragOperation = true; foreach (object item in items) validDragOperation = validDragOperation && (item is IOptimizer); } } } private void optimizerTreeView_DragOver(object sender, DragEventArgs e) { e.Effect = DragDropEffects.None; if (!validDragOperation) return; Point coordinates = treeView.PointToClient(new Point(e.X, e.Y)); TreeNode node = treeView.GetNodeAt(coordinates); Experiment experiment = null; BatchRun batchRun = null; Algorithm algorithm = null; if (node == null) experiment = Content; else { experiment = node.Tag as Experiment; batchRun = node.Tag as BatchRun; algorithm = node.Tag as Algorithm; } if (batchRun == null && experiment == null && algorithm == null) return; var data = e.Data.GetData(HeuristicLab.Common.Constants.DragDropDataFormat); if (algorithm != null) { var problem = data as IProblem; if (problem == null) return; if (!algorithm.ProblemType.IsAssignableFrom(problem.GetType())) return; } else if (batchRun != null) { var optimizer = data as IOptimizer; if (optimizer == null) return; if (batchRun == optimizer) return; if (optimizer.NestedOptimizers.Contains(batchRun)) return; } //do not allow recursive nesting of contents else if (experiment != null) { var optimizer = data as IOptimizer; IEnumerable optimizers = null; var enumerable = data as IEnumerable; if (enumerable != null) optimizers = enumerable.Cast(); if (experiment == optimizer) return; if (optimizer != null && optimizer.NestedOptimizers.Contains(experiment)) return; if (optimizers != null && optimizers.Any(x => x.NestedOptimizers.Contains(experiment))) return; } if ((e.KeyState & 32) == 32 && e.AllowedEffect.HasFlag(DragDropEffects.Link)) e.Effect = DragDropEffects.Link; // ALT key else if ((e.KeyState & 4) == 4 && e.AllowedEffect.HasFlag(DragDropEffects.Move)) e.Effect = DragDropEffects.Move; // SHIFT key else if (e.AllowedEffect.HasFlag(DragDropEffects.Copy)) e.Effect = DragDropEffects.Copy; else if (e.AllowedEffect.HasFlag(DragDropEffects.Move)) e.Effect = DragDropEffects.Move; else if (e.AllowedEffect.HasFlag(DragDropEffects.Link)) e.Effect = DragDropEffects.Link; } private void optimizerTreeView_DragDrop(object sender, DragEventArgs e) { Point coordinates = treeView.PointToClient(new Point(e.X, e.Y)); TreeNode node = treeView.GetNodeAt(coordinates); Algorithm algorithm = null; BatchRun batchRun = null; Experiment experiment = null; if (node == null) experiment = Content; else { algorithm = node.Tag as Algorithm; batchRun = node.Tag as BatchRun; experiment = node.Tag as Experiment; } var data = e.Data.GetData(HeuristicLab.Common.Constants.DragDropDataFormat); if (data is IProblem) { var problem = (IProblem)data; if (e.Effect.HasFlag(DragDropEffects.Copy)) problem = (IProblem)problem.Clone(); algorithm.Problem = problem; } else if (data is IOptimizer) { IOptimizer optimizer = (IOptimizer)data; if (e.Effect.HasFlag(DragDropEffects.Copy)) optimizer = (IOptimizer)optimizer.Clone(); if (batchRun != null) batchRun.Optimizer = optimizer; else if (experiment != null) experiment.Optimizers.Add(optimizer); else throw new NotSupportedException("Handling for specific type not implemented" + node.Tag.GetType()); } else if (data is IEnumerable) { IEnumerable optimizers = ((IEnumerable)data).Cast(); if (e.Effect.HasFlag(DragDropEffects.Copy)) { Cloner cloner = new Cloner(); optimizers = optimizers.Select(o => cloner.Clone(o)); } if (experiment != null) experiment.Optimizers.AddRange(optimizers); else throw new NotSupportedException("Handling for specific type not implemented" + node.Tag.GetType()); } } #endregion #region control events private void optimizerTreeView_NodeMouseDoubleClick(object sender, TreeNodeMouseClickEventArgs e) { if (rightClickOccured) return; if (e.X < e.Node.Bounds.Left || e.X > e.Node.Bounds.Right) return; e.Node.Toggle(); IContent optimizer = (IContent)e.Node.Tag; MainFormManager.MainForm.ShowContent(optimizer); } private void optimizerTreeview_NodeMouseClick(object sender, TreeNodeMouseClickEventArgs e) { if (e.X < e.Node.Bounds.Left || e.X > e.Node.Bounds.Right) return; treeView.SelectedNode = e.Node; detailsViewHost.Content = (IContent)e.Node.Tag; SetEnabledStateOfControls(); } private void optimizerTreeView_MouseDown(object sender, MouseEventArgs e) { // enables deselection of treeNodes if (e.Button != MouseButtons.Right) rightClickOccured = false; if (treeView.SelectedNode == null) return; Point coordinates = new Point(e.X, e.Y); TreeNode node = treeView.GetNodeAt(coordinates); if (node == null || coordinates.X < node.Bounds.Left || coordinates.X > node.Bounds.Right) { treeView.SelectedNode = null; detailsViewHost.Content = null; SetEnabledStateOfControls(); } } private void optimizerTreeView_KeyDown(object sender, KeyEventArgs e) { if (ReadOnly) return; if (e.KeyCode != Keys.Delete) return; if (treeView.SelectedNode == null) return; if (!(treeView.SelectedNode.Tag is IOptimizer)) return; var treeNode = treeView.SelectedNode; var optimizer = (IOptimizer)treeNode.Tag; if (treeNode.Parent == null) Content.Optimizers.Remove(optimizer); else { var batchRun = treeNode.Parent.Tag as BatchRun; var experiment = treeNode.Parent.Tag as Experiment; if (batchRun != null) batchRun.Optimizer = null; else if (experiment != null) experiment.Optimizers.Remove(optimizer); else throw new NotSupportedException("Handling for specific type not implemented" + treeView.SelectedNode.Tag.GetType()); } SetEnabledStateOfControls(); UpdateDetailsViewHost(); RebuildImageList(); } private bool rightClickOccured = true; private TreeNode toolStripMenuNode = null; private void optimizerTreeView_RightClick(object sender, EventArgs e) { rightClickOccured = true; Point coordinates = treeView.PointToClient(Cursor.Position); toolStripMenuNode = treeView.GetNodeAt(coordinates); if (toolStripMenuNode != null && coordinates.X >= toolStripMenuNode.Bounds.Left && coordinates.X <= toolStripMenuNode.Bounds.Right) { treeView.SelectedNode = toolStripMenuNode; detailsViewHost.Content = (IContent)toolStripMenuNode.Tag; SetEnabledStateOfControls(); ExpandToolStripMenuItem.Enabled = !toolStripMenuNode.IsExpanded && toolStripMenuNode.Nodes.Count > 0; ExpandToolStripMenuItem.Visible = !toolStripMenuNode.IsExpanded && toolStripMenuNode.Nodes.Count > 0; CollapseToolStripMenuItem.Enabled = toolStripMenuNode.IsExpanded; CollapseToolStripMenuItem.Visible = toolStripMenuNode.IsExpanded; EditNodeLabelToolStripMenuItem.Enabled = toolStripMenuNode.Tag != null && toolStripMenuNode.Tag is INamedItem; EditNodeLabelToolStripMenuItem.Visible = toolStripMenuNode.Tag != null && toolStripMenuNode.Tag is INamedItem; if (contextMenuStrip.Items.Cast().Any(item => item.Enabled)) contextMenuStrip.Show(Cursor.Position); } } private void ExpandToolStripMenuItem_Click(object sender, EventArgs e) { if (toolStripMenuNode != null) toolStripMenuNode.ExpandAll(); } private void CollapseToolStripMenuItem_Click(object sender, EventArgs e) { if (toolStripMenuNode != null) toolStripMenuNode.Collapse(); } private void EditNodeLabelToolStripMenuItem_Click(object sender, EventArgs e) { if (toolStripMenuNode != null) { if (!toolStripMenuNode.IsEditing) toolStripMenuNode.BeginEdit(); } } private void treeView_BeforeLabelEdit(object sender, NodeLabelEditEventArgs e) { if (Locked) e.CancelEdit = true; if (ReadOnly) e.CancelEdit = true; if (e.Node.Tag == null) e.CancelEdit = true; if (!(e.Node.Tag is NamedItem)) e.CancelEdit = true; } private void treeView_AfterLabelEdit(object sender, NodeLabelEditEventArgs e) { if (e.Label == null) return; e.Node.EndEdit(false); var namedItem = (INamedItem)e.Node.Tag; namedItem.Name = e.Label; } private void addButton_Click(object sender, System.EventArgs e) { if (typeSelectorDialog == null) typeSelectorDialog = new TypeSelectorDialog(); IAlgorithm algorithm = null; if (treeView.SelectedNode != null && (treeView.SelectedNode.Tag is IAlgorithm)) algorithm = (IAlgorithm)treeView.SelectedNode.Tag; if (algorithm == null) { typeSelectorDialog.Caption = "Select Optimizer"; typeSelectorDialog.TypeSelector.Caption = "Available Optimizers"; typeSelectorDialog.TypeSelector.Configure(typeof(IOptimizer), false, true); } else { typeSelectorDialog.Caption = "Select Problem"; typeSelectorDialog.TypeSelector.Caption = "Available Problems"; typeSelectorDialog.TypeSelector.Configure(algorithm.ProblemType, false, true); } if (typeSelectorDialog.ShowDialog(this) == DialogResult.OK) { try { if (algorithm == null) { IOptimizer optimizer = (IOptimizer)typeSelectorDialog.TypeSelector.CreateInstanceOfSelectedType(); if (treeView.SelectedNode == null) Content.Optimizers.Add(optimizer); else { var batchRun = treeView.SelectedNode.Tag as BatchRun; var experiment = treeView.SelectedNode.Tag as Experiment; if (batchRun != null) batchRun.Optimizer = optimizer; else if (experiment != null) experiment.Optimizers.Add(optimizer); else throw new NotSupportedException("Handling for specific type not implemented" + treeView.SelectedNode.Tag.GetType()); } } else { IProblem problem = (IProblem)typeSelectorDialog.TypeSelector.CreateInstanceOfSelectedType(); algorithm.Problem = problem; } } catch (Exception ex) { ErrorHandling.ShowErrorDialog(this, ex); } } } private void moveUpButton_Click(object sender, EventArgs e) { var optimizer = (IOptimizer)treeView.SelectedNode.Tag; Experiment experiment = null; if (treeView.SelectedNode.Parent == null) experiment = Content; else experiment = (Experiment)treeView.SelectedNode.Parent.Tag; int index = treeView.SelectedNode.Index; experiment.Optimizers.Reverse(index - 1, 2); treeView.SelectedNode = treeNodeTagMapping[optimizer].First(); SetEnabledStateOfControls(); UpdateDetailsViewHost(); RebuildImageList(); } private void moveDownButton_Click(object sender, EventArgs e) { var optimizer = (IOptimizer)treeView.SelectedNode.Tag; Experiment experiment = null; if (treeView.SelectedNode.Parent == null) experiment = Content; else experiment = (Experiment)treeView.SelectedNode.Parent.Tag; int index = treeView.SelectedNode.Index; experiment.Optimizers.Reverse(index, 2); treeView.SelectedNode = treeNodeTagMapping[optimizer].First(); SetEnabledStateOfControls(); UpdateDetailsViewHost(); RebuildImageList(); } private void removeButton_Click(object sender, EventArgs e) { var treeNode = treeView.SelectedNode; var optimizer = treeNode.Tag as IOptimizer; if (treeNode.Parent == null) Content.Optimizers.Remove(optimizer); else { var batchRun = treeNode.Parent.Tag as BatchRun; var experiment = treeNode.Parent.Tag as Experiment; if (batchRun != null) batchRun.Optimizer = null; else if (experiment != null) experiment.Optimizers.Remove(optimizer); else throw new NotSupportedException("Handling for specific type not implemented" + treeView.SelectedNode.Tag.GetType()); } SetEnabledStateOfControls(); UpdateDetailsViewHost(); RebuildImageList(); } private void showDetailsCheckBox_CheckedChanged(object sender, System.EventArgs e) { if (showDetailsCheckBox.Checked) { splitContainer.Panel2Collapsed = false; detailsGroupBox.Enabled = treeView.SelectedNode != null; detailsViewHost.Content = treeView.SelectedNode != null ? (IContent)treeView.SelectedNode.Tag : null; } else { splitContainer.Panel2Collapsed = true; detailsViewHost.Content = null; } } #endregion #region helpers private void UpdateDetailsViewHost() { if (treeView.SelectedNode != null) detailsViewHost.Content = (IContent)treeView.SelectedNode.Tag; else detailsViewHost.Content = null; } private TreeNode CreateTreeNode(IOptimizer optimizer) { TreeNode node = new TreeNode(optimizer.ToString()); node.Tag = optimizer; var algorithm = optimizer as IAlgorithm; if (algorithm != null) { foreach (TreeNode childNode in CreateAlgorithmChildNodes(algorithm)) node.Nodes.Add(childNode); } List nodes; if (!treeNodeTagMapping.TryGetValue(optimizer, out nodes)) { nodes = new List(); treeNodeTagMapping.Add(optimizer, nodes); RegisterNamedItemEvents(optimizer); } nodes.Add(node); foreach (TreeNode childNode in node.Nodes) { NamedItem namedItem = childNode.Tag as NamedItem; if (namedItem != null) { if (!treeNodeTagMapping.TryGetValue(namedItem, out nodes)) { nodes = new List(); treeNodeTagMapping.Add(namedItem, nodes); RegisterNamedItemEvents(namedItem); } nodes.Add(childNode); } } return node; } private IEnumerable CreateAlgorithmChildNodes(IAlgorithm algorithm) { TreeNode problemNode; if (algorithm.Problem != null) { problemNode = new TreeNode(algorithm.Problem.Name); problemNode.Tag = algorithm.Problem; } else { problemNode = new TreeNode("No Problem"); problemNode.Tag = null; } TreeNode parametersNode = new TreeNode("Parameters"); parametersNode.Tag = algorithm.Parameters; TreeNode resultsNode = new TreeNode("Results"); resultsNode.Tag = algorithm.Results; yield return problemNode; yield return parametersNode; yield return resultsNode; } private void DisposeTreeNode(TreeNode node) { var namedItem = node.Tag as INamedItem; if (namedItem == null) return; List nodes; if (!treeNodeTagMapping.TryGetValue(namedItem, out nodes)) throw new ArgumentException(); nodes.Remove(node); if (nodes.Count == 0) { treeNodeTagMapping.Remove(namedItem); DeregisterNamedItemEvents(namedItem); } } private IEnumerable IterateTreeNodes(TreeNode node = null) { TreeNodeCollection nodes; if (node == null) nodes = treeView.Nodes; else { nodes = node.Nodes; yield return node; } foreach (var childNode in nodes.OfType()) foreach (var n in IterateTreeNodes(childNode)) yield return n; } private void RebuildImageList() { if (InvokeRequired) { Invoke((Action)RebuildImageList); return; } treeView.BeginUpdate(); treeView.ImageList.Images.Clear(); var topLevelNodes = treeView.Nodes.OfType().ToArray(); var nodes = IterateTreeNodes().ToList(); treeView.Nodes.Clear(); foreach (TreeNode treeNode in nodes) { var item = (IItem)treeNode.Tag; treeView.ImageList.Images.Add(item == null ? HeuristicLab.Common.Resources.VSImageLibrary.Nothing : item.ItemImage); treeNode.ImageIndex = treeView.ImageList.Images.Count - 1; } treeView.Nodes.AddRange(topLevelNodes); treeView.EndUpdate(); } #endregion public sealed class CustomTreeView : System.Windows.Forms.TreeView { protected override void WndProc(ref System.Windows.Forms.Message m) { const int WM_RBUTTONDOWN = 0x204; if (m.Msg == WM_RBUTTONDOWN) { //Raise your custom event right click event to prevent node highlighting OnRightClick(); return; } base.WndProc(ref m); } public event EventHandler RightClick; private void OnRightClick() { var handler = RightClick; if (handler != null) RightClick(this, EventArgs.Empty); } } } }