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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
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
148
149
150
151
152
153
|
# -*- coding: utf-8 -*-
# pylint: disable=too-many-lines
from __future__ import print_function
from __future__ import unicode_literals
import io
import sys
from cmakelang.parse.common import TreeNode
def dump_tree(nodes, outfile=None, indent=None):
"""
Print a tree of node objects for debugging purposes
"""
if indent is None:
indent = ''
if outfile is None:
outfile = sys.stdout
for idx, node in enumerate(nodes):
outfile.write(indent)
if idx + 1 == len(nodes):
outfile.write('└─ ')
else:
outfile.write('├─ ')
noderep = repr(node)
if sys.version_info[0] < 3:
# python2
outfile.write(getattr(noderep, 'decode')('utf-8'))
else:
# python3
outfile.write(noderep)
outfile.write('\n')
if not hasattr(node, 'children'):
continue
if idx + 1 == len(nodes):
dump_tree(node.children, outfile, indent + ' ')
else:
dump_tree(node.children, outfile, indent + '│ ')
def dump_tree_tostr(nodes, indent=None):
outfile = io.StringIO()
dump_tree(nodes, outfile, indent)
return outfile.getvalue()
def dump_tree_upto(nodes, history, outfile=None, indent=None):
"""
Print a tree of node objects for debugging purposes
"""
if indent is None:
indent = ''
if outfile is None:
outfile = sys.stdout
for idx, node in enumerate(nodes):
outfile.write(indent)
if idx + 1 == len(nodes):
outfile.write('└─ ')
else:
outfile.write('├─ ')
if history[0] is node:
# ANSI red
outfile.write("\u001b[31m")
if sys.version_info[0] < 3:
outfile.write(repr(node).decode('utf-8'))
else:
outfile.write(repr(node))
# ANSI reset
outfile.write("\u001b[0m")
else:
if sys.version_info[0] < 3:
outfile.write(repr(node).decode('utf-8'))
else:
outfile.write(repr(node))
outfile.write('\n')
if not hasattr(node, 'children'):
continue
subhistory = history[1:]
if not subhistory:
continue
if idx + 1 == len(nodes):
dump_tree_upto(node.children, subhistory, outfile, indent + ' ')
else:
dump_tree_upto(node.children, subhistory, outfile, indent + '│ ')
def tree_string(nodes, history=None):
outfile = io.StringIO()
if history:
dump_tree_upto(nodes, history, outfile)
else:
dump_tree(nodes, outfile)
return outfile.getvalue()
def has_nontoken_children(node):
if not hasattr(node, 'children'):
return False
for child in node.children:
if isinstance(child, TreeNode):
return True
return False
def dump_tree_for_test(nodes, outfile=None, indent=None, increment=None):
"""
Print a tree of node objects for debugging purposes
"""
if increment is None:
increment = ' '
if indent is None:
indent = ''
if outfile is None:
outfile = sys.stdout
for node in nodes:
if not isinstance(node, TreeNode):
continue
outfile.write(indent)
outfile.write('({}, ['.format(node.node_type))
if has_nontoken_children(node):
outfile.write('\n')
dump_tree_for_test(node.children, outfile, indent + increment, increment)
outfile.write(indent)
outfile.write(']),\n')
def test_string(nodes, indent=None, increment=None):
if indent is None:
indent = ' ' * 10
if increment is None:
increment = ' '
outfile = io.StringIO()
dump_tree_for_test(nodes, outfile, indent, increment)
return outfile.getvalue()
|
