Module:Pokemon
Documentation for this module may be created at Module:Pokemon/doc
local p = {}
local DATA_FORMAT = 'CSV with header'
local COLUMNS = {
number = 'Pokedex Number',
name = 'Name',
type1 = 'Type1',
type2 = 'Type2',
ability1 = 'Ability 1',
ability2 = 'Ability 2',
ha = 'Hidden Ability',
generation = 'Generation',
hp = 'HP',
atk = 'Attack',
def = 'Defense',
spatk = 'Special Attack',
spdef = 'Special Defense',
spd = 'Speed',
catch = 'Catch Rate',
evhp = 'EVHP',
evatk = 'EVATK',
evdef = 'EVDEF',
evspa = 'EVSPA',
evspd = 'EVSPD',
evsp = 'EVSP',
male = 'Male Ratio',
height = 'Height',
weight = 'Weight',
npc = 'NPC',
exp = 'Base Experience',
rarity = 'Tier',
}
local GEN_NUMERALS = { 'I', 'II', 'III', 'IV', 'V', 'VI', 'VII', 'VIII', 'IX' }
local STATS = {
{ key = 'hp', label = 'HP', template = 'HP' },
{ key = 'atk', label = 'Atk', template = 'Attack' },
{ key = 'def', label = 'Def', template = 'Defense' },
{ key = 'spatk', label = 'Sp.Atk', template = 'Special Attack' },
{ key = 'spdef', label = 'Sp.Def', template = 'Special Defense' },
{ key = 'spd', label = 'Spd', template = 'Speed' },
}
local EV_KEYS = { 'evhp', 'evatk', 'evdef', 'evspa', 'evspd', 'evsp' }
local function nonEmpty(s)
if s == nil then return nil end
s = mw.text.trim(tostring(s))
if s == '' then return nil end
return s
end
local function templateArgs(frame)
local parent = frame:getParent()
return (parent and parent.args) or frame.args
end
local function padNumber(n)
n = tonumber(n) or 0
return string.format('%03d', n)
end
local PokemonForms = require('Module:PokemonForms')
local FORM_SUFFIXES = PokemonForms.FORM_SUFFIXES
local STRIP_SUFFIXES = PokemonForms.STRIP_SUFFIXES
local MEGA_VARIANTS = PokemonForms.MEGA_VARIANTS or { '', ' X', ' Y' }
-- The single data fetch: the whole raw list is loaded once per page render
-- and every lookup (by name, by number, form probing) reads from it.
local allRowsCache = nil
local numberIndexCache = nil
local function loadAllRows()
if allRowsCache then
return allRowsCache
end
local csvData = mw.ext.externalData.getExternalData{
url = 'https://wiki.pokemonrevolution.net/index.php?title=Special:GetData/PokemonRawList',
format = DATA_FORMAT,
data = COLUMNS,
}
allRowsCache = {}
numberIndexCache = {}
if csvData then
for _, row in ipairs(csvData) do
if nonEmpty(row.name) then
allRowsCache[row.name] = row
local num = nonEmpty(row.number)
-- First occurrence wins so base forms beat Megas sharing a number.
if num and numberIndexCache[num] == nil then
numberIndexCache[num] = row
end
end
end
end
return allRowsCache
end
local function hyphenSpaceVariant(s)
if mw.ustring.find(s, '-', 1, true) then
return (mw.ustring.gsub(s, '%-', ' '))
end
return (mw.ustring.gsub(s, ' ', '-'))
end
local function byName(name)
local rows = loadAllRows()
return rows[name] or rows[hyphenSpaceVariant(name)]
end
local function byNumber(number)
loadAllRows()
return numberIndexCache[tostring(number)]
end
-- Mega evolutions live in a separate CSV without a Special:GetData page, so
-- they are fetched per name through the filtered file source — the same path
-- Template:PokemonInfo/Mega has always used.
local MEGA_COLUMNS = {
number = 'Pokedex Number',
name = 'Name',
type1 = 'Type1',
type2 = 'Type2',
ability1 = 'Ability 1',
ability2 = 'Ability 2',
ha = 'Hidden Ability',
generation = 'Generation',
hp = 'HP',
atk = 'Attack',
def = 'Defense',
spatk = 'Special Attack',
spdef = 'Special Defense',
spd = 'Speed',
evhp = 'EVHP',
evatk = 'EVATK',
evdef = 'EVDEF',
evspa = 'EVSPA',
evspd = 'EVSPD',
evsp = 'EVSP',
height = 'Height',
weight = 'Weight',
stone = 'Stone',
}
local MEGA_SPEC
do
local parts = {}
for key, header in pairs(MEGA_COLUMNS) do
parts[#parts + 1] = key .. '=' .. header
end
MEGA_SPEC = table.concat(parts, ',')
end
local megaCache = {}
local function megaByName(name)
if megaCache[name] == nil then
local ok, values = pcall(function()
return mw.ext.externalData.getFileData{
source = 'data',
['file name'] = 'RawMegaPokemonList.csv',
format = DATA_FORMAT,
filters = 'Name=' .. name,
data = MEGA_SPEC,
}
end)
if ok and values and values.name ~= nil then
megaCache[name] = values
else
megaCache[name] = false
end
end
return megaCache[name] or nil
end
-- Page/raw-list name -> base name the form candidates are derived from
-- ("Alolan Rattata" -> "Rattata", "Eiscue-Ice Face" -> "Eiscue").
local function formStem(name)
for _, form in ipairs(FORM_SUFFIXES) do
local head = form.label .. ' '
if mw.ustring.sub(name, 1, mw.ustring.len(head)) == head then
return mw.ustring.sub(name, mw.ustring.len(head) + 1)
end
end
for _, suffix in ipairs(STRIP_SUFFIXES) do
for _, sep in ipairs({ '-', ' ' }) do
local tail = sep .. suffix
if mw.ustring.sub(name, -mw.ustring.len(tail)) == tail then
return mw.ustring.sub(name, 1, -mw.ustring.len(tail) - 1)
end
end
end
return name
end
local function rawListFormName(form, stem)
if form.rawList == 'suffix' then
return stem .. '-' .. form.suffix
elseif form.rawList == 'prefix-hyphen' then
return form.label .. '-' .. stem
elseif form.rawList == 'suffix-label' then
return stem .. '-' .. form.label
elseif form.rawList == 'suffix-space' then
return stem .. ' ' .. form.label
end
return form.label .. ' ' .. stem
end
local function typeColors(frame, pokeType)
return {
light = frame:expandTemplate{ title = pokeType .. ' color light' },
base = frame:expandTemplate{ title = pokeType .. ' color' },
dark = frame:expandTemplate{ title = pokeType .. ' color dark' },
}
end
local statColorCache = {}
local function statColors(frame, statTemplate)
if not statColorCache[statTemplate] then
statColorCache[statTemplate] = {
base = frame:expandTemplate{ title = statTemplate .. ' color' },
dark = frame:expandTemplate{ title = statTemplate .. ' color dark' },
}
end
return statColorCache[statTemplate]
end
local function navTableHtml(frame, data, colors, pageName)
local number = tonumber(data.number)
local prevCell = ''
if number and number > 1 then
local prev = byNumber(number - 1)
if prev then
prevCell = ' [[File:' .. prev.number .. 'Icon.png|' .. prev.name .. '|link=' .. prev.name ..
']] #' .. padNumber(prev.number) .. ' [[' .. prev.name .. ']]'
end
end
local nextCell = ''
if number then
local nxt = byNumber(number + 1)
if nxt then
nextCell = '[[' .. nxt.name .. ']] #' .. padNumber(nxt.number) .. ' [[File:' ..
nxt.number .. 'Icon.png|' .. nxt.name .. '|link=' .. nxt.name .. ']]'
end
end
return '<table style="margin-left: auto; margin-right: auto; margin-bottom: 15px; height: 60px; ' ..
'border: 4px #' .. colors.dark .. ' solid; font-weight: bold; width: 65%; color: #' .. colors.dark ..
'; border-radius: 20px; padding: 3px; background: #' .. colors.base .. ';">' ..
'<tr>' ..
'<td style="text-align: left; width: 36%; background: #' .. colors.light ..
'; border-radius: 15px 1px 1px 15px; border: 1px solid #' .. colors.dark .. ';">' .. prevCell .. '</td>' ..
'<td style="text-align: center; width: 28%; background: #' .. colors.light ..
'; border: 1px solid #' .. colors.dark .. ';">#' .. padNumber(number) .. ' - ' ..
pageName .. '</td>' ..
'<td style="text-align: right; width: 36%; padding-right: 10px; background: #' .. colors.light ..
'; border: 1px solid #' .. colors.dark .. '; border-radius: 1px 15px 15px 1px;">' .. nextCell .. '</td>' ..
'</tr></table>'
end
function p.navTable(frame)
local args = templateArgs(frame)
local pageName = mw.title.getCurrentTitle().text
local name = nonEmpty(args.Name) or pageName
local data = byName(name)
if not data then
return '<strong class="error">PokeNavTable: no data found for "' .. name .. '"</strong>'
end
return navTableHtml(frame, data, typeColors(frame, data.type1), pageName)
end
local function infoCell(colors, labelHtml, valueHtml)
return '<td colspan="3" style="width: 50%; border: 1px solid #' .. colors.dark .. '; background-color: #' .. colors.light ..
'; border-radius: 15px; padding: 5px;">' ..
'<p style="margin-bottom: 7px; font-weight: bold; color: #' .. colors.dark .. ';">' .. labelHtml .. '</p>' ..
'<span style="border-radius: 12px; padding: 4px 10px; border: 1px solid #' .. colors.dark ..
'; background-color: #FFFFFF;"> ' .. valueHtml .. ' </span></td>'
end
local function abilityCell(colors, label, ability, widthPercent)
if not ability then return '' end
local style = 'border-radius: 12px; padding: 4px 10px; border: 1px solid #' .. colors.dark ..
'; background-color: #FFFFFF;'
if widthPercent then
style = 'width: ' .. widthPercent .. '%; ' .. style
end
return '<td style="' .. style .. '"><b>' .. label .. '</b><br>[[' .. ability .. '_(ability)|' .. ability ..
']]</td>'
end
local function abilitiesRow(colors, data)
local entries = {
{ 'Ability 1', nonEmpty(data.ability1) },
{ 'Ability 2', nonEmpty(data.ability2) },
{ 'Hidden Ability', nonEmpty(data.ha) },
}
local present = {}
for _, entry in ipairs(entries) do
if entry[2] then present[#present + 1] = entry end
end
local widthPercent = (#present > 0) and string.format('%.2f', 100 / #present) or nil
local cells = {}
for _, entry in ipairs(present) do
cells[#cells + 1] = abilityCell(colors, entry[1], entry[2], widthPercent)
end
return table.concat(cells)
end
local function statValueCell(frame, stat, value)
local colors = statColors(frame, stat.template)
return '<td style="width: 16.66%; border: 1px solid #' .. colors.dark .. '; background-color: #' .. colors.base ..
'; border-radius: 12px; padding: 4px; font-size: 11px;">\'\'\'' .. stat.label .. '\'\'\'<br>' ..
value .. '</td>'
end
-- One horizontal stat bar, scaled against the stat ceiling of 255.
-- The fill is colored by the value (red -> yellow -> green -> teal); the
-- label keeps the stat's own color.
local function statBarRow(frame, stat, value)
local labelColors = statColors(frame, stat.template)
local pct = math.floor(value / 255 * 100 + 0.5)
if pct < 2 then pct = 2 end
if pct > 100 then pct = 100 end
local hue = math.floor(math.max(0, math.min(180, value * 180 / 255)) + 0.5)
local fill = 'hsl(' .. hue .. ', 85%, 45%)'
return '<tr>' ..
'<td style="text-align: right; padding: 2px 8px 2px 4px; font-weight: bold; font-size: 12px; ' ..
'color: #' .. labelColors.dark .. '; white-space: nowrap;">' .. stat.label .. '</td>' ..
'<td style="width: 100%; padding: 2px 0;"><div style="background-color: #FFFFFF; border: 1px solid ' ..
'#999999; border-radius: 8px; height: 13px; overflow: hidden;">' ..
'<div style="width: ' .. pct .. '%; height: 100%; background-color: ' .. fill .. ';"></div>' ..
'</div></td>' ..
'<td style="text-align: left; padding: 2px 4px 2px 8px; font-weight: bold; font-size: 12px; ' ..
'width: 34px;">' .. value .. '</td></tr>'
end
local MALE_COLOR = '3B6BF5'
local FEMALE_COLOR = 'F5629C'
local function genderRatioHtml(male)
if male == 'Genderless' then
return '<span style="display: inline-block; color: #000000; background-color: #FFFFFF; ' ..
'border-radius: 12px; border: 1px solid #000000; padding: 3px 12px; font-weight: bold;">' ..
'Genderless</span>'
end
-- ExternalData drops cells whose value is '0' (PHP empty('0') is true),
-- so a nil/blank ratio on an otherwise-complete row means 0% male.
local m = tonumber(male)
if m == nil then
m = nonEmpty(male) and 100 or 0
end
if m < 0 then m = 0 end
if m > 100 then m = 100 end
local f = 100 - m
local segments = ''
if m > 0 then
segments = segments .. '<div style="width: ' .. m .. '%; background-color: #' .. MALE_COLOR .. ';"></div>'
end
if f > 0 then
segments = segments .. '<div style="width: ' .. f .. '%; background-color: #' .. FEMALE_COLOR .. ';"></div>'
end
local labels = {}
if m > 0 then
labels[#labels + 1] = '<span style="color: #' .. MALE_COLOR .. '; font-weight: bold;">' .. m .. '% ♂</span>'
end
if f > 0 then
labels[#labels + 1] = '<span style="color: #' .. FEMALE_COLOR .. '; font-weight: bold;">' .. f .. '% ♀</span>'
end
return '<div style="width: 85%; margin: 4px auto 0;">' ..
'<div style="display: flex; height: 12px; border-radius: 8px; overflow: hidden; ' ..
'border: 1px solid #555555; background-color: #FFFFFF;">' .. segments .. '</div>' ..
'<div style="margin-top: 4px; font-size: 12px;">' .. table.concat(labels, ' · ') ..
'</div></div>'
end
local function categories(data, gen, bstCategory, args)
local cats = { '[[Category:Pokémon species]]' }
if gen then
cats[#cats + 1] = '[[Category:Generation ' .. gen .. ' Pokémon]]'
end
cats[#cats + 1] = '[[Category:' .. data.type1 .. '-type Pokémon]]'
if nonEmpty(data.type2) then
cats[#cats + 1] = '[[Category:' .. data.type2 .. '-type Pokémon]]'
end
cats[#cats + 1] = '[[Category:' .. data.rarity .. ' Pokémon]]'
if nonEmpty(data.npc) then
cats[#cats + 1] = '[[Category:NPC-obtainable Pokémon]]'
end
if not nonEmpty(args.EvolvesFrom) then
cats[#cats + 1] = '[[Category:Base-stage Pokémon]]'
end
cats[#cats + 1] = '[[Category:' .. bstCategory .. '-plus-BST Pokémon]]'
return table.concat(cats, ' ')
end
local function infoboxHtml(frame, args, data, displayName, width, pageName, includeNav)
local colors = typeColors(frame, data.type1)
-- Regional forms carry values like '1A'/'5A'; use the leading digits.
local gen = GEN_NUMERALS[tonumber(mw.ustring.match(tostring(data.generation or ''), '^%d+') or '')]
-- Types pill(s).
local typesHtml
if nonEmpty(data.type2) then
local type2Colors = typeColors(frame, data.type2)
typesHtml =
'<td style="width: 50%; background-color: #' .. colors.dark .. '; border-radius: 25px 1px 1px 25px;">' ..
'[[' .. data.type1 .. '_(type)|<span style="color: #FFFFFF;">' .. data.type1 .. '</span>]]</td>' ..
'<td style="width: 50%; background-color: #' .. type2Colors.dark .. '; border-radius: 1px 25px 25px 1px;">' ..
'[[' .. data.type2 .. '_(type)|<span style="color: #FFFFFF;">' .. data.type2 .. '</span>]]</td>'
else
typesHtml =
'<td style="width:100%; background-color: #' .. colors.dark .. '; border-radius: 25px;">' ..
'[[' .. data.type1 .. '_(type)|<span style="color: #FFFFFF;">' .. data.type1 .. '</span>]]</td>'
end
-- Base stats
local statRows, bst = {}, 0
for _, stat in ipairs(STATS) do
local value = tonumber(data[stat.key]) or 0
bst = bst + value
statRows[#statRows + 1] = statBarRow(frame, stat, value)
end
local bstCategory = math.floor(bst / 100 + 0.5) * 100
-- EV yield.
local evCells, evTotal = {}, 0
for i, stat in ipairs(STATS) do
local ev = tonumber(data[EV_KEYS[i]]) or 0
evTotal = evTotal + ev
evCells[#evCells + 1] = statValueCell(frame, stat, ev)
end
-- Catch rate.
local catchRaw = tonumber(data.catch) or 0
local catchPercent = string.format('%.2f', catchRaw / 255 / 3 * 100)
local buf = {}
if includeNav then
buf[#buf + 1] = navTableHtml(frame, data, colors, pageName)
end
buf[#buf + 1] = '<table align="right" class="informational-box" cellpadding="4" cellspacing="4" ' ..
'style="width: ' .. width .. '%; border: 4px solid #' .. colors.dark .. '; background-color: #' ..
colors.base .. '; padding: 4px; border-radius: 15px;">'
-- Header: dex number + icon, name.
buf[#buf + 1] = '<tr><td colspan="3" style="width: 50%; border: 1px solid #' .. colors.dark .. '; background-color: #' ..
colors.light .. '; border-radius: 15px;"><span style="margin-bottom: 7px; font-weight: bold; color:#' ..
colors.dark .. ';">#' .. padNumber(data.number) .. '</span> [[File:' .. data.number ..
'Icon.png]]</td><td colspan="3" style="width: 50%; border: 1px solid #' .. colors.dark .. '; background-color: #' ..
colors.light .. '; border-radius: 15px;"><span style="margin-bottom: 7px; font-weight: bold; color:#' ..
colors.dark .. ';">' .. displayName .. '</span></td></tr>'
-- Artwork.
buf[#buf + 1] = '<tr><td colspan="6" style="border: 1px solid #' .. colors.dark .. '; background-color: #' ..
colors.light .. '; border-radius: 15px; padding: 5px;"><div style="border-radius: 15px; ' ..
'text-align: center; padding: 5px; background-color: #FFFFFF;">[[File:' .. data.number .. data.name ..
'.png|150px]]</div></td></tr>'
-- Types + Generation.
local genHtml = '—'
if gen then
genHtml = '[[:Category:Generation ' .. gen .. ' Pokémon|<span style="color:#' ..
colors.dark .. ';">' .. gen .. '</span>]]'
end
buf[#buf + 1] = '<tr><td colspan="3" style="width: 50%; border: 1px solid #' .. colors.dark .. '; background-color: #' ..
colors.light .. '; padding: 11px; border-radius: 15px;"><p style="margin-bottom: 7px; font-weight: bold; ' ..
'color: #' .. colors.dark .. ';">Types</p><table align="center" style="width: 75%; font-weight: bold; ' ..
'padding: 3px;"><tr>' .. typesHtml .. '</tr></table></td>' ..
'<td colspan="3" style="width: 50%; border: 1px solid #' .. colors.dark .. '; background-color: #' .. colors.light ..
'; border-radius: 15px;"><p style="margin-bottom: 7px; font-weight: bold; color: #' .. colors.dark ..
';">Generation</p><span style="border-radius: 12px; padding: 4px 10px; border: 1px solid #' .. colors.dark ..
'; background-color: #FFFFFF;"> ' .. genHtml .. ' </span></td></tr>'
-- Abilities.
buf[#buf + 1] = '<tr><td colspan="6" style="border: 1px solid #' .. colors.dark .. '; background-color: #' ..
colors.light .. '; padding: 11px; border-radius: 15px; width: 100%;"><table width="100%">' ..
'<tr><th colspan="6"><p style="margin-bottom: 7px; font-weight: bold; color:#' .. colors.dark ..
';">[[Abilities|<span style="color:#' .. colors.dark .. ';">Abilities</span>]]</p></th></tr>' ..
'<tr>' .. abilitiesRow(colors, data) .. '</tr></table></td></tr>'
-- Rarity + Catch rate.
buf[#buf + 1] = '<tr>' ..
infoCell(colors, '[[List of Pokémon by Rarity Tier|<span style="color:#' .. colors.dark ..
';">Rarity Tier</span>]]', '[[:Category:' .. data.rarity .. ' Pokémon|<span style="color:#' ..
colors.dark .. ';">' .. data.rarity .. '</span>]]') ..
infoCell(colors, '[[Catch Rates|<span style="color:#' .. colors.dark .. ';">Catch rate</span>]]',
data.catch .. '/255 (' .. catchPercent .. '%)') ..
'</tr>'
-- Base stats.
buf[#buf + 1] = '<tr><td colspan="6" style="border: 1px solid #' .. colors.dark .. '; background-color: #' ..
colors.light .. '; border-radius: 15px; padding: 8px; width: 100%">' ..
'<table width="100%" style="border-collapse: collapse;">' ..
'<tr><th colspan="3"><p style="margin-bottom: 7px; font-weight: bold; color: #' .. colors.dark ..
';">[[Base stats|<span style="color: #' .. colors.dark .. ';">Base stats</span>]]</p></th></tr>' ..
table.concat(statRows) ..
'<tr><td colspan="3" style="text-align: center; padding-top: 6px;">' ..
'<div style="display: inline-block; background-color: #FFFFFF; border: 1px solid #' .. colors.dark ..
'; padding: 4px 14px; border-radius: 12px;">Base-stat total: ' .. bst .. '</div></td></tr>' ..
'</table></td></tr>'
-- EV yield.
buf[#buf + 1] = '<tr><td colspan="6" style="border: 1px solid #' .. colors.dark .. '; background-color: #' ..
colors.light .. '; border-radius: 15px; padding: 5px; width: 100%"><table width="100%">' ..
'<tr><th colspan="6">[[EVs|<span style="margin-bottom: 7px; font-weight: bold; color: #' .. colors.dark ..
';">EV Yield</span>]]</th></tr><tr>' .. table.concat(evCells) .. '</tr>' ..
'<tr><td colspan="6"><div style="margin: auto; background-color:#FFFFFF; border: 1px solid #' ..
colors.dark .. '; padding: 4px 14px; border-radius: 12px; margin-bottom: 5px; width: 50%;">' ..
'EV-yield total: ' .. evTotal .. '</div></td></tr></table></td></tr>'
-- Base experience + Gender ratio.
buf[#buf + 1] = '<tr>' ..
infoCell(colors, '[[Experience System#Base Experience|<span style="color:#' .. colors.dark ..
';">Base Experience</span>]]', data.exp) ..
'<td colspan="3" style="width: 50%; border: 1px solid #' .. colors.dark .. '; background-color: #' .. colors.light ..
'; border-radius: 15px; padding: 5px;"><p style="margin-bottom: 7px; font-weight: bold; color: #' ..
colors.dark .. ';">Gender Ratio</p>' .. genderRatioHtml(data.male) .. '</td>' ..
'</tr>'
-- Height + Weight.
buf[#buf + 1] = '<tr>' ..
infoCell(colors, 'Height', data.height .. 'm') ..
infoCell(colors, 'Weight', data.weight .. 'kg') ..
'</tr>'
buf[#buf + 1] = '</table>'
buf[#buf + 1] = categories(data, gen, bstCategory, args)
return table.concat(buf)
end
-- Infobox for a Mega evolution row from RawMegaPokemonList.csv: named
-- "Mega <name>", artwork <number><name>-Mega.png, a Mega Stone row, and no
-- nav/catch/rarity/gender/experience sections.
local function megaInfoboxHtml(frame, data, width)
local colors = typeColors(frame, data.type1)
local gen = GEN_NUMERALS[tonumber(mw.ustring.match(tostring(data.generation or ''), '^%d+') or '')]
local displayName = 'Mega ' .. (data.name or '')
local typesHtml
if nonEmpty(data.type2) then
local type2Colors = typeColors(frame, data.type2)
typesHtml =
'<td style="width: 50%; background-color: #' .. colors.dark .. '; border-radius: 25px 1px 1px 25px;">' ..
'[[' .. data.type1 .. '_(type)|<span style="color: #FFFFFF;">' .. data.type1 .. '</span>]]</td>' ..
'<td style="width: 50%; background-color: #' .. type2Colors.dark .. '; border-radius: 1px 25px 25px 1px;">' ..
'[[' .. data.type2 .. '_(type)|<span style="color: #FFFFFF;">' .. data.type2 .. '</span>]]</td>'
else
typesHtml =
'<td style="width:100%; background-color: #' .. colors.dark .. '; border-radius: 25px;">' ..
'[[' .. data.type1 .. '_(type)|<span style="color: #FFFFFF;">' .. data.type1 .. '</span>]]</td>'
end
local statRows, bst = {}, 0
for _, stat in ipairs(STATS) do
local value = tonumber(data[stat.key]) or 0
bst = bst + value
statRows[#statRows + 1] = statBarRow(frame, stat, value)
end
local evCells, evTotal = {}, 0
for i, stat in ipairs(STATS) do
local ev = tonumber(data[EV_KEYS[i]]) or 0
evTotal = evTotal + ev
evCells[#evCells + 1] = statValueCell(frame, stat, ev)
end
local buf = {}
buf[#buf + 1] = '<table align="right" class="informational-box" cellpadding="4" cellspacing="4" ' ..
'style="width: ' .. width .. '%; border: 4px solid #' .. colors.dark .. '; background-color: #' ..
colors.base .. '; padding: 4px; border-radius: 15px;">'
-- Header: name across the whole top line.
buf[#buf + 1] = '<tr><td colspan="6" style="border: 1px solid #' .. colors.dark .. '; background-color: #' ..
colors.light .. '; border-radius: 15px;"><span style="margin-bottom: 7px; font-weight: bold; color:#' ..
colors.dark .. ';">' .. displayName .. '</span></td></tr>'
-- Artwork.
buf[#buf + 1] = '<tr><td colspan="6" style="border: 1px solid #' .. colors.dark .. '; background-color: #' ..
colors.light .. '; border-radius: 15px; padding: 5px;"><div style="border-radius: 15px; ' ..
'text-align: center; padding: 5px; background-color: #FFFFFF;">[[File:' .. (data.number or '') .. (data.name or '') ..
'-Mega.png|150px]]</div></td></tr>'
-- Types + Generation.
local genHtml = '—'
if gen then
genHtml = '[[:Category:Generation ' .. gen .. ' Pokémon|<span style="color:#' ..
colors.dark .. ';">' .. gen .. '</span>]]'
end
buf[#buf + 1] = '<tr><td colspan="3" style="width: 50%; border: 1px solid #' .. colors.dark .. '; background-color: #' ..
colors.light .. '; padding: 11px; border-radius: 15px;"><p style="margin-bottom: 7px; font-weight: bold; ' ..
'color: #' .. colors.dark .. ';">Types</p><table align="center" style="width: 75%; font-weight: bold; ' ..
'padding: 3px;"><tr>' .. typesHtml .. '</tr></table></td>' ..
'<td colspan="3" style="width: 50%; border: 1px solid #' .. colors.dark .. '; background-color: #' .. colors.light ..
'; border-radius: 15px;"><p style="margin-bottom: 7px; font-weight: bold; color: #' .. colors.dark ..
';">Generation</p><span style="border-radius: 12px; padding: 4px 10px; border: 1px solid #' .. colors.dark ..
'; background-color: #FFFFFF;"> ' .. genHtml .. ' </span></td></tr>'
-- Abilities.
buf[#buf + 1] = '<tr><td colspan="6" style="border: 1px solid #' .. colors.dark .. '; background-color: #' ..
colors.light .. '; padding: 11px; border-radius: 15px; width: 100%;"><table width="100%">' ..
'<tr><th colspan="6"><p style="margin-bottom: 7px; font-weight: bold; color:#' .. colors.dark ..
';">[[Abilities|<span style="color:#' .. colors.dark .. ';">Abilities</span>]]</p></th></tr>' ..
'<tr>' .. abilitiesRow(colors, data) .. '</tr></table></td></tr>'
-- Mega Stone, linked to its quest page ("Venusaurite (quest)").
local stone = nonEmpty(data.stone)
local stoneHtml = '—'
if stone then
stoneHtml = '[[' .. stone .. ' (quest)|<span style="color:#' .. colors.dark .. ';">' .. stone .. '</span>]]'
end
buf[#buf + 1] = '<tr><td colspan="6" style="border: 1px solid #' .. colors.dark .. '; background-color: #' ..
colors.light .. '; border-radius: 15px; padding: 5px;">' ..
'<p style="margin-bottom: 7px; font-weight: bold; color: #' .. colors.dark .. ';">Mega Stone</p>' ..
'<span style="border-radius: 12px; padding: 4px 10px; border: 1px solid #' .. colors.dark ..
'; background-color: #FFFFFF;"> ' .. stoneHtml .. ' </span></td></tr>'
-- Base stats.
buf[#buf + 1] = '<tr><td colspan="6" style="border: 1px solid #' .. colors.dark .. '; background-color: #' ..
colors.light .. '; border-radius: 15px; padding: 8px; width: 100%">' ..
'<table width="100%" style="border-collapse: collapse;">' ..
'<tr><th colspan="3"><p style="margin-bottom: 7px; font-weight: bold; color: #' .. colors.dark ..
';">[[Base stats|<span style="color: #' .. colors.dark .. ';">Base stats</span>]]</p></th></tr>' ..
table.concat(statRows) ..
'<tr><td colspan="3" style="text-align: center; padding-top: 6px;">' ..
'<div style="display: inline-block; background-color: #FFFFFF; border: 1px solid #' .. colors.dark ..
'; padding: 4px 14px; border-radius: 12px;">Base-stat total: ' .. bst .. '</div></td></tr>' ..
'</table></td></tr>'
-- EV yield.
buf[#buf + 1] = '<tr><td colspan="6" style="border: 1px solid #' .. colors.dark .. '; background-color: #' ..
colors.light .. '; border-radius: 15px; padding: 5px; width: 100%"><table width="100%">' ..
'<tr><th colspan="6">[[EVs|<span style="margin-bottom: 7px; font-weight: bold; color: #' .. colors.dark ..
';">EV Yield</span>]]</th></tr><tr>' .. table.concat(evCells) .. '</tr>' ..
'<tr><td colspan="6"><div style="margin: auto; background-color:#FFFFFF; border: 1px solid #' ..
colors.dark .. '; padding: 4px 14px; border-radius: 12px; margin-bottom: 5px; width: 50%;">' ..
'EV-yield total: ' .. evTotal .. '</div></td></tr></table></td></tr>'
-- Height + Weight.
buf[#buf + 1] = '<tr>' ..
infoCell(colors, 'Height', (data.height or '—') .. 'm') ..
infoCell(colors, 'Weight', (data.weight or '—') .. 'kg') ..
'</tr>'
buf[#buf + 1] = '</table>'
buf[#buf + 1] = '[[Category:Mega-Evolvable Pokémon]]'
return table.concat(buf)
end
function p.infobox(frame)
local args = templateArgs(frame)
local pageName = mw.title.getCurrentTitle().text
local name = nonEmpty(args.Name) or pageName
local width = nonEmpty(args.Width) or '45'
local data = byName(name)
if not data then
return '<strong class="error">PokemonInfo: no data found for "' .. name .. '"</strong>'
end
-- Downstream templates ({{Tabs:MultiformTab}} etc.) color themselves via
-- {{#var:type}}, which the old wikitext infobox used to define.
frame:callParserFunction('#vardefine', 'type', data.type1 or 'Normal')
return infoboxHtml(frame, args, data, name, width, pageName, true)
end
-- Returns the page Pokémon's primary type as plain text, for use in template
-- wikitext: {{#vardefine:type|{{#invoke:Pokemon|primaryType}}}}. Variables
-- defined from inside Lua do not persist on this wiki, so the template layer
-- has to do the #vardefine itself.
function p.primaryType(frame)
local args = templateArgs(frame)
local name = nonEmpty(args.Name) or mw.title.getCurrentTitle().text
local data = byName(name) or byName(formStem(name))
return (data and nonEmpty(data.type1)) or 'Normal'
end
-- Name/index plumbing for the template-side tab shell: the index arrives as
-- the invoke's own first argument, while Name comes from the surrounding
-- template call.
local function requestedName(frame)
local parent = frame:getParent()
return nonEmpty(frame.args.Name)
or (parent and nonEmpty(parent.args.Name))
or mw.title.getCurrentTitle().text
end
local participantsCache = {}
-- All forms of the page's Pokémon, base first. parts[1].label is 'Normal'
-- unless a matched form dictates a baseLabel ('Male').
local function computeParticipants(frame, name)
if participantsCache[name] ~= nil then
return participantsCache[name] or nil
end
local rows = loadAllRows()
local function rowFor(n)
return rows[n] or rows[hyphenSpaceVariant(n)]
end
local stem = formStem(name)
local baseData = rowFor(stem) or rowFor(name)
if not baseData then
participantsCache[name] = false
return nil
end
local parts = { { data = baseData, label = 'Normal' } }
local seen = { [baseData.name] = true }
for _, form in ipairs(FORM_SUFFIXES) do
local row = rowFor(rawListFormName(form, stem))
if row and not seen[row.name] then
seen[row.name] = true
parts[#parts + 1] = { data = row, label = form.label }
if form.baseLabel then
parts[1].label = form.baseLabel
end
end
end
for _, variant in ipairs(MEGA_VARIANTS) do
local row = megaByName(stem .. variant)
if row then
parts[#parts + 1] = { data = row, label = 'Mega' .. variant, mega = true }
end
end
participantsCache[name] = parts
return parts
end
-- {{#invoke:Pokemon|formLabel|N}}: tab label of the Nth participant
-- (0 = base form, 1.. = extra forms); empty when there is no such form or the
-- Pokémon has no extra forms at all.
function p.formLabel(frame)
local idx = tonumber(frame.args[1]) or 0
local parts = computeParticipants(frame, requestedName(frame))
if not parts then
return ''
end
local part = parts[idx + 1]
return part and part.label or ''
end
-- {{#invoke:Pokemon|rawName}}: the page Pokémon's raw-list name, for the
-- template-side {{#get_file_data:}} filter that feeds downstream
-- {{#external_value:}} consumers.
function p.rawName(frame)
local name = requestedName(frame)
local parts = computeParticipants(frame, name)
return (parts and parts[1].data.name) or name
end
-- {{#invoke:Pokemon|formBox|N}}: the Nth participant's infobox HTML
-- (0 = base form), without the nav table. Width defaults to 100 (inside a
-- tab cell); pass Width=45 for the untabbed fallback.
function p.formBox(frame)
local idx = tonumber(frame.args[1]) or 0
local width = nonEmpty(frame.args.Width) or '100'
local name = requestedName(frame)
local parts = computeParticipants(frame, name)
if not parts then
return '<strong class="error">PokemonInfo: no data found for "' .. name .. '"</strong>'
end
local part = parts[idx + 1]
if not part then
return ''
end
if part.mega then
return megaInfoboxHtml(frame, part.data, width)
end
local parent = frame:getParent()
local args = (parent and parent.args) or frame.args
return infoboxHtml(frame, args, part.data, part.data.name, width, mw.title.getCurrentTitle().text, false)
end
-- Temporary diagnostic: {{#invoke:Pokemon|debugForms|Name=Meowth}}
function p.debugForms(frame)
local args = templateArgs(frame)
local name = nonEmpty(args.Name) or mw.title.getCurrentTitle().text
local rows = loadAllRows()
local count = 0
for _ in pairs(rows) do count = count + 1 end
local stem = formStem(name)
local out = { 'name=' .. name, 'stem=' .. stem, 'rows loaded=' .. count }
for _, form in ipairs(FORM_SUFFIXES) do
local candidate = rawListFormName(form, stem)
if rows[candidate] or rows[hyphenSpaceVariant(candidate)] then
out[#out + 1] = 'form found: ' .. form.label .. ' -> ' .. candidate
end
end
for _, variant in ipairs(MEGA_VARIANTS) do
if megaByName(stem .. variant) then
out[#out + 1] = 'mega found: Mega' .. variant .. ' -> ' .. stem .. variant
end
end
return table.concat(out, '; ')
end
-- Like infobox, but detects alternate forms via Module:PokemonForms and, when
-- any exist, renders one nav table plus a {{TabbedInfoBoxes}} block with one
-- infobox per form. Falls back to the plain single infobox otherwise.
function p.infoboxes(frame)
local args = templateArgs(frame)
local pageName = mw.title.getCurrentTitle().text
local name = nonEmpty(args.Name) or pageName
local rows = loadAllRows()
local function rowFor(n)
return rows[n] or rows[hyphenSpaceVariant(n)]
end
local stem = formStem(name)
local baseData = rowFor(stem) or rowFor(name)
if not baseData then
return '<strong class="error">PokemonInfo: no data found for "' .. name .. '"</strong>'
end
local baseName = baseData.name
-- Downstream templates ({{Tabs:MultiformTab}} etc.) color themselves via
-- {{#var:type}}, which the old wikitext infobox used to define.
frame:callParserFunction('#vardefine', 'type', baseData.type1 or 'Normal')
local forms = {}
local seen = { [baseName] = true }
for _, form in ipairs(FORM_SUFFIXES) do
local row = rowFor(rawListFormName(form, stem))
if row and not seen[row.name] then
seen[row.name] = true
forms[#forms + 1] = { label = form.label, data = row, baseLabel = form.baseLabel }
end
end
if #forms == 0 then
local width = nonEmpty(args.Width) or '45'
return infoboxHtml(frame, args, baseData, baseName, width, pageName, true)
end
local baseLabel = 'Normal'
for _, f in ipairs(forms) do
if f.baseLabel then
baseLabel = f.baseLabel
break
end
end
-- Native tab shell replicating {{TabbedInfoBoxes}} + {{Tabs:MultiformTab}} +
-- {{Tabs:MultiformTable}} without Extension:Variables, whose page variables
-- are not visible inside Scribunto expansions on this wiki. The onclick div
-- and the name-attributed td must go through {{#tag:html}}, as the original
-- templates do.
local shellColors = typeColors(frame, baseData.type1)
local shellWidth = nonEmpty(args.Width) or '45'
-- The {{#tag:html}} bits live in Template:PokemonTabHeader and
-- Template:PokemonTabContent: Extension:NamespaceHTML fatals when the tag
-- reaches it directly from Scribunto (callParserFunction or a top-level
-- preprocess string), but survives inside a transcluded template body —
-- the path {{Tabs:MultiformTab}} has always used.
local tabHeaders, boxes = {}, {}
local function addTab(tabName, label, boxHtml)
tabHeaders[#tabHeaders + 1] = frame:expandTemplate{
title = 'PokemonTabHeader',
args = { Tab = tabName, Label = label, Dark = shellColors.dark, Light = shellColors.light },
}
boxes[#boxes + 1] = frame:expandTemplate{
title = 'PokemonTabContent',
args = { Tab = tabName, Content = boxHtml },
}
end
addTab('Default', baseLabel, infoboxHtml(frame, args, baseData, baseName, '100', pageName, false))
for _, f in ipairs(forms) do
addTab(f.label, f.label, infoboxHtml(frame, args, f.data, f.data.name, '100', pageName, false))
end
return navTableHtml(frame, baseData, shellColors, pageName) ..
'<table class="informational-box" style="width: ' .. shellWidth .. '%;">' ..
'<tr class="tab" style="color: #' .. shellColors.dark .. ';">' .. table.concat(tabHeaders) .. '</tr>' ..
'<tr>' .. table.concat(boxes) .. '</tr></table>'
end
return p