source: code/branches/toonshader/src/tolua/lua/container.lua @ 8474

-- tolua: container abstract class
-- Written by Waldemar Celes
-- TeCGraf/PUC-Rio
-- Jul 1998
-- $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.

-- table to store namespaced typedefs/enums in global scope
global_typedefs = {}
global_enums = {}

-- Container class
-- Represents a container of features to be bound
-- to lua.
classContainer =
{
    curr = nil,
}
classContainer.__index = classContainer
setmetatable(classContainer,classFeature)

-- output tags
function classContainer:decltype ()
    push(self)
    local i=1
    while self[i] do
        self[i]:decltype()
        i = i+1
    end
    pop()
end


-- write support code
function classContainer:supcode ()

    if not self:check_public_access() then
        return
    end

    push(self)
    local i=1
    while self[i] do
        if self[i]:check_public_access() then
            self[i]:supcode()
        end
        i = i+1
    end
    pop()
end

function classContainer:hasvar ()
    local i=1
    while self[i] do
        if self[i]:isvariable() then
            return 1
        end
        i = i+1
    end
    return 0
end

-- Internal container constructor
function _Container (self)
    setmetatable(self,classContainer)
    self.n = 0
    self.typedefs = {tolua_n=0}
    self.usertypes = {}
    self.enums = {tolua_n=0}
    self.lnames = {}
    return self
end

-- push container
function push (t)
    t.prox = classContainer.curr
    classContainer.curr = t
end

-- pop container
function pop ()
    --print("name",classContainer.curr.name)
    --foreach(classContainer.curr.usertypes,print)
    --print("______________")
    classContainer.curr = classContainer.curr.prox
end

-- get current namespace
function getcurrnamespace ()
    return getnamespace(classContainer.curr)
end

-- append to current container
function append (t)
    return classContainer.curr:append(t)
end

-- append typedef to current container
function appendtypedef (t)
    return classContainer.curr:appendtypedef(t)
end

-- append usertype to current container
function appendusertype (t)
    return classContainer.curr:appendusertype(t)
end

-- append enum to current container
function appendenum (t)
    return classContainer.curr:appendenum(t)
end

-- substitute typedef
function applytypedef (mod,type)
    return classContainer.curr:applytypedef(mod,type)
end

-- check if is type
function findtype (type)
    local t = classContainer.curr:findtype(type)
    return t
end

-- check if is typedef
function istypedef (type)
    return classContainer.curr:istypedef(type)
end

-- get fulltype (with namespace)
function fulltype (t)
    local curr =  classContainer.curr
    while curr do
        if curr then
            if curr.typedefs and curr.typedefs[t] then
                return curr.typedefs[t]
            elseif curr.usertypes and curr.usertypes[t] then
                return curr.usertypes[t]
            end
        end
        curr = curr.prox
    end
    return t
end

-- checks if it requires collection
function classContainer:requirecollection (t)
    push(self)
    local i=1
    local r = false
    while self[i] do
        r = self[i]:requirecollection(t) or r
        i = i+1
    end
    pop()
    return r
end


-- get namesapce
function getnamespace (curr)
    local namespace = ''
    while curr do
        if curr and
            ( curr.classtype == 'class' or curr.classtype == 'namespace')
        then
            namespace = (curr.original_name or curr.name) .. '::' .. namespace
            --namespace = curr.name .. '::' .. namespace
        end
        curr = curr.prox
    end
    return namespace
end

-- get namespace (only namespace)
function getonlynamespace ()
    local curr = classContainer.curr
    local namespace = ''
    while curr do
        if curr.classtype == 'class' then
            return namespace
        elseif curr.classtype == 'namespace' then
            namespace = curr.name .. '::' .. namespace
        end
        curr = curr.prox
    end
    return namespace
end

-- check if is enum
function isenum (type)
    return classContainer.curr:isenum(type)
end

-- append feature to container
function classContainer:append (t)
    self.n = self.n + 1
    self[self.n] = t
    t.parent = self
end

