compute the offset of base classes

[lqt.git] / generator / generator.lua

#!/usr/bin/lua

--[[

Copyright (c) 2007-2008 Mauro Iazzi

Permission is hereby granted, free of charge, to any person
obtaining a copy of this software and associated documentation
files (the "Software"), to deal in the Software without
restriction, including without limitation the rights to use,
copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the
Software is furnished to do so, subject to the following
conditions:

The above copyright notice and this permission notice shall be
included in all copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
OTHER DEALINGS IN THE SOFTWARE.

--]]

local path = string.match(arg[0], '(.*/)[^%/]+') or ''
local filename = nil
local dirname = nil
local module_name = nil
local typefiles = {}
local filterfiles = {}
local output_includes = {
        '"lqt_common.hpp"',
}

do
        local i = 1
        while select(i, ...) do
                local argi = select(i, ...)
                if argi=='-n' then
                        i = i + 1
                        module_name = select(i, ...)
                elseif argi=='-i' then
                        i = i + 1
                        table.insert(output_includes, (select(i, ...)))
                elseif argi=='-t' then
                        i = i + 1
                        table.insert(typefiles, (select(i, ...)))
                elseif argi=='-f' then
                        i = i + 1
                        table.insert(filterfiles, (select(i, ...)))
                else
                        filename = filename and error'duplicate filename' or argi
                end
                i = i + 1
        end
end

local my_includes = ''
for _, i in ipairs(output_includes) do
        my_includes = my_includes .. '#include '..i..'\n'
end
output_includes = my_includes .. '\n'

local readfile = function(fn)
        local f = assert(io.open(fn))
        local s = f:read'*a'
        f:close()
        return s
end

local fprint = function(f)
        return function(...)
                for i = 1, select('#',...) do
                        f:write((i==1) and '' or '\t', tostring(select(i,...)))
                end
                f:write'\n'
                f:flush()
        end
end

local debug = fprint(io.stderr)
local print_enum = fprint(assert(io.open(module_name..'_src/'..module_name..'_enum.cpp', 'w')))
local print_slot_h = fprint(assert(io.open(module_name..'_src/'..module_name..'_slot.hpp', 'w')))
local print_slot_c = fprint(assert(io.open(module_name..'_src/'..module_name..'_slot.cpp', 'w')))

local xmlstream, idindex = dofile(path..'xml.lua')(readfile(filename))

----------------------------------------------------------------------------------

local copy_functions = function(index)
        local ret = {}
        for e in pairs(index) do
                if e.label:match'^Function' then
                        e.label = 'Function'
                        ret[e] = true
                end
        end
        return ret
end


local fix_arguments = function(all)
        local fullnames = {}
        for e in pairs(all or {}) do
                if e.xarg.fullname then fullnames[e.xarg.fullname] = true end
        end
        for a in pairs(all) do
                if a.label=='Argument'
                        and a.xarg.default=='1'
                        and (not string.match(a.xarg.defaultvalue, '^[-+]?%d+%.?%d*$'))
                        and a.xarg.defaultvalue~='true'
                        and a.xarg.defaultvalue~='false'
                        and (not string.match(a.xarg.defaultvalue, '^0[xX]%d+$')) then
                        local dv, call = string.match(a.xarg.defaultvalue, '(.-)(%b())')
                        dv = dv or a.xarg.defaultvalue
                        call = call or ''
                        if not fullnames[dv] then
                                dv = a.xarg.context..'::'..dv..call
                        end
                        if fullnames[dv] then
                                a.xarg.defaultvalue = dv..call
                        else
                                a.xarg.default = nil
                                a.xarg.defaultvalue = nil
                        end
                end
        end
        return all
end

local fix_functions = function(index)
        for f in pairs(index) do
                local args = {}
                for i, a in ipairs(f) do
                        -- avoid bogus 'void' arguments
                        if a.xarg.type_name=='void' and i==1 and f[2]==nil then break end
                        if a.label=='Argument' then
                                table.insert(args, a)
                        end
                end
                f.arguments = args
                f.return_type = f.xarg.type_name
                if f.xarg.type_name=='void' then
                        f.return_type = nil
                end
        end
        return index
end

local copy_enums = function(index)
        local ret = {}
        for e in pairs(index) do
                if e.label=='Enum'
                        and not string.match(e.xarg.fullname, '%b<>')
                        and e.xarg.access=='public' then
                        ret[e] = true
                end
        end
        return ret
end

