Factorio Forums

Code: Select all

local radius = 50
local maxRobotsCharged = 10
local chargeRate = 10

function UpdatePoles(event)
	for _,poleTable in pairs(global.robotChargerPoles) do
		pole = poleTable[0]
		chargingRobots = poleTable[1]
		
		--Remove any robots that are no longer valid
		robotsToRemove = {}
		for _,robot in pairs(chargingRobots) do
			if robot.valid == false then
				table.insert(robotsToRemove, robot)
			end
		end
		for _,robot in pairs(robotsToRemove) do
			table.remove(chargingRobots,robot)
		end
		
		--Only maxRobotsCharged number of robots charging at a time
		if #chargingRobots < maxRobotsCharged then
			length = #chargingRobots
			foundEntities = game.get_surface(1).find_entities{{-radius, -radius}, {radius, radius}}
			for _,entity in pairs(foundEntities) do
				--Find any sort of robot. Examples of robot names: "logistic-robot" "construction-robot" "logistic-robot-mark2" etc.
				if string.find(entity.name, "robot") ~= nil then
					if string.find(entity.name, "logistic") ~= nil or string.find(entity.name, "construction") ~= nil then
						table.insert(chargingRobots, entity)
						length = length + 1
						if length >= maxRobotsCharged then
							break
						end
					end
				end
			end
		end
		
		for _,robot in pairs(chargingRobots) do
			--TODO Suck energy out of pole and add it to the robots
		end
		--game.players[1].print("Found " .. #chargingRobots .. " robots.")
	end
end

function BuiltEntity(event)
	if event.created_entity.name == "robot-charger-pole" then
		poleTable = {}
		poleTable[0] = event.created_entity
		poleTable[1] = {}
		--game.players[1].print("Entity created")
		table.insert(global.robotChargerPoles, poleTable)
	end
end

function DestroyedEntity(event)
	poleTableToRemoveIndex = -1
	
	if event.entity.name == "robot-charger-pole" then
		for index,poleTable in pairs(global.robotChargerPoles) do
			if poleTable[0] == event.entity then
				poleTableToRemoveIndex = index
				break
			end
		end
		
		if poleTableToRemoveIndex ~= -1 then
			table.remove(global.robotChargerPoles, poleTableToRemoveIndex)
			else
			game.players[1].print("Couldn't find pole to destroy")
		end
	end
end

function Load(event)
	if global.robotChargerPoles == nil then
		global.robotChargerPoles = {}
	end
end

script.on_event(defines.events.on_tick, UpdatePoles)

script.on_event(defines.events.on_built_entity, BuiltEntity)
script.on_event(defines.events.on_robot_built_entity, BuiltEntity)

script.on_load(Load)

script.on_event(defines.events.on_preplayer_mined_item, DestroyedEntity)
script.on_event(defines.events.on_robot_pre_mined, DestroyedEntity)
script.on_event(defines.events.on_entity_died, DestroyedEntity)