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

Timestamp:

04/08/14 23:54:27 (10 years ago)

Author:

bburlacu

Message:

#1772: Some improvements on the way the genealogy graph and the lineages are drawn.

File:

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

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

@@ -111 +111 @@
 
         foreach (var node in nodes) {
-          var pen = new Pen(Color.LightGray);
-
-          var visualNode = new VisualGenealogyGraphNode(Chart, x, y, x + diameter, y + diameter, pen, null) {
+          var pen = new Pen(Color.DarkGray);
+          var brush = new SolidBrush(node.GetColor());
+
+          var visualNode = new VisualGenealogyGraphNode(Chart, x, y, x + diameter, y + diameter, pen, brush) {
             Data = node,
             ToolTipText = "Rank: " + node.Rank + nl +
@@ -129 +130 @@
 
       // add arcs
-      foreach (var node in GenealogyGraph.Nodes.Cast<IGenealogyGraphNode>()) {
-        if (node.InArcs == null) continue;
-        var visualNode = nodeMap[node];
+      foreach (var node in GenealogyGraph.Nodes) {
+        if (!node.InArcs.Any()) continue;
+        var visualNode = GetMappedNode(node);
         if (visualNode == null) continue;
         foreach (var arc in node.InArcs) {
@@ -184 +185 @@
       }
       var visualNodes = Chart.GetAllPrimitives(e.Location).Where(p => p is VisualGenealogyGraphNode).ToList();
-      if (visualNodes.Count <= 0) {
+      if (!visualNodes.Any()) {
         SelectedVisualNode = null;
         return;
@@ -201 +202 @@
         }
         // use a rectangle to highlight the currently selected genealogy graph node
-        var gNode = SelectedVisualNode.Data;
-        var visualNode = GetMappedNode(gNode);
+        var graphNode = SelectedVisualNode.Data;
+        var visualNode = GetMappedNode(graphNode);
 
         DrawLineage(visualNode, n => SimpleLineages ? n.IncomingArcs.Take(1) : n.IncomingArcs, a => a.Source);
-        visualNode.Brush = null;
         DrawLineage(visualNode, n => n.OutgoingArcs, a => a.Target);
       }
@@ -224 +224 @@
 
     private static void DrawLineage(VisualGenealogyGraphNode node, Func<VisualGenealogyGraphNode, IEnumerable<VisualGenealogyGraphArc>> arcSelector, Func<VisualGenealogyGraphArc, VisualGenealogyGraphNode> nodeSelector) {
-      if (node.Brush != null) return;
-      node.Brush = new SolidBrush(node.Data.GetColor());
+      var brush = (SolidBrush)node.Brush;
+      brush.Color = node.Data.GetColor();
       var arcs = arcSelector(node).ToList();
       foreach (var arc in arcs) {
@@ -267 +267 @@
         if (primitive is VisualGenealogyGraphArc) {
           primitive.Pen.Brush = new SolidBrush(Color.Transparent);
-        } else {
-          primitive.Brush = null;
+        } else if (primitive is VisualGenealogyGraphNode) {
+          var brush = (SolidBrush)primitive.Brush;
+          brush.Color = Color.Transparent;
+          primitive.Pen.Color = Color.DarkGray;
         }
       }
@@ -303 +305 @@
   internal static class Util {
     public static Color GetColor(this IGenealogyGraphNode node) {
-      var colorIndex = (int)(node.Quality * ColorGradient.Colors.Count);
-      return colorIndex < ColorGradient.Colors.Count ? ColorGradient.Colors[colorIndex] : Color.Transparent;
+      var colorIndex = (int)Math.Round(node.Quality * ColorGradient.Colors.Count);
+      if (colorIndex >= ColorGradient.Colors.Count) return ColorGradient.Colors.Last();
+      return ColorGradient.Colors[colorIndex];
     }
   }