From f29f01858d1b8c9e54b3293879796a4650823f60 Mon Sep 17 00:00:00 2001
From: Blaž Hrastnik
Date: Fri, 19 Mar 2021 11:14:13 +0900
Subject: Implement iter() and len() directly on Selection.

---
 helix-core/src/selection.rs | 28 +++++++++++++++++++++++-----
 1 file changed, 23 insertions(+), 5 deletions(-)

(limited to 'helix-core/src/selection.rs')

diff --git a/helix-core/src/selection.rs b/helix-core/src/selection.rs
index c34dff31..91edcf81 100644
--- a/helix-core/src/selection.rs
+++ b/helix-core/src/selection.rs
@@ -269,6 +269,25 @@ impl Selection {
     pub fn fragments<'a>(&'a self, text: RopeSlice<'a>) -> impl Iterator<Item = Cow<str>> + 'a {
         self.ranges.iter().map(move |range| range.fragment(text))
     }
+
+    #[inline(always)]
+    pub fn iter(&self) -> std::slice::Iter<'_, Range> {
+        self.ranges.iter()
+    }
+
+    #[inline(always)]
+    pub fn len(&self) -> usize {
+        self.ranges.len()
+    }
+}
+
+impl<'a> IntoIterator for &'a Selection {
+    type Item = &'a Range;
+    type IntoIter = std::slice::Iter<'a, Range>;
+
+    fn into_iter(self) -> std::slice::Iter<'a, Range> {
+        self.ranges().iter()
+    }
 }
 
 // TODO: checkSelection -> check if valid for doc length
@@ -279,7 +298,6 @@ pub fn keep_matches(
     regex: &crate::regex::Regex,
 ) -> Option<Selection> {
     let result: SmallVec<_> = selection
-        .ranges()
         .iter()
         .filter(|range| regex.is_match(&range.fragment(text)))
         .copied()
@@ -297,9 +315,9 @@ pub fn select_on_matches(
     selection: &Selection,
     regex: &crate::regex::Regex,
 ) -> Option<Selection> {
-    let mut result = SmallVec::with_capacity(selection.ranges().len());
+    let mut result = SmallVec::with_capacity(selection.len());
 
-    for sel in selection.ranges() {
+    for sel in selection {
         // TODO: can't avoid occasional allocations since Regex can't operate on chunks yet
         let fragment = sel.fragment(text);
 
@@ -331,9 +349,9 @@ pub fn split_on_matches(
     selection: &Selection,
     regex: &crate::regex::Regex,
 ) -> Selection {
-    let mut result = SmallVec::with_capacity(selection.ranges().len());
+    let mut result = SmallVec::with_capacity(selection.len());
 
-    for sel in selection.ranges() {
+    for sel in selection {
         // TODO: can't avoid occasional allocations since Regex can't operate on chunks yet
         let fragment = sel.fragment(text);
 
-- 
cgit v1.2.3-70-g09d2