Skip to content

Commit 642260b

Browse files
author
Roderick Bovee
committed
Cargo fmt
1 parent 13798e4 commit 642260b

File tree

3 files changed

+35
-35
lines changed

3 files changed

+35
-35
lines changed

src/bfield.rs

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,11 @@
11
use std::io;
22
use std::path::Path;
33

4-
use serde::Serialize;
54
use serde::de::DeserializeOwned;
5+
use serde::Serialize;
66

77
use crate::bfield_member::{BFieldLookup, BFieldMember, BFieldVal};
8-
use crate::marker::{to_marker};
9-
8+
use crate::marker::to_marker;
109

1110
pub struct BField<T> {
1211
members: Vec<BFieldMember<T>>,
@@ -19,9 +18,9 @@ impl<'a, T: Clone + DeserializeOwned + Serialize> BField<T> {
1918
pub fn create<P>(
2019
filename: P,
2120
size: usize,
22-
n_hashes: u8, // k
23-
marker_width: u8, // nu
24-
n_marker_bits: u8, // kappa
21+
n_hashes: u8, // k
22+
marker_width: u8, // nu
23+
n_marker_bits: u8, // kappa
2524
secondary_scaledown: f64, // beta
2625
max_scaledown: f64,
2726
n_secondaries: u8,
@@ -43,8 +42,14 @@ impl<'a, T: Clone + DeserializeOwned + Serialize> BField<T> {
4342
} else {
4443
None
4544
};
46-
let member =
47-
BFieldMember::create(file, cur_size, n_hashes, marker_width, n_marker_bits, params)?;
45+
let member = BFieldMember::create(
46+
file,
47+
cur_size,
48+
n_hashes,
49+
marker_width,
50+
n_marker_bits,
51+
params,
52+
)?;
4853
members.push(member);
4954
cur_size = f64::max(
5055
cur_size as f64 * secondary_scaledown,
@@ -84,13 +89,10 @@ impl<'a, T: Clone + DeserializeOwned + Serialize> BField<T> {
8489
if members.is_empty() {
8590
return Err(io::Error::new(
8691
io::ErrorKind::NotFound,
87-
format!("No Bfield found at {:?}", filename.as_ref())
92+
format!("No Bfield found at {:?}", filename.as_ref()),
8893
));
8994
}
90-
Ok(BField {
91-
members,
92-
read_only,
93-
})
95+
Ok(BField { members, read_only })
9496
}
9597

9698
#[cfg(feature = "legacy")]
@@ -178,9 +180,7 @@ impl<'a, T: Clone + DeserializeOwned + Serialize> BField<T> {
178180
}
179181

180182
pub fn info(&self) -> Vec<(usize, u8, u8, u8)> {
181-
self.members.iter().map(|m| {
182-
m.info()
183-
}).collect()
183+
self.members.iter().map(|m| m.info()).collect()
184184
}
185185
}
186186

src/bfield_member.rs

Lines changed: 19 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -6,21 +6,20 @@ use std::intrinsics;
66
use std::io;
77
use std::path::Path;
88

9-
use bincode::{serialize, deserialize, Infinite};
10-
use mmap_bitvec::{BitVector, MmapBitVec, BitVecSlice};
9+
use bincode::{deserialize, serialize, Infinite};
10+
use mmap_bitvec::{BitVecSlice, BitVector, MmapBitVec};
1111
use murmurhash3::murmurhash3_x64_128;
12-
use serde::Serialize;
1312
use serde::de::DeserializeOwned;
13+
use serde::Serialize;
1414
#[cfg(feature = "legacy")]
1515
use serde_json;
1616

1717
use crate::marker::{from_marker, to_marker};
1818

19-
2019
#[derive(Debug, Deserialize, Serialize)]
2120
pub(crate) struct BFieldParams<T> {
22-
n_hashes: u8, // k
23-
marker_width: u8, // nu
21+
n_hashes: u8, // k
22+
marker_width: u8, // nu
2423
n_marker_bits: u8, // kappa
2524
pub(crate) other: Option<T>,
2625
}
@@ -148,10 +147,8 @@ impl<T: Clone + DeserializeOwned + Serialize> BFieldMember<T> {
148147

149148
for marker_ix in 0usize..self.params.n_hashes as usize {
150149
let pos = marker_pos(hash, marker_ix, self.bitvec.size(), marker_width);
151-
self.bitvec.set_range(
152-
pos..pos + marker_width,
153-
aligned_marker,
154-
);
150+
self.bitvec
151+
.set_range(pos..pos + marker_width, aligned_marker);
155152
}
156153
}
157154

