Module:WeaponData

From Zero-K
Revision as of 03:18, 11 May 2024 by Histidine (talk | contribs)
Jump to navigation Jump to search
Documentation

This module is used to autogenerate the weapon infoboxes contained in unit infoboxes, by drawing data from Module:WeaponData/data. The intent is to ease updating of unit infoboxes on the wiki; ideally, only the one data page needing to be changed every release. The concept is taken from e.g. the Combat Card Data module on the Library of Ruina Wiki.

Unlike Module:UnitData, there is no template for this module. If it is desired to generate a weapon infobox outside its unit infobox, the module should be invoked directly, e.g.:

{{#invoke:WeaponData|printInfobox|defname=striderdetriment_gauss|name=Name override}}

Data page

Module:WeaponData/data is a central store of data used by Module:WeaponData to automatically populate weapon infoboxes.

The page is a Lua table written to the local file temp/weaponStats.lua by the Wiki Data Export (dbg_wiki_export.lua) widget in Zero-K. This widget should be run once each update and the data page on the wiki replaced accordingly. It should not be necessary to edit the data manually.


local getArgs = require('Module:Arguments').getArgs
local weapon = {}
local weaponData = mw.loadData('Module:WeaponData/data')

local function writeTemplateLine(key, value)
	if not key or not value then
		return ""
	end
	local str = "\n"
	str = str .. "| " .. key .. " = " .. value
	return str
end

-- args: key in a key-value pair (nil if not a pair); value to display; number of extra rows used in the infobox thus far; infobox string thus far
-- return values: appended string, number of rows (now incremented by one), infobox string with new text appended
local function writeCustomDataLine(key, value, count, currentStr)
	currentStr = currentStr or ''
	local append = ""
	if key then
		append = writeTemplateLine("customlabel"..count, key) .. writeTemplateLine("customdata"..count, value)
	else
		append = writeTemplateLine("special"..count, value)
	end
	append = append .. 'extra junk'
	currentStr = currentStr .. append
	count = count + 1
	return append, count, currentStr
end


function weapon.getData(frame)
	if not frame then return '' end
    local weaponDefName = frame.args[1]
    local wd = weaponData[weaponDefName]
    if not (wd) then return 'weapondef ' .. weaponDefName .. ' not found' end
    local property = frame.args[2]
    local result = wd[property]
    if not result then result = '' end
    return frame:preprocess(result) or ''
end

function weapon.printInfobox(frame)
	local weaponDefName = frame.args[1]
    if not (weaponData[weaponDefName]) then return 'weapondef ' .. weaponDefName .. ' not found' end
	return [[{{User:Histidine/Autoinfobox zkweapon
|defname=]]..weaponDefName..[[}}]]
end

function weapon.printInfoboxCustomRows(frame)
	local weaponDefName = frame.args[1]
    local wd = weaponData[weaponDefName]
    if not wd then return 'weapondef ' .. weaponDefName .. ' not found' end
    if not wd.extraData then return '' end
	
	local str = ''
	local count = 1
	for i=1, #wd.extraData do
		local entry = wd.extraData[i]
		local append = ''
		if type(entry) == 'table' then
			if entry.type == 'shielddrain' then
				append, count, str = writeCustomDataLine("Shield drain (HP/shot)", entry.drain, count, str)				
			elseif entry.type == 'needs_link' then
				append, count, str = writeCustomDataLine("Grid needed", entry.power, count, str)				
			elseif entry.type == 'spawn' then
				append, count, str = writeCustomDataLine("Spawns Unit", "[[" .. entry.name .. "]]", count, str)
				append, count, str = writeCustomDataLine("Spawn Life (s)", entry.expire, count, str)				
			elseif entry.type == 'areadamage' then			
				local typeName = entry.grav and "Gravity Well" or "Ground Burn"
				if not grav then
					append, count, str = writeCustomDataLine(typeName .. " DPS", entry.dps, count, str)
				end
				append, count, str = writeCustomDataLine(typeName .. " radius (elmo)", entry.radius, count, str)
				append, count, str = writeCustomDataLine(typeName .. " duration (s)", entry.duration, count, str)
				
			elseif entry.type == 'stockpile' then
				append, count, str = writeCustomDataLine("Stockpile time (s)", entry.time, count, str)
				if (entry.cost) then
					append, count, str = writeCustomDataLine("Stockpile cost (M)", entry.cost, count, str)
				end
			end
		else
			if entry == 'hitscan' then
				append, count, str = writeCustomDataLine(nil, "Instantly hits", count, str)
			elseif entry == 'ignoreshield' then
				append, count, str = writeCustomDataLine(nil, "Ignores shields", count, str)
			elseif entry == 'smoothsground' then
				append, count, str = writeCustomDataLine(nil, "Smooths ground", count, str)
			elseif entry == 'hightraj' then
				append, count, str = writeCustomDataLine(nil, "High trajectory", count, str)
			elseif entry == 'trajtoggle' then
				append, count, str = writeCustomDataLine(nil, "Trajectory toggle", count, str)
			elseif entry == 'watercapable' then
				append, count, str = writeCustomDataLine(nil, "Water capable", count, str)
			elseif entry == 'friendlyfire' then
				append, count, str = writeCustomDataLine(nil, "Potential friendly fire", count, str)
			elseif entry == 'nofriendlyfire' then
				append, count, str = writeCustomDataLine(nil, "No friendly fire", count, str)
			elseif entry == 'nogroundcollide' then
				append, count, str = writeCustomDataLine(nil, "Passes through ground", count, str)
			elseif entry == 'piercing' then
				append, count, str = writeCustomDataLine(nil, "Damage increase vs large units", count, str)
			elseif entry == 'damagefalloff' then
				append, count, str = writeCustomDataLine(nil, "Damage falls off with range", count, str)
			elseif entry == 'inaccuratevsmoving' then
				append, count, str = writeCustomDataLine(nil, "Inaccuracy vs moving targets", count, str)
			elseif entry == 'interceptedbyantinuke' then
				append, count, str = writeCustomDataLine(nil, "Can be shot down by antinukes", count, str)
			end
		end
	end
	return frame:preprocess(str) or ''
end

return weapon