Changeset 8248 for branches/HeuristicLab.EvolutionaryTracking/HeuristicLab.EvolutionaryTracking/3.4/SymbolicExpressionTreeGenealogyGraph.cs

Timestamp:

07/06/12 11:24:34 (12 years ago)

Author:

bburlacu

Message:

#1772: Separated instance-specific attributes of graph node objects from the generic genealogy graph, into metadata objects kept by the specific graph class which corresponds to the specific problem instance (so for instance the SymbolicExpressionTreeGenealogyGraph will keep info about node ranks and qualities etc because that info is specific to symbolic data analysis problems).

File:

• branches/HeuristicLab.EvolutionaryTracking/HeuristicLab.EvolutionaryTracking/3.4/SymbolicExpressionTreeGenealogyGraph.cs (modified) (3 diffs)

branches/HeuristicLab.EvolutionaryTracking/HeuristicLab.EvolutionaryTracking/3.4/SymbolicExpressionTreeGenealogyGraph.cs

@@ -11 +11 @@
   [StorableClass]
   public class SymbolicExpressionTreeGenealogyGraph : GenealogyGraph<ISymbolicExpressionTree> {
+
+    [Storable]
+    private readonly Dictionary<GenealogyGraphNode, NodeMetadata> _nodeInfo = new Dictionary<GenealogyGraphNode, NodeMetadata>();
+
     public SymbolicExpressionTreeGenealogyGraph() {
     }
@@ -27 +31 @@
     }
 
+    public override void AddNode(ISymbolicExpressionTree tree) {
+      if (HasNode(tree)) return;
+      var node = new GenealogyGraphNode(tree);
+      _nodeInfo[node] = new NodeMetadata();
+      base.AddNode(node);
+    }
+
+    public override void AddNode(GenealogyGraphNode node) {
+      var tree = (ISymbolicExpressionTree)node.Data;
+      if (HasNode(tree)) return;
+      _nodeInfo[node] = new NodeMetadata();
+      base.AddNode(node);
+    }
+
+    public NodeMetadata this[GenealogyGraphNode node] {
+      get {
+        NodeMetadata value;
+        _nodeInfo.TryGetValue(node, out value);
+        return value;
+      }
+      set {
+        _nodeInfo[node] = value;
+      }
+    }
+
     #region Fragment tracing
     public IEnumerable<ISymbolicExpressionTree> TraceFragment(ISymbolicExpressionTreeNode fragment, int mode = 0) {
@@ -33 +62 @@
     #endregion
   }
+
+  public class NodeMetadata {
+    public double Quality { get; set; }
+    public List<double> Ranks { get; set; }
+    public bool IsElite { get; set; }
+
+    public NodeMetadata() {
+      Ranks = new List<double>();
+    }
+  }
 }