Files
aux-addon-ClassicAPI/libs/green_t.lua
T
Manuel Simon Hirsig 843cfecbbf refactoring
2016-09-05 09:34:41 +02:00

139 lines
4.1 KiB
Lua

--green_t = module
setglobal('green_t', green_t and error(nil) or module)
local next, setn, type, setmetatable = next, table.setn, type, setmetatable
local wipe, release, acquire, acquire_temp, empty
-- TODO mandatory operation table mandate + comply/fulfill functions
local pool, pool_size, overflow_pool, auto_release = {}, 0, setmetatable({}, {__mode='k'}), {}
CreateFrame('Frame'):SetScript('OnUpdate', function()
for t in auto_release do release(t) end; wipe(auto_release)
end)
function wipe(t)
setmetatable(t, nil); setn(t, 0)
for k in t do t[k] = nil end
t.reset, t.reset = nil, 1
return t
end
public.wipe = wipe
function release(t)
wipe(t)
auto_release[t] = nil
if pool_size < 50 then
pool_size = pool_size + 1
pool[pool_size] = t
else
overflow_pool[t] = true
end
return
end
public.release = release
function public.ret(t)
if getn(t) > 0 then return tremove(t, 1), ret(t) else release(t) end
end
function acquire()
if pool_size > 0 then
pool_size = pool_size - 1
return pool[pool_size + 1]
end
local t = next(overflow_pool)
if t then overflow_pool[t] = nil; return t end
return {}
end
public.t.get = acquire
function acquire_temp() local t = acquire(); auto_release[t] = true; return t end
public.tt.get = acquire_temp
do local mt = {__newindex=nop}
function empty() return setmetatable(acquire_temp(), mt) end
public.empty.get = empty
end
do local f = function(_, v) if type(v) == 'table' then auto_release[v] = true end return v end
local mt = {__call=f, __sub=f}
public.temp {get=function() return setmetatable(acquire_temp(), mt) end, set=function(t) auto_release[t] = true end}
end
do local f = function(_, v) if type(v) == 'table' then auto_release[v] = nil end return v end
local mt = {__call=f, __sub=f}
public.perm {get=function() return setmetatable(acquire_temp(), mt) end, set=function(t) auto_release[t] = nil end}
end
do local mt, key = {}, nil
function mt:__unm() local temp = mt.__index; mt.__index = nil; return temp end
function mt:__index(k) key = k; return self end
function mt:__call(v) self[key] = v; key = nil return self end
function public.__(t) mt.__newindex = wipe(t); return setmetatable(acquire_temp(), mt) end
end
local function arg_chunk(k, n)
k = k or 1
n = n or 100
local str = k > n and '' or 'a' .. k
for i = k + 1, n do str = str .. ',a' .. i end
return str
end
function public.pseudo_vararg_function(body, upvals)
local upval_chunk = ''
for k in upvals or empty() do
upval_chunk = upval_chunk .. format('local %1$s = %1$s;', k)
end
local f = loadstring(format('%s return function(%s) %s end', upval_chunk, arg_chunk(), body)) or error()
setfenv(f, upvals or empty())
return (f or error())()
end
local function insert_chunk(mode)
local body = 'repeat '
if mode == 'k' then
for i = 2, 99 do
body = body .. format('if a%1$d == nil then break end; a1[a%1$d] = true;', i)
end
elseif mode == 'v' then
body = body .. 'if a2 == nil then break end; a1[1] = a2;'
for i = 3, 99 do
body = body .. format('if a%1$d == nil then setn(a1, %d); break end; a1[%d] = a%1$d;', i, i-2, i-1)
end
elseif mode == 'v0' then
body = body .. 'setn(a1, 98);'
for i = 2, 99 do
body = body .. format('a1[%d] = a%d;', i-1, i)
end
elseif mode == 'kv' then
for i = 2, 98, 2 do
body = body .. format('if a%1$d == nil then break end; a1[a%1$d] = a%d;', i, i+1)
end
end
return body .. 'if a100 ~= nil then error"Overflow." end until false;'
end
do
local function pseudo_literal(mode)
local upvals = {setmetatable=setmetatable, setn=table.setn, error=error}
local mt = {__call = pseudo_vararg_function(insert_chunk(mode) .. 'setmetatable(a1, nil); return a1', upvals)}
return function() return setmetatable(acquire(), mt) end
end
public.S.get = pseudo_literal'k'
public.A.get = pseudo_literal'v'
public.A0.get = pseudo_literal'v0'
public.T.get = pseudo_literal'kv'
end
--do
-- local body = ''
-- for i = 2, 100 do
-- body = body .. format('if a1 == %d then return %s end;', i, arg_chunk(i, 100))
-- end
-- public.select = pseudo_vararg_function(body)
--end
function public.select(i, ...) temp=arg
while i > 1 do i = i - 1; tremove(arg, i) end
return tremove(arg, 1), unpack(arg)
end