gered/pso_gc_tools
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

add to_array helper method

Browse source
This commit is contained in:
Gered 2021-05-16 16:31:14 -04:00
parent 93c2180da2
commit 54a707f860
1 changed files with 12 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
psoutils/src/bytes.rs
View file

@ -30,19 +30,13 @@ pub trait WriteAsBytes<T: WriteBytesExt> {
pub trait FixedLengthByteArrays { pub trait FixedLengthByteArrays {
fn as_unpadded_slice(&self) -> &[u8]; fn as_unpadded_slice(&self) -> &[u8];
fn to_fixed_length(&self, length: usize) -> Vec<u8>; fn to_fixed_length(&self, length: usize) -> Vec<u8>;
fn to_array<const N: usize>(&self) -> [u8; N];
} }
impl<T: AsRef<[u8]> + ?Sized> FixedLengthByteArrays for T { impl<T: AsRef<[u8]> + ?Sized> FixedLengthByteArrays for T {
fn as_unpadded_slice(&self) -> &[u8] { fn as_unpadded_slice(&self) -> &[u8] {
let end = self.as_ref().iter().take_while(|&b| *b != 0).count(); let end = self.as_ref().iter().take_while(|&b| *b != 0).count();
&self.as_ref()[0..end] &self.as_ref()[0..end]
/*
self.as_ref()
.iter()
.take_while(|&b| *b != 0u8)
.map(|b| *b)
.collect()
*/
} }
fn to_fixed_length(&self, length: usize) -> Vec<u8> { fn to_fixed_length(&self, length: usize) -> Vec<u8> {
@ -52,6 +46,17 @@ impl<T: AsRef<[u8]> + ?Sized> FixedLengthByteArrays for T {
} }
result result
} }
fn to_array<const N: usize>(&self) -> [u8; N] {
assert_ne!(N, 0);
let mut array = [0u8; N];
if N <= self.as_ref().len() {
array.copy_from_slice(&self.as_ref()[0..N]);
} else {
array[0..self.as_ref().len()].copy_from_slice(&self.as_ref())
}
array
}
} }
#[cfg(test)] #[cfg(test)]