108 lines
2.9 KiB
Rust
108 lines
2.9 KiB
Rust
use fixedbitset::FixedBitSet;
|
|
use std::sync::{Arc, Mutex};
|
|
|
|
use crate::io_engine::IoEngine;
|
|
use crate::pdata::array::{self, ArrayBlock};
|
|
use crate::pdata::array_walker::{ArrayVisitor, ArrayWalker};
|
|
use crate::pdata::space_map::*;
|
|
|
|
pub struct CheckedBitSet {
|
|
bits: FixedBitSet,
|
|
}
|
|
|
|
impl CheckedBitSet {
|
|
pub fn with_capacity(bits: usize) -> CheckedBitSet {
|
|
CheckedBitSet {
|
|
bits: FixedBitSet::with_capacity(bits << 1),
|
|
}
|
|
}
|
|
|
|
pub fn set(&mut self, bit: usize, enabled: bool) {
|
|
self.bits.set(bit << 1, true);
|
|
self.bits.set((bit << 1) + 1, enabled);
|
|
}
|
|
|
|
pub fn contains(&self, bit: usize) -> Option<bool> {
|
|
if !self.bits.contains(bit << 1) {
|
|
return None;
|
|
}
|
|
Some(self.bits.contains((bit << 1) + 1))
|
|
}
|
|
}
|
|
|
|
struct BitsetVisitor {
|
|
nr_entries: usize,
|
|
bits: Mutex<CheckedBitSet>,
|
|
}
|
|
|
|
impl BitsetVisitor {
|
|
pub fn new(nr_entries: usize) -> Self {
|
|
BitsetVisitor {
|
|
nr_entries,
|
|
bits: Mutex::new(CheckedBitSet::with_capacity(nr_entries)),
|
|
}
|
|
}
|
|
|
|
pub fn get_bitset(self) -> CheckedBitSet {
|
|
self.bits.into_inner().unwrap()
|
|
}
|
|
}
|
|
|
|
impl ArrayVisitor<u64> for BitsetVisitor {
|
|
fn visit(&self, index: u64, b: ArrayBlock<u64>) -> array::Result<()> {
|
|
let mut begin = index as usize * (b.header.max_entries as usize) << 6;
|
|
|
|
for i in 0..b.header.nr_entries as usize {
|
|
if begin > self.nr_entries as usize {
|
|
return Err(array::value_err("bitset size exceeds expectation".to_string()));
|
|
}
|
|
|
|
let end: usize = std::cmp::min(begin + 64, self.nr_entries as usize);
|
|
let mut mask = 1;
|
|
let bits = b.values[i];
|
|
|
|
for bi in begin..end {
|
|
self.bits.lock().unwrap().set(bi, bits & mask != 0);
|
|
mask <<= 1;
|
|
}
|
|
begin += 64;
|
|
}
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
// TODO: multi-threaded is possible
|
|
pub fn read_bitset(
|
|
engine: Arc<dyn IoEngine + Send + Sync>,
|
|
root: u64,
|
|
nr_entries: usize,
|
|
ignore_none_fatal: bool,
|
|
)-> (CheckedBitSet, Option<array::ArrayError>) {
|
|
let w = ArrayWalker::new(engine, ignore_none_fatal);
|
|
let mut v = BitsetVisitor::new(nr_entries);
|
|
let err = w.walk(&mut v, root);
|
|
let e = match err {
|
|
Ok(()) => None,
|
|
Err(e) => Some(e),
|
|
};
|
|
return (v.get_bitset(), e);
|
|
}
|
|
|
|
// TODO: multi-threaded is possible
|
|
pub fn read_bitset_with_sm(
|
|
engine: Arc<dyn IoEngine + Send + Sync>,
|
|
root: u64,
|
|
nr_entries: usize,
|
|
sm: Arc<Mutex<dyn SpaceMap + Send + Sync>>,
|
|
ignore_none_fatal: bool,
|
|
)-> array::Result<(CheckedBitSet, Option<array::ArrayError>)> {
|
|
let w = ArrayWalker::new_with_sm(engine, sm, ignore_none_fatal)?;
|
|
let mut v = BitsetVisitor::new(nr_entries);
|
|
let err = w.walk(&mut v, root);
|
|
let e = match err {
|
|
Ok(()) => None,
|
|
Err(e) => Some(e),
|
|
};
|
|
return Ok((v.get_bitset(), e));
|
|
}
|