source: code/trunk/src/tolua/lua/basic.lua @ 3368

-- tolua: basic utility functions
-- Written by Waldemar Celes
-- TeCGraf/PUC-Rio
-- Jul 1998
-- Last update: Apr 2003
-- $Id: $

-- This code is free software; you can redistribute it and/or modify it.
-- The software provided hereunder is on an "as is" basis, and
-- the author has no obligation to provide maintenance, support, updates,
-- enhancements, or modifications.


-- Basic C types and their corresponding Lua types
-- All occurrences of "char*" will be replaced by "_cstring",
-- and all occurrences of "void*" will be replaced by "_userdata"
_basic = {
    ['void'] = '',
    ['char'] = 'number',
    ['int'] = 'number',
    ['short'] = 'number',
    ['long'] = 'number',
    ['unsigned'] = 'number',
    ['float'] = 'number',
    ['double'] = 'number',
    ['_cstring'] = 'string',
    ['_userdata'] = 'userdata',
    ['char*'] = 'string',
    ['void*'] = 'userdata',
    ['bool'] = 'boolean',
    ['lua_Object'] = 'value',
    ['LUA_VALUE'] = 'value',    -- for compatibility with tolua 4.0
    ['lua_State*'] = 'state',
    ['_lstate'] = 'state',
    ['lua_Function'] = 'value',
}

_basic_ctype = {
    number = "lua_Number",
    string = "const char*",
    userdata = "void*",
    boolean = "bool",
    value = "int",
    state = "lua_State*",
}

-- functions the are used to do a 'raw push' of basic types
_basic_raw_push = {}

-- List of user defined types
-- Each type corresponds to a variable name that stores its tag value.
_usertype = {}

-- List of types that have to be collected
_collect = {}

-- List of types
_global_types = {n=0}
_global_types_hash = {}

-- list of classes
_global_classes = {}

-- List of enum constants
_global_enums = {}

-- List of auto renaming
_renaming = {}
function appendrenaming (s)
    local b,e,old,new = strfind(s,"%s*(.-)%s*@%s*(.-)%s*$")
    if not b then
        error("#Invalid renaming syntax; it should be of the form: pattern@pattern")
    end
    tinsert(_renaming,{old=old, new=new})
end

function applyrenaming (s)
    for i=1,getn(_renaming) do
        local m,n = gsub(s,_renaming[i].old,_renaming[i].new)
        if n ~= 0 then
            return m
        end
    end
    return nil
end

-- Error handler
function tolua_error (s,f)
    if _curr_code then
        print("***curr code for error is "..tostring(_curr_code))
        print(debug.traceback())
    end
    local out = _OUTPUT
    _OUTPUT = _STDERR
    if strsub(s,1,1) == '#' then
        write("\n** tolua: "..strsub(s,2)..".\n\n")
        if _curr_code then
            local _,_,s = strfind(_curr_code,"^%s*(.-\n)") -- extract first line
            if s==nil then s = _curr_code end
            s = gsub(s,"_userdata","void*") -- return with 'void*'
            s = gsub(s,"_cstring","char*")  -- return with 'char*'
            s = gsub(s,"_lstate","lua_State*")  -- return with 'lua_State*'
            write("Code being processed:\n"..s.."\n")
        end
    else
        if not f then f = "(f is nil)" end
        print("\n** tolua internal error: "..f..s..".\n\n")
        return
    end
    _OUTPUT = out
end

function warning (msg)
    local out = _OUTPUT
    _OUTPUT = _STDERR
    write("\n** tolua warning: "..msg..".\n\n")
    _OUTPUT = out
end

-- register an user defined type: returns full type
function regtype (t)
    --if isbasic(t) then
    --    return t
    --end
    local ft = findtype(t)

    if not _usertype[ft] then
        return appendusertype(t)
    end
    return ft
end

-- return type name: returns full type
function typevar(type)
    if type == '' or type == 'void' then
        return type
    else
        local ft = findtype(type)
        if ft then
            return ft
        end
        _usertype[type] = type
        return type
    end
end

-- check if basic type
function isbasic (type)
    local t = gsub(type,'const ','')
    local m,t = applytypedef('', t)
    local b = _basic[t]
    if b then
        return b,_basic_ctype[b]
    end
    return nil
end

