Class: Awkward::Visitor

Inherits:
Object
  • Object
show all
Defined in:
lib/awkward/visitor.rb

Defined Under Namespace

Classes: Edge, Node

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize ⇒ Visitor

Returns a new instance of Visitor.



13
14
15
16
17
18
19
 
# File 'lib/awkward/visitor.rb', line 13

def initialize
  @nodes     = []
  @stack     = []
  @edges     = []
  @callstack = []
  @seen      = {}
end

Instance Attribute Details

#edges ⇒ Object (readonly)

Returns the value of attribute edges.



11
12
13
 
# File 'lib/awkward/visitor.rb', line 11

def edges
  @edges
end

Instance Method Details

#accept(o) ⇒ Object 



25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
 
# File 'lib/awkward/visitor.rb', line 25

def accept o
  return connect(@stack.last, @seen[o.object_id]) if @seen.key? o.object_id

  node = Node.new o

  @seen[o.object_id] = node

  connect(@stack.last, node) unless @callstack.empty?

  @stack.push node

  send "visit_#{o.class.name.gsub('::', '_')}", o

  @stack.pop
end

#nodes ⇒ Object 



21
22
23
 
# File 'lib/awkward/visitor.rb', line 21

def nodes
  @seen.values
end

#to_dot ⇒ Object 



41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
 
# File 'lib/awkward/visitor.rb', line 41

def to_dot
  dot = "digraph \"Graph\" {\nnode [width=0.375,height=0.25,shape = \"record\"];\n  eodot\n\n  nodes.each do |node|\n    dot.concat <<-eonode\n    \#{node.object_id} [label=\"<f0> \#{[node.name, node.value].compact.join('|')}\"];\n    eonode\n  end\n\n  edges.each do |edge|\n    dot.concat <<-eoedge\n    \#{edge.left.object_id} -> \#{edge.right.object_id} [label=\"\#{edge.name}\"];\n    eoedge\n  end\n  dot + \"}\"\nend\n"