Changeset 15561 for branches/HeuristicLab.EvolutionTracking/HeuristicLab.EvolutionTracking.Views/3.4/GenealogyGraphChart.cs

Timestamp:

12/29/17 16:43:29 (6 years ago)

Author:

bburlacu

Message:

#1772: Refactor SymbolicDataAnalysisGenealogyGraphView

File:

• branches/HeuristicLab.EvolutionTracking/HeuristicLab.EvolutionTracking.Views/3.4/GenealogyGraphChart.cs (modified) (12 diffs)

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

@@ -32 +32 @@
   public partial class GenealogyGraphChart : ChartControl {
     private IGenealogyGraph genealogyGraph;
-    private const double XIncrement = 30;
-    private const double YIncrement = 30;
+    private const double XIncrement = 25;
+    private const double YIncrement = 25;
     private const double Diameter = 20;
     private readonly Brush defaultBrush;
@@ -44 +44 @@
     public bool SimpleLineages { get; set; }
     public bool LockGenealogy { get; set; }
-    public bool TraceFragments { get; set; }
+    public Func<IGenealogyGraphNode, Color> ColorSelector { get; set; }
     #endregion
 
@@ -98 +98 @@
     }
 
-    public GenealogyGraphChart() : base(new Chart(0, 0, 800, 600)) {
+    public GenealogyGraphChart() : base(new GridlessChart(0, 0, 800, 600)) {
       InitializeComponent();
       AddChartModes(new PanChartMode(this), new ZoomInChartMode(this), new ZoomOutChartMode(this), new SelectChartMode(this));
       defaultBrush = new SolidBrush(Color.Transparent);
       defaultPen = new Pen(Color.DarkGray);
+
+      ColorSelector = node => ColorGradient.Colors[(int)Math.Round(node.Quality * 255)];
     }
 
@@ -114 +116 @@
 
       foreach (var rank in ranks) {
+        var rect = new Visualization.Rectangle(Chart, new PointD(x, y), new PointD(x + diameter, y + diameter));
+        var font = new Font(FontFamily.GenericSansSerif, 12);
+        var genLabel = new LabeledPrimitive(rect, rank.Rank.ToString(), font);
+        Chart.Group.Add(genLabel);
+
+        x += xIncrement;
+
         var nodes = rank.Nodes.ToList();
         nodes.Sort((a, b) => b.CompareTo(a)); // sort descending by quality
@@ -119 +128 @@
 
         foreach (var node in nodes) {
-          var brush = new SolidBrush(node.GetColor());
+          var brush = new SolidBrush(ColorSelector(node));
           var visualNode = new VisualGenealogyGraphNode(Chart, x, y, x + diameter, y + diameter, defaultPen, brush) {
             Data = node,
@@ -199 +208 @@
         var visualNode = GetMappedNode(graphNode);
 
-        DrawLineage(visualNode, n => SimpleLineages ? n.IncomingArcs.Take(1) : n.IncomingArcs, a => a.Source);
+        DrawLineage(visualNode, n => SimpleLineages ? n.IncomingArcs.Take(1) : n.IncomingArcs, a => a.Source, ColorSelector);
         ((SolidBrush)visualNode.Brush).Color = Color.Transparent;
-        DrawLineage(visualNode, n => n.OutgoingArcs, a => a.Target);
+        DrawLineage(visualNode, n => n.OutgoingArcs, a => a.Target, ColorSelector);
       }
 
@@ -212 +221 @@
 
     #region drawing routines
-    private static void DrawLineage(VisualGenealogyGraphNode node, Func<VisualGenealogyGraphNode, IEnumerable<VisualGenealogyGraphArc>> arcSelector, Func<VisualGenealogyGraphArc, VisualGenealogyGraphNode> nodeSelector) {
+    private static void DrawLineage(VisualGenealogyGraphNode node, Func<VisualGenealogyGraphNode, IEnumerable<VisualGenealogyGraphArc>> arcSelector, Func<VisualGenealogyGraphArc, VisualGenealogyGraphNode> nodeSelector, Func<IGenealogyGraphNode, Color> getNodeColor) {
       var brush = (SolidBrush)node.Brush;
       if (brush.Color != Color.Transparent) return; // this lineage was already drawn (avoid redrawing common ancestors)
-      brush.Color = node.Data.GetColor();
+      brush.Color = getNodeColor(node.Data);
       var arcs = arcSelector(node);
       var pen = new Pen(Color.Transparent);
@@ -224 +233 @@
         var end = new Point((int)arc.End.X, (int)arc.End.Y);
         arc.Pen = pen;
-        arc.Pen.Brush = new LinearGradientBrush(start, end, source.GetColor(), target.GetColor());
-        DrawLineage(nodeSelector(arc), arcSelector, nodeSelector);
+        arc.Pen.Brush = new LinearGradientBrush(start, end, getNodeColor(source), getNodeColor(target));
+        DrawLineage(nodeSelector(arc), arcSelector, nodeSelector, getNodeColor);
       }
     }
@@ -287 +296 @@
       foreach (var node in nodes) {
         var graphNode = GetMappedNode(node);
-        graphNode.Brush = new SolidBrush(node.GetColor());
+        graphNode.Brush = new SolidBrush(ColorSelector(node));
       }
     }
@@ -302 +311 @@
     public void HighlightAll() {
       foreach (var visualNode in nodeMap.Values) {
-        visualNode.Brush = new SolidBrush(visualNode.Data.GetColor());
-      }
+        visualNode.Brush = new SolidBrush(ColorSelector(visualNode.Data));
+        Chart.IntoForeground(visualNode);
+      }
+      // draw arcs first and then nodes
       foreach (var arc in arcMap.Values) {
         var source = arc.Source.Data;
@@ -309 +320 @@
         var start = new Point((int)arc.Start.X, (int)arc.Start.Y);
         var end = new Point((int)arc.End.X, (int)arc.End.Y);
-        arc.Pen.Brush = new LinearGradientBrush(start, end, source.GetColor(), target.GetColor());
+        arc.Pen = new Pen(new LinearGradientBrush(start, end, ColorSelector(source), ColorSelector(target)));
+        Chart.IntoBackground(arc);
       }
     }
@@ -319 +331 @@
       var end = new Point((int)arc.End.X, (int)arc.End.Y);
       arc.Pen = new Pen(Color.Transparent);
-      arc.Pen.Brush = new LinearGradientBrush(start, end, source.GetColor(), target.GetColor());
+      arc.Pen = new Pen(new LinearGradientBrush(start, end, ColorSelector(source), ColorSelector(target)));
+      Chart.IntoBackground(arc);
     }
     #endregion
-  }
-
-  internal static class Util {
-    public static Color GetColor(this IGenealogyGraphNode node) {
-      if (double.IsNaN(node.Quality))
-        return ColorGradient.Colors[0];
-      var colorIndex = (int)Math.Round(node.Quality * ColorGradient.Colors.Count);
-      if (colorIndex >= ColorGradient.Colors.Count) return ColorGradient.Colors.Last();
-      return ColorGradient.Colors[colorIndex];
+
+    // workaround to disable the chart grid
+    private class GridlessChart : Chart {
+      public GridlessChart(PointD lowerLeft, PointD upperRight) : base(lowerLeft, upperRight) {
+        Grid = null;
+      }
+
+      public GridlessChart(double x1, double y1, double x2, double y2) : this(new PointD(x1, y1), new PointD(x2, y2)) { }
     }
   }