-- append typedef
function classContainer:appendtypedef (t)
    local namespace = getnamespace(classContainer.curr)
    self.typedefs.tolua_n = self.typedefs.tolua_n + 1
    self.typedefs[self.typedefs.tolua_n] = t
    self.typedefs[t.utype] = namespace .. t.utype
    global_typedefs[namespace..t.utype] = t
    t.ftype = findtype(t.type) or t.type
    --print("appending typedef "..t.utype.." as "..namespace..t.utype.." with ftype "..t.ftype)
    append_global_type(namespace..t.utype)
    if t.ftype and isenum(t.ftype) then
        global_enums[namespace..t.utype] = true
    end
end

-- append usertype: return full type
function classContainer:appendusertype (t)
    local container
    if t == (self.original_name or self.name) then
        container = self.prox
    else
        container = self
    end
    local ft = getnamespace(container) .. t
    container.usertypes[t] = ft
    _usertype[ft] = ft
    return ft
end

-- append enum
function classContainer:appendenum (t)
    local namespace = getnamespace(classContainer.curr)
    self.enums.tolua_n = self.enums.tolua_n + 1
    self.enums[self.enums.tolua_n] = t
    global_enums[namespace..t.name] = t
end

-- determine lua function name overload
function classContainer:overload (lname)
    if not self.lnames[lname] then
        self.lnames[lname] = 0
    else
        self.lnames[lname] = self.lnames[lname] + 1
    end
    return format("%02d",self.lnames[lname])
end

-- applies typedef: returns the 'the facto' modifier and type
function classContainer:applytypedef (mod,type)
    if global_typedefs[type] then
        --print("found typedef "..global_typedefs[type].type)
        local mod1, type1 = global_typedefs[type].mod, global_typedefs[type].ftype
        local mod2, type2 = applytypedef(mod.." "..mod1, type1)
        --return mod2 .. ' ' .. mod1, type2
        return mod2, type2
    end
    do return mod,type end
end

-- check if it is a typedef
function classContainer:istypedef (type)
    local env = self
    while env do
        if env.typedefs then
            local i=1
            while env.typedefs[i] do
                if env.typedefs[i].utype == type then
                    return type
                end
                i = i+1
            end
        end
        env = env.parent
    end
    return nil
end

function find_enum_var(var)

    if tonumber(var) then return var end

    local c = classContainer.curr
    while c do
        local ns = getnamespace(c)
        for k,v in pairs(_global_enums) do
            if match_type(var, v, ns) then
                return v
            end
        end
        if c.base and c.base ~= '' then
            c = _global_classes[c:findtype(c.base)]
        else
            c = nil
        end
    end

    return var
end

-- check if is a registered type: return full type or nil
function classContainer:findtype (t)

    t = string.gsub(t, "=.*", "")
    if _basic[t] then
     return t
    end

    local _,_,em = string.find(t, "([&%*])%s*$")
    t = string.gsub(t, "%s*([&%*])%s*$", "")
    p = self
    while p and type(p)=='table' do
        local st = getnamespace(p)

        for i=_global_types.n,1,-1 do -- in reverse order

            if match_type(t, _global_types[i], st) then
                return _global_types[i]..(em or "")
            end
        end
        if p.base and p.base ~= '' and p.base ~= t then
            --print("type is "..t..", p is "..p.base.." self.type is "..self.type.." self.name is "..self.name)
            p = _global_classes[p:findtype(p.base)]
        else
            p = nil
        end
    end

    return nil
end

function append_global_type(t, class)
    _global_types.n = _global_types.n +1
    _global_types[_global_types.n] = t
    _global_types_hash[t] = 1
    if class then append_class_type(t, class) end
end

function append_class_type(t,class)
    if _global_classes[t] then
        class.flags = _global_classes[t].flags
        class.lnames = _global_classes[t].lnames
        if _global_classes[t].base and (_global_classes[t].base ~= '') then
            class.base = _global_classes[t].base or class.base
        end
    end
    _global_classes[t] = class
    class.flags = class.flags or {}
end

function match_type(childtype, regtype, st)
--print("findtype "..childtype..", "..regtype..", "..st)
    local b,e = string.find(regtype, childtype, -string.len(childtype), true)
    if b then

        if e == string.len(regtype) and
                (b == 1 or (string.sub(regtype, b-1, b-1) == ':' and
                string.sub(regtype, 1, b-1) == string.sub(st, 1, b-1))) then
            return true
        end
    end

    return false
