Jump to content

Module:PHL sports overall tally

From Wikipedia, the free encyclopedia
This is an old revision of this page, as edited by McVahl (talk | contribs) at 05:47, 31 March 2020 (reduce globals). The present address (URL) is a permanent link to this revision, which may differ significantly from the current revision.

require('Module:No globals')

local p = {}

local div = {
	senior = { 'M', 'W', 'C'},
	junior = { 'B', 'G', 'C'}
}

local evt = { 
	['BSKB'] = {  1, 'bball',    'Basketball' },
	['3X3B'] = {  2, '3x3bball', '3x3 basketball' },
	['INVB'] = {  3, 'vball',    'Volleyball (indoor)' },
	['BCVB'] = {  4, 'bvolley',  'Volleyball (beach)' },
	['SWMM'] = {  5, 'swim',	 'Swimming' },
	['CHSS'] = {  6, 'chess',    'Chess' },
	['TNNS'] = {  7, 'tennis',   'Tennis' },
	['TBTN'] = {  8, 'tbltenn',  'Table tennis' },
	['BDMT'] = {  9, 'badmnt',   'Badminton' },
	['TKWD'] = { 10, 'taekwd',   'Taekwondo' },
	['JUDO'] = { 11, 'judo',     'Judo' },
	['BSBL'] = { 12, 'bsball',   'Baseball' },
	['SFBL'] = { 13, 'sfball',   'Softball' },
	['FTBL'] = { 14, 'ftball',   'Football' },
	['ATHL'] = { 15, 'track',    'Athletics' },
	['FENC'] = { 16, 'fencing',  'Fencing' }
}

local colors = {
	{ '#FFCE09', 'Champion'},
	{ '#BABBBC', 'Runner-up' },
	{ '#D6A461', 'Third place' },
	['WD'] = { '#FFBBBB', 'Withdrew' },
	['NT'] = { nil, 'No team' }
}

local function isnotempty(s)
	return s and s:match('^%s*(.-)%s*$') ~= ''
end

local function stripwhitespace(text)
	return text:match("^%s*(.-)%s*$")
end

local function findchamp(teams, t, r, ptsbyrank)
	local found = teams[t].res[r].pts == ptsbyrank[1]
	if found or (t == 1) then return found else return findchamp(teams, t - 1, r, ptsbyrank) end
end

