DAI: improved AttackersBehaviour and AtatckersHandler

LuaRules/Gadgets/ai/EvoRTS/attackerbehaviour.lua

@@ -0,0 +1,95 @@
+shard_include( "attackers" )
+
+function IsAttacker(unit)
+	for i,name in ipairs(attackerlist) do
+		if name == unit:Internal():Name() then
+			return true
+		end
+	end
+	return false
+end
+
+AttackerBehaviour = class(Behaviour)
+
+function AttackerBehaviour:Init()
+	--self.ai.game:SendToConsole("attacker!")
+	--self.game:AddMarker({ x = startPosx, y = startPosy, z = startPosz }, "my start position")
+end
+
+function AttackerBehaviour:OwnerBuilt()
+	self.ai.attackhandler:AddRecruit(self)
+	self.attacking = true
+	self.active = true
+end
+
+
+function AttackerBehaviour:OwnerDead()
+	self.ai.attackhandler:RemoveRecruit(self)
+end
+
+function AttackerBehaviour:OwnerIdle()
+	self.attacking = true
+	self.active = true
+	self.ai.attackhandler:AddRecruit(self)
+end
+
+function AttackerBehaviour:AttackCell(cell)
+	local unit = self.unit:Internal()
+	local currenthealth = unit:GetHealth()
+	local maxhealth = unit:GetMaxHealth()
+	local startPosx, startPosy, startPosz = Spring.GetTeamStartPosition(self.ai.id)
+	if currenthealth >= maxhealth then
+		p = api.Position()
+		p.x = cell.posx
+		p.z = cell.posz
+		p.y = 0
+		self.target = p
+		self.attacking = true
+		self.ai.attackhandler:AddRecruit(self)
+		if self.active then
+			self.unit:Internal():MoveAndFire(self.target)
+		else
+			self.unit:ElectBehaviour()
+		end
+	else
+		p = api.Position()
+		p.x = startPosx + math.random(0,1000) - math.random(0,1000)
+		p.z = startPosz + math.random(0,1000) - math.random(0,1000)
+		p.y = startPosy
+		self.target = p
+		self.attacking = false
+		self.ai.attackhandler:AddRecruit(self)
+		if self.active then
+			self.unit:Internal():Move(self.target)
+		else
+			self.unit:ElectBehaviour()
+		end
+	end
+end
+
+function AttackerBehaviour:Priority()
+	if not self.attacking then
+		return 0
+	else
+		return 100
+	end
+end
+
+function AttackerBehaviour:Activate()
+	self.active = true
+	if self.target then
+		self.unit:Internal():MoveAndFire(self.target)
+		self.target = nil
+		self.ai.attackhandler:AddRecruit(self)
+	else
+		self.ai.attackhandler:AddRecruit(self)
+	end
+end
+
+
+function AttackerBehaviour:OwnerDied()
+	self.ai.attackhandler:RemoveRecruit(self)
+	self.attacking = nil
+	self.active = nil
+	self.unit = nil
+end

LuaRules/Gadgets/ai/EvoRTS/attackhandler.lua

