Module:Sandbox/LVL: Difference between revisions

LVL (talk | contribs)
No edit summary
LVL (talk | contribs)
No edit summary
 
(124 intermediate revisions by the same user not shown)
Line 1: Line 1:
local p = {};
local p = {}
local heroes_data = mw.loadJsonData("Data:Sandbox.json")
 
local attributes_data = mw.loadJsonData("Data:AttributeData.json")
local heroes_data       = mw.loadJsonData("Data:HeroData.json")
local attribute_orders = mw.loadJsonData("Data:StatInfoboxOrder.json")
local attributes_data   = mw.loadJsonData("Data:AttributeData.json")
local util_module = require('Module:Utilities')
local attribute_orders = mw.loadJsonData("Data:StatInfoboxOrder.json")
local lang_module = require('Module:Lang')
local util_module       = require('Module:Utilities')
local lang_module       = require('Module:Lang')
local dictionary_module = require('Module:Dictionary')
local dictionary_module = require('Module:Dictionary')
local attribute_module = require('Module:AttributeData')
local attribute_module = require('Module:AttributeData')
local hero_data_module  = require('Module:HeroData')


-- returns the table of a specific item, used by external modules
local function get_nested_stat_value(hero_data, stat_key)
function p.get_json_item(name)
    if hero_data[stat_key] ~= nil then
for i,v in pairs(heroes_data) do
        return hero_data[stat_key]
if (v["Name"] == name) then
    elseif hero_data.Weapon and hero_data.Weapon[stat_key] ~= nil then
return v
        return hero_data.Weapon[stat_key]
end
    elseif hero_data.Weapon and hero_data.Weapon.AltFire and hero_data.Weapon.AltFire[stat_key] ~= nil then
end
        return hero_data.Weapon.AltFire[stat_key]
return nil
    end
    return 0
end
end


-- Returns an array of item tables that have the same properties
local function localize(key, fallback)
-- @function  get_similar_items
    local result = lang_module.get_string(key)
-- @param      {string}
    if result == "" or result == nil then
-- @return    {array of tables}
        result = util_module.add_space_before_cap(fallback) ..
local function get_similar_items(property)
                mw.getCurrentFrame():expandTemplate{title = "MissingValveTranslationTooltip"}
local similarItems = {}
    end
for _, v in pairs(heroes_data) do
    return result
if (v[property] ~= nil) then
table.insert(similarItems, v)
end
end
return similarItems
end
end


-- returns the key of the specified hero's english name
-- =========================================================================
function p.get_hero_key(name)
--  DPS calculation helpers
for i,v in pairs(heroes_data) do
-- =========================================================================
if (v["Name"] == name) then
 
return i
local function calculate_dps(stats, dps_type)
end
    local rps = stats.RoundsPerSecond or 0
end
    if rps == 0 then return 0 end
return nil
 
end
    local bullets_per_shot = (stats.HitOnceAcrossAllBullets and 1) or (stats.BulletsPerShot or 1)
    local burst_count = stats.BulletsPerBurst or 1
    local cycle_time = 1 / rps
    local total_cycle_time = cycle_time * burst_count


    if total_cycle_time == 0 then return 0 end


