1
0
mirror of https://github.com/Wyrrrd/Autofilter.git synced 2026-03-06 15:19:22 +01:00
Files
Autofilter/control.lua
2020-02-22 04:11:21 +01:00

47 lines
1.4 KiB
Lua

--control.lua
--functions definitions
local function get_items_by_content(inserter,inventory)
local items = {}
for item,count in pairs(inventory.get_contents()) do
if #items < inserter.filter_slot_count then
items[#items+1] = item
end
end
return items
end
local function get_items_by_filter(inserter,inventory)
local items = {}
for slot = 1,#inventory do
if #items < inserter.filter_slot_count and inventory.get_filter(slot) then
items[#items+1] = inventory.get_filter(slot)
end
end
return items
end
local function on_built_entity(event)
local inserter = event.created_entity
if inserter.filter_slot_count then
local pickup = inserter.surface.find_entities_filtered({position = inserter.pickup_position, limit = 1})
if #pickup > 0 then
local inventory = pickup[1].get_output_inventory()
if inventory then
local mode = game.players[event.player_index].mod_settings["Autofilter_Mode"].value
if mode == "content" and not inventory.is_empty() then
for slot,item in pairs(get_items_by_content(inserter,inventory)) do
inserter.set_filter(slot,item)
end
elseif mode == "filter" and inventory.is_filtered() then
for slot,item in pairs(get_items_by_filter(inserter,inventory)) do
inserter.set_filter(slot,item)
end
end
end
end
end
end
--event handling
script.on_event(defines.events.on_built_entity, on_built_entity,{{filter="type", type = "inserter"}})