mirror/zig
1
0
Fork 0
mirror of https://github.com/ziglang/zig.git synced 2025-12-16 11:13:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

std.enums.IndexedSet: Add initOne and initMany

Browse Source
This commit is contained in:
Jay Petacat 2023-03-25 00:03:22 -06:00 committed by Veikka Tuominen
parent 49e33a2f23
commit fcc86832d6
1 changed files with 14 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
lib/std/enums.zig
View File

@ -775,6 +775,18 @@ pub fn IndexedSet(comptime I: type, comptime Ext: fn (type) type) type {
return .{ .bits = BitSet.initFull() }; return .{ .bits = BitSet.initFull() };
} }
/// Returns a set containing multiple keys.
pub fn initMany(keys: []const Key) Self {
var set = initEmpty();
for (keys) |key| set.insert(key);
return set;
}
/// Returns a set containing a single key.
pub fn initOne(key: Key) Self {
return initMany(&[_]Key{key});
}
/// Returns the number of keys in the set. /// Returns the number of keys in the set.
pub fn count(self: Self) usize { pub fn count(self: Self) usize {
return self.bits.count(); return self.bits.count();
@ -900,20 +912,8 @@ test "pure EnumSet fns" {
const empty = EnumSet(Suit).initEmpty(); const empty = EnumSet(Suit).initEmpty();
const full = EnumSet(Suit).initFull(); const full = EnumSet(Suit).initFull();
const black = EnumSet(Suit).initMany(&[_]Suit{ .spades, .clubs });
const black = black: { const red = EnumSet(Suit).initMany(&[_]Suit{ .hearts, .diamonds });
var set = EnumSet(Suit).initEmpty();
set.insert(.spades);
set.insert(.clubs);
break :black set;
};
const red = red: {
var set = EnumSet(Suit).initEmpty();
set.insert(.hearts);
set.insert(.diamonds);
break :red set;
};
try testing.expect(empty.eql(empty)); try testing.expect(empty.eql(empty));
try testing.expect(full.eql(full)); try testing.expect(full.eql(full));