Skip to content

Commit 8a94d20

Browse files
Simplified code
1 parent 32d074e commit 8a94d20

File tree

1 file changed

+3
-8
lines changed

1 file changed

+3
-8
lines changed

src/ThreadedSparseArrays.jl

Lines changed: 3 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -17,15 +17,10 @@ end
1717
1818
Returns an iterator splitting the range `1:n` into `min(k,n)` parts of (almost) equal size.
1919
"""
20-
RangeIterator(n::Int,k::Int) = RangeIterator(min(n,k),divrem(n,k)...)
20+
RangeIterator(n::Int, k::Int) = RangeIterator(min(n,k),divrem(n,k)...)
2121
Base.length(it::RangeIterator) = it.k
22-
#Base.iterate(it::RangeIterator, i::Int=1) = i>it.k ? nothing : (((i-1)*it.d+min(i-1,it.r)+1):(i*it.d+min(i,it.r)), i+1)
23-
function Base.iterate(it::RangeIterator, i::Int=1)
24-
i>it.k && return nothing
25-
e = i*it.d + min(i,it.r)
26-
s = e - it.d + (i>it.r)
27-
(s:e,i+1)
28-
end
22+
endpos(it::RangeIterator, i::Int) = i*it.d+min(i,it.r)
23+
Base.iterate(it::RangeIterator, i::Int=1) = i>it.k ? nothing : (endpos(it,i-1)+1:endpos(it,i), i+1)
2924

3025

3126
# * ThreadedSparseMatrixCSC

0 commit comments

Comments
 (0)