summaryrefslogtreecommitdiffstats
path: root/vendor/itertools/src/lazy_buffer.rs
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-06-19 09:26:03 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-06-19 09:26:03 +0000
commit9918693037dce8aa4bb6f08741b6812923486c18 (patch)
tree21d2b40bec7e6a7ea664acee056eb3d08e15a1cf /vendor/itertools/src/lazy_buffer.rs
parentReleasing progress-linux version 1.75.0+dfsg1-5~progress7.99u1. (diff)
downloadrustc-9918693037dce8aa4bb6f08741b6812923486c18.tar.xz
rustc-9918693037dce8aa4bb6f08741b6812923486c18.zip
Merging upstream version 1.76.0+dfsg1.
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'vendor/itertools/src/lazy_buffer.rs')
-rw-r--r--vendor/itertools/src/lazy_buffer.rs30
1 files changed, 16 insertions, 14 deletions
diff --git a/vendor/itertools/src/lazy_buffer.rs b/vendor/itertools/src/lazy_buffer.rs
index ca24062aa..38c7d405b 100644
--- a/vendor/itertools/src/lazy_buffer.rs
+++ b/vendor/itertools/src/lazy_buffer.rs
@@ -1,10 +1,12 @@
-use std::ops::Index;
use alloc::vec::Vec;
+use std::iter::Fuse;
+use std::ops::Index;
+
+use crate::size_hint::{self, SizeHint};
#[derive(Debug, Clone)]
pub struct LazyBuffer<I: Iterator> {
- pub it: I,
- done: bool,
+ it: Fuse<I>,
buffer: Vec<I::Item>,
}
@@ -14,8 +16,7 @@ where
{
pub fn new(it: I) -> LazyBuffer<I> {
LazyBuffer {
- it,
- done: false,
+ it: it.fuse(),
buffer: Vec::new(),
}
}
@@ -24,27 +25,28 @@ where
self.buffer.len()
}
+ pub fn size_hint(&self) -> SizeHint {
+ size_hint::add_scalar(self.it.size_hint(), self.len())
+ }
+
+ pub fn count(self) -> usize {
+ self.len() + self.it.count()
+ }
+
pub fn get_next(&mut self) -> bool {
- if self.done {
- return false;
- }
if let Some(x) = self.it.next() {
self.buffer.push(x);
true
} else {
- self.done = true;
false
}
}
pub fn prefill(&mut self, len: usize) {
let buffer_len = self.buffer.len();
-
- if !self.done && len > buffer_len {
+ if len > buffer_len {
let delta = len - buffer_len;
-
self.buffer.extend(self.it.by_ref().take(delta));
- self.done = self.buffer.len() < len;
}
}
}
@@ -53,7 +55,7 @@ impl<I, J> Index<J> for LazyBuffer<I>
where
I: Iterator,
I::Item: Sized,
- Vec<I::Item>: Index<J>
+ Vec<I::Item>: Index<J>,
{
type Output = <Vec<I::Item> as Index<J>>::Output;