mirror of
https://github.com/ComfyFactory/ComfyFactorio.git
synced 2025-01-24 03:47:58 +02:00
585 lines
26 KiB
Lua
585 lines
26 KiB
Lua
local Public = {}
|
|
local math_floor = math.floor
|
|
|
|
local Table = require 'modules.scrap_towny_ffa.table'
|
|
|
|
local biters = {
|
|
[1] = "small-biter",
|
|
[2] = "medium-biter",
|
|
[3] = "big-biter",
|
|
[4] = "behemoth-biter"
|
|
}
|
|
|
|
local spitters = {
|
|
[1] = "small-spitter",
|
|
[2] = "medium-spitter",
|
|
[3] = "big-spitter",
|
|
[4] = "behemoth-spitter"
|
|
}
|
|
|
|
local worms = {
|
|
[1] = "small-worm-turret",
|
|
[2] = "medium-worm-turret",
|
|
[3] = "big-worm-turret",
|
|
[4] = "behemoth-worm-turret"
|
|
}
|
|
|
|
-- evolution max distance in tiles
|
|
local max_evolution_distance = 1024
|
|
local max_pollution_behemoth = 256
|
|
local max_pollution_big = 64
|
|
local max_pollution_medium = 16
|
|
local max_factor = 0.8
|
|
|
|
-- technology weights (biter, spitter, worm)
|
|
local technology_weights = {
|
|
['advanced-electronics'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['advanced-electronics-2'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['advanced-material-processing'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['advanced-material-processing-2'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['advanced-oil-processing'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['artillery']={biter=1, spitter=1, worm=1},
|
|
['artillery-shell-range-1']={biter=1, spitter=1, worm=1},
|
|
['artillery-shell-speed-1']={biter=1, spitter=1, worm=1},
|
|
['atomic-bomb']={biter=1, spitter=1, worm=1},
|
|
['automated-rail-transportation'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['automation'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['automation-2'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['automation-3'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['automobilism'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['battery'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['battery-equipment'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['battery-mk2-equipment'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['belt-immunity-equipment'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['braking-force-1'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['braking-force-2'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['braking-force-3'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['braking-force-4'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['braking-force-5'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['braking-force-6'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['braking-force-7'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['chemical-science-pack'] = { biter = 125, spitter = 125, worm = 125 },
|
|
['circuit-network'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['cliff-explosives'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['coal-liquefaction'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['concrete'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['construction-robotics'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['defender'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['destroyer'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['discharge-defense-equipment'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['distractor'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['effect-transmission'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['effectivity-module'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['effectivity-module-2'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['effectivity-module-3'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['electric-energy-accumulators'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['electric-energy-distribution-1'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['electric-energy-distribution-2'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['electric-engine'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['electronics'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['energy-shield-equipment'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['energy-shield-mk2-equipment'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['energy-weapons-damage-1'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['energy-weapons-damage-2'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['energy-weapons-damage-3'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['energy-weapons-damage-4'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['energy-weapons-damage-5'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['energy-weapons-damage-6'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['energy-weapons-damage-7'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['engine'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['exoskeleton-equipment'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['explosive-rocketry']={biter=1, spitter=1, worm=1},
|
|
['explosives'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['fast-inserter'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['flamethrower'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['flammables'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['fluid-handling'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['fluid-wagon'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['follower-robot-count-1'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['follower-robot-count-2'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['follower-robot-count-3'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['follower-robot-count-4'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['follower-robot-count-5'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['follower-robot-count-6'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['follower-robot-count-7'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['fusion-reactor-equipment'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['gate'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['gun-turret'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['heavy-armor'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['improved-equipment'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['inserter-capacity-bonus-1'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['inserter-capacity-bonus-3'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['inserter-capacity-bonus-4'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['inserter-capacity-bonus-5'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['inserter-capacity-bonus-6'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['inserter-capacity-bonus-7'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['kovarex-enrichment-process'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['land-mine'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['landfill'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['laser'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['laser-turret-speed-1'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['laser-turret-speed-2'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['laser-turret-speed-3'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['laser-turret-speed-4'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['laser-turret-speed-5'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['laser-turret-speed-6'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['laser-turret-speed-7'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['laser-turret'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['logistic-robotics'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['logistic-science-pack'] = { biter = 25, spitter = 25, worm = 25 },
|
|
['logistic-system'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['logistics'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['logistics-2'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['logistics-3'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['low-density-structure'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['lubricant'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['military'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['military-2'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['military-3'] = { biter = 5, spitter = 5, worm = 51 },
|
|
['military-4'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['military-science-pack'] = { biter = 50, spitter = 50, worm = 50 },
|
|
['mining-productivity-1'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['mining-productivity-2'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['mining-productivity-3'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['mining-productivity-4'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['modular-armor'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['modules'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['night-vision-equipment'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['nuclear-fuel-reprocessing'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['nuclear-power'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['oil-processing'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['optics'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['personal-laser-defense-equipment'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['personal-roboport-equipment'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['personal-roboport-mk2-equipment'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['physical-projectile-damage-1'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['physical-projectile-damage-2'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['physical-projectile-damage-3'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['physical-projectile-damage-4'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['physical-projectile-damage-5'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['physical-projectile-damage-6'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['physical-projectile-damage-7'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['plastics'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['power-armor'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['power-armor-mk2'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['production-science-pack'] = { biter = 250, spitter = 250, worm = 250 },
|
|
['productivity-module'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['productivity-module-2'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['productivity-module-3'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['rail-signals'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['railway'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['refined-flammables-1'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['refined-flammables-2'] = { biter = 5, spitter = 5, worm = 51 },
|
|
['refined-flammables-3'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['refined-flammables-4'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['refined-flammables-5'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['refined-flammables-6'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['refined-flammables-7'] = { biter = 5, spitter = 5, worm = 51 },
|
|
['research-speed-1'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['research-speed-2'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['research-speed-3'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['research-speed-4'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['research-speed-5'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['research-speed-6'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['robotics'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['rocket-control-unit'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['rocket-fuel'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['rocket-silo'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['rocketry']={biter=1, spitter=1, worm=1},
|
|
['solar-energy'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['solar-panel-equipment'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['space-science-pack'] = { biter = 1000, spitter = 1000, worm = 1000 },
|
|
['speed-module'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['speed-module-2'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['speed-module-3'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['spidertron'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['stack-inserter'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['steel-axe'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['steel-processing'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['stone-wall'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['stronger-explosives-1'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['stronger-explosives-2'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['stronger-explosives-3'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['stronger-explosives-4'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['stronger-explosives-5'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['stronger-explosives-6'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['stronger-explosives-7'] = { biter = 5, spitter = 5, worm = 51 },
|
|
['sulfur-processing'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['tank'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['toolbelt'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['uranium-ammo'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['uranium-processing'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['utility-science-pack'] = { biter = 500, spitter = 500, worm = 500 },
|
|
['weapon-shooting-speed-1'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['weapon-shooting-speed-2'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['weapon-shooting-speed-3'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['weapon-shooting-speed-4'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['weapon-shooting-speed-5'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['weapon-shooting-speed-6'] = { biter = 5, spitter = 5, worm = 5 },
|
|
['worker-robots-speed-1'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['worker-robots-speed-2'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['worker-robots-speed-3'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['worker-robots-speed-4'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['worker-robots-speed-5'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['worker-robots-speed-6'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['worker-robots-storage-1'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['worker-robots-storage-2'] = { biter = 1, spitter = 1, worm = 1 },
|
|
['worker-robots-storage-3'] = { biter = 1, spitter = 1, worm = 1 },
|
|
}
|
|
|
|
local max_biter_weight = 0
|
|
local max_spitter_weight = 0
|
|
local max_worm_weight = 0
|
|
for _, weight in pairs(technology_weights) do
|
|
max_biter_weight = max_biter_weight + weight.biter
|
|
max_spitter_weight = max_spitter_weight + weight.spitter
|
|
max_worm_weight = max_worm_weight + weight.worm
|
|
end
|
|
max_biter_weight = max_biter_weight * max_factor
|
|
max_spitter_weight = max_spitter_weight * max_factor
|
|
max_worm_weight = max_worm_weight * max_factor
|
|
|
|
local function get_unit_size(evolution)
|
|
-- returns a value 0-3 that represents the unit size
|
|
|
|
-- basically evo values of: 0%, 10%, 30%, 60%, 80%, 100%
|
|
-- small unit chances are 100%, 100%, 50%, 25%, 12.5%, 0%
|
|
-- medium unit chances are 0%, 0%, 50%, 25%, 12.5%, 0%
|
|
-- big unit chances are 0%, 0%, 0%, 50%, 25%, 0%
|
|
-- behemoth unit chances are 0%, 0%, 0%, 0%, 50%, 100%
|
|
-- and curve accordingly in between evo values
|
|
|
|
-- magic stuff happens here
|
|
if (evolution < 0.10) then return 1 end
|
|
if (evolution >= 0.10 and evolution < 0.40) then
|
|
local r = (evolution - 0.10) * 5
|
|
if math.random() < 0.5 then return 1 end
|
|
if math.random() < r then return 1 end
|
|
return 2
|
|
end
|
|
if (evolution >= 0.30 and evolution < 0.60) then
|
|
local r = (evolution - 0.30) * 3.3333
|
|
if math.random() < 0.5 then
|
|
if math.random() < 0.5 then
|
|
return 1
|
|
else
|
|
if math.random() < r then return 1 else return 2 end
|
|
end
|
|
else
|
|
if math.random() < r then return 2 else return 3 end
|
|
end
|
|
end
|
|
if (evolution >= 0.60 and evolution < 0.80) then
|
|
local r = (evolution - 0.60) * 5
|
|
if math.random() < 0.5 then
|
|
if math.random() < 0.5 then
|
|
if math.random() < r then return 1 else return 2 end
|
|
else
|
|
if math.random() < r then return 2 else return 3 end
|
|
end
|
|
else
|
|
if math.random() < r then return 3 else return 4 end
|
|
end
|
|
end
|
|
if (evolution >= 0.80 and evolution < 1.0) then
|
|
local r = (evolution - 0.80) * 5
|
|
if math.random() < 0.5 then
|
|
if math.random() < r then
|
|
if math.random() < r then
|
|
return 1
|
|
else return 2 end
|
|
else return 3 end
|
|
else return 4 end
|
|
end
|
|
if (evolution >= 1.0) then return 4 end
|
|
end
|
|
|
|
local function distance_squared(pos1, pos2)
|
|
-- calculate the distance squared
|
|
local dx = pos1.x - pos2.x
|
|
local dy = pos1.y - pos2.y
|
|
local d2 = dx * dx + dy * dy
|
|
return d2
|
|
end
|
|
|
|
local function get_relative_biter_evolution(position)
|
|
local ffatable = Table.get_table()
|
|
local relative_evolution = 0.0
|
|
local max_d2 = max_evolution_distance * max_evolution_distance
|
|
-- for all of the teams
|
|
local teams = ffatable.town_centers
|
|
for _, town_center in pairs(teams) do
|
|
local market_position = town_center.market.position
|
|
-- calculate the distance squared
|
|
local d2 = distance_squared(position, market_position)
|
|
if d2 < max_d2 then
|
|
-- get the distance factor (0.0-1.0)
|
|
local distance_factor = 1.0 - d2 / max_d2;
|
|
-- get the evolution factor (0.0-1.0)
|
|
if not town_center.evolution then town_center.evolution = {} end
|
|
if town_center.evolution.biters == nil then town_center.evolution.biters = 0.0 end
|
|
local evolution_factor = town_center.evolution.biters
|
|
local evo = distance_factor * evolution_factor
|
|
relative_evolution = math.max(relative_evolution, evo)
|
|
end
|
|
end
|
|
return relative_evolution
|
|
end
|
|
|
|
local function get_relative_spitter_evolution(position)
|
|
local ffatable = Table.get_table()
|
|
local relative_evolution = 0.0
|
|
local max_d2 = max_evolution_distance * max_evolution_distance
|
|
-- for all of the teams
|
|
local teams = ffatable.town_centers
|
|
for _, town_center in pairs(teams) do
|
|
local market_position = town_center.market.position
|
|
-- calculate the distance squared
|
|
local d2 = distance_squared(position, market_position)
|
|
if d2 < max_d2 then
|
|
-- get the distance factor (0.0-1.0)
|
|
local distance_factor = 1.0 - d2 / max_d2;
|
|
-- get the evolution factor (0.0-1.0)
|
|
if not town_center.evolution then town_center.evolution = {} end
|
|
if town_center.evolution.spitters == nil then town_center.evolution.spitters = 0.0 end
|
|
local evolution_factor = town_center.evolution.spitters
|
|
local evo = distance_factor * evolution_factor
|
|
relative_evolution = math.max(relative_evolution, evo)
|
|
end
|
|
end
|
|
return relative_evolution
|
|
end
|
|
|
|
local function get_relative_worm_evolution(position)
|
|
local ffatable = Table.get_table()
|
|
local relative_evolution = 0.0
|
|
local max_d2 = max_evolution_distance * max_evolution_distance
|
|
-- for all of the teams
|
|
local teams = ffatable.town_centers
|
|
for _, town_center in pairs(teams) do
|
|
local market_position = town_center.market.position
|
|
-- calculate the distance squared
|
|
local d2 = distance_squared(position, market_position)
|
|
if d2 < max_d2 then
|
|
-- get the distance factor (0.0-1.0)
|
|
local distance_factor = 1.0 - d2 / max_d2;
|
|
-- get the evolution factor (0.0-1.0)
|
|
if not town_center.evolution then town_center.evolution = {} end
|
|
if town_center.evolution.worms == nil then town_center.evolution.worms = 0.0 end
|
|
local evolution_factor = town_center.evolution.worms
|
|
local evo = distance_factor * evolution_factor
|
|
relative_evolution = math.max(relative_evolution, evo)
|
|
end
|
|
end
|
|
return relative_evolution
|
|
end
|
|
|
|
function Public.get_evolution(position)
|
|
return get_relative_biter_evolution(position)
|
|
end
|
|
|
|
function Public.get_biter_evolution(entity)
|
|
return get_relative_biter_evolution(entity.position)
|
|
end
|
|
|
|
function Public.get_spitter_evolution(entity)
|
|
return get_relative_spitter_evolution(entity.position)
|
|
end
|
|
|
|
function Public.get_worm_evolution(entity)
|
|
return get_relative_worm_evolution(entity.position)
|
|
end
|
|
|
|
local function get_nearby_location(position, surface, radius, entity_name)
|
|
return surface.find_non_colliding_position(entity_name, position, radius, 0.5, false)
|
|
end
|
|
|
|
local function set_biter_type(entity)
|
|
-- checks nearby evolution levels for bases and returns an appropriately leveled type
|
|
local position = entity.position
|
|
local evo = get_relative_biter_evolution(position)
|
|
local unit_size = get_unit_size(evo)
|
|
local entity_name = biters[unit_size]
|
|
if entity.name == entity_name then return end
|
|
local surface = entity.surface
|
|
local pollution = surface.get_pollution(position)
|
|
local behemoth = math_floor(pollution / max_pollution_behemoth)
|
|
local big = math_floor((pollution - (behemoth * max_pollution_behemoth)) / max_pollution_big)
|
|
local medium = math_floor((pollution - (behemoth * max_pollution_behemoth) - (big * max_pollution_big)) / max_pollution_medium)
|
|
local small = pollution - (behemoth * max_pollution_behemoth) - (big * max_pollution_big) - (medium * max_pollution_medium) + 1
|
|
|
|
if entity.valid then
|
|
for _ = 1, behemoth do
|
|
local e = surface.create_entity({ name = biters[4], position = get_nearby_location(position, surface, 5, biters[4]) })
|
|
e.copy_settings(entity);
|
|
e.ai_settings.allow_try_return_to_spawner = true
|
|
end
|
|
for _ = 1, big do
|
|
local e = surface.create_entity({ name = biters[3], position = get_nearby_location(position, surface, 5, biters[3]) })
|
|
e.copy_settings(entity);
|
|
e.ai_settings.allow_try_return_to_spawner = true
|
|
end
|
|
for _ = 1, medium do
|
|
local e = surface.create_entity({ name = biters[2], position = get_nearby_location(position, surface, 5, biters[2]) })
|
|
e.copy_settings(entity);
|
|
e.ai_settings.allow_try_return_to_spawner = true
|
|
end
|
|
for _ = 1, small do
|
|
local e = surface.create_entity({ name = biters[1], position = get_nearby_location(position, surface, 5, biters[1]) })
|
|
e.copy_settings(entity);
|
|
e.ai_settings.allow_try_return_to_spawner = true
|
|
end
|
|
local e = surface.create_entity({ name = entity_name, position = get_nearby_location(position, surface, 5, entity_name) })
|
|
e.copy_settings(entity);
|
|
e.ai_settings.allow_try_return_to_spawner = true
|
|
entity.destroy()
|
|
--log("spawned " .. entity_name)
|
|
end
|
|
end
|
|
|
|
local function set_spitter_type(entity)
|
|
-- checks nearby evolution levels for bases and returns an appropriately leveled type
|
|
local position = entity.position
|
|
local evo = get_relative_spitter_evolution(position)
|
|
local unit_size = get_unit_size(evo)
|
|
local entity_name = spitters[unit_size]
|
|
if entity.name == entity_name then return end
|
|
local surface = entity.surface
|
|
local pollution = surface.get_pollution(position)
|
|
local behemoth = math_floor(pollution / max_pollution_behemoth)
|
|
local big = math_floor((pollution - (behemoth * max_pollution_behemoth)) / max_pollution_big)
|
|
local medium = math_floor((pollution - (behemoth * max_pollution_behemoth) - (big * max_pollution_big)) / max_pollution_medium)
|
|
local small = pollution - (behemoth * max_pollution_behemoth) - (big * max_pollution_big) - (medium * max_pollution_medium) + 1
|
|
|
|
if entity.valid then
|
|
for _ = 1, behemoth do
|
|
local e = surface.create_entity({ name = spitters[4], position = get_nearby_location(position, surface, 5, spitters[4]) })
|
|
e.copy_settings(entity);
|
|
e.ai_settings.allow_try_return_to_spawner = true
|
|
end
|
|
for _ = 1, big do
|
|
local e = surface.create_entity({ name = spitters[3], position = get_nearby_location(position, surface, 5, spitters[3]) })
|
|
e.copy_settings(entity);
|
|
e.ai_settings.allow_try_return_to_spawner = true
|
|
end
|
|
for _ = 1, medium do
|
|
local e = surface.create_entity({ name = spitters[2], position = get_nearby_location(position, surface, 5, spitters[2]) })
|
|
e.copy_settings(entity);
|
|
e.ai_settings.allow_try_return_to_spawner = true
|
|
end
|
|
for _ = 1, small do
|
|
local e = surface.create_entity({ name = spitters[1], position = get_nearby_location(position, surface, 5, spitters[1]) })
|
|
e.copy_settings(entity);
|
|
e.ai_settings.allow_try_return_to_spawner = true
|
|
end
|
|
local e = surface.create_entity({ name = entity_name, position = get_nearby_location(position, surface, 5, entity_name) })
|
|
e.copy_settings(entity);
|
|
e.ai_settings.allow_try_return_to_spawner = true
|
|
entity.destroy()
|
|
--log("spawned " .. entity_name)
|
|
end
|
|
end
|
|
|
|
local function set_worm_type(entity)
|
|
-- checks nearby evolution levels for bases and returns an appropriately leveled type
|
|
local position = entity.position
|
|
local evo = get_relative_worm_evolution(position)
|
|
local unit_size = get_unit_size(evo)
|
|
local entity_name = worms[unit_size]
|
|
if entity.name == entity_name then return end
|
|
local surface = entity.surface
|
|
if entity.valid then
|
|
entity.destroy()
|
|
surface.create_entity({ name = entity_name, position = position })
|
|
--log("spawned " .. entity_name)
|
|
end
|
|
end
|
|
|
|
local function is_biter(entity)
|
|
if entity == nil or not entity.valid then return false end
|
|
if entity.name == 'small-biter' or entity.name == 'medium-biter' or entity.name == 'big-biter' or entity.name == 'behemoth-biter' then
|
|
return true
|
|
end
|
|
return false
|
|
end
|
|
|
|
local function is_spitter(entity)
|
|
if entity == nil or not entity.valid then return false end
|
|
if entity.name == 'small-spitter' or entity.name == 'medium-spitter' or entity.name == 'big-spitter' or entity.name == 'behemoth-spitter' then
|
|
return true
|
|
end
|
|
return false
|
|
end
|
|
|
|
local function is_worm(entity)
|
|
if entity == nil or not entity.valid then return false end
|
|
if entity.name == 'small-worm-turret' or entity.name == 'medium-worm-turret' or entity.name == 'big-worm-turret' or entity.name == 'behemoth-worm-turret' then
|
|
return true
|
|
end
|
|
return false
|
|
end
|
|
|
|
local function update_evolution(force_name, technology)
|
|
if technology == nil then return end
|
|
local ffatable = Table.get_table()
|
|
-- update evolution based on research completed (weighted)
|
|
local town_center = ffatable.town_centers[force_name]
|
|
-- town_center is a reference to a global table
|
|
if not town_center then return end
|
|
-- initialize if not already
|
|
local evo = town_center.evolution
|
|
-- get the weights for this technology
|
|
local weight = technology_weights[technology]
|
|
if weight == nil then
|
|
log("no technology_weights for " .. technology)
|
|
return
|
|
end
|
|
local biter_weight = weight.biter
|
|
local spitter_weight = weight.spitter
|
|
local worm_weight = weight.worm
|
|
-- update the evolution values (0.0 to 1.0)
|
|
local b = (biter_weight / max_biter_weight)
|
|
local s = (spitter_weight / max_spitter_weight)
|
|
local w = (worm_weight / max_worm_weight)
|
|
b = b + evo.biters
|
|
s = s + evo.spitters
|
|
w = w + evo.worms
|
|
evo.biters = b
|
|
evo.spitters = s
|
|
evo.worms = w
|
|
end
|
|
|
|
local function on_research_finished(event)
|
|
local research = event.research
|
|
local force_name = research.force.name
|
|
local technology = research.name
|
|
update_evolution(force_name, technology)
|
|
end
|
|
|
|
local function on_entity_spawned(event)
|
|
local entity = event.entity
|
|
-- check the unit type and handle appropriately
|
|
if is_biter(entity) then
|
|
set_biter_type(entity)
|
|
end
|
|
if is_spitter(entity) then
|
|
set_spitter_type(entity)
|
|
end
|
|
if is_worm(entity) then
|
|
set_worm_type(entity)
|
|
end
|
|
end
|
|
|
|
local function on_biter_base_built(event)
|
|
local entity = event.entity
|
|
if is_worm(entity) then
|
|
set_worm_type(entity)
|
|
end
|
|
end
|
|
|
|
local Event = require 'utils.event'
|
|
Event.add(defines.events.on_research_finished, on_research_finished)
|
|
Event.add(defines.events.on_entity_spawned, on_entity_spawned)
|
|
Event.add(defines.events.on_biter_base_built, on_biter_base_built)
|
|
|
|
return Public |