made builtin classes more complete
[swftools.git] / lib / as3 / mklib.c
1 /* code.c
2
3    File to generate builtin.c
4
5    Copyright (c) 2008 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     char*id = malloc(strlen(package)+strlen(name)+10);
70     strcpy(id, "&");
71     strcat(id, package);
72     strcat(id, "_");
73     strcat(id, name);
74     fixstring(id+1);
75     return id;
76 }
77 static char* mkid2(const char*cls, const char*member)
78 {
79     char*id = malloc(strlen(cls)+strlen(member)+10);
80     strcpy(id, cls);
81     strcat(id, "_");
82     strcat(id, member);
83     fixstring(id);
84     return id;
85 }
86 #define mkid(s) ((s)?mkcid((s)->package, (s)->name):"0")
87 #define mkptr(s) ((s)?mkptr2((s)->package, (s)->name):"0")
88
89 static array_t*tosort=0;
90 static int compare_classes(const void*v1, const void*v2)
91 {
92     int x1 = *(int*)v1;
93     int x2 = *(int*)v2;
94     abc_class_t*c1 = array_getvalue(tosort, x1);
95     abc_class_t*c2 = array_getvalue(tosort, x2);
96     int i = strcmp(c1->classname->ns->name, c2->classname->ns->name);
97     if(i)
98         return i;
99     return strcmp(c1->classname->name, c2->classname->name);
100 }
101
102 static int compare_traits(const void*v1, const void*v2)
103 {
104     trait_t* x1 = *(trait_t**)v1;
105     trait_t* x2 = *(trait_t**)v2;
106     int i = strcmp(x1->name->ns->name, x2->name->ns->name);
107     if(i)
108         return i;
109     return strcmp(x1->name->name, x2->name->name);
110 }
111
112 dict_t* builtin_getclasses()
113 {
114     return dict_new2(&slotinfo_type);
115 }
116
117 extern dict_t*registry_classes;
118
119 char*mktype(slotinfo_t*s) 
120 {
121     if(s->kind == INFOTYPE_CLASS) {
122         return "classinfo_t";
123     } else if(s->kind == INFOTYPE_METHOD) {
124         return "methodinfo_t";
125     } else if(s->kind == INFOTYPE_SLOT) {
126         return "varinfo_t";
127     }
128 }
129
130 void write_slotinfo(FILE*fi, slotinfo_t*s, char*id, char*prefix);
131
132 void write_slotinfo_decl(FILE*fi, slotinfo_t*s, char*prefix)
133 {
134     fprintf(fi, "%s", prefix);
135     char*id = mkid(s);
136     fprintf(fi, "static %s %s;\n", mktype(s), id);
137
138     if(s->kind == INFOTYPE_CLASS) {
139         classinfo_t*c = (classinfo_t*)s;
140         dict_t*d = &c->members;
141         int t;
142         for(t=0;t<d->hashsize;t++) {
143             dictentry_t*l = d->slots[t];
144             while(l) {
145                 slotinfo_t*s2 = (slotinfo_t*)l->data;
146                 fprintf(fi, "static %s %s;\n", mktype(s2), mkid2(id, s2->name));
147                 l = l->next;
148             }
149         }
150     }
151 }
152 void write_initinfo(FILE*fi, slotinfo_t*s, char*prefix)
153 {
154     if(s->kind == INFOTYPE_CLASS) {
155         classinfo_t*c = (classinfo_t*)s;
156         fprintf(fi, "%s", prefix);
157         char*id = mkid(c);
158         dict_t*d = &c->members;
159         fprintf(fi, "dict_init(&%s.members, %d);\n", id, d->hashsize);
160         int t;
161         for(t=0;t<d->hashsize;t++) {
162             dictentry_t*l = d->slots[t];
163             while(l) {
164                 slotinfo_t*s2 = (slotinfo_t*)l->data;
165                 fprintf(fi, "%s", prefix);
166                 fprintf(fi, "dict_put(&%s.members, \"%s\", &%s);\n", id, s2->name, mkid2(id, s2->name));
167                 l = l->next;
168             }
169         }
170     }
171 }
172 void write_slotinfo(FILE*fi, slotinfo_t*s, char*id, char*prefix)
173 {
174     fprintf(fi, "%s", prefix);
175     fprintf(fi, "static %s %s = {", mktype(s), id);
176     fprintf(fi, "0x%02x, 0x%02x, 0x%02x, 0x%02x, ", s->kind, s->subtype, s->flags, s->access);
177     if(s->package)
178         fprintf(fi, "\"%s\", ", s->package);
179     else
180         fprintf(fi, "0, ");
181     
182     if(s->name)
183         fprintf(fi, "\"%s\", ", s->name);
184     else
185         fprintf(fi, "0, ");
186
187     fprintf(fi, "%d, ", s->slot);
188
189     if(s->kind == INFOTYPE_CLASS) {
190         classinfo_t*c = (classinfo_t*)s;
191         fprintf(fi, "%s, ", mkptr(c->superclass));
192         fprintf(fi, "interfaces: {");
193         int t;
194         for(t=0;c->interfaces[t];t++) {
195             fprintf(fi, "%s", mkptr(c->interfaces[t]));
196             fprintf(fi, ", ");
197         }
198         fprintf(fi, "0}};\n");
199     }
200     if(s->kind == INFOTYPE_METHOD) {
201         methodinfo_t*m = (methodinfo_t*)s;
202         fprintf(fi, "%s, ", mkptr(m->return_type));
203         fprintf(fi, "%s, ", mkptr(m->parent));
204         fprintf(fi, "0"); // params TODO
205         fprintf(fi, "};\n");
206     }
207     if(s->kind == INFOTYPE_SLOT) {
208         varinfo_t*m = (varinfo_t*)s;
209         fprintf(fi, "%s, ", mkptr(m->type));
210         fprintf(fi, "%s, ", mkptr(m->parent));
211         fprintf(fi, "0"); // value TODO
212         fprintf(fi, "};\n");
213     }
214     
215     if(s->kind == INFOTYPE_CLASS) {
216         classinfo_t*c = (classinfo_t*)s;
217         dict_t*d = &c->members;
218         int t;
219         for(t=0;t<d->hashsize;t++) {
220             dictentry_t*l = d->slots[t];
221             while(l) {
222                 slotinfo_t*s2 = (slotinfo_t*)l->data;
223                 write_slotinfo(fi, s2, mkid2(id,s2->name), prefix);
224                 l = l->next;
225             }
226         }
227     }
228 }
229
230 int main()
231 {
232     registry_classes = builtin_getclasses();
233
234     as3_import_abc("/home/kramm/c/swftools/lib/as3/builtin.abc");
235     as3_import_abc("/home/kramm/c/swftools/lib/as3/playerglobal.abc");
236
237     FILE*fi = fopen("builtin.c", "wb");
238
239     int t;
240     int pass;
241
242
243     for(pass=1;pass<=3;pass++) {
244         if(pass==1) {
245             fprintf(fi, "#include \"builtin.h\"\n");
246             fprintf(fi, "\n");
247         }
248         if(pass==3) {
249             fprintf(fi, "dict_t* builtin_getclasses()\n");
250             fprintf(fi, "{\n");
251             fprintf(fi, "    dict_t*d = dict_new2(&slotinfo_type);\n");
252         }
253         for(t=0;t<registry_classes->hashsize;t++) {
254             dictentry_t*l = registry_classes->slots[t];
255             while(l) {
256                 slotinfo_t*s = (slotinfo_t*)l->key;
257                 //printf("%08x %s %s\n", s, s->package, s->name);
258                 char*id = mkid(s);
259                 if(pass==1) {
260                     write_slotinfo_decl(fi, s, "");
261                 }
262                 if(pass==2) {
263                     write_slotinfo(fi, s, mkid(s), "");
264                 }
265                 if(pass==3) {
266                     fprintf(fi, "    dict_put(d, &%s, &%s);\n", id, id);
267                     write_initinfo(fi, s, "    ");
268                 }
269                 l = l->next;
270             }
271         }
272     }
273     fprintf(fi, "    return d;\n");
274     fprintf(fi, "}\n");
275
276 }