Factorio Forums

Code: Select all

require "util"
--local game_ready = true -- this is sure desync and would also cause init_energy on each game load

-- search exsisting energy things
function init_energy() 
	global.energy = {}
	global.energy_sources = {"generator", "solar-panel", "reactor", "accumulator"}
	for _, source in ipairs(global.energy_sources) do
		global.energy[source] = {}
		for _,surf in pairs(game.surfaces) do
			local ents = surf.find_entities_filtered({type = source})
			for _, ent in pairs(ents) do
				global.energy[source][ent.unit_number] = ent--you don't need position, besides that can be changed
			end
		end
	end
end

function add_energy_entity(entity)
	if global.energy_sources[entity.type] then--you already hava a dictionary, no need to search through it
		global.energy[entity.type][entity.unit_number] = entity
	end
end
function delete_energy_entity(entity)
	if global.energy_sources[entity.type] then
		global.energy[entity.type][entity.unit_number] = nil
	end
end
function get_entitys_energy(source)
	local cur = 0
	local max = 0
	for _, ent in pairs(global.energy[source]) do
		cur = cur + ent.energy
		max = max + ent.electric_buffer_size
	end
	local energy = {cur = cur, max = max}
	return energy
end