Add .extract() for extracting underlying Vec

jonhoo · jonhoo · commit 4cdeba386ca3 · 2016-04-18T17:50:29.000-04:00
diff --git a/src/array.rs b/src/array.rs
@@ -144,6 +144,12 @@ impl<T> Array<T> {
     pub fn iter_mut<'a>(&'a mut self) -> IterMut<'a, T> {
         IterMut { inner: self.data.iter_mut() }
     }
+
+    /// Returns the underlying data vector for this Array in the
+    /// higher-dimensional equivalent of row-major order.
+    pub fn extract(self) -> Vec<T> {
+        self.data
+    }
 }
 
 /// A trait implemented by types that can index into an `Array`.
diff --git a/src/lib.rs b/src/lib.rs
@@ -48,6 +48,16 @@ mod tests {
         assert_eq!(2, a[1]);
     }
 
+    #[test]
+    fn test_extract() {
+        let a = Array::from_vec(vec![0i32, 1, 2], -1);
+        let a = a.extract();
+        assert_eq!(a.len(), 3);
+        assert_eq!(0, a[0]);
+        assert_eq!(1, a[1]);
+        assert_eq!(2, a[2]);
+    }
+
     #[test]
     fn test_2d_slice_get() {
         let mut a = Array::from_vec(vec![0i32, 1, 2], -1);

Original file line number	Diff line number	Diff line change
`@@ -144,6 +144,12 @@ impl<T> Array<T> {`
`144`	`144`	`pub fn iter_mut<'a>(&'a mut self) -> IterMut<'a, T> {`
`145`	`145`	`IterMut { inner: self.data.iter_mut() }`
`146`	`146`	`}`
	`147`	`+`
	`148`	`+ /// Returns the underlying data vector for this Array in the`
	`149`	`+ /// higher-dimensional equivalent of row-major order.`
	`150`	`+ pub fn extract(self) -> Vec<T> {`
	`151`	`+ self.data`
	`152`	`+ }`
`147`	`153`	`}`
`148`	`154`
`149`	`155`	/// A trait implemented by types that can index into an `Array`.