jyapayne/Nim
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Do we want we to use the typesystem like this?

Browse source
This commit is contained in:
Hans Raaf 2015-03-06 16:01:05 +01:00
commit 9e2f79251c
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
lib/pure/collections/sequtils.nim
View file

@ -104,7 +104,7 @@ proc zip*[S, T](seq1: seq[S], seq2: seq[T]): seq[tuple[a: S, b: T]] =
newSeq(result, m) newSeq(result, m)
for i in 0 .. m-1: result[i] = (seq1[i], seq2[i]) for i in 0 .. m-1: result[i] = (seq1[i], seq2[i])
proc distribute*[T](s: seq[T], num: int, spread = true): seq[seq[T]] = proc distribute*[T](s: seq[T], num: Positive, spread = true): seq[seq[T]] =
## Splits and distributes a sequence `s` into `num` sub sequences. ## Splits and distributes a sequence `s` into `num` sub sequences.
## ##
## Returns a sequence of `num` sequences. For some input values this is the ## Returns a sequence of `num` sequences. For some input values this is the
@ -131,11 +131,12 @@ proc distribute*[T](s: seq[T], num: int, spread = true): seq[seq[T]] =
## assert numbers.distribute(6)[0] == @[1, 2] ## assert numbers.distribute(6)[0] == @[1, 2]
## assert numbers.distribute(6)[5] == @[7] ## assert numbers.distribute(6)[5] == @[7]
assert(not s.isNil, "`s` can't be nil") assert(not s.isNil, "`s` can't be nil")
assert(num > 0, "`num` has to be greater than zero")
if num < 2: if num < 2:
result = @[s] result = @[s]
return return
let num = cast[int](num)
# Create the result and calculate the stride size and the remainder if any. # Create the result and calculate the stride size and the remainder if any.
result = newSeq[seq[T]](num) result = newSeq[seq[T]](num)
var var