From 7b2becfb2189d6587b01b19c53b836b9658b1b1b Mon Sep 17 00:00:00 2001
From: Jakob Stoklund Olesen <jolesen@mozilla.com>
Date: Tue, 11 Apr 2017 16:03:49 -0700
Subject: [PATCH] Add EntityList::first().

This is the same as slice::first(), except it returns the first element
by value.

The implementation can avoid checking the list length since empty lists
already have a special representation.
---
 lib/cretonne/src/entity_list.rs | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/lib/cretonne/src/entity_list.rs b/lib/cretonne/src/entity_list.rs
index 7e11019529..3b7493094e 100644
--- a/lib/cretonne/src/entity_list.rs
+++ b/lib/cretonne/src/entity_list.rs
@@ -260,6 +260,15 @@ impl<T: EntityRef> EntityList<T> {
         self.as_slice(pool).get(index).cloned()
     }
 
+    /// Get the first element from the list.
+    pub fn first(&self, pool: &ListPool<T>) -> Option<T> {
+        if self.is_empty() {
+            None
+        } else {
+            Some(pool.data[self.index as usize])
+        }
+    }
+
     /// Get the list as a mutable slice.
     pub fn as_mut_slice<'a>(&'a mut self, pool: &'a mut ListPool<T>) -> &'a mut [T] {
         let idx = self.index as usize;
@@ -507,6 +516,7 @@ mod tests {
         assert!(list.is_empty());
         assert_eq!(list.len(pool), 0);
         assert_eq!(list.as_slice(pool), &[]);
+        assert_eq!(list.first(pool), None);
     }
 
     #[test]
@@ -523,6 +533,7 @@ mod tests {
         assert_eq!(list.len(pool), 1);
         assert!(!list.is_empty());
         assert_eq!(list.as_slice(pool), &[i1]);
+        assert_eq!(list.first(pool), Some(i1));
         assert_eq!(list.get(0, pool), Some(i1));
         assert_eq!(list.get(1, pool), None);
 
@@ -530,6 +541,7 @@ mod tests {
         assert_eq!(list.len(pool), 2);
         assert!(!list.is_empty());
         assert_eq!(list.as_slice(pool), &[i1, i2]);
+        assert_eq!(list.first(pool), Some(i1));
         assert_eq!(list.get(0, pool), Some(i1));
         assert_eq!(list.get(1, pool), Some(i2));
         assert_eq!(list.get(2, pool), None);
@@ -538,6 +550,7 @@ mod tests {
         assert_eq!(list.len(pool), 3);
         assert!(!list.is_empty());
         assert_eq!(list.as_slice(pool), &[i1, i2, i3]);
+        assert_eq!(list.first(pool), Some(i1));
         assert_eq!(list.get(0, pool), Some(i1));
         assert_eq!(list.get(1, pool), Some(i2));
         assert_eq!(list.get(2, pool), Some(i3));
@@ -548,6 +561,7 @@ mod tests {
         assert_eq!(list.len(pool), 4);
         assert!(!list.is_empty());
         assert_eq!(list.as_slice(pool), &[i1, i2, i3, i4]);
+        assert_eq!(list.first(pool), Some(i1));
         assert_eq!(list.get(0, pool), Some(i1));
         assert_eq!(list.get(1, pool), Some(i2));
         assert_eq!(list.get(2, pool), Some(i3));