mirror of
https://github.com/veden/Rampant.git
synced 2025-01-05 22:53:24 +02:00
124 lines
4.2 KiB
Lua
124 lines
4.2 KiB
Lua
local aiBuilding = {}
|
|
|
|
-- imports
|
|
|
|
local constants = require("Constants")
|
|
local mapUtils = require("MapUtils")
|
|
local unitGroupUtils = require("UnitGroupUtils")
|
|
local neighborUtils = require("NeighborUtils")
|
|
package.path = "../?.lua;" .. package.path
|
|
local config = require("config")
|
|
|
|
-- constants
|
|
|
|
local BASE_PHEROMONE = constants.BASE_PHEROMONE
|
|
local PLAYER_PHEROMONE = constants.PLAYER_PHEROMONE
|
|
local MOVEMENT_PHEROMONE = constants.MOVEMENT_PHEROMONE
|
|
|
|
local AI_SQUAD_COST = constants.AI_SQUAD_COST
|
|
local AI_VENGENCE_SQUAD_COST = constants.AI_VENGENCE_SQUAD_COST
|
|
|
|
local RALLY_TRIGGERED = constants.RALLY_TRIGGERED
|
|
local INTERVAL_LOGIC = constants.INTERVAL_LOGIC
|
|
|
|
local HALF_CHUNK_SIZE = constants.HALF_CHUNK_SIZE
|
|
local TRIPLE_CHUNK_SIZE = constants.TRIPLE_CHUNK_SIZE
|
|
local NORTH_SOUTH_PASSABLE = constants.NORTH_SOUTH_PASSABLE
|
|
local EAST_WEST_PASSABLE = constants.EAST_WEST_PASSABLE
|
|
|
|
local RALLY_CRY_DISTANCE = constants.RALLY_CRY_DISTANCE
|
|
|
|
local DEFINES_COMMAND_GROUP = defines.command.group
|
|
local DEFINES_DISTRACTION_NONE = defines.distraction.none
|
|
|
|
local NEST_COUNT = constants.NEST_COUNT
|
|
|
|
-- imported functions
|
|
|
|
local getNeighborChunks = mapUtils.getNeighborChunks
|
|
local getChunkByIndex = mapUtils.getChunkByIndex
|
|
local scoreNeighbors = neighborUtils.scoreNeighbors
|
|
local createSquad = unitGroupUtils.createSquad
|
|
local attackWaveScaling = config.attackWaveScaling
|
|
|
|
-- module code
|
|
|
|
local function attackWaveValidCandidate(chunk, natives, surface)
|
|
local total = 0;
|
|
|
|
if natives.attackUsePlayer then
|
|
local playerPheromone = chunk[PLAYER_PHEROMONE]
|
|
if (playerPheromone > natives.attackPlayerThreshold) then
|
|
total = total + chunk[PLAYER_PHEROMONE]
|
|
end
|
|
end
|
|
if natives.attackUsePollution then
|
|
total = total + surface.get_pollution(chunk)
|
|
end
|
|
|
|
return total > natives.attackWaveThreshold
|
|
end
|
|
|
|
local function scoreUnitGroupLocation(squad, neighborChunk, surface)
|
|
return surface.get_pollution(neighborChunk) + neighborChunk[PLAYER_PHEROMONE] + neighborChunk[MOVEMENT_PHEROMONE] + neighborChunk[BASE_PHEROMONE]
|
|
end
|
|
|
|
local function validUnitGroupLocation(x, chunk, neighborChunk)
|
|
return neighborChunk[NORTH_SOUTH_PASSABLE] and neighborChunk[EAST_WEST_PASSABLE] and (neighborChunk[NEST_COUNT] ~= 0)
|
|
end
|
|
|
|
function aiBuilding.rallyUnits(chunk, regionMap, surface, natives, tick, tempNeighbors)
|
|
if (tick - chunk[RALLY_TRIGGERED] > INTERVAL_LOGIC) and (natives.points >= AI_VENGENCE_SQUAD_COST) then
|
|
chunk[RALLY_TRIGGERED] = tick
|
|
local cX = chunk.cX
|
|
local cY = chunk.cY
|
|
for x=cX - RALLY_CRY_DISTANCE, cX + RALLY_CRY_DISTANCE do
|
|
for y=cY - RALLY_CRY_DISTANCE, cY + RALLY_CRY_DISTANCE do
|
|
local rallyChunk = getChunkByIndex(regionMap, x, y)
|
|
if rallyChunk and (rallyChunk[NEST_COUNT] ~= 0) and (x ~= cX) and (y ~= cY) then
|
|
aiBuilding.formSquads(regionMap, surface, natives, rallyChunk, AI_VENGENCE_SQUAD_COST, tempNeighbors)
|
|
if (natives.points < AI_VENGENCE_SQUAD_COST) then
|
|
return
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
function aiBuilding.formSquads(regionMap, surface, natives, chunk, cost, tempNeighbors)
|
|
local valid = ((cost == AI_VENGENCE_SQUAD_COST) or
|
|
((cost == AI_SQUAD_COST) and attackWaveValidCandidate(chunk, natives, surface)))
|
|
|
|
if valid and (math.random() < natives.formSquadThreshold) then
|
|
|
|
local squadPath, _ = scoreNeighbors(chunk,
|
|
getNeighborChunks(regionMap, chunk.cX, chunk.cY, tempNeighbors),
|
|
validUnitGroupLocation,
|
|
scoreUnitGroupLocation,
|
|
nil,
|
|
surface,
|
|
false)
|
|
if squadPath then
|
|
local squadPosition = { x = squadPath.x + HALF_CHUNK_SIZE,
|
|
y = squadPath.y + HALF_CHUNK_SIZE }
|
|
|
|
local squad = createSquad(squadPosition, surface, natives)
|
|
|
|
squad.rabid = math.random() < 0.03
|
|
|
|
local scaledWaveSize = attackWaveScaling(natives)
|
|
local foundUnits = surface.set_multi_command({ command = { type = DEFINES_COMMAND_GROUP,
|
|
group = squad.group,
|
|
distraction = DEFINES_DISTRACTION_NONE },
|
|
unit_count = scaledWaveSize,
|
|
unit_search_distance = TRIPLE_CHUNK_SIZE })
|
|
if (foundUnits > 0) then
|
|
natives.points = natives.points - cost
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
return aiBuilding
|