Changeset 12225 for branches/HeuristicLab.EvolutionTracking/HeuristicLab.Problems.DataAnalysis.Symbolic.Views/3.4/Tracking/SymbolicDataAnalysisGenealogyGraphView.cs

Timestamp:

03/19/15 01:10:21 (9 years ago)

Author:

bburlacu

Message:

#1772: Work in progress for calculating sampling counts for subtrees in the population: updated TraceCalculator to aggregate tracing statistics, updated SymbolicDataAnalysisGenealogyGraphView, added SymbolicDataAnalysisSubtreeSampleCountAnalyzer.

File:

• branches/HeuristicLab.EvolutionTracking/HeuristicLab.Problems.DataAnalysis.Symbolic.Views/3.4/Tracking/SymbolicDataAnalysisGenealogyGraphView.cs (modified) (9 diffs)

branches/HeuristicLab.EvolutionTracking/HeuristicLab.Problems.DataAnalysis.Symbolic.Views/3.4/Tracking/SymbolicDataAnalysisGenealogyGraphView.cs

@@ -41 +41 @@
     private ISymbolicExpressionTreeNode selectedSubtree;
     private ISymbolicExpressionTreeNode SelectedSubtree {
-      get {
-        return selectedSubtree;
-      }
+      get { return selectedSubtree; }
       set {
-        if (selectedSubtree == value) return;
+        if (value == null || selectedSubtree == value) return;
         selectedSubtree = value;
         ClonedSubtree = (ISymbolicExpressionTreeNode)selectedSubtree.Clone();
       }
     }
-
+    // the clone is necessary in order to deselect parts of a tree (by removing subtrees) 
+    // and do "high level" matching of the remaining part
     private ISymbolicExpressionTreeNode ClonedSubtree { get; set; }
+
+    // we use the symbolic expression tree chart to call the drawing routines
+    // and highlight relevant tree parts
+    public SymbolicExpressionTreeChart SymbolicExpressionTreeChart {
+      get {
+        var view = (GraphicalSymbolicExpressionTreeView)base.viewHost.ActiveView;
+        return view == null ? null : view.SymbolicExpressionTreeChart;
+      }
+    }
 
     public SymbolicDataAnalysisGenealogyGraphView() {
@@ -61 +69 @@
     protected override void OnContentChanged() {
       base.OnContentChanged();
-      if (Content != null) { genealogyGraphChart.GenealogyGraph = Content; }
-    }
-    #endregion
-
-    public SymbolicExpressionTreeChart SymbolicExpressionTreeChart {
-      get {
-        var view = (GraphicalSymbolicExpressionTreeView)base.viewHost.ActiveView;
-        return view == null ? null : view.SymbolicExpressionTreeChart;
-      }
-    }
-
-    protected override void RegisterContentEvents() {
-      base.RegisterContentEvents();
-      if (SymbolicExpressionTreeChart != null)
-        SymbolicExpressionTreeChart.SymbolicExpressionTreeNodeClicked += treeChart_SymbolicExpressionTreeNodeClicked;
-    }
-
-    protected override void DeregisterContentEvents() {
-      if (SymbolicExpressionTreeChart != null)
-        SymbolicExpressionTreeChart.SymbolicExpressionTreeNodeClicked -= treeChart_SymbolicExpressionTreeNodeClicked;
-      base.DeregisterContentEvents();
+      if (Content != null && Content != genealogyGraphChart.GenealogyGraph) {
+        genealogyGraphChart.GenealogyGraph = Content;
+      }
     }
 
@@ -110 +100 @@
         var fragment = data as IFragment<ISymbolicExpressionTreeNode>;
         var td = data as TraceData;
+        //        double max = Content.Vertices.SelectMany(x => x.Data.IterateNodesPrefix()).Max(x => x.NodeWeight);
+
         if (fragment != null) {
-          treeChart_HighlightSubtree(graphNode.Data.NodeAt(fragment.Index1));
+          //          treeChart_HighlightSubtree(graphNode.Data.NodeAt(fragment.Index1));
+          var nodes = graphNode.Data.IterateNodesPrefix().ToList();
+          int c = ColorGradient.Colors.Count - 1;
+          foreach (var n in nodes.Skip(1)) { // skip program root symbol 
+            //            var i = (int)Math.Round(n.NodeWeight / max * c);
+            //            const int alpha = 50;
+            //            var fillColor = Color.FromArgb(alpha, ColorGradient.Colors[i]);
+
+            //            treeChart_HighlightSubtree(n, null, fillColor);
+          }
         } else if (td != null) {
+
           var nodes = graphNode.Data.IterateNodesPrefix().ToList();
+          int c = ColorGradient.Colors.Count - 1;
+          foreach (var n in nodes.Skip(1)) { // skip program root symbol 
+            //            var i = (int)Math.Round(n.NodeWeight / max * c);
+            //            const int alpha = 50;
+            //            var fillColor = Color.FromArgb(alpha, ColorGradient.Colors[i]);
+
+            //            treeChart_HighlightSubtree(n, null, fillColor);
+          }
+
           treeChart_HighlightSubtree(nodes[td.LastSubtreeIndex], Color.Orange);
-          treeChart_HighlightSubtree(nodes[td.LastFragmentIndex], null, Color.PowderBlue);
+          treeChart_HighlightSubtree(nodes[td.LastFragmentIndex], Color.CornflowerBlue);
         }
       }
@@ -125 +136 @@
       // highlight the selected subtree inside the displayed tree on the right hand side
       treeChart_ClearColors();
-      treeChart_HighlightSubtree(subtree);
+      treeChart_HighlightSubtree(subtree, Color.Black, Color.RoyalBlue);
 
       bool trace = genealogyGraphChart.TraceFragments; // check whether we are in 'trace' or 'match' mode
