ComfyFactorio/maps/scrap_towny_ffa/combat_balance.lua

local Public = {}

local player_ammo_damage_starting_modifiers = {
    --['artillery-shell'] = -0.75,
    --['biological'] = -0.5,
    ['bullet'] = 0,
    ['cannon-shell'] = -0.5,
    ['capsule'] = 0,
    ['beam'] = -0.5,
    ['laser'] = -0.5,
    ['electric'] = -0.5,
    ['flamethrower'] = 0,
    ['grenade'] = -0.5,
    ['landmine'] = -0.75,
    --['melee'] = 2,
    --['rocket'] = -0.75,
    ['shotgun-shell'] = 0
}
local player_ammo_damage_modifiers = {
    --['artillery-shell'] = -0.75,
    --['biological'] = -0.5,
    ['bullet'] = 0,
    ['cannon-shell'] = -0.5,
    ['capsule'] = 0,
    ['beam'] = -0.5,
    ['laser'] = -0.5,
    ['electric'] = -0.5,
    ['flamethrower'] = 0,
    ['grenade'] = -0.5,
    ['landmine'] = -0.5,
    --['melee'] = 0,
    --['rocket'] = -0.5,
    ['shotgun-shell'] = 0
}
local player_gun_speed_modifiers = {
    --['artillery-shell'] = -0.75,
    --['biological'] = -0.5,
    ['bullet'] = 0,
    ['cannon-shell'] = -0.5,
    ['capsule'] = -0.5,
    ['beam'] = -0.5,
    ['laser'] = 0,
    ['electric'] = -0.5,
    ['flamethrower'] = 0,
    ['grenade'] = -0.5,
    ['landmine'] = 0,
    --['melee'] = 1,
    --['rocket'] = -0.75,
    ['shotgun-shell'] = 0
}

--local player_turrets_research_modifiers = {
--    ['gun-turret'] = 0,
--    ['laser-turret'] = 1.5,
--    ['flamethrower-turret'] = 0
--}

--local enemy_ammo_starting_modifiers = {
--    ['artillery-shell'] = 3,
--    ['biological'] = 3,
--    ['bullet'] = 2,
--    ['cannon-shell'] = 0,
--    ['capsule'] = 0,
--    ['beam'] = 0,
--    ['laser'] = 0,
--    ['electric'] = 0,
--    ['flamethrower'] = 0,
--    ['grenade'] = 0,
--    ['landmine'] = 0,
--    ['melee'] = 1,
--    ['rocket'] = 0,
--    ['shotgun-shell'] = 0
--}

--local enemy_ammo_evolution_modifiers = {
--    ['artillery-shell'] = 1,
--    ['biological'] = 2,
--    ['bullet'] = 1,
--    --['cannon-shell'] = 1,
--    ['capsule'] = 1,
--    ['beam'] = 1,
--    ['laser'] = 1,
--    ['electric'] = 1,
--    ['flamethrower'] = 2,
--    --['grenade'] = 1,
--    --['landmine'] = 1,
--    ['melee'] = 2
--    --['rocket'] = 1,
--    --['shotgun-shell'] = 1
--}

function Public.init_player_weapon_damage(force)
    for k, v in pairs(player_ammo_damage_starting_modifiers) do
        force.set_ammo_damage_modifier(k, v)
    end

    for k, v in pairs(player_gun_speed_modifiers) do
        force.set_gun_speed_modifier(k, v)
    end

    force.set_turret_attack_modifier('laser-turret', 3)
end

--local function init_enemy_weapon_damage()
--    local e_force = game.forces['enemy']
--
--    for k, v in pairs(enemy_ammo_starting_modifiers) do
--        e_force.set_ammo_damage_modifier(k, v)
--    end
--end
--
--local function enemy_weapon_damage()
--    local f = game.forces.enemy
--
--    local ef = f.evolution_factor
--
--    for k, v in pairs(enemy_ammo_evolution_modifiers) do
--        local base = enemy_ammo_starting_modifiers[k]
--
--        local new = base + v * ef
--        f.set_ammo_damage_modifier(k, new)
--    end
--end

-- After a research is finished and the game applied the modifier, we reduce modifiers to achieve the reduction
local function research_finished(event)
    local r = event.research
    local p_force = r.force

    for _, e in ipairs(r.effects) do
        local t = e.type

        if t == 'ammo-damage' then
            local category = e.ammo_category
            local factor = player_ammo_damage_modifiers[category]

            if factor then
                local current_m = p_force.get_ammo_damage_modifier(category)
                p_force.set_ammo_damage_modifier(category, current_m + factor * e.modifier)
            end
        --elseif t == 'turret-attack' then  -- NOTE: this doesn't trigger for laser turrets :-(
        --    local category = e.turret_id
        --    local factor = player_turrets_research_modifiers[category]
        --    game.print("XDB cat " .. category)
        --    if factor then
        --        local current_m = p_force.get_turret_attack_modifier(category)
        --        game.print("XDB mod " .. current_m .. " -> " .. current_m + factor * e.modifier)
        --        p_force.set_turret_attack_modifier(category, current_m + factor * e.modifier)
        --    end
        elseif t == 'gun-speed' then
            local category = e.ammo_category
            local factor = player_gun_speed_modifiers[category]

            if factor then
                local current_m = p_force.get_gun_speed_modifier(category)
                p_force.set_gun_speed_modifier(category, current_m + factor * e.modifier)
            end
        end
    end
end

local Event = require 'utils.event'
--Event.on_init(init_enemy_weapon_damage)
--Event.on_nth_tick(18000, enemy_weapon_damage)
Event.add(defines.events.on_research_finished, research_finished)
return Public