Factorio-Paranoidal_mod/even-distribution/scripts/util.lua

local util = {}

function util.doEvery(tick, func, args)
	if (game.tick % tick) == 0 then func(args) end
end

-- remove leading and trailing whitespaces
function util.trim(str)
  return str and (str:gsub("^%s*(.-)%s*$", "%1")) or ""
end

-- trim and also remove multiple whitespaces
function util.fullTrim(str)
  return (util.trim(str):gsub("[ \t\r\n]*[\r\n][ \t\r\n]*", "\r\n"):gsub("[ \t]+", " "))
end

-- escape string for use with regex
local quotepattern = '(['..("%^$().[]*+-?"):gsub("(.)", "%%%1")..'])'
function util.escape(str)
    return str:gsub(quotepattern, "%%%1")
end

function util.isValid(object)
	return object and object.valid
end

function util.destroyIfValid(object)
	if util.isValid(object) then object.destroy() end
end

function util.isValidStack(stack)
	return util.isValid(stack) and stack.valid_for_read
end

function util.isValidPlayer(player) -- valid, connected and alive player
	return util.isValid(player) and 
		   player.connected and (
		       player.controller_type == defines.controllers.character or
		       player.controller_type == defines.controllers.god or
		       player.controller_type == defines.controllers.editor
		   )
end

function util.isCraftingMachine(entity)
	return entity.type == "furnace" or entity.type == "assembling-machine" or entity.type == "rocket-silo"
end

function util.enablesRobots(tech)
	local recipes = game.recipe_prototypes
	local items   = game.item_prototypes

	for __,effect in pairs(tech.effects) do
		if effect.type == "unlock-recipe" then
			local recipe = recipes[effect.recipe]
			for __,product in pairs(recipe.products) do
				if product.type == "item" then 
					local entity = items[product.name].place_result
					if entity and entity.type == "roboport" then
						return true
					end
				end
			end
		end
	end

	return false
end

function util.hasRobotsEnabled(force)
	for __,tech in pairs(force.technologies) do
		if tech.researched and util.enablesRobots(tech) then return true end
	end

	return false
end

function util.shallowCopy(original) -- Creates a shallow copy of a table
    local copy = {}
    for key,value in pairs(original) do copy[key] = value  end
    return copy
end

function util.hasInventory(prototype)
	for __,index in pairs(defines.inventory) do
		if prototype.get_inventory_size(index) then return true end
	end
	return false
end

function util.countTable(tbl)
	local count = 0
	for __ in pairs(tbl) do count = count + 1 end
	return count
end

function util.isEmpty(tbl) -- empty table
	return type(tbl) == "table" and next(tbl) == nil
end

function util.isFilled(tbl) -- filled table
	return type(tbl) == "table" and next(tbl) ~= nil
end

function util.distribute(entities, totalItems, func)
	local insertAmount = math.floor(totalItems / #entities)
	local remainder = totalItems % #entities
	
	for entity in util.epairs(entities) do
		if util.isValid(entity) then
			local amount = insertAmount
			if remainder > 0 then
				amount = amount + 1
				remainder = remainder - 1
			end

			func(entity, amount)
		end
	end
end

function util.epairs(tbl) -- iterator for tables with entity based indices
	local id, start
	local surface, x, y
	local tbls, tblsx, value
	
	local tblId = rawget(tbl, "id")
	local tblPos = rawget(tbl, "pos")
	if tblId then
		start = true
	elseif not tblPos then
		return function () end
	end
	
	return function () 		-- Iterator
		if id or start then
			start = false
			id, value = next(tblId, id)
			
			if value == nil then
				if not tblPos then return end
				surface, tbls = next(tblPos)
				if not tbls then return end
				x, tblsx = next(tbls)
			else
				return value
			end
		end
		
		while value == nil do
			y, value = next(tblsx, y)
			
			if value == nil then
				x, tblsx = next(tbls, x)
				y = nil
				
				if not tblsx then
					surface, tbls = next(tblPos, surface)
					if not tbls then return end
					x, tblsx = next(tbls, x)
					y = nil
				end
			end
		end
		
		return value
	end
end

return util