| ... | ... | @@ -41,8 +41,8 @@ pub fn AlignedArrayList(comptime T: type, comptime A: u29) type { |
| 41 | 41 | return self.items[0..self.len]; |
| 42 | 42 | } |
| 43 | 43 | |
| 44 | | pub fn at(self: Self, n: usize) T { |
| 45 | | return self.toSliceConst()[n]; |
| 44 | pub fn at(self: Self, i: usize) T { |
| 45 | return self.toSliceConst()[i]; |
| 46 | 46 | } |
| 47 | 47 | |
| 48 | 48 | /// Sets the value at index `i`, or returns `error.OutOfBounds` if |
| ... | ... | @@ -102,6 +102,26 @@ pub fn AlignedArrayList(comptime T: type, comptime A: u29) type { |
| 102 | 102 | new_item_ptr.* = item; |
| 103 | 103 | } |
| 104 | 104 | |
| 105 | /// Removes the element at the specified index and returns it. |
| 106 | /// The empty slot is filled from the end of the list. |
| 107 | pub fn swapRemove(self: *Self, i: usize) T { |
| 108 | if (self.len - 1 == i) return self.pop(); |
| 109 | |
| 110 | const slice = self.toSlice(); |
| 111 | const old_item = slice[i]; |
| 112 | slice[i] = self.pop(); |
| 113 | return old_item; |
| 114 | } |
| 115 | |
| 116 | pub fn removeOrError(self: *Self, n: usize) !T { |
| 117 | if (n >= self.len) return error.OutOfBounds; |
| 118 | if (self.len - 1 == n) return self.pop(); |
| 119 | |
| 120 | var old_item = self.at(n); |
| 121 | try self.setOrError(n, self.pop()); |
| 122 | return old_item; |
| 123 | } |
| 124 | |
| 105 | 125 | pub fn appendSlice(self: *Self, items: []align(A) const T) !void { |
| 106 | 126 | try self.ensureCapacity(self.len + items.len); |
| 107 | 127 | mem.copy(T, self.items[self.len..], items); |
| ... | ... | @@ -232,6 +252,33 @@ test "basic ArrayList test" { |
| 232 | 252 | assert(list.pop() == 33); |
| 233 | 253 | } |
| 234 | 254 | |
| 255 | test "std.ArrayList.swapRemove" { |
| 256 | var list = ArrayList(i32).init(debug.global_allocator); |
| 257 | defer list.deinit(); |
| 258 | |
| 259 | try list.append(1); |
| 260 | try list.append(2); |
| 261 | try list.append(3); |
| 262 | try list.append(4); |
| 263 | try list.append(5); |
| 264 | try list.append(6); |
| 265 | try list.append(7); |
| 266 | |
| 267 | //remove from middle |
| 268 | assert(list.swapRemove(3) == 4); |
| 269 | assert(list.at(3) == 7); |
| 270 | assert(list.len == 6); |
| 271 | |
| 272 | //remove from end |
| 273 | assert(list.swapRemove(5) == 6); |
| 274 | assert(list.len == 5); |
| 275 | |
| 276 | //remove from front |
| 277 | assert(list.swapRemove(0) == 1); |
| 278 | assert(list.at(0) == 5); |
| 279 | assert(list.len == 4); |
| 280 | } |
| 281 | |
| 235 | 282 | test "iterator ArrayList test" { |
| 236 | 283 | var list = ArrayList(i32).init(debug.global_allocator); |
| 237 | 284 | defer list.deinit(); |
Andrew Kelley <andrew@ziglang.org> 2018-07-14 10:02:13-04:00