Mercurial > pub > dyncall > dyncall
diff test/call_suite_aggrs/mk-cases.lua @ 485:0c68b3f91367
- renamed suite_aggrs to call_suite_aggrs for consistency (callback version will be called callback_suite_aggrs)
author | Tassilo Philipp |
---|---|
date | Thu, 17 Mar 2022 15:41:26 +0100 |
parents | test/suite_aggrs/mk-cases.lua@0f3b6898078d |
children | da5232da6270 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/test/call_suite_aggrs/mk-cases.lua Thu Mar 17 15:41:26 2022 +0100 @@ -0,0 +1,250 @@ +require"config" +require"math" +local max_numargs = 0 + +local aggrs = { } +local seen_aggrs = { } + + +function canon_type(t) + -- aggregate types start with special (closing) char + c = ({ ['}'] = 'struct ', ['>'] = 'union ' })[t:sub(1,1)] + if c ~= nil then + return c..'A'..t:sub(2) + end + return t +end + +function trim(l) return l:gsub("^%s+",""):gsub("%s+$","") end +function mkcase(id,sig) + local sig = trim(sig) + local h = { "/* ",id,":",sig," */ " } + local t = { "" } + local pos = 0 + local n_nest = 0 + local aggr = { } + local aggr_sig = { } + aggr[0] = { } -- non-sequential [0] collects all non-aggr types + aggr_sig[0] = '' + for i = 1, #sig do + local name = "a"..pos + local ch = sig:sub(i,i) + + + -- aggregate nest level change? + if ch == '{' or ch == '<' then + n_nest = n_nest + 1 + aggr[n_nest] = { } + aggr_sig[n_nest] = '' + end + + aggr_sig[n_nest] = aggr_sig[n_nest]..ch + + -- array? Just append to name of member var from prev loop + if ch:match('[%[%]0123456789]') ~= nil then + aggr[n_nest][#aggr[n_nest]] = aggr[n_nest][#aggr[n_nest]]..ch + else + + if ch == '}' or ch == '>' then + -- register yet unseen aggregates, key is sig, val is body and name + if seen_aggrs[aggr_sig[n_nest]] == nil then + aggrs[#aggrs+1] = aggr_sig[n_nest] + ch = ch..#aggrs + seen_aggrs[aggr_sig[n_nest]] = { aggr[n_nest], ch } + end + ch = seen_aggrs[aggr_sig[n_nest]][2] + + n_nest = n_nest - 1 + aggr_sig[n_nest] = aggr_sig[n_nest]..aggr_sig[n_nest+1] + end + + if ch ~= '{' and ch ~= '}' and ch ~= '<' and ch ~= '>' then + aggr[n_nest][#aggr[n_nest]+1] = canon_type(ch) + aggr[n_nest][#aggr[n_nest]+1] = 'm'..(#aggr[n_nest] >> 1) + end + + -- no nesting (= actual func args), generate case code + if n_nest == 0 then + h[#h+1] = canon_type(ch) + -- aggregate types have more than one + if #h[#h] > 1 then + if aggrcpsimple then + t[#t+1] = '*('..h[#h]..'*)V_a['..pos.."]="..name..";" + else + t[#t+1] = 'f_cp'..h[#h]:sub(8)..'(V_a['..pos.."],&"..name..");" + end + if aggrmutabletest then + t[#t] = t[#t]..'memset(&'..name..',0,sizeof('..name..'));' + end + else + t[#t+1] = "V_"..ch.."["..pos.."]="..name..";" + end + + -- is return type or func arg? + if pos == 0 then + h[#h+1] = " f"..id.."(" + h[#h+1] = '' + t[#t] = '' -- clear; aggr return type handled explicitly + else + h[#h+1] = ' '..name + h[#h+1] = "," + end + + pos = pos + 1 + end + end + end + max_numargs = math.max(max_numargs, pos-1) + h[#h] = "){" + if #h[6] == 1 then + t[#t+1] = "ret_"..h[6].."("..(pos-1)..")}\n" + else + t[#t+1] = "ret_a("..(pos-1)..","..h[6]..")}\n" + end + return table.concat(h,"")..table.concat(t,"") +end + +function mkfuntab(n) + local s = { "funptr G_funtab[] = {\n"} + for i = 0, n-1 do + s[#s+1] = "\t(funptr)&f"..i..",\n" + end + s[#s+1] = "};\n" + return table.concat(s,"") +end + +function mksigtab(sigs) + local s = { "char const * G_sigtab[] = {\n"} + for k,v in pairs(sigs) do + s[#s+1] = '\t"'..v..'",\n' + end + s[#s+1] = "};\n" + return table.concat(s,"") +end + +function split_array_decl(s) + local name = s + local n = nil -- not an array + local array_i = s:find('%[') + if array_i ~= nil then + name = name:sub(1, array_i-1) + n = tonumber(s:sub(array_i):match('[0123456789]+')) + end + return { name, n } +end + +function mkall() + local lineno = 0 + local sigtab = { } + local cases = '' + for line in io.lines() do + local sig = trim(line) + cases = cases..mkcase(lineno,sig) + sigtab[#sigtab+1] = sig + lineno = lineno + 1 + end + + agg_sizes = {} + agg_sigs = {} + agg_names = {} + for a = 1, #aggrs do + local k = aggrs[a] + local v = seen_aggrs[k] + local at = canon_type(v[2]) -- aggregate type + local am = v[1] -- aggregate members + + agg_sizes[#agg_sizes + 1] = 'sizeof('..at..')' + agg_sigs [#agg_sigs + 1] = k + agg_names[#agg_names + 1] = at:sub(8) + + -- aggregate def + io.write('/* '..k..' */\n') + if aggrpacking ~= 0 then + local pack = aggrpacking + if pack < 0 then + pack = math.floor(math.pow(2,math.floor(math.log(math.random(math.abs(pack)),2)))) + end + io.write('#pragma pack(push,'..pack..')\n') + end + + io.write(at..' { ') + for i = 1, #am, 2 do + io.write(am[i]..' '..am[i+1]..'; ') + end + io.write("};\n") + + if aggrpacking ~= 0 then + io.write('#pragma pack(pop)\n') + end + + -- aggregate cp and cmp funcs + s = { + 'void f_cp'..at:sub(8)..'('..at..' *x, const '..at..' *y) { ', + 'int f_cmp'..at:sub(8)..'(const '..at..' *x, const '..at..' *y) { return ' + } + o = { '=', '==', 'f_cp', 'f_cmp', '; ', ' && ', '', '1' } + for t = 1, 2 do + if t ~= 1 or aggrcpsimple == false then + io.write(s[t]) + local b = {} + for i = 1, #am, 2 do + local m = split_array_decl(am[i+1]) + local fmt = '' + if m[2] ~= nil then -- need array suffixes? + fmt = '[%d]' + else + m[2] = 1 + end + + for j = 1, m[2] do + name = m[1]..string.format(fmt, j-1) + if string.match(am[i], ' ') then -- aggregate canonical types contain at least one space + b[#b+1] = o[t+2]..am[i]:sub(8)..'(&x->'..name..', &y->'..name..')' + else + b[#b+1] = 'x->'..name..' '..o[t]..' y->'..name + end + end + end + if #b == 0 then -- to handle empty aggregates + b[1] = o[t+6] + end + io.write(table.concat(b,o[t+4]).."; };\n") + end + end + + -- convenient dcnewstruct helper funcs + io.write('DCaggr* f_touchdcst'..at:sub(8)..'() {\n\tstatic DCaggr* at = NULL;\n\tif(!at) {\n\t\tat = dcNewAggr('..(#am>>1)..', sizeof('..at..'), DC_TRUE);\n\t\t') + for i = 1, #am, 2 do + local m = split_array_decl(am[i+1]) + if m[2] == nil then -- need array suffixes? + m[2] = 1 + end + if string.match(am[i], ' ') then -- aggregate canonical types contain at least one space + --io.write('dcAggrField(at, DC_SIGCHAR_AGGREGATE, offsetof('..at..', '..m[1]..'), '..m[2]..', f_touchdcst'..am[i]:sub(8)..'());\n\t\t') + io.write("AFa("..at..','..m[1]..','..m[2]..','..am[i]:sub(8)..')\n\t\t') + else + --io.write("dcAggrField(at, '"..am[i].."', offsetof("..at..', '..m[1]..'), '..m[2]..');\n\t\t') + io.write("AF('"..am[i].."',"..at..','..m[1]..','..m[2]..')\n\t\t') + end + end + io.write("dcCloseAggr(at);\n\t}\n\treturn at;\n};\n") + end + + -- make table.concat work + if #agg_names > 0 then + table.insert(agg_names, 1, '') + end + + io.write(cases) + io.write(mkfuntab(lineno)) + io.write(mksigtab(sigtab)) + io.write('const char* G_agg_sigs[] = {\n\t"'..table.concat(agg_sigs, '",\n\t"')..'"\n};\n') + io.write('int G_agg_sizes[] = {\n\t'..table.concat(agg_sizes, ',\n\t')..'\n};\n') + io.write('funptr G_agg_touchdcstfuncs[] = {'..string.sub(table.concat(agg_names, ',\n\t(funptr)&f_touchdcst'),2)..'\n};\n') + io.write('funptr G_agg_cmpfuncs[] = {'..string.sub(table.concat(agg_names, ',\n\t(funptr)&f_cmp'),2)..'\n};\n') + io.write("int G_maxargs = "..max_numargs..";\n") +end + +math.randomseed(seed) +mkall() +