motiejus/zig
1
Fork 0
Code Releases Activity

Merge pull request #7634 from daurnimator/packed-init-all-to

Browse Source 
Add PackedIntArray .initAllTo function
This commit is contained in:
Andrew Kelley
2021-01-01 13:23:18 -08:00
committed by GitHub
parent dfacac916f 03e2c53747
commit 4cc374c639
1 changed files with 25 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
lib/std/packed_int_array.zig
View File

@@ -203,6 +203,14 @@ pub fn PackedIntArrayEndian(comptime Int: type, comptime endian: builtin.Endian,
return self;
}
///Initialize all entries of a packed array to the same value
pub fn initAllTo(int: Int) Self {
// TODO: use `var self = @as(Self, undefined);` https://github.com/ziglang/zig/issues/7635
var self = Self{ .bytes = [_]u8{0} ** total_bytes };
self.setAll(int);
return self;
}
///Return the Int stored at index
pub fn get(self: Self, index: usize) Int {
debug.assert(index < int_count);
@@ -215,6 +223,14 @@ pub fn PackedIntArrayEndian(comptime Int: type, comptime endian: builtin.Endian,
return Io.set(&self.bytes, index, 0, int);
}
///Set all entries of a packed array to the same value
pub fn setAll(self: *Self, int: Int) void {
var i: usize = 0;
while (i < int_count) : (i += 1) {
self.set(i, int);
}
}
///Create a PackedIntSlice of the array from given start to given end
pub fn slice(self: *Self, start: usize, end: usize) PackedIntSliceEndian(Int, endian) {
debug.assert(start < int_count);
@@ -365,7 +381,15 @@ test "PackedIntArray init" {
const PackedArray = PackedIntArray(u3, 8);
var packed_array = PackedArray.init([_]u3{ 0, 1, 2, 3, 4, 5, 6, 7 });
var i = @as(usize, 0);
while (i < packed_array.len()) : (i += 1) testing.expect(packed_array.get(i) == i);
while (i < packed_array.len()) : (i += 1) testing.expectEqual(@intCast(u3, i), packed_array.get(i));
}
test "PackedIntArray initAllTo" {
if (we_are_testing_this_with_stage1_which_leaks_comptime_memory) return error.SkipZigTest;
const PackedArray = PackedIntArray(u3, 8);
var packed_array = PackedArray.initAllTo(5);
var i = @as(usize, 0);
while (i < packed_array.len()) : (i += 1) testing.expectEqual(@as(u3, 5), packed_array.get(i));
}
test "PackedIntSlice" {