aboutsummaryrefslogtreecommitdiff
path: root/vendor/rayon/src/iter/chunks.rs
diff options
context:
space:
mode:
Diffstat (limited to 'vendor/rayon/src/iter/chunks.rs')
-rw-r--r--vendor/rayon/src/iter/chunks.rs226
1 files changed, 0 insertions, 226 deletions
diff --git a/vendor/rayon/src/iter/chunks.rs b/vendor/rayon/src/iter/chunks.rs
deleted file mode 100644
index ec48278..0000000
--- a/vendor/rayon/src/iter/chunks.rs
+++ /dev/null
@@ -1,226 +0,0 @@
-use std::cmp::min;
-
-use super::plumbing::*;
-use super::*;
-use crate::math::div_round_up;
-
-/// `Chunks` is an iterator that groups elements of an underlying iterator.
-///
-/// This struct is created by the [`chunks()`] method on [`IndexedParallelIterator`]
-///
-/// [`chunks()`]: trait.IndexedParallelIterator.html#method.chunks
-/// [`IndexedParallelIterator`]: trait.IndexedParallelIterator.html
-#[must_use = "iterator adaptors are lazy and do nothing unless consumed"]
-#[derive(Debug, Clone)]
-pub struct Chunks<I>
-where
- I: IndexedParallelIterator,
-{
- size: usize,
- i: I,
-}
-
-impl<I> Chunks<I>
-where
- I: IndexedParallelIterator,
-{
- /// Creates a new `Chunks` iterator
- pub(super) fn new(i: I, size: usize) -> Self {
- Chunks { i, size }
- }
-}
-
-impl<I> ParallelIterator for Chunks<I>
-where
- I: IndexedParallelIterator,
-{
- type Item = Vec<I::Item>;
-
- fn drive_unindexed<C>(self, consumer: C) -> C::Result
- where
- C: Consumer<Vec<I::Item>>,
- {
- bridge(self, consumer)
- }
-
- fn opt_len(&self) -> Option<usize> {
- Some(self.len())
- }
-}
-
-impl<I> IndexedParallelIterator for Chunks<I>
-where
- I: IndexedParallelIterator,
-{
- fn drive<C>(self, consumer: C) -> C::Result
- where
- C: Consumer<Self::Item>,
- {
- bridge(self, consumer)
- }
-
- fn len(&self) -> usize {
- div_round_up(self.i.len(), self.size)
- }
-
- fn with_producer<CB>(self, callback: CB) -> CB::Output
- where
- CB: ProducerCallback<Self::Item>,
- {
- let len = self.i.len();
- return self.i.with_producer(Callback {
- size: self.size,
- len,
- callback,
- });
-
- struct Callback<CB> {
- size: usize,
- len: usize,
- callback: CB,
- }
-
- impl<T, CB> ProducerCallback<T> for Callback<CB>
- where
- CB: ProducerCallback<Vec<T>>,
- {
- type Output = CB::Output;
-
- fn callback<P>(self, base: P) -> CB::Output
- where
- P: Producer<Item = T>,
- {
- let producer = ChunkProducer::new(self.size, self.len, base, Vec::from_iter);
- self.callback.callback(producer)
- }
- }
- }
-}
-
-pub(super) struct ChunkProducer<P, F> {
- chunk_size: usize,
- len: usize,
- base: P,
- map: F,
-}
-
-impl<P, F> ChunkProducer<P, F> {
- pub(super) fn new(chunk_size: usize, len: usize, base: P, map: F) -> Self {
- Self {
- chunk_size,
- len,
- base,
- map,
- }
- }
-}
-
-impl<P, F, T> Producer for ChunkProducer<P, F>
-where
- P: Producer,
- F: Fn(P::IntoIter) -> T + Send + Clone,
-{
- type Item = T;
- type IntoIter = std::iter::Map<ChunkSeq<P>, F>;
-
- fn into_iter(self) -> Self::IntoIter {
- let chunks = ChunkSeq {
- chunk_size: self.chunk_size,
- len: self.len,
- inner: if self.len > 0 { Some(self.base) } else { None },
- };
- chunks.map(self.map)
- }
-
- fn split_at(self, index: usize) -> (Self, Self) {
- let elem_index = min(index * self.chunk_size, self.len);
- let (left, right) = self.base.split_at(elem_index);
- (
- ChunkProducer {
- chunk_size: self.chunk_size,
- len: elem_index,
- base: left,
- map: self.map.clone(),
- },
- ChunkProducer {
- chunk_size: self.chunk_size,
- len: self.len - elem_index,
- base: right,
- map: self.map,
- },
- )
- }
-
- fn min_len(&self) -> usize {
- div_round_up(self.base.min_len(), self.chunk_size)
- }
-
- fn max_len(&self) -> usize {
- self.base.max_len() / self.chunk_size
- }
-}
-
-pub(super) struct ChunkSeq<P> {
- chunk_size: usize,
- len: usize,
- inner: Option<P>,
-}
-
-impl<P> Iterator for ChunkSeq<P>
-where
- P: Producer,
-{
- type Item = P::IntoIter;
-
- fn next(&mut self) -> Option<Self::Item> {
- let producer = self.inner.take()?;
- if self.len > self.chunk_size {
- let (left, right) = producer.split_at(self.chunk_size);
- self.inner = Some(right);
- self.len -= self.chunk_size;
- Some(left.into_iter())
- } else {
- debug_assert!(self.len > 0);
- self.len = 0;
- Some(producer.into_iter())
- }
- }
-
- fn size_hint(&self) -> (usize, Option<usize>) {
- let len = self.len();
- (len, Some(len))
- }
-}
-
-impl<P> ExactSizeIterator for ChunkSeq<P>
-where
- P: Producer,
-{
- #[inline]
- fn len(&self) -> usize {
- div_round_up(self.len, self.chunk_size)
- }
-}
-
-impl<P> DoubleEndedIterator for ChunkSeq<P>
-where
- P: Producer,
-{
- fn next_back(&mut self) -> Option<Self::Item> {
- let producer = self.inner.take()?;
- if self.len > self.chunk_size {
- let mut size = self.len % self.chunk_size;
- if size == 0 {
- size = self.chunk_size;
- }
- let (left, right) = producer.split_at(self.len - size);
- self.inner = Some(left);
- self.len -= size;
- Some(right.into_iter())
- } else {
- debug_assert!(self.len > 0);
- self.len = 0;
- Some(producer.into_iter())
- }
- }
-}