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

Timestamp:

02/01/15 23:17:56 (10 years ago)

Author:

bburlacu

Message:

#1772: Added some useful functionality to the GenealogyGraphView.

File:

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

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

@@ -40 +40 @@
     private readonly Pen defaultPen;
 
+    private Dictionary<IGenealogyGraphNode, VisualGenealogyGraphNode> nodeMap;
+    private Dictionary<Tuple<VisualGenealogyGraphNode, VisualGenealogyGraphNode>, VisualGenealogyGraphArc> arcMap;
+
     public IGenealogyGraph GenealogyGraph {
       get { return genealogyGraph; }
@@ -63 +66 @@
     }
 
-    private Dictionary<IGenealogyGraphNode, VisualGenealogyGraphNode> nodeMap;
-    private Dictionary<Tuple<VisualGenealogyGraphNode, VisualGenealogyGraphNode>, VisualGenealogyGraphArc> arcMap;
+    public bool UpdateEnabled {
+      get { return Chart.UpdateEnabled; }
+      set { Chart.UpdateEnabled = value; }
+    }
+
+    public void EnforceUpdate() {
+      Chart.EnforceUpdate();
+    }
 
     #region chart modes
@@ -225 +234 @@
 
       // update
-      Chart.UpdateEnabled = true;
-      Chart.EnforceUpdate();
+      UpdateEnabled = true;
+      EnforceUpdate();
 
       if (SelectedVisualNode != null)
-        /* emit clicked event */
-        OnGenealogyGraphNodeClicked(SelectedVisualNode, e);
+        OnGenealogyGraphNodeClicked(SelectedVisualNode, e); // emit clicked event
 
       base.pictureBox_MouseUp(sender, e);
@@ -306 +314 @@
     }
 
+    public void ClearArcs() {
+      foreach (var primitive in Chart.Group.Primitives.OfType<VisualGenealogyGraphArc>()) {
+        primitive.Pen = Pens.Transparent;
+      }
+    }
+
+    public void ClearNodes() {
+      foreach (var primitive in Chart.Group.Primitives.OfType<VisualGenealogyGraphNode>()) {
+        primitive.Brush = new SolidBrush(Color.Transparent);
+        primitive.Pen = new Pen(Color.LightGray);
+      }
+    }
+
     public virtual void ClearPrimitives() {
       foreach (var primitive in Chart.Group.Primitives) {
@@ -313 +334 @@
           primitive.Brush = new SolidBrush(Color.Transparent);
           primitive.Pen = new Pen(Color.DarkGray);
-        }
-      }
-    }
-
-    public void HighlightNodes(IEnumerable<IGenealogyGraphNode> nodes) {
-      Chart.UpdateEnabled = false;
-      ClearPrimitives();
+          //          primitive.Pen = Pens.Transparent; // remove the node contour
+        }
+      }
+    }
+
+    public void HighlightNodes(IEnumerable<IGenealogyGraphNode> nodes, bool clearPrimitives = true) {
+      if (clearPrimitives)
+        ClearPrimitives();
+
       foreach (var node in nodes) {
         var graphNode = GetMappedNode(node);
         graphNode.Brush = new SolidBrush(node.GetColor());
       }
-      Chart.UpdateEnabled = true;
-      Chart.EnforceUpdate();
+    }
+
+    public void HighlightNodes(IEnumerable<IGenealogyGraphNode> nodes, Color color, bool clearPrimitives = true) {
+      if (clearPrimitives)
+        ClearPrimitives();
+
+      foreach (var node in nodes.Select(GetMappedNode)) {
+        node.Brush = new SolidBrush(color);
+      }
     }
 
     public void HighlightAll() {
-      Chart.UpdateEnabled = false;
       foreach (var visualNode in nodeMap.Values) {
         visualNode.Brush = new SolidBrush(visualNode.Data.GetColor());
@@ -340 +369 @@
         arc.Pen.Brush = new LinearGradientBrush(start, end, source.GetColor(), target.GetColor());
       }
-      Chart.UpdateEnabled = true;
-      Chart.EnforceUpdate();
     }