godot-cpp/binding_generator.py

#!python

import json

# comment.

classes = []

def generate_bindings(path):
    
    global classes
    classes = json.load(open(path))
    
    icalls = set()
    
    for c in classes:
        # print c['name']
        used_classes = get_used_classes(c)
        
        header = generate_class_header(used_classes, c)
        
        impl = generate_class_implementation(icalls, used_classes, c)
        
        header_file = open("include/" + strip_name(c["name"]) + ".hpp", "w+")
        header_file.write(header)
        
        source_file = open("src/" + strip_name(c["name"]) + ".cpp", "w+")
        source_file.write(impl)
    
    
    icall_header_file = open("src/__icalls.hpp", "w+")
    icall_header_file.write(generate_icall_header(icalls))
    
    icall_source_file = open("src/__icalls.cpp", "w+")
    icall_source_file.write(generate_icall_implementation(icalls))


def is_reference_type(t):
    for c in classes:
        if c['name'] != t:
            continue
        if c['is_reference']:
            return True
    return False

def make_gdnative_type(t):
    if is_enum(t):
        return remove_enum_prefix(t) + " "
    elif is_class_type(t):
        if is_reference_type(t):
            return "Ref<" + strip_name(t) + "> "
        else:
            return strip_name(t) + " *"
    else:
        if t == "int":
            return "int64_t "
        if t == "float" or t == "real":
            return "double "
        return strip_name(t) + " "

