You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

61 lines
1.1 KiB

use std::borrow::Borrow;
/// Iter that converts 2 `u8`s into 1 `u16`
pub struct Bit16Iter<I>
where I: Iterator,
<I as Iterator>::Item: Borrow<u8>
{
iter: I,
}
impl<I> Iterator for Bit16Iter<I>
where I: Iterator,
<I as Iterator>::Item: Borrow<u8>
{
type Item = u16;
fn next(&mut self) -> Option<Self::Item>
{
/*let mut c = 0u16;
unsafe {
if let Some(a) = self.iter.next() {
crate::reinterpret::bytes_mut(&mut c)[0] = *a.borrow();
} else {
return None;
}
if let Some(b) = self.iter.next() {
crate::reinterpret::bytes_mut(&mut c)[1] = *b.borrow();
}
}*/
let mut ar = [0u8; std::mem::size_of::<u16>()];
if let Some(a) = self.iter.next()
{
ar[0] = *a.borrow();
} else {
return None;
}
if let Some(b) = self.iter.next()
{
ar[1] = *b.borrow();
}
Some(u16::from_le_bytes(ar))
}
}
pub trait Bit16IterExt: Iterator + Sized
where <Self as Iterator>::Item: Borrow<u8>
{
fn into_16(self) -> Bit16Iter<Self>;
}
impl<I> Bit16IterExt for I
where I: Iterator,
<I as Iterator>::Item: Borrow<u8>
{
fn into_16(self) -> Bit16Iter<Self>
{
Bit16Iter{
iter: self
}
}
}