Jump to content

Module:Tree chart/sandbox: Difference between revisions

From Wikipedia, the free encyclopedia
Content deleted Content added
Jackmcbarn (talk | contribs)
rm v.name
Jackmcbarn (talk | contribs)
simiplify p.main
Line 45: Line 45:
local args = require('Module:Arguments').getArgs(frame, {wrappers = 'Template:Chart', trim = false, removeBlanks = false})
local args = require('Module:Arguments').getArgs(frame, {wrappers = 'Template:Chart', trim = false, removeBlanks = false})
local cell_args = {}
local cell_args = {}
for _, v in ipairs(args) do
for _, val in ipairs(args) do
local w = v:match('^%s*(.-)%s*$')
local trimmedVal = val:match('^%s*(.-)%s*$')
if w == '' then
if trimmedVal == '' then
w = '$'
trimmedVal = '$'
end
end
if cells[trimmedVal] then
local cell_x
table.insert(cell_args, trimmedVal)
if cells[w] then
cell_x = w
else
else
-- Unnamed params behave weirdly
-- Unnamed params behave weirdly
-- white space at the front counts for param_{{{1}}}, but not whitespace at the end, so remove it
-- white space at the front counts for param_{{{1}}}, but not whitespace at the end, so remove it
local w2 = v:gsub('%s+$','')
local rightTrimmedVal = val:gsub('%s+$','')
table.insert(cell_args, {
cell_x = {}
cell_x.text = args[w] or '{{{'..w..'}}}'
text = args[trimmedVal] or ('{{{'..trimmedVal..'}}}'),
cell_x.colspan = args['colspan_'..w2] or args.colspan
colspan = args['colspan_'..rightTrimmedVal] or args.colspan,
cell_x.rowspan = args['rowspan_'..w2] or args.rowspan
rowspan = args['rowspan_'..rightTrimmedVal] or args.rowspan,
cell_x.thick = args['border_'..w2] or args.border
thick = args['border_'..rightTrimmedVal] or args.border,
cell_x.css = args['boxstyle_'..w2] or args.boxstyle
css = args['boxstyle_'..rightTrimmedVal] or args.boxstyle
})
end
end
table.insert(cell_args,cell_x)
end
end

Revision as of 00:27, 17 December 2014

require('Module:No globals')

local p = {}

local cells = mw.loadData('Module:TreeChart/data')

local function make_cell(builder,name,key)
	local props = cells[name][key]
	if not props then
		return
	end
	for _, v in ipairs(props) do
		builder:tag('td',{selfClosing=true})
				:css(v.style or {})
				:attr(v.attr or {})
	end
end

function p._main(cell_args)
	local ret = mw.html.create()
	local top = ret:tag('tr')
						:css{ height = '1px',
								['text-align'] = 'center' }
	local bottom = ret:tag('tr')
						:css{ height = '1px',
								['text-align'] = 'center' }
	for _, v in ipairs(cell_args) do
		if type(v) == 'string' then
			make_cell(top,v,'t')
			make_cell(bottom,v,'b')
		else
			top:tag('td')
				:attr{ colspan = v.colspan or 6,
						rowspan = v.rowspan or 2 }
				:css{ padding = '0.2em',
						border = (v.thick or '2') .. 'px solid black' }
				:cssText(v.css)
				:wikitext(v.text)
		end
	end
	return tostring(ret)
end

function p.main(frame)
	local args = require('Module:Arguments').getArgs(frame, {wrappers = 'Template:Chart', trim = false, removeBlanks = false})
	local cell_args = {}
	for _, val in ipairs(args) do
		local trimmedVal = val:match('^%s*(.-)%s*$')
		if trimmedVal == '' then
			trimmedVal = '$'
		end
		if cells[trimmedVal] then
			table.insert(cell_args, trimmedVal)
		else
			-- Unnamed params behave weirdly
			-- white space at the front counts for param_{{{1}}}, but not whitespace at the end, so remove it
			local rightTrimmedVal = val:gsub('%s+$','')
			table.insert(cell_args, {
				text = args[trimmedVal] or ('{{{'..trimmedVal..'}}}'),
				colspan = args['colspan_'..rightTrimmedVal] or args.colspan,
				rowspan = args['rowspan_'..rightTrimmedVal] or args.rowspan,
				thick = args['border_'..rightTrimmedVal] or args.border,
				css = args['boxstyle_'..rightTrimmedVal] or args.boxstyle
			})
		end
	end
	
	return p._main(cell_args)
end

return p