about summary refs log tree commit diff
path: root/src/libcore/iter
diff options
context:
space:
mode:
authorkennytm <kennytm@gmail.com>2018-06-19 04:08:20 +0800
committerkennytm <kennytm@gmail.com>2018-07-13 09:53:36 +0800
commit0d7e9933d3cac85bc1f11dc0fec67fcad77784ca (patch)
tree1c0488fda11d56a39db22edb943b7e8e536d0b7d /src/libcore/iter
parent7db82ccd765cbfe55c3d8a2c434bc6f9b986843d (diff)
downloadrust-0d7e9933d3cac85bc1f11dc0fec67fcad77784ca.tar.gz
rust-0d7e9933d3cac85bc1f11dc0fec67fcad77784ca.zip
Change RangeInclusive to a three-field struct.
Fix #45222.
Diffstat (limited to 'src/libcore/iter')
-rw-r--r--src/libcore/iter/range.rs100
1 files changed, 30 insertions, 70 deletions
diff --git a/src/libcore/iter/range.rs b/src/libcore/iter/range.rs
index 0b279f66b88..16849e84f27 100644
--- a/src/libcore/iter/range.rs
+++ b/src/libcore/iter/range.rs
@@ -10,7 +10,7 @@
 
 use convert::TryFrom;
 use mem;
-use ops::{self, Add, Sub, Try};
+use ops::{self, Add, Sub};
 use usize;
 
 use super::{FusedIterator, TrustedLen};
@@ -330,23 +330,23 @@ impl<A: Step> Iterator for ops::RangeInclusive<A> {
 
     #[inline]
     fn next(&mut self) -> Option<A> {
-        if self.start <= self.end {
-            if self.start < self.end {
-                let n = self.start.add_one();
-                Some(mem::replace(&mut self.start, n))
-            } else {
-                let last = self.start.replace_one();
-                self.end.replace_zero();
-                Some(last)
-            }
+        if self.is_empty() {
+            self.is_iterating = Some(false);
+            return None;
+        }
+        if self.start < self.end {
+            let n = self.start.add_one();
+            self.is_iterating = Some(true);
+            Some(mem::replace(&mut self.start, n))
         } else {
-            None
+            self.is_iterating = Some(false);
+            Some(self.start.clone())
         }
     }
 
     #[inline]
     fn size_hint(&self) -> (usize, Option<usize>) {
-        if !(self.start <= self.end) {
+        if self.is_empty() {
             return (0, Some(0));
         }
 
@@ -358,25 +358,29 @@ impl<A: Step> Iterator for ops::RangeInclusive<A> {
 
     #[inline]
     fn nth(&mut self, n: usize) -> Option<A> {
+        if self.is_empty() {
+            self.is_iterating = Some(false);
+            return None;
+        }
+
         if let Some(plus_n) = self.start.add_usize(n) {
             use cmp::Ordering::*;
 
             match plus_n.partial_cmp(&self.end) {
                 Some(Less) => {
+                    self.is_iterating = Some(true);
                     self.start = plus_n.add_one();
                     return Some(plus_n)
                 }
                 Some(Equal) => {
-                    self.start.replace_one();
-                    self.end.replace_zero();
+                    self.is_iterating = Some(false);
                     return Some(plus_n)
                 }
                 _ => {}
             }
         }
 
-        self.start.replace_one();
-        self.end.replace_zero();
+        self.is_iterating = Some(false);
         None
     }
 
@@ -394,68 +398,24 @@ impl<A: Step> Iterator for ops::RangeInclusive<A> {
     fn max(mut self) -> Option<A> {
         self.next_back()
     }
-
-    #[inline]
-    fn try_fold<B, F, R>(&mut self, init: B, mut f: F) -> R where
-        Self: Sized, F: FnMut(B, Self::Item) -> R, R: Try<Ok=B>
-    {
-        let mut accum = init;
-        if self.start <= self.end {
-            loop {
-                let (x, done) =
-                    if self.start < self.end {
-                        let n = self.start.add_one();
-                        (mem::replace(&mut self.start, n), false)
-                    } else {
-                        self.end.replace_zero();
-                        (self.start.replace_one(), true)
-                    };
-                accum = f(accum, x)?;
-                if done { break }
-            }
-        }
-        Try::from_ok(accum)
-    }
 }
 
 #[stable(feature = "inclusive_range", since = "1.26.0")]
 impl<A: Step> DoubleEndedIterator for ops::RangeInclusive<A> {
     #[inline]
     fn next_back(&mut self) -> Option<A> {
-        if self.start <= self.end {
-            if self.start < self.end {
-                let n = self.end.sub_one();
-                Some(mem::replace(&mut self.end, n))
-            } else {
-                let last = self.end.replace_zero();
-                self.start.replace_one();
-                Some(last)
-            }
-        } else {
-            None
+        if self.is_empty() {
+            self.is_iterating = Some(false);
+            return None;
         }
-    }
-
-    #[inline]
-    fn try_rfold<B, F, R>(&mut self, init: B, mut f: F) -> R where
-        Self: Sized, F: FnMut(B, Self::Item) -> R, R: Try<Ok=B>
-    {
-        let mut accum = init;
-        if self.start <= self.end {
-            loop {
-                let (x, done) =
-                    if self.start < self.end {
-                        let n = self.end.sub_one();
-                        (mem::replace(&mut self.end, n), false)
-                    } else {
-                        self.start.replace_one();
-                        (self.end.replace_zero(), true)
-                    };
-                accum = f(accum, x)?;
-                if done { break }
-            }
+        if self.start < self.end {
+            let n = self.end.sub_one();
+            self.is_iterating = Some(true);
+            Some(mem::replace(&mut self.end, n))
+        } else {
+            self.is_iterating = Some(false);
+            Some(self.end.clone())
         }
-        Try::from_ok(accum)
     }
 }