Class: TreeNode

Inherits:
AbsNode show all
Defined in:
lib/treevisitor/tree_node.rb

Overview

TreeNode can contains other TreeNode (children) and can contains LeafNode (leves)

TreeNode @childs -1—n-> TreeNode

@leaves -1---n-> LeafNode

Instance Attribute Summary collapse

Attributes inherited from AbsNode

#name, #next, #parent, #prev

Instance Method Summary collapse

Methods inherited from AbsNode

#depth, #path, #path_with_prefix, #prefix_path, #prefix_path=, #root, #to_s

Constructor Details

#initialize(name, parent = nil) ⇒ TreeNode

Returns a new instance of TreeNode.



15
16
17
18
19
20
21
22
 
# File 'lib/treevisitor/tree_node.rb', line 15

def initialize( name, parent = nil )
  @leaves = []
  @children = []
  super( name )
  if parent
    parent.add_child( self )
  end
end

Instance Attribute Details

#children ⇒ Object (readonly)

Returns the value of attribute children.



13
14
15
 
# File 'lib/treevisitor/tree_node.rb', line 13

def children
  @children
end

#leaves ⇒ Object (readonly)

Returns the value of attribute leaves.



12
13
14
 
# File 'lib/treevisitor/tree_node.rb', line 12

def leaves
  @leaves
end

Instance Method Details

#accept(visitor) ⇒ Object

return the visitor



105
106
107
108
109
110
111
112
113
114
115
 
# File 'lib/treevisitor/tree_node.rb', line 105

def accept( visitor )
  visitor.enter_tree_node( self )
  @leaves.each{ |leaf|
    leaf.accept( visitor )
  }
  @children.each { |child|
    child.accept( visitor )
  }
  visitor.exit_tree_node( self )
  visitor
end

#add_child(tree_node) ⇒ Object 



68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
 
# File 'lib/treevisitor/tree_node.rb', line 68

def add_child( tree_node )
  return if tree_node.parent == self
  if not tree_node.parent.nil?
    tree_node.remove_from_parent
  else
    tree_node.prefix_path = nil
  end  
  tree_node.invalidate
  tree_node.parent = self
  if @children.length > 0
    @children.last.next = tree_node
    tree_node.prev = @children.last
  else
    tree_node.prev = nil
  end
  tree_node.next = nil
  @children << tree_node
end

#add_leaf(leaf) ⇒ Object 



51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
 
# File 'lib/treevisitor/tree_node.rb', line 51

def add_leaf( leaf )
  return if leaf.parent == self
  if not leaf.parent.nil?
    leaf.remove_from_parent
  end  
  leaf.parent = self
  if @leaves.length > 0
    @leaves.last.next = leaf
    leaf.prev = @leaves.last
  else
    leaf.prev = nil
  end
  leaf.next = nil
  leaf.invalidate
  @leaves << leaf
end

#find(name) ⇒ Object 



87
88
89
90
91
92
93
94
95
96
97
98
99
100
 
# File 'lib/treevisitor/tree_node.rb', line 87

def find( name )
  return self if self.name == name
  
  leaf = @leaves.find { |l| l.name == name }
  if leaf
    return leaf
  end
  
  @children.each {|c|
    node = c.find(name)
    return node if node
  }
  nil
end

#invalidate ⇒ Object

invalidate cached info invalidate propagates form parent to children and leaves



32
33
34
35
36
 
# File 'lib/treevisitor/tree_node.rb', line 32

def invalidate
  super
  @children.each{ |c| c.invalidate }
  @leaves.each{ |l| l.invalidate }
end

#nr_children ⇒ Object 



47
48
49
 
# File 'lib/treevisitor/tree_node.rb', line 47

def nr_children
  @children.length + @children.inject(0) { |sum, child| sum + child.nr_children }
end

#nr_leaves ⇒ Object 



43
44
45
 
# File 'lib/treevisitor/tree_node.rb', line 43

def nr_leaves
  @leaves.length + @children.inject(0) { |sum, child| sum + child.nr_leaves }
end

#nr_nodes ⇒ Object 



38
39
40
41
 
# File 'lib/treevisitor/tree_node.rb', line 38

def nr_nodes
  nr = @leaves.length + @children.length
  @children.inject( nr ) { |nr,c| nr + c.nr_nodes }
end

#root? ⇒ Boolean

Returns:

  • (Boolean) 


24
25
26
 
# File 'lib/treevisitor/tree_node.rb', line 24

def root?
  @parent.nil?
end

#to_str(prefix = "") ⇒ Object 



117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
 
# File 'lib/treevisitor/tree_node.rb', line 117

def to_str( prefix= "" )
  str = ""

  if root?
    str << to_s << "\n"
  else
    str << prefix 
    if self.next 
      str << "|-- "
    else
      str << "\`-- "
    end
    str << to_s << "\n"
    prefix += self.next ? "|   " : "    "
  end
  
  @leaves.each do |leaf|
    str << prefix
    if !leaf.next.nil? or !@children.empty?
      str << "|-- "
    else
      str << "\`-- "
    end
    str <<  leaf.to_s << "\n"
  end

  @children.each do |child|
    str << child.to_str( prefix )
  end
  str
end