select_indices 3.0.0

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

use std::ops::Index;
use crate::indexed_type::Indexed;
use crate::immutable::iter::ParSelectIndicesIter;
use rayon::{
    prelude::*,
    iter::plumbing::{ Consumer, UnindexedConsumer },
};

impl<'a, Data, Indices> ParallelIterator for ParSelectIndicesIter<'a, Data, Indices, Indexed>
where
    Data: ?Sized + Index<Indices::Item> + Sync,
    Data::Output: 'a + Sync,
    Indices: ParallelIterator,
    Indices::Item: Copy,
{
    type Item = (Indices::Item, &'a Data::Output);

    fn drive_unindexed<C>(self, consumer: C) -> C::Result
    where
        C: UnindexedConsumer<Self::Item>
    {
        let data = self.data;
        self.indices.map(|index| {
            (index, &data[index])
        }).drive_unindexed(consumer)
    }
}

impl<'a, Data, Indices> IndexedParallelIterator for ParSelectIndicesIter<'a, Data, Indices, Indexed>
where
    Data: ?Sized + Index<Indices::Item> + Sync,
    Data::Output: 'a + Sync,
    Indices: IndexedParallelIterator,
    Indices::Item: Copy,
{
    fn len(&self) -> usize {
        self.indices.len()
    }

    fn drive<C: Consumer<Self::Item>>(self, consumer: C) -> C::Result {
        let data = self.data;
        self.indices.map(|index| {
            (index, &data[index])
        }).drive(consumer)
    }

    fn with_producer<CB: rayon::iter::plumbing::ProducerCallback<Self::Item>>(self, callback: CB) -> CB::Output {
        let data = self.data;
        self.indices.map(|index| {
            (index, &data[index])
        }).with_producer(callback)
    }
}