def generate_class_header(used_classes, c):

    source = []
    source.append("#ifndef GODOT_CPP_" + strip_name(c["name"]).upper() + "_HPP")
    source.append("#define GODOT_CPP_" + strip_name(c["name"]).upper() + "_HPP")
    source.append("")
    source.append("")
    
    source.append("#include <gdnative_api_struct.gen.h>")
    source.append("#include <stdint.h>")
    source.append("")
    
    
    source.append("#include <core/CoreTypes.hpp>")

    class_name = strip_name(c["name"])

    # Ref<T> is not included in object.h in Godot either,
    # so don't include it here because it's not needed
    if class_name != "Object" and class_name != "Reference":
        source.append("#include <core/Ref.hpp>")


    included = []

    for used_class in used_classes:
        if is_enum(used_class) and is_nested_type(used_class):
            used_class_name = remove_enum_prefix(extract_nested_type(used_class))
            if used_class_name not in included:
                included.append(used_class_name)
                source.append("#include \"" + used_class_name + ".hpp\"")
        elif is_enum(used_class) and is_nested_type(used_class) and not is_nested_type(used_class, class_name):
            used_class_name = remove_enum_prefix(used_class)
            if used_class_name not in included:
                included.append(used_class_name)
                source.append("#include \"" + used_class_name + ".hpp\"")

    source.append("")
    
    if c["base_class"] != "":
        source.append("#include \"" + strip_name(c["base_class"]) + ".hpp\"")
        
        
    
    source.append("namespace godot {")
    source.append("")


    for used_type in used_classes:
        if is_enum(used_type) or is_nested_type(used_type, class_name):
            continue
        else:
            source.append("class " + strip_name(used_type) + ";")
    
    
    source.append("")
    
    
    # generate the class definition here
    source.append("class " + class_name + ("" if c["base_class"] == "" else (" : public " + strip_name(c["base_class"])) ) + " {")
    
    source.append("public:")
    source.append("")

    # ___get_class_name
    source.append("\tstatic inline char *___get_class_name() { return (char *) \"" + strip_name(c["name"]) + "\"; }")

    source.append("\tstatic inline Object *___get_from_variant(Variant a) { return (Object *) a; }")

    enum_values = []

    source.append("\n\t// enums")
    for enum in c["enums"]:
        source.append("\tenum class " + strip_name(enum["name"]) + " {")
        for value in enum["values"]:
            source.append("\t\t" + remove_nested_type_prefix(value) + " = " + str(enum["values"][value]) + ",")
            enum_values.append(value)
        source.append("\t};")

    source.append("\n\t// constants")
    
    for name in c["constants"]:
        if name not in enum_values:
            source.append("\tconst static int " + name + " = " + str(c["constants"][name]) + ";")

    
    if c["instanciable"]:
        source.append("\tstatic void *operator new(size_t);")
        
        source.append("\tstatic void operator delete(void *);")
    
    source.append("\n\t// methods")
    
    for method in c["methods"]:
        
        method_signature = ""
        
        method_signature += "static " if c["singleton"] else ""
        method_signature += make_gdnative_type(method["return_type"])
        method_signature += escape_cpp(method["name"]) + "("
            
            
        has_default_argument = False
        
        for i, argument in enumerate(method["arguments"]):
            method_signature += "const " + make_gdnative_type(argument["type"])
            method_signature += escape_cpp(argument["name"])
            
            
            # default arguments
            def escape_default_arg(_type, default_value):
                if _type == "Color":
                    return "Color(" + default_value + ")"
                if _type == "bool" or _type == "int":
                    return default_value.lower()
                if _type == "Array":
                    return "Array()"
                if _type in ["PoolVector2Array", "PoolStringArray", "PoolVector3Array", "PoolColorArray"]:
                    return _type + "()"
                if _type == "Vector2":
                    return "Vector2" + default_value
                if _type == "Vector3":
                    return "Vector3" + default_value
                if _type == "Transform":
                    return "Transform()"
                if _type == "Transform2D":
                    return "Transform2D()"
                if _type == "Rect2":
                    return "Rect2" + default_value
                if _type == "Variant":
                    return "Variant()" if default_value == "Null" else default_value
                if _type == "String":
                    return "\"" + default_value + "\""
                if _type == "RID":
                    return "RID()"
                
                if default_value == "Null" or default_value == "[Object:null]":
                    return "nullptr"
                    
                return default_value
                    
            
            
            
            if argument["has_default_value"] or has_default_argument:
                method_signature += " = " + escape_default_arg(argument["type"], argument["default_value"])
                has_default_argument = True
            
            
            
            if i != len(method["arguments"]) - 1:
                method_signature += ", "
                
        if method["has_varargs"]:
            if len(method["arguments"]) > 0:
                method_signature += ", "
            method_signature += "const Array& __var_args = Array()"
            
        method_signature += ")" + (" const" if method["is_const"] and not c["singleton"] else "")
            

        source.append("\t" + method_signature + ";")
    
    
    source.append("};")
    source.append("")
    
    
    source.append("}")
    source.append("")
    
    
    
    source.append("#endif")
    
    
    return "\n".join(source)
 




