select_indices 3.0.0-alpha

Iterators for taking multiple shared/exclusive references from a slice
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
63
64
65
66
67
68
69

use rayon::iter::plumbing::Producer;
use std::slice::from_raw_parts_mut;

use crate::SelectIndicesBase;

use num_traits::{ PrimInt, ToPrimitive };

impl<'a, T: Send, I: Copy + Clone + PrimInt + ToPrimitive + Sync> Producer for SelectIndicesBase<'a, &'a mut [T], I>
{
    type Item = &'a mut T;

    type IntoIter = Self;

    fn into_iter(self) -> Self::IntoIter {
        return self;
    }

    fn split_at(self, index: usize) -> (Self, Self) {
        let data_len = self.data.len();
        let ptr = self.data.as_mut_ptr();
        let refs = unsafe { (
            from_raw_parts_mut(ptr.clone(), data_len),
            from_raw_parts_mut(ptr, data_len)
        ) };
        let split = self.indices.split_at(index); // TODO: Unchecked
        return (
            SelectIndicesBase {
                data: refs.0,
                indices: split.0,
                start: 0,
                end: split.0.len(),
            }.into(),
            SelectIndicesBase {
                data: refs.1,
                indices: split.1,
                start: 0,
                end: split.1.len()
            }.into(),
        );
    }
}

impl<'a, T: Sync, I: Copy + Clone + PrimInt + ToPrimitive + Sync> Producer for SelectIndicesBase<'a, &'a [T], I>
{
    type Item = &'a T;
    type IntoIter = Self;

    fn into_iter(self) -> Self::IntoIter {
        return self;
    }

    fn split_at(self, index: usize) -> (Self, Self) {
        let split = self.indices.split_at(index);
        return (
            SelectIndicesBase {
                data: self.data,
                indices: split.0,
                start: 0,
                end: split.0.len(),
            }.into(),
            SelectIndicesBase {
                data: self.data,
                indices: split.1,
                start: 0,
                end: split.1.len()
            }.into(),
        )
    }
}