use super::*;
use core::convert::{TryFrom, TryInto};
#[cfg(feature = "serde")]
use core::marker::PhantomData;
#[cfg(feature = "serde")]
use serde::de::{
Deserialize, Deserializer, Error as DeserializeError, SeqAccess, Visitor,
};
#[cfg(feature = "serde")]
use serde::ser::{Serialize, SerializeSeq, Serializer};
#[macro_export]
macro_rules! array_vec {
($array_type:ty => $($elem:expr),* $(,)?) => {
{
let mut av: $crate::ArrayVec<$array_type> = Default::default();
$( av.push($elem); )*
av
}
};
($array_type:ty) => {
$crate::ArrayVec::<$array_type>::default()
};
($($elem:expr),*) => {
$crate::array_vec!(_ => $($elem),*)
};
($elem:expr; $n:expr) => {
$crate::ArrayVec::from([$elem; $n])
};
() => {
$crate::array_vec!(_)
};
}
#[repr(C)]
pub struct ArrayVec<A> {
len: u16,
pub(crate) data: A,
}
impl<A> Clone for ArrayVec<A>
where
A: Array + Clone,
A::Item: Clone,
{
#[inline]
fn clone(&self) -> Self {
Self { data: self.data.clone(), len: self.len }
}
#[inline]
fn clone_from(&mut self, o: &Self) {
let iter = self
.data
.as_slice_mut()
.iter_mut()
.zip(o.data.as_slice())
.take(self.len.max(o.len) as usize);
for (dst, src) in iter {
dst.clone_from(src)
}
if let Some(to_drop) =
self.data.as_slice_mut().get_mut((o.len as usize)..(self.len as usize))
{
to_drop.iter_mut().for_each(|x| drop(take(x)));
}
self.len = o.len;
}
}
impl<A> Copy for ArrayVec<A>
where
A: Array + Copy,
A::Item: Copy,
{
}
impl<A: Array> Default for ArrayVec<A> {
fn default() -> Self {
Self { len: 0, data: A::default() }
}
}
impl<A: Array> Deref for ArrayVec<A> {
type Target = [A::Item];
#[inline(always)]
#[must_use]
fn deref(&self) -> &Self::Target {
&self.data.as_slice()[..self.len as usize]
}
}
impl<A: Array> DerefMut for ArrayVec<A> {
#[inline(always)]
#[must_use]
fn deref_mut(&mut self) -> &mut Self::Target {
&mut self.data.as_slice_mut()[..self.len as usize]
}
}
impl<A: Array, I: SliceIndex<[A::Item]>> Index<I> for ArrayVec<A> {
type Output = <I as SliceIndex<[A::Item]>>::Output;
#[inline(always)]
#[must_use]
fn index(&self, index: I) -> &Self::Output {
&self.deref()[index]
}
}
impl<A: Array, I: SliceIndex<[A::Item]>> IndexMut<I> for ArrayVec<A> {
#[inline(always)]
#[must_use]
fn index_mut(&mut self, index: I) -> &mut Self::Output {
&mut self.deref_mut()[index]
}
}
#[cfg(feature = "serde")]
#[cfg_attr(docs_rs, doc(cfg(feature = "serde")))]
impl<A: Array> Serialize for ArrayVec<A>
where
A::Item: Serialize,
{
#[must_use]
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: Serializer,
{
let mut seq = serializer.serialize_seq(Some(self.len()))?;
for element in self.iter() {
seq.serialize_element(element)?;
}
seq.end()
}
}
#[cfg(feature = "serde")]
#[cfg_attr(docs_rs, doc(cfg(feature = "serde")))]
impl<'de, A: Array> Deserialize<'de> for ArrayVec<A>
where
A::Item: Deserialize<'de>,
{
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: Deserializer<'de>,
{
deserializer.deserialize_seq(ArrayVecVisitor(PhantomData))
}
}
#[cfg(all(feature = "arbitrary", feature = "nightly_const_generics"))]
#[cfg_attr(
docs_rs,
doc(cfg(all(feature = "arbitrary", feature = "nightly_const_generics")))
)]
impl<'a, T, const N: usize> arbitrary::Arbitrary<'a> for ArrayVec<[T; N]>
where
T: arbitrary::Arbitrary<'a> + Default,
{
fn arbitrary(u: &mut arbitrary::Unstructured<'a>) -> arbitrary::Result<Self> {
let v = <[T; N]>::arbitrary(u)?;
let av = ArrayVec::from(v);
Ok(av)
}
}
impl<A: Array> ArrayVec<A> {
#[inline]
pub fn append(&mut self, other: &mut Self) {
assert!(
self.try_append(other).is_none(),
"ArrayVec::append> total length {} exceeds capacity {}!",
self.len() + other.len(),
A::CAPACITY
);
}
#[inline]
pub fn try_append<'other>(
&mut self, other: &'other mut Self,
) -> Option<&'other mut Self> {
let new_len = self.len() + other.len();
if new_len > A::CAPACITY {
return Some(other);
}
let iter = other.iter_mut().map(take);
for item in iter {
self.push(item);
}
other.set_len(0);
return None;
}
#[inline(always)]
#[must_use]
pub fn as_mut_ptr(&mut self) -> *mut A::Item {
self.data.as_slice_mut().as_mut_ptr()
}
#[inline(always)]
#[must_use]
pub fn as_mut_slice(&mut self) -> &mut [A::Item] {
self.deref_mut()
}
#[inline(always)]
#[must_use]
pub fn as_ptr(&self) -> *const A::Item {
self.data.as_slice().as_ptr()
}
#[inline(always)]
#[must_use]
pub fn as_slice(&self) -> &[A::Item] {
self.deref()
}
#[inline(always)]
#[must_use]
pub fn capacity(&self) -> usize {
self.data.as_slice().len()
}
#[inline(always)]
pub fn clear(&mut self) {
self.truncate(0)
}
#[inline]
pub fn drain<R>(&mut self, range: R) -> ArrayVecDrain<'_, A::Item>
where
R: RangeBounds<usize>,
{
ArrayVecDrain::new(self, range)
}
#[inline]
pub fn into_inner(self) -> A {
self.data
}
#[inline]
pub fn extend_from_slice(&mut self, sli: &[A::Item])
where
A::Item: Clone,
{
if sli.is_empty() {
return;
}
let new_len = self.len as usize + sli.len();
assert!(
new_len <= A::CAPACITY,
"ArrayVec::extend_from_slice> total length {} exceeds capacity {}!",
new_len,
A::CAPACITY
);
let target = &mut self.data.as_slice_mut()[self.len as usize..new_len];
target.clone_from_slice(sli);
self.set_len(new_len);
}
#[inline]
pub fn fill<I: IntoIterator<Item = A::Item>>(
&mut self, iter: I,
) -> I::IntoIter {
let mut iter = iter.into_iter();
let mut pushed = 0;
let to_take = self.capacity() - self.len();
let target = &mut self.data.as_slice_mut()[self.len as usize..];
for element in iter.by_ref().take(to_take) {
target[pushed] = element;
pushed += 1;
}
self.len += pushed as u16;
iter
}
#[inline]
#[must_use]
#[allow(clippy::match_wild_err_arm)]
pub fn from_array_len(data: A, len: usize) -> Self {
match Self::try_from_array_len(data, len) {
Ok(out) => out,
Err(_) => panic!(
"ArrayVec::from_array_len> length {} exceeds capacity {}!",
len,
A::CAPACITY
),
}
}
#[inline]
pub fn insert(&mut self, index: usize, item: A::Item) {
let x = self.try_insert(index, item);
assert!(x.is_none(), "ArrayVec::insert> capacity overflow!");
}
#[inline]
pub fn try_insert(
&mut self, index: usize, mut item: A::Item,
) -> Option<A::Item> {
assert!(
index <= self.len as usize,
"ArrayVec::try_insert> index {} is out of bounds {}",
index,
self.len
);
if (self.len as usize) < A::CAPACITY {
self.len += 1;
} else {
return Some(item);
}
let target = &mut self.as_mut_slice()[index..];
for i in 0..target.len() {
core::mem::swap(&mut item, &mut target[i]);
}
return None;
}
#[inline(always)]
#[must_use]
pub fn is_empty(&self) -> bool {
self.len == 0
}
#[inline(always)]
#[must_use]
pub fn len(&self) -> usize {
self.len as usize
}
#[inline(always)]
#[must_use]
pub fn new() -> Self {
Self::default()
}
#[inline]
pub fn pop(&mut self) -> Option<A::Item> {
if self.len > 0 {
self.len -= 1;
let out = take(&mut self.data.as_slice_mut()[self.len as usize]);
Some(out)
} else {
None
}
}
#[inline(always)]
pub fn push(&mut self, val: A::Item) {
let x = self.try_push(val);
assert!(x.is_none(), "ArrayVec::push> capacity overflow!");
}
#[inline(always)]
pub fn try_push(&mut self, val: A::Item) -> Option<A::Item> {
debug_assert!(self.len as usize <= A::CAPACITY);
let itemref = match self.data.as_slice_mut().get_mut(self.len as usize) {
None => return Some(val),
Some(x) => x,
};
*itemref = val;
self.len += 1;
return None;
}
#[inline]
pub fn remove(&mut self, index: usize) -> A::Item {
let targets: &mut [A::Item] = &mut self.deref_mut()[index..];
let item = take(&mut targets[0]);
for i in 0..targets.len() - 1 {
targets.swap(i, i + 1);
}
self.len -= 1;
item
}
#[inline]
pub fn resize(&mut self, new_len: usize, new_val: A::Item)
where
A::Item: Clone,
{
self.resize_with(new_len, || new_val.clone())
}
#[inline]
pub fn resize_with<F: FnMut() -> A::Item>(
&mut self, new_len: usize, mut f: F,
) {
match new_len.checked_sub(self.len as usize) {
None => self.truncate(new_len),
Some(new_elements) => {
for _ in 0..new_elements {
self.push(f());
}
}
}
}
#[inline]
pub fn retain<F: FnMut(&A::Item) -> bool>(&mut self, mut acceptable: F) {
struct JoinOnDrop<'vec, Item> {
items: &'vec mut [Item],
done_end: usize,
tail_start: usize,
}
impl<Item> Drop for JoinOnDrop<'_, Item> {
fn drop(&mut self) {
self.items[self.done_end..].rotate_left(self.tail_start);
}
}
let mut rest = JoinOnDrop {
items: &mut self.data.as_slice_mut()[..self.len as usize],
done_end: 0,
tail_start: 0,
};
let len = self.len as usize;
for idx in 0..len {
if !acceptable(&rest.items[idx]) {
let _ = take(&mut rest.items[idx]);
self.len -= 1;
rest.tail_start += 1;
} else {
rest.items.swap(rest.done_end, idx);
rest.done_end += 1;
}
}
}
#[inline(always)]
pub fn set_len(&mut self, new_len: usize) {
if new_len > A::CAPACITY {
panic!(
"ArrayVec::set_len> new length {} exceeds capacity {}",
new_len,
A::CAPACITY
)
}
let new_len: u16 = new_len
.try_into()
.expect("ArrayVec::set_len> new length is not in range 0..=u16::MAX");
self.len = new_len;
}
#[inline]
pub fn split_off(&mut self, at: usize) -> Self {
if at > self.len() {
panic!(
"ArrayVec::split_off> at value {} exceeds length of {}",
at, self.len
);
}
let mut new = Self::default();
let moves = &mut self.as_mut_slice()[at..];
let split_len = moves.len();
let targets = &mut new.data.as_slice_mut()[..split_len];
moves.swap_with_slice(targets);
new.len = split_len as u16;
self.len = at as u16;
new
}
#[inline]
pub fn splice<R, I>(
&mut self, range: R, replacement: I,
) -> ArrayVecSplice<'_, A, core::iter::Fuse<I::IntoIter>>
where
R: RangeBounds<usize>,
I: IntoIterator<Item = A::Item>,
{
use core::ops::Bound;
let start = match range.start_bound() {
Bound::Included(x) => *x,
Bound::Excluded(x) => x.saturating_add(1),
Bound::Unbounded => 0,
};
let end = match range.end_bound() {
Bound::Included(x) => x.saturating_add(1),
Bound::Excluded(x) => *x,
Bound::Unbounded => self.len(),
};
assert!(
start <= end,
"ArrayVec::splice> Illegal range, {} to {}",
start,
end
);
assert!(
end <= self.len(),
"ArrayVec::splice> Range ends at {} but length is only {}!",
end,
self.len()
);
ArrayVecSplice {
removal_start: start,
removal_end: end,
parent: self,
replacement: replacement.into_iter().fuse(),
}
}
#[inline]
pub fn swap_remove(&mut self, index: usize) -> A::Item {
assert!(
index < self.len(),
"ArrayVec::swap_remove> index {} is out of bounds {}",
index,
self.len
);
if index == self.len() - 1 {
self.pop().unwrap()
} else {
let i = self.pop().unwrap();
replace(&mut self[index], i)
}
}
#[inline]
pub fn truncate(&mut self, new_len: usize) {
if new_len >= self.len as usize {
return;
}
if needs_drop::<A::Item>() {
let len = self.len as usize;
self.data.as_slice_mut()[new_len..len]
.iter_mut()
.map(take)
.for_each(drop);
}
self.len = new_len as u16;
}
#[inline]
pub fn try_from_array_len(data: A, len: usize) -> Result<Self, A> {
if len <= A::CAPACITY {
Ok(Self { data, len: len as u16 })
} else {
Err(data)
}
}
}
impl<A> ArrayVec<A> {
#[inline]
#[must_use]
pub const fn from_array_empty(data: A) -> Self {
Self { data, len: 0 }
}
}
#[cfg(feature = "grab_spare_slice")]
impl<A: Array> ArrayVec<A> {
#[inline(always)]
pub fn grab_spare_slice(&self) -> &[A::Item] {
&self.data.as_slice()[self.len as usize..]
}
#[inline(always)]
pub fn grab_spare_slice_mut(&mut self) -> &mut [A::Item] {
&mut self.data.as_slice_mut()[self.len as usize..]
}
}
#[cfg(feature = "nightly_slice_partition_dedup")]
impl<A: Array> ArrayVec<A> {
#[inline(always)]
pub fn dedup(&mut self)
where
A::Item: PartialEq,
{
self.dedup_by(|a, b| a == b)
}
#[inline(always)]
pub fn dedup_by<F>(&mut self, same_bucket: F)
where
F: FnMut(&mut A::Item, &mut A::Item) -> bool,
{
let len = {
let (dedup, _) = self.as_mut_slice().partition_dedup_by(same_bucket);
dedup.len()
};
self.truncate(len);
}
#[inline(always)]
pub fn dedup_by_key<F, K>(&mut self, mut key: F)
where
F: FnMut(&mut A::Item) -> K,
K: PartialEq,
{
self.dedup_by(|a, b| key(a) == key(b))
}
}
pub struct ArrayVecSplice<'p, A: Array, I: Iterator<Item = A::Item>> {
parent: &'p mut ArrayVec<A>,
removal_start: usize,
removal_end: usize,
replacement: I,
}
impl<'p, A: Array, I: Iterator<Item = A::Item>> Iterator
for ArrayVecSplice<'p, A, I>
{
type Item = A::Item;
#[inline]
fn next(&mut self) -> Option<A::Item> {
if self.removal_start < self.removal_end {
match self.replacement.next() {
Some(replacement) => {
let removed = core::mem::replace(
&mut self.parent[self.removal_start],
replacement,
);
self.removal_start += 1;
Some(removed)
}
None => {
let removed = self.parent.remove(self.removal_start);
self.removal_end -= 1;
Some(removed)
}
}
} else {
None
}
}
#[inline]
fn size_hint(&self) -> (usize, Option<usize>) {
let len = self.len();
(len, Some(len))
}
}
impl<'p, A, I> ExactSizeIterator for ArrayVecSplice<'p, A, I>
where
A: Array,
I: Iterator<Item = A::Item>,
{
#[inline]
fn len(&self) -> usize {
self.removal_end - self.removal_start
}
}
impl<'p, A, I> FusedIterator for ArrayVecSplice<'p, A, I>
where
A: Array,
I: Iterator<Item = A::Item>,
{
}
impl<'p, A, I> DoubleEndedIterator for ArrayVecSplice<'p, A, I>
where
A: Array,
I: Iterator<Item = A::Item> + DoubleEndedIterator,
{
#[inline]
fn next_back(&mut self) -> Option<A::Item> {
if self.removal_start < self.removal_end {
match self.replacement.next_back() {
Some(replacement) => {
let removed = core::mem::replace(
&mut self.parent[self.removal_end - 1],
replacement,
);
self.removal_end -= 1;
Some(removed)
}
None => {
let removed = self.parent.remove(self.removal_end - 1);
self.removal_end -= 1;
Some(removed)
}
}
} else {
None
}
}
}
impl<'p, A: Array, I: Iterator<Item = A::Item>> Drop
for ArrayVecSplice<'p, A, I>
{
fn drop(&mut self) {
for _ in self.by_ref() {}
for replacement in self.replacement.by_ref() {
self.parent.insert(self.removal_end, replacement);
self.removal_end += 1;
}
}
}
impl<A: Array> AsMut<[A::Item]> for ArrayVec<A> {
#[inline(always)]
#[must_use]
fn as_mut(&mut self) -> &mut [A::Item] {
&mut *self
}
}
impl<A: Array> AsRef<[A::Item]> for ArrayVec<A> {
#[inline(always)]
#[must_use]
fn as_ref(&self) -> &[A::Item] {
&*self
}
}
impl<A: Array> Borrow<[A::Item]> for ArrayVec<A> {
#[inline(always)]
#[must_use]
fn borrow(&self) -> &[A::Item] {
&*self
}
}
impl<A: Array> BorrowMut<[A::Item]> for ArrayVec<A> {
#[inline(always)]
#[must_use]
fn borrow_mut(&mut self) -> &mut [A::Item] {
&mut *self
}
}
impl<A: Array> Extend<A::Item> for ArrayVec<A> {
#[inline]
fn extend<T: IntoIterator<Item = A::Item>>(&mut self, iter: T) {
for t in iter {
self.push(t)
}
}
}
impl<A: Array> From<A> for ArrayVec<A> {
#[inline(always)]
#[must_use]
fn from(data: A) -> Self {
let len: u16 = data
.as_slice()
.len()
.try_into()
.expect("ArrayVec::from> length must be in range 0..=u16::MAX");
Self { len, data }
}
}
#[derive(Debug, Copy, Clone)]
pub struct TryFromSliceError(());
impl core::fmt::Display for TryFromSliceError {
fn fmt(&self, f: &mut Formatter<'_>) -> core::fmt::Result {
f.write_str("could not convert slice to ArrayVec")
}
}
#[cfg(feature = "std")]
impl std::error::Error for TryFromSliceError {}
impl<T, A> TryFrom<&'_ [T]> for ArrayVec<A>
where
T: Clone + Default,
A: Array<Item = T>,
{
type Error = TryFromSliceError;
#[inline]
#[must_use]
fn try_from(slice: &[T]) -> Result<Self, Self::Error> {
if slice.len() > A::CAPACITY {
Err(TryFromSliceError(()))
} else {
let mut arr = ArrayVec::new();
arr.set_len(slice.len());
arr.as_mut_slice().clone_from_slice(slice);
Ok(arr)
}
}
}
impl<A: Array> FromIterator<A::Item> for ArrayVec<A> {
#[inline]
#[must_use]
fn from_iter<T: IntoIterator<Item = A::Item>>(iter: T) -> Self {
let mut av = Self::default();
for i in iter {
av.push(i)
}
av
}
}
pub struct ArrayVecIterator<A: Array> {
base: u16,
tail: u16,
data: A,
}
impl<A: Array> ArrayVecIterator<A> {
#[inline]
#[must_use]
pub fn as_slice(&self) -> &[A::Item] {
&self.data.as_slice()[self.base as usize..self.tail as usize]
}
}
impl<A: Array> FusedIterator for ArrayVecIterator<A> {}
impl<A: Array> Iterator for ArrayVecIterator<A> {
type Item = A::Item;
#[inline]
fn next(&mut self) -> Option<Self::Item> {
let slice =
&mut self.data.as_slice_mut()[self.base as usize..self.tail as usize];
let itemref = slice.first_mut()?;
self.base += 1;
return Some(take(itemref));
}
#[inline(always)]
#[must_use]
fn size_hint(&self) -> (usize, Option<usize>) {
let s = self.tail - self.base;
let s = s as usize;
(s, Some(s))
}
#[inline(always)]
fn count(self) -> usize {
self.size_hint().0
}
#[inline]
fn last(mut self) -> Option<Self::Item> {
self.next_back()
}
#[inline]
fn nth(&mut self, n: usize) -> Option<A::Item> {
let slice = &mut self.data.as_slice_mut();
let slice = &mut slice[self.base as usize..self.tail as usize];
if let Some(x) = slice.get_mut(n) {
self.base += n as u16 + 1;
return Some(take(x));
}
self.base = self.tail;
return None;
}
}
impl<A: Array> DoubleEndedIterator for ArrayVecIterator<A> {
#[inline]
fn next_back(&mut self) -> Option<Self::Item> {
let slice =
&mut self.data.as_slice_mut()[self.base as usize..self.tail as usize];
let item = slice.last_mut()?;
self.tail -= 1;
return Some(take(item));
}
#[cfg(feature = "rustc_1_40")]
#[inline]
fn nth_back(&mut self, n: usize) -> Option<Self::Item> {
let base = self.base as usize;
let tail = self.tail as usize;
let slice = &mut self.data.as_slice_mut()[base..tail];
let n = n.saturating_add(1);
if let Some(n) = slice.len().checked_sub(n) {
let item = &mut slice[n];
self.tail = self.base + n as u16;
return Some(take(item));
}
self.tail = self.base;
return None;
}
}
impl<A: Array> Debug for ArrayVecIterator<A>
where
A::Item: Debug,
{
#[allow(clippy::missing_inline_in_public_items)]
fn fmt(&self, f: &mut Formatter<'_>) -> core::fmt::Result {
f.debug_tuple("ArrayVecIterator").field(&self.as_slice()).finish()
}
}
impl<A: Array> IntoIterator for ArrayVec<A> {
type Item = A::Item;
type IntoIter = ArrayVecIterator<A>;
#[inline(always)]
#[must_use]
fn into_iter(self) -> Self::IntoIter {
ArrayVecIterator { base: 0, tail: self.len, data: self.data }
}
}
impl<'a, A: Array> IntoIterator for &'a mut ArrayVec<A> {
type Item = &'a mut A::Item;
type IntoIter = core::slice::IterMut<'a, A::Item>;
#[inline(always)]
#[must_use]
fn into_iter(self) -> Self::IntoIter {
self.iter_mut()
}
}
impl<'a, A: Array> IntoIterator for &'a ArrayVec<A> {
type Item = &'a A::Item;
type IntoIter = core::slice::Iter<'a, A::Item>;
#[inline(always)]
#[must_use]
fn into_iter(self) -> Self::IntoIter {
self.iter()
}
}
impl<A: Array> PartialEq for ArrayVec<A>
where
A::Item: PartialEq,
{
#[inline]
#[must_use]
fn eq(&self, other: &Self) -> bool {
self.as_slice().eq(other.as_slice())
}
}
impl<A: Array> Eq for ArrayVec<A> where A::Item: Eq {}
impl<A: Array> PartialOrd for ArrayVec<A>
where
A::Item: PartialOrd,
{
#[inline]
#[must_use]
fn partial_cmp(&self, other: &Self) -> Option<core::cmp::Ordering> {
self.as_slice().partial_cmp(other.as_slice())
}
}
impl<A: Array> Ord for ArrayVec<A>
where
A::Item: Ord,
{
#[inline]
#[must_use]
fn cmp(&self, other: &Self) -> core::cmp::Ordering {
self.as_slice().cmp(other.as_slice())
}
}
impl<A: Array> PartialEq<&A> for ArrayVec<A>
where
A::Item: PartialEq,
{
#[inline]
#[must_use]
fn eq(&self, other: &&A) -> bool {
self.as_slice().eq(other.as_slice())
}
}
impl<A: Array> PartialEq<&[A::Item]> for ArrayVec<A>
where
A::Item: PartialEq,
{
#[inline]
#[must_use]
fn eq(&self, other: &&[A::Item]) -> bool {
self.as_slice().eq(*other)
}
}
impl<A: Array> Hash for ArrayVec<A>
where
A::Item: Hash,
{
#[inline]
fn hash<H: Hasher>(&self, state: &mut H) {
self.as_slice().hash(state)
}
}
#[cfg(feature = "experimental_write_impl")]
impl<A: Array<Item = u8>> core::fmt::Write for ArrayVec<A> {
fn write_str(&mut self, s: &str) -> core::fmt::Result {
let my_len = self.len();
let str_len = s.as_bytes().len();
if my_len + str_len <= A::CAPACITY {
let remainder = &mut self.data.as_slice_mut()[my_len..];
let target = &mut remainder[..str_len];
target.copy_from_slice(s.as_bytes());
Ok(())
} else {
Err(core::fmt::Error)
}
}
}
impl<A: Array> Binary for ArrayVec<A>
where
A::Item: Binary,
{
#[allow(clippy::missing_inline_in_public_items)]
fn fmt(&self, f: &mut Formatter) -> core::fmt::Result {
write!(f, "[")?;
if f.alternate() {
write!(f, "\n ")?;
}
for (i, elem) in self.iter().enumerate() {
if i > 0 {
write!(f, ",{}", if f.alternate() { "\n " } else { " " })?;
}
Binary::fmt(elem, f)?;
}
if f.alternate() {
write!(f, ",\n")?;
}
write!(f, "]")
}
}
impl<A: Array> Debug for ArrayVec<A>
where
A::Item: Debug,
{
#[allow(clippy::missing_inline_in_public_items)]
fn fmt(&self, f: &mut Formatter) -> core::fmt::Result {
write!(f, "[")?;
if f.alternate() {
write!(f, "\n ")?;
}
for (i, elem) in self.iter().enumerate() {
if i > 0 {
write!(f, ",{}", if f.alternate() { "\n " } else { " " })?;
}
Debug::fmt(elem, f)?;
}
if f.alternate() {
write!(f, ",\n")?;
}
write!(f, "]")
}
}
impl<A: Array> Display for ArrayVec<A>
where
A::Item: Display,
{
#[allow(clippy::missing_inline_in_public_items)]
fn fmt(&self, f: &mut Formatter) -> core::fmt::Result {
write!(f, "[")?;
if f.alternate() {
write!(f, "\n ")?;
}
for (i, elem) in self.iter().enumerate() {
if i > 0 {
write!(f, ",{}", if f.alternate() { "\n " } else { " " })?;
}
Display::fmt(elem, f)?;
}
if f.alternate() {
write!(f, ",\n")?;
}
write!(f, "]")
}
}
impl<A: Array> LowerExp for ArrayVec<A>
where
A::Item: LowerExp,
{
#[allow(clippy::missing_inline_in_public_items)]
fn fmt(&self, f: &mut Formatter) -> core::fmt::Result {
write!(f, "[")?;
if f.alternate() {
write!(f, "\n ")?;
}
for (i, elem) in self.iter().enumerate() {
if i > 0 {
write!(f, ",{}", if f.alternate() { "\n " } else { " " })?;
}
LowerExp::fmt(elem, f)?;
}
if f.alternate() {
write!(f, ",\n")?;
}
write!(f, "]")
}
}
impl<A: Array> LowerHex for ArrayVec<A>
where
A::Item: LowerHex,
{
#[allow(clippy::missing_inline_in_public_items)]
fn fmt(&self, f: &mut Formatter) -> core::fmt::Result {
write!(f, "[")?;
if f.alternate() {
write!(f, "\n ")?;
}
for (i, elem) in self.iter().enumerate() {
if i > 0 {
write!(f, ",{}", if f.alternate() { "\n " } else { " " })?;
}
LowerHex::fmt(elem, f)?;
}
if f.alternate() {
write!(f, ",\n")?;
}
write!(f, "]")
}
}
impl<A: Array> Octal for ArrayVec<A>
where
A::Item: Octal,
{
#[allow(clippy::missing_inline_in_public_items)]
fn fmt(&self, f: &mut Formatter) -> core::fmt::Result {
write!(f, "[")?;
if f.alternate() {
write!(f, "\n ")?;
}
for (i, elem) in self.iter().enumerate() {
if i > 0 {
write!(f, ",{}", if f.alternate() { "\n " } else { " " })?;
}
Octal::fmt(elem, f)?;
}
if f.alternate() {
write!(f, ",\n")?;
}
write!(f, "]")
}
}
impl<A: Array> Pointer for ArrayVec<A>
where
A::Item: Pointer,
{
#[allow(clippy::missing_inline_in_public_items)]
fn fmt(&self, f: &mut Formatter) -> core::fmt::Result {
write!(f, "[")?;
if f.alternate() {
write!(f, "\n ")?;
}
for (i, elem) in self.iter().enumerate() {
if i > 0 {
write!(f, ",{}", if f.alternate() { "\n " } else { " " })?;
}
Pointer::fmt(elem, f)?;
}
if f.alternate() {
write!(f, ",\n")?;
}
write!(f, "]")
}
}
impl<A: Array> UpperExp for ArrayVec<A>
where
A::Item: UpperExp,
{
#[allow(clippy::missing_inline_in_public_items)]
fn fmt(&self, f: &mut Formatter) -> core::fmt::Result {
write!(f, "[")?;
if f.alternate() {
write!(f, "\n ")?;
}
for (i, elem) in self.iter().enumerate() {
if i > 0 {
write!(f, ",{}", if f.alternate() { "\n " } else { " " })?;
}
UpperExp::fmt(elem, f)?;
}
if f.alternate() {
write!(f, ",\n")?;
}
write!(f, "]")
}
}
impl<A: Array> UpperHex for ArrayVec<A>
where
A::Item: UpperHex,
{
#[allow(clippy::missing_inline_in_public_items)]
fn fmt(&self, f: &mut Formatter) -> core::fmt::Result {
write!(f, "[")?;
if f.alternate() {
write!(f, "\n ")?;
}
for (i, elem) in self.iter().enumerate() {
if i > 0 {
write!(f, ",{}", if f.alternate() { "\n " } else { " " })?;
}
UpperHex::fmt(elem, f)?;
}
if f.alternate() {
write!(f, ",\n")?;
}
write!(f, "]")
}
}
#[cfg(feature = "alloc")]
use alloc::vec::Vec;
#[cfg(all(feature = "alloc", feature = "rustc_1_57"))]
use alloc::collections::TryReserveError;
#[cfg(feature = "alloc")]
impl<A: Array> ArrayVec<A> {
pub fn drain_to_vec_and_reserve(&mut self, n: usize) -> Vec<A::Item> {
let cap = n + self.len();
let mut v = Vec::with_capacity(cap);
let iter = self.iter_mut().map(take);
v.extend(iter);
self.set_len(0);
return v;
}
#[cfg(feature = "rustc_1_57")]
pub fn try_drain_to_vec_and_reserve(
&mut self, n: usize,
) -> Result<Vec<A::Item>, TryReserveError> {
let cap = n + self.len();
let mut v = Vec::new();
v.try_reserve(cap)?;
let iter = self.iter_mut().map(take);
v.extend(iter);
self.set_len(0);
return Ok(v);
}
pub fn drain_to_vec(&mut self) -> Vec<A::Item> {
self.drain_to_vec_and_reserve(0)
}
#[cfg(feature = "rustc_1_57")]
pub fn try_drain_to_vec(&mut self) -> Result<Vec<A::Item>, TryReserveError> {
self.try_drain_to_vec_and_reserve(0)
}
}
#[cfg(feature = "serde")]
struct ArrayVecVisitor<A: Array>(PhantomData<A>);
#[cfg(feature = "serde")]
impl<'de, A: Array> Visitor<'de> for ArrayVecVisitor<A>
where
A::Item: Deserialize<'de>,
{
type Value = ArrayVec<A>;
fn expecting(
&self, formatter: &mut core::fmt::Formatter,
) -> core::fmt::Result {
formatter.write_str("a sequence")
}
fn visit_seq<S>(self, mut seq: S) -> Result<Self::Value, S::Error>
where
S: SeqAccess<'de>,
{
let mut new_arrayvec: ArrayVec<A> = Default::default();
let mut idx = 0usize;
while let Some(value) = seq.next_element()? {
if new_arrayvec.len() >= new_arrayvec.capacity() {
return Err(DeserializeError::invalid_length(idx, &self));
}
new_arrayvec.push(value);
idx = idx + 1;
}
Ok(new_arrayvec)
}
}