Support generic-array as arrayvec backend

Author: bluss

Cargo.toml

@@ -19,6 +19,10 @@ version = "0.1.8"
 path = "nodrop"
 default-features = false
 
+[dependencies.generic-array]
+version = "0.5"
+optional = true
+
 [features]
 default = ["std"]
 std = ["odds/std", "nodrop/std"]

src/array.rs

@@ -19,6 +19,24 @@ pub trait Index : PartialEq + Copy {
     fn from(usize) -> Self;
 }
 
+#[cfg(feature = "generic-array")]
+unsafe impl<T, U> Array for ::generic_array::GenericArray<T, U>
+    where U: ::generic_array::ArrayLength<T>
+{
+    type Item = T;
+    type Index = usize;
+    fn as_ptr(&self) -> *const Self::Item {
+        (**self).as_ptr()
+    }
+    fn as_mut_ptr(&mut self) -> *mut Self::Item {
+        (**self).as_mut_ptr()
+    }
+    fn capacity() -> usize {
+        U::to_usize()
+    }
+
+}
+
 impl Index for u8 {
     #[inline(always)]
     fn to_usize(self) -> usize { self as usize }
@@ -33,6 +51,13 @@ impl Index for u16 {
     fn from(ix: usize) ->  Self { ix as u16 }
 }
 
+impl Index for usize {
+    #[inline(always)]
+    fn to_usize(self) -> usize { self }
+    #[inline(always)]
+    fn from(ix: usize) ->  Self { ix }
+}
+
 macro_rules! fix_array_impl {
     ($index_type:ty, $len:expr ) => (
         unsafe impl<T> Array for [T; $len] {

src/lib.rs

@@ -18,6 +18,9 @@
 extern crate odds;
 extern crate nodrop;
 
+#[cfg(feature = "generic-array")]
+extern crate generic_array;
+
 #[cfg(not(feature="std"))]
 extern crate core as std;
 

tests/generic_array.rs

@@ -0,0 +1,23 @@
+#![cfg(feature = "generic-array")]
+
+extern crate arrayvec;
+#[macro_use]
+extern crate generic_array;
+
+use arrayvec::ArrayVec;
+
+use generic_array::GenericArray;
+
+use generic_array::typenum::U41;
+
+#[test]
+fn test_simple() {
+    let mut vec: ArrayVec<GenericArray<i32, U41>> = ArrayVec::new();
+
+    assert_eq!(vec.len(), 0);
+    assert_eq!(vec.capacity(), 41);
+    vec.extend(0..20);
+    assert_eq!(vec.len(), 20);
+    assert_eq!(&vec[..5], &[0, 1, 2, 3, 4]);
+}
+