From 1de96a2cc4bf2d1be1404690d6a867eb0127d533 Mon Sep 17 00:00:00 2001 From: Chris Boesch <48591413+chrboesch@users.noreply.github.com> Date: Thu, 29 Dec 2022 00:24:57 +0100 Subject: [PATCH] Add the two functions 'getLast' and 'getLastOrNull' to ArrayListAligned/ArrayListAlignedUnmanaged. --- lib/std/array_list.zig | 28 ++++++++++++++++++++++++++++ 1 file changed, 28 insertions(+) diff --git a/lib/std/array_list.zig b/lib/std/array_list.zig index c6df2a66eb..852c81f140 100644 --- a/lib/std/array_list.zig +++ b/lib/std/array_list.zig @@ -468,6 +468,20 @@ pub fn ArrayListAligned(comptime T: type, comptime alignment: ?u29) type { pub fn unusedCapacitySlice(self: Self) Slice { return self.allocatedSlice()[self.items.len..]; } + + /// Return the last element from the list. + /// Asserts the list has at least one item. + pub fn getLast(self: *Self) T { + const val = self.items[self.items.len - 1]; + return val; + } + + /// Return the last element from the list, or + /// return `null` if list is empty. + pub fn getLastOrNull(self: *Self) ?T { + if (self.items.len == 0) return null; + return self.getLast(); + } }; } @@ -913,6 +927,20 @@ pub fn ArrayListAlignedUnmanaged(comptime T: type, comptime alignment: ?u29) typ pub fn unusedCapacitySlice(self: Self) Slice { return self.allocatedSlice()[self.items.len..]; } + + /// Return the last element from the list. + /// Asserts the list has at least one item. + pub fn getLast(self: *Self) T { + const val = self.items[self.items.len - 1]; + return val; + } + + /// Return the last element from the list, or + /// return `null` if list is empty. + pub fn getLastOrNull(self: *Self) ?T { + if (self.items.len == 0) return null; + return self.getLast(); + } }; }