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
|
#!/usr/bin/env python3
import json
import sys
def script2json(script):
ret = {}
if script[0]['_type'] != 'Label': raise TypeError('obj does not start with Label, wrong file?')
for inst in script:
inst_type = inst['_type']
if inst_type == 'Label':
label = []
ret[inst['name']] = label
continue
func = statement_printer_dict.get(inst_type, print_Unknown)
label.append([inst_type] + func(inst))
return ret
def print_Say(stmt):
return [stmt['who'], stmt['what'], stmt['with_']]
def print_Jump(stmt):
if stmt['expression']:
raise NotImplementedError()
return [stmt['target']]
def print_Scene(stmt):
return [stmt['imspec'], stmt['atl']]
def print_With(stmt):
return [stmt['expr']]
def print_Show(stmt):
return [stmt['imspec'], stmt['atl']]
def print_Hide(stmt):
return [stmt['imspec']]
def print_Python(stmt):
return [stmt['code']]
def print_Return(stmt):
return [stmt['expression']]
def print_UserStatement(stmt):
return [stmt['line'], stmt['parsed']]
def print_Init(stmt):
raise NotImplementedError()
def print_Image(stmt):
raise NotImplementedError()
def print_Transform(stmt):
raise NotImplementedError()
def print_Menu(stmt):
if stmt['with_'] != 'menueffect':
raise NotImplementedError()
if stmt['set'] is not None:
raise NotImplementedError()
return []
for item in stmt['items']:
# caption
f.write(u"\"%s\"" % (escape_string(item[0]), ))
if item[2] is not None:
# condition
if item[1] != 'True':
f.write(u" if %s" % (item[1], ))
f.write(u':\n')
for inner_stmt in item[2]:
print_statement(f, inner_stmt, indent_level + 2)
else:
f.write(u'\n')
def print_Pass(stmt):
return []
def print_Call(stmt):
f.write(u"call ")
if stmt['expression']:
raise NotImplementedError()
return [stmt['label'], stmt['arguments']]
def print_If(stmt):
f.write(u"if %s:\n" % (stmt['entries'][0][0], ))
for inner_stmt in stmt['entries'][0][1]:
print_statement(f, inner_stmt, indent_level + 1)
if len(stmt['entries']) >= 2:
if stmt['entries'][-1][0].strip() == 'True':
else_entry = stmt['entries'][-1]
elif_entries = stmt['entries'][1:-1]
else:
else_entry = None
elif_entries = stmt['entries']
for case in elif_entries:
indent(f, indent_level)
f.write(u"elif %s:\n" % (case[0], ))
for inner_stmt in case[1]:
print_statement(f, inner_stmt, indent_level + 1)
if else_entry is not None:
indent(f, indent_level)
f.write(u"else:\n")
for inner_stmt in else_entry[1]:
print_statement(f, inner_stmt, indent_level + 1)
def print_EarlyPython(stmt):
print_Python(stmt, early=True)
statement_printer_dict = {
"Say": print_Say,
"Jump": print_Jump,
"Scene": print_Scene,
"With": print_With,
"Show": print_Show,
"Hide": print_Hide,
"Python": print_Python,
"Return": print_Return,
"UserStatement": print_UserStatement,
"Init": print_Init,
"Image": print_Image,
"Transform": print_Transform,
"Menu": print_Menu,
"Pass": print_Pass,
"Call": print_Call,
"If": print_If,
"EarlyPython": print_EarlyPython,
}
def print_Unknown(stmt):
raise NotImplementedError("Unknown AST node: %s" % type(stmt).__name__)
with open(sys.argv[1], 'r') as f:
output = script2json(json.load(f))
json.dump(output, open(sys.argv[2], 'w'), separators=(',', ':'))
|
