tiny.simd.BitSet

1pub fn BitSet(comptime size: usize) type {2    requireCapacity(size);3    const word_count = wordsFor(size);45    return struct {6        words: [word_count]BitSet64 = @as([word_count]BitSet64, @splat(.{})),78        const Self = @This();9        pub const capacity: usize = size;10        pub const max_size: usize = capacity;1112        pub fn maxSize(_: *const Self) usize {13            return max_size;14        }1516        pub fn set(self: *Self, index: usize) void {17            std.debug.assert(index < max_size);18            self.words[wordIndex(index)].set(bitIndex(index));19            std.debug.assert(self.get(index));20        }2122        pub fn clear(self: *Self, index: usize) void {23            std.debug.assert(index < max_size);24            self.words[wordIndex(index)].clear(bitIndex(index));25            std.debug.assert(!self.get(index));26        }2728        pub fn get(self: *const Self, index: usize) bool {29            std.debug.assert(index < max_size);30            return self.words[wordIndex(index)].get(bitIndex(index));31        }3233        pub fn any(self: *const Self) bool {34            for (&self.words) |*word| {35                if (word.any()) return true;36            }37            return false;38        }3940        pub fn all(self: *const Self) bool {41            for (self.words[0 .. word_count - 1]) |word| {42                if (!word.all()) return false;43            }44            const remainder = max_size % 64;45            return if (remainder == 0)46                self.words[word_count - 1].all()47            else48                self.words[word_count - 1].count() == remainder;49        }5051        pub fn first(self: *const Self) usize {52            std.debug.assert(self.any());53            for (&self.words, 0..) |*word, word_index| {54                if (word.any()) return word_index * 64 + word.first();55            }56            unreachable;57        }5859        pub fn first0(self: *const Self) usize {60            std.debug.assert(!self.all());61            for (&self.words, 0..) |*word, word_index| {62                if (!word.all()) {63                    const index = word_index * 64 + word.first0();64                    std.debug.assert(index < max_size);65                    return index;66                }67            }68            unreachable;69        }7071        pub fn foreach(self: *const Self, function: anytype) void {72            for (&self.words, 0..) |*word, word_index| {73                visitWord(word.get64(), word_index * 64, function);74            }75        }7677        pub fn count(self: *const Self) usize {78            var total: usize = 0;79            for (&self.words) |*word| total += word.count();80            std.debug.assert(total <= max_size);81            return total;82        }83    };84}

1pub const BitSet = bitset.BitSet;