def generate_class_implementation(icalls, used_classes, c):
    class_name = strip_name(c["name"])
    source = []
    source.append("#include \"" + class_name + ".hpp\"")
    source.append("")
    source.append("")
    
    source.append("#include <core/GodotGlobal.hpp>")
    source.append("#include <core/CoreTypes.hpp>")
    source.append("#include <core/Ref.hpp>")
    
    source.append("#include <core/Godot.hpp>")
    source.append("")
    
    
    source.append("#include \"__icalls.hpp\"")
    source.append("")
    source.append("")
    
    for used_class in used_classes:
        if is_enum(used_class):
            continue
        else:
            source.append("#include \"" + strip_name(used_class) + ".hpp\"")
    
    source.append("")
    source.append("")
    
    source.append("namespace godot {")
    
    
    core_object_name = ("___static_object_" + strip_name(c["name"])) if c["singleton"] else "this"
    
    
    source.append("")
    source.append("")
    
    if c["singleton"]:
        source.append("static godot_object *" + core_object_name + ";")
        source.append("")
        source.append("")
        
        # FIXME Test if inlining has a huge impact on binary size
        source.append("static inline void ___singleton_init()")
        source.append("{")
        source.append("\tif (" + core_object_name + " == nullptr) {")
        source.append("\t\t" + core_object_name + " = godot::api->godot_global_get_singleton((char *) \"" + strip_name(c["name"]) + "\");")
        source.append("\t}")
        source.append("}")
        
        source.append("")
        source.append("")
    
    
    
    if c["instanciable"]:
        source.append("void *" + strip_name(c["name"]) + "::operator new(size_t)")
        source.append("{")
        source.append("\treturn godot::api->godot_get_class_constructor((char *)\"" + c["name"] + "\")();")
        source.append("}")
        
        source.append("void " + strip_name(c["name"]) + "::operator delete(void *ptr)")
        source.append("{")
        source.append("\tgodot::api->godot_object_destroy((godot_object *)ptr);")
        source.append("}")
    
    for method in c["methods"]:
        method_signature = ""
        
        method_signature += make_gdnative_type(method["return_type"])
        method_signature += strip_name(c["name"]) + "::" + escape_cpp(method["name"]) + "("
        
        for i, argument in enumerate(method["arguments"]):
            method_signature += "const " + make_gdnative_type(argument["type"])
            method_signature += escape_cpp(argument["name"])
            
            if i != len(method["arguments"]) - 1:
                method_signature += ", "
                
        if method["has_varargs"]:
            if len(method["arguments"]) > 0:
                method_signature += ", "
            method_signature += "const Array& __var_args"
            
        method_signature += ")" + (" const" if method["is_const"] and not c["singleton"] else "")
    
        source.append(method_signature + " {")
        
        
        
        if c["singleton"]:
            source.append("\t___singleton_init();")
        
        
        source.append("\tstatic godot_method_bind *mb = nullptr;")
        source.append("\tif (mb == nullptr) {")
        source.append("\t\tmb = godot::api->godot_method_bind_get_method(\"" + c["name"] +"\", \"" + method["name"] + "\");")
        source.append("\t}")
        
        return_statement = ""
        
        if method["return_type"] != "void":
            if is_class_type(method["return_type"]):
                if is_enum(method["return_type"]):
                    return_statement += "return (" + remove_enum_prefix(method["return_type"]) + ") "
                elif is_reference_type(method["return_type"]):
                    return_statement += "return Ref<" + strip_name(method["return_type"]) + ">::__internal_constructor(";
                else:
                    return_statement += "return " + ("(" + strip_name(method["return_type"]) + " *) " if is_class_type(method["return_type"]) else "")
            else:
                return_statement += "return "
        
        def get_icall_type_name(name):
            if is_enum(name):
                return "int"
            if is_class_type(name):
                return "Object"
            return name
        
        
        
        if method["has_varargs"]:

            if len(method["arguments"]) != 0:
                source.append("\tVariant __given_args[" + str(len(method["arguments"])) + "];")
            
            for i, argument in enumerate(method["arguments"]):
                source.append("\tgodot::api->godot_variant_new_nil((godot_variant *) &__given_args[" + str(i) + "]);")
                
            source.append("")
            
            
            for i, argument in enumerate(method["arguments"]):
                source.append("\t__given_args[" + str(i) + "] = " + escape_cpp(argument["name"]) + ";")
            
            source.append("")
            
            size = ""
            if method["has_varargs"]:
                size = "(__var_args.size() + " + str(len(method["arguments"])) + ")"
            else:
                size = "(" + str(len(method["arguments"])) + ")"
            
            source.append("\tgodot_variant **__args = (godot_variant **) alloca(sizeof(godot_variant *) * " + size + ");")
            
            source.append("")
            
            for i, argument in enumerate(method["arguments"]):
                source.append("\t__args[" + str(i) + "] = (godot_variant *) &__given_args[" + str(i) + "];")
                
            source.append("")
            
            if method["has_varargs"]:
                source.append("\tfor (int i = 0; i < __var_args.size(); i++) {")
                source.append("\t\t__args[i + " + str(len(method["arguments"])) + "] = (godot_variant *) &((Array &) __var_args)[i];")
                source.append("\t}")
           
            source.append("")
            
            source.append("\tVariant __result;")
            source.append("\t*(godot_variant *) &__result = godot::api->godot_method_bind_call(mb, (godot_object *) " + core_object_name + ", (const godot_variant **) __args, " + size + ", nullptr);")
            
            source.append("")
            
            
            for i, argument in enumerate(method["arguments"]):
                source.append("\tgodot::api->godot_variant_destroy((godot_variant *) &__given_args[" + str(i) + "]);")
                
            source.append("")
            
            if method["return_type"] != "void":
                cast = ""
                if is_class_type(method["return_type"]):
                    if is_reference_type(method["return_type"]):
                        cast += "Ref<" + strip_name(method["return_type"]) + ">::__internal_constructor(__result);"
                    else:
                        cast += "(" + strip_name(method["return_type"]) + " *) (Object *) __result;"
                else:
                    cast += "__result;"
                source.append("\treturn " + cast)
            
            
            
        else:
        
            args = []
            for arg in method["arguments"]:
                args.append(get_icall_type_name(arg["type"]))
            
            icall_ret_type = get_icall_type_name(method["return_type"])
            
            icall_sig = tuple((icall_ret_type, tuple(args)))
            
            icalls.add(icall_sig)
            
            icall_name = get_icall_name(icall_sig)
            
            return_statement += icall_name + "(mb, (godot_object *) " + core_object_name
            
            for arg in method["arguments"]:
                return_statement += ", " + escape_cpp(arg["name"]) + (".ptr()" if is_reference_type(arg["type"]) else "")
                
            return_statement += ")"
            
            source.append("\t" + return_statement + (")" if is_reference_type(method["return_type"]) else "") + ";")
        
        source.append("}")
        source.append("")

    
    
    source.append("}")
    
    
    return "\n".join(source)