end

function findtype_on_childs(self, t)

    local tchild
    if self.classtype == 'class' or self.classtype == 'namespace' then
        for k,v in ipairs(self) do
            if v.classtype == 'class' or v.classtype == 'namespace' then
                if v.typedefs and v.typedefs[t] then
                 return v.typedefs[t]
                elseif v.usertypes and v.usertypes[t] then
                 return v.usertypes[t]
                end
                tchild = findtype_on_childs(v, t)
                if tchild then return tchild end
            end
        end
    end
    return nil

end

function classContainer:isenum (type)
    if global_enums[type] then
        return type
    else
        return false
    end

    local basetype = gsub(type,"^.*::","")
    local env = self
    while env do
        if env.enums then
            local i=1
            while env.enums[i] do
                if env.enums[i].name == basetype then
                    return true
                end
                i = i+1
            end
        end
        env = env.parent
    end
    return false
end

methodisvirtual = false -- a global

-- parse chunk
function classContainer:doparse (s)
    --print ("parse "..s)

    -- try the parser hook
    do
        local sub = parser_hook(s)
        if sub then
            return sub
        end
    end

    -- try the null statement
    do
        local b,e,code = string.find(s, "^%s*;")
        if b then
            return strsub(s,e+1)
        end
    end

    -- try empty verbatim line
    do
        local b,e,code = string.find(s, "^%s*$\n")
        if b then
            return strsub(s,e+1)
        end
    end

    -- try Lua code
    do
        local b,e,code = strfind(s,"^%s*(%b\1\2)")
        if b then
            Code(strsub(code,2,-2))
            return strsub(s,e+1)
        end
    end

    -- try C code
    do
        local b,e,code = strfind(s,"^%s*(%b\3\4)")
        if b then
            code = '{'..strsub(code,2,-2)..'\n}\n'
            Verbatim(code,'r')        -- verbatim code for 'r'egister fragment
            return strsub(s,e+1)
        end
    end

    -- try C code for preamble section
    do
        local b,e,code = string.find(s, "^%s*(%b\5\6)")
        if b then
            code = string.sub(code, 2, -2).."\n"
            Verbatim(code, '')
            return string.sub(s, e+1)
        end
    end

    -- try default_property directive
    do
        local b,e,ptype = strfind(s, "^%s*TOLUA_PROPERTY_TYPE%s*%(+%s*([^%)%s]*)%s*%)+%s*;?")
        if b then
            if not ptype or ptype == "" then
                ptype = "default"
            end
            self:set_property_type(ptype)
            return strsub(s, e+1)
        end
    end

    -- try protected_destructor directive
    do
        local b,e = string.find(s, "^%s*TOLUA_PROTECTED_DESTRUCTOR%s*;?")
        if b then
            if self.set_protected_destructor then
                self:set_protected_destructor(true)
            end
            return strsub(s, e+1)
        end
    end

    -- try 'extern' keyword
    do
        local b,e = string.find(s, "^%s*extern%s+")
        if b then
            -- do nothing
            return strsub(s, e+1)
        end
    end

    -- try 'virtual' keyworkd
    do
        local b,e = string.find(s, "^%s*virtual%s+")
        if b then
            methodisvirtual = true
            return strsub(s, e+1)
        end
    end

    -- try labels (public, private, etc)
    do
        local b,e = string.find(s, "^%s*%w*%s*:[^:]")
        if b then
            return strsub(s, e) -- preserve the [^:]
        end
    end

    -- try module
    do
        local b,e,name,body = strfind(s,"^%s*module%s%s*([_%w][_%w]*)%s*(%b{})%s*")
        if b then
            _curr_code = strsub(s,b,e)
            Module(name,body)
            return strsub(s,e+1)
        end
    end

    -- try namesapce
    do
        local b,e,name,body = strfind(s,"^%s*namespace%s%s*([_%w][_%w]*)%s*(%b{})%s*;?")
        if b then
            _curr_code = strsub(s,b,e)
            Namespace(name,body)
            return strsub(s,e+1)
        end
    end

    -- try define
    do
        local b,e,name = strfind(s,"^%s*#define%s%s*([^%s]*)[^\n]*\n%s*")
        if b then
            _curr_code = strsub(s,b,e)
            Define(name)
            return strsub(s,e+1)
        end
    end

    -- try enumerates

    do
        local b,e,name,body,varname = strfind(s,"^%s*enum%s+(%S*)%s*(%b{})%s*([^%s;]*)%s*;?%s*")
        if b then
            --error("#Sorry, declaration of enums and variables on the same statement is not supported.\nDeclare your variable separately (example: '"..name.." "..varname..";')")
            _curr_code = strsub(s,b,e)
            Enumerate(name,body,varname)
            return strsub(s,e+1)
        end
    end

    -- do
    --    local b,e,name,body = strfind(s,"^%s*enum%s+(%S*)%s*(%b{})%s*;?%s*")
    --    if b then
    --        _curr_code = strsub(s,b,e)
    --        Enumerate(name,body)
    --        return strsub(s,e+1)
    --    end
    -- end

    do
        local b,e,body,name = strfind(s,"^%s*typedef%s+enum[^{]*(%b{})%s*([%w_][^%s]*)%s*;%s*")
        if b then
        _curr_code = strsub(s,b,e)
        Enumerate(name,body)
        return strsub(s,e+1)
        end
    end

    -- try operator
    do
        local b,e,decl,kind,arg,const = strfind(s,"^%s*([_%w][_%w%s%*&:<>,]-%s+operator)%s*([^%s][^%s]*)%s*(%b())%s*(c?o?n?s?t?)%s*;%s*")
        if not b then
            -- try inline
            b,e,decl,kind,arg,const = strfind(s,"^%s*([_%w][_%w%s%*&:<>,]-%s+operator)%s*([^%s][^%s]*)%s*(%b())%s*(c?o?n?s?t?)[%s\n]*%b{}%s*;?%s*")
        end
        if not b then
            -- try cast operator
            b,e,decl,kind,arg,const = strfind(s, "^%s*(operator)%s+([%w_:%d<>%*%&%s]+)%s*(%b())%s*(c?o?n?s?t?)");
            if b then
                local _,ie = string.find(s, "^%s*%b{}", e+1)
                if ie then
                    e = ie
                end
            end
        end
        if b then
            _curr_code = strsub(s,b,e)
            Operator(decl,kind,arg,const)
            return strsub(s,e+1)
        end
    end

    -- try function
    do
        --local b,e,decl,arg,const = strfind(s,"^%s*([~_%w][_@%w%s%*&:<>]*[_%w])%s*(%b())%s*(c?o?n?s?t?)%s*=?%s*0?%s*;%s*")
        local b,e,decl,arg,const,virt = strfind(s,"^%s*([^%(\n]+)%s*(%b())%s*(c?o?n?s?t?)%s*(=?%s*0?)%s*;%s*")
        if not b then
            -- try function with template
            b,e,decl,arg,const = strfind(s,"^%s*([~_%w][_@%w%s%*&:<>]*[_%w]%b<>)%s*(%b())%s*(c?o?n?s?t?)%s*=?%s*0?%s*;%s*")
        end
        if not b then
            -- try a single letter function name
            b,e,decl,arg,const = strfind(s,"^%s*([_%w])%s*(%b())%s*(c?o?n?s?t?)%s*;%s*")
        end
        if b then
            if virt and string.find(virt, "[=0]") then
                if self.flags then
                    self.flags.pure_virtual = true
                end
            end
            _curr_code = strsub(s,b,e)
            Function(decl,arg,const)
            return strsub(s,e+1)
        end
    end

    -- try inline function
    do
        local b,e,decl,arg,const = strfind(s,"^%s*([^%(\n]+)%s*(%b())%s*(c?o?n?s?t?)[^;{]*%b{}%s*;?%s*")
        --local b,e,decl,arg,const = strfind(s,"^%s*([~_%w][_@%w%s%*&:<>]*[_%w>])%s*(%b())%s*(c?o?n?s?t?)[^;]*%b{}%s*;?%s*")
        if not b then
            -- try a single letter function name
            b,e,decl,arg,const = strfind(s,"^%s*([_%w])%s*(%b())%s*(c?o?n?s?t?).-%b{}%s*;?%s*")
        end
        if b then
            _curr_code = strsub(s,b,e)
            Function(decl,arg,const)
            return strsub(s,e+1)
        end
    end

    -- try class
    do
        local b,e,name,base,body
        base = '' body = ''
        b,e,name = strfind(s,"^%s*class%s*([_%w][_%w@]*)%s*;")  -- dummy class
        if not b then
            b,e,name = strfind(s,"^%s*struct%s*([_%w][_%w@]*)%s*;")    -- dummy struct
            if not b then
                b,e,name,base,body = strfind(s,"^%s*class%s*([_%w][_%w@]*)%s*(.-)%s*(%b{})%s*;%s*")
                if not b then
                    b,e,name,base,body = strfind(s,"^%s*struct%s*([_%w][_%w@]*)%s*(.-)%s*(%b{})%s*;%s*")
                    if not b then
                        b,e,name,base,body = strfind(s,"^%s*union%s*([_%w][_%w@]*)%s*(.-)%s*(%b{})%s*;%s*")
                        if not b then
                            base = ''
                            b,e,body,name = strfind(s,"^%s*typedef%s%s*struct%s%s*[_%w]*%s*(%b{})%s*([_%w][_%w@]*)%s*;%s*")
                        end
                    end
                end
            end
        end
        if b then
            if base ~= '' then
                base = string.gsub(base, "^%s*:%s*", "")
                base = string.gsub(base, "%s*public%s*", "")
                base = split(base, ",")
                --local b,e
                --b,e,base = strfind(base,".-([_%w][_%w<>,:]*)$")
            else
                base = {}
            end
            _curr_code = strsub(s,b,e)
            Class(name,base,body)
            return strsub(s,e+1)
        end
    end

    -- try typedef
    do
        local b,e,types = strfind(s,"^%s*typedef%s%s*(.-)%s*;%s*")
        if b then
            _curr_code = strsub(s,b,e)
            Typedef(types)
            return strsub(s,e+1)
        end
    end

    -- try variable
    do
        local b,e,decl = strfind(s,"^%s*([_%w][_@%s%w%d%*&:<>,]*[_%w%d])%s*;%s*")
        if b then
            _curr_code = strsub(s,b,e)

            local list = split_c_tokens(decl, ",")
            Variable(list[1])
            if list.n > 1 then
                local _,_,type = strfind(list[1], "(.-)%s+([^%s]*)$");

                local i =2;
                while list[i] do
                    Variable(type.." "..list[i])
                    i=i+1
                end
            end
            --Variable(decl)
            return strsub(s,e+1)
        end
    end

    -- try string
    do
        local b,e,decl = strfind(s,"^%s*([_%w]?[_%s%w%d]-char%s+[_@%w%d]*%s*%[%s*%S+%s*%])%s*;%s*")
        if b then
            _curr_code = strsub(s,b,e)
            Variable(decl)
            return strsub(s,e+1)
        end
    end

    -- try array
    do
        local b,e,decl = strfind(s,"^%s*([_%w][][_@%s%w%d%*&:]*[]_%w%d])%s*;%s*")
        if b then
            _curr_code = strsub(s,b,e)
            Array(decl)
            return strsub(s,e+1)
        end
    end

    -- no matching
    if gsub(s,"%s%s*","") ~= "" then
        _curr_code = s
        error("#parse error")
    else
        return ""
    end

end

function classContainer:parse (s)

    self.curr_member_access = nil

    while s ~= '' do
        s = self:doparse(s)
        methodisvirtual = false
    end
end


-- property types

function get_property_type()

    return classContainer.curr:get_property_type()
end

function classContainer:set_property_type(ptype)
    ptype = string.gsub(ptype, "^%s*", "")
    ptype = string.gsub(ptype, "%s*$", "")

    self.property_type = ptype
end

function classContainer:get_property_type()
    return self.property_type or (self.parent and self.parent:get_property_type()) or "default"
end