Changeset 10269 for branches/HeuristicLab.EvolutionTracking/HeuristicLab.EvolutionTracking.Views/3.4/GenealogyGraphView.cs

Timestamp:

12/23/13 11:42:24 (10 years ago)

Author:

bburlacu

Message:

#1772: Added HeuristicLab.Problems.DataAnalysis.Symbolic and HeuristicLab.Problems.DataAnalysis.Symbolic.Views and integrated some modifications from the old branch.

File:

• branches/HeuristicLab.EvolutionTracking/HeuristicLab.EvolutionTracking.Views/3.4/GenealogyGraphView.cs (modified) (8 diffs)

branches/HeuristicLab.EvolutionTracking/HeuristicLab.EvolutionTracking.Views/3.4/GenealogyGraphView.cs

@@ -21 +21 @@
 
 using System;
-using System.Collections.Generic;
-using System.Drawing;
 using System.Linq;
 using System.Windows.Forms;
 using HeuristicLab.Core.Views;
-using HeuristicLab.Encodings.SymbolicExpressionTreeEncoding;
-using HeuristicLab.Encodings.SymbolicExpressionTreeEncoding.Views;
 using HeuristicLab.MainForm;
-using HeuristicLab.Problems.DataAnalysis.Symbolic;
 
 namespace HeuristicLab.EvolutionaryTracking.Views {
-  [View("SymbolicExpressionTreeGenealogyGraph")]
-  [Content(typeof(SymbolicExpressionTreeGenealogyGraph), IsDefaultView = true)]
+  [View("GenealogyGraph")]
+  [Content(typeof(GenealogyGraph), IsDefaultView = true)]
   public sealed partial class GenealogyGraphView : ItemView {
-    private VisualSymbolicExpressionTreeNode selectedVisualSymbExprTreeNode;
-    private SymbolicExpressionTreeNodeSimilarityComparer comparer;
-    private Dictionary<ISymbolicExpressionTreeNode, ISymbolicExpressionTreeNode> clonedNodes;
-
-    public new SymbolicExpressionTreeGenealogyGraph Content {
-      get { return (SymbolicExpressionTreeGenealogyGraph)base.Content; }
+    public new GenealogyGraph Content {
+      get { return (GenealogyGraph)base.Content; }
       set { base.Content = value; }
     }
@@ -51 +42 @@
       this.simpleLineagesCheckBox.Image = Common.Resources.VSImageLibrary.ArrowDown;
       this.lockGenealogyCheckBox.Image = Common.Resources.VSImageLibrary.ProtectForm;
-
-      comparer = new SymbolicExpressionTreeNodeSimilarityComparer {
-        MatchConstantValues = true,
-        MatchVariableWeights = true,
-        MatchVariableNames = true
-      };
     }
 
     protected override void DeregisterContentEvents() {
       // TODO: Deregister your event handlers here
-      genealogyGraphChart.GenealogyGraphNodeClicked -= graphChart_GenealogyGraphNodeClicked;
-      symbolicExpressionTreeChart.SymbolicExpressionTreeNodeClicked -= treeChart_SymbolicExpressionTreeNodeClicked;
-      symbolicExpressionTreeChart.SymbolicExpressionTreeNodeDoubleClicked -= treeChart_SymbolicExpressionTreeNodeDoubleClicked;
-      Content.GraphUpdated -= genealogyGraphUpdated;
+      Content.GraphUpdated -= GenealogyGraphUpdated;
       base.DeregisterContentEvents();
     }
@@ -71 +53 @@
       base.RegisterContentEvents();
       // TODO: Register your event handlers here
-      genealogyGraphChart.GenealogyGraphNodeClicked += graphChart_GenealogyGraphNodeClicked;
-      symbolicExpressionTreeChart.SymbolicExpressionTreeNodeClicked += treeChart_SymbolicExpressionTreeNodeClicked;
-      symbolicExpressionTreeChart.SymbolicExpressionTreeNodeDoubleClicked += treeChart_SymbolicExpressionTreeNodeDoubleClicked;
-      Content.GraphUpdated += genealogyGraphUpdated;
+      Content.GraphUpdated += GenealogyGraphUpdated;
     }
 
@@ -81 +60 @@
     protected override void OnContentChanged() {
       base.OnContentChanged();
-      if (Content == null)
-        genealogyGraphChart.Graph = null;
-      else {
-        genealogyGraphChart.Graph = Content;
-        //        var best = Content.Nodes.OrderByDescending(x => x.Quality).First();
-        //        symbolicExpressionTreeChart.Tree = best.SymbolicExpressionTree;
-      }
+      genealogyGraphChart.GenealogyGraph = Content ?? null;
     }
-
     private void Content_GraphChanged(object sender, EventArgs e) {
     }
-
     protected override void SetEnabledStateOfControls() {
       base.SetEnabledStateOfControls();
@@ -100 +71 @@
 
     #region Event Handlers (child controls)
-
-    // TODO: Put event handlers of child controls here.
-    private void graphChart_GenealogyGraphNodeClicked(object sender, MouseEventArgs e) {
-      // the type hierarchy goes like this: VisualGenealogyGraphNode --> GenealogyGraphNode --> SymbolicExpressionTree
-      var visualGenealogyGraphNode = (VisualGenealogyGraphNode)sender;
-      var genealogyGraphNode = visualGenealogyGraphNode.Data;
-      symbolicExpressionTreeChart.Tree = genealogyGraphNode.SymbolicExpressionTree;
-
-      if (genealogyGraphNode.InEdges == null) return;
-      var arc = genealogyGraphNode.InEdges.Last(x => x.Source != x.Target);
-      if (arc == null || arc.Data == null) return;
-      var fragment = (IFragment)arc.Data;
-      if (fragment.Root == null) return;
-      foreach (var node in fragment.Root.IterateNodesBreadth()) {
-        var visualSymbExprTreeNode = symbolicExpressionTreeChart.GetVisualSymbolicExpressionTreeNode(node);
-        visualSymbExprTreeNode.LineColor = Color.OrangeRed;
-      }
-      symbolicExpressionTreeChart.Repaint();
+    private void GenealogyGraphUpdated(object sender, EventArgs args) {
+      genealogyGraphChart.GenealogyGraph = Content;
     }
-
-    private void genealogyGraphUpdated(object sender, EventArgs args) {
-      genealogyGraphChart.Graph = Content;
-    }
-
-    private void treeChart_SymbolicExpressionTreeNodeClicked(object sender, MouseEventArgs e) {
-      genealogyGraphChart.Chart.UpdateEnabled = false;
-      switch (e.Button) {
-        case MouseButtons.Left: {
-            selectedVisualSymbExprTreeNode = (VisualSymbolicExpressionTreeNode)sender;
-            if (selectedVisualSymbExprTreeNode == null) return;
-            ISymbolicExpressionTreeNode selectedSubtree = selectedVisualSymbExprTreeNode.SymbolicExpressionTreeNode;
-            var clonedSubtree = (ISymbolicExpressionTreeNode)selectedSubtree.Clone();
-            // map original nodes to cloned nodes
-            clonedNodes = selectedSubtree.IterateNodesPrefix()
-                                         .Zip(clonedSubtree.IterateNodesPrefix(), (original, clone) => new { original, clone })
-                                         .ToDictionary(p => p.original, p => p.clone);
-
-            foreach (var node in symbolicExpressionTreeChart.Tree.IterateNodesPrefix()) {
-              symbolicExpressionTreeChart.GetVisualSymbolicExpressionTreeNode(node).FillColor = Color.Transparent;
-            }
-            foreach (var node in selectedSubtree.IterateNodesPrefix()) {
-              symbolicExpressionTreeChart.GetVisualSymbolicExpressionTreeNode(node).FillColor = Color.LightBlue;
-            }
-            break;
-          }
-        case MouseButtons.Middle: {
-            var visualNode = (VisualSymbolicExpressionTreeNode)sender;
-            if (selectedVisualSymbExprTreeNode == null || selectedVisualSymbExprTreeNode == visualNode) return;
-            var selectedNode = visualNode.SymbolicExpressionTreeNode;
-            if (clonedNodes.ContainsKey(selectedNode)) {
-              var selectedClone = clonedNodes[selectedNode];
-              var parent = selectedClone.Parent;
-              if (parent == null) break;
-              int index = parent.IndexOfSubtree(selectedClone);
-              if (index == -1) break;
-              parent.RemoveSubtree(parent.IndexOfSubtree(selectedClone));
-              foreach (var node in selectedNode.IterateNodesPrefix()) {
-                symbolicExpressionTreeChart.GetVisualSymbolicExpressionTreeNode(node).FillColor = Color.Transparent;
-              }
-            }
-            break;
-          }
-      }
-      if (selectedVisualSymbExprTreeNode == null || clonedNodes == null) return;
-      // update visual graph nodes that contain matching trees
-      MatchNodesAndRepaint();
-      // refresh the tree chart
-      symbolicExpressionTreeChart.Repaint();
-    }
-
     void MatchNodesAndRepaint() {
       genealogyGraphChart.Chart.UpdateEnabled = false;
-      genealogyGraphChart.ClearAllNodes(); // clear node colors
-      var fragment = new Fragment { Root = clonedNodes[selectedVisualSymbExprTreeNode.SymbolicExpressionTreeNode] };
-      var fragmentLength = fragment.Length;
-      // highlight nodes
-      genealogyGraphChart.HighlightNodes(from node in genealogyGraphChart.Graph.Nodes
-                                         let tree = node.SymbolicExpressionTree
-                                         where tree.Length >= fragmentLength
-                                         where tree.Root.ContainsFragment(fragment, comparer)
-                                         select tree);
-      // highlight edges that contain a matching fragment
-      var fragmentSimilarityComparer = new SymbolicExpressionTreeFragmentSimilarityComparer {
-        SimilarityComparer = comparer
-      };
-      var matchingEdges = genealogyGraphChart.Graph.Nodes.Where(n => n.InEdges != null)
-                           .SelectMany(n => n.InEdges).Where(edge => edge.Data != null && ((IFragment)edge.Data).Root != null)
-                           .Where(edge => fragmentSimilarityComparer.Equals(fragment, (IFragment)edge.Data));
-      genealogyGraphChart.HighlightArcs(matchingEdges, Color.DodgerBlue);
-
+      genealogyGraphChart.ClearPrimitives(); // clear node colors
 
       genealogyGraphChart.Chart.UpdateEnabled = true;
@@ -198 +85 @@
     }
     #endregion
-
-    private void matchConstantsCheckBox_CheckedChanged(object sender, EventArgs e) {
-      comparer.MatchConstantValues = matchConstantsCheckBox.Checked;
-      if (selectedVisualSymbExprTreeNode != null)
-        MatchNodesAndRepaint();
-    }
-
-    private void matchVariableWeightsCheckBox_CheckedChanged(object sender, EventArgs e) {
-      comparer.MatchVariableWeights = matchVariableWeightsCheckBox.Checked;
-      if (selectedVisualSymbExprTreeNode != null)
-        MatchNodesAndRepaint();
-    }
-
-    private void matchVariableNamesCheckBox_CheckedChanged(object sender, EventArgs e) {
-      comparer.MatchVariableNames = matchVariableNamesCheckBox.Checked;
-      if (selectedVisualSymbExprTreeNode != null)
-        MatchNodesAndRepaint();
-    }
-
     private void simpleLineagesCheckBox_CheckedChanged(object sender, EventArgs e) {
       genealogyGraphChart.SimpleLineages = simpleLineagesCheckBox.Checked;
@@ -222 +90 @@
 
     private void highlightAllButton_Click(object sender, EventArgs e) {
-      symbolicExpressionTreeChart.Repaint();
-      genealogyGraphChart.HighlightAll();
     }
 
@@ -231 +97 @@
 
     private void highlightRootParentsButton_Click(object sender, EventArgs e) {
-      var nodes = genealogyGraphChart.Graph.Nodes.Where(n => n.InEdges != null && n.InEdges.Count == 2).Select(n => (SymbolicExpressionTreeGenealogyGraphNode)n.InEdges[0].Source).ToList();
+      var nodes = genealogyGraphChart.GenealogyGraph.Nodes.Where(n => n.InArcs != null && n.InArcs.Count == 2).Select(n => (GenealogyGraphNode)n.InArcs[0].Source).ToList();
       genealogyGraphChart.HighlightNodes(nodes);
     }
 
     private void highlightSecondaryParentsButton_Click(object sender, EventArgs e) {
-      var nodes = genealogyGraphChart.Graph.Nodes.Where(n => n.InEdges != null && n.InEdges.Count == 2).Select(n => (SymbolicExpressionTreeGenealogyGraphNode)n.InEdges[1].Source).ToList();
+      var nodes = genealogyGraphChart.GenealogyGraph.Nodes.Where(n => n.InArcs != null && n.InArcs.Count == 2).Select(n => (GenealogyGraphNode)n.InArcs[1].Source).ToList();
       genealogyGraphChart.HighlightNodes(nodes);
     }