rayon/
array.rs

1//! Parallel iterator types for [arrays] (`[T; N]`)
2//!
3//! You will rarely need to interact with this module directly unless you need
4//! to name one of the iterator types.
5//!
6//! [arrays]: https://doc.rust-lang.org/std/primitive.array.html
7
8use crate::iter::plumbing::*;
9use crate::iter::*;
10use crate::slice::{Iter, IterMut};
11use crate::vec::DrainProducer;
12use std::mem::ManuallyDrop;
13
14impl<'data, T: Sync + 'data, const N: usize> IntoParallelIterator for &'data [T; N] {
15    type Item = &'data T;
16    type Iter = Iter<'data, T>;
17
18    fn into_par_iter(self) -> Self::Iter {
19        <&[T]>::into_par_iter(self)
20    }
21}
22
23impl<'data, T: Send + 'data, const N: usize> IntoParallelIterator for &'data mut [T; N] {
24    type Item = &'data mut T;
25    type Iter = IterMut<'data, T>;
26
27    fn into_par_iter(self) -> Self::Iter {
28        <&mut [T]>::into_par_iter(self)
29    }
30}
31
32impl<T: Send, const N: usize> IntoParallelIterator for [T; N] {
33    type Item = T;
34    type Iter = IntoIter<T, N>;
35
36    fn into_par_iter(self) -> Self::Iter {
37        IntoIter { array: self }
38    }
39}
40
41/// Parallel iterator that moves out of an array.
42#[derive(Debug, Clone)]
43pub struct IntoIter<T: Send, const N: usize> {
44    array: [T; N],
45}
46
47impl<T: Send, const N: usize> ParallelIterator for IntoIter<T, N> {
48    type Item = T;
49
50    fn drive_unindexed<C>(self, consumer: C) -> C::Result
51    where
52        C: UnindexedConsumer<Self::Item>,
53    {
54        bridge(self, consumer)
55    }
56
57    fn opt_len(&self) -> Option<usize> {
58        Some(N)
59    }
60}
61
62impl<T: Send, const N: usize> IndexedParallelIterator for IntoIter<T, N> {
63    fn drive<C>(self, consumer: C) -> C::Result
64    where
65        C: Consumer<Self::Item>,
66    {
67        bridge(self, consumer)
68    }
69
70    fn len(&self) -> usize {
71        N
72    }
73
74    fn with_producer<CB>(self, callback: CB) -> CB::Output
75    where
76        CB: ProducerCallback<Self::Item>,
77    {
78        unsafe {
79            // Drain every item, and then the local array can just fall out of scope.
80            let mut array = ManuallyDrop::new(self.array);
81            let producer = DrainProducer::new(array.as_mut_slice());
82            callback.callback(producer)
83        }
84    }
85}
rayon/array.rs

rayon/
array.rs
