source: branches/HeuristicLab.EvolutionTracking/HeuristicLab.Problems.DataAnalysis.Symbolic/3.4/Tracking/SymbolicDataAnalysisExpressionTracing.cs @ 10886

#region License Information
/* HeuristicLab
 * Copyright (C) 2002-2014 Heuristic and Evolutionary Algorithms Laboratory (HEAL)
 *
 * This file is part of HeuristicLab.
 *
 * HeuristicLab is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * HeuristicLab is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with HeuristicLab. If not, see <http://www.gnu.org/licenses/>.
 */
#endregion

using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
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>>();
      var nodes = Trace(graphNode, subtreeIndex);
      foreach (var n in nodes) {
        graph.AddVertex(n);
      }
      return graph;
    }

    private static IEnumerable<IFragmentNode> Trace(IGraphNode graphNode, int subtreeIndex, FragmentNode parentNode = null) {
      var node = graphNode; // current node
      var index = subtreeIndex; // current index
      var parent = parentNode; // current parent

      while (true) {
        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
          };
          if (parent != null) {
            var arc = new GenealogyGraphArc(parent, fragmentNode);
            parent.AddForwardArc(arc);
            fragmentNode.AddReverseArc(arc);
          }
          yield return fragmentNode; // no tracing if there are no parents
          break;
        }
        var parents = node.Parents.ToList();
        var fragment = (IFragment<ISymbolicExpressionTreeNode>)node.InArcs.Last().Data;

        if (node.IsElite || fragment == null) {
          // skip elite, go up the graph to original individual
          node = parents[0];
          continue;
        }
        var fragmentLength = fragment.Root.GetLength();
        var tree = node.Content;
        var subtree = tree.NodeAt(index);
        var subtreeLength = subtree.GetLength();

        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
          };
          if (parent != null) {
            var arc = new GenealogyGraphArc(parent, fragmentNode);
            parent.AddForwardArc(arc);
            fragmentNode.AddReverseArc(arc);
          }
          yield return fragmentNode;
          var up = Trace(parents[0], fragment.Index1, fragmentNode);
          foreach (var v in up) { yield return v; }
          break;
        }

        if (parents.Count == 2) {
          // we are tracing a crossover operation
          if (fragment.Index1 == index) {
            node = parents[1]; // take second parent which contains the fragment
            index = fragment.Index2;
            continue;
          }

          if (fragment.Index1 < index) {
            if (fragment.Index1 + fragmentLength > index) {
              node = parents[1]; // fragment contains subtree, take second parent
              index += fragment.Index2 - fragment.Index1; // the subtree has the same index relative to the fragment
            } else {
              // fragment distinct from subtree
              node = parents[0]; // take first parent which contains the subtree
              index += node.Content.NodeAt(fragment.Index1).GetLength() - fragmentLength; // subtreeIndex must be adjusted according to swapped fragments sizes
            }
            continue;
          }

          if (fragment.Index1 > index) {
            if (fragment.Index1 < index + subtreeLength) {
              // 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
              };
              if (parent != null) {
                var arc = new GenealogyGraphArc(parent, fragmentNode);
                parent.AddForwardArc(arc);
                fragmentNode.AddReverseArc(arc);
              }
              //              fragmentNode.Content.Index1 = fragment.Index1 - index;
              yield return fragmentNode;

              if (parents[1].Content.NodeAt(fragment.Index2).GetLength() != fragmentLength) {
                throw new Exception("Fragment lengths should match!");
              }
              // the two paths below ("left" and "right") correspond to the subtree and the fragment, respectively
              var left = Trace(parents[0], index, fragmentNode); // trace subtree
              var right = Trace(parents[1], fragment.Index2, fragmentNode); // trace fragment
              foreach (var v in left.Concat(right)) { yield return v; }
              break;
            } else {
              node = parents[0]; // fragment distinct from subtree: update node, index remains unchanged
            }
          }
        }
      }
    }

    private static string ViewAsText(this ISymbolicExpressionTreeNode root) {
      var writer = new StringWriter();
      SymbolicExpressionTreeHierarchicalFormatter.RenderNode(writer, root, string.Empty);
      return writer.ToString();
    }
    #region some helper methods for shortening the tracing code
    private static ISymbolicExpressionTreeNode NodeAt(this ISymbolicExpressionTree tree, int position) {
      return NodeAt(tree.Root, position);
    }
    private static ISymbolicExpressionTreeNode NodeAt(this ISymbolicExpressionTreeNode root, int position) {
      return root.IterateNodesPrefix().ElementAt(position);
    }
    #endregion
  }
}