mirror of
https://github.com/veden/Rampant.git
synced 2024-12-30 21:19:46 +02:00
393 lines
14 KiB
Lua
393 lines
14 KiB
Lua
if (squadAttackG) then
|
|
return squadAttackG
|
|
end
|
|
local squadAttack = {}
|
|
|
|
-- imports
|
|
|
|
local constants = require("Constants")
|
|
local mapUtils = require("MapUtils")
|
|
local movementUtils = require("MovementUtils")
|
|
local mathUtils = require("MathUtils")
|
|
local chunkPropertyUtils = require("ChunkPropertyUtils")
|
|
|
|
-- constants
|
|
|
|
local PLAYER_PHEROMONE = constants.PLAYER_PHEROMONE
|
|
local BASE_PHEROMONE = constants.BASE_PHEROMONE
|
|
local RESOURCE_PHEROMONE = constants.RESOURCE_PHEROMONE
|
|
|
|
local TEN_DEATH_PHEROMONE_GENERATOR_AMOUNT = constants.TEN_DEATH_PHEROMONE_GENERATOR_AMOUNT
|
|
local FIVE_DEATH_PHEROMONE_GENERATOR_AMOUNT = constants.FIVE_DEATH_PHEROMONE_GENERATOR_AMOUNT
|
|
|
|
local SQUAD_BUILDING = constants.SQUAD_BUILDING
|
|
|
|
local SQUAD_RAIDING = constants.SQUAD_RAIDING
|
|
local SQUAD_SETTLING = constants.SQUAD_SETTLING
|
|
local SQUAD_GUARDING = constants.SQUAD_GUARDING
|
|
local SQUAD_RETREATING = constants.SQUAD_RETREATING
|
|
|
|
local AI_STATE_SIEGE = constants.AI_STATE_SIEGE
|
|
|
|
local PLAYER_PHEROMONE_MULTIPLER = constants.PLAYER_PHEROMONE_MULTIPLER
|
|
|
|
local DEFINES_DISTRACTION_NONE = defines.distraction.none
|
|
local DEFINES_DISTRACTION_BY_ENEMY = defines.distraction.by_enemy
|
|
local DEFINES_DISTRACTION_BY_ANYTHING = defines.distraction.by_anything
|
|
|
|
-- imported functions
|
|
|
|
local euclideanDistancePoints = mathUtils.euclideanDistancePoints
|
|
|
|
local findMovementPosition = movementUtils.findMovementPosition
|
|
|
|
local removeSquadFromChunk = chunkPropertyUtils.removeSquadFromChunk
|
|
local addDeathGenerator = chunkPropertyUtils.addDeathGenerator
|
|
local getDeathGenerator = chunkPropertyUtils.getDeathGenerator
|
|
|
|
local getNestCount = chunkPropertyUtils.getNestCount
|
|
|
|
local getNeighborChunks = mapUtils.getNeighborChunks
|
|
local addSquadToChunk = chunkPropertyUtils.addSquadToChunk
|
|
local getChunkByXY = mapUtils.getChunkByXY
|
|
local positionToChunkXY = mapUtils.positionToChunkXY
|
|
local addMovementPenalty = movementUtils.addMovementPenalty
|
|
local positionFromDirectionAndFlat = mapUtils.positionFromDirectionAndFlat
|
|
|
|
local euclideanDistanceNamed = mathUtils.euclideanDistanceNamed
|
|
|
|
local getPlayerBaseGenerator = chunkPropertyUtils.getPlayerBaseGenerator
|
|
local getResourceGenerator = chunkPropertyUtils.getResourceGenerator
|
|
|
|
local scoreNeighborsForAttack = movementUtils.scoreNeighborsForAttack
|
|
local scoreNeighborsForSettling = movementUtils.scoreNeighborsForSettling
|
|
|
|
-- module code
|
|
|
|
local function scoreResourceLocationKamikaze(_, neighborChunk)
|
|
local settle = neighborChunk[RESOURCE_PHEROMONE]
|
|
return settle - (neighborChunk[PLAYER_PHEROMONE] * PLAYER_PHEROMONE_MULTIPLER)
|
|
end
|
|
|
|
local function scoreSiegeLocationKamikaze(_, neighborChunk)
|
|
local settle = neighborChunk[BASE_PHEROMONE] +
|
|
neighborChunk[RESOURCE_PHEROMONE] + (neighborChunk[PLAYER_PHEROMONE] * PLAYER_PHEROMONE_MULTIPLER)
|
|
|
|
return settle
|
|
end
|
|
|
|
local function scoreResourceLocation(map, neighborChunk)
|
|
local settle = -getDeathGenerator(map, neighborChunk) + neighborChunk[RESOURCE_PHEROMONE]
|
|
return settle - (neighborChunk[PLAYER_PHEROMONE] * PLAYER_PHEROMONE_MULTIPLER)
|
|
end
|
|
|
|
local function scoreSiegeLocation(map, neighborChunk)
|
|
local settle = -getDeathGenerator(map, neighborChunk) + neighborChunk[BASE_PHEROMONE] +
|
|
neighborChunk[RESOURCE_PHEROMONE] + (neighborChunk[PLAYER_PHEROMONE] * PLAYER_PHEROMONE_MULTIPLER)
|
|
|
|
return settle
|
|
end
|
|
|
|
local function scoreAttackLocation(map, neighborChunk)
|
|
local damage = -getDeathGenerator(map, neighborChunk) + neighborChunk[BASE_PHEROMONE] +
|
|
(neighborChunk[PLAYER_PHEROMONE] * PLAYER_PHEROMONE_MULTIPLER)
|
|
return damage
|
|
end
|
|
|
|
local function scoreAttackKamikazeLocation(_, neighborChunk)
|
|
local damage = neighborChunk[BASE_PHEROMONE] + (neighborChunk[PLAYER_PHEROMONE] * PLAYER_PHEROMONE_MULTIPLER)
|
|
return damage
|
|
end
|
|
|
|
local function settleMove(map, squad)
|
|
local universe = map.universe
|
|
local targetPosition = universe.position
|
|
local targetPosition2 = universe.position2
|
|
local group = squad.group
|
|
|
|
local groupPosition = group.position
|
|
local x, y = positionToChunkXY(groupPosition)
|
|
local chunk = getChunkByXY(map, x, y)
|
|
local scoreFunction = scoreResourceLocation
|
|
if (map.state == AI_STATE_SIEGE) then
|
|
if squad.kamikaze then
|
|
scoreFunction = scoreSiegeLocationKamikaze
|
|
else
|
|
scoreFunction = scoreSiegeLocation
|
|
end
|
|
elseif squad.kamikaze then
|
|
scoreFunction = scoreResourceLocationKamikaze
|
|
end
|
|
addDeathGenerator(map, chunk, FIVE_DEATH_PHEROMONE_GENERATOR_AMOUNT)
|
|
addSquadToChunk(map, chunk, squad)
|
|
addMovementPenalty(squad, chunk)
|
|
local distance = euclideanDistancePoints(groupPosition.x,
|
|
groupPosition.y,
|
|
squad.originPosition.x,
|
|
squad.originPosition.y)
|
|
local cmd
|
|
local position
|
|
local surface = map.surface
|
|
|
|
if (distance >= squad.maxDistance) or ((getResourceGenerator(map, chunk) ~= 0) and (getNestCount(map, chunk) == 0))
|
|
then
|
|
position = findMovementPosition(surface, groupPosition)
|
|
|
|
if not position then
|
|
position = groupPosition
|
|
end
|
|
|
|
targetPosition.x = position.x
|
|
targetPosition.y = position.y
|
|
|
|
cmd = universe.settleCommand
|
|
if squad.kamikaze then
|
|
cmd.distraction = DEFINES_DISTRACTION_NONE
|
|
else
|
|
cmd.distraction = DEFINES_DISTRACTION_BY_ENEMY
|
|
end
|
|
|
|
squad.status = SQUAD_BUILDING
|
|
|
|
group.set_command(cmd)
|
|
else
|
|
local attackChunk,
|
|
attackDirection,
|
|
nextAttackChunk,
|
|
nextAttackDirection = scoreNeighborsForSettling(map,
|
|
chunk,
|
|
getNeighborChunks(map, x, y),
|
|
scoreFunction)
|
|
|
|
if (attackChunk == -1) then
|
|
cmd = universe.wonderCommand
|
|
group.set_command(cmd)
|
|
return
|
|
elseif (attackDirection ~= 0) then
|
|
local attackPlayerThreshold = universe.attackPlayerThreshold
|
|
|
|
if (nextAttackChunk ~= -1) then
|
|
attackChunk = nextAttackChunk
|
|
positionFromDirectionAndFlat(attackDirection, groupPosition, targetPosition)
|
|
positionFromDirectionAndFlat(nextAttackDirection, targetPosition, targetPosition2)
|
|
position = findMovementPosition(surface, targetPosition2)
|
|
else
|
|
positionFromDirectionAndFlat(attackDirection, groupPosition, targetPosition)
|
|
position = findMovementPosition(surface, targetPosition)
|
|
end
|
|
|
|
if position then
|
|
targetPosition.x = position.x
|
|
targetPosition.y = position.y
|
|
if nextAttackChunk then
|
|
addDeathGenerator(map, nextAttackChunk, FIVE_DEATH_PHEROMONE_GENERATOR_AMOUNT)
|
|
else
|
|
addDeathGenerator(map, attackChunk, FIVE_DEATH_PHEROMONE_GENERATOR_AMOUNT)
|
|
end
|
|
else
|
|
cmd = universe.wonderCommand
|
|
group.set_command(cmd)
|
|
return
|
|
end
|
|
|
|
if (getPlayerBaseGenerator(map, attackChunk) ~= 0) or
|
|
(attackChunk[PLAYER_PHEROMONE] >= attackPlayerThreshold)
|
|
then
|
|
cmd = universe.attackCommand
|
|
|
|
if not squad.rabid then
|
|
squad.frenzy = true
|
|
squad.frenzyPosition.x = groupPosition.x
|
|
squad.frenzyPosition.y = groupPosition.y
|
|
end
|
|
else
|
|
cmd = universe.moveCommand
|
|
if squad.rabid or squad.kamikaze then
|
|
cmd.distraction = DEFINES_DISTRACTION_NONE
|
|
else
|
|
cmd.distraction = DEFINES_DISTRACTION_BY_ENEMY
|
|
end
|
|
end
|
|
else
|
|
cmd = universe.settleCommand
|
|
cmd.destination.x = groupPosition.x
|
|
cmd.destination.y = groupPosition.y
|
|
|
|
if squad.kamikaze then
|
|
cmd.distraction = DEFINES_DISTRACTION_NONE
|
|
else
|
|
cmd.distraction = DEFINES_DISTRACTION_BY_ENEMY
|
|
end
|
|
|
|
squad.status = SQUAD_BUILDING
|
|
end
|
|
|
|
group.set_command(cmd)
|
|
end
|
|
end
|
|
|
|
local function attackMove(map, squad)
|
|
|
|
local universe = map.universe
|
|
local targetPosition = universe.position
|
|
local targetPosition2 = universe.position2
|
|
|
|
local group = squad.group
|
|
|
|
local surface = map.surface
|
|
local position
|
|
local groupPosition = group.position
|
|
local x, y = positionToChunkXY(groupPosition)
|
|
local chunk = getChunkByXY(map, x, y)
|
|
local attackScorer = scoreAttackLocation
|
|
if squad.kamikaze then
|
|
attackScorer = scoreAttackKamikazeLocation
|
|
end
|
|
local squadChunk = squad.chunk
|
|
addDeathGenerator(map, squadChunk, FIVE_DEATH_PHEROMONE_GENERATOR_AMOUNT)
|
|
addSquadToChunk(map, chunk, squad)
|
|
addMovementPenalty(squad, chunk)
|
|
squad.frenzy = (squad.frenzy and (euclideanDistanceNamed(groupPosition, squad.frenzyPosition) < 100))
|
|
local attackChunk, attackDirection,
|
|
nextAttackChunk, nextAttackDirection = scoreNeighborsForAttack(map,
|
|
chunk,
|
|
getNeighborChunks(map, x, y),
|
|
attackScorer)
|
|
local cmd
|
|
if (attackChunk == -1) then
|
|
cmd = universe.wonderCommand
|
|
group.set_command(cmd)
|
|
return
|
|
elseif (nextAttackChunk ~= -1) then
|
|
attackChunk = nextAttackChunk
|
|
positionFromDirectionAndFlat(attackDirection, groupPosition, targetPosition)
|
|
positionFromDirectionAndFlat(nextAttackDirection, targetPosition, targetPosition2)
|
|
position = findMovementPosition(surface, targetPosition2)
|
|
else
|
|
positionFromDirectionAndFlat(attackDirection, groupPosition, targetPosition)
|
|
position = findMovementPosition(surface, targetPosition)
|
|
end
|
|
|
|
if not position then
|
|
cmd = universe.wonderCommand
|
|
group.set_command(cmd)
|
|
return
|
|
else
|
|
targetPosition.x = position.x
|
|
targetPosition.y = position.y
|
|
if nextAttackChunk then
|
|
addDeathGenerator(map, nextAttackChunk, FIVE_DEATH_PHEROMONE_GENERATOR_AMOUNT)
|
|
else
|
|
addDeathGenerator(map, attackChunk, FIVE_DEATH_PHEROMONE_GENERATOR_AMOUNT)
|
|
end
|
|
end
|
|
|
|
if (getPlayerBaseGenerator(map, attackChunk) ~= 0) and
|
|
(attackChunk[PLAYER_PHEROMONE] >= universe.attackPlayerThreshold)
|
|
then
|
|
cmd = universe.attackCommand
|
|
|
|
if not squad.rabid then
|
|
squad.frenzy = true
|
|
squad.frenzyPosition.x = groupPosition.x
|
|
squad.frenzyPosition.y = groupPosition.y
|
|
end
|
|
else
|
|
cmd = universe.moveCommand
|
|
if squad.rabid or squad.frenzy then
|
|
cmd.distraction = DEFINES_DISTRACTION_BY_ANYTHING
|
|
else
|
|
cmd.distraction = DEFINES_DISTRACTION_BY_ENEMY
|
|
end
|
|
end
|
|
|
|
group.set_command(cmd)
|
|
end
|
|
|
|
local function buildMove(map, squad)
|
|
local group = squad.group
|
|
local universe = map.universe
|
|
local position = universe.position
|
|
local groupPosition = findMovementPosition(map.surface, group.position)
|
|
|
|
if not groupPosition then
|
|
groupPosition = group.position
|
|
end
|
|
|
|
position.x = groupPosition.x
|
|
position.y = groupPosition.y
|
|
|
|
group.set_command(universe.compoundSettleCommand)
|
|
end
|
|
|
|
function squadAttack.cleanSquads(map)
|
|
local squads = map.groupNumberToSquad
|
|
local k = map.squadIterator
|
|
local squad
|
|
if not k then
|
|
k, squad = next(squads, k)
|
|
else
|
|
squad = squads[k]
|
|
end
|
|
if not k then
|
|
map.squadIterator = nil
|
|
if (table_size(squads) == 0) then
|
|
-- this is needed as the next command remembers the max length a table has been
|
|
map.groupNumberToSquad = {}
|
|
end
|
|
else
|
|
map.squadIterator = next(squads, k)
|
|
local group = squad.group
|
|
if not group.valid then
|
|
addDeathGenerator(map, squad.chunk, FIVE_DEATH_PHEROMONE_GENERATOR_AMOUNT)
|
|
removeSquadFromChunk(map, squad)
|
|
if (map.regroupIterator == k) then
|
|
map.regroupIterator = nil
|
|
end
|
|
local universe = map.universe
|
|
if squad.settlers then
|
|
universe.builderCount = universe.builderCount - 1
|
|
else
|
|
universe.squadCount = universe.squadCount - 1
|
|
end
|
|
squads[k] = nil
|
|
elseif (group.state == 4) then
|
|
squadAttack.squadDispatch(map, squad)
|
|
end
|
|
end
|
|
end
|
|
|
|
function squadAttack.squadDispatch(map, squad)
|
|
local group = squad.group
|
|
if group and group.valid then
|
|
local status = squad.status
|
|
if (status == SQUAD_RAIDING) then
|
|
attackMove(map, squad)
|
|
elseif (status == SQUAD_SETTLING) then
|
|
settleMove(map, squad)
|
|
elseif (status == SQUAD_RETREATING) then
|
|
if squad.settlers then
|
|
squad.status = SQUAD_SETTLING
|
|
settleMove(map, squad)
|
|
else
|
|
squad.status = SQUAD_RAIDING
|
|
attackMove(map, squad)
|
|
end
|
|
elseif (status == SQUAD_BUILDING) then
|
|
removeSquadFromChunk(map, squad)
|
|
buildMove(map, squad)
|
|
elseif (status == SQUAD_GUARDING) then
|
|
if squad.settlers then
|
|
squad.status = SQUAD_SETTLING
|
|
settleMove(map, squad)
|
|
else
|
|
squad.status = SQUAD_RAIDING
|
|
attackMove(map, squad)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
squadAttackG = squadAttack
|
|
return squadAttack
|