use std::mem::size_of;
use std::marker::PhantomData;
use std::ops::Index;
pub use rawslice::SliceIter;
#[derive(Debug)]
pub struct SliceCopyIter<'a, T: 'a> {
ptr: *const T,
end: *const T,
ty: PhantomData<&'a T>,
}
impl<'a, T> Copy for SliceCopyIter<'a, T> { }
impl<'a, T> Clone for SliceCopyIter<'a, T> {
fn clone(&self) -> Self { *self }
}
impl<'a, T> SliceCopyIter<'a, T>
where T: Copy
{
#[inline]
pub unsafe fn new(ptr: *const T, end: *const T) -> Self {
assert!(size_of::<T>() != 0);
SliceCopyIter {
ptr: ptr,
end: end,
ty: PhantomData,
}
}
pub fn into_raw(self) -> (*const T, *const T) {
(self.ptr, self.end)
}
pub fn start(&self) -> *const T {
self.ptr
}
pub fn end(&self) -> *const T {
self.end
}
pub unsafe fn start_mut(&mut self) -> &mut *const T {
&mut self.ptr
}
pub unsafe fn end_mut(&mut self) -> &mut *const T {
&mut self.end
}
pub fn peek_next(&self) -> Option<<Self as Iterator>::Item> {
if self.ptr != self.end {
unsafe {
Some(*self.ptr)
}
} else {
None
}
}
}
impl<'a, T> Iterator for SliceCopyIter<'a, T>
where T: Copy,
{
type Item = T;
#[inline]
fn next(&mut self) -> Option<Self::Item> {
if self.ptr != self.end {
unsafe {
let elt = Some(*self.ptr);
self.ptr = self.ptr.offset(1);
elt
}
} else {
None
}
}
fn size_hint(&self) -> (usize, Option<usize>) {
let len = (self.end as usize - self.ptr as usize) / size_of::<T>();
(len, Some(len))
}
fn count(self) -> usize {
self.len()
}
fn last(mut self) -> Option<Self::Item> {
self.next_back()
}
}
impl<'a, T> DoubleEndedIterator for SliceCopyIter<'a, T>
where T: Copy
{
#[inline]
fn next_back(&mut self) -> Option<Self::Item> {
if self.ptr != self.end {
unsafe {
self.end = self.end.offset(-1);
let elt = Some(*self.end);
elt
}
} else {
None
}
}
}
impl<'a, T> ExactSizeIterator for SliceCopyIter<'a, T> where T: Copy { }
impl<'a, T> From<&'a [T]> for SliceCopyIter<'a, T>
where T: Copy
{
fn from(slice: &'a [T]) -> Self {
assert!(size_of::<T>() != 0);
unsafe {
let ptr = slice.as_ptr();
let end = ptr.offset(slice.len() as isize);
SliceCopyIter::new(ptr, end)
}
}
}
impl<'a, T> Default for SliceCopyIter<'a, T>
where T: Copy
{
fn default() -> Self {
unsafe {
SliceCopyIter::new(0x1 as *const T, 0x1 as *const T)
}
}
}
impl<'a, T> Index<usize> for SliceCopyIter<'a, T>
where T: Copy
{
type Output = T;
fn index(&self, i: usize) -> &T {
assert!(i < self.len());
unsafe {
&*self.ptr.offset(i as isize)
}
}
}