Factorio Forums

Code: Select all

-- data-final-fixes.lua

local function get_item(name)
    for typeName in pairs(defines.prototypes.item) do
        local items = data.raw[typeName]
        if items and items[name] then
            return items[name]
        end
    end
    return
end

local function minimum_slots_for_harvest(tree)
    if not tree.minable then
        return 0
    end

    if tree.minable.results then
        local result_slots = 0
        for _, result in pairs(tree.minable.results) do
            if result.type == "item" then
                local amount = result.amount
                if amount == nil then amount = result.amount_max end
                result_slots = result_slots + math.ceil(amount / get_item(result.name).stack_size)
            end
        end
        return result_slots
    end

    if tree.minable.result then
        return math.ceil(tree.minable.count / get_item(tree.minable.result).stack_size)
    end

    return 0 -- No results
end

-- Calculate maximum number of slots to harvest any possible trees
local default_output_size = 1

for _, tree in pairs(data.raw["tree"]) do
    default_output_size = math.max(default_output_size, minimum_slots_for_harvest(tree))
end

for _, plant in pairs(data.raw["plant"]) do
    default_output_size = math.max(default_output_size, minimum_slots_for_harvest(plant))
end

for _, entity in pairs(data.raw["agricultural-tower"]) do
    if entity.output_inventory_size == nil then
        entity.output_inventory_size = default_output_size
    end
end