| ... | ... | @@ -203,6 +203,14 @@ pub fn PackedIntArrayEndian(comptime Int: type, comptime endian: builtin.Endian, |
| 203 | 203 | return self; |
| 204 | 204 | } |
| 205 | 205 | |
| 206 | ///Initialize all entries of a packed array to the same value |
| 207 | pub fn initAllTo(int: Int) Self { |
| 208 | // TODO: use `var self = @as(Self, undefined);` https://github.com/ziglang/zig/issues/7635 |
| 209 | var self = Self{ .bytes = [_]u8{0} ** total_bytes }; |
| 210 | self.setAll(int); |
| 211 | return self; |
| 212 | } |
| 213 | |
| 206 | 214 | ///Return the Int stored at index |
| 207 | 215 | pub fn get(self: Self, index: usize) Int { |
| 208 | 216 | debug.assert(index < int_count); |
| ... | ... | @@ -215,6 +223,14 @@ pub fn PackedIntArrayEndian(comptime Int: type, comptime endian: builtin.Endian, |
| 215 | 223 | return Io.set(&self.bytes, index, 0, int); |
| 216 | 224 | } |
| 217 | 225 | |
| 226 | ///Set all entries of a packed array to the same value |
| 227 | pub fn setAll(self: *Self, int: Int) void { |
| 228 | var i: usize = 0; |
| 229 | while (i < int_count) : (i += 1) { |
| 230 | self.set(i, int); |
| 231 | } |
| 232 | } |
| 233 | |
| 218 | 234 | ///Create a PackedIntSlice of the array from given start to given end |
| 219 | 235 | pub fn slice(self: *Self, start: usize, end: usize) PackedIntSliceEndian(Int, endian) { |
| 220 | 236 | debug.assert(start < int_count); |
| ... | ... | @@ -365,7 +381,15 @@ test "PackedIntArray init" { |
| 365 | 381 | const PackedArray = PackedIntArray(u3, 8); |
| 366 | 382 | var packed_array = PackedArray.init([_]u3{ 0, 1, 2, 3, 4, 5, 6, 7 }); |
| 367 | 383 | var i = @as(usize, 0); |
| 368 | | while (i < packed_array.len()) : (i += 1) testing.expect(packed_array.get(i) == i); |
| 384 | while (i < packed_array.len()) : (i += 1) testing.expectEqual(@intCast(u3, i), packed_array.get(i)); |
| 385 | } |
| 386 | |
| 387 | test "PackedIntArray initAllTo" { |
| 388 | if (we_are_testing_this_with_stage1_which_leaks_comptime_memory) return error.SkipZigTest; |
| 389 | const PackedArray = PackedIntArray(u3, 8); |
| 390 | var packed_array = PackedArray.initAllTo(5); |
| 391 | var i = @as(usize, 0); |
| 392 | while (i < packed_array.len()) : (i += 1) testing.expectEqual(@as(u3, 5), packed_array.get(i)); |
| 369 | 393 | } |
| 370 | 394 | |
| 371 | 395 | test "PackedIntSlice" { |
Andrew Kelley <andrew@ziglang.org> 2021-01-01 13:23:18-08:00
GitHub <noreply@github.com> 2021-01-01 13:23:18-08:00
Signed by PGP key 4AEE18F83AFDEB23
lib