@@ -136 +147 @@
             var graphNode = (IGenealogyGraphNode<ISymbolicExpressionTree>)genealogyGraphChart.SelectedGraphNode;
             var subtreeIndex = graphNode.Data.IterateNodesPrefix().ToList().IndexOf(subtree);
-            var traceGraph = TraceCalculator.TraceSubtree(graphNode, subtreeIndex);
+            var traceGraph = TraceCalculator.TraceSubtree(graphNode, subtreeIndex, updateVertexWeights: true, updateSubtreeWeights: true, cacheTraceNodes: true);
             if (traceGraph.Vertices.Any()) {
               genealogyGraphChart.UpdateEnabled = false;
-              genealogyGraphChart.ClearArcs();
-              var genealogyNodes = traceGraph.Vertices.Select(v => Content.GetByContent(v.Data));
-              genealogyGraphChart.HighlightNodes(genealogyNodes, Color.Black, false);
+              genealogyGraphChart.ClearPrimitives();
+              var traceVertices = traceGraph.Vertices.ToList();
+              var genealogyNodes = traceVertices.Select(v => Content.GetByContent(v.Data)).ToList();
+              // color graph vertices according to their weight
+              double max = traceGraph.Vertices.Max(x => x.Data.IterateNodesPrefix().Max(n => x.Weight));
+              for (int i = 0; i < genealogyNodes.Count; ++i) {
+                var g = genealogyNodes[i];
+                g.Weight = traceVertices[i].Weight;
+                var v = genealogyGraphChart.GetMappedNode(g);
+                if (v != null) {
+                  v.Brush = new SolidBrush(ColorGradient.Colors[(int)Math.Round(g.Weight * (ColorGradient.Colors.Count - 1) / max)]);
+                }
+              }
+
               genealogyGraphChart.UpdateEnabled = true;
               genealogyGraphChart.EnforceUpdate();
+
               if (openNew_CheckBox.Checked)
                 MainFormManager.MainForm.ShowContent(traceGraph); // display the fragment graph on the screen
-
-              var max = traceGraph.Vertices.Max(x => x.Weight);
-
-              genealogyGraphChart.UpdateEnabled = false;
-              foreach (var traceNode in traceGraph.Vertices) {
-                var g = Content.GetByContent(traceNode.Data);
-                g.Weight = traceNode.Weight;
-                var v = genealogyGraphChart.GetMappedNode(g);
-                if (v != null) {
-                  int i = (int)Math.Round(g.Weight * (ColorGradient.Colors.Count - 1) / max);
-
-                  v.Brush = new SolidBrush(ColorGradient.Colors[i]);
-                }
-              }
-              genealogyGraphChart.UpdateEnabled = true;
-              genealogyGraphChart.EnforceUpdate();
             }
           } else {
@@ -188 +195 @@
           break;
       }
+    }
+    #endregion
+
+    protected override void RegisterContentEvents() {
+      base.RegisterContentEvents();
+      if (SymbolicExpressionTreeChart != null)
+        SymbolicExpressionTreeChart.SymbolicExpressionTreeNodeClicked += treeChart_SymbolicExpressionTreeNodeClicked;
+    }
+
+    protected override void DeregisterContentEvents() {
+      if (SymbolicExpressionTreeChart != null)
+        SymbolicExpressionTreeChart.SymbolicExpressionTreeNodeClicked -= treeChart_SymbolicExpressionTreeNodeClicked;
+      base.DeregisterContentEvents();
     }
 
@@ -210 +230 @@
 
     private void treeChart_HighlightSubtree(ISymbolicExpressionTreeNode subtree, Color? lineColor = null, Color? fillColor = null) {
-      var lc = lineColor ?? Color.RoyalBlue;
-      var fc = fillColor ?? Color.White;
+      if (lineColor == null && fillColor == null)
+        return;
+
       foreach (var s in subtree.IterateNodesPrefix()) {
         var visualNode = SymbolicExpressionTreeChart.GetVisualSymbolicExpressionTreeNode(s);
-        visualNode.LineColor = lc;
-        visualNode.FillColor = fc;
-
-        foreach (var c in s.Subtrees) {
-          var visualArc = SymbolicExpressionTreeChart.GetVisualSymbolicExpressionTreeNodeConnection(s, c);
-          visualArc.LineColor = lc;
+        if (lineColor != null) {
+          visualNode.LineColor = (Color)lineColor;
+          foreach (var c in s.Subtrees) {
+            var visualArc = SymbolicExpressionTreeChart.GetVisualSymbolicExpressionTreeNodeConnection(s, c);
+            visualArc.LineColor = (Color)lineColor;
+          }
         }
+        if (fillColor != null)
+          visualNode.FillColor = (Color)fillColor;
       }
       SymbolicExpressionTreeChart.RepaintNodes();
     }
 
+    #region navigate the genealogy / trace graph
     private void navigateLeftButton_Click(object sender, EventArgs e) {
       var graphNode = (IGenealogyGraphNode<ISymbolicExpressionTree>)genealogyGraphChart.SelectedGraphNode;
@@ -267 +291 @@
       }
     }
+    #endregion
   }
+
+  // bogus class needed in order to be able to "design" the view
   public class SymbolicDataAnalysisGenealogyGraphViewDesignable : GenealogyGraphView<ISymbolicExpressionTree> { }
 
+  #region helper methods
   internal static class Util {
     internal static ISymbolicExpressionTreeNode NodeAt(this ISymbolicExpressionTree tree, int position) {
@@ -278 +306 @@
     }
   }
+  #endregion
 }