diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-28 14:29:10 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-28 14:29:10 +0000 |
commit | 2aa4a82499d4becd2284cdb482213d541b8804dd (patch) | |
tree | b80bf8bf13c3766139fbacc530efd0dd9d54394c /third_party/rust/itertools-0.8.0/src/intersperse.rs | |
parent | Initial commit. (diff) | |
download | firefox-2aa4a82499d4becd2284cdb482213d541b8804dd.tar.xz firefox-2aa4a82499d4becd2284cdb482213d541b8804dd.zip |
Adding upstream version 86.0.1.upstream/86.0.1upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'third_party/rust/itertools-0.8.0/src/intersperse.rs')
-rw-r--r-- | third_party/rust/itertools-0.8.0/src/intersperse.rs | 60 |
1 files changed, 60 insertions, 0 deletions
diff --git a/third_party/rust/itertools-0.8.0/src/intersperse.rs b/third_party/rust/itertools-0.8.0/src/intersperse.rs new file mode 100644 index 0000000000..b1dc732117 --- /dev/null +++ b/third_party/rust/itertools-0.8.0/src/intersperse.rs @@ -0,0 +1,60 @@ +use std::iter::Fuse; +use super::size_hint; + +#[derive(Clone)] +/// An iterator adaptor to insert a particular value +/// between each element of the adapted iterator. +/// +/// Iterator element type is `I::Item` +/// +/// This iterator is *fused*. +/// +/// See [`.intersperse()`](../trait.Itertools.html#method.intersperse) for more information. +#[must_use = "iterator adaptors are lazy and do nothing unless consumed"] +#[derive(Debug)] +pub struct Intersperse<I> + where I: Iterator +{ + element: I::Item, + iter: Fuse<I>, + peek: Option<I::Item>, +} + +/// Create a new Intersperse iterator +pub fn intersperse<I>(iter: I, elt: I::Item) -> Intersperse<I> + where I: Iterator +{ + let mut iter = iter.fuse(); + Intersperse { + peek: iter.next(), + iter: iter, + element: elt, + } +} + +impl<I> Iterator for Intersperse<I> + where I: Iterator, + I::Item: Clone +{ + type Item = I::Item; + #[inline] + fn next(&mut self) -> Option<I::Item> { + if self.peek.is_some() { + self.peek.take() + } else { + self.peek = self.iter.next(); + if self.peek.is_some() { + Some(self.element.clone()) + } else { + None + } + } + } + + fn size_hint(&self) -> (usize, Option<usize>) { + // 2 * SH + { 1 or 0 } + let has_peek = self.peek.is_some() as usize; + let sh = self.iter.size_hint(); + size_hint::add_scalar(size_hint::add(sh, sh), has_peek) + } +} |