local fill_enums = function(index)
        for e in pairs(index) do
                local values = {}
                for _, v in ipairs(e) do
                        if v.label=='Enumerator' then
                                table.insert(values, v)
                        end
                end
                e.values = values
        end
        return index
end

local copy_classes = function(index)
        local ret = {}
        for e in pairs(index) do
                if e.label=='Class'
                        and e.xarg.access~='private'
                        and not e.xarg.fullname:match'%b<>' then
                        ret[e] = true
                end
        end
        return ret
end

local fill_virtuals = function(index)
        local classes = {}
        for c in pairs(index) do
                classes[c.xarg.fullname] = c
        end
        local get_virtuals
        get_virtuals = function(c)
                local ret = {}
                for _, f in ipairs(c) do
                        if f.label=='Function' and f.xarg.virtual=='1' then
                                local n = string.match(f.xarg.name, '~') or f.xarg.name
                                if n~='~' then ret[n] = f end
                        end
                end
                for b in string.gmatch(c.xarg.bases or '', '([^;]+);') do
                        local base = classes[b]
                        if type(base)=='table' then
                                local bv = get_virtuals(base)
                                for n, f in pairs(bv) do
                                        if not ret[n] then ret[n] = f end
                                end
                        end
                end
                for _, f in ipairs(c) do
                        if f.label=='Function'
                                and f.xarg.access~='private'
                                and (ret[string.match(f.xarg.name, '~') or f.xarg.name]) then
                                f.xarg.virtual = '1'
                                local n = string.match(f.xarg.name, '~')or f.xarg.name
                                ret[n] = f
                        end
                end
                return ret
        end
        for c in pairs(index) do
                c.virtuals = get_virtuals(c)
                for _, f in pairs(c.virtuals) do
                        if f.xarg.abstract=='1' then c.abstract=true break end
                end
        end
        return index
end

local distinguish_methods = function(index)
        for c in pairs(index) do
                local construct, destruct, normal = {}, nil, {}
                local n = c.xarg.name
                local copy = nil
                for _, f in ipairs(c) do
                        if n==f.xarg.name then
                                table.insert(construct, f)
                        elseif f.xarg.name:match'~' then
                                destruct = f
                        else
                                if (not string.match(f.xarg.name, '^operator%W'))
                                        and (not f.xarg.member_template_parameters) then
                                        table.insert(normal, f)
                                end
                        end
                end
                c.constructors = construct
                c.destructor = destruct
                c.methods = normal
        end
        return index
end

local fill_public_destr = function(index)
        local classes = {}
        for c in pairs(index) do
                classes[c.xarg.fullname] = c
        end
        local destr_is_public
        destr_is_public = function(c)
                if c.destructor then
                        return c.destructor.xarg.access=='public'
                else
                        for b in string.gmatch(c.xarg.bases or '', '([^;]+);') do
                                local base = classes[b]
                                if base and not destr_is_public(base) then
                                        return false
                                end
                        end
                        return true
                end
        end
        for c in pairs(index) do
                c.public_destr = destr_is_public(c)
        end
        return index
end

local fill_copy_constructor = function(index)
        local classes = {}
        for c in pairs(index) do
                classes[c.xarg.name] = c
        end
        for c in pairs(index) do
                local copy = nil
                for _, f in ipairs(c.constructors) do
                        if #(f.arguments)==1
                                and f.arguments[1].xarg.type_name==c.xarg.fullname..' const&' then
                                copy = f
                                break
                        end
                end
                c.copy_constructor = copy
        end
        local copy_constr_is_public
        copy_constr_is_public = function(c)
                if c.copy_constructor then
                        return (c.copy_constructor.xarg.access=='public')
                        or (c.copy_constructor.xarg.access=='protected')
                else
                        local ret = nil
                        for b in string.gmatch(c.xarg.bases or '', '([^;]+);') do
                                local base = classes[b]
                                if base and not copy_constr_is_public(base) then
                                        return false
                                end
                        end
                        return true
                end
        end
        for c in pairs(index) do
                c.public_constr = copy_constr_is_public(c)
        end
        return index
end

local fill_typesystem_with_enums = function(enums, types)
        local etype = function(en)
                return {
                        push = function(n)
                                return 'lqtL_pushenum(L, '..n..', "'..en..'")', 1
                        end,
                        get = function(n)
                                return 'static_cast<'..en..'>'
                                ..'(lqtL_toenum(L, '..n..', "'..en..'"))', 1
                        end,
                        test = function(n)
                                return 'lqtL_isenum(L, '..n..', "'..en..'")', 1
                        end,
                }
        end
        local ret = {}
        for e in pairs(enums) do
                if types[e.xarg.fullname]==nil then
                        ret[e] = true
                        types[e.xarg.fullname] = etype(e.xarg.fullname)
                else
                        --io.stderr:write(e.xarg.fullname, ': already present\n')
                end
        end
        return ret
