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

std.MultiArrayList: add set and get to Slice

Browse Source
This commit is contained in:
Jacob Young 2023-03-26 03:27:35 -04:00
parent 86625c5a75
commit 0e5e001278
1 changed files with 16 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
lib/std/multi_array_list.zig
View File

@ -49,6 +49,20 @@ pub fn MultiArrayList(comptime S: type) type {
return casted_ptr[0..self.len]; return casted_ptr[0..self.len];
} }
pub fn set(self: Slice, index: usize, elem: S) void {
inline for (fields) |field_info| {
self.items(@field(Field, field_info.name))[index] = @field(elem, field_info.name);
}
}
pub fn get(self: Slice, index: usize) S {
var elem: S = undefined;
inline for (fields) |field_info| {
@field(elem, field_info.name) = self.items(@field(Field, field_info.name))[index];
}
return elem;
}
pub fn toMultiArrayList(self: Slice) Self { pub fn toMultiArrayList(self: Slice) Self {
if (self.ptrs.len == 0) { if (self.ptrs.len == 0) {
return .{}; return .{};
@ -156,20 +170,12 @@ pub fn MultiArrayList(comptime S: type) type {
/// Overwrite one array element with new data. /// Overwrite one array element with new data.
pub fn set(self: *Self, index: usize, elem: S) void { pub fn set(self: *Self, index: usize, elem: S) void {
const slices = self.slice(); return self.slice().set(index, elem);
inline for (fields, 0..) |field_info, i| {
slices.items(@intToEnum(Field, i))[index] = @field(elem, field_info.name);
}
} }
/// Obtain all the data for one array element. /// Obtain all the data for one array element.
pub fn get(self: Self, index: usize) S { pub fn get(self: Self, index: usize) S {
const slices = self.slice(); return self.slice().get(index);
var result: S = undefined;
inline for (fields, 0..) |field_info, i| {
@field(result, field_info.name) = slices.items(@intToEnum(Field, i))[index];
}
return result;
} }
/// Extend the list by 1 element. Allocates more memory as necessary. /// Extend the list by 1 element. Allocates more memory as necessary.