local function getevtdisp(teams, division, ptsbyrank)
	local r = 1
	local evt_disp = {}
	for ke, ve in pairs(evt) do
		for kd, vd in pairs(div[division]) do
			local found = findchamp(teams, #teams, r, ptsbyrank)
			table.insert(evt_disp, found)
			r = r + 1
		end
	end
	return evt_disp
end

local function getindex(map, value)
	for k, v in pairs(map) do
		if v == value then return tonumber(k) end
	end
	return tonumber(99)
end

local function getvte(frame, args)
	local baselink = frame:getParent():getTitle()
	if mw.title.getCurrentTitle().text == baselink then	baselink = '' end
	local vtetemplate = args['tname'] or (baselink ~= '' and (':' .. baselink)) or ''

	if vtetemplate ~= '' then
		return frame:expandTemplate{ title = 'navbar', args = { mini=1, style='float:left', brackets=1, vtetemplate} }
	end
	return nil
end

local function getbg(pts, raw, ptsbyrank)
	if	   ptsbyrank[1] == pts then return colors[1][1]
	elseif ptsbyrank[2] == pts then return colors[2][1]
	elseif ptsbyrank[3] == pts then return colors[3][1]
	elseif raw == 'WD'		 then return colors.WD[1]
	else	return nil
	end
end

local function getdisptext(pts,raw,ptsbyrank)
	if pts ~= 0 then return pts
	elseif string.match(raw,'([?|WD])') then return raw
	else return ptsbyrank.NT
	end
end

local function prefillvalues(args,teams,division,ptsbyrank,ovptsonly)
	local tally = {}
	for kt, vt in pairs(teams) do
		local res, subtotal, overall, gold, silver, bronze = {}, {}, 0, 0, 0, 0
		
		local evt_disp = getevtdisp(teams,division,ptsbyrank)
		for kr, vr in pairs(vt.res) do
			if evt_disp[kr] then
				subtotal[vr.div] = (tonumber(subtotal[vr.div]) or 0) + vr.pts
				if (vr.pts == ptsbyrank[1]) then gold = gold + 1
				elseif (vr.pts == ptsbyrank[2]) then silver = silver + 1
				elseif (vr.pts == ptsbyrank[3]) then bronze = bronze + 1 end
				table.insert(res, vr)
			end
		end
		for kd, vd in pairs(div[division]) do overall = overall + tonumber(subtotal[vd] or 0) end

		if ovptsonly then
			overall = tonumber(args['pts_'..vt.code]) or overall
		end
		
		table.sort(res, function (a, b) return evt[a.evt][1] < evt[b.evt][1] or (evt[a.evt][1] == evt[b.evt][1] and getindex(div[division],a.div) < getindex(div[division],b.div)) end)
		table.insert(tally, { rank = vt.rank, code = vt.code, team = vt.name, res = res, subtotal = subtotal, overall = overall, medals = { gold, silver, bronze } })
	end
	table.sort(tally, function (a, b)
		if not ovptsonly then
			return a.overall > b.overall or (a.overall == b.overall
				and (a.medals[1] > b.medals[1]
				or (a.medals[1] == b.medals[1] and a.medals[2] > b.medals[2])
				or (a.medals[1] == b.medals[1] and a.medals[2] == b.medals[2] and a.medals[3] > b.medals[3])
				or (a.medals[1] == b.medals[1] and a.medals[2] == b.medals[2] and a.medals[3] == b.medals[3] and a.code < b.code)))
		else return a.overall > b.overall or (a.overall == b.overall and a.rank < b.rank) end
		end)
	return tally
end

local function medaltable(frame,args,tally,division)
	local mMedals = require('Module:Medals table')
	args['team'] = 'Team'
	args['event'] = 'inst'
	args['flag_template'] = args['team_template'] or 'UAAPteam'
	for kt, vt in pairs(tally) do
		local name = args['name_'..vt.code]
		if division == 'junior' and isnotempty(args['j_short_'..vt.code]) then
			args['name_'..vt.code] = frame:expandTemplate{title = args['flag_template'], args = { vt.code, division, inst = args['j_short_'..vt.code] } }
		elseif not isnotempty(name) or name == nil then
			args['name_'..vt.code] = vt.name
		end
		if stripwhitespace(args['status_'..vt.code] or '') == 'H' then
			args['host_'..vt.code] = 'yes'
			args['host'] = 'Season host'
		end
		args['gold_'..vt.code] = vt.medals[1]
		args['silver_'..vt.code] = vt.medals[2]
		args['bronze_'..vt.code] = vt.medals[3]
	end
	return mMedals.createTable(frame, args)
end

local function buildtable(frame,args,teams,division,ptsbyrank,showmedals,ovptsonly,sumsonly,vtetemplate)
	local tally = prefillvalues(args,teams,division,ptsbyrank,ovptsonly)
	
	if showmedals then
		return medaltable(frame,args,tally,division)
	end
	
	local root = mw.html.create()
	local footer = mw.html.create()
	local abbr = mw.html.create('abbr')
	root = root:tag('table')
		:addClass('wikitable')
		:addClass('plainrowheaders')
		:css('font-size', (ovptsonly or sumsonly) and '100%' or '95%')
		:css('text-align', 'center')
	
	-- header row (1)
	local evts = tally[1].res
	local divs = div[division]
	local row = root:tag('tr')
	local celltype = not ovptsonly and 'th' or 'td'
	local showc, showhost = false, false
	
	if not ovptsonly then
		row:tag('th')
			:attr('scope', 'col')
			:attr('colspan', '2')
			:wikitext(getvte(frame,args))
		
		abbr:attr('title', 'Mixed or co-ed'):wikitext(divs[3])
		
		-- column spanning by event
		local prevspan, prevcell, prevevt = 0, nil, nil
		for ke, ve in pairs(evts) do
			local evtname = evt[ve.evt][3]
			if (ve.div == divs[3]) then
				showc = true
			end
			if not sumsonly then
				if (prevevt == ve.evt) then
					prevspan = prevspan + 1
					prevcell
						:attr('colspan', prevspan)
				else
					prevspan = 1
					prevcell = row:tag('th')
						:attr('scope', 'col')
						:wikitext(string.format('[[File:%s pictogram.svg|20px|link=|%s]]', evtname, evtname))
					
					prevevt = ve.evt
				end
			end
		end

		row:tag('th')
			:attr('scope', 'col')
			:attr('colspan', showc and 4 or 3)
			:css('border-left-width', '3px')
			:wikitext('Total')
	end
	
	-- header row (2)
	row = root:tag('tr')
	row:tag('th')
			:attr('scope', 'col')
			:attr('width', '50px')
			:wikitext('Rank')
		:tag('th')
			:attr('scope', 'col')
			:attr('width', '90px')
			:wikitext('Team')
	
	if not (ovptsonly or sumsonly) then
		for ke, ve in pairs(evts) do
			row:tag('th')
				:attr('scope', 'col')
				:attr('width', '22px')
				:wikitext(ve.div == divs[3] and tostring(abbr) or ve.div)
		end
	end
	
	if not ovptsonly then
		for kd, vd in pairs(divs) do
			if (not showc and vd == divs[3]) then break
			else
				row:tag('th')
					:attr('scope', 'col')
					:attr('width', '22px')
					:css('border-left-width', (kd == 1) and '3px' or nil)
					:wikitext(vd == divs[3] and tostring(abbr) or vd)
			end
		end
	end
	
	row:tag('th'):attr('scope', 'col'):wikitext(ovptsonly and 'Points' or 'Overall')
			
	-- team row
	for ka, va in pairs(tally) do
		local overall = 0
		local teamtext = va.team

		if stripwhitespace(args['status_'..va.code] or '') == 'H' then
			showhost = true
			teamtext = va.team..'&nbsp;<b>(H)</b>'
		end
		
		row = root:tag('tr')
		row:tag(celltype)
				:attr('scope', 'row')
				:css('text-align', 'center')
				:wikitext(ka)
			:tag('td')
				:attr('scope', 'row')
				:css('white-space', 'nowrap')
				:css('text-align', 'left')
				:wikitext(teamtext)
		
		if not ovptsonly then
			if not sumsonly then
				for kr, vr in pairs(va.res) do
					row:tag('td')
						:css('background-color', getbg(vr.pts,vr.raw,ptsbyrank))
						:wikitext(getdisptext(vr.pts,vr.raw,ptsbyrank))
				end
			end
			
			for kd, vd in pairs(divs) do
				if (not showc and vd == divs[3]) then break
				else row:tag('td')
					:css('border-left-width', (kd == 1) and '3px' or nil)
					:wikitext(va.subtotal[vd] or 0)
				end
			end
		end
		
		row:tag(celltype)
			:attr('scope', 'row')
			:css('font-weight', 'bold')
			:css('text-align', 'center')
			:wikitext(va.overall)
	end
	
	local legend = footer:tag('div'):attr('class', 'reflist')
	if showhost then
		legend:tag('span')
			:css('font-weight', 'bold')
			:wikitext('(H)')
			:done()
		:wikitext('&nbsp;Season host')
		if ovptsonly or sumsonly then legend:wikitext('.') end
	end
	if not (ovptsonly or sumsonly) then
		local firsttag = not showhost
		for kp, vp in pairs(ptsbyrank) do
			if firsttag == false then legend:wikitext('; ') end
			legend:tag('span')
				:css('margin', '0')
				:css('white-space', 'nowrap')
				:tag('span')
					:addClass('legend-text')
					:css('border', 'none')
					:css('padding', '1px .3em')
					:css('background-color', getbg(vp,nil,ptsbyrank))
					:css('font-size', '95%')
					:css('border', '1px solid #BBB')
					:css('line-height', '1.25')
					:css('text-align', 'center')
					:wikitext(vp)
					:done()
				:wikitext(' = ' .. (colors[kp] or colors.NT)[2])
			firsttag = false
		end
		legend:wikitext('.')
	end
	
	return tostring(root)..tostring(footer)
end

function p.main(frame)
	local getArgs = require('Module:Arguments').getArgs
	local args = getArgs(frame, { parentFirst = true })
	
	local yesno = require('Module:Yesno')
	local showmedals = yesno(args['show_medals'] or 'n')
	local ovptsonly = yesno(args['overall_pts_only'] or 'n')
	local sumsonly = yesno(args['subtotals_only'] or 'n')
	local division = (args['division'] or 'senior'):lower()
	local ptsbyrank = { tonumber(args['pts_1st']) or 15, tonumber(args['pts_2nd']) or 12, tonumber(args['pts_3rd']) or 10, NT = '&mdash;' }
	
	local team_list = {}
	local tally = {}
	local template = args['team_template'] or 'UAAPteam'
	for ka, va in pairs(args) do
		-- Process team args
		local i = tostring(ka):match('^team([%d]+)%s*$') or '0'
		if (tonumber(i) > 0 and isnotempty(va)) then
			local res, t = {}, args['team' .. i]
			local sname = args['short_' .. t]
			if division == 'junior' and isnotempty(args['j_short_' .. t]) then sname = args['j_short_' .. t] end
			local tname = args['name_' .. t] or
				(isnotempty(sname) and
					frame:expandTemplate{title = template, args = { t, division, name = sname } } or 
					frame:expandTemplate{title = template, args = { t, division, 'short' } }
				)
			for ke, ve in pairs(evt) do
				for kd, vd in pairs(div[division]) do
					local cvd = vd
					if (kd == 3) then cvd = division:sub(1,1) end
					local evt_pts = stripwhitespace(args[cvd:lower()..'_'..ke..'_'..t] or args[t..'-'..cvd:lower()..'_'.. ve[2]:lower()] or '')
					table.insert(res, { div = vd, evt = ke, raw = evt_pts, pts = tonumber(evt_pts) or 0 })
				end
			end
			table.insert(team_list, {rank = i, code = t, name = tname, res = res})
		end
	end
	return buildtable(frame,args,team_list,division,ptsbyrank,showmedals,ovptsonly,sumsonly)
end

return p