1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
# Base class for formatted textual dump of a "model"
#
class Puppet::Pops::Model::TreeDumper
attr_accessor :indent_count
def initialize initial_indentation = 0
@@dump_visitor ||= Puppet::Pops::Visitor.new(nil,"dump",0,0)
@indent_count = initial_indentation
end
def dump(o)
format(do_dump(o))
end
def do_dump(o)
@@dump_visitor.visit_this_0(self, o)
end
def indent
" " * indent_count
end
def format(x)
result = ""
parts = format_r(x)
parts.each_index do |i|
if i > 0
# separate with space unless previous ends with whitespace or (
result << ' ' if parts[i] != ")" && parts[i-1] !~ /.*(?:\s+|\()$/ && parts[i] !~ /^\s+/
end
result << parts[i].to_s
end
result
end
def format_r(x)
result = []
case x
when :break
result << "\n" + indent
when :indent
@indent_count += 1
when :dedent
@indent_count -= 1
when Array
result << '('
result += x.collect {|a| format_r(a) }.flatten
result << ')'
when Symbol
result << x.to_s # Allows Symbols in arrays e.g. ["text", =>, "text"]
else
result << x
end
result
end
def is_nop? o
o.nil? || o.is_a?(Puppet::Pops::Model::Nop)
end
end
|
