72 lines
1.6 KiB
Rust
72 lines
1.6 KiB
Rust
/// A vector of bits
|
|
#[derive(Clone)]
|
|
pub struct BitVec {
|
|
data: Vec<u8>,
|
|
}
|
|
|
|
impl BitVec {
|
|
pub fn new(size: usize) -> BitVec {
|
|
assert_eq!(size % 8, 0);
|
|
Self {
|
|
data: vec![0; size / 8],
|
|
}
|
|
}
|
|
|
|
pub fn load(data: &[u8]) -> BitVec {
|
|
Self {
|
|
data: Vec::from(data),
|
|
}
|
|
}
|
|
|
|
pub fn set(&mut self, index: usize, value: bool) -> bool {
|
|
let (byte_index, bit_mask) = self.get_indexes(index);
|
|
|
|
let byte = self.data[byte_index];
|
|
let old_value = byte & bit_mask != 0;
|
|
|
|
self.data[byte_index] = if value {
|
|
byte | bit_mask
|
|
} else {
|
|
byte & (u8::MAX ^ bit_mask)
|
|
};
|
|
|
|
return old_value;
|
|
}
|
|
|
|
pub fn get(&self, index: usize) -> bool {
|
|
let (byte_index, bit_mask) = self.get_indexes(index);
|
|
return self.data[byte_index] & bit_mask != 0;
|
|
}
|
|
|
|
pub fn fill(&mut self, value: bool) {
|
|
let byte: u8 = if value { 0xFF } else { 0x00 };
|
|
self.data.fill(byte);
|
|
}
|
|
|
|
pub fn len(&self) -> usize {
|
|
self.data.len() * 8
|
|
}
|
|
|
|
fn get_indexes(&self, index: usize) -> (usize, u8) {
|
|
let byte_index = index / 8;
|
|
let bit_in_byte_index = 7 - index % 8;
|
|
let bit_mask: u8 = 1 << bit_in_byte_index;
|
|
return (byte_index, bit_mask);
|
|
}
|
|
}
|
|
|
|
impl Into<Vec<u8>> for BitVec {
|
|
fn into(self) -> Vec<u8> {
|
|
self.data
|
|
}
|
|
}
|
|
|
|
impl std::fmt::Debug for BitVec {
|
|
fn fmt(&self, fmt: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
fmt.debug_struct("BitVec")
|
|
.field("len", &self.len())
|
|
.field("data", &self.data)
|
|
.finish()
|
|
}
|
|
}
|