Changeset 10888 for branches/HeuristicLab.EvolutionTracking/HeuristicLab.Problems.DataAnalysis.Symbolic/3.4/Tracking/SymbolicDataAnalysisExpressionTracing.cs

Timestamp:

05/26/14 01:56:35 (10 years ago)

Author:

bburlacu

Message:

#1772: Introduced separate class for FragmentNodes and adjusted tracing code. Fixed small bug creating the genealogy graph.

File:

• branches/HeuristicLab.EvolutionTracking/HeuristicLab.Problems.DataAnalysis.Symbolic/3.4/Tracking/SymbolicDataAnalysisExpressionTracing.cs (modified) (6 diffs)

branches/HeuristicLab.EvolutionTracking/HeuristicLab.Problems.DataAnalysis.Symbolic/3.4/Tracking/SymbolicDataAnalysisExpressionTracing.cs

@@ -26 +26 @@
 using HeuristicLab.Encodings.SymbolicExpressionTreeEncoding;
 using HeuristicLab.EvolutionTracking;
-using FragmentNode = HeuristicLab.EvolutionTracking.GenealogyGraphNode<HeuristicLab.EvolutionTracking.IFragment<HeuristicLab.Encodings.SymbolicExpressionTreeEncoding.ISymbolicExpressionTreeNode>>;
-using IFragmentNode = HeuristicLab.EvolutionTracking.IGenealogyGraphNode<HeuristicLab.EvolutionTracking.IFragment<HeuristicLab.Encodings.SymbolicExpressionTreeEncoding.ISymbolicExpressionTreeNode>>;
-using IGraphNode = HeuristicLab.EvolutionTracking.IGenealogyGraphNode<HeuristicLab.Encodings.SymbolicExpressionTreeEncoding.ISymbolicExpressionTree>;
 
 namespace HeuristicLab.Problems.DataAnalysis.Symbolic {
   public static class SymbolicDataAnalysisExpressionTracing {
-    public static IGenealogyGraph<IFragment<ISymbolicExpressionTreeNode>> TraceSubtree(IGenealogyGraphNode<ISymbolicExpressionTree> graphNode, int subtreeIndex) {
-      var graph = new GenealogyGraph<IFragment<ISymbolicExpressionTreeNode>>();
+    public static FragmentGraph TraceSubtree(IGenealogyGraphNode<ISymbolicExpressionTree> graphNode, int subtreeIndex) {
+      var graph = new FragmentGraph { AllowDuplicateContent = true };
       var nodes = Trace(graphNode, subtreeIndex);
       foreach (var n in nodes) {
@@ -40 +37 @@
       return graph;
     }
-
-    private static IEnumerable<IFragmentNode> Trace(IGraphNode graphNode, int subtreeIndex, FragmentNode parentNode = null) {
+    /// <summary>
+    /// Traces a subtree in a genealogy of individuals, tracking down it's genetic components
+    /// </summary>
+    /// <param name="graphNode">The point in the genealogy where to start the tracing from</param>
+    /// <param name="subtreeIndex">The traced subtree preorder index in the current tree</param>
+    /// <param name="parentNode">The parent node in the fragment graph if there is any</param>
+    /// <returns></returns>
+    private static IEnumerable<FragmentNode> Trace(IGenealogyGraphNode<ISymbolicExpressionTree> graphNode, int subtreeIndex, FragmentNode parentNode = null) {
       var node = graphNode; // current node
       var index = subtreeIndex; // current index
@@ -49 +52 @@
         if (!node.Parents.Any()) {
           var fragmentNode = new FragmentNode {
-            Content = new Fragment<ISymbolicExpressionTreeNode> {
-              Root = node.Content.Root,
-              //              Root = node.Content.NodeAt(index)
-              Index1 = index
-            },
-            Rank = node.Rank,
-            Quality = node.Quality
+            Content = node,
+            SubtreeIndex = index,
           };
           if (parent != null) {
-            var arc = new GenealogyGraphArc(parent, fragmentNode);
+            var arc = new Arc(parent, fragmentNode);
             parent.AddForwardArc(arc);
             fragmentNode.AddReverseArc(arc);
@@ -78 +76 @@
         var subtreeLength = subtree.GetLength();
 
+        #region mutation tracing
         if (parents.Count == 1) {
           // we are tracing a mutation operation
           var fragmentNode = new FragmentNode {
-            Content = new Fragment<ISymbolicExpressionTreeNode> {
-              Root = tree.Root,
-              //              Root = subtree
-              Index1 = index,
-              Index2 = fragment.Index1
-            },
-            Rank = node.Rank,
-            Quality = node.Quality
+            Content = node,
+            SubtreeIndex = index,
+            FragmentIndex = fragment.Index1
           };
           if (parent != null) {
-            var arc = new GenealogyGraphArc(parent, fragmentNode);
+            var arc = new Arc(parent, fragmentNode);
             parent.AddForwardArc(arc);
             fragmentNode.AddReverseArc(arc);
           }
+          node = parents[0];
+          if (subtreeIndex == fragment.Index1) {
+            // index stays the same
+          } else if (fragment.Index1 < subtreeIndex) {
+            if (subtreeIndex < fragment.Index1 + fragmentLength) {
+              // subtree contained in fragment, index stays the same
+            } else {
+              index += node.Content.NodeAt(fragment.Index1).GetLength() - fragmentLength;
+            }
+          } else if (subtreeIndex < fragment.Index1) {
+            if (fragment.Index1 < subtreeIndex + subtreeLength) {
+              // subtree contains fragment
+              index = fragment.Index1;
+            } else {
+              // index stays the same
+            }
+          }
+
           yield return fragmentNode;
-          var up = Trace(parents[0], fragment.Index1, fragmentNode);
+          var up = Trace(node, index, fragmentNode);
           foreach (var v in up) { yield return v; }
           break;
         }
+        #endregion
 
+        #region crossover tracing
         if (parents.Count == 2) {
           // we are tracing a crossover operation
@@ -125 +139 @@
               // subtree contains fragment => bifurcation point in the fragment graph
               var fragmentNode = new FragmentNode {
-                Content = new Fragment<ISymbolicExpressionTreeNode> {
-                  Root = tree.Root,
-                  Index1 = index,
-                  Index2 = fragment.Index1
-                },
-                Rank = node.Rank,
-                Quality = node.Quality
+                Content = node,
+                SubtreeIndex = index,
+                FragmentIndex = fragment.Index1
               };
               if (parent != null) {
-                var arc = new GenealogyGraphArc(parent, fragmentNode);
+                var arc = new Arc(parent, fragmentNode);
                 parent.AddForwardArc(arc);
                 fragmentNode.AddReverseArc(arc);
               }
-              //              fragmentNode.Content.Index1 = fragment.Index1 - index;
               yield return fragmentNode;
 
@@ -154 +163 @@
           }
         }
+        #endregion
       }
     }