local p = {}
local concat = table.concat
local insert = table.insert
local format = mw.ustring.format
local tConfig = mw.loadData("Module:Tennis performance timeline/data")
local rounds = tConfig.rounds
local tournaments = tConfig.tournaments
local environments = tConfig.environments
local surfaces = tConfig.surfaces
local tOrders = tConfig.orders
local curYear = os.date("!*t").year
local calendar = mw.loadData("Module:Tennis performance timeline/calendar")
local genders = {
men = "Men's",
women = "Women's"
}
local matchTypes = {
singles = "Singles",
doubles = "Doubles"
}
--[[Utility functions]]
local function checkNonNil(value, type)
if value == nil then
error("Expected " .. type .. ", but is nil", 2)
end
return value
end
local function checkFormat(str, pattern, type)
if str == mw.ustring.match(str, pattern) then
return str
else
error("Invalid " .. type .. ": " .. str, 2)
end
end
local function checkYear(year)
checkNonNil(year, "year")
return checkFormat(year, "%d%d%d%d", "year")
end
local function checkNum(num, diagMsg)
diagMsg = "number" .. (diagMsg and " for " .. diagMsg or "")
checkNonNil(num, diagMsg);
return checkFormat(num, "%d+", diagMsg)
end
local function checkMember(elem, arr, type, diagMsg)
if arr[elem] then
return elem
else
diagMsg = type .. (diagMsg and " for " .. diagMsg or "")
checkNonNil(elem, diagMsg)
local message = {}
insert(message, "Invalid ")
insert(message, diagMsg)
insert(message, ": ")
insert(message, elem)
error(concat(message))
end
end
-- Format an HTML element with link, tooltip, and colors.
local function tooltip(tag, link, tooltip, text, spec)
spec = spec or {}
if spec.color then
tag:css('color', spec.color)
end
if spec.italic then
tag:wikitext("''");
end
if spec.bold then
tag:wikitext("'''");
end
if link then
tag:wikitext("[[" .. link .. "|")
end
if tooltip then
if spec.abbr then
tag:tag('abbr'):attr('title', tooltip):wikitext(text)
else
tag:attr('title', tooltip):wikitext(text)
end
else
tag:wikitext(text)
end
if link then tag:wikitext("]]") end
if spec.bold then
tag:wikitext("'''");
end
if spec.italic then
tag:wikitext("''");
end
end
-- Substitute "$[`param`]$" appearing in `str` with `value`.
-- For example, subst("$year$ ATP Tour", "year", "1990") -> "1990 ATP Tour"
local function subst(str, param, value)
if str == nil then return str end
return str:gsub("%$" .. param .. "%$", value)
end
local function tr()
return mw.html.create('tr')
end
local function th(row)
return row:tag('th')
end
local function td(row)
return row:tag('td')
end
local years
local data
local usedRounds
-- parser for data tables
local function parse(entry, year, tStats)
local entryType = type(entry)
if entryType == "string" then
return entry
elseif entryType == "table" then
if entry.type == "chrono" then
local numericYear = tonumber(year)
for _,elem in ipairs(entry) do
if type(elem) == "table" and numericYear >= elem[1] then
return parse(elem[2], year, tStats)
end
end
return parse(entry.default, year, tStats)
elseif entry.type == "switch" then
local param = entry.param
local arg = param == "year" and year or param == "gender" and data.gender or tStats[param]
if entry[arg] then return parse(entry[arg], year, tStats) end
return parse(entry.default, year, tStats)
else
return entry
end
end
end
-- Transform `param` entry in `array` with supplied data.
local function transform(array, param, data, tournament, year, country)
local entry = array[param]
entry = subst(entry, "gender", genders[data.gender])
entry = subst(entry, "matchType", matchTypes[data.matchType])
entry = subst(entry, "year", year)
if data[tournament] and data[tournament][year] then
local tStats = data[tournament][year]
if tournaments[tournament] and tournaments[tournament].region then
entry = subst(entry, "region",
parse(tournaments[tournament].region[country]
or tournaments[tournament].region.default, year, tStats))
end
if tournaments[tournament].group then
entry = subst(entry, "group", tournaments[tournament].group[tStats.group] or "")
end
end
array[param] = entry
end
-- Return wiki page title for the tournament in a given year.
local function annualTournamentLink(year, tStats, tInfo)
local annualLink = tInfo.annualLink
return parse(annualLink, year, tStats)
end
--[[-
Prepare the header row of the performance timeline table.
@return three elements:
- table header wikitext
- header row
- the first cell in the header row
]]
local function header()
local rows = {}
local row = tr()
local headerCell = th(row):attr('scope', 'col'):addClass('unsortable')
local tooltipSpec = {abbr = true}
if years.amateur or years.professional then
headerCell:attr('rowspan', 2)
local eras = {}
if years.amateur then
eras[#eras+1] = {years.amateur, "Amateur"}
end
if years.professional then
eras[#eras+1] = {years.professional, "Professional"}
end
eras[#eras+1] = {"1968", "Open Era"}
eras[#eras+1] = {tostring(years.last + 1)}
local lastEra = "?"
local idx = 1
for _,era in ipairs(eras) do
local count = 0
while years[idx] and years[idx] ~= era[1] do
count = count + 1
idx = idx + 1
end
if count > 0 then
th(row):attr('scope', 'col'):attr('colspan', count):wikitext(lastEra)
end
lastEra = era[2]
end
tooltip(th(row):attr('scope', 'col'):attr('rowspan', 2), nil, "Strike rate", "SR", tooltipSpec)
tooltip(th(row):attr('scope', 'col'):attr('rowspan', 2), nil, "Win–Loss", "W–L", tooltipSpec)
th(row):attr('scope', 'col'):attr('rowspan', 2):wikitext("Win %")
insert(rows, row)
row = tr()
end
if data.categories.count > 1 then
headerCell:attr('colspan', 2):wikitext("Tournament")
end
for _,year in ipairs(years) do
local link = subst(parse(tConfig.tours.link, year), "year", year)
th(row):attr('scope', 'col'):addClass('unsortable')
:wikitext(link and format("[[%s|%s]]", link, year) or year)
end
if #rows == 0 then
tooltip(th(row):attr('scope', 'col'), nil, "Strike rate", "SR", tooltipSpec)
tooltip(th(row):attr('scope', 'col'), nil, "Win–Loss", "W–L", tooltipSpec)
th(row):attr('scope', 'col'):wikitext("Win %")
end
insert(rows, row)
-- Add hatnote if needed.
local headerText = {}
if years.last and data.last and tournaments[data.last] then
local tInfo = tournaments[data.last]
local annualLink = {link = annualTournamentLink(years.last, nil, tInfo)}
transform(annualLink, "link", data, data.last, years.last)
local hatnote = {}
insert(hatnote, "''This table includes results through the conclusion of the [[")
annualLink = annualLink.link:gsub(" – .*$", "")
local annualName = annualLink
local annualSubst = tInfo.annualSubst or {}
for _,subst in ipairs(annualSubst) do
annualName = annualName:gsub(subst[1], subst[2])
end
if annualLink ~= annualName then
insert(hatnote, annualLink)
insert(hatnote, "|")
end
insert(hatnote, annualName)
insert(hatnote, "]].''")
insert(headerText, concat(hatnote))
end
insert(headerText, "{|class=\"plainrowheaders wikitable sortable\" style=text-align:center")
return concat(headerText, "\n"), rows, headerCell
end
local function outputSpan(row, span, seenRounds)
local cell = td(row):attr('colspan', span.span)
tooltip(cell, nil,
span.span < span.info.minSpellCols and span.info.tooltip,
span.span >= span.info.minSpellCols and span.info.tooltip or span.round,
span.info)
if seenRounds and span.span < span.info.minSpellCols then
seenRounds[span.info.round] = span.info
end
end
local footnoteCount
-- Return the year a given tournament was first held.
local function eventFirstYear(entries, yearInfos, year, yearEntry)
year = tonumber(year)
local startYear
if #entries > 0 then
local endYear = entries[#entries][4]
for testYear = endYear + 1, year do
local testYearTournament = parse(yearInfos, testYear)
if testYearTournament ~= entries[#entries][1] then
entries[#entries][4] = testYear - 1
break;
end
end
for testYear = year, endYear, -1 do
local testYearTournament = parse(yearInfos, testYear)
if testYearTournament ~= yearEntry then
startYear = testYear + 1
break;
end
end
end
return startYear
end
local frame_
-- Format a footnote.
local function footnoteText(footnoteChunks, wikilinkFn)
if #footnoteChunks > 1 then
local footnote = {}
local footnoteChunk = {"Held as"}
for pos,entry in ipairs(footnoteChunks) do
local link, name, abbr = wikilinkFn(entry)
insert(footnoteChunk, format("[[%s%s]]",
link and link .. "|" or "", name))
-- Add abbr if doesn't appear in name
if abbr and not name:match(abbr) then
insert(footnoteChunk, "(" .. abbr .. ")")
end
if entry[3] == entry[4] then
-- One-year event
insert(footnoteChunk, "in")
insert(footnoteChunk, entry[3])
else
if pos > 1 then
insert(footnoteChunk, "from")
insert(footnoteChunk, entry[3])
end
if pos < #footnoteChunks then
insert(footnoteChunk, pos == 1 and "until" or "to")
insert(footnoteChunk, entry[4])
end
end
insert(footnote, concat(footnoteChunk, (" ")))
footnoteChunk = {}
end
footnote[#footnote] = "and " .. footnote[#footnote]
return frame_:callParserFunction{name = '#tag:ref', args = {
concat(footnote, #footnote > 2 and ", " or " "),
group = "lower-alpha"
}}
end
end
local function winLossStatsSummary(row, stats, boldmarkup, summaryRowspan)
boldmarkup = boldmarkup or ""
-- − is (nonbreaking) minus sign.
local srCell = td(row):attr('data-sort-value', stats.count > 0 and stats.champs / stats.count or -1)
:addClass('nowrap')
:wikitext(format("%s%d / %d%s", boldmarkup, stats.champs, stats.count, boldmarkup))
local matches = stats.wins + stats.losses
local wlCell = td(row):attr('data-sort-value', matches > 0 and stats.wins / matches or -1)
local wrCell = td(row)
if summaryRowspan then
wlCell:attr('rowspan', summaryRowspan)
wrCell:attr('rowspan', summaryRowspan)
srCell:attr('rowspan', summaryRowspan)
end
if matches > 0 then
wlCell:wikitext(format("%s%d–%d%s", boldmarkup, stats.wins, stats.losses, boldmarkup))
wlCell:addClass("nowrap")
wrCell:wikitext(format("%s%.2f%%%s", boldmarkup, stats.wins * 100 / matches, boldmarkup))
else
wlCell:wikitext(format("%s–%s", boldmarkup, boldmarkup))
wrCell:attr('data-sort-value', '-1%')
:wikitext(format("%s–%s", boldmarkup, boldmarkup))
end
end
-- Prepare a Win-Loss row in the performance timeline table.
local function winLossStatsRow(row, info, stats, bold, summaryRowspan)
local boldmarkup = bold and "'''" or ""
local headerName = info.name and info.name .. " " or ""
th(row):attr('scope', 'row')
:wikitext(format("%s%sWin–Loss%s", boldmarkup, headerName, boldmarkup))
local span
for _,year in ipairs(years) do
local yStats = stats[year]
local display = {}
if yStats and yStats.wins + yStats.losses > 0 then
display.text = format("%s%d–%d%s", boldmarkup, yStats.wins, yStats.losses, boldmarkup)
bg_col = 'background-color:#EAECF0'
else
display.text = format("%s–%s", boldmarkup, boldmarkup)
bg_col = 'background-color:#EAECF0'
if info.absence then
local aConfig = parse(info.absence, year)
if aConfig then
display.text = aConfig.round
display.config = aConfig
end
end
end
if span then
if span.info.round == display.text then
span.span = span.span + 1
else
outputSpan(row, span)
span = nil
end
end
if not span then
if display.config and display.config.span then
span = {round = display.text, span = 1, info = display.config}
else
td(row):wikitext(display.text)
:addClass("nowrap")
end
end
end
if span then
outputSpan(row, span)
span = nil
end
winLossStatsSummary(row, stats, boldmarkup, summaryRowspan)
end
-- Return true if the player appears in a given tournament.
local function hasTournamentAppearance(tournament)
local tournamentType = type(tournament)
if tournamentType == "string" then
if data[tournament] then
return true
end
elseif tournamentType == "table" then
if tournament.type == "chrono" then
for _,entry in ipairs(tournament) do
if data[entry[2]] then
return true
end
end
if data[tournament.default] then
return true
end
else
-- TODO other table type
end
end
return false
end
-- Create a fresh statistics table.
local function statsFactory()
return {count = 0, champs = 0, finals = 0, wins = 0, losses = 0}
end
-- Generate performance timeline rows for a given tournament level.
local function body(level, levelHeaderCell)
local entries = {}
local stats = statsFactory()
local levelInfo = parse(tOrders[level])
local levelInfos = {}
local levelLastAppearance
for pos,tournament in ipairs(levelInfo) do
if hasTournamentAppearance(tournament) then
local tStats = statsFactory()
local row = tr()
if #entries == 0 and data.categories.count > 1 then
levelHeaderCell = th(row):attr('scope', 'row')
:css('width', '6.5em')
:css('max-width', '10em')
end
local headerCell = th(row):attr('scope', 'row')
local tInfos = {}
local lastAppearance
local seenRounds = {}
local span
local country = data.country.default
for _,year in ipairs(years) do
country = data.country[year] or country
local yearLevelName = parse(levelInfo.name or levelInfo.tooltip, year)
if #entries == 0 and (#levelInfos == 0 or levelInfos[#levelInfos][1] ~= yearLevelName) then
-- Add footnote noting series transition.
local startYear = eventFirstYear(levelInfos, levelInfo.name or levelInfo.tooltip, year, yearLevelName)
insert(levelInfos, {
yearLevelName,
{link = parse(levelInfo.link, year), abbr = parse(levelInfo.abbr, year)},
startYear, tonumber(year)
})
else
levelInfos[#levelInfos][4] = tonumber(year)
end
local yearTournament = parse(tournament, year)
local tInfo = checkNonNil(tournaments[yearTournament], "entry for " .. yearTournament)
if #tInfos == 0 or tInfos[#tInfos][2] ~= tInfo then
-- Add footnote noting tournament transition.
local startYear = eventFirstYear(tInfos, tournament, year, yearTournament)
insert(tInfos, {yearTournament, tInfo, startYear, tonumber(year)})
else
tInfos[#tInfos][4] = tonumber(year)
end
local display = {}
if data[yearTournament] and data[yearTournament][year] then
if not levelLastAppearance or levelLastAppearance < year then
levelLastAppearance = year
end
lastAppearance = tInfo
local tyStats = data[yearTournament][year]
if not rounds[tyStats.round].nocount then
tStats.count = tStats.count + 1
stats.count = stats.count + 1
end
if rounds[tyStats.round].strike then
tStats.champs = tStats.champs + 1
stats.champs = stats.champs + 1
end
if not stats[year] then
stats[year] = statsFactory()
end
tStats.wins = tStats.wins + tyStats.wins
stats[year].wins = stats[year].wins + tyStats.wins
stats.wins = stats.wins + tyStats.wins
tStats.losses = tStats.losses + tyStats.losses
stats[year].losses = stats[year].losses + tyStats.losses
stats.losses = stats.losses + tyStats.losses
local annualLink = annualTournamentLink(year, tyStats, tInfo)
display.round = tyStats.round
display.group = tyStats.group
display.link = annualLink
transform(display, "link", data, yearTournament, year, country)
else
display.round = parse(tInfo.absence, year) or "A"
end
local round = rounds[display.round]
if round.absence then
local absence = false
if year < years.last or not data.last and tonumber(year) < curYear then
absence = true
elseif year == years.last and data.last and calendar[year] and calendar[year][data.gender] then
local tWeek = calendar[year][data.gender].week[yearTournament]
local lWeek = calendar[year][data.gender].week[data.last]
if tWeek and lWeek and tWeek <= lWeek then
absence = true
end
end
if not absence then display.round = nil end
end
local roundInfo = {}
setmetatable(roundInfo, {__index = rounds[display.round]})
transform(roundInfo, "tooltip", data, yearTournament, year)
if roundInfo.group then
roundInfo.round = display.round .. display.group
else
roundInfo.round = display.round
end
display.round = roundInfo.name or roundInfo.round
if span then
if span.round == display.round then
span.span = span.span + 1
else
outputSpan(row, span, seenRounds)
span = nil
end
end
if not span then
if roundInfo.span then
span = {round = display.round, span = 1, info = roundInfo}
else
local cell = td(row)
if roundInfo.round then
if roundInfo.bgcolor then
cell:css('background', roundInfo.bgcolor)
end
tooltip(cell, display.link, roundInfo.tooltip, display.round, roundInfo)
seenRounds[roundInfo.round] = roundInfo
end
end
end
end
if span then
outputSpan(row, span, seenRounds)
span = nil
end
if lastAppearance then
headerCell:wikitext(
format("[[%s%s]]",
lastAppearance.link and lastAppearance.link .. "|" or
lastAppearance.abbr and lastAppearance.name .. "|" or "",
lastAppearance.abbr or lastAppearance.name))
if #tInfos > 1 then
local footnote = footnoteText(tInfos,
function(entry)
local tInfo = entry[2]
return tInfo.link, tInfo.name, tInfo.abbr
end)
headerCell:wikitext(footnote)
footnoteCount = footnoteCount + 1
end
winLossStatsSummary(row, tStats)
insert(entries, row)
for _,roundInfo in pairs(seenRounds) do
local round = roundInfo.name and not usedRounds[roundInfo.name] and roundInfo.name or roundInfo.round
usedRounds[round] = roundInfo
end
end
end
end
if #entries == 0 then return nil end
if levelHeaderCell then
local levelLink = parse(levelInfo.link, levelLastAppearance)
local levelName = parse(levelInfo.name, levelLastAppearance)
local levelAbbr = parse(levelInfo.abbr, levelLastAppearance)
local levelTooltip = parse(levelInfo.tooltip, levelLastAppearance)
local levelString = levelAbbr or levelName
if data.categories.count > 1 then
levelHeaderCell:attr('rowspan', #entries + 1)
end
tooltip(levelHeaderCell, levelLink or levelAbbr and levelName,
levelTooltip, levelString, {bold = true, abbr = true})
if #levelInfos > 1 then
local footnote = footnoteText(levelInfos,
function(entry)
return entry[2].link, entry[1], entry[2].abbr
end)
levelHeaderCell:wikitext(footnote)
footnoteCount = footnoteCount + 1
end
end
local row = tr()
winLossStatsRow(row, {}, stats, true)
insert(entries, row)
local result = {}
for _,entry in ipairs(entries) do
insert(result, tostring(entry))
end
return concat(result, "\n")
end
-- Generate rows for career performance timeline.
local function summary(envSummary, headerCell)
local entries = {}
local stats = statsFactory()
local environmentInfo = tOrders.environments
local surfaceInfo = tOrders.surfaces
local surfaceCount = 0
for _,environment in ipairs(environmentInfo) do
if data[environment] then
for _,surface in ipairs(surfaceInfo) do
if data[environment][surface] then
surfaceCount = surfaceCount + 1
end
end
end
end
if surfaceCount == 0 then return nil end
-- Aggregate data.
local eStats = {}
local sStats = {}
for _,env in ipairs(environmentInfo) do
if data[env] then
for _,surface in ipairs(surfaceInfo) do
if data[env][surface] then
for _,year in ipairs(years) do
local esyStats = data[env][surface][year]
if esyStats then
if not eStats[env] then
eStats[env] = statsFactory()
end
if not eStats[env][year] then
eStats[env][year] = statsFactory()
end
if not sStats[surface] then
sStats[surface] = statsFactory()
end
if not sStats[surface][year] then
sStats[surface][year] = statsFactory()
end
if not stats[year] then
stats[year] = statsFactory()
end
for key,value in pairs(esyStats) do
sStats[surface][year][key] = sStats[surface][year][key] + value
sStats[surface][key] = sStats[surface][key] + value
eStats[env][year][key] = eStats[env][year][key] + value
eStats[env][key] = eStats[env][key] + value
stats[year][key] = stats[year][key] + value
stats[key] = stats[key] + value
end
end
end
end
end
end
end
local function venueWinLossStatsRow(venues, vInfo, vStats)
for _,venue in ipairs(venues) do
if vStats[venue] then
local row = tr()
if #entries == 0 and data.categories.count > 1 then
-- Add header cell.
headerCell = th(row):attr('scope', 'row')
end
winLossStatsRow(row, vInfo[venue], vStats[venue])
insert(entries, row)
end
end
end
venueWinLossStatsRow(surfaceInfo, surfaces, sStats)
if envSummary then
venueWinLossStatsRow(environmentInfo, environments, eStats)
end
local row = tr()
winLossStatsRow(row, {name = "Overall"}, stats, true, 2)
insert(entries, row)
local row = tr()
local wrHdrCell = th(row):attr('scope', 'row'):wikitext("'''Win %'''")
for _,year in ipairs(years) do
local cellContent = "'''–'''"
if stats[year] then
local wins = stats[year].wins
local losses = stats[year].losses
local matches = wins + losses
if matches > 0 then
cellContent = format("'''%.1f%%'''", wins * 100 / matches)
end
end
td(row):wikitext(cellContent)
end
insert(entries, row)
if headerCell then
if data.categories.count > 1 then
headerCell:attr('rowspan', #entries)
end
headerCell:wikitext("'''Career'''")
end
local function counterStatsRow(row, name, type, bold)
local boldmarkup = bold and "'''" or ""
local rowHdrCell = th(row):attr('scope', 'row'):css('text-align', 'right')
:wikitext(format("%s%s%s", boldmarkup, name, boldmarkup))
if data.categories.count > 1 then rowHdrCell:attr('colspan', 2) end
for _,year in ipairs(years) do
td(row):wikitext(format("%s%s%s",
boldmarkup,
stats[year] and tostring(stats[year][type]) or "–",
boldmarkup))
end
td(row):attr('colspan', 2)
:wikitext(format("%s%d total%s", boldmarkup, stats[type], boldmarkup))
end
local row = tr():addClass('sortbottom')
counterStatsRow(row, "Tournaments played", "count")
td(row):wikitext(format("'''%.1f%%'''", stats.champs * 100 / stats.count))
insert(entries, row)
if stats.finals > 0 then
local row = tr():addClass('sortbottom')
counterStatsRow(row, "Finals reached", "finals")
td(row):attr('rowspan', 2)
:wikitext(format("'''%.1f%%'''", stats.champs * 100 / stats.finals))
insert(entries, row)
local row = tr():addClass('sortbottom')
counterStatsRow(row, "Titles", "champs", true)
insert(entries, row)
end
local row = tr():addClass('sortbottom')
local yearEndHdrCell = th(row):attr('scope', 'row'):css('text-align', 'right')
:wikitext("'''Year-end ranking'''")
if data.categories.count > 1 then yearEndHdrCell:attr('colspan', 2) end
for _,year in ipairs(years) do
local cell = td(row)
if data.rank and data.rank[year] then
local rank = data.rank[year]
local rankConfig = tConfig.rankings[rank] or {}
if rankConfig.bgcolor then
cell:css('background', rankConfig.bgcolor)
end
local boldmarkup = rankConfig.bold and "'''" or ""
cell:wikitext(format("%s%s%s", boldmarkup, rank, boldmarkup))
end
end
local cell = td(row):attr('colspan', 3)
if data.prizemoney then
tooltip(cell, nil, "Career prize money", data.prizemoney, {bold = true, abbr = true})
end
insert(entries, row)
local result = {}
for _,entry in ipairs(entries) do
insert(result, tostring(entry))
end
return concat(result, "\n")
end
-- Generate wikitext to conclude performance timeline table, including footnotes.
local function footer()
local result = {}
insert(result, "|-\n|}")
if footnoteCount > 0 then
local reflistTag = mw.html.create("div")
reflistTag:addClass("reflist"):css('list-style-type', 'lower-alpha')
:wikitext(frame_:callParserFunction{
name = '#tag:references', args = {"", group = "lower-alpha"}
})
insert(result, tostring(reflistTag))
end
return concat(result, "\n")
end
-- Return true if the player appears in a given tournament level.
local function hasLevelAppearance(level)
local levelInfo = parse(tOrders[level])
for _,tournament in ipairs(levelInfo) do
if hasTournamentAppearance(tournament) then return true end
end
return false
end
function p._main(args, frame)
frame_ = frame
data = {}
years = {}
usedRounds = {}
footnoteCount = 0
data.gender = args.gender or "men"
data.matchType = args.matchType or "singles"
data.country = {}
data.country.default = args.country or "UNK"
local idx = 1
local environmentSummary = true
local year
while args[idx] do
local arg = args[idx]
if arg == "year" then
idx = idx + 1
year = checkYear(args[idx])
if years.last and year <= years.last then
error(format("Nonincreasing year: %s appears after %s", year, years.last))
end
years.last = year
insert(years, year)
elseif arg == "country" then
idx = idx + 1
local country = checkNonNil(args[idx], year .. " country")
data.country[year] = args[idx]
elseif arg == "amateur" then
years.amateur = year
elseif arg == "professional" then
years.professional = year
elseif tournaments[arg] then
local tournament = arg
local diagMsg = year .." " .. tournament
local tStats = {}
idx = idx + 1
local round = args[idx]
-- Handle zones for Davis Cup.
if round and round:sub(1, 2) == "WG" then
tStats.round = "WG"
tStats.group = round:sub(3)
elseif round and round:sub(1, 2) == "PO" then
tStats.round = "PO"
tStats.group = round:sub(3)
elseif round and round:sub(1, 1) == "Z" then
tStats.round = "Z"
tStats.group = checkNum(round:sub(2), diagMsg .. " zone")
else
tStats.round = round
end
checkMember(tStats.round, rounds, "round", diagMsg)
idx = idx + 1
tStats.wins = checkNum(args[idx], diagMsg)
idx = idx + 1
tStats.losses = checkNum(args[idx], diagMsg)
if data[tournament] == nil then data[tournament] = {} end
data[tournament][year] = tStats
elseif environments[arg] then
local environment = arg
local diagMsg = year .. " " .. " " .. environment
idx = idx + 1
local surface = checkNonNil(args[idx], diagMsg .. " surface")
if surfaces[surface] then
diagMsg = diagMsg .. " " .. surface
local sStats = {}
idx = idx + 1
sStats.count = checkNum(args[idx], diagMsg)
idx = idx + 1
sStats.wins = checkNum(args[idx], diagMsg)
idx = idx + 1
sStats.losses = checkNum(args[idx], diagMsg)
idx = idx + 1
sStats.champs = checkNum(args[idx], diagMsg)
idx = idx + 1
sStats.finals = sStats.champs + checkNum(args[idx], diagMsg)
if data[environment] == nil then data[environment] = {} end
if data[environment][surface] == nil then
data[environment][surface] = {}
end
data[environment][surface][year] = sStats
else
error(format("Unknown surface (%s %s): %s", year, environment, arg))
end
elseif surfaces[arg] then
local surface = arg
local diagMsg = year .. " " .. surface
local sStats = {}
idx = idx + 1
sStats.count = checkNum(args[idx], diagMsg)
idx = idx + 1
sStats.wins = checkNum(args[idx], diagMsg)
idx = idx + 1
sStats.losses = checkNum(args[idx], diagMsg)
idx = idx + 1
sStats.champs = checkNum(args[idx], diagMsg)
idx = idx + 1
sStats.finals = sStats.champs + checkNum(args[idx], diagMsg)
if data.outdoor == nil then data.outdoor = {} end
if data["outdoor"][surface] == nil then data["outdoor"][surface] = {} end
data["outdoor"][surface][year] = sStats
-- Disable summary by environment.
environmentSummary = false
elseif arg == "rank" then
idx = idx + 1
if data.rank == nil then data.rank = {} end
data.rank[year] = checkNum(args[idx], year .. " rank")
else
error(format("Unknown argument at position %d (%s): %s", idx, year, arg))
end
idx = idx + 1
end
data.prizemoney = args.prizemoney
data.last = args.last
data.categories = {}
if args.types then
local count = 0
for _,type in ipairs(mw.text.split(args.types, ",")) do
if type == "Career" or tConfig.orders[type] and hasLevelAppearance(type) then
data.categories[type] = true
count = count + 1
end
end
data.categories.count = count
else
local count = 0
for _,type in ipairs(parse(tConfig.orders.order)) do
if hasLevelAppearance(type) then
data.categories[type] = true
count = count + 1
end
end
data.categories.Career = true
data.categories.count = count + 1
end
local result = {}
local tableHeader, headerRows, headerCell = header()
insert(result, tableHeader)
local function insertHeaderRowsIfNeeded()
if #result == 1 then
for _,headerRow in ipairs(headerRows) do
insert(result, tostring(headerRow))
end
end
end
for _,level in ipairs(parse(tConfig.orders.order)) do
if data.categories[level] then
local levelRows = body(level, data.categories.count == 1 and headerCell)
insertHeaderRowsIfNeeded()
insert(result, levelRows)
end
end
if data.categories.Career then
local careerRows = summary(environmentSummary, data.categories.count == 1 and headerCell)
insertHeaderRowsIfNeeded()
if careerRows then insert(result, careerRows) end
end
insert(result, footer())
return concat(result, "\n")
end
function p.main(frame)
-- Import module function to work with passed arguments
local getArgs = require('Module:Arguments').getArgs
local args = getArgs(frame)
return p._main(args, frame)
end
return p
