use crate::SingleItemStorage;
#[cfg(feature = "alloc")]
use alloc::{string::String, vec::Vec};
pub trait Capacity {
fn capacity(&self) -> usize;
}
impl<T> Capacity for &T
where
T: Capacity,
{
#[inline]
fn capacity(&self) -> usize {
(*self).capacity()
}
}
impl Capacity for () {
#[inline]
fn capacity(&self) -> usize {
0
}
}
impl<T> Capacity for Option<T> {
#[inline]
fn capacity(&self) -> usize {
1
}
}
impl<T> Capacity for SingleItemStorage<T> {
#[inline]
fn capacity(&self) -> usize {
1
}
}
impl<T, const N: usize> Capacity for [T; N] {
#[inline]
fn capacity(&self) -> usize {
N
}
}
impl<T> Capacity for &'_ [T] {
#[inline]
fn capacity(&self) -> usize {
self.len()
}
}
impl<T> Capacity for &'_ mut [T] {
#[inline]
fn capacity(&self) -> usize {
self.len()
}
}
#[cfg(feature = "alloc")]
impl Capacity for String {
#[inline]
fn capacity(&self) -> usize {
self.capacity()
}
}
#[cfg(feature = "alloc")]
impl<T> Capacity for Vec<T> {
#[inline]
fn capacity(&self) -> usize {
self.capacity()
}
}
#[cfg(feature = "arrayvec")]
impl<const N: usize> Capacity for arrayvec::ArrayString<N> {
#[inline]
fn capacity(&self) -> usize {
self.capacity()
}
}
#[cfg(feature = "arrayvec")]
impl<T, const N: usize> Capacity for arrayvec::ArrayVec<T, N> {
#[inline]
fn capacity(&self) -> usize {
self.capacity()
}
}
#[cfg(feature = "smallvec")]
impl<A> Capacity for smallvec::SmallVec<A>
where
A: smallvec::Array,
{
#[inline]
fn capacity(&self) -> usize {
self.capacity()
}
}
#[cfg(feature = "tinyvec")]
impl<A> Capacity for tinyvec::ArrayVec<A>
where
A: tinyvec::Array,
A::Item: Default,
{
#[inline]
fn capacity(&self) -> usize {
self.capacity()
}
}
#[cfg(all(feature = "alloc", feature = "tinyvec"))]
impl<A> Capacity for tinyvec::TinyVec<A>
where
A: tinyvec::Array,
A::Item: Default,
{
#[inline]
fn capacity(&self) -> usize {
self.capacity()
}
}