use crate::{IterWrapper, Length, SingleItemStorage};
#[cfg(feature = "alloc")]
use alloc::{
collections::{BTreeMap, BTreeSet},
string::String,
vec::Vec,
};
#[cfg(feature = "std")]
use std::collections::{HashMap, HashSet};
pub trait SizeHint {
fn size_hint(&self) -> (usize, Option<usize>);
}
impl<T> SizeHint for IterWrapper<T>
where
T: Iterator,
{
#[inline]
fn size_hint(&self) -> (usize, Option<usize>) {
self.0.size_hint()
}
}
impl SizeHint for () {
#[inline]
fn size_hint(&self) -> (usize, Option<usize>) {
(0, Some(0))
}
}
impl<T> SizeHint for Option<T> {
#[inline]
fn size_hint(&self) -> (usize, Option<usize>) {
(self.length(), Some(self.length()))
}
}
impl<T> SizeHint for SingleItemStorage<T> {
#[inline]
fn size_hint(&self) -> (usize, Option<usize>) {
(1, Some(1))
}
}
impl<T> SizeHint for &T
where
T: SizeHint,
{
#[inline]
fn size_hint(&self) -> (usize, Option<usize>) {
(*self).size_hint()
}
}
impl<T> SizeHint for [T] {
#[inline]
fn size_hint(&self) -> (usize, Option<usize>) {
(self.len(), Some(self.len()))
}
}
impl<T> SizeHint for &'_ [T] {
#[inline]
fn size_hint(&self) -> (usize, Option<usize>) {
(self.len(), Some(self.len()))
}
}
impl<T> SizeHint for &'_ mut [T] {
#[inline]
fn size_hint(&self) -> (usize, Option<usize>) {
(self.len(), Some(self.len()))
}
}
impl<T, const N: usize> SizeHint for [T; N] {
#[inline]
fn size_hint(&self) -> (usize, Option<usize>) {
(self.len(), Some(self.len()))
}
}
#[cfg(feature = "alloc")]
impl<K, V> SizeHint for BTreeMap<K, V> {
#[inline]
fn size_hint(&self) -> (usize, Option<usize>) {
(self.len(), Some(self.len()))
}
}
#[cfg(feature = "alloc")]
impl<V> SizeHint for BTreeSet<V> {
#[inline]
fn size_hint(&self) -> (usize, Option<usize>) {
(self.len(), Some(self.len()))
}
}
#[cfg(feature = "std")]
impl<K, V> SizeHint for HashMap<K, V> {
#[inline]
fn size_hint(&self) -> (usize, Option<usize>) {
(self.len(), Some(self.len()))
}
}
#[cfg(feature = "std")]
impl<V> SizeHint for HashSet<V> {
#[inline]
fn size_hint(&self) -> (usize, Option<usize>) {
(self.len(), Some(self.len()))
}
}
#[cfg(feature = "alloc")]
impl SizeHint for String {
#[inline]
fn size_hint(&self) -> (usize, Option<usize>) {
(self.len(), Some(self.len()))
}
}
#[cfg(feature = "alloc")]
impl<T> SizeHint for Vec<T> {
#[inline]
fn size_hint(&self) -> (usize, Option<usize>) {
(self.len(), Some(self.len()))
}
}
#[cfg(feature = "arrayvec")]
impl<const N: usize> SizeHint for arrayvec::ArrayString<N> {
#[inline]
fn size_hint(&self) -> (usize, Option<usize>) {
(self.len(), Some(self.len()))
}
}
#[cfg(feature = "arrayvec")]
impl<T, const N: usize> SizeHint for arrayvec::ArrayVec<T, N> {
#[inline]
fn size_hint(&self) -> (usize, Option<usize>) {
(self.len(), Some(self.len()))
}
}
#[cfg(feature = "smallvec")]
impl<A> SizeHint for smallvec::SmallVec<A>
where
A: smallvec::Array,
{
#[inline]
fn size_hint(&self) -> (usize, Option<usize>) {
(self.len(), Some(self.len()))
}
}
#[cfg(feature = "staticvec")]
impl<T, const N: usize> SizeHint for staticvec::StaticVec<T, N> {
#[inline]
fn size_hint(&self) -> (usize, Option<usize>) {
(self.len(), Some(self.len()))
}
}
#[cfg(feature = "tinyvec")]
impl<A> SizeHint for tinyvec::ArrayVec<A>
where
A: tinyvec::Array,
A::Item: Default,
{
#[inline]
fn size_hint(&self) -> (usize, Option<usize>) {
(self.len(), Some(self.len()))
}
}
#[cfg(feature = "tinyvec")]
impl<A> SizeHint for tinyvec::TinyVec<A>
where
A: tinyvec::Array,
A::Item: Default,
{
#[inline]
fn size_hint(&self) -> (usize, Option<usize>) {
(self.len(), Some(self.len()))
}
}