@@ -0,0 +1,180 @@
+AttackHandler = class(Module)
+
+function AttackHandler:Name()
+	return "AttackHandler"
+end
+
+function AttackHandler:internalName()
+	return "attackhandler"
+end
+
+function AttackHandler:Init()
+	self.recruits = {}
+	self.counter = 1
+end
+
+function AttackHandler:Update()
+-- stagger targetting if multiple shards are in the game
+	local f = self.game:Frame() + self.game:GetTeamID() 
+	if math.mod(f,30) == 0 then
+		self:DoTargetting()
+	end
+end
+
+function AttackHandler:UnitDead(engineunit)
+	if engineunit:Team() == self.game:GetTeamID() then
+		--self.counter = self.counter - 0.2
+		--self.counter = math.max(self.counter,8)
+		-- try and clean up dead recruits where possible
+		for i,v in ipairs(self.recruits) do
+			if v.engineID == engineunit:ID() then
+				table.remove(self.recruits,i)
+				break
+			end
+		end
+	end
+end
+
+function AttackHandler:DoTargetting()
+	if #self.recruits > self.counter then
+
+		--[[ try and catch invalid recruits and remove them, then reevaluate
+		local nrecruits = {}
+		for i,v in ipairs(self.recruits) do
+			if v.unit ~= nil then
+				table.insert(nrecruits,v)
+				break
+			end
+		end
+		self.recruits = nrecruits
+		if #nrecruits > self.counter then
+			
+		else
+			return
+		end]]--
+
+		-- find somewhere to attack
+		local cells = {}
+		local celllist = {}
+		local mapdimensions = self.game.map:MapDimensions()
+		--enemies = game:GetEnemies()
+		local enemies = self.game:GetEnemies()
+
+		if enemies and #enemies > 0 then
+			-- figure out where all the enemies are!
+			for i=1,#enemies do
+				local e = enemies[i]
+
+				if e ~= nil then
+					pos = e:GetPosition()
+					px = pos.x - math.fmod(pos.x,400)
+					pz = pos.z - math.fmod(pos.z,400)
+					px = px/400
+					pz = pz/400
+					if cells[px] == nil then
+						cells[px] = {}
+					end
+					if cells[px][pz] == nil then
+						local newcell = { count = 0, posx = 0,posz=0,}
+						cells[px][pz] = newcell
+						table.insert(celllist,newcell)
+					end
+					cell = cells[px][pz]
+					cell.count = cell.count + self:ScoreUnit(e)
+
+					-- we dont want to target the center of the cell encase its a ledge with nothing
+					-- on it etc so target this units position instead
+					cell.posx = pos.x
+					cell.posz = pos.z
+
+					-- @TODO: The unit chosen may not be the best unit to target, ideally pick the
+					-- one closest to the average location of all units in that grid
+				end
+
+			end
+
+			local bestCell = nil
+			-- now find the smallest nonvacant cell to go lynch!
+			for i=1,#celllist do
+				local cell = celllist[i]
+				if bestCell == nil then
+					bestCell = cell
+				else
+					if cell.count < bestCell.count then
+						bestCell = cell
+					end
+				end
+			end
+
+			-- if we have a cell then lets go attack it!
+			if bestCell ~= nil then
+				for i,recruit in ipairs(self.recruits) do
+					recruit:AttackCell(bestCell)
+				end
+
+				--self.counter = self.counter + 0.2
+				--self.counter = math.min(self.counter,20)
+
+				-- remove all our recruits!
+				--self.recruits = {}
+			end
+		end
+
+		-- cleanup
+		cellist = nil
+		cells = nil
+		mapdimensions = nil
+
+	end
+end
+
+function AttackHandler:IsRecruit(attkbehaviour)
+	for i,v in ipairs(self.recruits) do
+		if v.engineID == attkbehaviour.engineID then
+			return true
+		end
+	end
+	return false
+end
+
+function AttackHandler:AddRecruit(attkbehaviour)
+	if attkbehaviour.unit == nil then
+		self.game:SendToConsole( "null unit in attack beh found ")
+		return
+	end
+	if not self:IsRecruit(attkbehaviour) then
+		table.insert(self.recruits,attkbehaviour)
+	end
+end
+
+function AttackHandler:RemoveRecruit(attkbehaviour)
+	for i,v in ipairs(self.recruits) do
+		if v.engineID == attkbehaviour.engineID then
+			table.remove(self.recruits,i)
+			return true
+		end
+	end
+	return false
+end
+
+-- How much is this unit worth?
+-- 
+-- Idea: add a table with hardcoded values,
+-- and use said values if a units found in
+-- that table to highlight strategic value
+function AttackHandler:ScoreUnit(unit)
+	local value = 1
+	--[[
+	if unit:CanMove() then
+		if unit:CanBuild() then
+			value = value + 1
+		end
+	else
+		value = value + 2
+		if unit:CanBuild() then
+			value = value + 2
+		end
+	end
+	--]]
+	return value
+end