new registry format
[swftools.git] / lib / as3 / registry.c
index b0a8d9b..ca02bec 100644 (file)
 #include "registry.h"
 #include "builtin.h"
 
-static dict_t*classes=0;
+dict_t*registry_classes=0;
 
 // ----------------------- class signature ------------------------------
 
-char class_signature_equals(class_signature_t*c1, class_signature_t*c2)
+char slotinfo_equals(slotinfo_t*c1, slotinfo_t*c2)
 {
     if(!!c1 != !!c2)
         return 0;
@@ -41,7 +41,7 @@ char class_signature_equals(class_signature_t*c1, class_signature_t*c2)
     }
     return 0;
 }
-static unsigned int class_signature_hash(class_signature_t*c)
+static unsigned int slotinfo_hash(slotinfo_t*c)
 {
     unsigned int hash = 0;
     hash = crc32_add_string(hash, c->package);
@@ -50,68 +50,151 @@ static unsigned int class_signature_hash(class_signature_t*c)
 }
 
 static void* dummy_clone(void*other) {return other;}
-static void dummy_destroy(class_signature_t*c) {}
+static void dummy_destroy(slotinfo_t*c) {}
 
-type_t class_signature_type = {
-    hash: (hash_func)class_signature_hash,
-    equals: (equals_func)class_signature_equals,
-    /* all signatures are static */
-    dup: (dup_func)dummy_clone,
-    free: (free_func)dummy_destroy,
-};
-
-// ----------------------- function signature ------------------------------
-
-static char function_signature_equals(function_signature_t*f1, function_signature_t*f2)
-{
-    return !strcmp(f1->name, f2->name);
-}
-static unsigned int function_signature_hash(function_signature_t*f)
-{
-    return crc32_add_string(0, f->name);
-}
-type_t function_signature_type = {
-    hash: (hash_func)function_signature_hash,
-    equals: (equals_func)function_signature_equals,
-    /* all signatures are static */
-    dup: (dup_func)dummy_clone,
+type_t slotinfo_type = {
+    hash: (hash_func)slotinfo_hash,
+    equals: (equals_func)slotinfo_equals,
+    dup: (dup_func)dummy_clone, // all signatures are static
     free: (free_func)dummy_destroy,
 };
 
 // ------------------------- constructors --------------------------------
 
-class_signature_t* class_signature_register(int access, char*package, char*name)
+#define AVERAGE_NUMBER_OF_MEMBERS 8
+classinfo_t* classinfo_register(int access, const char*package, const char*name, int num_interfaces)
 {
-    class_signature_t*c = malloc(sizeof(class_signature_t));
+    classinfo_t*c = rfx_calloc(sizeof(classinfo_t)+(sizeof(classinfo_t*)*(num_interfaces+1)));
+    c->interfaces[0] = 0;
+    c->kind = INFOTYPE_CLASS;
     c->access = access;
     c->package = package;
     c->name = name;
-    dict_put(classes, c, c);
+    dict_put(registry_classes, c, c);
+    dict_init(&c->members,AVERAGE_NUMBER_OF_MEMBERS);
     return c;
 }
+methodinfo_t* methodinfo_register_onclass(classinfo_t*cls, U8 access, const char*name)
+{
+    NEW(methodinfo_t,m);
+    m->kind = INFOTYPE_METHOD;
+    m->access = access;
+    m->name = name;
+    m->parent = cls;
+    dict_put(&cls->members, name, m);
+    return m;
+}
+varinfo_t* varinfo_register_onclass(classinfo_t*cls, U8 access, const char*name)
+{
+    NEW(varinfo_t,m);
+    m->kind = INFOTYPE_SLOT;
+    m->access = access;
+    m->name = name;
+    m->parent = cls;
+    dict_put(&cls->members, name, m);
+    return m;
+}
+methodinfo_t* methodinfo_register_global(U8 access, const char*package, const char*name)
+{
+    NEW(methodinfo_t, m);
+    m->kind = INFOTYPE_METHOD;
+    m->flags = FLAG_STATIC;
+    m->access = access;
+    m->package = package;
+    m->name = name;
+    m->parent = 0;
+    dict_put(registry_classes, m, m);
+    return m;
+}
+varinfo_t* varinfo_register_global(U8 access, const char*package, const char*name)
+{
+    NEW(varinfo_t, m);
+    m->kind = INFOTYPE_SLOT;
+    m->flags = FLAG_STATIC;
+    m->access = access;
+    m->package = package;
+    m->name = name;
+    m->parent = 0;
+    dict_put(registry_classes, m, m);
+    return m;
+}
 
 // --------------- builtin classes (from builtin.c) ----------------------
 
 void registry_init()
 {
-    classes = builtin_getclasses();
+    if(!registry_classes)
+        registry_classes = builtin_getclasses();
 }
-class_signature_t* registry_safefindclass(const char*package, const char*name)
+slotinfo_t* registry_find(const char*package, const char*name)
 {
-    class_signature_t*c = registry_findclass(package, name);
-    assert(c);
+    assert(registry_classes);
+    slotinfo_t tmp;
+    tmp.package = package;
+    tmp.name = name;
+    slotinfo_t* c = (slotinfo_t*)dict_lookup(registry_classes, &tmp);
+    /*if(c)
+        printf("%s.%s->%08x (%s.%s)\n", package, name, c, c->package, c->name);*/
     return c;
 }
-class_signature_t* registry_findclass(const char*package, const char*name)
+slotinfo_t* registry_safefind(const char*package, const char*name)
 {
-    assert(classes);
-    class_signature_t tmp;
-    tmp.package = package;
-    tmp.name = name;
-    class_signature_t* c = (class_signature_t*)dict_lookup(classes, &tmp);
+    slotinfo_t*c = registry_find(package, name);
+    if(!c) {
+        printf("%s.%s\n", package, name);
+    }
+    assert(c);
     return c;
 }
-void registry_fill_multiname(multiname_t*m, namespace_t*n, class_signature_t*c)
+void registry_dump()
+{
+    int t;
+    for(t=0;t<registry_classes->hashsize;t++) {
+        dictentry_t*e = registry_classes->slots[t];
+        while(e) {
+            slotinfo_t*i = (slotinfo_t*)e->key;
+            printf("[%s] %s.%s\n", access2str(i->access), i->package, i->name);
+            e = e->next;
+        }
+    }
+}
+
+memberinfo_t* registry_findmember(classinfo_t*cls, const char*name, char recursive)
+{
+    if(!recursive) {
+        return (memberinfo_t*)dict_lookup(&cls->members, name);
+    }
+    /* look at classes directly extended by this class */
+    slotinfo_t*m = 0;
+    classinfo_t*s = cls;
+
+    if(recursive>1) // check *only* superclasses
+        s = s->superclass;
+
+    while(s) {
+        m = (slotinfo_t*)dict_lookup(&s->members, name);
+        if(m) {
+            return (memberinfo_t*)m;
+        }
+        s = s->superclass;
+    }
+    /* look at interfaces, and parent interfaces */
+    int t=0;
+    while(cls->interfaces[t]) {
+        classinfo_t*s = cls->interfaces[t];
+        while(s) {
+            m = (slotinfo_t*)dict_lookup(&s->members, name);
+            if(m) {
+                return (memberinfo_t*)m;
+            }
+            s = s->superclass;
+        }
+        t++;
+    }
+    return 0;
+
+}
+void registry_fill_multiname(multiname_t*m, namespace_t*n, slotinfo_t*c)
 {
     m->type = QNAME;
     m->ns = n;
@@ -120,31 +203,141 @@ void registry_fill_multiname(multiname_t*m, namespace_t*n, class_signature_t*c)
     m->name = c->name;
     m->namespace_set = 0;
 }
-multiname_t* class_signature_to_multiname(class_signature_t*cls)
+multiname_t* classinfo_to_multiname(slotinfo_t*cls)
 {
     if(!cls)
         return 0;
     multiname_t*m=0;
-    namespace_t*ns = namespace_new(cls->access, cls->package);
-    return multiname_new(ns,cls->name);
+    namespace_t ns = {cls->access, (char*)cls->package};
+    return multiname_new(&ns,cls->name);
 }
 
+// ----------------------- memberinfo methods ------------------------------
+
+/* hacky code to wrap a variable or function into a "type"
+   object, but keep a pointer to the "value" */
+static dict_t* functionobjects = 0;
+classinfo_t* slotinfo_asclass(slotinfo_t*f) {
+    if(!functionobjects) {
+        functionobjects = dict_new2(&ptr_type);
+    } else {
+        classinfo_t*c = dict_lookup(functionobjects, f);
+        if(c)
+            return c;
+    }
+
+    classinfo_t*c = rfx_calloc(sizeof(classinfo_t)+sizeof(classinfo_t*));
+    c->access = ACCESS_PUBLIC;
+    c->package = "";
+    if(f->kind == INFOTYPE_METHOD)
+        c->name = "Function";
+    else if(f->kind == INFOTYPE_CLASS)
+        c->name = "Class";
+    else if(f->kind == INFOTYPE_SLOT)
+        c->name = "Object";
+    else {
+        c->name = "undefined";
+    }
+    
+    dict_init(&c->members,1);
+    c->data = f;
+    dict_put(functionobjects, f, c);
+    return c;
+}
+
+classinfo_t* slotinfo_gettype(slotinfo_t*f)
+{
+    if(f) {
+       if(f->kind == INFOTYPE_METHOD) {
+           return slotinfo_asclass(f);
+       } else if(f->kind == INFOTYPE_SLOT) {
+           varinfo_t*v = (varinfo_t*)f;
+           return v->type;
+       } else 
+           return 0;
+    } else {
+       return registry_getanytype();
+    }
+}
 // ----------------------- builtin types ------------------------------
-class_signature_t* registry_getanytype() {return 0;/*FIXME*/}
+classinfo_t* registry_getanytype() {return 0;}
 
-class_signature_t* registry_getobjectclass() {return registry_safefindclass("", "Object");}
-class_signature_t* registry_getstringclass() {return registry_safefindclass("", "String");}
-class_signature_t* registry_getintclass() {return registry_safefindclass("", "int");}
-class_signature_t* registry_getuintclass() {return registry_safefindclass("", "uint");}
-class_signature_t* registry_getbooleanclass() {return registry_safefindclass("", "Boolean");}
-class_signature_t* registry_getnumberclass() {return registry_safefindclass("", "Number");}
-class_signature_t* registry_getMovieClip() {return registry_safefindclass("flash.display", "MovieClip");}
+char registry_isfunctionclass(classinfo_t*c) {
+    return (c && c->package && c->name && 
+            !strcmp(c->package, "") && !strcmp(c->name, "Function"));
+}
+char registry_isclassclass(classinfo_t*c) {
+    return (c && c->package && c->name && 
+            !strcmp(c->package, "") && !strcmp(c->name, "Class"));
+}
+
+classinfo_t* registry_getobjectclass() {
+    static classinfo_t*c = 0;
+    if(!c) c = (classinfo_t*)registry_safefind("", "Object");
+    return c;
+}
+classinfo_t* registry_getstringclass() {
+    static classinfo_t*c = 0;
+    if(!c) c = (classinfo_t*)registry_safefind("", "String");
+    return c;
+}
+classinfo_t* registry_getarrayclass() {
+    static classinfo_t*c = 0;
+    if(!c) c = (classinfo_t*)registry_safefind("", "Array");
+    return c;
+}
+classinfo_t* registry_getintclass() {
+    static classinfo_t*c = 0;
+    if(!c) c = (classinfo_t*)registry_safefind("", "int");
+    return c;
+}
+classinfo_t* registry_getuintclass() {
+    static classinfo_t*c = 0;
+    if(!c) c = (classinfo_t*)registry_safefind("", "uint");
+    return c;
+}
+classinfo_t* registry_getbooleanclass() {
+    static classinfo_t*c = 0;
+    if(!c) c = (classinfo_t*)registry_safefind("", "Boolean");
+    return c;
+}
+classinfo_t* registry_getnumberclass() {
+    static classinfo_t*c = 0;
+    if(!c) c = (classinfo_t*)registry_safefind("", "Number");
+    return c;
+}
+classinfo_t* registry_getregexpclass() {
+    static classinfo_t*c = 0;
+    if(!c) c = (classinfo_t*)registry_safefind("", "RegExp");
+    return c;
+}
+classinfo_t* registry_getMovieClip() {
+    static classinfo_t*c = 0;
+    if(!c) c = (classinfo_t*)registry_safefind("flash.display", "MovieClip");
+    return c;
+}
 
 // ----------------------- builtin dummy types -------------------------
-class_signature_t nullclass = {
-    ACCESS_PACKAGE, 0, "", "null", 0, 0, 0,
+classinfo_t nullclass = {
+    INFOTYPE_CLASS,0,0,ACCESS_PACKAGE, "", "null", 0, 0, 0
 };
-class_signature_t* registry_getnullclass() {
+classinfo_t* registry_getnullclass() {
     return &nullclass;
 }
 
+namespace_t access2namespace(U8 access, char*package)
+{
+    namespace_t ns;
+    ns.access = access;
+    ns.name = package;
+    return ns;
+}
+
+char* infotypename(slotinfo_t*s)
+{
+    if(s->kind == INFOTYPE_CLASS) return "class";
+    else if(s->kind == INFOTYPE_SLOT) return "member";
+    else if(s->kind == INFOTYPE_METHOD) return "method";
+    else return "object";
+}
+