end

local fill_typesystem_with_classes = function(classes, types)
        local pointer_t = function(fn)
                return {
                        -- the argument is a pointer to class
                        push = function(n)
                                return 'lqtL_passudata(L, '..n..', "'..fn..'*")', 1
                        end,
                        get = function(n)
                                return 'static_cast<'..fn..'*>'
                                ..'(lqtL_toudata(L, '..n..', "'..fn..'*"))', 1
                        end,
                        test = function(n)
                                return 'lqtL_isudata(L, '..n..', "'..fn..'*")', 1
                        end,
                }
        end
        local pointer_const_t = function(fn)
                return {
                        -- the argument is a pointer to constant class instance
                        push = function(n)
                                return 'lqtL_passudata(L, '..n..', "'..fn..'*")', 1
                        end,
                        get = function(n)
                                return 'static_cast<'..fn..'*>'
                                ..'(lqtL_toudata(L, '..n..', "'..fn..'*"))', 1
                        end,
                        test = function(n)
                                return 'lqtL_isudata(L, '..n..', "'..fn..'*")', 1
                        end,
                }
        end
        local ref_t = function(fn)
                return {
                        -- the argument is a reference to class
                        push = function(n)
                                return 'lqtL_passudata(L, &'..n..', "'..fn..'*")', 1
                        end,
                        get = function(n)
                                return '*static_cast<'..fn..'*>'
                                ..'(lqtL_toudata(L, '..n..', "'..fn..'*"))', 1
                        end,
                        test = function(n)
                                return 'lqtL_isudata(L, '..n..', "'..fn..'*")', 1
                        end,
                }
        end
        local instance_t = function(fn)
                return {
                        -- the argument is the class itself
                        push = function(n)
                                return 'lqtL_copyudata(L, &'..n..', "'..fn..'*")', 1
                        end,
                        get = function(n)
                                return '*static_cast<'..fn..'*>'
                                ..'(lqtL_toudata(L, '..n..', "'..fn..'*"))', 1
                        end,
                        test = function(n)
                                return 'lqtL_isudata(L, '..n..', "'..fn..'*")', 1
                        end,
                }
        end
        local const_ref_t = function(fn)
                return {
                        -- the argument is a pointer to class
                        push = function(n)
                                return 'lqtL_copyudata(L, &'..n..', "'..fn..'*")', 1, string.gsub(fn, ' const&$', '')
                        end,
                        get = function(n)
                                return '*static_cast<'..fn..'*>'
                                ..'(lqtL_toudata(L, '..n..', "'..fn..'*"))', 1
                        end,
                        test = function(n)
                                return 'lqtL_isudata(L, '..n..', "'..fn..'*")', 1
                        end,
                }
        end
        local ret = {}
        for c in pairs(classes) do
                if types[c.xarg.fullname]==nil then
                        ret[c] = true
                        types[c.xarg.fullname..'*'] = pointer_t(c.xarg.fullname)
                        types[c.xarg.fullname..' const*'] =  pointer_const_t(c.xarg.fullname)
                        types[c.xarg.fullname..'&'] = ref_t(c.xarg.fullname)
                        if c.public_constr and c.shell then
                                local shellname = 'lqt_shell_'..string.gsub(c.xarg.fullname, '::', '_LQT_')
                                types[c.xarg.fullname] = instance_t(c.xarg.fullname, shellname)
                                types[c.xarg.fullname..' const&'] = const_ref_t(c.xarg.fullname, shellname)
                        end
                end
        end
        return ret
end

local argument_name = function(tn, an)
        local ret
        if string.match(tn, '%(%*%)') then
                ret = string.gsub(tn, '%(%*%)', '(*'..an..')', 1)
        elseif string.match(tn, '%[.*%]') then
                ret = string.gsub(tn, '(%[.*%])', an..'%1')
        else
                ret = tn .. ' ' .. an
        end
        return ret
end

