renamed code_dump() to code_dump2(), code_dump() only takes one argument
[swftools.git] / lib / as3 / mkabc.py
1 #!/usr/bin/python
2 #
3 # mkops.py
4 #
5 # Generate opcodes.h, opcodes.h
6 #
7 # Copyright (c) 2008 Matthias Kramm <kramm@quiss.org>
8 #
9 # This program is free software; you can redistribute it and/or modify
10 # it under the terms of the GNU General Public License as published by
11 # the Free Software Foundation; either version 2 of the License, or
12 # (at your option) any later version.
13 #
14 # This program is distributed in the hope that it will be useful,
15 # but WITHOUT ANY WARRANTY; without even the implied warranty of
16 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17 # GNU General Public License for more details.
18 #
19 # You should have received a copy of the GNU General Public License
20 # along with this program; if not, write to the Free Software
21 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
22
23 import re
24
25 fi = open("code.c", "rb")
26 foc = open("opcodes.c", "wb")
27 foh = open("opcodes.h", "wb")
28
29 foh.write("#ifndef __opcodes_h__\n")
30 foh.write("#define __opcodes_h__\n")
31 foh.write("#include \"abc.h\"\n")
32 foh.write("#include \"pool.h\"\n")
33 foh.write("#include \"code.h\"\n")
34
35 foc.write("#include \"opcodes.h\"\n")
36
37 R = re.compile('{(0x..),\s*"([^"]*)"\s*,\s*"([^"]*)"[^}]*}\s*')
38
39 for line in fi.readlines():
40     line = line.strip()
41     m = R.match(line)
42     if m:
43         op,name,params = m.group(1),m.group(2),m.group(3)
44
45         iterations=1
46         if "2" in params or "s" in params:
47             iterations=2
48
49         for iteration in range(iterations):
50             if iteration==1:
51                 name=name+"2"
52             params = params.strip()
53             paramstr = ""
54             seen = {}
55             names = []
56
57             for c in params:
58                 paramstr += ", "
59                 if c == "2":
60                     if iteration==0:
61                         type,pname="char*","name"
62                     else:
63                         type,pname="multiname_t*","name"
64                 elif c == "s":
65                     if iteration==0:
66                         type,pname="char*","name"
67                     else:
68                         type,pname="string_t*","s"
69                 elif c in "nubs":
70                     type,pname="int","v"
71                 elif c == "m":
72                     type,pname="abc_method_t*","m"
73                 elif c == "i":
74                     type,pname="abc_method_body_t*","m"
75                 elif c == "c":
76                     type,pname="abc_class_t*","m"
77                 elif c == "j":
78                     type,pname="code_t*","label"
79                 elif c == "S":
80                     type,pname="lookupswitch_t*","l"
81                 elif c == "D":
82                     type,pname="void*","debuginfo"
83                 elif c == "r":
84                     type,pname="int","reg"
85                 elif c == "f":
86                     type,pname="double","f"
87                 elif c == "I":
88                     type,pname="int","i"
89                 elif c == "U":
90                     type,pname="unsigned int","u"
91                 else:
92                     raise "Unknown type "+c
93                 paramstr += type
94                 paramstr += " "
95                 if pname in seen:
96                     seen[pname]+=1
97                     pname += str(seen[pname])
98                 else:
99                     seen[pname]=1
100                 paramstr += pname
101                 names += [pname]
102
103             foh.write("code_t* abc_%s(code_t*prev%s);\n" % (name, paramstr))
104
105             foc.write("code_t* abc_%s(code_t*prev%s)\n" % (name, paramstr))
106             foc.write("{\n")
107             foc.write("    code_t*self = add_opcode(prev, %s);\n" % op)
108             i = 0
109             for pname,c in zip(names,params):
110                 if(c == "2"):
111                     if iteration==0:
112                      foc.write("    self->data[%d] = multiname_fromstring(%s);\n" % (i,pname));
113                     else:
114                      foc.write("    self->data[%d] = multiname_clone(%s);\n" % (i,pname));
115                 elif(c in "nur"):
116                     foc.write("    self->data[%d] = (void*)(ptroff_t)%s;\n" % (i,pname))
117                 elif(c in "IU"):
118                     foc.write("    self->data[%d] = (void*)(ptroff_t)%s;\n" % (i,pname))
119                 elif(c in "f"):
120                     foc.write("    double*fp = malloc(sizeof(double));\n")
121                     foc.write("    *fp = %s;\n" % (pname))
122                     foc.write("    self->data[%d] = fp;\n" % (i))
123                 elif(c == "b"):
124                     foc.write("    self->data[%d] = (void*)(ptroff_t)%s;\n" % (i,pname))
125                 elif(c == "s"):
126                     if iteration==0:
127                         foc.write("    self->data[%d] = string_new4(%s);\n" % (i,pname))
128                     else:
129                         foc.write("    self->data[%d] = string_dup3(%s);\n" % (i,pname))
130                 elif(c == "m"):
131                     foc.write("    self->data[%d] = %s;\n" % (i,pname))
132                 elif(c == "c"):
133                     foc.write("    self->data[%d] = %s;\n" % (i,pname))
134                 elif(c == "i"):
135                     foc.write("    self->data[%d] = %s;\n" % (i,pname))
136                 elif(c == "j"):
137                     foc.write("    self->data[%d] = 0; //placeholder\n" % i)
138                     foc.write("    self->branch = %s;\n" % pname)
139                 elif(c == "S"):
140                     foc.write("    self->data[%d] = %s;\n" % (i,pname))
141                 elif(c == "D"):
142                     foc.write("    /* FIXME: write debuginfo %s */\n" % pname)
143                 else:
144                     raise "Unknown type "+c
145                 i = i+1
146             foc.write("    return self;\n")
147             foc.write("}\n")
148
149             foh.write("#define "+name+"(")
150             foh.write(",".join(["method"]+names))
151             foh.write(") (method->code = abc_"+name+"(")
152             foh.write(",".join(["method->code"]+names))
153             foh.write("))\n")
154
155             foh.write("#define OPCODE_"+name.upper()+" "+op+"\n")
156
157 foh.write("#endif\n")
158
159 foh.close()
160 foc.close()
161 fi.close()
162 #{0x75, "convert_d", ""},
163