1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

use crate::Operator;

/// Operated iterator.
#[derive(Debug, Clone, Copy)]
pub struct Operated<It, P> {
    source: It,
    op: P,
}

impl<It, P> Iterator for Operated<It, P>
where
    It: Iterator,
    P: Operator<It::Item>,
{
    type Item = P::Output;

    fn next(&mut self) -> Option<Self::Item> {
        self.source.next().map(|input| self.op.next(input))
    }
}

/// Iterator extension trait for indicators.
pub trait IndicatorIteratorExt: Iterator {
    /// Apply an indicator to the iterator.
    fn indicator<P>(self, op: P) -> Operated<Self, P>
    where
        Self: Sized,
        P: Operator<Self::Item>,
    {
        Operated { source: self, op }
    }
}

impl<It: Iterator> IndicatorIteratorExt for It {}