BLI: add Array.last method

This makes it consistent with Vector and Span.

2 files changed, 24 insertions, 0 deletions

diff --git a/source/blender/blenlib/BLI_array.hh b/source/blender/blenlib/BLI_array.hh
index 0ffe6b9a750..d6b7ab03203 100644
--- a/source/blender/blenlib/BLI_array.hh
+++ b/source/blender/blenlib/BLI_array.hh
@@ -269,6 +269,21 @@ class Array {
   }
 
   /**
+   * Return a reference to the last element in the array.
+   * This invokes undefined behavior when the array is empty.
+   */
+  const T &last() const
+  {
+    BLI_assert(size_ > 0);
+    return *(data_ + size_ - 1);
+  }
+  T &last()
+  {
+    BLI_assert(size_ > 0);
+    return *(data_ + size_ - 1);
+  }
+
+  /**
    * Get a pointer to the beginning of the array.
    */
   const T *data() const
diff --git a/source/blender/blenlib/tests/BLI_array_test.cc b/source/blender/blenlib/tests/BLI_array_test.cc
index 251cff833f7..3d45a9f5277 100644
--- a/source/blender/blenlib/tests/BLI_array_test.cc
+++ b/source/blender/blenlib/tests/BLI_array_test.cc
@@ -227,4 +227,13 @@ TEST(array, MoveAssignmentExceptions)
   EXPECT_ANY_THROW({ array_moved = std::move(array); });
 }
 
+TEST(array, Last)
+{
+  Array<int> array = {5, 7, 8, 9};
+  EXPECT_EQ(array.last(), 9);
+  array.last() = 1;
+  EXPECT_EQ(array[3], 1);
+  EXPECT_EQ(const_cast<const Array<int> &>(array).last(), 1);
+}
+
 }  // namespace blender::tests