mirrors/Nim
1
0
Fork 0
mirror of https://github.com/nim-lang/Nim.git synced 2026-04-18 21:40:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1403 from def-/newseqwith

Browse Source 
Add newSeqWith
This commit is contained in:
Andreas Rumpf
2014-08-12 20:00:25 +02:00
parent 15eea198be 61a6ecfd94
commit b1bfd51200
1 changed files with 24 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
lib/pure/collections/sequtils.nim
View File

@@ -409,6 +409,23 @@ template mapIt*(varSeq, pred: expr) =
let it {.inject.} = varSeq[i]
varSeq[i] = pred
template newSeqWith*(len: int, init: expr): expr =
## creates a new sequence, calling `init` to initialize each value. Example:
##
## .. code-block:: nimrod
## var seq2D = newSeqWith(20, newSeq[bool](10))
## seq2D[0][0] = true
## seq2D[1][0] = true
## seq2D[0][1] = true
##
## import math
## var seqRand = newSeqWith(20, random(10))
## echo seqRand
var result {.gensym.} = newSeq[type(init)](len)
for i in 0 .. <len:
result[i] = init
result
when isMainModule:
import strutils
block: # concat test
@@ -557,4 +574,11 @@ when isMainModule:
doAssert b.distribute(5, true)[4].len == 5
doAssert b.distribute(5, false)[4].len == 2
block: # newSeqWith tests
var seq2D = newSeqWith(4, newSeq[bool](2))
seq2D[0][0] = true
seq2D[1][0] = true
seq2D[0][1] = true
doAssert seq2D == @[@[true, true], @[true, false], @[false, false], @[false, false]]
echo "Finished doc tests"