local fill_wrapper_code = function(f, types)
        if f.wrapper_code then return f end
        local stackn, argn = 1, 1
        local wrap, line = '  int oldtop = lua_gettop(L);\n', ''
        if f.xarg.abstract then return nil end
        if f.xarg.member_of_class and f.xarg.static~='1' then
                if not types[f.xarg.member_of_class..'*'] then return nil end -- print(f.xarg.member_of_class) return nil end
                local sget, sn = types[f.xarg.member_of_class..'*'].get(stackn)
                wrap = wrap .. '  ' .. f.xarg.member_of_class .. '* self = ' .. sget .. ';\n'
                stackn = stackn + sn
                wrap = wrap .. [[
  if (NULL==self) {
    lua_pushstring(L, "this pointer is NULL");
    lua_error(L);
  }
]]
                --print(sget, sn)
                line = 'self->'..f.xarg.fullname..'('
        else
                line = f.xarg.fullname..'('
        end
        for i, a in ipairs(f.arguments) do
                if not types[a.xarg.type_name] then return nil end
                local aget, an, arg_as = types[a.xarg.type_name].get(stackn)
                wrap = wrap .. '  ' .. argument_name(arg_as or a.xarg.type_name, 'arg'..argn) .. ' = '
                if a.xarg.default=='1' and an>0 then
                        wrap = wrap .. 'lua_isnoneornil(L, '..stackn..')'
                        for j = stackn+1,stackn+an-1 do
                                wrap = wrap .. ' && lua_isnoneornil(L, '..j..')'
                        end
                        local dv = a.xarg.defaultvalue
                        wrap = wrap .. ' ? static_cast< ' .. a.xarg.type_name .. ' >(' .. dv .. ') : '
                end
                wrap = wrap .. aget .. ';\n'
                line = line .. (argn==1 and 'arg' or ', arg') .. argn
                stackn = stackn + an
                argn = argn + 1
        end
        line = line .. ')'
        -- FIXME: hack follows for constructors
        if f.calling_line then line = f.calling_line end
        if f.return_type then line = f.return_type .. ' ret = ' .. line end
        wrap = wrap .. '  ' .. line .. ';\n  lua_settop(L, oldtop);\n' -- lua_pop(L, '..stackn..');\n'
        if f.return_type then
                if not types[f.return_type] then return nil end
                local rput, rn = types[f.return_type].push'ret'
                wrap = wrap .. '  luaL_checkstack(L, '..rn..', "cannot grow stack for return value");\n'
                wrap = wrap .. '  '..rput..';\n  return '..rn..';\n'
        else
                wrap = wrap .. '  return 0;\n'
        end
        f.wrapper_code = wrap
        return f
end

local fill_test_code = function(f, types)
        local stackn = 1
        local test = ''
        if f.xarg.member_of_class and f.xarg.static~='1' then
                if not types[f.xarg.member_of_class..'*'] then return nil end -- print(f.xarg.member_of_class) return nil end
                local stest, sn = types[f.xarg.member_of_class..'*'].test(stackn)
                test = test .. ' && ' .. stest
                stackn = stackn + sn
        end
        for i, a in ipairs(f.arguments) do
                if not types[a.xarg.type_name] then return nil end -- print(a.xarg.type_name) return nil end
                local atest, an = types[a.xarg.type_name].test(stackn)
                if a.xarg.default=='1' and an>0 then
                        test = test .. ' && (lqtL_missarg(L, ' .. stackn .. ', ' .. an .. ') || '
                        test = test .. atest .. ')'
                else
                        test = test .. ' && ' .. atest
                end
                stackn = stackn + an
        end
        -- can't make use of default values if I fix number of args
        test = '(lua_gettop(L)<' .. stackn .. ')' .. test
        f.test_code = test
        return f
end

local fill_wrappers = function(functions, types)
        local ret = {}
        for f in pairs(functions) do
                f = fill_wrapper_code(f, types)
                if f then
                        f = assert(fill_test_code(f, types), f.xarg.fullname) -- MUST pass
                        ret[f] = true
                        --local out = 'extern "C" int lqt_bind'..f.xarg.id..' (lua_State *L) {\n'
                        --.. f.wrapper_code .. '}\n'
                        --print(out)
                end
        end
        return ret
end

local make_pushlines = function(args, types)
        local pushlines, stack = '', 0
        for i, a in ipairs(args) do
                if not types[a.xarg.type_name] then return nil end
                local apush, an = types[a.xarg.type_name].push('arg'..i)
                pushlines = pushlines .. '    ' .. apush .. ';\n'
                stack = stack + an
        end
        return pushlines, stack
end

