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

added mget for TSet

Browse source
This commit is contained in:
Andreas Rumpf 2014-04-13 00:33:06 +02:00
commit 455c3c19ca
1 changed files with 9 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

9
lib/pure/collections/sets.nim
View file

@ -68,6 +68,15 @@ template rawInsertImpl() {.dirty.} =
proc rawGet[A](s: TSet[A], key: A): int = proc rawGet[A](s: TSet[A], key: A): int =
rawGetImpl() rawGetImpl()
proc mget*[A](s: var TSet[A], key: A): var A =
## returns the element that is actually stored in 's' which has the same
## value as 'key' or raises the ``EInvalidKey`` exception. This is useful
## when one overloaded 'hash' and '==' but still needs reference semantics
## for sharing.
var index = rawGet(s, key)
if index >= 0: result = t.data[index].key
else: raise newException(EInvalidKey, "key not found: " & $key)
proc contains*[A](s: TSet[A], key: A): bool = proc contains*[A](s: TSet[A], key: A): bool =
## returns true iff `key` is in `s`. ## returns true iff `key` is in `s`.
var index = rawGet(s, key) var index = rawGet(s, key)