refactored relocation
[swftools.git] / lib / as3 / mklib.c
1 /* mklib.c
2
3    File to generate builtin.c
4
5    Copyright (c) 2008,2009 Matthias Kramm <kramm@quiss.org>
6  
7    This program is free software; you can redistribute it and/or modify
8    it under the terms of the GNU General Public License as published by
9    the Free Software Foundation; either version 2 of the License, or
10    (at your option) any later version.
11
12    This program is distributed in the hope that it will be useful,
13    but WITHOUT ANY WARRANTY; without even the implied warranty of
14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15    GNU General Public License for more details.
16
17    You should have received a copy of the GNU General Public License
18    along with this program; if not, write to the Free Software
19    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
20
21 #include <stdlib.h>
22 #include <stdio.h>
23 #include <unistd.h>
24 #include <memory.h>
25 #include "../rfxswf.h"
26 #include "../os.h"
27 #include "../q.h"
28 #include "pool.h"
29 #include "files.h"
30 #include "tokenizer.h"
31 #include "parser.tab.h"
32 #include "parser.h"
33 #include "import.h"
34
35 void fixstring(char*s)
36 {
37     char first=1;
38     for(;*s;s++) {
39         if(!((*s>='a' && *s<='z') || (*s>='A' && *s<='Z') || 
40                     (*s>='0' && *s<='9' && !first))) {
41             *s = '_';
42         }
43         first = 0;
44     }
45 }
46
47 static char* mkpid(const char*package)
48 {
49     char*id = malloc(strlen(package)+20);
50     strcpy(id, "package_");
51     if(!*package) 
52         strcat(id, "global");
53     else
54         strcat(id, package);
55     fixstring(id);
56     return id;
57 }
58 static char* mkcid(const char*package, const char*name)
59 {
60     char*id = malloc(strlen(package)+strlen(name)+10);
61     strcpy(id, package);
62     strcat(id, "_");
63     strcat(id, name);
64     fixstring(id);
65     return id;
66 }
67 static char* mkptr2(const char*package, const char*name)
68 {
69     if(!package[0] && !strcmp(name, "void")) {
70         return "&voidclass";
71     }
72     char*id = malloc(strlen(package)+strlen(name)+10);
73     strcpy(id, "&");
74     strcat(id, package);
75     strcat(id, "_");
76     strcat(id, name);
77     fixstring(id+1);
78     return id;
79 }
80 static char* mkid2(const char*cls, const char*member)
81 {
82     char*id = malloc(strlen(cls)+strlen(member)+10);
83     strcpy(id, cls);
84     strcat(id, "_");
85     strcat(id, member);
86     fixstring(id);
87     return id;
88 }
89 #define mkid(s) ((s)?mkcid((s)->package, (s)->name):"0")
90 #define mkptr(s) ((s)?mkptr2((s)->package, (s)->name):"0")
91
92 static array_t*tosort=0;
93 static int compare_classes(const void*v1, const void*v2)
94 {
95     int x1 = *(int*)v1;
96     int x2 = *(int*)v2;
97     abc_class_t*c1 = array_getvalue(tosort, x1);
98     abc_class_t*c2 = array_getvalue(tosort, x2);
99     int i = strcmp(c1->classname->ns->name, c2->classname->ns->name);
100     if(i)
101         return i;
102     return strcmp(c1->classname->name, c2->classname->name);
103 }
104
105 static int compare_traits(const void*v1, const void*v2)
106 {
107     trait_t* x1 = *(trait_t**)v1;
108     trait_t* x2 = *(trait_t**)v2;
109     int i = strcmp(x1->name->ns->name, x2->name->ns->name);
110     if(i)
111         return i;
112     return strcmp(x1->name->name, x2->name->name);
113 }
114
115 dict_t* builtin_getclasses()
116 {
117     return dict_new2(&slotinfo_type);
118 }
119
120 extern dict_t*registry_classes;
121
122 char*mktype(slotinfo_t*s) 
123 {
124     if(s->kind == INFOTYPE_CLASS) {
125         return "classinfo_t";
126     } else if(s->kind == INFOTYPE_METHOD) {
127         return "methodinfo_t";
128     } else if(s->kind == INFOTYPE_VAR) {
129         return "varinfo_t";
130     }
131     return "**ERROR**";
132 }
133
134 void write_slotinfo(FILE*fi, slotinfo_t*s, char*id, char*prefix);
135
136 void write_slotinfo_decl(FILE*fi, slotinfo_t*s, char*prefix)
137 {
138     fprintf(fi, "%s", prefix);
139     char*id = mkid(s);
140     fprintf(fi, "static %s %s;\n", mktype(s), id);
141
142     if(s->kind == INFOTYPE_CLASS) {
143         classinfo_t*c = (classinfo_t*)s;
144         dict_t*d = &c->members;
145         DICT_ITERATE_DATA(d, slotinfo_t*, s1) {
146             fprintf(fi, "static %s %s;\n", mktype(s1), mkid2(id, s1->name));
147         }
148         DICT_ITERATE_DATA(d, slotinfo_t*, s2) {
149             fprintf(fi, "static %s %s;\n", mktype(s2), mkid2(id, s2->name));
150         }
151     }
152 }
153 void write_initinfo(FILE*fi, slotinfo_t*s, char*prefix)
154 {
155     if(s->kind == INFOTYPE_CLASS) {
156         classinfo_t*c = (classinfo_t*)s;
157         char*id = mkid(c);
158         dict_t*d1 = &c->members;
159         dict_t*d2 = &c->static_members;
160         fprintf(fi, "%s", prefix);
161         fprintf(fi, "dict_init2(&%s.members, &memberinfo_type, %d);\n", id, d1->hashsize);
162         fprintf(fi, "%s", prefix);
163         fprintf(fi, "dict_init2(&%s.static_members, &memberinfo_type, %d);\n", id, d2->hashsize);
164         int t;
165         DICT_ITERATE_DATA(d1,slotinfo_t*,s1) {
166            fprintf(fi, "%s", prefix);
167            char*id2 = mkid2(id, s1->name);
168            fprintf(fi, "dict_put(&%s.members, &%s, &%s);\n", id, id2, id2);
169         }
170         DICT_ITERATE_DATA(d2,slotinfo_t*,s2) {
171            fprintf(fi, "%s", prefix);
172            char*id2 = mkid2(id, s2->name);
173            fprintf(fi, "dict_put(&%s.static_members, &%s, &%s);\n", id, id2, id2);
174         }
175     }
176 }
177
178 void write_constant(FILE*fi, constant_t*value, char*id, char*prefix)
179 {
180     if(NS_TYPE(value->type)) {
181         fprintf(fi, "%s", prefix);
182         fprintf(fi, "static namespace_t %s_constant_ns = {0x%02x, \"%s\"};\n", id, value->ns->access, value->ns->name);
183     } else if(value->type == CONSTANT_STRING) {
184         fprintf(fi, "%s", prefix);
185         fprintf(fi, "static string_t %s_constant_s = {\"%s\", %d};\n", id, value->s->str, value->s->len);
186     }
187     fprintf(fi, "%s", prefix);
188     fprintf(fi, "static constant_t %s_constant = ", id);
189     fprintf(fi, "{type: %d", value->type);
190     if(NS_TYPE(value->type)) {
191         fprintf(fi, ", &%s_constant_ns", id);
192     } else if(value->type == CONSTANT_INT) {
193         fprintf(fi, ",i: %d,", value->type);
194     } else if(value->type == CONSTANT_UINT) {
195         fprintf(fi, ",u: %u", value->u);
196     } else if(value->type == CONSTANT_FLOAT) {
197         if(!isnan(value->f) && !isinf(value->f))
198             fprintf(fi, ", %f", value->f);
199     } else if(value->type == CONSTANT_STRING) {
200         fprintf(fi, ", &%s_constant_s", id);
201     }
202     fprintf(fi, "};\n");
203 }
204
205 void write_slotinfo(FILE*fi, slotinfo_t*s, char*id, char*prefix)
206 {
207     if(s->kind == INFOTYPE_VAR) {
208         varinfo_t*v = (varinfo_t*)s;
209         if(v->value) {
210             write_constant(fi, v->value, id, prefix);
211         }
212     }
213     fprintf(fi, "%s", prefix);
214     fprintf(fi, "static %s %s = {", mktype(s), id);
215     fprintf(fi, "0x%02x, 0x%02x, 0x%02x, 0x%02x, ", s->kind, s->subtype, s->flags, s->access);
216     if(s->package)
217         fprintf(fi, "\"%s\", ", s->package);
218     else
219         fprintf(fi, "0, ");
220     
221     if(s->name)
222         fprintf(fi, "\"%s\", ", s->name);
223     else
224         fprintf(fi, "0, ");
225
226     fprintf(fi, "%d, ", s->slot);
227
228     if(s->kind == INFOTYPE_CLASS) {
229         classinfo_t*c = (classinfo_t*)s;
230         fprintf(fi, "%s, ", mkptr(c->superclass));
231         fprintf(fi, "interfaces: {");
232         int t;
233         for(t=0;c->interfaces[t];t++) {
234             fprintf(fi, "%s", mkptr(c->interfaces[t]));
235             fprintf(fi, ", ");
236         }
237         fprintf(fi, "0}};\n");
238     }
239     if(s->kind == INFOTYPE_METHOD) {
240         methodinfo_t*m = (methodinfo_t*)s;
241         fprintf(fi, "%s, ", mkptr(m->return_type));
242         fprintf(fi, "%s, ", mkptr(m->parent));
243         fprintf(fi, "0"); // params TODO
244         fprintf(fi, "};\n");
245     }
246     if(s->kind == INFOTYPE_VAR) {
247         varinfo_t*m = (varinfo_t*)s;
248         fprintf(fi, "%s, ", mkptr(m->type));
249         fprintf(fi, "%s, ", mkptr(m->parent));
250         if(!m->value) fprintf(fi, "0");
251         else          fprintf(fi, "&%s_constant", id);
252         fprintf(fi, "};\n");
253     }
254     
255     if(s->kind == INFOTYPE_CLASS) {
256         classinfo_t*c = (classinfo_t*)s;
257         dict_t*d = &c->members;
258         DICT_ITERATE_DATA(d, slotinfo_t*, s1) {
259             write_slotinfo(fi, s1, mkid2(id,s1->name), prefix);
260         }
261         d = &c->static_members;
262         DICT_ITERATE_DATA(d, slotinfo_t*, s2) {
263             write_slotinfo(fi, s2, mkid2(id,s2->name), prefix);
264         }
265     }
266 }
267
268 int main()
269 {
270     registry_classes = builtin_getclasses();
271
272     as3_import_abc("builtin.abc");
273     as3_import_abc("playerglobal.abc");
274
275     FILE*fi = fopen("builtin.c", "wb");
276
277     int t;
278     int pass;
279
280
281     for(pass=1;pass<=3;pass++) {
282         if(pass==1) {
283             fprintf(fi, "#include \"builtin.h\"\n");
284             fprintf(fi, "\n");
285         }
286         if(pass==3) {
287             fprintf(fi, "dict_t* builtin_getclasses()\n");
288             fprintf(fi, "{\n");
289             fprintf(fi, "    dict_t*d = dict_new2(&slotinfo_type);\n");
290         }
291         for(t=0;t<registry_classes->hashsize;t++) {
292             dictentry_t*l = registry_classes->slots[t];
293             while(l) {
294                 slotinfo_t*s = (slotinfo_t*)l->key;
295                 //printf("%08x %s %s\n", s, s->package, s->name);
296                 char*id = mkid(s);
297                 if(pass==1) {
298                     write_slotinfo_decl(fi, s, "");
299                 }
300                 if(pass==2) {
301                     write_slotinfo(fi, s, mkid(s), "");
302                 }
303                 if(pass==3) {
304                     fprintf(fi, "    dict_put(d, &%s, &%s);\n", id, id);
305                     write_initinfo(fi, s, "    ");
306                 }
307                 l = l->next;
308             }
309         }
310     }
311     fprintf(fi, "    _NaN_constant.f =  __builtin_nan(\"\");\n");
312     fprintf(fi, "    _Infinity_constant.f = __builtin_inf();\n");
313     fprintf(fi, "    return d;\n");
314     fprintf(fi, "}\n");
315     return 0;
316 }