Add core::iter::adapters::PeekMap and core::iter::Peekable::peek_map

Author: MiguelX413

library/core/src/iter/adapters/mod.rs

@@ -18,6 +18,7 @@ mod intersperse;
 mod map;
 mod map_while;
 mod map_windows;
+mod peek_map;
 mod peekable;
 mod rev;
 mod scan;
@@ -31,8 +32,9 @@ mod zip;
 #[stable(feature = "rust1", since = "1.0.0")]
 pub use self::{
     chain::Chain, cycle::Cycle, enumerate::Enumerate, filter::Filter, filter_map::FilterMap,
-    flatten::FlatMap, fuse::Fuse, inspect::Inspect, map::Map, peekable::Peekable, rev::Rev,
-    scan::Scan, skip::Skip, skip_while::SkipWhile, take::Take, take_while::TakeWhile, zip::Zip,
+    flatten::FlatMap, fuse::Fuse, inspect::Inspect, map::Map, peek_map::PeekMap,
+    peekable::Peekable, rev::Rev, scan::Scan, skip::Skip, skip_while::SkipWhile, take::Take,
+    take_while::TakeWhile, zip::Zip,
 };
 
 #[unstable(feature = "iter_array_chunks", reason = "recently added", issue = "100450")]

library/core/src/iter/adapters/peek_map.rs

@@ -0,0 +1,105 @@
+use crate::iter::adapters::{SourceIter, TrustedRandomAccess, TrustedRandomAccessNoCoerce};
+use crate::iter::{FusedIterator, InPlaceIterable, Peekable, TrustedFused, TrustedLen};
+use crate::num::NonZeroUsize;
+
+/// An iterator that maps the values of `iter` with `f`.
+#[must_use = "iterators are lazy and do nothing unless consumed"]
+#[stable(feature = "rust1", since = "1.0.0")]
+pub struct PeekMap<P, F> {
+    pub(crate) peekable: P,
+    f: F,
+}
+
+impl<I: Iterator, F> PeekMap<Peekable<I>, F> {
+    pub(in crate::iter) fn new(peekable: Peekable<I>, f: F) -> PeekMap<Peekable<I>, F> {
+        PeekMap { peekable, f }
+    }
+}
+
+#[stable(feature = "rust1", since = "1.0.0")]
+impl<B, I: Iterator, F> Iterator for PeekMap<Peekable<I>, F>
+where
+    F: FnMut(I::Item, Option<&I::Item>) -> B,
+{
+    type Item = B;
+
+    #[inline]
+    fn next(&mut self) -> Option<B> {
+        self.peekable.next().map(|f| (f, self.peekable.peek())).map(&mut self.f)
+    }
+
+    #[inline]
+    fn size_hint(&self) -> (usize, Option<usize>) {
+        self.peekable.size_hint()
+    }
+}
+
+#[stable(feature = "rust1", since = "1.0.0")]
+impl<B, I: ExactSizeIterator, F> ExactSizeIterator for PeekMap<Peekable<I>, F>
+where
+    F: FnMut(I::Item, Option<&I::Item>) -> B,
+{
+    fn len(&self) -> usize {
+        self.peekable.len()
+    }
+
+    fn is_empty(&self) -> bool {
+        self.peekable.is_empty()
+    }
+}
+
+#[stable(feature = "fused", since = "1.26.0")]
+impl<B, I: FusedIterator, F> FusedIterator for PeekMap<Peekable<I>, F> where
+    F: FnMut(I::Item, Option<&I::Item>) -> B
+{
+}
+
+#[unstable(issue = "none", feature = "trusted_fused")]
+unsafe impl<I: Iterator, F> TrustedFused for PeekMap<Peekable<I>, F> where I: TrustedFused {}
+
+#[unstable(feature = "trusted_len", issue = "37572")]
+unsafe impl<B, I, F> TrustedLen for PeekMap<Peekable<I>, F>
+where
+    I: TrustedLen,
+    F: FnMut(I::Item, Option<&I::Item>) -> B,
+{
+}
+
+#[doc(hidden)]
+#[unstable(feature = "trusted_random_access", issue = "none")]
+unsafe impl<I: Iterator, F> TrustedRandomAccess for PeekMap<Peekable<I>, F> where
+    I: TrustedRandomAccess
+{
+}
+
+#[doc(hidden)]
+#[unstable(feature = "trusted_random_access", issue = "none")]
+unsafe impl<I, F> TrustedRandomAccessNoCoerce for PeekMap<Peekable<I>, F>
+where
+    I: TrustedRandomAccessNoCoerce,
+{
+    const MAY_HAVE_SIDE_EFFECT: bool = true;
+}
+
+#[unstable(issue = "none", feature = "inplace_iteration")]
+unsafe impl<I: Iterator, F> SourceIter for PeekMap<Peekable<I>, F>
+where
+    I: SourceIter,
+{
+    type Source = I::Source;
+
+    #[inline]
+    unsafe fn as_inner(&mut self) -> &mut I::Source {
+        // SAFETY: unsafe function forwarding to unsafe function with the same requirements
+        unsafe { SourceIter::as_inner(&mut self.peekable.as_inner()) }
+    }
+}
+
+#[unstable(issue = "none", feature = "inplace_iteration")]
+unsafe impl<I: Iterator, F> InPlaceIterable for PeekMap<Peekable<I>, F>
+where
+    I: InPlaceIterable,
+{
+    const EXPAND_BY: Option<NonZeroUsize> = I::EXPAND_BY;
+    const MERGE_BY: Option<NonZeroUsize> = I::MERGE_BY;
+}

library/core/src/iter/adapters/peekable.rs

@@ -1,4 +1,4 @@
-use crate::iter::{adapters::SourceIter, FusedIterator, TrustedLen};
+use crate::iter::{adapters::PeekMap, adapters::SourceIter, FusedIterator, TrustedLen};
 use crate::ops::{ControlFlow, Try};
 
 /// An iterator with a `peek()` that returns an optional reference to the next
@@ -316,6 +316,14 @@ impl<I: Iterator> Peekable<I> {
     {
         self.next_if(|next| next == expected)
     }
+
+    #[unstable(feature = "peek_map", issue = "118474")]
+    pub fn peek_map<B, F>(self, f: F) -> PeekMap<Self, F>
+    where
+        F: FnMut(I::Item, Option<&I::Item>) -> B,
+    {
+        PeekMap::new(self, f)
+    }
 }
 
 #[unstable(feature = "trusted_len", issue = "37572")]