eeric 0.0.25

An easily embeddable RV64I core with MFDV extensions
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
55
56
57
58
59
60
61
62

pub trait IterMaskExt: Iterator {
    // An iterator that maps the value of the iterator if the `mask.next()` yields 1 and `destination.next()` otherwise.

    fn masked_map<Mask, Destination, Func>(
        self,
        mask: Mask,
        dest: Destination,
        func: Func,
    ) -> MaskedMap<Self, Mask, Destination, Func>
    where
        Self: Sized,
        Mask: Iterator,
        Destination: Iterator,
        Func: FnMut(Self::Item) -> Destination::Item,
    {
        MaskedMap {
            values: self,
            mask,
            dest,
            func,
        }
    }
}

pub struct MaskedMap<I, Mask, Destination, Func>
where
    I: Iterator,
    Mask: Iterator,
    Destination: Iterator,
    Func: FnMut(I::Item) -> Destination::Item,
{
    values: I,
    mask: Mask,
    dest: Destination,
    func: Func,
}

impl<I, Mask, Destination, Func> Iterator for MaskedMap<I, Mask, Destination, Func>
where
    I: Iterator,
    Mask: Iterator,
    Mask::Item: PartialEq<u64>,
    Destination: Iterator,
    Func: FnMut(I::Item) -> Destination::Item,
{
    type Item = Destination::Item;

    fn next(&mut self) -> Option<Self::Item> {
        let iter_item = self.values.next();
        let dest_item = self.dest.next();

        self.mask.next().map(|m| {
            if m == 1 {
                (&mut self.func)(iter_item.unwrap())
            } else {
                dest_item.unwrap()
            }
        })
    }
}

impl<I: Iterator> IterMaskExt for I {}