Factorio Forums

Code: Select all

local DETECTORS_TO_PROCESS = 30

local pollutionDetectors
local pollutionRuntime

local function onConfigChange()
    if not pollutionRuntime then
	if not global.pollutionRuntime then
	    global.pollutionRuntime = {}
	end
	pollutionRuntime = global.pollutionRuntime
    end

    if (pollutionRuntime.version ~= 1) then
	pollutionRuntime.processEvent = (math.ceil(game.tick / 60) + 1) * 60

	print(pollutionRuntime.processEvent)
	pollutionRuntime.queueIndex = 1
	pollutionRuntime.virtualSignalParam = {
	    parameters={
		{
		    index=1,
		    signal={
			type="item",
			name="pollution"
		    },
		    count=-1
		}
	    }
	}
	pollutionRuntime.version = 1
    end
    
end

local function onInit()
    global.pollutionDetectors = {}
    global.pollutionRuntime = {}

    pollutionDetectors = global.pollutionDetectors
    pollutionRuntime = global.pollutionRuntime
    
    onConfigChange()
end

local function onLoad()    
    pollutionDetectors = global.pollutionDetectors
    pollutionRuntime = global.pollutionRuntime
end

-- Logic

local function addPDToTable(event)
    local entity = event.created_entity
    if (entity.name == "pollution-detector") then
	entity.operable = false
	pollutionDetectors[#pollutionDetectors+1] = entity
    end
end

local function removePollutionDetector(event)
    local entity = event.entity
    if (entity.name == "pollution-detector") then
	for i = #pollutionDetectors,1,-1 do
	    local pd = pollutionDetectors[i]
	    if pd.valid then 
		if (entity == pd) then
		    table.remove(pollutionDetectors, i)
		    break
		end
	    else
		table.remove(pollutionDetectors, i)
	    end
	end
    end
end

local function setPollutionValue(entity)
    if (entity.valid) then
	local pollutionSignal = pollutionRuntime.virtualSignalParam
	pollutionSignal.parameters[1].count = math.floor(entity.surface.get_pollution(entity.position))
	entity.get_control_behavior().parameters = pollutionSignal
    end
end

local function onTick(event)
    local tick = event.tick
    if (pollutionRuntime.processEvent == tick) then
	pollutionRuntime.processEvent = pollutionRuntime.processEvent + 60
	local startIndex = pollutionRuntime.queueIndex
	local endIndex = math.min(#pollutionDetectors, startIndex + DETECTORS_TO_PROCESS)
	for i = startIndex, endIndex do
	    setPollutionValue(pollutionDetectors[i])
	end
	if (endIndex == #pollutionDetectors) then
	    pollutionRuntime.queueIndex = 1
	else
	    pollutionRuntime.queueIndex = endIndex
	end
    end
end

-- Events
script.on_init(onInit)

script.on_load(onLoad)

script.on_configuration_changed(onConfigChange)

script.on_event({defines.events.on_built_entity,
		 defines.events.on_robot_built_entity}, addPDToTable)

script.on_event({defines.events.on_player_mined_entity,
		 defines.events.on_robot_mined_entity,
		 defines.events.on_entity_died}, removePollutionDetector)

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