Class: PrintTreeNodeVisitor2

Inherits:
TreeNodeVisitor show all
Defined in:
lib/treevisitor/visitors/print_tree_node_visitor2.rb

Overview

Prints node using “to_s” instead of name TODO: join this with PrintTreeNodeVisitor

Instance Method Summary collapse

Constructor Details

#initialize ⇒ PrintTreeNodeVisitor2

Returns a new instance of PrintTreeNodeVisitor2.



6
7
8
 
# File 'lib/treevisitor/visitors/print_tree_node_visitor2.rb', line 6

def initialize
  @depth = 0
end

Instance Method Details

#enter_tree_node(tree_node) ⇒ Object 



10
11
12
13
14
15
16
17
18
19
20
21
22
23
 
# File 'lib/treevisitor/visitors/print_tree_node_visitor2.rb', line 10

def enter_tree_node( tree_node )

  str = ""
  (0...@depth).step {
    str << " |-"
  }

  if @depth == 0
    puts str + tree_node.to_s
  else
    puts str + tree_node.to_s
  end
  @depth += 1
end

#exit_tree_node(tree_node) ⇒ Object 



25
26
27
 
# File 'lib/treevisitor/visitors/print_tree_node_visitor2.rb', line 25

def exit_tree_node( tree_node )
  @depth -= 1
end

#visit_leaf_node(leaf_node) ⇒ Object 



29
30
31
32
33
34
35
36
 
# File 'lib/treevisitor/visitors/print_tree_node_visitor2.rb', line 29

def visit_leaf_node( leaf_node )
  str = ""
  (0...@depth-1).step {
    str << " |-"
  }
  str << " |  "
  puts str + leaf_node.to_s
end