260
edits
No edit summary |
No edit summary |
||
| Line 1: | Line 1: | ||
-- Module:Tree chart | -- Module:Tree chart | ||
-- | -- Defensive renderer for table-based tree pieces | ||
require('strict') | -- Avoids hard require('strict') so it won't fail on installs without that library. | ||
-- try to load 'strict' if available, but don't error if it's missing | |||
pcall(function() require('strict') end) | |||
local p = {} | local p = {} | ||
| Line 8: | Line 11: | ||
local cells = mw.loadData('Module:Tree chart/data') | local cells = mw.loadData('Module:Tree chart/data') | ||
-- internal renderer: | -- internal renderer: accepts an array where each entry is either: | ||
-- * a string ( | -- * a string key (looks up cells[key]) or | ||
-- * a table describing a custom cell (text, colspan, rowspan, etc) | -- * a table describing a custom cell (text, colspan, rowspan, etc) | ||
function p._main(cell_args) | function p._main(cell_args) | ||
local ret = mw.html.create() | local ret = mw.html.create() | ||
local tbl = ret:tag('table') | local tbl = ret:tag('table') | ||
| Line 18: | Line 20: | ||
:css{ borderCollapse = 'collapse', ['vertical-align'] = 'top' } | :css{ borderCollapse = 'collapse', ['vertical-align'] = 'top' } | ||
local top = tbl:tag('tr'):css{ height = '1px', ['text-align'] = 'center' } | local top = tbl:tag('tr'):css{ height = '1px', ['text-align'] = 'center' } | ||
local bottom = tbl:tag('tr'):css{ height = '1px', ['text-align'] = 'center' } | local bottom = tbl:tag('tr'):css{ height = '1px', ['text-align'] = 'center' } | ||
| Line 26: | Line 27: | ||
local celldef = cells[v] | local celldef = cells[v] | ||
if celldef then | if celldef then | ||
if celldef.t then | if celldef.t then | ||
top:wikitext(celldef.t) | top:wikitext(celldef.t) | ||
else | else | ||
top:tag('td'):wikitext('') | top:tag('td'):wikitext('') | ||
end | end | ||
if celldef.b then | if celldef.b then | ||
bottom:wikitext(celldef.b) | bottom:wikitext(celldef.b) | ||
| Line 40: | Line 38: | ||
end | end | ||
else | else | ||
-- unknown symbol | -- unknown symbol => placeholder cells so columns align | ||
top:tag('td'):wikitext('') | top:tag('td'):wikitext('') | ||
bottom:tag('td'):wikitext('') | bottom:tag('td'):wikitext('') | ||
end | end | ||
else | else | ||
-- custom cell | -- custom cell (table) from unnamed params | ||
local colspan = v.colspan or cell_args.colspan or 6 | local colspan = v.colspan or cell_args.colspan or 6 | ||
local rowspan = v.rowspan or cell_args.rowspan or 2 | local rowspan = v.rowspan or cell_args.rowspan or 2 | ||
local border = (v.border or cell_args.border or '2') .. 'px solid' | local border = (v.border or cell_args.border or '2') .. 'px solid' | ||
top:tag('td') | top:tag('td') | ||
:attr{ colspan = tostring(colspan), rowspan = tostring(rowspan) } | :attr{ colspan = tostring(colspan), rowspan = tostring(rowspan) } | ||
| Line 57: | Line 54: | ||
:wikitext(v.text or '') | :wikitext(v.text or '') | ||
if tonumber(rowspan) == nil or tonumber(rowspan) < 2 then | if tonumber(rowspan) == nil or tonumber(rowspan) < 2 then | ||
bottom:tag('td'):wikitext('') | bottom:tag('td'):wikitext('') | ||
| Line 68: | Line 63: | ||
end | end | ||
function p.main(frame) | function p.main(frame) | ||
local args = require('Module:Arguments').getArgs(frame, { | local args = require('Module:Arguments').getArgs(frame, { | ||
| Line 76: | Line 70: | ||
}) | }) | ||
local cell_args = { | local cell_args = { | ||
colspan = args.colspan, | colspan = args.colspan, | ||
| Line 84: | Line 77: | ||
} | } | ||
for _, val in ipairs(args) do | for _, val in ipairs(args) do | ||
local trimmedVal = val:match('^%s*(.-)%s*$') | local trimmedVal = val:match('^%s*(.-)%s*$') | ||
if trimmedVal == '' then | if trimmedVal == '' then | ||
trimmedVal = '$' | trimmedVal = '$' | ||
end | end | ||
| Line 95: | Line 86: | ||
table.insert(cell_args, trimmedVal) | table.insert(cell_args, trimmedVal) | ||
else | else | ||
local rightTrimmedVal = val:gsub('%s+$','') | local rightTrimmedVal = val:gsub('%s+$','') | ||
local custom = { | local custom = { | ||
text = args[trimmedVal] or ('{{{'..trimmedVal..'}}}'), | text = args[trimmedVal] or ('{{{'..trimmedVal..'}}}'), | ||
| Line 111: | Line 98: | ||
end | end | ||
return p._main(cell_args) | return p._main(cell_args) | ||
end | end | ||
return p | return p | ||