use std::num::Int;
macro_rules! clone_fields {
($name:ident, $($field:ident),+) => (
fn clone(&self) -> Self
{
$name {
$(
$field : self . $field .clone()
),*
}
}
);
}
pub struct Intersperse<I: Iterator> {
element: I::Item,
iter: I,
peek: Option<I::Item>,
}
impl<I: Iterator> Clone for Intersperse<I> where
I: Clone,
I::Item: Clone,
{
clone_fields!(Intersperse, element, iter, peek);
}
impl<I: Iterator> Intersperse<I>
{
pub fn new(mut iter: I, elt: I::Item) -> Self
{
Intersperse{peek: iter.next(), iter: iter, element: elt}
}
}
impl<I: Iterator> Iterator for Intersperse<I> where
I::Item: Clone,
{
type Item = I::Item;
#[inline]
fn next(&mut self) -> Option<I::Item>
{
if self.peek.is_some() {
self.peek.take()
} else {
self.peek = self.iter.next();
if self.peek.is_some() {
Some(self.element.clone())
} else {
None
}
}
}
fn size_hint(&self) -> (usize, Option<usize>)
{
let (mut low, mut hi) = self.iter.size_hint();
if low > 0 {
low = low.saturating_add((low - 1));
}
hi = hi.and_then(|x| if x > 0 {
x.checked_add(x - 1)
} else { Some (x) });
(low, hi)
}
}