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

Add setutils.complement, setutils.fullSet (#17066)

Browse source
This commit is contained in:
Jason Beetham 2021-02-18 18:33:28 -07:00 • committed by GitHub
commit 35ded02074
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 55 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

25
lib/std/setutils.nim
View file

@ -14,7 +14,7 @@
## * `std/packedsets <packedsets.html>`_
## * `std/sets <sets.html>`_
import std/typetraits
import std/[typetraits, macros]
#[
type SetElement* = char|byte|bool|int16|uint16|enum|uint8|int8
@ -35,3 +35,26 @@ template toSet*(iter: untyped): untyped =
for x in iter:
incl(result, x)
result
macro enmRange(enm: typed): untyped = result = newNimNode(nnkCurly).add(enm.getType[1][1..^1])
proc fullSet*(T: typedesc): auto {.inline.} =
## Returns a full set of all valid elements.
runnableExamples:
assert {true, false} == fullSet(bool)
when T is Ordinal:
{T.low..T.high}
else: # Hole filled enum
enmRange(T)
proc complement*[T](s: set[T]): set[T] {.inline.} =
## Returns the complement of the set.
## Can also be thought of as inverting the set.
runnableExamples:
type Colors = enum
red, green = 3, blue
assert complement({red, blue}) == {green}
assert complement({red, green, blue}).card == 0
assert complement({range[0..10](0), 1, 2, 3}) == {range[0..10](4), 5, 6, 7, 8, 9, 10}
assert complement({'0'..'9'}) == {0.char..255.char} - {'0'..'9'}
fullSet(T) - s