Factorio Forums

Code: Select all

-- circuit-handling utilities
--
function ckey(e,wire,post)
    local pos=e.position
    return pos.x/2^21+pos.y/2^42 + (post or 0)/2^44 + (wire or 0)/2^46
    end
--
function iterate_copper_network(e)
    local ret, seen = {e}, {[ckey(e)]=1}
    return function()
        local r,k = ret[#ret] ret[#ret]=nil
        if r then
            for _,n in next,r.neighbours.copper do
                k = ckey(n)
                if not seen[k] then seen[k]=1 ret[#ret+1]=n end
                end
            end
        return r
        end
    end
-- arith/logic output is somewhat arbitrarily default initial exit
circuit={['arithmetic-combinator']=2, ['decider-combinator']=2}
function iterate_network(e,w)
    if w=='copper' then return iterate_copper_network(e) end
    local wire = defines.wire_type[w]
    local ret, seen = {{e=e, post=circuit[e.type] or 1}}, {[ckey(e,wire,post)]=1}
    return function()
        local r,k = ret[#ret] ret[#ret]=nil
        if r then
            for _,n in next,r.e.circuit_connection_definitions or {} do
                if n.wire == wire and n.source_circuit_id == r.post then
                    local e,post = n.target_entity, n.target_circuit_id
                    k = ckey(e,wire,post)
                    if not seen[k] then
                        seen[k]=1
                        ret[#ret+1] = {e=e, post=n.target_circuit_id}
                        end
                    end
                end
            return r.e
            end
        return nil
        end
    end
-- ease dealing with circuit signals
function read_signals(e,w,c)
    local r = {item={}, fluid={}, virtual={}}
    local n = e.get_circuit_network(
            defines.wire_type[w], c or circuit[e.type] or 1)
    for _,s in next,n and n.signals or {} do
        r[s.signal.type][s.signal.name] = s.count
        end
    return r
    end