Class: VisualizeRuby::Parser::If

Inherits:
Base
  • Object
show all
Includes:
Conditions
Defined in:
lib/visualize_ruby/parser/if.rb

Instance Method Summary collapse

Methods included from Conditions

#set_conditions

Methods inherited from Base

#initialize

Constructor Details

This class inherits a constructor from VisualizeRuby::Parser::Base

Instance Method Details

#branch(on_bool) ⇒ Object 



22
23
24
25
26
27
 
# File 'lib/visualize_ruby/parser/if.rb', line 22

def branch(on_bool)
  on_bool_nodes, on_bool_edges = Parser.new(ast: on_bool).parse
  on_bool_node                 = on_bool_nodes.first
  nodes.concat(on_bool_nodes)
  return on_bool_node, on_bool_edges
end

#parse ⇒ Array<VisualizeRuby::Node>, Array<VisualizeRuby::Edge>

Returns:



6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
 
# File 'lib/visualize_ruby/parser/if.rb', line 6

def parse
  break_ast

  condition_nodes               = set_conditions(condition)
  on_true_node, on_true_edges   = branch(on_true)
  on_false_node, on_false_edges = branch(on_false) if on_false

  condition_nodes.each do |condition_node|
    edges << Edge.new(name: "true", nodes: [condition_node, on_true_node])
    edges << Edge.new(name: "false", nodes: [condition_node, on_false_node]) if on_false
  end
  edges.concat(on_false_edges) if on_false
  edges.concat(on_true_edges)
  return [nodes, edges]
end