source: code/branches/weapon/src/tolua/lua/container.lua @ 2255

-- 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