diff options
| author | Артём Павлов [Artyom Pavlov] <newpavlov@gmail.com> | 2018-11-18 21:39:23 +0300 |
|---|---|---|
| committer | Артём Павлов [Artyom Pavlov] <newpavlov@gmail.com> | 2018-11-18 21:39:23 +0300 |
| commit | 126b71f69099df4ee2b0654f56709ab937eceb2b (patch) | |
| tree | 3df2fc90237f1a9814ffb3cb52161c22b67d91e6 /src/libcore | |
| parent | d1c2cbf90a12bce0d4214ae27feda4f7d73c0510 (diff) | |
| download | rust-126b71f69099df4ee2b0654f56709ab937eceb2b.tar.gz rust-126b71f69099df4ee2b0654f56709ab937eceb2b.zip | |
revert
Diffstat (limited to 'src/libcore')
| -rw-r--r-- | src/libcore/iter/mod.rs | 82 | ||||
| -rw-r--r-- | src/libcore/tests/iter.rs | 7 |
2 files changed, 9 insertions, 80 deletions
diff --git a/src/libcore/iter/mod.rs b/src/libcore/iter/mod.rs index c42fb7019c7..509068843d1 100644 --- a/src/libcore/iter/mod.rs +++ b/src/libcore/iter/mod.rs @@ -319,10 +319,9 @@ use cmp; use fmt; use iter_private::TrustedRandomAccess; -use ops::{self, Try}; +use ops::Try; use usize; use intrinsics; -use mem; #[stable(feature = "rust1", since = "1.0.0")] pub use self::iterator::Iterator; @@ -673,7 +672,12 @@ impl<I> Iterator for StepBy<I> where I: Iterator { #[inline] fn next(&mut self) -> Option<Self::Item> { - <Self as StepBySpecIterator>::spec_next(self) + if self.first_take { + self.first_take = false; + self.iter.next() + } else { + self.iter.nth(self.step) + } } #[inline] @@ -733,78 +737,6 @@ impl<I> Iterator for StepBy<I> where I: Iterator { } } -// hidden trait for specializing iterator methods -// could be generalized but is currently only used for StepBy -trait StepBySpecIterator { - type Item; - fn spec_next(&mut self) -> Option<Self::Item>; -} - -impl<I> StepBySpecIterator for StepBy<I> -where - I: Iterator, -{ - type Item = I::Item; - - #[inline] - default fn spec_next(&mut self) -> Option<I::Item> { - if self.first_take { - self.first_take = false; - self.iter.next() - } else { - self.iter.nth(self.step) - } - } -} - -impl<T> StepBySpecIterator for StepBy<ops::Range<T>> -where - T: Step, -{ - #[inline] - fn spec_next(&mut self) -> Option<Self::Item> { - self.first_take = false; - if !(self.iter.start < self.iter.end) { - return None; - } - // add 1 to self.step to get original step size back - // it was decremented for the general case on construction - if let Some(n) = self.iter.start.add_usize(self.step+1) { - let next = mem::replace(&mut self.iter.start, n); - Some(next) - } else { - let last = self.iter.start.clone(); - self.iter.start = self.iter.end.clone(); - Some(last) - } - } -} - -impl<T> StepBySpecIterator for StepBy<ops::RangeInclusive<T>> -where - T: Step, -{ - #[inline] - fn spec_next(&mut self) -> Option<Self::Item> { - self.first_take = false; - self.iter.compute_is_empty(); - if self.iter.is_empty.unwrap_or_default() { - return None; - } - // add 1 to self.step to get original step size back - // it was decremented for the general case on construction - if let Some(n) = self.iter.start.add_usize(self.step+1) { - self.iter.is_empty = Some(!(n <= self.iter.end)); - let next = mem::replace(&mut self.iter.start, n); - Some(next) - } else { - let last = self.iter.start.clone(); - self.iter.is_empty = Some(true); - Some(last) - } - } -} - // StepBy can only make the iterator shorter, so the len will still fit. #[stable(feature = "iterator_step_by", since = "1.28.0")] impl<I> ExactSizeIterator for StepBy<I> where I: ExactSizeIterator {} diff --git a/src/libcore/tests/iter.rs b/src/libcore/tests/iter.rs index 2476c07cbd9..fcf7f0cd5d7 100644 --- a/src/libcore/tests/iter.rs +++ b/src/libcore/tests/iter.rs @@ -1619,11 +1619,8 @@ fn test_range_step() { } #[test] -fn test_range_inclusive_step() { - assert_eq!((0..=50).step_by(10).collect::<Vec<_>>(), [0, 10, 20, 30, 40, 50]); - assert_eq!((0..=5).step_by(1).collect::<Vec<_>>(), [0, 1, 2, 3, 4, 5]); - assert_eq!((200..=255u8).step_by(10).collect::<Vec<_>>(), [200, 210, 220, 230, 240, 250]); - assert_eq!((250..=255u8).step_by(1).collect::<Vec<_>>(), [250, 251, 252, 253, 254, 255]); +fn test_step_by_skip() { + assert_eq!((0..640).step_by(128).skip(1).collect::<Vec<_>>(), [128, 256, 384, 512]); } #[test] |
