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
|
# JSON generation
import json
def walk_enums(s):
r = []
for e in s:
d = []
d.append(e.name)
for b in e.block:
d.append(b)
d.append({'enumtype': e.enumtype})
r.append(d)
return r
def walk_services(s):
r = {}
for e in s:
d = {'reply': e.reply}
if e.stream:
d['stream'] = True
if e.events:
d['events'] = e.events
r[e.caller] = d
return r
def walk_defs(s):
r = []
for t in s:
d = []
d.append(t.name)
for b in t.block:
f = []
if b.type == 'Field':
f = [b.fieldtype, b.fieldname]
elif b.type == 'Array':
if b.lengthfield:
f = [b.fieldtype, b.fieldname, b.length, b.lengthfield]
else:
f = [b.fieldtype, b.fieldname, b.length]
elif b.type == 'Union':
print('UNION')
else:
raise ValueError("Error in processing array type %s" % b)
d.append(f)
if t.crc:
c = {}
c['crc'] = "{0:#0{1}x}".format(t.crc, 10)
d.append(c)
r.append(d)
return r
#
# Plugin entry point
#
def run(filename, s, file_crc):
j = {}
j['types'] = walk_defs([o for o in s['types'] if o.__class__.__name__ == 'Typedef'])
j['messages'] = walk_defs(s['Define'])
j['unions'] = walk_defs([o for o in s['types'] if o.__class__.__name__ == 'Union'])
j['enums'] = walk_enums([o for o in s['types'] if o.__class__.__name__ == 'Enum'])
j['services'] = walk_services(s['Service'])
j['vl_api_version'] = hex(file_crc)
return json.dumps(j, indent=4, separators=(',', ': '))
|
