summary refs log tree commit diff stats
path: root/lib/system/sets.nim
blob: f9f3eb32b634db05d48f712e590d0535cf2ae412 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
#
#
#            Nimrod's Runtime Library
#        (c) Copyright 2009 Andreas Rumpf
#
#    See the file "copying.txt", included in this
#    distribution, for details about the copyright.
#

# set handling

type
  TNimSet = array [0..4*2048-1, int8]

proc countBits32(n: int32): int {.compilerproc.} =
  var v = n
  v = v -% ((v shr 1'i32) and 0x55555555'i32)
  v = (v and 0x33333333'i32) +% ((v shr 2'i32) and 0x33333333'i32)
  result = ((v +% (v shr 4'i32) and 0xF0F0F0F'i32) *% 0x1010101'i32) shr 24'i32

proc countBits64(n: int64): int {.compilerproc.} = 
  result = countBits32(toU32(n and 0xffff'i64)) +
           countBits32(toU32(n shr 16'i64))

proc cardSet(s: TNimSet, len: int): int {.compilerproc.} =
  result = 0
  for i in countup(0, len-1):
    inc(result, countBits32(int32(ze(s[i]))))