Module:Infobox3cols

-- -- This module implements -- -- The initial version was created by modifying Module:Infobox -- local p = {}

local navbar = require('Module:Navbar')._navbar

local args = {} local origArgs local root

local function union(t1, t2) -- Returns the union of the values of two tables, as a sequence. local vals = {} for k, v in pairs(t1) do       vals[v] = true end for k, v in pairs(t2) do       vals[v] = true end local ret = {} for k, v in pairs(vals) do       table.insert(ret, k)    end return ret end

local function getArgNums(prefix,suffix) -- Returns a table containing the numbers of the arguments that exist -- for the specified prefix. For example, if the prefix was 'data', and -- 'data1', 'data2', and 'data5' exist, it would return {1, 2, 5}. local nums = {} for k, v in pairs(args) do       local num = tostring(k):match('^' .. prefix .. '([0-9]%d*)' .. suffix .. '$') if num then table.insert(nums, tonumber(num)) end end table.sort(nums) return nums end

local function addRow(rowArgs) -- Adds a row to the infobox, with either a header cell -- or a label/data cell combination. if rowArgs.header then root :tag('tr') :addClass(rowArgs.rowclass) :cssText(rowArgs.rowstyle) :attr('id', rowArgs.rowid) :tag('th') :attr('colspan', 4) :attr('id', rowArgs.headerid) :addClass(rowArgs.class) :addClass(args.headerclass) :css('text-align', 'center') :cssText(args.headerstyle) :cssText(rowArgs.rowcellstyle) :cssText(args.extracellstyles[rowArgs.num .. 'h']) :wikitext(rowArgs.header) elseif rowArgs.label then if rowArgs.data then local row = root:tag('tr') row:addClass(rowArgs.rowclass) row:cssText(rowArgs.rowstyle) row:attr('id', rowArgs.rowid) row :tag('th') :attr('scope', 'row') :attr('id', rowArgs.labelid) :cssText(args.labelstyle) :cssText(rowArgs.rowcellstyle) :cssText(args.extracellstyles[rowArgs.num .. 'l']) :wikitext(rowArgs.label) :done local dataCell = row:tag('td') dataCell :attr('colspan', 3) :attr('id', rowArgs.dataid) :addClass(rowArgs.class) :cssText(rowArgs.datastyle) :cssText(rowArgs.rowcellstyle) :cssText(args.extracellstyles[rowArgs.num]) :newline :wikitext(rowArgs.data) elseif rowArgs.dataa or rowArgs.datab then local row = root:tag('tr') row:addClass(rowArgs.rowclass) row:cssText(rowArgs.rowstyle) row:attr('id', rowArgs.rowid) row :tag('th') :attr('scope', 'row') :attr('id', rowArgs.labelid) :cssText(args.labelstyle) :cssText(rowArgs.rowcellstyle) :cssText(args.extracellstyles[rowArgs.num .. 'h']) :wikitext(rowArgs.label) :done

local dataCella = row:tag('td') dataCella :attr('id', rowArgs.dataaid) :addClass(rowArgs.classa) :cssText(rowArgs.dataastyle) :cssText(rowArgs.rowcellstyle) :cssText(args.extracellstyles[rowArgs.num .. 'a']) :newline :wikitext(rowArgs.dataa) if rowArgs.renderb then local dataCellb = row:tag('td') dataCellb :attr('id', rowArgs.databid) :addClass(rowArgs.classb) :cssText(rowArgs.databstyle) :cssText(rowArgs.rowcellstyle) :cssText(args.extracellstyles[rowArgs.num .. 'b']) :newline :wikitext(rowArgs.datab) end if rowArgs.renderc then local dataCellc = row:tag('td') dataCellc :attr('id', rowArgs.datacid) :addClass(rowArgs.classc) :cssText(rowArgs.datacstyle) :cssText(rowArgs.rowcellstyle) :cssText(args.extracellstyles[rowArgs.num .. 'c']) :newline :wikitext(rowArgs.datac) end end elseif rowArgs.data then local row = root:tag('tr') row:addClass(rowArgs.rowclass) row:cssText(rowArgs.rowstyle) row:attr('id', rowArgs.rowid) local dataCell = row:tag('td') dataCell :attr('colspan', 4) :attr('id', rowArgs.dataid) :addClass(rowArgs.class) :css('text-align', 'center') :cssText(rowArgs.datastyle) :cssText(rowArgs.rowcellstyle) :cssText(args.extracellstyles[rowArgs.num]) :newline :wikitext(rowArgs.data) end end

local function renderTitle if not args.title then return end

root :tag('caption') :addClass(args.titleclass) :cssText(args.titlestyle) :wikitext(args.title) end

local function renderAboveRow if not args.above then return end root :tag('tr') :tag('th') :attr('colspan', 4) :addClass(args.aboveclass) :css('text-align', 'center') :css('font-size', '125%') :css('font-weight', 'bold') :cssText(args.abovestyle) :wikitext(args.above) end

