noir-compute 0.2.0

Network of Operators In Rust
Documentation 
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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54

use std::cmp::Ordering;

use super::{super::*, FoldFirst};
use crate::operator::{Data, DataKey, Operator};
use crate::stream::{KeyedStream, WindowedStream};

impl<Key, Out, WindowDescr, OperatorChain> WindowedStream<OperatorChain, Out, WindowDescr>
where
    WindowDescr: WindowDescription<Out>,
    OperatorChain: Operator<Out = (Key, Out)> + 'static,
    Key: DataKey,
    Out: Data + Ord,
{
    pub fn max(self) -> KeyedStream<impl Operator<Out = (Key, Out)>> {
        let acc = FoldFirst::<Out, _>::new(|max, x| {
            if x > *max {
                *max = x
            }
        });
        self.add_window_operator("WindowMax", acc)
    }
}

impl<Key, Out, WindowDescr, OperatorChain> WindowedStream<OperatorChain, Out, WindowDescr>
where
    WindowDescr: WindowDescription<Out>,
    OperatorChain: Operator<Out = (Key, Out)> + 'static,
    Key: DataKey,
    Out: Data,
{
    pub fn max_by_key<K: Ord, F: Fn(&Out) -> K + Clone + Send + 'static>(
        self,
        get_key: F,
    ) -> KeyedStream<impl Operator<Out = (Key, Out)>> {
        let acc = FoldFirst::<Out, _>::new(move |max, x| {
            if (get_key)(&x) > (get_key)(max) {
                *max = x
            }
        });
        self.add_window_operator("WindowMax", acc)
    }

    pub fn max_by<F: Fn(&Out, &Out) -> Ordering + Clone + Send + 'static>(
        self,
        compare: F,
    ) -> KeyedStream<impl Operator<Out = (Key, Out)>> {
        let acc = FoldFirst::<Out, _>::new(move |max, x| {
            if (compare)(&x, max).is_gt() {
                *max = x
            }
        });
        self.add_window_operator("WindowMax", acc)
    }
}