def generate_icall_header(icalls):
    
    source = []
    source.append("#ifndef GODOT_CPP__ICALLS_HPP")
    source.append("#define GODOT_CPP__ICALLS_HPP")
    
    source.append("")
    
    source.append("#include <gdnative_api_struct.gen.h>")
    source.append("#include <stdint.h>")
    source.append("")
    
    source.append("#include <core/CoreTypes.hpp>")
    source.append("#include \"Object.hpp\"")
    source.append("")
    source.append("")
    
    source.append("namespace godot {")
    source.append("")
    
    for icall in icalls:
        ret_type = icall[0]
        args = icall[1]
        
        method_signature = ""
        
        method_signature += return_type(ret_type) + get_icall_name(icall) + "(godot_method_bind *mb, godot_object *inst"
        
        for arg in args:
            method_signature += ", const "
            
            if is_core_type(arg):
                method_signature += arg + "&"
            elif is_primitive(arg):
                method_signature += arg
            else:
                method_signature += "Object *"
        
        method_signature += ");"
        
        source.append(method_signature)
    
    source.append("")
    
    source.append("}")
    source.append("")
    
    source.append("#endif")
    
    return "\n".join(source)
    

def generate_icall_implementation(icalls):
    source = []
    source.append("#include \"__icalls.hpp\"")
    
    source.append("")
    
    source.append("#include <gdnative_api_struct.gen.h>")
    source.append("#include <stdint.h>")
    source.append("")
    
    source.append("#include <core/GodotGlobal.hpp>")
    source.append("#include <core/CoreTypes.hpp>")
    source.append("#include \"Object.hpp\"")
    source.append("")
    source.append("")
    
    source.append("namespace godot {")
    source.append("")
    
    for icall in icalls:
        ret_type = icall[0]
        args = icall[1]
        
        method_signature = ""
        
        method_signature += return_type(ret_type) + get_icall_name(icall) + "(godot_method_bind *mb, godot_object *inst"
        
        for i, arg in enumerate(args):
            method_signature += ", const "
            
            if is_core_type(arg):
                method_signature += arg + "& "
            elif is_primitive(arg):
                method_signature += arg + " "
            else:
                method_signature += "Object *"
            
            method_signature += "arg" + str(i)
        
        method_signature += ")"
        
        source.append(method_signature + " {")
        
        if ret_type != "void":
            source.append("\t" + return_type(ret_type) + "ret;")
            if is_class_type(ret_type):
                source.append("\tret = nullptr;")
        
        
        source.append("\tconst void *args[" + ("1" if len(args) == 0 else "") + "] = {")
        
        for i, arg in enumerate(args):
            
            wrapped_argument = "\t\t"
            if is_primitive(arg) or is_core_type(arg):
                wrapped_argument += "(void *) &arg" + str(i)
            else:
                wrapped_argument += "(void *) arg" + str(i)
            
            wrapped_argument += ","
            source.append(wrapped_argument)
        
        source.append("\t};")
        source.append("")
        
        source.append("\tgodot::api->godot_method_bind_ptrcall(mb, inst, args, " + ("nullptr" if ret_type == "void" else "&ret") + ");")
        
        if ret_type != "void":
            source.append("\treturn ret;")
        
        source.append("}")
    
    source.append("}")
    source.append("")
    
    return "\n".join(source)



