local function renderBelowRow if not args.below then return end root :tag('tr') :tag('td') :attr('colspan', 4) :addClass(args.belowclass) :css('text-align', 'center') :cssText(args.belowstyle) :newline :wikitext(args.below) end

local function renderSubheaders if args.subheader then args.subheader1 = args.subheader end if args.subheaderrowclass then args.subheaderrowclass1 = args.subheaderrowclass end local subheadernums = getArgNums('subheader','') for k, num in ipairs(subheadernums) do       addRow({            data = args['subheader' .. tostring(num)],            datastyle = args.subheaderstyle or args['subheaderstyle' .. tostring(num)],            class = args.subheaderclass,            rowclass = args['subheaderrowclass' .. tostring(num)]        }) end end

local function renderImages if args.image then args.image1 = args.image end if args.caption then args.caption1 = args.caption end local imagenums = getArgNums('image','') for k, num in ipairs(imagenums) do       local caption = args['caption' .. tostring(num)] local data = mw.html.create:wikitext(args['image' .. tostring(num)]) if caption then data :tag('div') :cssText(args.captionstyle) :wikitext(caption) end addRow({           data = tostring(data),            datastyle = args.imagestyle,            class = args.imageclass,            rowclass = args['imagerowclass' .. tostring(num)]        }) end end

local function renderRows -- Gets the union of the header and data argument numbers, -- and renders them all in order using addRow. local rownums = union(getArgNums('header',''), getArgNums('data','[ab]?')) local datab_count = #(getArgNums('data','b')) local datac_count = #(getArgNums('data','c'))

table.sort(rownums) for k, num in ipairs(rownums) do       addRow({        	num = tostring(num),            renderb = datab_count > 0,            renderc = datac_count > 0,            header = args['header' .. tostring(num)],            label = args['label' .. tostring(num)],            data = args['data' .. tostring(num)],            datastyle = args.datastyle,            class = args['class' .. tostring(num)],            dataa = args['data' .. tostring(num) .. 'a'],            dataastyle = args.datastylea,            classa = args['class' .. tostring(num) .. 'a'],            datab = args['data' .. tostring(num) .. 'b'],            databstyle = args.datastyleb,            classb = args['class' .. tostring(num) .. 'b'],            datac = args['data' .. tostring(num) .. 'c'],            datacstyle = args.datastylec,            classc = args['class' .. tostring(num) .. 'c'],            rowclass = args['rowclass' .. tostring(num)], rowstyle = args['rowstyle' .. tostring(num)], rowcellstyle = args['rowcellstyle' .. tostring(num)], dataid = args['dataid' .. tostring(num)], dataaid = args['dataid' .. tostring(num) .. 'a'], dataaib = args['dataid' .. tostring(num) .. 'b'], dataaic = args['dataid' .. tostring(num) .. 'c'], labelid = args['labelid' .. tostring(num)], headerid = args['headerid' .. tostring(num)], rowid = args['rowid' .. tostring(num)] })   end end

local function renderNavBar if not args.name then return end root :tag('tr') :tag('td') :attr('colspan', '4') :css('text-align', 'right') :wikitext(navbar{                   args.name,                    mini = 1,                }) end

local function renderItalicTitle local italicTitle = args['italic title'] and mw.ustring.lower(args['italic title']) if italicTitle == '' or italicTitle == 'force' or italicTitle == 'yes' then root:wikitext(mw.getCurrentFrame:expandTemplate({title = 'italic title'})) end end

local function renderTrackingCategories if args.decat ~= 'yes' then if #(getArgNums('data','[abc]?')) == 0 and mw.title.getCurrentTitle.namespace == 0 then root:wikitext('') end end end

local function _infobox -- Specify the overall layout of the infobox, with special settings -- if the infobox is used as a 'child' inside another infobox.

root = mw.html.create('table') root :addClass('infobox') :addClass(args.bodyclass) if args.child == 'yes' or args.subbox == 'yes' then root :css('padding', '0') :css('border', 'none') :css('margin', (args.subbox == 'yes') and '-3px' or 'auto') :css('width', 'auto') :css('min-width', '100%') :css('font-size', '100%') :css('clear', 'none') :css('float', 'none') :css('background-color', 'transparent') else root :css('width', '22em') end

root:cssText(args.bodystyle) renderTitle renderAboveRow renderSubheaders renderImages renderRows renderBelowRow renderNavBar renderItalicTitle renderTrackingCategories return tostring(root) end

local function preprocessSingleArg(argName) -- If the argument exists and isn't blank, add it to the argument table. -- Blank arguments are treated as nil to match the behaviour of ParserFunctions. if origArgs[argName] and origArgs[argName] ~= '' then args[argName] = origArgs[argName] end end

