34 lines
1.1 KiB
Nim
34 lines
1.1 KiB
Nim
import std/sets, ../objects, ../templateutils
|
|
|
|
when not declared(initHashSet):
|
|
template initHashSet*[T](initialSize = 64): auto =
|
|
initSet[T](initialSize)
|
|
|
|
template init*[T](_: type HashSet[T]): auto = initHashSet[T]()
|
|
template init*[T](_: type HashSet[T], defaultSize: int): auto = initHashSet[T](defaultSize)
|
|
|
|
# TODO: This should work, but Nim can't handle it
|
|
# template init*[T](_: type HashSet[T], args: varargs[untyped]): auto = initHashSet[T](args)
|
|
|
|
template init*[T](_: type OrderedSet[T]): auto = initOrderedSet[T]()
|
|
template init*[T](_: type OrderedSet[T], initialSize: int): auto = initOrderedSet[T](initialSize)
|
|
# template init*[T](_: type OrderedSet[T], args: varargs[untyped]): auto = initOrderedSet[T](args)
|
|
|
|
template init*[T](_: type set[T]): auto =
|
|
var x: set[T]
|
|
x
|
|
|
|
template keepItIf*[T](setParam: var HashSet[T], itPredicate: untyped) =
|
|
bind evalTemplateParamOnce
|
|
evalTemplateParamOnce(setParam, s):
|
|
var itemsToDelete: seq[T]
|
|
|
|
for it {.inject.} in s:
|
|
if not itPredicate:
|
|
itemsToDelete.add(it)
|
|
|
|
for item in itemsToDelete:
|
|
s.excl item
|
|
|
|
export sets, objects
|