def return_type(t):
    if is_class_type(t):
        return "Object *"
    if t == "int":
        return "int64_t "
    if t == "float" or t == "real":
        return "double "
    return t + " "


def get_icall_name(sig):
    ret_type = sig[0]
    args = sig[1]
    
    name = "___godot_icall_"
    name += strip_name(ret_type)
    for arg in args:
        name += "_" + strip_name(arg)
    
    return name






def get_used_classes(c):
    classes = []
    for method in c["methods"]:
        if is_class_type(method["return_type"]) and not (method["return_type"] in classes):
            classes.append(method["return_type"])
        
        for arg in method["arguments"]:
            if is_class_type(arg["type"]) and not (arg["type"] in classes):
                classes.append(arg["type"])
    return classes








def strip_name(name):
    if name[0] == '_':
        return name[1:]
    return name

def extract_nested_type(nested_type):
    return strip_name(nested_type[:nested_type.find("::")])

def remove_nested_type_prefix(name):
    return name if name.find("::") == -1 else strip_name(name[name.find("::") + 2:])

def remove_enum_prefix(name):
    return strip_name(name[name.find("enum.") + 5:])

def is_nested_type(name, type = ""):
    return name.find(type + "::") != -1

def is_enum(name):
    return name.find("enum.") == 0

def is_class_type(name):
    return not is_core_type(name) and not is_primitive(name)

def is_core_type(name):
    core_types = ["Array",
                  "Basis",
                  "Color",
                  "Dictionary",
                  "Error",
                  "NodePath",
                  "Plane",
                  "PoolByteArray",
                  "PoolIntArray",
                  "PoolRealArray",
                  "PoolStringArray",
                  "PoolVector2Array",
                  "PoolVector3Array",
                  "PoolColorArray",
                  "Quat",
                  "Rect2",
                  "AABB",
                  "RID",
                  "String",
                  "Transform",
                  "Transform2D",
                  "Variant",
                  "Vector2",
                  "Vector3"]
    return name in core_types




def is_primitive(name):
    core_types = ["int", "bool", "real", "float", "void"]
    return name in core_types

def escape_cpp(name):
    escapes = {
        "class":    "_class",
        "char":     "_char",
        "short":    "_short",
        "bool":     "_bool",
        "int":      "_int",
        "default":  "_default",
        "case":     "_case",
        "switch":   "_switch",
        "export":   "_export",
        "template": "_template",
        "new":      "new_",
        "operator": "_operator"
    }
    if name in escapes:
        return escapes[name]
    return name