Contents

# -*- coding: utf-8 -*-
module TreeVisitor
  #
  # Clone a tree_node, nodes are duplicated.
  # Node content are not duplicated!
  #
  class CloneTreeNodeVisitor # < BasicTreeNodeVisitor

    #
    # Contains the cloned tree node after the visit
    #
    attr_reader :cloned_root

    def initialize
      super
      @cloned_root = nil
      @stack = []
    end

    def enter_node( tree_node )
      if @stack.empty?
        cloned_tree_node = TreeNode.new( tree_node.content )
        @cloned_root = cloned_tree_node
      else
        cloned_tree_node = TreeNode.new( tree_node.content, @stack.last )
      end
      @stack.push( cloned_tree_node )
    end

    def exit_node( tree_node )
      @stack.pop
    end

    def visit_leaf( leaf_node )
      LeafNode.new( leaf_node.content, @stack.last )
    end

  end
end

Version data entries

5 entries across 5 versions & 2 rubygems

Version	Path
tree.rb-0.3.0	lib/treevisitor/visitors/clone_tree_node_visitor.rb
treevisitor-0.2.3	lib/treevisitor/visitors/clone_tree_node_visitor.rb
treevisitor-0.2.2	lib/treevisitor/visitors/clone_tree_node_visitor.rb
treevisitor-0.2.1	lib/treevisitor/visitors/clone_tree_node_visitor.rb
treevisitor-0.2.0	lib/treevisitor/visitors/clone_tree_node_visitor.rb