#!/usr/bin/python3
import argparse
import code
import json
import os
import readline
import signal
import sys
import datetime
import xml.etree.ElementTree
C_STRUCT = ['typedef struct XXX {', '}XXX;']
HEADER_GUARD = ['\n#ifndef _XXX_H\n#define _XXX_H\n', '#endif //end of inclusion guard\n\n']
EXTERN_C = ['#ifdef __cplusplus\nextern "C" {\n#endif\n', '#ifdef __cplusplus\n}\n#endif //end of extern c\n']
BEGIN_NOTE = "//Generated Automatically by luatablegen."
HEADER_LIST = ['#include "HHHlua.h"\n', '#include "HHHlauxlib.h"\n',
'#include "HHHlualib.h"\n', '#include <inttypes.h>\n',
'#include <stdbool.h>\n']
CONVERT = ['static XXX* convert_XXX (lua_State* __ls, int index) {\n',
'\tXXX* dummy = (XXX*)lua_touserdata(__ls, index);\n',
'\tif (dummy == NULL) printf("XXX:bad user data type.\\n");\n',
'\treturn dummy;\n}\n']
CHECK = ['static XXX* check_XXX(lua_State* __ls, int index) {\n',
'\tXXX* dummy;\n',
'\tdummy = (XXX*)luaL_checkudata(__ls, index, "XXX");\n',
'\tif (dummy == NULL) printf("XXX:bad user data type.\\n");\n',
'\treturn dummy;\n}\n']
#CHECK = ['static XXX* check_XXX(lua_State* __ls, int index) {\n',
# '\tXXX* dummy;\n',
# '\tluaL_checktype(__ls, index, LUA_TUSERDATA);\n'
# '\tdummy = (XXX*)luaL_checkudata(__ls, index, "XXX");\n',
# '\tif (dummy == NULL) printf("XXX:bad user data type.\\n");\n',
# '\treturn dummy;\n}\n']
PUSH_SELF = [ 'XXX* push_XXX(lua_State* __ls) {\n',
'\tlua_checkstack(__ls, 3);\n',
'\tXXX* dummy = lua_newuserdata(__ls, sizeof(XXX));\n',
'\tluaL_getmetatable(__ls, "XXX");\n',
'\tlua_setmetatable(__ls, -2);\n',
'\tlua_pushlightuserdata(__ls, dummy);\n',
'\tlua_pushvalue(__ls, -2);\n',
'\tlua_settable(__ls, LUA_REGISTRYINDEX);\n',
'\treturn dummy;\n}\n']
PUSH_ARGS = ['int XXX_push_args(lua_State* __ls, XXX* _st) {\n',
'\tlua_checkstack(__ls, NNN);\n', '\treturn NNN;\n}\n']
NEW = ['int new_XXX(lua_State* __ls) {\n', '\tlua_checkstack(__ls, NNN);\n',
'\tXXX* dummy = push_XXX(__ls);\n', '\treturn 1;\n}\n']
GETTER_GEN = ['static int getter_XXX_YYY(lua_State* __ls) {\n',
'\tXXX* dummy = check_XXX(__ls, 1);\n',
'\tlua_pop(__ls, -1);\n',
'\treturn 1;\n}\n']
SETTER_GEN = ['static int setter_XXX_YYY(lua_State* __ls) {\n',
'\tXXX* dummy = check_XXX(__ls, 1);\n',
'\tlua_settop(__ls, 1);\n',
'\treturn 1;\n}\n']
REGISTER_TABLE_METHODS = ['static const luaL_Reg XXX_methods[] = {\n',
'\t{0,0}\n};\n']
REGISTER_META = ['static const luaL_Reg XXX_meta[] = {\n',
'\t{0, 0}\n};\n']
# table register for global lua tables
TABLE_REGISTER_G = ['int XXX_register(lua_State* __ls, char* reg_str) {\n',
'lua_checkstack(__ls, 4);\n'
'lua_newtable(__ls);\n',
'luaL_setfuncs(__ls, XXX_methods, 0);\n',
'lua_setglobal(__ls, "XXX");\n',
'luaL_newmetatable(__ls, reg_str);\n',
'luaL_setfuncs(__ls, XXX_meta, 0);\n',
'lua_pushliteral(__ls, "__index");\n',
'lua_pushvalue(__ls, -3);\n',
'lua_rawset(__ls, -3);\n',
'lua_pushliteral(__ls, "__metatable");\n',
'lua_pushvalue(__ls, -3);\n',
'lua_rawset(__ls, -3);\n',
'lua_setglobal(__ls , "XXX");\n'
'return 0;\n}\n']
# table register function for anonymous lua tables
TABLE_REGISTER = ['int XXX_register(lua_State* __ls, char* reg_str) {\n',
'lua_checkstack(__ls, 4);\n'
'lua_newtable(__ls);\n',
'luaL_setfuncs(__ls, XXX_methods, 0);\n',
'luaL_newmetatable(__ls, reg_str);\n',
'luaL_setfuncs(__ls, XXX_methods, 0);\n',
'luaL_setfuncs(__ls, XXX_meta, 0);\n',
'lua_pushliteral(__ls, "__index");\n',
'lua_pushvalue(__ls, -3);\n',
'lua_rawset(__ls, -3);\n',
'lua_pushliteral(__ls, "__metatable");\n',
'lua_pushvalue(__ls, -3);\n',
'lua_rawset(__ls, -3);\n',
'lua_setglobal(__ls , "XXX");\n'
'return 0;\n}\n']
SOURCE_FILE_NAME='XXX_luatablegen.c'
HEADER_FILE_NAME='XXX_luatablegen.h'
LUA_PUSH_TABLE = """
int pushluatable_YYY(lua_State* ls, XXX array, uint64_t count) {
if (!lua_checkstack(ls, 3)) {
printf("Not enough space on the lua stack.");
return -1;
}
lua_newtable(ls);
uint64_t i = 0U;
for (int j = 0; j < count; ++j) {
lua_pushinteger(ls, i+1);
WWW_push_args(ls, array[i]);
new_WWW(ls);
lua_settable(ls, -3);
i++;
}
return 0;
}
"""
LUA_PUSH_TABLE_SIMPLE_TYPE = """
int pushluatable_YYY(lua_State* ls, XXX array, uint64_t count) {
if (!lua_checkstack(ls, 3)) {
printf("Not enough space on the lua stack.");
return -1;
}
lua_newtable(ls);
uint64_t i = 1U;
for (int j = 0; j < count; ++j) {
lua_pushinteger(ls, i+1);
lua_pushZZZ(ls, array[i]);
lua_settable(ls, -3);
i++;
}
return 0;
}
"""
LUA_PUSH_TABLE_SIMPLE_TYPE_SIG = 'int pushluatable_YYY(lua_State* ls, XXX array, uint64_t count);\n'
LUA_PUSH_TABLE_SIG = "int pushluatable_YYY(lua_State* ls, XXX array, uint64_t count);\n"
LUA_PUSH_TABLE_CALL = "pushluatable_YYY(lua_State* ls, WWW, XXX array, ZZZ);\n"
LUA_LIB = ["local XXX = {}\n\n", "return XXX\n"]
LUA_SETMETA_NEW = ["setmetatable(XXX, {__call =\n", "\tfunction(selfAAA)\n",
"\t\tlocal t = self.new(AAA)\n", "\t\treturn t\n\tend\n\t}\n)\n"]
LUA_TO_GENERIC = "lua_to_YYY(__ls, ZZZ);\n"
LUA_TO_GENERIC_DEF = "YYY lua_to_YYY(lua_State* ls, XXX array, ZZZ) {}\n"
def lua_type_resolver(type_str):
if type_str == "int8":
return "integer"
elif type_str == "uint8":
return "integer"
elif type_str == "int16":
return "integer"
elif type_str == "uint16":
return "integer"
elif type_str == "int32":
return "integer"
elif type_str == "uint32":
return "integer"
elif type_str == "int64":
return "integer"
elif type_str == "uint64":
return "integer"
elif type_str == "int128":
return "integer"
elif type_str == "uint128":
return "integer"
elif type_str == "float":
return "number"
elif type_str == "double":
return "number"
elif type_str == "bool":
return "integer"
elif type_str == "uchar":
return "integer"
elif type_str == "schar":
return "integer"
elif type_str == "string":
return "string"
else: return "lightuserdata"
def simple_type_resovler(type_str):
if type_str == "int8":
return "int8_t"
elif type_str == "uint8":
return "uint8_t"
elif type_str == "int16":
return "int16_t"
elif type_str == "uint16":
return "uint16_t"
elif type_str == "int32":
return "int32_t"
elif type_str == "uint32":
return "uint32_t"
elif type_str == "int64":
return "int64_t"
elif type_str == "uint64":
return "uint64_t"
elif type_str == "int128":
return "int128_t"
elif type_str == "uint128":
return "uint128_t"
elif type_str == "float":
return "float"
elif type_str == "double":
return "double"
elif type_str == "bool":
return "uint8_t"
elif type_str == "uchar":
return "uint8_t"
elif type_str == "schar":
return "int8_t"
elif type_str == "string":
return "char*"
elif type_str == "FT::conditional":
return "void*"
else: return type_str
def get_eq_lua_type(type_str):
if type_str == "int8":
return "integer"
elif type_str == "uint8":
return "integer"
elif type_str == "int16":
return "integer"
elif type_str == "uint16":
return "integer"
elif type_str == "int32":
return "integer"
elif type_str == "uint32":
return "integer"
elif type_str == "int64":
return "integer"
elif type_str == "uint64":
return "integer"
elif type_str == "int128":
return "integer"
elif type_str == "uint128":
return "integer"
elif type_str == "float":
return "number"
elif type_str == "double":
return "number"
elif type_str == "bool":
return "integer"
elif type_str == "uchar":
return "integer"
elif type_str == "schar":
return "integer"
elif type_str == "string":
return "string"
elif type_str == "FT::conditional":
return "lightuserdata"
else: return None
def type_resolver(elem, elem_list):
if "isaggregate" in elem.attrib:
type_str = elem.attrib["name"]
else:
type_str = elem.attrib["type"]
type_name = elem.attrib["name"]
if type_str == "int8":
return "int8_t"
elif type_str == "uint8":
return "uint8_t"
elif type_str == "int16":
return "int16_t"
elif type_str == "uint16":
return "uint16_t"
elif type_str == "int32":
return "int32_t"
elif type_str == "uint32":
return "uint32_t"
elif type_str == "int64":
return "int64_t"
elif type_str == "uint64":
return "uint64_t"
elif type_str == "int128":
return "int128_t"
elif type_str == "uint128":
return "uint128_t"
elif type_str == "float":
return "float"
elif type_str == "double":
return "double"
elif type_str == "bool":
return "uint8_t"
elif type_str == "uchar":
return "uint8_t"
elif type_str == "schar":
return "int8_t"
elif type_str == "string":
return "char*"
elif type_str == "FT::conditional":
return "void*"
elif type_str.find("self::") == 0:
for node in elem_list:
if elem.attrib["type"][6:] == node.tag:
return node.attrib["name"]
else: return type_str
def get_full_path(path, name):
if path[-1] == "/": return path + name
else: return path + "/" + name
def get_def_node(type_str, elem_list):
for node in elem_list:
if type_str == node.attrib["name"]:
return node
def get_def_node_tag(type_str, elem_list):
for node in elem_list:
if type_str == node.tag:
return node
def get_elem_count(elem):
if "count" in elem.attrib:
try:
if str(int(elem.attrib["count"])) == elem.attrib["count"]:
return int(elem.attrib["count"])
else: return -1
except ValueError:
return -1
else:
return 1
def get_count_node(elem, parent):
if "count" in elem.attrib:
count_node_name = elem.attrib["count"][6:]
else: return None
for child in parent:
if child.tag == count_node_name:
return child
return None
# FIXME-incomplete
def get_lua_push_func(gen_type):
if gen_type == "integer": return "lua_pushinteger(__ls, XXX)"
elif gen_type == "string": return "lua_pushstring(__ls, XXX)"
elif gen_type == "number": return "lua_pushnumber(__ls, XXX)"
else: return None
def get_lua_pop_func(gen_type):
if gen_type == "integer": return "lua_tointeger(__ls, XXX)"
elif gen_type == "string": return "lua_tostring(__ls, XXX)"
elif gen_type == "number": return "lua_tonumber(__ls, XXX)"
else: return None
def get_cond_node(elem, parent):
if "condition" in elem.attrib:
cond_node_name = elem.attrib["condition"][6:]
for child in parent:
if child.tag == cond_node_name:
return child
else:
return None
def SigHandler_SIGINT(signum, frame):
print()
sys.exit(0)
def get_filename(filename):
if filename[-1] == "/":
c_source = filename + c_filename
else:
c_source = "/" + c_filename
class Argparser(object):
def __init__(self):
parser = argparse.ArgumentParser()
parser.add_argument("--out", type=str, help="output directory")
parser.add_argument("--name", type=str, help="will be used to generate some custom names")
parser.add_argument("--tbg", type=str, help="the table gen file")
parser.add_argument("--pre", type=str, help="path to source code file to add after header guard/extern c")
parser.add_argument("--post", type=str, help="path to source code file to add before header guard/extern c end")
parser.add_argument("--luaheader", type=str, help="path to lua header files")
parser.add_argument("--dbg", action="store_true", help="debug", default=False)
parser.add_argument("--singlefile", action="store_true", help="should all the generated code be added to a single file", default=False)
parser.add_argument("--makemacro", action="store_true", help="generate a makefile containing all objects in a macro to be included by another makefile", default=False)
parser.add_argument("--anon", action="store_true", help="generate anonymous lua tables if true, global if false", default=True)
parser.add_argument("--useuuid", action="store_true", help="use uuids to register metatables instead of the name of the metatable", default=True)
parser.add_argument("--outfile", type=str, help="name of the output file if signlefile is set, ignored otherwise")
parser.add_argument("--headeraggr", type=str, help="header aggregate file name")
parser.add_argument("--lualibpath", type=str, help="where the lua module file will be placed")
parser.add_argument("--lualibname", type=str, help="the name for the table")
parser.add_argument("--docpath", type=str, help="where the doc file will be placed")
parser.add_argument("--xml", type=str, help="same as --tbg but use an xml file instead")
parser.add_argument("--tbldefs", type=str, help="path to the definitions tablegen creates")
self.args = parser.parse_args()
class TbgParser(object):
def __init__(self, argparser):
#self.tbg_file = json.load(open(argparser.args.tbg))
self.argparser = argparser
self.time = datetime.datetime.now().isoformat()
print(self.time)
self.def_elems = []
self.read_elems = []
def begin(self, c_source, struct_name, h_filename, is_source):
c_source.write("\n")
c_source.write("// automatically generated by luatablegen\n")
c_source.write("// " + self.time + "\n")
for header in HEADER_LIST:
if self.argparser.args.luaheader:
c_source.write(header.replace("HHH", self.argparser.args.luaheader+"/"))
else:
c_source.write(header.replace("HHH", ""))
c_source.write('#include "./tabledefs.h"\n')
if not is_source: c_source.write(HEADER_GUARD[0].replace("XXX", struct_name))
if not is_source: c_source.write(EXTERN_C[0])
if is_source: c_source.write("#include " + '"./' +h_filename+ '"\n')
c_source.write("\n")
if self.argparser.args.pre:
pre_file = open(self.argparser.args.pre)
for line in pre_file:
c_source.write(line)
pre_file.close()
c_source.write("\n")
def gen_struct_header_xml(self):
self.struct_source_h = self.argparser.args.out + "/structs.h"
self.struct_source_c = self.argparser.args.out + "/structs.c"
struct_source = open(self.struct_source_h, "w")
struct_source_c = open(get_full_path(self.argparser.args.out, "structs.c"), "w")
struct_source.write("// automatically generated by luatablegen\n")
struct_source_c.write("// automatically generated by luatablegen\n")
struct_source.write("// " + self.time + "\n")
struct_source_c.write("// " + self.time + "\n")
struct_source.write("#ifndef FT_STRUCTS_H\n#define FT_STRUCTS_H\n")
struct_source.write('#ifdef __cplusplus__\nextern "C" {\n#endif\n')
struct_source_c.write('#include "structs.h"\n')
struct_source_c.write('#include "stdlib.h"\n')
struct_source_c.write('#include "stdio.h"\n')
struct_source.write('#include <unistd.h>\n')
struct_source.write('#include <inttypes.h>\n')
"""
if self.argparser.args.structsinclude:
copy(self.argparser.args.structsinclude, self.argparser.args.outdir)
pos = self.argparser.args.structsinclude.rfind("/")
sub = self.argparser.args.structsinclude[pos+1:]
struct_source.write('#include "' + sub + '"\n\n')
"""
for child in self.def_elems + self.read_elems:
struct_source.write("typedef struct {\n")
if not "isaggregate" in child.attrib:
ref_type = type_resolver(child, self.def_elems + self.read_elems)
def_node = get_def_node(ref_type, self.def_elems + self.read_elems)
pointer = str()
if "count" in child.attrib:
if child.attrib["count"] != "1":
pointer = "*"
if def_node:
struct_source.write(ref_type + pointer + "* " + child.attrib["name"] + ";\n")
else:
struct_source.write(ref_type + pointer + " " + child.attrib["name"] + ";\n")
for childer in child:
ref_type = type_resolver(childer, self.def_elems + self.read_elems)
def_node = get_def_node(ref_type, self.def_elems + self.read_elems)
pointer = str()
if "count" in childer.attrib:
if childer.attrib["count"] != "1":
pointer = "*"
if def_node:
struct_source.write(ref_type + pointer + "* " + childer.attrib["name"] + ";\n")
else:
struct_source.write(ref_type + pointer + " " + childer.attrib["name"] + ";\n")
struct_source.write("}" + child.attrib["name"] + ";\n\n")
struct_source.write('#ifdef __cplusplus__\n}\n#endif\n')
struct_source.write("#endif\n")
#struct_source.write(text.last_comment)
def gen_lua_table_push_def(self, node, struct_name, parent):
type_name = type_resolver(child, self.def_elems+self.read_elems)
type_ref_node = get_def_node(type_name, self.def_elems+self.read_elems)
count = get_elem_count(node, self.def_elems+self.read_elems)
#count_node = get_count_node(node, parent)
count_node_name = str()
if count_node: count_node_name = count_node.attrib["name"]
if count == -1:
count_node_name = node.attrib["count"][6:]
if count == 1:
pointer = ""
else:
pointer += "*"
yyy = node.atttrib["name"]
if type_ref_node:
pointer += "*"
xxx = type_ref_node.attrib["name"]
zzz = "push_" + type_ref_node.attrib["name"]
return LUA_PUSH_TABLE.replace("XXX", xxx+pointer).replace("YYY", yyy).replace("ZZZ", zzz)
else:
xxx = node.attrib["name"]
zzz = "lua_push" + node.attrib["luatype"]
return LUA_PUSH_TABLE_SIMPLE_TYPE.replace("XXX", xxx+pointer).replace("YYY", yyy).replace("ZZZ", zzz)
def gen_lua_table_push_call(self, node, arg_pos, parent):
type_name = type_resolver(node, self.def_elems+self.read_elems)
type_ref_node = get_def_node(type_name, self.def_elems+self.read_elems)
count = get_elem_count(node)
count_node = get_count_node(node, parent)
count_node_name = str()
if count_node != None:
count_node_name = count_node.attrib["name"]
pointer = ""
if count == 1:
pointer = ""
else:
pointer += "*"
yyy = node.attrib["name"]
if type_ref_node:
pointer += "*"
xxx = type_ref_node.attrib["name"]
zzz = "push_" + type_ref_node.attrib["name"]
else:
xxx = node.attrib["name"]
zzz = "lua_push" + node.attrib["luatype"]
dummy = str()
if count == 1:
dummy = "\tpush_" + type_resolver(node, self.elems) +"(__ls, dummy->"+node.attrib["name"]+");\n"
elif count > 1:
dummy = LUA_PUSH_TABLE_CALL.replace("XXX", xxx+pointer).replace("YYY", yyy).replace("WWW", repr(arg_pos)).replace("ZZZ", count)
else:
dummy = LUA_PUSH_TABLE_CALL.replace("XXX", xxx+pointer).replace("YYY", yyy).replace("WWW", repr(arg_pos)).replace("ZZZ", count_node_name)
return [type_resolver(node, self.elems) + pointer + node.attrib["name"], dummy]
def gen_luato_generic(self, struct_name, field_name, arg_pos):
parent = get_def_node(struct_name, self.elems)
child = get_def_node(field_name, self.elems)
return "check_" + struct_name + "(__ls," + repr(arg_pos) + ");\n"
def struct(self, c_source, field_names, field_types, struct_name):
c_source.write("typedef struct {\n")
for field_type, field_name in zip(field_types, field_names):
c_source.write("\t" + field_type + " " + field_name + ";\n")
c_source.write("}" +struct_name+ ";\n")
c_source.write("\n")
def convert(self, c_source, struct_name):
for line in CONVERT:
c_source.write(line.replace("XXX", struct_name))
c_source.write("\n")
def check(self, c_source, struct_name):
node = get_def_node(struct_name, self.elems)
has_conditional = False
for elem in node:
if "conditional" in elem.attrib:
has_conditional = True
if has_conditional == True:
c_source.write('static XXX* check_XXX(lua_State* __ls, int index) {\n'.replace("XXX", struct_name))
c_source.write('\tXXX* dummy;\n'.replace("XXX", struct_name))
c_source.write('dummy = lua_touserdata(__ls, index);\n'.replace("XXX", struct_name))
c_source.write('\tif (dummy == NULL) printf("XXX:bad user data type.\\n");\n'.replace("XXX", struct_name))
c_source.write('\treturn dummy;\n}\n')
return
for line in CHECK:
c_source.write(line.replace("XXX", struct_name))
c_source.write("\n")
def push_self(self, c_source, struct_name):
for line in PUSH_SELF:
c_source.write(line.replace("XXX", struct_name))
c_source.write("\n")
def read_xml(self):
tree = xml.etree.ElementTree.parse(self.argparser.args.xml)
root = tree.getroot()
read_tree = xml.etree.ElementTree.Element("read")
def_tree = xml.etree.ElementTree.Element("def")
for child in root:
if child.tag == "Read":
read_tree = child
if child.tag == "Definition":
def_tree = child
for child in read_tree:
self.read_elems.append(child)
for child in def_tree:
self.def_elems.append(child)
read_iter = read_tree.iter(tag=None)
def_iter = def_tree.iter(tag=None)
self.read_iter = read_iter
self.def_iter = def_iter
self.struct_names = []
self.lua_types = []
lua_type = []
self.field_names = []
field_name = []
self.field_types = []
field_type = []
for node in self.read_elems+self.def_elems:
self.struct_names.append(node.attrib["name"])
for child in node:
field_name.append(child.attrib["name"])
field_type.append(child.attrib["type"])
lua_type.append(child.attrib["luatype"])
self.field_types.append(field_type)
field_type = []
self.field_names.append(field_name)
field_name = []
self.lua_types.append(lua_type)
lua_type = []
self.elems = self.def_elems + self.read_elems
def push_args(self, c_source, struct_name, field_names, lua_types):
dummy = str()
c_source.write(PUSH_ARGS[0].replace("XXX", struct_name))
c_source.write("if (_st == NULL) return 0;\n")
c_source.write("\tlua_checkstack(__ls, " + repr(len(field_names)) + ");\n")
if not field_names:
orig_node = get_def_node(struct_name, self.elems)
lua_type = orig_node.attrib["luatype"]
field_name = orig_node.attrib["name"]
if lua_type == "integer": dummy = "\tlua_pushinteger(__ls, _st->"+field_name+");\n"
elif lua_type == "number": dummy = "\tlua_pushnumber(__ls, _st->"+field_name+");\n"
elif lua_type == "string": dummy = "\tlua_pushstring(__ls, _st->"+field_name+");\n"
elif lua_type == "boolean": dummy = "\tlua_pushboolean(__ls, _st->"+field_name+");\n"
else: print("badf lua type")
c_source.write(dummy)
for field_name, lua_type in zip(field_names, lua_types):
if lua_type == "integer": dummy = "\tlua_pushinteger(__ls, _st->"+field_name+");\n"
elif lua_type == "lightuserdata": dummy = "\tlua_pushlightuserdata(__ls, _st->"+field_name+");\n"
elif lua_type == "number": dummy = "\tlua_pushnumber(__ls, _st->"+field_name+");\n"
elif lua_type == "string": dummy = "\tlua_pushstring(__ls, _st->"+field_name+");\n"
elif lua_type == "boolean": dummy = "\tlua_pushboolean(__ls, _st->"+field_name+");\n"
elif lua_type == "table":
parent = get_def_node(struct_name, self.elems)
child = get_def_node(field_name, self.elems)
count_node_name = str()
if not child:
for kid in parent:
if kid.attrib["name"] == field_name: child = kid
count = get_elem_count(child)
count_node = get_count_node(child, parent)
#print("parent:" + parent.attrib["name"])
#print("child:" + child.attrib["name"])
#if count_node != None: print("count node:" + count_node.attrib["name"])
if count_node != None: count_node_name = count_node.attrib["name"]
if count == 1:
dummy = "\tpush_" + type_resolver(child, self.elems) +"(__ls, _st->"+field_name+");\n"
elif count > 1:
dummy = "\tpushluatable_" + type_resolver(child, self.elems) +"(__ls, _st->"+field_name+", _st->"+count+");\n"
else:
dummy = "\tpushluatable_" + type_resolver(child, self.elems) +"(__ls, _st->"+field_name+", _st->"+count_node_name+");\n"
elif lua_type == "conditional":
parent = get_def_node(struct_name, self.elems)
child = get_def_node(field_name, self.elems)
if not child:
for kid in parent:
if kid.attrib["name"] == field_name: child = kid
cond_node = get_def_node_tag(child.attrib["condition"][6:], [child for child in parent])
for childer in child:
c_source.write("if (_st->" + cond_node.attrib["name"] + "==" + childer.text + ")\n")
if childer.attrib["luatype"] == "integer": c_source.write("lua_pushinteger(__ls, _st->" + child.attrib["name"] + ");\n")
elif childer.attrib["luatype"] == "number":c_source.write("lua_pushnumber(__ls, _st->" + child.attrib["name"] + ");\n")
elif childer.attrib["luatype"] == "string":c_source.write("lua_pushstring(__ls, _st->" + child.attrib["name"] + ");\n")
elif childer.attrib["luatype"] == "lightuserdata":
count = get_elem_count(childer)
if count == 1:
ref_type_node = get_def_node_tag(childer.attrib["type"][6:], self.elems)
if ref_type_node != None:
c_source.write("push_" + ref_type_node.attrib["name"] + "(__ls, _st->" + child.attrib["name"] + ");\n")
else:
c_source.write("lua_pushlightuserdata(__ls, _st->" + child.attrib["name"] + ");\n")
# FIXME
else: pass
else:
print("bad lua_type entry in the json file")
sys.exit(1)
c_source.write(dummy)
dummy = str()
if not field_names:
c_source.write(PUSH_ARGS[2].replace("NNN", "1"))
else:
c_source.write(PUSH_ARGS[2].replace("NNN", repr(len(field_names))))
c_source.write("\n")
def new(self, c_source, struct_name, field_types, field_names, lua_types):
dummy = str()
rev_counter = -len(field_types)
c_source.write(NEW[0].replace("XXX", struct_name))
if not field_names:
c_source.write("\tlua_checkstack(__ls, 1);\n")
else:
c_source.write("\tlua_checkstack(__ls, " + repr(len(field_names)) + ");\n")
if not field_names:
orig_node = get_def_node(struct_name, self.elems)
lua_type = orig_node.attrib["luatype"]
field_name = orig_node.attrib["name"]
field_type = orig_node.attrib["type"]
if lua_type == "integer": dummy = "\t"+simple_type_resovler(field_type) +" "+field_name +"_s"+" = "+"luaL_optinteger(__ls,-1,0);\n"
elif lua_type == "string":dummy = "\t"+simple_type_resovler(field_type) +" "+field_name+" = "+"lua_tostring(__ls,-1,0);\n"
c_source.write(dummy)
for lua_type, field_name, field_type in zip(lua_types, field_names, field_types):
parent = get_def_node(struct_name, self.elems)
child = get_def_node(field_name, self.elems)
for kid in parent:
if kid.attrib["name"] == field_name: child = kid
if lua_type == "integer": dummy = "\t"+simple_type_resovler(field_type) +" "+field_name+" = "+"luaL_optinteger(__ls,"+repr(rev_counter)+",0);\n"
elif lua_type == "lightuserdata":
parent_node = get_def_node(struct_name, self.elems)
self_node = get_def_node(field_name, parent_node)
count = get_elem_count(self_node)
if field_type.find("self::") == 0:
ptr = ""
if count != 1: ptr = "*"
child_node = get_def_node_tag(field_type[6:], self.elems)
dummy = "\t"+child_node.attrib["name"] + ptr +"* "+field_name+" = "+"lua_touserdata(__ls,"+repr(rev_counter)+");\n"
else:
ptr = str()
if count != 1: ptr = "*"
if type_resolver(child, self.elems) != field_type:
dummy = "\t"+type_resolver(child, self.elems) + ptr + " "+field_name+" = "+"lua_touserdata(__ls,"+repr(rev_counter)+");\n"
else:
dummy = "\t"+field_type+" "+field_name+" = "+"lua_touserdata(__ls,"+repr(rev_counter)+");\n"
elif lua_type == "number": pass
elif lua_type == "string":dummy = "\t"+simple_type_resovler(field_type) +" "+field_name+" = "+"lua_tostring(__ls,"+repr(rev_counter)+");\n"
elif lua_type == "boolean": pass
elif lua_type == "table":
temp = self.gen_lua_table_push_call(child, rev_counter, parent)
temp2 = self.gen_luato_generic(struct_name, field_name, rev_counter)
dummy = temp[0] + "=" + temp2
elif lua_type == "conditional":
dummy = "void*" + child.attrib["name"] + ";\n"
for kind in child:
#print(kind.tag)
#print(child.attrib["condition"][6:])
cond_node = get_def_node_tag(child.attrib["condition"][6:], parent)
lua_eq_type = get_eq_lua_type(kind.attrib["type"])
push = str()
#if lua_eq_type == "integer": push = child.attrib["name"]+"=luaL_optinteger(__ls,"+repr(rev_counter)+",0);\n"
#elif lua_eq_type == "number":push = child.attrib["name"]+"=lua_tonumber(__ls,"+repr(rev_counter)+");\n"
#elif lua_eq_type == "string":push = child.attrib["name"]+"=lua_tostring(__ls,"+repr(rev_counter)+");\n"
#elif lua_eq_type == "lightuserdata":push = child.attrib["name"]+"=lua_touserdata(__ls,"+repr(rev_counter)+");\n"
#elif lua_eq_type == None:push = child.attrib["name"]+"=lua_touserdata(__ls,"+repr(rev_counter)+");\n"
#else: print("this was not supposed to happen...")
push = child.attrib["name"]+"=lua_touserdata(__ls,"+repr(rev_counter)+");\n"
dummy += "if (" + cond_node.attrib["name"] + " ==" + kind.text+ ") {"+push+"}\n"
#dummy = "void* " + child.attrib["name"] + "=" + self.gen_luato_generic(struct_name, field_name, rev_counter)
else:
print("bad lua_type entry in the json file")
sys.exit(1)
rev_counter += 1
c_source.write(dummy)
dummy = str()
if not field_names:
c_source.write("lua_pop(__ls,1);\n")
else:
c_source.write("lua_pop(__ls,"+repr(len(field_types))+");\n")
c_source.write(NEW[2].replace("XXX", struct_name))
for field_name in field_names:
c_source.write("\tdummy->" + field_name + " = " + field_name + ";\n")
if not field_names:
orig_node = get_def_node(struct_name, self.elems)
lua_type = orig_node.attrib["luatype"]
field_name = orig_node.attrib["name"]
field_type = orig_node.attrib["type"]
c_source.write("\tdummy->" + field_name + " = " + field_name + "_s" + ";\n")
c_source.write(NEW[3].replace("XXX", struct_name))
c_source.write("\n")
def getter(self, c_source, struct_name, field_names, field_types, lua_types):
dummy = str()
for field_name, lua_type, field_type in zip(field_names, lua_types, field_types):
c_source.write(GETTER_GEN[0].replace("XXX", struct_name).replace("YYY", field_name))
c_source.write(GETTER_GEN[1].replace("XXX", struct_name))
c_source.write(GETTER_GEN[2])
parent = get_def_node(struct_name, self.elems)
#child = get_def_node(field_name, self.elems)
for kid in parent:
if field_name == kid.attrib["name"]: child = kid
count = get_elem_count(child)
count_node = get_count_node(child, parent)
count_node_name = str()
if count_node != None: count_node_name = count_node.attrib["name"]
ref_node_type = get_def_node_tag(child.attrib["type"][6:], self.elems)
if not child:
for kid in parent:
if kid.attrib["name"] == field_name: child = kid
if lua_type == "integer": dummy = "\tlua_pushinteger(__ls, dummy->"+field_name+");\n"
elif lua_type == "lightuserdata":
if count == 1:
dummy = "lua_pushlightuserdata(__ls, dummy->"+child.attrib["name"]+");\n"
dummy += "lua_gettable(__ls, LUA_REGISTRYINDEX);\n"
#dummy += 'luaL_getmetatable(__ls, "'+ref_node_type.attrib["name"]+'");\n'
#dummy += "lua_setmetatable(__ls, -2);\n"
#dummy = ref_node_type.attrib["name"]+ "_push_args(__ls, dummy->"+field_name+");\nnew_" + ref_node_type.attrib["name"] + "(__ls);\n"
#FIXME
else:
count_replacer = str()
if count > 1: count_replacer = repr(count)
else:
count_replacer = count_node_name
dummy = "lua_checkstack(__ls, 3);\nlua_newtable(__ls);\n"
dummy += "for (uint64_t i = 0; i < dummy->" + count_replacer + " ; ++i) {\nlua_pushinteger(__ls, i+1);\n"
if ref_node_type != None:
dummy += "if (dummy->" +field_name+ "[i] != NULL) {\n"
dummy += "lua_pushlightuserdata(__ls, dummy->"+field_name+"[i]);\n"
dummy += "lua_gettable(__ls, LUA_REGISTRYINDEX);\n"
#dummy += 'luaL_getmetatable(__ls,"'+ref_node_type.attrib["name"]+'");\n'
#dummy += "lua_setmetatable(__ls, -2);\n"
dummy += "} else {\nlua_pop(__ls, 1);\n continue;\n}"
else:
eq_lua_type = get_eq_lua_type(field_type)
dummy += "lua_push"+eq_lua_type+"(__ls, dummy->"+field_name+"[i]);\n"
dummy += "lua_settable(__ls, -3);\n}\n"
elif lua_type == "number": dummy = "\tlua_pushnumber(__ls, dummy->"+field_name+");\n"
elif lua_type == "string": dummy = "\tlua_pushstring(__ls, dummy->"+field_name+");\n"
elif lua_type == "boolean": dummy = "\tlua_pushboolean(__ls, dummy->"+field_name+");\n"
elif lua_type == "table":
if count == 1:
dummy = "\tpush_" + type_resolver(child, self.elems) +"(__ls, dummy->"+field_name+");\n"
elif count > 1:
dummy = "\tpushluatable_" + type_resolver(child, self.elems) +"(__ls, dummy->"+field_name+", dummy->"+count+");\n"
else:
dummy = "\tpushluatable_" + type_resolver(child, self.elems) +"(__ls, dummy->"+field_name+", dummy->"+count_node_name+");\n"
elif lua_type == "conditional":
#FIXME-wont work properly for counts greater than 1
for kind in child:
cond_node = get_def_node_tag(child.attrib["condition"][6:], parent)
lua_eq_type = get_eq_lua_type(kind.attrib["type"])
push = str()
if lua_eq_type == "integer": push = "lua_pushinteger(__ls, dummy->"+child.attrib["name"]+");\n"
elif lua_eq_type == "number": push = "lua_pushnumber(__ls, dummy->"+child.attrib["name"]+");\n"
elif lua_eq_type == "string": push = "lua_pushstring(__ls, dummy->"+child.attrib["name"]+");\n"
elif lua_eq_type == "lightuserdata": push = ""
elif lua_eq_type == None:
type_node = get_def_node_tag(kind.attrib["type"][6:], self.elems)
#print(kind.attrib["type"])
#push = type_node.attrib["name"]+"_push_args(__ls, dummy->"+child.attrib["name"]+");\n"
#push += "new_" + type_node.attrib["name"] + "(__ls);\n"
push = "lua_pushlightuserdata(__ls, dummy->"+child.attrib["name"]+");\n"
push += "lua_gettable(__ls, LUA_REGISTRYINDEX);\n"
#push += 'luaL_getmetatable(__ls, "'+type_node.attrib["name"]+'");\n'
#push += "lua_setmetatable(__ls, -2);\n"
else: print("this was not supposed to happen...")
dummy += "if (dummy->" + cond_node.attrib["name"] + " ==" + kind.text+ ") {"+push+"}\n"
else:
print("bad lua_type entry in the json file")
sys.exit(1)
c_source.write(dummy)
dummy = str()
c_source.write(GETTER_GEN[3])
c_source.write("\n")
def setter(self, c_source, struct_name, field_names, field_types, lua_types):
dummy = str()
for field_name, lua_type in zip(field_names, lua_types):
parent = get_def_node(struct_name, self.elems)
node = get_def_node(field_name, parent)
type_node = get_def_node_tag(node.attrib["type"][6:], self.elems)
count = get_elem_count(node)
c_source.write(SETTER_GEN[0].replace("XXX", struct_name).replace("YYY", field_name))
c_source.write(SETTER_GEN[1].replace("XXX", struct_name))
if lua_type == "integer": dummy = "\tdummy->" + field_name + " = " + "luaL_checkinteger(__ls, 2);\n"
elif lua_type == "lightuserdata":
if type_node != None:
type_replacement = type_node.attrib["name"]
else:
type_replacement = simple_type_resovler(node.attrib["type"])
#dummy += "dummy->" +field_name+ "=calloc(sizeof(" +type_replacement+ ")*table_length,1);\n"
if count == 1:
#dummy += "free(dummy->" + field_name + ");\n"
#dummy += "dummy->" +field_name+ "=calloc(sizeof(" +type_replacement+ "),1);\n"
dummy += "dummy->" + field_name + "= luaL_checkudata(__ls, -1,\""+type_replacement+"\");\n"
dummy += "lua_pop(__ls, 1);\n"
else:
dummy = "if (!lua_checkstack(__ls, 3)) {printf(\"error\"\n);return 0;}\n"
dummy += "int table_length = lua_rawlen(__ls, 2);\nfree(dummy->"+field_name+");\n"
#dummy += "dummy->" +field_name+ "=calloc(sizeof(" +type_replacement+ ")*table_length,1);\n"
dummy += "dummy->" +field_name+ "=lua_newuserdata(__ls, sizeof(" +type_replacement+ ")*table_length);\n"
dummy += "for (int i = 1; i <= table_length; ++i) {\n lua_rawgeti(__ls, 2, i);\n"
real_type = node.attrib["type"]
real_type_string = lua_type_resolver(real_type)
if real_type_string == "lightuserdata":
dummy += "dummy->" + field_name + "[i-1] = luaL_checkudata(__ls , -1, \""+type_replacement+"\");\n"
elif real_type_string == "integer":
dummy += "dummy->" + field_name + "[i-1] = luaL_checkinteger(__ls , -1);\n"
elif real_type_string == "string":
dummy += "dummy->" + field_name + "[i-1] = luaL_checkstring(__ls , -1);\n"
dummy += "lua_pop(__ls, 1);\n}\n"
elif lua_type == "number": dummy ="\tdummy->" + field_name + " = " + "luaL_checknumber(__ls, 2);\n"
elif lua_type == "string": dummy ="\tdummy->" + field_name + " = " + "luaL_checkstring(__ls, 2);\n"
elif lua_type == "boolean": pass
elif lua_type == "table": dummy = "\t;\n"
elif lua_type == "conditional":
if type_node != None:
type_replacement = type_node.attrib["name"]
else:
type_replacement = simple_type_resovler(node.attrib["type"])
cond_node = get_cond_node(node, parent)
if count == 1:
#-FIXME i should really do something about memory management.this is getting out of hand...
#dummy += "free(dummy->" + field_name + ");\n"
for con_child in node:
child_def_node = get_def_node(con_child.attrib["type"][6:], self.elems)
dummy += "if (dummy->"+cond_node.attrib["name"]+" == "+con_child.text+"){"
if child_def_node:
#dummy += "dummy->" +field_name+ "=calloc(sizeof(" +child_def_node.attrib["name"]+ "),1);\n"
dummy += "dummy->" + field_name + "= luaL_checkudata(__ls, -1,\""+child_def_node.attrib["name"]+"\");}\n"
else:
con_child_type_node = get_def_node_tag(con_child.attrib["type"][6:], self.elems)
if con_child_type_node: # for user-defined structs
#dummy += "dummy->" +field_name+ "=calloc(sizeof(" +con_child_type_node.attrib["name"]+ "),1);\n"
dummy += "dummy->" + field_name + "= luaL_checkudata(__ls, -1,\""+con_child_type_node.attrib["name"]+"\");}\n"
else: #for simple types
real_type_string = lua_type_resolver(con_child.attrib["type"])
#dummy += "dummy->" +field_name+ "=calloc(sizeof(" +simple_type_resovler(con_child.attrib["type"])+ "),1);\n"
lua_push_func_str = get_lua_pop_func(real_type_string)
dummy += "dummy->" + field_name + "="+ lua_push_func_str.replace("XXX", "-1")+";}\n"
#dummy += "lua_pop(__ls, 1);\n"
# FIXME- not implemented for count greater than one
else:
dummy = "if (!lua_checkstack(__ls, 3)) {printf(\"error\"\n);return 0;}\n"
dummy += "int table_length = lua_rawlen(__ls, 2);\n"
#dummy += "free(dummy->"+field_name+");\n"
#dummy += "dummy->" +field_name+ "=calloc(sizeof(" +type_replacement+ ")*table_length,1);\n"
dummy += "for (int i = 1; i <= table_length; ++i) {\n lua_rawgeti(__ls, 2, i);\n"
real_type = node.attrib["type"]
real_type_string = lua_type_resolver(real_type)
if real_type_string == "lightuserdata":
dummy += "dummy->" + field_name + "[i-1] = luaL_checkudata(__ls , -1, \""+type_replacement+"\");\n"
elif real_type_string == "integer":
dummy += "dummy->" + field_name + "[i-1] = luaL_checkinteger(__ls , -1);\n"
elif real_type_string == "string":
dummy += "dummy->" + field_name + "[i-1] = luaL_checkstring(__ls , -1);\n"
dummy += "lua_pop(__ls, 1);\n}\n"
else:
print("bad lua_type entry in the json file")
sys.exit(1)
c_source.write(dummy)
dummy = str()
c_source.write(SETTER_GEN[2])
c_source.write(SETTER_GEN[3])
c_source.write("\n")
def gc(self):
dummy = str()
for field_name, lua_type in zip(field_names, lua_types):
parent = get_def_node(struct_name, self.elems)
node = get_def_node(field_name, parent)
type_node = get_def_node_tag(node.attrib["type"][6:], self.elems)
count = get_elem_count(node)
if lua_type == "integer": pass
elif lua_type == "lightuserdata": pass
elif lua_type == "number": pass
elif lua_type == "string": pass
elif lua_type == "boolean": pass
elif lua_type == "table": pass
elif lua_type == "conditional": pass
c_source.write(dummy)
dummy = str()
c_source.write("\n")
def tostring(self):
pass
def register_table_methods(self, c_source, struct_name, field_names):
c_source.write(REGISTER_TABLE_METHODS[0].replace("XXX", struct_name))
c_source.write('\t{"new", ' + "new_" + struct_name + "},\n")
for field_name in field_names:
c_source.write("\t{" + '"set_' + field_name + '"' + ", " + "setter_"+struct_name +"_"+ field_name + "},\n")
for field_name in field_names:
c_source.write("\t{" + '"' + field_name + '", ' + "getter_"+struct_name+"_"+field_name+"},\n")
c_source.write(REGISTER_TABLE_METHODS[1])
c_source.write("\n")
def register_table_meta(self, c_source, struct_name):
c_source.write(REGISTER_META[0].replace("XXX", struct_name))
c_source.write(REGISTER_META[1])
c_source.write("\n")
def register_table(self, c_source, struct_name, length):
# if anon tables were selected
if self.argparser.args.anon:
for line in TABLE_REGISTER:
c_source.write(line.replace("XXX", struct_name))
# if global tables were selected
else:
for line in TABLE_REGISTER_G:
c_source.write(line.replace("XXX", struct_name))
def end(self, c_source, is_source):
if self.argparser.args.post:
c_source.write("\n")
post_file = open(self.argparser.args.post)
for line in post_file:
c_source.write(line)
post_file.clsoe()
c_source.write("\n")
if not is_source: c_source.write(EXTERN_C[1])
if not is_source: c_source.write(HEADER_GUARD[1])
c_source.write("\n")
def docgen_md(self, d_source, struct_name, field_names, field_types, lua_types):
d_source.write("## " + "__" + struct_name + "__" + ":\n")
d_source.write("\n")
d_source.write("### " + "_" + "getter fields" + "_" + ":\n")
for field_name,lua_type in zip(field_names, lua_types):
d_source.write(struct_name + ":" + field_name + "()" + " -- ")
if lua_type == "lightuserdata":
d_source.write("return type: " + field_name + "_t" + "<br/>" + "\n")
else:
d_source.write("return type: " + lua_type + "<br/>" + "\n")
d_source.write("\n")
d_source.write("### " + "_" + "setter fields" + "_" + ":\n")
for field_name,lua_type in zip(field_names, lua_types):
d_source.write(struct_name + ":set_" + field_name + "()" + " -- ")
if lua_type == "lightuserdata":
d_source.write("arg type: " + field_name + "_t" + "<br/>" + "\n")
else:
d_source.write("arg type: " + lua_type + "<br/>" + "\n")
d_source.write("\n")
d_source.write("### " + "_" + "constructors" + "_" + ":\n")
d_source.write(struct_name + ":new() -- needs all the args<br/>\n")
d_source.write(struct_name + "() -- lazy constructor<br/>\n")
d_source.write("\n")
d_source.write("\n")
def luagen(self, l_source, struct_name, field_names, field_types, lua_types):
l_source.write(LUA_SETMETA_NEW[0].replace("XXX", struct_name))
arg_list_str = str()
for i in range(0, len(field_names)):
arg_list_str += ", arg" + repr(i)
l_source.write(LUA_SETMETA_NEW[1].replace("AAA", arg_list_str))
l_source.write(LUA_SETMETA_NEW[2].replace("AAA", arg_list_str[2:]))
l_source.write(LUA_SETMETA_NEW[3])
arg_list_str = str()
l_source.write("\n")
def gen_table_def(self):
tbl_source = open(self.argparser.args.tbldefs + "/tabledefs.c", "w")
tbl_header = open(self.argparser.args.tbldefs + "/tabledefs.h", "w")
tbl_source.write("// automatically generated by luatablegen\n")
tbl_header.write("// automatically generated by luatablegen\n")
tbl_source.write("//" + self.time + "\n")
tbl_header.write("//" + self.time + "\n")
for header in HEADER_LIST[0:4]:
if self.argparser.args.luaheader:
tbl_source.write(header.replace("HHH", self.argparser.args.luaheader+"/"))
tbl_header.write(header.replace("HHH", self.argparser.args.luaheader+"/"))
else:
tbl_source.write(header.replace("HHH", ""))
tbl_header.write(header.replace("HHH", ""))
tbl_source.write('#include "./structs.h"\n')
tbl_header.write('#include "./structs.h"\n')
tbl_tag_list = []
simple_table_list = []
for elem in self.elems:
for node in elem:
count_replacement = ""
type_name = type_resolver(node, self.def_elems+self.read_elems)
type_ref_node = get_def_node(type_name, self.def_elems+self.read_elems)
# if node has attribute aggregate
if type_ref_node and type_ref_node.tag not in tbl_tag_list:
tbl_tag_list.append(type_ref_node.tag)
count = get_elem_count(node)
pointer = ""
if count == -1:
for node2 in elem:
if node2.tag == node.attrib["count"][6:]:
count_replacement = node2.attrib["name"]
if count == 1:
pointer = "*"
else:
pointer += "*"
yyy = node.attrib["name"]
if type_ref_node:
pointer += "*"
xxx = type_ref_node.attrib["name"]
zzz = "push_" + type_ref_node.attrib["name"]
else:
xxx = node.attrib["name"]
zzz = "lua_push" + node.attrib["luatype"]
#if pointer == "*": continue
tbl_source.write(LUA_PUSH_TABLE.replace("XXX", xxx+pointer).replace("YYY", xxx).replace("WWW", xxx))
tbl_header.write(LUA_PUSH_TABLE_SIG.replace("XXX", xxx+pointer).replace("YYY", xxx))
# if node is simple type
else:
count = get_elem_count(node)
simple_type = simple_type_resovler(node.attrib["type"])
if count != 1 and simple_type not in simple_table_list:
simple_table_list.append(simple_type)
yyy = node.attrib["name"]
xxx = simple_type_resovler(node.attrib["type"])
# lightuserdata types are being handled elsewhere
if simple_type == "lightuserdata": continue
lua_type = lua_type_resolver(node.attrib["type"])
tbl_source.write(LUA_PUSH_TABLE_SIMPLE_TYPE.replace("YYY", xxx).replace("XXX", simple_type+"*").replace("ZZZ", lua_type))
tbl_header.write(LUA_PUSH_TABLE_SIMPLE_TYPE_SIG.replace("YYY", xxx).replace("XXX", simple_type+"*"))
def run(self):
header_aggr_list = []
table_reg_list = []
self.read_xml()
self.gen_table_def()
self.gen_struct_header_xml()
if self.argparser.args.singlefile:
c_source = open(self.argparser.args.outfile, "w")
if self.argparser.args.docpath:
d_source = open(self.argparser.args.docpath, "w")
d_source.write("The lazy constructors are inside wasm.lua.\n")
d_source.write("```lua\nlocal wasm = require(\"wasm\")\n```\n")
if self.argparser.args.lualibpath:
l_source = open(self.argparser.args.lualibpath, "w")
l_source.write("-- automatically generated by luatablegen\n")
l_source.write("-- " + self.time + "\n")
l_source.write(LUA_LIB[0].replace("XXX", self.argparser.args.lualibname))
#for k, v in self.tbg_file.items():
for struct_name, field_names, field_types, lua_types in zip(self.struct_names, self.field_names, self.field_types, self.lua_types):
if not self.argparser.args.singlefile:
c_filename = struct_name + "_tablegen.c"
h_filename = struct_name + "_tablegen.h"
if self.argparser.args.out[-1] == "/":
c_source = open(self.argparser.args.out + c_filename, "w")
header_aggr_list.append("./" + h_filename)
h_source = open(self.argparser.args.out + h_filename, "w")
else:
c_source = open(self.argparser.args.out + "/" + c_filename, "w")
header_aggr_list.append("./" + h_filename)
h_source = open(self.argparser.args.out + "/" + h_filename, "w")
# source file
self.begin(c_source, struct_name, h_filename, True)
self.convert(c_source, struct_name)
self.check(c_source, struct_name)
self.push_self(c_source, struct_name)
self.push_args(c_source, struct_name, field_names, lua_types)
self.new(c_source, struct_name, field_types, field_names, lua_types)
self.getter(c_source, struct_name, field_names, field_types, lua_types)
self.setter(c_source, struct_name, field_names, field_types, lua_types)
self.register_table_methods(c_source, struct_name, field_names)
self.register_table_meta(c_source, struct_name)
self.register_table(c_source, struct_name, len(self.struct_names))
self.end(c_source, True)
if not self.argparser.args.singlefile: c_source.close()
# header file
self.begin(h_source, struct_name, h_filename, False)
h_source.write(CONVERT[0].replace("XXX", struct_name).replace(" {\n", ";\n"))
h_source.write(CHECK[0].replace("XXX", struct_name).replace(" {\n", ";\n"))
h_source.write(PUSH_SELF[0].replace("XXX", struct_name).replace(" {\n", ";\n"))
h_source.write(PUSH_ARGS[0].replace("XXX", struct_name).replace(" {\n", ";\n"))
h_source.write(NEW[0].replace("XXX", struct_name).replace(" {\n", ";\n"))
for field_name, lua_type in zip(field_names, lua_types):
h_source.write(GETTER_GEN[0].replace("XXX", struct_name).replace("YYY", field_name).replace(" {\n", ";\n"))
for field_name, lua_type in zip(field_names, lua_types):
h_source.write(SETTER_GEN[0].replace("XXX", struct_name).replace("YYY", field_name).replace(" {\n", ";\n"))
table_reg_list.append(struct_name + '_register(__ls,"'+struct_name+'");\n')
h_source.write(TABLE_REGISTER[0].replace("XXX", struct_name).replace(" {\n", ";\n"))
self.end(h_source, False)
# docs
if self.argparser.args.docpath:
self.docgen_md(d_source, struct_name, field_names, field_types, lua_types)
if self.argparser.args.lualibpath:
self.luagen(l_source, struct_name, field_names, field_types, lua_types)
# header aggregate
if self.argparser.args.headeraggr:
name = self.argparser.args.headeraggr
dummy = name[name.rfind("/"):]
aggr_header = open(self.argparser.args.headeraggr.replace(".h", ".c"), "w")
aggr_header_h = open(self.argparser.args.headeraggr, "w")
aggr_header.write("// automatically generated by luatablegen\n")
aggr_header_h.write("// automatically generated by luatablegen\n")
aggr_header.write("// " + self.time + "\n")
aggr_header_h.write("// " + self.time + "\n")
aggr_header_h.write(HEADER_GUARD[0].replace("XXX", "WASM_TABLES_AGGR"))
aggr_header_h.write(EXTERN_C[0])
aggr_header.write("\n")
for item in header_aggr_list:
aggr_header.write("#include " + '"' + item + '"\n')
aggr_header_h.write("#include " + '"' + item + '"\n')
aggr_header.write("#include " + '".' + dummy + '"\n')
aggr_header.write("\n")
aggr_header.write("#pragma weak reg_tablegen_tables_"+self.argparser.args.name + "\n")
aggr_header.write("void reg_tablegen_tables_"+self.argparser.args.name+"(lua_State* __ls) {\n")
aggr_header_h.write("void reg_tablegen_tables_"+self.argparser.args.name+"(lua_State* __ls);\n")
for func_sig in table_reg_list:
aggr_header.write("\t" + func_sig)
if self.argparser.args.anon:
pass
else:
aggr_header.write("\t" + "lua_pop(__ls, 1);\n")
aggr_header.write("}\n")
aggr_header_h.write(EXTERN_C[1])
aggr_header_h.write(HEADER_GUARD[1])
aggr_header.write("\n")
if self.argparser.args.makemacro:
if self.argparser.args.out[-1] == "/":
m_source = open(self.argparser.args.out + "tablegen.mk", "w")
else:
m_source = open(self.argparser.args.out + "/" + "tablegen.mk", "w")
# generate lua module
#self.luagen()
if self.argparser.args.docpath:
d_source.write("_automatically generated by luatablegen._<br/>\n")
d_source.write("_" + self.time + "_")
if self.argparser.args.lualibpath:
#l_source = open(self.argparser.args.lualibpath, "w")
l_source.write(LUA_LIB[1].replace("XXX", self.argparser.args.lualibname))
# write code here
def premain(argparser):
signal.signal(signal.SIGINT, SigHandler_SIGINT)
#here
parser = TbgParser(argparser)
parser.run()
def main():
argparser = Argparser()
if argparser.args.dbg:
try:
premain(argparser)
except Exception as e:
print(e.__doc__)
if e.message: print(e.message)
variables = globals().copy()
variables.update(locals())
shell = code.InteractiveConsole(variables)
shell.interact(banner="DEBUG REPL")
else:
premain(argparser)
if __name__ == "__main__":
main()