use std::ops;
#[derive(Clone, Copy)]
pub struct ArrayRef<'a, T> {
data: &'a [T],
}
impl<'a, T> ArrayRef<'a, T> {
#[inline]
pub fn new(data: &'a [T]) -> Self {
Self { data }
}
#[inline]
pub fn empty() -> Self {
Self { data: &[] }
}
#[inline]
pub fn size(&self) -> usize {
self.data.len()
}
#[inline]
pub fn len(&self) -> usize {
self.data.len()
}
#[inline]
pub fn is_empty(&self) -> bool {
self.data.is_empty()
}
#[inline]
pub fn get(&self, index: usize) -> Option<&T> {
self.data.get(index)
}
#[inline]
pub fn front(&self) -> Option<&T> {
self.data.first()
}
#[inline]
pub fn back(&self) -> Option<&T> {
self.data.last()
}
#[inline]
pub fn slice(&self, start: usize, n: usize) -> ArrayRef<'a, T> {
if start >= self.data.len() {
return ArrayRef::empty();
}
let end = (start + n).min(self.data.len());
ArrayRef::new(&self.data[start..end])
}
#[inline]
pub fn drop_front(&self, n: usize) -> ArrayRef<'a, T> {
let start = n.min(self.data.len());
ArrayRef::new(&self.data[start..])
}
#[inline]
pub fn drop_back(&self, n: usize) -> ArrayRef<'a, T> {
let n = n.min(self.data.len());
ArrayRef::new(&self.data[..self.data.len() - n])
}
#[inline]
pub fn take_front(&self, n: usize) -> ArrayRef<'a, T> {
let end = n.min(self.data.len());
ArrayRef::new(&self.data[..end])
}
#[inline]
pub fn take_back(&self, n: usize) -> ArrayRef<'a, T> {
let n = n.min(self.data.len());
ArrayRef::new(&self.data[self.data.len() - n..])
}
#[inline]
pub fn data_ptr(&self) -> *const T {
self.data.as_ptr()
}
#[inline]
pub fn equals(&self, other: &[T]) -> bool
where
T: PartialEq,
{
self.data == other
}
#[inline]
pub fn iter(&self) -> impl Iterator<Item = &T> + '_ {
self.data.iter()
}
#[inline]
pub fn as_slice(&self) -> &'a [T] {
self.data
}
}
impl<'a, T> ops::Index<usize> for ArrayRef<'a, T> {
type Output = T;
fn index(&self, index: usize) -> &Self::Output {
&self.data[index]
}
}
impl<'a, T: PartialEq> PartialEq for ArrayRef<'a, T> {
fn eq(&self, other: &Self) -> bool {
self.data == other.data
}
}
impl<'a, T: Eq> Eq for ArrayRef<'a, T> {}
impl<'a, T: fmt::Debug> fmt::Debug for ArrayRef<'a, T> {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
self.data.fmt(f)
}
}
impl<'a, T> From<&'a [T]> for ArrayRef<'a, T> {
fn from(data: &'a [T]) -> Self {
ArrayRef::new(data)
}
}
impl<'a, T, const N: usize> From<&'a [T; N]> for ArrayRef<'a, T> {
fn from(data: &'a [T; N]) -> Self {
ArrayRef::new(data.as_slice())
}
}
impl<'a, T> IntoIterator for ArrayRef<'a, T> {
type Item = &'a T;
type IntoIter = std::slice::Iter<'a, T>;
fn into_iter(self) -> Self::IntoIter {
self.data.iter()
}
}
use std::fmt;