--{{#invoke:HeroData|get_hero_var|HERO_NAME|STAT_NAME|sig_figs_or_localize}}--
     local base_damage = stats.BulletDamage or 0
--sig_figs optional for rounding floats
     if dps_type == 'burst' then
p.get_hero_var = function(frame)
        return base_damage * bullets_per_shot * burst_count / total_cycle_time
     local hero_name = frame.args[1]
    local hero_stat_key = frame.args[2]
    local sig_figs_or_localize = frame.args[3]
   
    local hero = heroes_data[hero_name]
     if(hero == nil) then hero = p.get_json_item(hero_name) end
    if(hero == nil) then return "Hero Not Found" end
   
    -- Check top-level first, then Weapon, then AltFire
    local var_value = hero[hero_stat_key]
    if var_value == nil and hero.Weapon then
        var_value = hero.Weapon[hero_stat_key]
     end
     end
     if var_value == nil and hero.Weapon and hero.Weapon.AltFire then
 
         var_value = hero.Weapon.AltFire[hero_stat_key]
    local clip_size = stats.ClipSize or 0
     if clip_size <= 0 then
         return base_damage * bullets_per_shot * burst_count / total_cycle_time
     end
     end
   
 
    if(var_value == nil) then return 0 end
     local reload_time
   
     if stats.ReloadSingle then
     --round
         reload_time = (stats.ReloadTime or 0) * clip_size
     if (sig_figs_or_localize ~= nil and tonumber(sig_figs_or_localize) ~= nil) then
     else
         var_value = util_module.round_to_sig_fig(var_value, sig_figs_or_localize)
        reload_time = stats.ReloadTime or 0
        if (var_value == nil) then return "get_hero_var() error with rounding" end
     end
   
    --localize
    if (sig_figs_or_localize == "true") then
        return lang_module.get_string(var_value)
     end
     end
    reload_time = reload_time + (stats.ReloadDelay or 0)


     return var_value
     local time_to_empty_clip = (clip_size / burst_count) * total_cycle_time
    local damage_from_clip = base_damage * bullets_per_shot * clip_size
    local total_time = time_to_empty_clip + reload_time
    if total_time == 0 then return 0 end
    return damage_from_clip / total_time
end
end


--{{#invoke:HeroData|get_alt_fire_var|HERO_NAME|STAT_NAME|sig_figs_or_localize}}--
-- Compute how much DPS changes when all relevant component scalings are applied.
p.get_alt_fire_var = function(frame)
-- Used for alt-fire rows where explicit DPS/SustainedDPS scaling keys are missing.
     local hero_name = frame.args[1]
local function compute_dps_scaling(hero_data, base_stats, dps_type, scaling_type)
     local stat_key = frame.args[2]
     local scaling_source = hero_data[scaling_type .. "Scaling"] or {}
     local sig_figs_or_localize = frame.args[3]
 
   
     local component_scalings = {}
     local hero = heroes_data[hero_name]
     local possible_keys = {
     if(hero == nil) then hero = p.get_json_item(hero_name) end
        "BulletDamage", "RoundsPerSecond", "ClipSize", "ReloadTime",
    if(hero == nil) then return "Hero Not Found" end
        "ReloadDelay", "BulletsPerShot", "BulletsPerBurst"
   
     }
    if not (hero.Weapon and hero.Weapon.AltFire) then
     for _, key in ipairs(possible_keys) do
         return "No Alt Fire"
        local alt_key = key .. "AltFire"
        if scaling_source[alt_key] then
            component_scalings[key] = scaling_source[alt_key]
         end
     end
     end
      
 
     local var_value = hero.Weapon.AltFire[stat_key]
     -- ClipSize is shared between fire modes; fall back to primary scaling
    if(var_value == nil) then return 0 end
     if base_stats.ClipSize and not component_scalings.ClipSize then
   
        local primary_clip_scale = scaling_source["ClipSize"]
    --round
        if primary_clip_scale then
    if (sig_figs_or_localize ~= nil and tonumber(sig_figs_or_localize) ~= nil) then
            component_scalings.ClipSize = primary_clip_scale
         var_value = util_module.round_to_sig_fig(var_value, sig_figs_or_localize)
         end
     end
     end
   
 
    --localize
     if next(component_scalings) == nil then
     if (sig_figs_or_localize == "true") then
         return 0
         return lang_module.get_string(var_value)
     end
     end


     return var_value
    local base_dps = calculate_dps(base_stats, dps_type)
    local scaled_stats = {}
    for k, v in pairs(base_stats) do scaled_stats[k] = v end
    for comp_key, scale_val in pairs(component_scalings) do
        if scaled_stats[comp_key] ~= nil then
            scaled_stats[comp_key] = scaled_stats[comp_key] + scale_val
        end
    end
    local scaled_dps = calculate_dps(scaled_stats, dps_type)
     return scaled_dps - base_dps
end
end


--{{#invoke:HeroData|get_list_elem|HERO_NAME|VAR|NUMBER|LOCALIZE}}
-- =========================================================================
p.get_list_elem = function(frame)
--  Main module function
local hero_name = frame.args[1]
-- =========================================================================
local var = frame.args[2]
 
local number_str = frame.args[3]
p.write_hero_comparison_table = function(frame)
local number_int = tonumber(number_str)
    local power_increases = tonumber(frame.args[1]) or 0
local localize = frame.args[4]
    local spirit_power    = tonumber(frame.args[2]) or 0
local hero = heroes_data[hero_name] --check if hero key is passed instead
    local max_power      = tonumber(frame.args[3]) or 25
if(hero == nil) then hero = p.get_json_item(hero_name) end --check if hero name is passed
    local max_spirit      = tonumber(frame.args[4]) or 500
if(hero == nil) then return "Hero Not Found" end --both invalid, error
 
if (hero == nil) then return "Hero " .. hero_name .. " not found" end
    local display_scaling_icons = (power_increases == 0 and spirit_power == 0)
local list = hero[var]
 
if (list == nil) then return "" end
    local body_str = ""
local element = list[number_int]
 
if (element == nil) then return "" end
    local stats_to_include = {
        Weapon = {
if localize=="true" then  
            "DPS", "SustainedDPS", "BulletDamage", "RoundsPerSecond", "FireRate",
element = lang_module.get_string(element)
            "ClipSize", "ReloadTime", "ReloadDelay", "ReloadSingle", "BulletsPerShot", "BulletsPerBurst",
end
            "BurstInterShotInterval", "LightMeleeDamage", "HeavyMeleeDamage",
            "BulletSpeed", "FalloffStartRange", "FalloffEndRange",
return element
            "CritDamageBonusPercent", "BulletRadius", "RoundsPerSecondAtMaxSpin", "SpinAcceleration", "SpinDeceleration"
end
        },
        Vitality = {
            "MaxHealth", "BaseHealthRegen", "BulletResist", "TechResist", "MeleeResist",
            "CritDamageReceivedPercent", "DebuffResist", "BulletLifesteal", "MaxMoveSpeed",
            "SprintSpeed", "StaminaCooldown", "Stamina", "GroundDashSpeed", "GravityChange"
        },
        Spirit = { "TechPower" }
    }
 
    -- Collect and sort heroes alphabetically, excluding disabled/in-development
    local sorted_heroes = {}
    for hero_key, hero_data in pairs(heroes_data) do
        if not hero_data["InDevelopment"] and not hero_data["IsDisabled"] then
            table.insert(sorted_heroes, {key = hero_key, data = hero_data})
        end
    end
    table.sort(sorted_heroes, function(a, b)
        local function get_sort_name(name)
            return (name:gsub("^The ", ""))
        end
        return get_sort_name(a.data["Name"] or a.key) < get_sort_name(b.data["Name"] or b.key)
    end)
 
    -- Build a single stat cell (<td>) with data attributes for JS recalculation
    local function buildStatCell(hero_data, hero_key, attr_key, is_alt_fire, category)
        -- Conversion factors for stats that need unit changes (e.g. metres → centimetres)
        local unit_conversion = {
            BulletRadius = 100  -- metres to centimetres
        }
        local conv_factor = unit_conversion[attr_key] or 1
 
        local base_value
        if is_alt_fire then
            if attr_key == "ClipSize" or attr_key == "ReloadTime" then
                base_value = get_nested_stat_value(hero_data, attr_key)
            elseif hero_data.Weapon and hero_data.Weapon.AltFire
                and hero_data.Weapon.AltFire[attr_key] ~= nil then
                base_value = hero_data.Weapon.AltFire[attr_key]
            else
                base_value = get_nested_stat_value(hero_data, attr_key)
            end
        else
            base_value = get_nested_stat_value(hero_data, attr_key)
        end


p.get_ability_key = function(frame)
        -- Apply unit conversion to the base value
local hero_key = frame.args[1]
        if type(base_value) == "number" then
local bound_slot_number = frame.args[2]
            base_value = base_value * conv_factor
local hero_data = heroes_data[hero_key]
        end
if (hero_data == nil) then return "Hero key "..hero_key.. " not found" end
local bound_abilities_data = hero_data["BoundAbilities"]
if (bound_abilities_data == nil) then return "Hero key " .. hero_key.. " has no BoundAbilities" end
return bound_abilities_data[tonumber(bound_slot_number)]["Key"]
end


p.write_role_playstyle_quote = function(frame)
        local stat_value = base_value
local hero_key = frame.args[1]
local hero_data = heroes_data[hero_key]
if (hero_data == nil) then return hero_key.." not found" end
local role_key = hero_data["Role"]
local role_localized = lang_module.get_string(role_key, nil, 'en')
local playstyle_key = hero_data["Playstyle"]
local playstyle_localized = lang_module.get_string(playstyle_key, nil, 'en')
local str = "<b>" .. role_localized .. '</b><br>' .. playstyle_localized
local template_args = {}
template_args[1] = ""
template_args[2] = str
return frame:expandTemplate{title = 'Quotation', args = template_args}
end


p.write_default_items = function(frame)
        -- Scaling lookup
local hero_key = frame.args[1] --unlocalized
        local scaling_data
if (hero_key == nil) then return "No hero key provided" end
        if is_alt_fire then
local str = ""
            if attr_key == "ClipSize" or attr_key == "ReloadTime" then
local hero = heroes_data[hero_key]
                scaling_data = hero_data_module.get_hero_scaling_data(hero_data, attr_key)
if (hero == nil) then return "Hero not found, must be unlocalized" end
            elseif (attr_key == "DPS" or attr_key == "SustainedDPS")
local template_title = 'PageRef'
                and hero_data.Weapon and hero_data.Weapon.AltFire then
for i, item_key in ipairs(hero["RecommendedItems"]) do
                -- Alt-fire DPS/SustainedDPS: use explicit key if present, otherwise compute from components
template_args = {}
                local direct_scaling = hero_data_module.get_hero_scaling_data(hero_data, attr_key .. "AltFire")
template_args[1] = lang_module.get_string(item_key, 'en')
                if direct_scaling and next(direct_scaling) ~= nil then
template_args['alt_name'] = localize(item_key, item_key)
                    scaling_data = direct_scaling
local expanded_template = mw.getCurrentFrame():expandTemplate{ title = template_title, args = template_args }
                else
str = str .. "* " .. expanded_template .. "\n"
                    local alt_stats = hero_data.Weapon.AltFire
end
                    local dps_type = attr_key == "DPS" and 'burst' or 'sustained'
return str
                    local level_scale = compute_dps_scaling(hero_data, alt_stats, dps_type, "Level")
end
                    local spirit_scale = compute_dps_scaling(hero_data, alt_stats, dps_type, "Spirit")
                    level_scale = util_module.round_to_sig_fig(level_scale, 5)
                    spirit_scale = util_module.round_to_sig_fig(spirit_scale, 5)
                    scaling_data = {}
                    if level_scale ~= 0 then scaling_data[level_scale] = "Level" end
                    if spirit_scale ~= 0 then scaling_data[spirit_scale] = "Spirit" end
                    if next(scaling_data) == nil then scaling_data = nil end
                end
            elseif hero_data.Weapon and hero_data.Weapon.AltFire
                and hero_data.Weapon.AltFire[attr_key] ~= nil then
                scaling_data = hero_data_module.get_hero_scaling_data(hero_data, attr_key .. "AltFire")
            else
                scaling_data = hero_data_module.get_hero_scaling_data(hero_data, attr_key)
            end
        else
            scaling_data = hero_data_module.get_hero_scaling_data(hero_data, attr_key)
        end


--If the hero scales with the stat, it returns {{Ss|value}} or {{Ls|value}}, else blank string
        local scaling_strs = ""
--{{#invoke:HeroData|get_hero_scalar|HERO_NAME|SCALING_TYPE|STAT_NAME|sig_figs_or_localize}}--
        local spirit_scale = 0
p.get_hero_scalar = function(frame)
        local level_scale  = 0
local hero_key = frame.args[1]
        local spirit_val = 0
local scaling_type = frame.args[2]
        local level_val  = 0
local hero_stat_key = frame.args[3]
        if scaling_data ~= nil then
local sig_figs_or_localize = frame.args[4]
            for scaling_val, scaling_type in pairs(scaling_data) do
                if scaling_type == "Spirit" then
local hero_data = heroes_data[hero_key]
                    spirit_val = scaling_val * conv_factor  -- apply unit conversion
if(hero_data == nil) then hero_data = p.get_json_item(hero_key) end
                elseif scaling_type == "Level" then
if(hero_data == nil) then return "Hero not found." end
                    level_val = scaling_val * conv_factor  -- apply unit conversion
                end
local scaling_data = hero_data[scaling_type .. "Scaling"]
            end
if scaling_data == nil then return scaling_type .. "Scaling data not found for hero" .. hero_key end
local scaling_value = scaling_data[hero_stat_key]
if scaling_value == nil then return scaling_type .. "Scaling data does not have key " .. hero_stat_key .. " for hero " .. hero_key end


--round
            -- Output Spirit icon before Level icon
if (sig_figs_or_localize ~= nil and tonumber(sig_figs_or_localize) ~= nil) then
            local function append_scaling(val, stype)
scaling_value = util_module.round_to_sig_fig(scaling_value, sig_figs_or_localize)
                if val ~= 0 then
if (scaling_value == nil) then return "get_hero_scalar() error with rounding" end
                    local scaling_str = hero_data_module.write_scalar_str(val, stype, true)
end
                    if scaling_str ~= "" then
                        scaling_strs = scaling_strs .. " " .. scaling_str
--localize
                    end
if (sig_figs_or_localize == "true") then
                end
return lang_module.get_string(scaling_value)
            end
end
            append_scaling(spirit_val, "Spirit")
            append_scaling(level_val, "Level")
local scaling_str = p.write_scalar_str(scaling_value, scaling_type)
if (scaling_str ~= "") then scaling_str = " " .. scaling_str end
return scaling_str
end


-- Retrieve scaling string of a hero's given stat, if it has scaling, else return blank
            if type(stat_value) == "number" then
-- Scaling string meaning the expanded template {{Ss|scalar}} or {{Ls|scalar}}
                stat_value = stat_value + (spirit_power * spirit_val)
function p.write_scalar_str(scaling_value, scaling_type)
                stat_value = stat_value + (power_increases * level_val)
local scaling_abbrevs = {Spirit = "Ss", Level = "PI"}
            end
-- Return blank if it doesnt scale
if (scaling_value == 0) then return "" end
-- Round it
scaling_value = util_module.round_to_sig_fig(scaling_value, 3)
--The hero has a scaling value with this stat
local template_title = "Template:" .. scaling_abbrevs[scaling_type] --scaling type's abbreviation
local template_args = {}
template_args["1"] = scaling_value --store in 1st arg for {{{1}}} to grab it from {{SS}} or {{LS}} template
local template_call = mw.getCurrentFrame():expandTemplate{ title = template_title, args = template_args }
return template_call
end


-- Retrieve scaling value and type of a hero's given stat, if it has scaling, else return 0
            spirit_scale = spirit_val
function p.get_hero_scaling_data(hero_data, hero_stat_key)
            level_scale  = level_val
local scaling_type_full
        end
local scaling_data
local scaling_value
local scaling_types = {"Spirit", "Level"}
local scaling_data_returned = {}
for index, scaling_type in ipairs(scaling_types) do
scaling_type_full = scaling_type .. "Scaling"
scaling_data = hero_data[scaling_type_full]
--If the scaling data exists
if (scaling_data ~= nil) then
scaling_value = scaling_data[hero_stat_key]
--If the stat scales
if (scaling_value ~= nil) then
scaling_data_returned[scaling_value] = scaling_type
end
end
end
return scaling_data_returned
end


--HERO_NAME in english
         if attr_key == "TechPower" and spirit_scale == 0 then
--{{#invoke:HeroData|write_infobox|HERO_NAME}}--
            spirit_scale = 1.0
p.write_infobox = function(frame)
            if type(stat_value) == "number" then
-- Get hero data
                stat_value = stat_value + (spirit_power * spirit_scale)
    hero_key = frame.args[1]
            end
    hero_data = heroes_data[hero_key]
        end
    if(hero_data == nil) then return "Hero " .. hero_key .. " Not Found" end
   
    local infobox_attributes = {
         Weapon = {'DPS','ClipSize','RoundsPerSecond','ReloadTime'},
        Vitality = {'MaxHealth','BulletResist','TechResist','MaxMoveSpeed'}
    }
   
    -- Declarations and initializations
    local category_data = attribute_module.get_category_data()
    local stats
    local stat_data
    local stat_value
    local stat_text
    local label
    local postfix
    local should_display
    local image_file_name
    local image_file
    local icon_and_value_str
    local template_args = {}
    local stat_values = {Weapon = "", Vitality = ""}
   
    -- Add the main parameters
    template_args["name_english"] = lang_module.get_string(hero_key, 'en')
    template_args["name_localized"] = localize(hero_key, hero_key)
   
    -- Iterate attribute categories
    for category, stats in pairs(infobox_attributes) do
-- Iterate stats
for _, stat_name in ipairs(stats) do
-- Confirm its in attribute data and retrieve it
attribute_data = attributes_data[category]
if (attribute_data == nil) then return "Category " .. category .. " in infobox_attributes is not in Data:AttributeData" end
stat_data = attribute_data[stat_name]
if (stat_data == nil) then return "Attribute " .. stat_name .. " in infobox_attributes is not in Data:AttributeData" end
-- gets the stat's value
stat_value = hero_data[stat_name]
if (stat_value == nil) then
    -- Check nested weapon stats
    if hero_data.Weapon and hero_data.Weapon[stat_name] ~= nil then
        stat_value = hero_data.Weapon[stat_name]
    elseif hero_data.Weapon and hero_data.Weapon.AltFire and hero_data.Weapon.AltFire[stat_name] ~= nil then
        stat_value = hero_data.Weapon.AltFire[stat_name]
    else
        stat_value = 0 --default to 0 if not present
    end
end
--Round value to 3 significant figures
stat_value = util_module.round_to_sig_fig(stat_value, 3)
-- get label and postfix
label = localize(stat_data["label"], stat_name)
postfix = stat_data["postfix"]
if (postfix == nil) then
postfix = ""
else
-- light grey for postfixes
postfix = lang_module.get_string(postfix)
if (postfix == nil) then
postfix = ""
end
postfix = '<span style="color: #666666;">' .. postfix .. "</span>"
end
-- if a language is missing the postfix, use no postfix
-- Check if icon file exists, and if not, don't use any image
image_file_name = 'File:AttributeIcon' .. stat_name .. '.png'
image_file = util_module.get_image_file(image_file_name, 15, stat_name)
-- Create the template'd icon
local icon_color = attribute_module.get_attr_icon_color(stat_name)
icon_template_title = mw.title.new("Template:Icon/" .. icon_color)
icon_template_args = {}
icon_template_args[1] = image_file
icon_template_args[2] = stat_value .. postfix
icon_and_value_str = mw.getCurrentFrame():expandTemplate{ title = icon_template_title, args = icon_template_args }
label =  ',<span style="color: #999999; background-color: #2F2F2F; display: inline-flex; width: 100%; height: 100%; justify-content: center; align-items: center;">' .. label .. '</span>'
-- First add localized stat name for left column
stat_values[category] = stat_values[category] .. label
-- Then add icon and stat value for right column
stat_values[category] = stat_values[category] .. ',' .. icon_and_value_str
end
-- Add the stat values in the category to i.e weapon_values and vitality_values parameters
template_args[string.lower(category) .. "_values"] = stat_values[category]
end
   
    -- Use expandTemplate to evaluate the Infobox_hero template
    local template_title = mw.title.new("User:Sur")
    local expanded_template = mw.getCurrentFrame():expandTemplate{ title = template_title, args = template_args }
   
    return expanded_template
end


        local innate_spirit_scale =
            (hero_data["LevelScaling"] and hero_data["LevelScaling"]["TechPower"]) or 0


--{{#invoke:HeroData|write_stat_infoboxes|HERO_KEY}}
        if not display_scaling_icons then scaling_strs = "" end
--Creates {{Infobox_stat}}'s' for the 3 categories Weapon, Vitality, Spirit
p.write_stat_infoboxes = function(frame)
local hero_key = frame.args[1]
if(hero_key == nil) then return "Hero parameter missing" end
-- Use expandTemplate to evaluate the Infobox_hero template
    local template_title = mw.title.new("Template:Infobox_stat") --name of the template
    local template_calls = '<div style="display: flex; flex-direction: column;">' --all template calls concatenated
    local template_call --current template call
local template_args = {} --current template arguments
local cell_values --current cell values
local cell_value --current cell value
local label --current stat's label
local postfix --current stat's postfix
local stat_value --current stat's numerical value in the hero data
local hero_data = heroes_data[hero_key]
local stats --stats of the current category
local image_file_name --name of the image_file to check
local image_file --mw returned image file
local extra_blank_cell --add a blank cell after certain stats
local stats_to_add_blank_after = {CritDamageReceivedScale = true}
if (hero_data == nil) then return "Hero Not Found" end
local category_data = attribute_module.get_category_data()
local should_display
for _, category in ipairs(attribute_orders["category_order"]) do
if (category ~= "Spirit") then --hide Spirit section for now
stats = attributes_data[category]
local category_values = category_data[category]
template_args["box_name"] = category_values.unlocalized_name
template_args["box_rgb"] = category_values.rgb
template_args["num_cols"] = 2
cell_values = ""
-- Determine cell values
for _, stat_name in ipairs(attribute_orders[category]["attribute_order"]) do
local stat_data = stats[stat_name]
if stat_data == nil then return "Stat " .. stat_name .. " from StatInfoboxOrder has no data in AttributeData" end
-- gets the stat's value if it has that stat, default to 0 if not
stat_value = hero_data[stat_name]
if (stat_value == nil) then
    -- Check nested weapon stats
    if hero_data.Weapon and hero_data.Weapon[stat_name] ~= nil then
        stat_value = hero_data.Weapon[stat_name]
    elseif hero_data.Weapon and hero_data.Weapon.AltFire and hero_data.Weapon.AltFire[stat_name] ~= nil then
        stat_value = hero_data.Weapon.AltFire[stat_name]
    else
        stat_value = 0 --default to 0 if not present
    end
end
--Round value to 3 significant figures
stat_value = util_module.round_to_sig_fig(stat_value, 3)
-- get label and postfix
label = localize(stat_data["label"], stat_name)
postfix = stat_data["postfix"]
if (postfix == nil) then
postfix = ""
else
-- light grey for postfixes
postfix = '<span style="color: #666666;">' .. lang_module.get_string(postfix) .. "</span>"
end
-- if a language is missing the postfix, use no postfix
if (postfix == nil) then
postfix = ""
end
-- Check if icon file exists, and if not, don't use any image
image_file_name = 'File:AttributeIcon' .. stat_name .. '.png'
-- 15px and link to stat name (page name might not match perfectly yet)
image_file = util_module.get_image_file(image_file_name, 15, stat_name)
-- Create the template'd icon
local icon_color = attribute_module.get_attr_icon_color(stat_name)
if (icon_color == "Brown") then
image_file = '<span style="position: relative; bottom: 2px; filter: invert(42%) sepia(10%) saturate(2912%) hue-rotate(351deg) brightness(90%) contrast(87%);">' .. image_file .. '</span>'
end
-- use white instead of grey if grey is returned
-- Add an empty cell following some stats to align them correctly as seen in game
if (stats_to_add_blank_after[stat_name]) then
extra_blank_cell = " ," --prefixed space is needed
else
extra_blank_cell = ""
end
-- slightly lighter color for stat name
label = '<span style="color: #3d3d3d;">' .. label .. '</span>'
-- Add the scaling str if it scales
local scaling_data = p.get_hero_scaling_data(hero_data, stat_name)
local scaling_strs = ""
if (scaling_data ~= nil) then
for scaling_value, scaling_type in pairs(scaling_data) do
local scaling_str = p.write_scalar_str(scaling_value, scaling_type)
if (scaling_str ~= "") then scaling_str = " " .. scaling_str end
scaling_strs = scaling_strs .. scaling_str
end
end
-- Set cell value as "Icon StatvaluePostfix Statname Scaling,"
-- If icon file already exists, set it to #REDIRECT to the duplicate file page
cell_value = image_file .. " " .. stat_value  .. postfix .. " " .. label  .. scaling_strs .. "," .. extra_blank_cell
cell_values = cell_values .. cell_value --add value to values list
end
template_args["cell_values"] = cell_values
-- Write current call
template_call = mw.getCurrentFrame():expandTemplate{ title = template_title, args = template_args }
-- Add call to the set of calls
template_calls = template_calls .. "\n" .. template_call
end
end
return template_calls .. "</div>"
end


local function get_nested_stat_value(hero_data, stat_key)
        if type(stat_value) == "boolean" then
    -- Check top-level, then Weapon, then AltFire
            stat_value = tostring(stat_value)
    if hero_data[stat_key] ~= nil then
         else
         return hero_data[stat_key]
            stat_value = util_module.round_to_sig_fig(stat_value, 5)
    elseif hero_data.Weapon and hero_data.Weapon[stat_key] ~= nil then
         end
        return hero_data.Weapon[stat_key]
    elseif hero_data.Weapon and hero_data.Weapon.AltFire and hero_data.Weapon.AltFire[stat_key] ~= nil then
         return hero_data.Weapon.AltFire[stat_key]
    end
    return 0
end


--writes the massive hero comparison table for a specific PI and SP
        local cell_inner = string.format(
-- scaling icons are shown only if PI and SP are both 0 or both not provided
             '<span class="stat-num">%s</span><span class="stat-scaling">%s</span>',
--{{#invoke:HeroData|write_hero_comparison_table|POWER_INCREASES|SPIRIT_POWER}}
            stat_value,
p.write_hero_comparison_table = function(frame)
            scaling_strs
-- process inputs
        )
local power_increases = frame.args[1]
if (power_increases == nil) then power_increases = 0 end
local spirit_power = frame.args[2]
if (spirit_power == nil) then spirit_power = 0 end
-- Show scaling icons if power/increases/spirit power are 0/not specified
local display_scaling_icons = false
if (power_increases == 0 and spirit_power == 0) then
display_scaling_icons = true
end
-- Initializations and declarations
local row_str = ""
local body_str = ""
local in_development
local is_disabled
local template_title = ""
local template_args = {}
local hero_icon
local hero_td_style
local scalar_str
-- Add hero comp stats to
local stats_to_include = {
Weapon = {
"DPS",
            "SustainedDPS",
            "BulletDamage",
            "RoundsPerSecond",
            "FireRate",
            "ClipSize",
            "ReloadTime",
            "ReloadDelay",
            "BulletsPerShot",
            "BulletsPerBurst",
            "BurstInterShotInterval",
            "LightMeleeDamage",
            "HeavyMeleeDamage",
            "ReloadSingle",
            "BulletSpeed",
            "BulletGravityScale",
            "FalloffStartRange",
            "FalloffEndRange",
            "BonusAttackRange",
            "RoundsPerSecondAtMaxSpin",
            "SpinAcceleration",
            "SpinDeceleration"
},
Vitality = {
"MaxHealth",
            "BaseHealthRegen",
            "BulletResist",
            "TechResist",
            "CritDamageReceivedScale",
             "MaxMoveSpeed",
            "SprintSpeed",
            "StaminaCooldown",
            "Stamina"
}
}
-- Iterate heroes
for hero_key, hero_data in pairs(heroes_data) do
-- Ensure they are not in development and they are an active hero
in_development = hero_data["InDevelopment"]
if (not in_development) then
is_disabled = hero_data["IsDisabled"]
if (not is_disabled) then
--Add the row's stats
row_str = ""
-- Retrieve hero's localized name
hero_name = localize(hero_key, hero_key)
-- Retrieve hero's english name, used for hero icon
hero_name_en = hero_data["Name"]
-- Expand hero icon
template_title = "Template:HeroIcon"
template_args[1] = hero_name_en
template_args[2] = hero_name
hero_icon = mw.getCurrentFrame():expandTemplate{ title = template_title, args = template_args }
-- First column in each row is hero name
row_str = row_str .. "<td>" .. hero_icon .. "</td>"
-- Consecutive columns are then stats
-- Iterate categories
for _, category in ipairs(attribute_orders["category_order"]) do
category_attrs = attributes_data[category]
-- Iterate attributes within the category
if (stats_to_include[category] ~= nil) then
for _, attr_key in ipairs(stats_to_include[category]) do
--Retrieve the stats value from hero data
stat_value = get_nested_stat_value(hero_data, attr_key)
-- Retrieve scaling val and str if it scales
local scaling_data = p.get_hero_scaling_data(hero_data, attr_key)
local scaling_strs = ""
if (scaling_data ~= nil) then
for scaling_value, scaling_type in pairs(scaling_data) do
local scaling_str = p.write_scalar_str(scaling_value, scaling_type)
if (scaling_str ~= "") then scaling_str = " " .. scaling_str end
if (scaling_str ~= nil) then
scaling_strs = scaling_strs .. scaling_str
end
-- Scale the stat value
if (scaling_type == "Spirit") then
stat_value = stat_value + (spirit_power * scaling_value)
elseif (scaling_type == "Level") then
if attr_key == 'TechResist' or attr_key == 'BulletResist' then
-- each PI for resists is multiplicative
stat_value = (stat_value/100 + 1 - (1-scaling_value/100)^power_increases)*100
else
stat_value = stat_value + (power_increases * scaling_value)
end
end
end
end
if (not display_scaling_icons) then scaling_strs = "" end
-- Convert from boolean to string, or round it
if (type(stat_value)) == 'boolean' then
stat_value = tostring(stat_value)
else
stat_value = util_module.round_to_sig_fig(stat_value, 3)
end
-- Add it to the row
row_str = row_str .. "<td>" .. stat_value .. scaling_strs .. "</td>"
end
end
end
-- Add row to the body
row_str = "<tr>" .. row_str .. "</tr>"
body_str = body_str .. row_str
first_hero = false
end
end
end
-- First header is Hero
local headers_str = "<th>" .. "Hero" .. "</th>"
local header_style = ""
local category_data = attribute_module.get_category_data()
local postfix
-- Used for keys that are not localized by Valve
local postfix_key_map = {
["ReloadDelay"] = "StatDesc_ReloadTime_postfix",
["BulletsPerShot"] = "",
["BulletsPerBurst"] = "",
["BurstInterShotInterval"] = "StatDesc_ReloadTime_postfix",
["ReloadSingle"] = "",
["BonusAttackRange"] = "StatDesc_WeaponRangeFalloffMax_postfix",
["SustainedDPS"] = "DPS_postfix",
["RoundsPerSecondAtMaxSpin"] = "",
["SpinAcceleration"] = "BonusFireRate_postfix",
["SpinDeceleration"] = "BonusFireRate_postfix"
}
-- Iterate stats again for displaying headers with their respective color
for _, category in ipairs(attribute_orders["category_order"]) do
category_attrs = attributes_data[category]
category_rgb = category_data[category]["rgb"]
-- Iterate attributes within the category
if (stats_to_include[category] ~= nil) then
for _, attr_key in ipairs(stats_to_include[category]) do
attr_data = category_attrs[attr_key]
if (attr_data ~= nil) then
-- Localize, fallback to attr_key with proper spaces
attr_localized = lang_module.get_string(attr_data["label"])
if (attr_localized == nil or attr_localized == "") then attr_localized = util_module.add_space_before_cap(attr_key) end
-- Get postfix
postfix = lang_module.get_string(attr_data["postfix"])
if (postfix == nil or postfix == "") then
postfix = ""
else
postfix = " (" .. postfix .. ")"
end
else -- If not in attributes data, use dictionary translate and postfix
attr_localized = dictionary_module.translate(attr_key)
postfix = lang_module.get_string(postfix_key_map[attr_key])
if postfix == nil then
return "attr_key " .. attr_key .. " must be added to postfix_key_map"
end
if postfix ~= "" then
postfix = " (" .. postfix .. ")"
end
end
header_style = ' style="background-color: ' .. "rgb(" .. category_rgb .. ') ;"'
headers_str = headers_str .. "<th" .. header_style .. ">" .. attr_localized .. postfix .. "</th>"
end
end
end
headers_str = "<tr>" .. headers_str .. "</tr>"


local table_str = '<table class="wikitable sortable" style="table-layout: auto; width: 100%;">' .. headers_str .. body_str .. "</table>"
        local weapon_table = hero_data.Weapon
local output_str = '<div style="overflow-x: auto; width: 100%;">' .. table_str .. '</div>'
        if is_alt_fire and weapon_table and weapon_table.AltFire then
return output_str
            weapon_table = weapon_table.AltFire
end
        end
        local hit_once = "false"
        if (attr_key == "DPS" or attr_key == "SustainedDPS")
            and weapon_table
            and weapon_table.HitOnceAcrossAllBullets
        then
            hit_once = "true"
        end


function localize(key, fallback)
        local data_attrs = string.format(
local result = lang_module.get_string(key)
            'data-stat-name="%s" data-base="%s" data-spirit-scale="%s" data-level-scale="%s" data-innate-spirit-scale="%s" data-sort-value="%s" data-hit-once="%s"',
if (result == "") or (result == nil) then
            attr_key,
result = util_module.add_space_before_cap(fallback) .. mw.getCurrentFrame():expandTemplate{title="MissingValveTranslationTooltip"}
            tostring(type(base_value) == "number" and util_module.round_to_sig_fig(base_value, 3) or base_value),
end
            tonumber(spirit_scale) or 0,
return result
            (attr_key == "TechPower") and "0" or (tonumber(level_scale) or 0),
end
            innate_spirit_scale or 0,
            stat_value,
            hit_once
        )


-- Mapping from hero name to internal tag name
        return string.format('<td style="white-space: nowrap;" %s>%s</td>', data_attrs, cell_inner)
local TAG_NAME_MAP = {
     end
["Abrams"] = "Abrams",
["Bebop"] = "Bebop",
["Billy"] = "Punkgoat",
["Calico"] = "Nano",
["The Doorman"] = "Doorman",
["Drifter"] = "Drifter",
["Dynamo"] = "Dynamo",
["Grey Talon"] = "Orion",
["Haze"] = "Haze",
    ["Holliday"] = "Astro",
    ["Infernus"] = "Inferno",
    ["Ivy"] = "Tengu",
    ["Kelvin"] = "Kelvin",
    ["Lady Geist"] = "Geist",
    ["Lash"] = "Lash",
    ["McGinnis"] = "Engineer",
    ["Mina"] = "VampireBat",
    ["Mirage"] = "Mirage",
    ["Mo & Krill"] = "Digger",
    ["Paige"] = "Bookworm",
    ["Paradox"] = "Chrono",
    ["Pocket"] = "Synth",
     ["Seven"] = "Gigawatt",
    ["Shiv"] = "Shiv",
    ["Sinclair"] = "Magician",
    ["Victor"] = "Frank",
    ["Vindicta"] = "Vindicta",
    ["Viscous"] = "Viscous",
    ["Vyper"] = "Viper",
    ["Warden"] = "Warden",
    ["Wraith"] = "Wraith",
    ["Yamato"] = "Yamato",
}


-- Function to call hero tags based on name
    -- Generate a hero's primary stat row
p.get_hero_tag = function (frame)
    local function generatePrimaryRow(hero_data, hero_key, has_alt)
    local name = frame.args[1]
        local row_str = ""
    local number = tonumber(frame.args[2])
        local hero_name_local = localize(hero_key, hero_key)
        local hero_name_en = hero_data["Name"]
        local template_args = {[1] = hero_name_en, l1 = hero_name_local}
        local hero_icon = mw.getCurrentFrame():expandTemplate{ title = "Template:HeroIcon", args = template_args }
        local hero_cell_content = hero_icon
        if has_alt then
            hero_cell_content = '[[#alt-fire-' .. hero_key .. '|+]] ' .. hero_cell_content
        end
        local sort_name = (hero_data["Name"] or hero_key):gsub("^The ", "")
        row_str = row_str .. '<td style="position: sticky; left: 0; z-index: 10; background-color: var(--background-color-base-2); isolation: isolate; overflow: hidden; min-width: 150px;" data-sort-value="' .. sort_name .. '">' .. hero_cell_content .. '</td>'
        for _, category in ipairs(attribute_orders["category_order"]) do
            if stats_to_include[category] ~= nil then
                for _, attr_key in ipairs(stats_to_include[category]) do
                    row_str = row_str .. buildStatCell(hero_data, hero_key, attr_key, false, category)
                end
            end
        end
        return "<tr>" .. row_str .. "</tr>"
    end


     -- Validate input
     -- Generate an expandable alt-fire sub-row (weapon stats only)
     if not name then
     local function generateAltFireSubRow(hero_data, hero_key)
         return "Error: Missing name input."
        local row_str = ""
        local hero_name_local = localize(hero_key, hero_key)
        local hero_name_en = hero_data["Name"]
        local template_args = {[1] = hero_name_en, l1 = hero_name_local}
        local hero_icon = mw.getCurrentFrame():expandTemplate{ title = "Template:HeroIcon", args = template_args }
        local hero_cell_content = hero_icon .. ' <small style="color:#aaa;">(Alt‑fire)</small>'
        local sort_name = (hero_data["Name"] or hero_key):gsub("^The ", "")
        row_str = row_str .. '<td style="position: sticky; left: 0; z-index: 9; background-color: #202122; isolation: isolate; overflow: hidden; min-width: 150px;" data-sort-value="' .. sort_name .. '">' .. hero_cell_content .. '</td>'
        for _, category in ipairs(attribute_orders["category_order"]) do
            if stats_to_include[category] ~= nil then
                for _, attr_key in ipairs(stats_to_include[category]) do
                    if category == "Weapon" then
                        row_str = row_str .. buildStatCell(hero_data, hero_key, attr_key, true, category)
                    else
                        row_str = row_str .. '<td></td>'
                    end
                end
            end
        end
         return '<tr class="alt-fire-sub" data-parent="' .. hero_key .. '">' .. row_str .. '</tr>'
     end
     end


     if number ~= 1 and number ~= 2 and number ~= 3 then
     -- Build all body rows
         return string.format("Error: Invalid number '%s'. Must be 1, 2, or 3.", tostring(frame.args[2]))
    for _, hero_entry in ipairs(sorted_heroes) do
        local hero_key  = hero_entry.key
        local hero_data = hero_entry.data
        local has_alt = hero_data.Weapon and hero_data.Weapon.AltFire and true or false
         body_str = body_str .. generatePrimaryRow(hero_data, hero_key, has_alt)
        if has_alt then
            body_str = body_str .. generateAltFireSubRow(hero_data, hero_key)
        end
     end
     end


     local tagname = TAG_NAME_MAP[name]
     -- Pre-pass: determine which stats have any scaling, and which have both types.
    if not tagname then
    -- Used to set column widths in the header.
        return string.format("Error: Unknown name '%s'.", name)
    local stats_with_any_scaling  = {}
    local stats_with_both_scaling = {}
    for _, hero_entry in ipairs(sorted_heroes) do
        local hero_data = hero_entry.data
        for _, category in ipairs(attribute_orders["category_order"]) do
            if stats_to_include[category] ~= nil then
                for _, attr_key in ipairs(stats_to_include[category]) do
                    -- Primary scaling
                    local scaling_data = hero_data_module.get_hero_scaling_data(hero_data, attr_key)
                    if scaling_data ~= nil and next(scaling_data) ~= nil then
                        stats_with_any_scaling[attr_key] = true
                        local has_spirit, has_level = false, false
                        for _, scaling_type in pairs(scaling_data) do
                            if scaling_type == "Spirit" then has_spirit = true
                            elseif scaling_type == "Level" then has_level = true end
                        end
                        if has_spirit and has_level then stats_with_both_scaling[attr_key] = true end
                    end
 
                    -- Alt-fire scaling (only for stats present in the AltFire table)
                    if hero_data.Weapon
                        and hero_data.Weapon.AltFire
                        and hero_data.Weapon.AltFire[attr_key] ~= nil
                    then
                        if attr_key ~= "ClipSize" then
                            local alt_scaling = hero_data_module.get_hero_scaling_data(hero_data, attr_key .. "AltFire")
                            if alt_scaling and next(alt_scaling) ~= nil then
                                stats_with_any_scaling[attr_key] = true
                                local has_spirit_alt, has_level_alt = false, false
                                for _, stype in pairs(alt_scaling) do
                                    if stype == "Spirit" then has_spirit_alt = true
                                    elseif stype == "Level" then has_level_alt = true end
                                end
                                if has_spirit_alt and has_level_alt then stats_with_both_scaling[attr_key] = true end
                            else
                                if attr_key == "DPS" or attr_key == "SustainedDPS" then
                                    local alt_stats = hero_data.Weapon.AltFire
                                    local dps_type = attr_key == "DPS" and 'burst' or 'sustained'
                                    local lv = compute_dps_scaling(hero_data, alt_stats, dps_type, "Level")
                                    local sp = compute_dps_scaling(hero_data, alt_stats, dps_type, "Spirit")
                                    lv = util_module.round_to_sig_fig(lv, 5)
                                    sp = util_module.round_to_sig_fig(sp, 5)
                                    if lv ~= 0 or sp ~= 0 then
                                        stats_with_any_scaling[attr_key] = true
                                        if lv ~= 0 and sp ~= 0 then
                                            stats_with_both_scaling[attr_key] = true
                                        end
                                    end
                                end
                            end
                        end
                    end
                end
            end
        end
     end
     end


     -- Build localization key
     -- Build table header row
     local key = string.format("Citadel_%s_HeroTag_%d", tagname, number)
     local headers_str = '<th style="position: sticky; left: 0; top: -1px; z-index: 12; background-color: var(--background-color-base-5); isolation: isolate; min-width: 150px;">Hero</th>'
    local category_data = attribute_module.get_category_data()
    local postfix_key_map = {
        ["ReloadDelay"] = "StatDesc_ReloadTime_postfix",
        ["BulletsPerShot"] = "",
        ["BulletsPerBurst"] = "",
        ["BurstInterShotInterval"] = "StatDesc_ReloadTime_postfix",
        ["ReloadSingle"] = "",
        ["BonusAttackRange"] = "StatDesc_WeaponRangeFalloffMax_postfix",
        ["SustainedDPS"] = "DPS_postfix",
        ["RoundsPerSecondAtMaxSpin"] = "",
        ["CritDamageBonusPercent"]  = "StatDesc_CritDamageBonusScale_postfix",
["CritDamageReceivedPercent"] = "StatDesc_CritDamageReceivedScale_postfix",
        ["BulletLifesteal"] = "BulletLifestealPercentHero_postfix",
        ["GroundDashSpeed"] = "DashSpeed_postfix"
    }
    for _, category in ipairs(attribute_orders["category_order"]) do
        local category_attrs = attributes_data[category]
        local category_rgb  = category_data[category]["rgb"]
        if stats_to_include[category] ~= nil then
            for _, attr_key in ipairs(stats_to_include[category]) do
                local attr_data = category_attrs[attr_key]
               
                -- Fallback: if the stat is missing from this category, check all other categories
                if attr_data == nil then
                    for _, fallback_cat_data in pairs(attributes_data) do
                        if fallback_cat_data[attr_key] ~= nil then
                            attr_data = fallback_cat_data[attr_key]
                            break
                        end
                    end
                end


    -- Call Lang.get_string
                local attr_localized, postfix
    local result = lang_module.get_string(key)
                if attr_data ~= nil then
                    attr_localized = lang_module.get_string(attr_data["label"])
                    if attr_localized == nil or attr_localized == "" then
                        attr_localized = util_module.add_space_before_cap(attr_key)
                    end
                    postfix = lang_module.get_string(attr_data["postfix"])
                    if postfix == nil or postfix == "" then postfix = ""
                    else postfix = " (" .. postfix .. ")" end
                    if attr_key == "BulletRadius" then postfix = " (cm)" end
                else
                    attr_localized = dictionary_module.translate(attr_key)
                    postfix = lang_module.get_string(postfix_key_map[attr_key])
                    if postfix == nil then return "attr_key " .. attr_key .. " must be added to postfix_key_map" end
                    if postfix ~= "" then postfix = " (" .. postfix .. ")" end
                end
                local th_style = 'position: sticky; top: -1px; z-index: 3; background-color: rgb(' .. category_rgb .. ');'
                if stats_with_both_scaling[attr_key] then
                    th_style = th_style .. ' min-width: 130px;'
                elseif stats_with_any_scaling[attr_key] then
                    th_style = th_style .. ' min-width: 75px;'
                end
                headers_str = headers_str .. '<th style="' .. th_style .. '">' .. attr_localized .. postfix .. "</th>"
            end
        end
    end
    headers_str = "<tr>" .. headers_str .. "</tr>"


     return result or ("[Missing string for key: " .. key .. "]")
     return string.format(
        '<div id="hero-comparison-container" data-max-power="%s" data-max-spirit="%s">' ..
        '<div style="overflow: auto; max-height: 70vh; width: 100%%;">' ..
        '<table class="wikitable sortable" style="table-layout: auto; width: 100%%;" id="hero-comparison-table">%s%s</table>' ..
        '</div></div>',
        max_power, max_spirit, headers_str, body_str
    )
end
end


return p
return p