Changeset 10746 for branches/HeuristicLab.EvolutionTracking/HeuristicLab.Encodings.SymbolicExpressionTreeEncoding.Views

Timestamp:

04/14/14 17:15:32 (10 years ago)

Author:

bburlacu

Message:

#1772: Small improvements to FragmentGraphView, moved tracking classes to separate folder.

File:

• branches/HeuristicLab.EvolutionTracking/HeuristicLab.Encodings.SymbolicExpressionTreeEncoding.Views/3.4/SymbolicExpressionTreeTile.cs (modified) (8 diffs)

branches/HeuristicLab.EvolutionTracking/HeuristicLab.Encodings.SymbolicExpressionTreeEncoding.Views/3.4/SymbolicExpressionTreeTile.cs

@@ -23 +23 @@
 using System.Collections.Generic;
 using System.Drawing;
+using System.Drawing.Drawing2D;
 using System.Linq;
 using HeuristicLab.Common;
@@ -37 +38 @@
     public int PreferredNodeWidth { get; set; }
     public int PreferredNodeHeight { get; set; }
+    public ReingoldTilfordLayoutEngine<ISymbolicExpressionTreeNode> LayoutEngine { get; set; }
+    private readonly Dictionary<IPrimitive, ISymbolicExpressionTreeNode> primitivesToNodes;
+    private readonly Dictionary<ISymbolicExpressionTreeNode, IPrimitive> nodesToPrimitives;
 
     private Size size;
@@ -70 +74 @@
     }
 
+    public IPrimitive GetPrimitive(ISymbolicExpressionTreeNode node) {
+      IPrimitive primitive;
+      nodesToPrimitives.TryGetValue(node, out primitive);
+      return primitive;
+    }
+
+    public ISymbolicExpressionTreeNode GetExpressionTreeNode(IPrimitive primitive) {
+      ISymbolicExpressionTreeNode node;
+      primitivesToNodes.TryGetValue(primitive, out node);
+      return node;
+    }
+
     private ISymbolicExpressionTree symbolicExpressionTree;
     public ISymbolicExpressionTree SymbolicExpressionTree {
@@ -81 +97 @@
     new private void Clear() {
       Group.Clear();
-      primitiveMap.Clear();
+      primitivesToNodes.Clear();
+      nodesToPrimitives.Clear();
     }
 
@@ -92 +109 @@
       }
     }
-    public ReingoldTilfordLayoutEngine<ISymbolicExpressionTreeNode> LayoutEngine { get; set; }
-    private readonly Dictionary<IPrimitive, ISymbolicExpressionTreeNode> primitiveMap;
 
     public SymbolicExpressionTreeTile(IChart chart)
       : base(chart) {
-      primitiveMap = new Dictionary<IPrimitive, ISymbolicExpressionTreeNode>();
+      primitivesToNodes = new Dictionary<IPrimitive, ISymbolicExpressionTreeNode>();
+      nodesToPrimitives = new Dictionary<ISymbolicExpressionTreeNode, IPrimitive>();
       PreferredNodeWidth = 80;
       PreferredNodeHeight = 40;
@@ -124 +140 @@
         var upperRight = new PointD(visualNode.X + preferredNodeWidth, visualNode.Y + preferredNodeHeight);
         var node = visualNode.Content;
+
         RectangularPrimitiveBase rectangularPrimitive;
         if (node.SubtreeCount == 0) {
@@ -131 +148 @@
         }
 
-        primitiveMap.Add(rectangularPrimitive, node); // to be able to retrieve nodes via primitives
+        primitivesToNodes.Add(rectangularPrimitive, node); // to be able to retrieve nodes via primitives
+        nodesToPrimitives.Add(node, rectangularPrimitive);
         this.Add(rectangularPrimitive);
 
@@ -179 +197 @@
       }
     }
+
+    public override void Draw(Graphics graphics) {
+      graphics.SmoothingMode = SmoothingMode.HighQuality;
+      base.Draw(graphics);
+    }
   }
 }