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
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
|
import giscanner
class Node(object):
pass
class Function(Node):
def __init__(self, name, retval, parameters, symbol):
self.name = name
self.retval = retval
self.parameters = parameters
self.symbol = symbol
def __repr__(self):
return 'Function(%r, %r, %r)' % (self.name, self.retval,
self.parameters)
class Parameter(Node):
def __init__(self, name, type):
self.name = name
self.type = type
def __repr__(self):
return 'Parameter(%r, %r)' % (self.name, self.type)
class Enum(Node):
def __init__(self, name, members):
self.name = name
self.members = members
def __repr__(self):
return 'Enum(%r, %r)' % (self.name, self.members)
class Member(Node):
def __init__(self, name, value):
self.name = name
self.value = value
def __repr__(self):
return 'Member(%r, %r)' % (self.name, self.value)
class Struct(Node):
def __init__(self, name):
self.name = name
def __repr__(self):
return 'Struct(%r)' % (self.name,)
class Return(Node):
def __init__(self, type):
self.type = type
def __repr__(self):
return 'Return(%r)' % (self.type,)
class Class(Node):
def __init__(self, name, parent):
self.name = name
self.parent = parent
self.methods = []
self.constructors = []
self.properties = []
def __repr__(self):
return '%s(%r, %r, %r)' % (
self.__class__.__name__,
self.name, self.parent, self.methods)
class Interface(Node):
def __init__(self, name):
self.name = name
self.methods = []
self.properties = []
def __repr__(self):
return '%s(%r, %r)' % (
self.__class__.__name__,
self.name, self.methods)
class Constant(Node):
def __init__(self, name, type, value):
self.name = name
self.type = type
self.value = value
def __repr__(self):
return 'Constant(%r, %r, %r)' % (
self.name, self.type, self.value)
class Property(Node):
def __init__(self, name, type):
self.name = name
self.type = type
def __repr__(self):
return '%s(%r, %r, %r)' % (
self.__class__.__name__,
self.name, self.type, self.value)
class Callback(Node):
def __init__(self, name, retval, parameters):
self.name = name
self.retval = retval
self.parameters = parameters
def __repr__(self):
return 'Callback(%r, %r, %r)' % (
self.name, self.retval, self.parameters)
class TreeBuilder(object):
def __init__(self, generator):
self.generator = generator
self.nodes = []
self._traverse()
def get_nodes(self):
for node in self.nodes:
yield node
def _traverse(self):
for symbol in self.generator.get_symbols():
node = self._traverse_one(symbol)
if node is not None:
self.nodes.append(node)
def _traverse_one(self, symbol, stype=None):
# Skip private symbols
if symbol.ident.startswith('_'):
return
if stype is None:
stype = symbol.type
if stype == giscanner.CSYMBOL_TYPE_FUNCTION:
return self._create_function(symbol)
elif stype == giscanner.CSYMBOL_TYPE_TYPEDEF:
if (symbol.base_type.type == giscanner.CTYPE_POINTER and
symbol.base_type.base_type.type == giscanner.CTYPE_FUNCTION):
return self._create_callback(symbol)
else:
return self._traverse_one(symbol, symbol.base_type.type)
elif stype == giscanner.CSYMBOL_TYPE_STRUCT:
return self._create_struct(symbol)
elif stype == giscanner.CSYMBOL_TYPE_ENUM:
return self._create_enum(symbol)
else:
print 'BUILDER: unhandled symbol', symbol.type
def _create_enum(self, symbol):
members = []
for child in symbol.base_type.child_list:
members.append(Member(child.ident,
child.const_int))
return Enum(symbol.ident, members)
def _create_function(self, symbol):
parameters = self._create_parameters(symbol.base_type)
retval = Return(self._create_source_type(symbol.base_type.base_type))
return Function(symbol.ident, retval, list(parameters), symbol.ident)
def _create_source_type(self, source_type):
if source_type.type == giscanner.CTYPE_VOID:
value = 'void'
elif source_type.type == giscanner.CTYPE_BASIC_TYPE:
value = source_type.name
elif source_type.type == giscanner.CTYPE_TYPEDEF:
value = source_type.name
elif source_type.type == giscanner.CTYPE_POINTER:
value = self._create_source_type(source_type.base_type) + '*'
else:
print 'BUILDER: Unhandled source type: %d' % (source_type.type,)
value = '???'
return value
def _create_parameters(self, base_type):
for child in base_type.child_list:
yield self._create_parameter(child)
def _create_parameter(self, symbol):
return Parameter(symbol.ident,
self._create_source_type(symbol.base_type))
def _create_struct(self, symbol):
return Struct(symbol.ident)
def _create_callback(self, symbol):
parameters = self._create_parameters(symbol.base_type.base_type)
retval = Return(self._create_source_type(symbol.base_type.base_type.base_type))
return Callback(symbol.ident, retval, list(parameters))
|