-- split string using a token
function split (s,t)
    local l = {n=0}
    local f = function (s)
        l.n = l.n + 1
        l[l.n] = s
        return ""
    end
    local p = "%s*(.-)%s*"..t.."%s*"
    s = gsub(s,"^%s+","")
    s = gsub(s,"%s+$","")
    s = gsub(s,p,f)
    l.n = l.n + 1
    l[l.n] = gsub(s,"(%s%s*)$","")
    return l
end

-- splits a string using a pattern, considering the spacial cases of C code (templates, function parameters, etc)
-- pattern can't contain the '^' (as used to identify the begining of the line)
-- also strips whitespace
function split_c_tokens(s, pat)

    s = string.gsub(s, "^%s*", "")
    s = string.gsub(s, "%s*$", "")

    local token_begin = 1
    local token_end = 1
    local ofs = 1
    local ret = {n=0}

    function add_token(ofs)

        local t = string.sub(s, token_begin, ofs)
        t = string.gsub(t, "^%s*", "")
        t = string.gsub(t, "%s*$", "")
        ret.n = ret.n + 1
        ret[ret.n] = t
    end

    while ofs <= string.len(s) do

        local sub = string.sub(s, ofs, -1)
        local b,e = string.find(sub, "^"..pat)
        if b then
            add_token(ofs-1)
            ofs = ofs+e
            token_begin = ofs
        else
            local char = string.sub(s, ofs, ofs)
            if char == "(" or char == "<" then

                local block
                if char == "(" then block = "^%b()" end
                if char == "<" then block = "^%b<>" end

                b,e = string.find(sub, block)
                if not b then
                    -- unterminated block?
                    ofs = ofs+1
                else
                    ofs = ofs + e
                end

            else
                ofs = ofs+1
            end
        end

    end
    add_token(ofs)
    --if ret.n == 0 then

    --    ret.n=1
    --    ret[1] = ""
    --end

    return ret

end

-- concatenate strings of a table
function concat (t,f,l,jstr)
    jstr = jstr or " "
    local s = ''
    local i=f
    while i<=l do
        s = s..t[i]
        i = i+1
        if i <= l then s = s..jstr end
    end
    return s
end

-- concatenate all parameters, following output rules
function concatparam (line, ...)
    local i=1
    while i<=arg.n do
        if _cont and not strfind(_cont,'[%(,"]') and
            strfind(arg[i],"^[%a_~]") then
            line = line .. ' '
        end
        line = line .. arg[i]
        if arg[i] ~= '' then
            _cont = strsub(arg[i],-1,-1)
        end
        i = i+1
    end
    if strfind(arg[arg.n],"[%/%)%;%{%}]$") then
        _cont=nil line = line .. '\n'
    end
    return line
end

-- output line
function output (...)
    local i=1
    while i<=arg.n do
        if _cont and not strfind(_cont,'[%(,"]') and
            strfind(arg[i],"^[%a_~]") then
            write(' ')
        end
        write(arg[i])
        if arg[i] ~= '' then
            _cont = strsub(arg[i],-1,-1)
        end
        i = i+1
    end
    if strfind(arg[arg.n],"[%/%)%;%{%}]$") then
        _cont=nil write('\n')
    end
end

function get_property_methods(ptype, name)

    if get_property_methods_hook and get_property_methods_hook(ptype,name) then
        return get_property_methods_hook(ptype, name)
    end

    if ptype == "default" then -- get_name, set_name
        return "get_"..name, "set_"..name
    end

    if ptype == "qt" then -- name, setName
        return name, "set"..string.upper(string.sub(name, 1, 1))..string.sub(name, 2, -1)
    end

    if ptype == "overload" then -- name, name
        return name,name
    end

    return nil
end

-------------- the hooks

-- called right after processing the $[ichl]file directives,
-- right before processing anything else
-- takes the package object as the parameter
function preprocess_hook(p)
    -- p.code has all the input code from the pkg
end


-- called for every $ifile directive
-- takes a table with a string called 'code' inside, the filename, and any extra arguments
-- passed to $ifile. no return value
function include_file_hook(t, filename, ...)

end

-- called after processing anything that's not code (like '$renaming', comments, etc)
-- and right before parsing the actual code.
-- takes the Package object with all the code on the 'code' key. no return value
function preparse_hook(package)

end


-- called after writing all the output.
-- takes the Package object
function post_output_hook(package)

end


-- called from 'get_property_methods' to get the methods to retrieve a property
-- according to its type
function get_property_methods_hook(property_type, name)

end

-- called from ClassContainer:doparse with the string being parsed
-- return nil, or a substring
function parser_hook(s)

    return nil
end