enumerable_take.jl at 192a00e ⋅ queryverse/QueryOperators.jl

struct EnumerableTake{T,S} <: Enumerable
    source::S
    n::Int
end

function take(source::Enumerable, n::Integer)
    T = eltype(source)
    S = typeof(source)
    return EnumerableTake{T,S}(source, Int(n))
end

Base.IteratorSize(::Type{EnumerableTake{T,S}}) where {T,S} = haslength(S)

Base.eltype(::Type{EnumerableTake{T,S}}) where {T,S} = T

Base.length(iter::EnumerableTake{T,S}) where {T,S} = min(length(iter.source),iter.n)

function Base.iterate(iter::EnumerableTake{T,S}) where {T,S}
    ret = iterate(iter.source)

    if ret===nothing
        return nothing
    elseif iter.n==0
        return nothing
    else
        return ret[1], (ret[2],1)
    end
end

function Base.iterate(iter::EnumerableTake{T,S}, state) where {T,S}
    if state[2]==iter.n
        return nothing
    else
        ret = iterate(iter.source, state[1])

        if ret===nothing
            return nothing
        else
            return ret[1], (ret[2], state[2]+1)
        end
    end
end

1		struct EnumerableTake{T,S} <: Enumerable
2	23	source::S
3		n::Int
4		end
5
6		function take(source::Enumerable, n::Integer)
7	23	T = eltype(source)
8		S = typeof(source)
9	23	return EnumerableTake{T,S}(source, Int(n))
10		end
11
12	0	Base.IteratorSize(::Type{EnumerableTake{T,S}}) where {T,S} = haslength(S)
13
14	0	Base.eltype(::Type{EnumerableTake{T,S}}) where {T,S} = T
15
16	5	Base.length(iter::EnumerableTake{T,S}) where {T,S} = min(length(iter.source),iter.n)
17
18		function Base.iterate(iter::EnumerableTake{T,S}) where {T,S}
19	23	ret = iterate(iter.source)
20
21	23	if ret===nothing
22	0	return nothing
23	23	elseif iter.n==0
24	0	return nothing
25		else
26	23	return ret[1], (ret[2],1)
27		end
28		end
29
30		function Base.iterate(iter::EnumerableTake{T,S}, state) where {T,S}
31	23	if state[2]==iter.n
32	23	return nothing
33		else
34	18	ret = iterate(iter.source, state[1])
35
36	23	if ret===nothing
37	0	return nothing
38		else
39	23	return ret[1], (ret[2], state[2]+1)
40		end
41		end
42		end

Navigation	Overlay
`t` Navigate files	`h` Toggle hits
`y` Change url to tip of branch	`m` Toggle misses
`b / v` Jump to prev/next hit line	`p` Toggle partial
`z / x` Jump to prev/next missed or partial line	`1..9` Toggle flags
`shift + o` Open current page in GitHub	`a` Toggle all on
`/ or ?` Show keyboard shortcuts dialog	`c` Toggle context lines or commits