local virtual_overload = function(v, types)
        local ret = ''
        if v.virtual_overload then return v end
        -- make return type
        if v.return_type and not types[v.return_type] then return nil end
        local rget, rn = '', 0
        if v.return_type then rget, rn, ret_as = types[v.return_type].get'oldtop+1' end
        local retget = (v.return_type and argument_name(ret_as or v.return_type, 'ret')
        .. ' = ' .. rget .. ';' or '') .. 'lua_settop(L, oldtop);return'
        .. (v.return_type and ' ret' or '')
        -- make argument push
        local pushlines, stack = make_pushlines(v.arguments, types)
        if not pushlines then return nil end
        -- make lua call
        local luacall = 'lua_pcall(L, '..(stack+1)..', '..rn..', 0)'
        -- make prototype and fallback
        local proto = (v.return_type or 'void')..' ;;'..v.xarg.name..' ('
        local fallback = ''
        for i, a in ipairs(v.arguments) do
                proto = proto .. (i>1 and ', ' or '')
                .. argument_name(a.xarg.type_name, 'arg'..i)
                fallback = fallback .. (i>1 and ', arg' or 'arg') .. i
        end
        proto = proto .. ')' .. (v.xarg.constant=='1' and ' const' or '')
        fallback = (v.return_type and 'return this->' or 'this->')
        .. v.xarg.fullname .. '(' .. fallback .. ');\n}\n'
        ret = proto .. [[ {
  int oldtop = lua_gettop(L);
  lqtL_pushudata(L, this, "]]..v.xarg.member_of_class..[[*");
  lua_getfield(L, -1, "]]..v.xarg.name..[[");
  if (lua_isfunction(L, -1)) {
    lua_insert(L, -2);
]] .. pushlines .. [[
    if (!]]..luacall..[[) {
      ]]..retget..[[;
    }
  }
  lua_settop(L, oldtop);
  ]] .. fallback
        v.virtual_overload = ret
        v.virtual_proto = string.gsub(proto, ';;', '', 1)
        return v
end

local fill_shell_class = function(c, types)
        local shellname = 'lqt_shell_'..string.gsub(c.xarg.fullname, '::', '_LQT_')
        local shell = 'class ' .. shellname .. ' : public ' .. c.xarg.fullname .. ' {\npublic:\n'
        shell = shell .. '  lua_State *L;\n'
        for _, constr in ipairs(c.constructors) do
                if constr.xarg.access~='private' then
                        local cline = '  '..shellname..' (lua_State *l'
                        local argline = ''
                        for i, a in ipairs(constr.arguments) do
                                cline = cline .. ', ' .. argument_name(a.xarg.type_name, 'arg'..i)
                                argline = argline .. (i>1 and ', arg' or 'arg') .. i
                        end
                        cline = cline .. ') : ' .. c.xarg.fullname
                                .. '(' .. argline .. '), L(l) '
                                .. '{ lqtL_register(L, this); }\n'
                        shell = shell .. cline
                end
        end
        if c.copy_constructor==nil and c.public_constr then
                local cline = '  '..shellname..' (lua_State *l, '..c.xarg.fullname..' const& arg1)'
                cline = cline .. ' : ' .. c.xarg.fullname .. '(arg1), L(l) {}\n'
                shell = shell .. cline
        end
        for i, v in pairs(c.virtuals) do
                if v.xarg.access~='private' then
                        if v.virtual_proto then shell = shell .. '  virtual ' .. v.virtual_proto .. ';\n' end
                end
        end
        shell = shell .. '  ~'..shellname..'() { lqtL_unregister(L, this); }\n'
        shell = shell .. '};\n'
        c.shell_class = shell
        return c
end

local fill_virtual_overloads = function(classes, types)
        for c in pairs(classes) do
                for i, v in pairs(c.virtuals) do
                        if v.xarg.access~='private' then
                                local vret = virtual_overload(v, types)
                        end
                end
        end
        return classes
end

local fill_shell_classes = function(classes, types)
        local ret = {}
        for c in pairs(classes) do
                if c.shell then
                        c = fill_shell_class(c, types)
                        if c then ret[c] = true end
                end
                ret[c] = true
        end
        return ret
end

local print_shell_classes = function(classes)
        local fhead = nil
        for c in pairs(classes) do
                if fhead then fhead:close() end
                local n = string.gsub(c.xarg.fullname, '::', '_LQT_')
                fhead = assert(io.open(module_name..'_src/'..module_name..'_head_'..n..'.hpp', 'w'))
                local print_head = function(...)
                        fhead:write(...)
                        fhead:write'\n'
                end
                print_head('#ifndef LQT_HEAD_'..n)
                print_head('#define LQT_HEAD_'..n)
                print_head(output_includes)
                print_head('#include <'..string.match(c.xarg.fullname, '^[^:]+')..'>')
                print_head''
                if c.shell then
                        if c then
                                print_head(c.shell_class)
                        else
                                --io.stderr:write(c.fullname, '\n')
                        end
                end
                print_head('#endif // LQT_HEAD_'..n)
        end
        if fhead then fhead:close() end
        return classes
end

local print_virtual_overloads = function(classes)
        for c in pairs(classes) do
                if c.shell then
                        local vo = ''
                        local shellname = 'lqt_shell_'..string.gsub(c.xarg.fullname, '::', '_LQT_')
                        for _,v in pairs(c.virtuals) do
                                if v.virtual_overload then
                                        vo = vo .. string.gsub(v.virtual_overload, ';;', shellname..'::', 1)
                                end
                        end
                        c.virtual_overloads = vo
                end
        end
        return classes
end

local print_wrappers = function(index)
        for c in pairs(index) do
                local meta = {}
                local wrappers = ''
                for _, f in ipairs(c.methods) do
                        if f.wrapper_code and f.xarg.virtual~='1' then
                                local out = 'static int lqt_bind'..f.xarg.id
                                ..' (lua_State *L) {\n'.. f.wrapper_code .. '}\n'
                                if f.xarg.access=='public' then
                                        --print_meta(out)
                                        wrappers = wrappers .. out .. '\n'
                                        meta[f] = f.xarg.name
                                end
                        end
                end
                if c.shell then
                        for _, f in ipairs(c.constructors) do
                                if f.wrapper_code then
                                        local out = 'static int lqt_bind'..f.xarg.id
                                            ..' (lua_State *L) {\n'.. f.wrapper_code .. '}\n'
                                        if f.xarg.access=='public' then
                                                --print_meta(out)
                                                wrappers = wrappers .. out .. '\n'
                                                meta[f] = 'new'
                                        end
                                end
                        end
                        --local shellname = 'lqt_shell_'..string.gsub(c.xarg.fullname, '::', '_LQT_')
                        local out = 'static int lqt_delete'..c.xarg.id..' (lua_State *L) {\n'
                        out = out ..'  '..c.xarg.fullname..' *p = static_cast<'
                                ..c.xarg.fullname..'*>(lqtL_toudata(L, 1, "'..c.xarg.fullname..'*"));\n'
                        out = out .. '  if (p) delete p;\n  return 0;\n}\n'
                        --print_meta(out)
                        wrappers = wrappers .. out .. '\n'
                end
                c.meta = meta
                c.wrappers = wrappers
        end
        return index
end

local print_metatable = function(c)
        local methods = {}
        local wrappers = c.wrappers
        for m, n in pairs(c.meta) do
                methods[n] = methods[n] or {}
                table.insert(methods[n], m)
        end
        for n, l in pairs(methods) do
                local disp = 'static int lqt_dispatcher_'..n..c.xarg.id..' (lua_State *L) {\n'
                for _, f in ipairs(l) do
                        disp = disp..'  if ('..f.test_code..') return lqt_bind'..f.xarg.id..'(L);\n'
                end
                disp = disp .. '  lua_settop(L, 0);\n'
                disp = disp .. '  lua_pushstring(L, "incorrect or extra arguments");\n'
                disp = disp .. '  return lua_error(L);\n}\n' 
                --print_meta(disp)
                wrappers = wrappers .. disp .. '\n'
        end
        local metatable = 'static luaL_Reg lqt_metatable'..c.xarg.id..'[] = {\n'
        for n, l in pairs(methods) do
                metatable = metatable .. '  { "'..n..'", lqt_dispatcher_'..n..c.xarg.id..' },\n'
        end
        if c.shell then
                metatable = metatable .. '  { "delete", lqt_delete'..c.xarg.id..' },\n'
        end
        metatable = metatable .. '  { 0, 0 },\n};\n'
        --print_meta(metatable)
        wrappers = wrappers .. metatable .. '\n'
        local bases = ''
        for b in string.gmatch(c.xarg.bases or '', '([^;]*);') do
                bases = bases .. '{"' .. b .. '*", (char*)(void*)static_cast<'..b..'*>(static_cast<'..c.xarg.fullname..'*>(NULL))-(char*)NULL}, '
        end
        bases = 'static lqt_Base lqt_base'..c.xarg.id..'[] = { '..bases..'{NULL, NULL} };\n'
        --print_meta(bases)
        wrappers = wrappers .. bases .. '\n'
        c.wrappers = wrappers
        return c
end

local print_metatables = function(classes)
        for c in pairs(classes) do
                print_metatable(c)
        end
        return classes
end

local print_single_class = function(c)
        local n = string.gsub(c.xarg.fullname, '::', '_LQT_')
        local fmeta = assert(io.open(module_name..'_src/'..module_name..'_meta_'..n..'.cpp', 'w'))
        local print_meta = function(...)
                fmeta:write(...)
                fmeta:write'\n'
        end
        print_meta('#include "'..module_name..'_head_'..n..'.hpp'..'"\n\n')
        print_meta(c.wrappers)
        if c.virtual_overloads then
                print_meta(c.virtual_overloads)
        end
        print_meta('extern "C" int luaopen_'..n..' (lua_State *L) {')
        print_meta('\tlqtL_createclass(L, "'
                ..n..'*", lqt_metatable'
                ..c.xarg.id..', lqt_base'
                ..c.xarg.id..');')
        print_meta'\treturn 0;'
        print_meta'}'
        print_meta''
        fmeta:close()
end

local print_class_list = function(classes)
        local big_picture = {}
        for c in pairs(classes) do
                local n = string.gsub(c.xarg.fullname, '::', '_LQT_')
                print_single_class(c)
                table.insert(big_picture, 'luaopen_'..n)
        end
        if fmeta then fmeta:close() end
        fmeta = assert(io.open(module_name..'_src/'..module_name..'_meta.cpp', 'w'))
        local print_meta = function(...)
                fmeta:write(...)
                fmeta:write'\n'
        end
        print_meta('#include "lqt_common.hpp"')
        for _, p in ipairs(big_picture) do
                print_meta('extern "C" int '..p..' (lua_State *);')
        end
        print_meta('extern "C" int lqt_slot (lua_State *);')
        print_meta('int lqt_create_enums_'..module_name..' (lua_State *);')
        print_meta('extern "C" int luaopen_'..module_name..' (lua_State *L) {')
        for _, p in ipairs(big_picture) do
                print_meta('\t'..p..'(L);')
        end
        print_meta('\tlqt_create_enums_'..module_name..'(L);')
        print_meta('\tlua_pushcfunction(L, lqt_slot);')
        print_meta('\tlua_setglobal(L, "newslot");')
        print_meta('\treturn 0;\n}')
        if fmeta then fmeta:close() end
        return classes
end

local fix_methods_wrappers = function(classes)
        for c in pairs(classes) do
                c.shell = (not c.abstract) and c.public_destr
                for _, constr in ipairs(c.constructors) do
                        local shellname = 'lqt_shell_'..string.gsub(c.xarg.fullname, '::', '_LQT_')
                        constr.calling_line = '*new '..shellname..'(L'
                        for i=1,#(constr.arguments) do
                                constr.calling_line = constr.calling_line .. ', arg' .. i
                        end
                        constr.calling_line = constr.calling_line .. ')'
                        constr.xarg.static = '1'
                        constr.return_type = constr.xarg.type_base..'&'
                end
                if c.destructor then
                        c.destructor.return_type = nil
                end
        end
        return classes
end

local print_enum_tables = function(enums)
        for e in pairs(enums) do
                local table = 'static lqt_Enum lqt_enum'..e.xarg.id..'[] = {\n'
                --io.stderr:write(e.xarg.fullname, '\t', #e.values, '\n')
                for _,v in pairs(e.values) do
                        table = table .. '  { "' .. v.xarg.name
                                .. '", static_cast<int>('..v.xarg.fullname..') },\n'
                end
                table = table .. '  { 0, 0 }\n'
                table = table .. '};\n'
                e.enum_table = table
                print_enum(table)
        end
        return enums
end
local print_enum_creator = function(enums, n)
        local out = 'static lqt_Enumlist lqt_enum_list[] = {\n'
        for e in pairs(enums) do
                out = out..'  { lqt_enum'..e.xarg.id..', "'..e.xarg.fullname..'" },\n'
        end
        out = out..'  { 0, 0 },\n};\n'
        out = out .. 'void lqt_create_enums_'..n..' (lua_State *L) {\n'
        out = out .. '  lqtL_createenumlist(L, lqt_enum_list);  return;\n}\n'
        print_enum(out)
        return enums
end

local copy_signals = function(functions)
        local ret = {}
        for f in pairs(functions) do
                if f.xarg.signal=='1' then
                        ret[f] = 1
                end
        end
        return ret
end

local slots_for_signals = function(signals, types)
        local ret = {}
        for sig in pairs(signals) do
                local args, comma = '(', ''
                for i, a in ipairs(sig.arguments) do
                        args = args .. comma .. a.xarg.type_name .. ' arg'..i
                        comma = ', '
                end
                args = args .. ')'
                local pushlines, stack = make_pushlines(sig.arguments, types)
                if not ret['void slot '..args] and pushlines then
                        ret['void slot '..args] = 'void LqtSlotAcceptor::slot '..args..[[ {
  int oldtop = lua_gettop(L);
  lqtL_pushudata(L, this, "QObject*");
  lua_getfield(L, -1, "slot]]..string.gsub(args, ' arg.', '')..[[");
  if (lua_isnil(L, -1)) {
    lua_pop(L, 1);
    lua_getfield(L, -1, "slot");
  }
  if (!lua_isfunction(L, -1)) {
    lua_settop(L, oldtop);
    return;
  }
  lua_insert(L, -2);
]] .. pushlines .. [[
  if (lua_pcall(L, ]]..stack..[[+1, 0, 0)) {
          lua_error(L);
  }
  lua_settop(L, oldtop);
}
]]
                end
        end
        return ret
end

local print_slots = function(s)
        print_slot_h'class LqtSlotAcceptor : public QObject {'
        print_slot_h'  Q_OBJECT'
        print_slot_h'  lua_State *L;'
        print_slot_h'  public:'
        print_slot_h'  LqtSlotAcceptor(lua_State *l, QObject *p=NULL) : QObject(p), L(l) { lqtL_register(L, this); }'
        print_slot_h'  virtual ~LqtSlotAcceptor() { lqtL_unregister(L, this); }'
        print_slot_h'  public slots:'
        for p, b in pairs(s) do
                print_slot_h('  '..p..';')
        end
        print_slot_h'};\n'
        for p, b in pairs(s) do
                print_slot_c(b)
        end
        print_slot_c[[

extern "C" int lqt_slot (lua_State *L) {
        QObject *parent = static_cast<QObject*>(lqtL_toudata(L, 1, "QObject*"));
        lqtL_passudata(L, new LqtSlotAcceptor(L, parent), "QObject*");
        return 1;
}
]]
end


--------------------------------------------------------------------------------------

local typesystem = {}
do
        local ts = {}
        for i, ft in ipairs(typefiles) do
                ts = loadfile(ft)(ts)
        end
        setmetatable(typesystem, {
                __newindex = function(t, k, v)
                        --debug('added type', k)
                        ts[k] = v
                end,
                __index = function(t, k)
                        local ret = ts[k]
                        --if not ret then debug("unknown type:", tostring(k), ret) end
                        return ret
                end,
        })
end

fix_arguments(idindex) -- fixes default arguments if they are context-relative
local functions = copy_functions(idindex) -- picks functions and fixes label
local functions = fix_functions(functions) -- fixes name and fullname and fills arguments

local enums = copy_enums(idindex) -- picks enums if public
local enums = fill_enums(enums) -- fills field "values"

local classes = copy_classes(idindex) -- picks classes if not private and not blacklisted
local classes = fill_virtuals(classes) -- does that, destructor ("~") excluded
local classes = distinguish_methods(classes) -- does that
local classes = fill_public_destr(classes) -- does that: checks if destructor is public
local classes = fill_copy_constructor(classes) -- does that: checks if copy contructor is public or protected
local classes = fix_methods_wrappers(classes)

for _, f in ipairs(filterfiles) do
        classes, enums = loadfile(f)(classes, enums)
end

local enums = fill_typesystem_with_enums(enums, typesystem) -- does that
local classes = fill_typesystem_with_classes(classes, typesystem)

local functions = fill_wrappers(functions, typesystem)
local classes = fill_virtual_overloads(classes, typesystem) -- does that
local classes = fill_shell_classes(classes, typesystem) -- does that

local signals = copy_signals(functions)
local slots = slots_for_signals(signals, typesystem)


------------- BEGIN OUTPUT



print_enum(output_includes)
print_slot_h(output_includes)
print_slot_c('#include "'..module_name..'_slot.hpp'..'"\n\n')


local classes = print_shell_classes(classes) -- does that
local classes = print_virtual_overloads(classes, typesystem) -- does that
local enums = print_enum_tables(enums) -- does that
local enums = print_enum_creator(enums, module_name) -- does that + print enum list
local classes = print_wrappers(classes) -- just compiles metatable list
local classes = print_metatables(classes) -- just collects the wrappers + generates dispatchers
local classes = print_class_list(classes) -- does that + prints everything related to class

local slots = print_slots(slots)

--print_openmodule(module_name) -- does that