Merge pull request #60 from orxfun/iter_over_range

iter_over_range

Author: orxfun

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "orx-split-vec"
-version = "3.9.0"
+version = "3.10.0"
 edition = "2021"
 authors = ["orxfun <[email protected]>"]
 description = "An efficient constant access time vector with dynamic capacity and pinned elements."
@@ -11,7 +11,7 @@ categories = ["data-structures", "rust-patterns", "no-std"]
 
 [dependencies]
 orx-pseudo-default = { version = "1.4", default-features = false }
-orx-pinned-vec = "3.9"
+orx-pinned-vec = "3.10"
 
 [[bench]]
 name = "serial_access"

src/concurrent_pinned_vec.rs

@@ -266,6 +266,17 @@ impl<T, G: GrowthWithConstantTimeAccess> ConcurrentPinnedVec<T> for ConcurrentSp
         self.slices(0..len).into_iter().flat_map(|x| x.iter())
     }
 
+    unsafe fn iter_over_range<'a, R: RangeBounds<usize>>(
+        &'a self,
+        range: R,
+    ) -> impl Iterator<Item = &'a T> + 'a
+    where
+        T: 'a,
+    {
+        let [a, b] = orx_pinned_vec::utils::slice::vec_range_limits(&range, None);
+        self.slices(a..b).into_iter().flat_map(|x| x.iter())
+    }
+
     unsafe fn slices_mut<R: RangeBounds<usize>>(
         &self,
         range: R,

tests/con_iter_over_range.rs

@@ -0,0 +1,30 @@
+use orx_split_vec::*;
+
+#[test]
+fn iter_over_range() {
+    let vec = SplitVec::<_, Doubling>::from_iter([0, 1, 2, 3, 4, 5, 6].into_iter());
+    let con_vec = vec.into_concurrent();
+
+    unsafe {
+        let vec: Vec<_> = con_vec.iter_over_range(..7).copied().collect();
+        assert_eq!(vec, &[0, 1, 2, 3, 4, 5, 6]);
+
+        let vec: Vec<_> = con_vec.iter_over_range(..4).copied().collect();
+        assert_eq!(vec, &[0, 1, 2, 3]);
+
+        let vec: Vec<_> = con_vec.iter_over_range(1..7).copied().collect();
+        assert_eq!(vec, &[1, 2, 3, 4, 5, 6]);
+
+        let vec: Vec<_> = con_vec.iter_over_range(1..4).copied().collect();
+        assert_eq!(vec, &[1, 2, 3]);
+
+        let vec: Vec<_> = con_vec.iter_over_range(4..4).copied().collect();
+        assert_eq!(vec, &[]);
+
+        let vec: Vec<_> = con_vec.iter_over_range(4..3).copied().collect();
+        assert_eq!(vec, &[]);
+
+        let vec: Vec<_> = con_vec.iter_over_range(1..=4).copied().collect();
+        assert_eq!(vec, &[1, 2, 3, 4]);
+    }
+}