slicetools 0.3.0

Add extra iterators to slices.
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

#[cfg(test)] mod tests;

use ::StreamerMut;

pub struct CartesianProductMut<'a, T: 'a, U: 'a> {
    slice: (&'a mut [T], &'a mut [U]),
    index: (usize, usize),
}

impl <'a, 'it, T, U> StreamerMut<'it, (&'it mut T, &'it mut U)> for CartesianProductMut<'a, T, U> {
    fn next(&'it mut self) -> Option<(&'it mut T, &'it mut U)> {
        if self.index.0 >= self.slice.0.len() {
            return None;
        }
        if self.index.1 >= self.slice.1.len() {
            self.index.0 += 1;
            self.index.1 = 0;
        }
        match self.slice.0.get_mut(self.index.0) {
            Some(first) => {
                let ret = Some((first, self.slice.1.get_mut(self.index.1).unwrap()));
                self.index.1 += 1;
                ret
            },
            None => None,
        }
    }
}

impl<'a, T, U> CartesianProductMut<'a, T, U> {
    pub fn new(s1: &'a mut [T], s2: &'a mut [U]) -> CartesianProductMut<'a, T, U> {
        CartesianProductMut{ slice: (s1, s2), index: (0, 0) }
    }
}