OC Parallel Library

local list = {}
coroutine = require("coroutine")
os = require("os")

function list.create( first, ... )
        if first ~= nil then
            if type( first ) ~= "function" then
                error( "Expected function, got "..type( first ), 3 )
        end
                return coroutine.create(first), list.create( ... )
    end
    return nil
end

local function runUntilLimit( _routines, _limit )
    local count = #_routines
    local living = count

    local tFilters = {}
    local eventData = {}
    while true do
        for n=1,count do
                local r = _routines[n]
                if r then
                        if tFilters[r] == nil or tFilters[r] == eventData[1] or eventData[1] == "terminate" then
                                local ok, param = coroutine.resume( r, table.unpack(eventData) )
                                        if not ok then
                                                error( param, 0 )
                                        else
                                                tFilters[r] = param
                                        end
                                        if coroutine.status( r ) == "dead" then
                                                _routines[n] = nil
                                                living = living - 1
                                                if living <= _limit then
                                                        return n
                                                end
                                        end
                                end
                end
        end
                for n=1,count do
                local r = _routines[n]
                        if r and coroutine.status( r ) == "dead" then
                                _routines[n] = nil
                                living = living - 1
                                if living <= _limit then
                                        return n
                                end
                        end
                end
        eventData = { os.pullEventRaw() }
    end
end

function list.waitForAny( ... )
    local routines = { list.create( ... ) }
    return runUntilLimit( routines, #routines - 1 )
end

function list.waitForAll( ... )
    local routines = { list.create( ... ) }
        runUntilLimit( routines, 0 )

end

return list --return the list of functions