local function preprocessArgs(prefixTable, step) -- Assign the parameters with the given prefixes to the args table, in order, in batches -- of the step size specified. This is to prevent references etc. from appearing in the -- wrong order. The prefixTable should be an array containing tables, each of which has -- two possible fields, a "prefix" string and a "depend" table. The function always parses -- parameters containing the "prefix" string, but only parses parameters in the "depend" -- table if the prefix parameter is present and non-blank. if type(prefixTable) ~= 'table' then error("Non-table value detected for the prefix table", 2) end if type(step) ~= 'number' then error("Invalid step value detected", 2) end -- Get arguments without a number suffix, and check for bad input. for i,v in ipairs(prefixTable) do       if type(v) ~= 'table' or type(v.prefix) ~= "string" or (v.depend and type(v.depend) ~= 'table') then error('Invalid input detected to preprocessArgs prefix table', 2) end preprocessSingleArg(v.prefix) -- Only parse the depend parameter if the prefix parameter is present and not blank. if args[v.prefix] and v.depend then for j, dependValue in ipairs(v.depend) do               if type(dependValue) ~= 'string' then error('Invalid "depend" parameter value detected in preprocessArgs') end preprocessSingleArg(dependValue) end end end

-- Get arguments with number suffixes. local a = 0 -- Counter variable. local moreArgumentsExist = true while moreArgumentsExist == true do       moreArgumentsExist = false for i = a, a + step - 1 do           for j,v in ipairs(prefixTable) do                local prefixArgName = v.prefix .. tostring(i) .. (v.suffix or '') if origArgs[prefixArgName] then moreArgumentsExist = true -- Do another loop if any arguments are found, even blank ones. preprocessSingleArg(prefixArgName) end -- Process the depend table if the prefix argument is present and not blank, or               -- we are processing "prefix1" and "prefix" is present and not blank, and -- if the depend table is present. if v.depend and (args[prefixArgName] or (i == 1 and args[v.prefix])) then for j,dependValue in ipairs(v.depend) do                       local dependArgName = dependValue .. tostring(i) .. (v.dependsuffix or '') preprocessSingleArg(dependArgName) end end end end a = a + step end end function p.infobox(frame) -- If called via #invoke, use the args passed into the invoking template. -- Otherwise, for testing purposes, assume args are being passed directly in. if frame == mw.getCurrentFrame then origArgs = frame:getParent.args else origArgs = frame end -- Parse the data parameters in the same order that the old did, so that -- references etc. will display in the expected places. Parameters that depend on   -- another parameter are only processed if that parameter is present, to avoid -- phantom references appearing in article reference lists. preprocessSingleArg('child') preprocessSingleArg('bodyclass') preprocessSingleArg('subbox') preprocessSingleArg('bodystyle') preprocessSingleArg('title') preprocessSingleArg('titleclass') preprocessSingleArg('titlestyle') preprocessSingleArg('above') preprocessSingleArg('aboveclass') preprocessSingleArg('abovestyle') preprocessArgs({       {prefix = 'subheader', depend = {'subheaderstyle', 'subheaderrowclass'}}    }, 10) preprocessSingleArg('subheaderstyle') preprocessSingleArg('subheaderclass') preprocessSingleArg('image') preprocessSingleArg('caption') preprocessArgs({       {prefix = 'image', depend = {'caption', 'imagerowclass'}}    }, 10) preprocessSingleArg('captionstyle') preprocessSingleArg('imagestyle') preprocessSingleArg('imageclass') preprocessArgs({       {prefix = 'header'},        {prefix = 'data', depend = {'label'}},        {prefix = 'data', suffix = 'a', depend = {'label'}},        {prefix = 'data', suffix = 'a', depend = {'data'}, dependsuffix='c'},        {prefix = 'data', suffix = 'b', depend = {'label'}},        {prefix = 'data', suffix = 'b', depend = {'data'}, dependsuffix='c'},        {prefix = 'rowclass'},        {prefix = 'rowstyle'},        {prefix = 'rowcellstyle'},        {prefix = 'class'},        {prefix = 'dataid'},        {prefix = 'labelid'},        {prefix = 'headerid'},        {prefix = 'rowid'}    }, 50) preprocessSingleArg('headerclass') preprocessSingleArg('headerstyle') preprocessSingleArg('labelstyle') preprocessSingleArg('datastyle') preprocessSingleArg('datastylea') preprocessSingleArg('datastyleb') preprocessSingleArg('datastylec') preprocessSingleArg('below') preprocessSingleArg('belowclass') preprocessSingleArg('belowstyle') preprocessSingleArg('name') args['italic title'] = origArgs['italic title'] -- different behaviour if blank or absent preprocessSingleArg('decat')

args['extracellstyles'] = {} for line in mw.text.gsplit(origArgs['extracellstyles'] or '', '\n') do		local equals = line:find('=') if equals then for i in mw.text.gsplit(line:sub(1, equals - 1), '%s*,%s*') do				args.extracellstyles[i] = line:sub(equals + 1) end end end return _infobox end

return p