diff --git a/src/libcore/iterator.rs b/src/libcore/iterator.rs index 4a3a98df0d0ea..8bbf843085809 100644 --- a/src/libcore/iterator.rs +++ b/src/libcore/iterator.rs @@ -312,6 +312,19 @@ impl> Iterator for TakeIterator { } } +pub struct ScanIterator<'self, A, B, T, St> { + priv iter: T, + priv f: &'self fn(&mut St, A) -> Option, + state: St +} + +impl<'self, A, B, T: Iterator, St> Iterator for ScanIterator<'self, A, B, T, St> { + #[inline] + fn next(&mut self) -> Option { + self.iter.next().chain(|a| (self.f)(&mut self.state, a)) + } +} + pub struct UnfoldrIterator<'self, A, St> { priv f: &'self fn(&mut St) -> Option, state: St @@ -335,16 +348,25 @@ impl<'self, A, St> Iterator for UnfoldrIterator<'self, A, St> { } } -pub struct ScanIterator<'self, A, B, T, St> { - priv iter: T, - priv f: &'self fn(&mut St, A) -> Option, - state: St +/// An infinite iterator starting at `start` and advancing by `step` with each iteration +pub struct Counter { + state: A, + step: A } -impl<'self, A, B, T: Iterator, St> Iterator for ScanIterator<'self, A, B, T, St> { - #[inline] - fn next(&mut self) -> Option { - self.iter.next().chain(|a| (self.f)(&mut self.state, a)) +pub impl Counter { + #[inline(always)] + fn new(start: A, step: A) -> Counter { + Counter{state: start, step: step} + } +} + +impl + Clone> Iterator for Counter { + #[inline(always)] + fn next(&mut self) -> Option { + let result = self.state.clone(); + self.state = self.state.add(&self.step); // FIXME: #6050 + Some(result) } } @@ -353,6 +375,13 @@ mod tests { use super::*; use prelude::*; + #[test] + fn test_counter_to_vec() { + let mut it = Counter::new(0, 5).take(10); + let xs = iter::iter_to_vec(|f| it.advance(f)); + assert_eq!(xs, ~[0, 5, 10, 15, 20, 25, 30, 35, 40, 45]); + } + #[test] fn test_iterator_chain() { let xs = [0u, 1, 2, 3, 4, 5];