@@ -168,7 +165,7 @@ impl<T: Clone + DeserializeOwned + Serialize> BFieldMember<T> {
168165
let existing_marker = self.get_raw(key, k);
169166

170167
match existing_marker.count_ones().cmp(&k) {
171-
Ordering::Greater => false, // already indeterminate
168+
Ordering::Greater => false, // already indeterminate
172169
Ordering::Equal => {
173170
// value already in b-field, but is it correct?
174171
if existing_marker == correct_marker {
@@ -186,12 +183,12 @@ impl<T: Clone + DeserializeOwned + Serialize> BFieldMember<T> {
186183
// mask out the existing!
187184
self.insert_raw(key, new_marker);
188185
false
189-
},
186+
}
190187
Ordering::Less => {
191188
// nothing present; insert the value
192189
self.insert_raw(key, correct_marker);
193190
true
194-
},
191+
}
195192
}
196193
}
197194

@@ -211,7 +208,7 @@ impl<T: Clone + DeserializeOwned + Serialize> BFieldMember<T> {
211208
let marker_width = self.params.marker_width as usize;
212209
let hash = murmurhash3_x64_128(key, 0);
213210
let mut merged_marker = BitVecSlice::max_value();
214-
let mut positions: [usize; 16] = [0; 16]; // support up to 16 hashes
211+
let mut positions: [usize; 16] = [0; 16]; // support up to 16 hashes
215212
for marker_ix in 0usize..self.params.n_hashes as usize {
216213
let pos = marker_pos(hash, marker_ix, self.bitvec.size(), marker_width);
217214
positions[marker_ix] = pos;
@@ -229,7 +226,7 @@ impl<T: Clone + DeserializeOwned + Serialize> BFieldMember<T> {
229226

230227
assert!(self.params.n_hashes <= 16);
231228
for pos in positions.iter().take(self.params.n_hashes as usize) {
232-
let marker = self.bitvec.get_range(*pos..*pos + marker_width);
229+
let marker = self.bitvec.get_range(*pos..*pos + marker_width);
233230
merged_marker &= marker;
234231
if merged_marker.count_ones() < k {
235232
return 0;
@@ -239,7 +236,12 @@ impl<T: Clone + DeserializeOwned + Serialize> BFieldMember<T> {
239236
}
240237

241238
pub fn info(&self) -> (usize, u8, u8, u8) {
242-
(self.bitvec.size(), self.params.n_hashes, self.params.marker_width, self.params.n_marker_bits)
239+
(
240+
self.bitvec.size(),
241+
self.params.n_hashes,
242+
self.params.marker_width,
243+
self.params.n_marker_bits,
244+
)
243245
}
244246
}
245247

@@ -319,7 +321,7 @@ fn test_bfield_bits_set() {
319321
bfield.insert(b"test2", 200);
320322
assert_eq!(bfield.bitvec.rank(0..128), 16);
321323
bfield.insert(b"test3", 300);
322-
assert!(bfield.bitvec.rank(0..128) < 24); // 23 bits set
324+
assert!(bfield.bitvec.rank(0..128) < 24); // 23 bits set
323325
}
324326

325327
#[test]

src/marker.rs

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,6 @@ pub fn to_marker(value: BFieldVal, k: u8) -> BitVecSlice {
4242
}
4343
}
4444

45-
4645
#[cfg(not(feature = "marker_lookup"))]
4746
pub fn to_marker(value: BFieldVal, k: u8) -> BitVecSlice {
4847
// set the appropriate number of bits in the marker
@@ -142,7 +141,6 @@ fn next_marker(marker: BitVecSlice) -> BitVecSlice {
142141
(t + 1) | (((!t & (t + 1)) - 1) >> (marker.trailing_zeros() + 1))
143142
}
144143

145-
146144
#[test]
147145
fn test_next_marker() {
148146
assert_eq!(next_marker(0b1), 0b10);

0 commit comments

Comments
 (0)