Skip to content

Commit 75d9312

Browse files
committed
refactor
1 parent 5d4b0ea commit 75d9312

File tree

1 file changed

+15
-17
lines changed

1 file changed

+15
-17
lines changed

lib/dictBuilder/zdict.rs

Lines changed: 15 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -61,10 +61,10 @@ impl DictItem {
6161
}
6262
}
6363

64-
const MINRATIO: u32 = 4;
64+
const MINRATIO: usize = 4;
6565
const ZDICT_MAX_SAMPLES_SIZE: usize = 2000 << 20;
6666
#[expect(deprecated)]
67-
const ZDICT_MIN_SAMPLES_SIZE: usize = ZDICT_CONTENTSIZE_MIN as usize * MINRATIO as usize;
67+
const ZDICT_MIN_SAMPLES_SIZE: usize = ZDICT_CONTENTSIZE_MIN as usize * MINRATIO;
6868

6969
const NOISELENGTH: usize = 32;
7070
static g_selectivity_default: u32 = 9;
@@ -152,7 +152,7 @@ unsafe fn ZDICT_analyzePos(
152152
suffix_slice: &[u32],
153153
mut start: u32,
154154
buffer: &[u8],
155-
minRatio: u32,
155+
minRatio: usize,
156156
notificationLevel: u32,
157157
) -> DictItem {
158158
let mut lengthList = [0u32; LLIMIT];
@@ -221,7 +221,7 @@ unsafe fn ZDICT_analyzePos(
221221
}
222222

223223
// exit if not found a minimum number of repetitions
224-
if end.wrapping_sub(start) < minRatio {
224+
if end.wrapping_sub(start) < minRatio as u32 {
225225
for idx in start..end {
226226
doneMarks[suffix(idx as usize) as usize] = true;
227227
}
@@ -269,7 +269,7 @@ unsafe fn ZDICT_analyzePos(
269269
selectedCount = currentCount;
270270
selectedID = currentID;
271271
}
272-
if selectedCount < minRatio {
272+
if selectedCount < minRatio as u32 {
273273
break;
274274
}
275275
refinedStart = selectedID;
@@ -327,7 +327,7 @@ unsafe fn ZDICT_analyzePos(
327327
let mut u_0: core::ffi::c_uint = 0;
328328
u_0 = (LLIMIT - 1) as core::ffi::c_uint;
329329
while u_0 >= MINMATCHLENGTH as core::ffi::c_uint {
330-
if cumulLength[u_0 as usize] >= minRatio {
330+
if cumulLength[u_0 as usize] >= minRatio as u32 {
331331
break;
332332
}
333333
u_0 = u_0.wrapping_sub(1);
@@ -525,7 +525,7 @@ fn ZDICT_insertDictItem(table: &mut [DictItem], elt: DictItem, buffer: &[u8]) {
525525
table[0].pos = nextElt.wrapping_add(1);
526526
}
527527

528-
unsafe fn ZDICT_dictSize(dictList: &[DictItem]) -> u32 {
528+
fn ZDICT_dictSize(dictList: &[DictItem]) -> u32 {
529529
let mut u: u32 = 0;
530530
let mut dictSize = 0u32;
531531
u = 1;
@@ -540,9 +540,9 @@ unsafe fn ZDICT_trainBuffer_legacy(
540540
dictList: &mut [DictItem],
541541
buffer: &[u8],
542542
mut bufferSize: size_t,
543-
fileSizes: *const size_t,
544-
mut nbFiles: core::ffi::c_uint,
545-
mut minRatio: core::ffi::c_uint,
543+
fileSizes: &[size_t],
544+
mut nbFiles: usize,
545+
mut minRatio: usize,
546546
notificationLevel: u32,
547547
) -> size_t {
548548
let mut displayClock = Instant::now();
@@ -566,7 +566,7 @@ unsafe fn ZDICT_trainBuffer_legacy(
566566
}
567567
while bufferSize > ZDICT_MAX_SAMPLES_SIZE {
568568
nbFiles = nbFiles.wrapping_sub(1);
569-
bufferSize = bufferSize.wrapping_sub(*fileSizes.offset(nbFiles as isize));
569+
bufferSize = bufferSize.wrapping_sub(fileSizes[nbFiles]);
570570
}
571571

572572
// sort
@@ -599,7 +599,7 @@ unsafe fn ZDICT_trainBuffer_legacy(
599599
// filePos[0] is intentionally left 0
600600
for pos in 1..nbFiles as size_t {
601601
filePos[pos] =
602-
(filePos[pos - 1] as size_t).wrapping_add(*fileSizes.add(pos.wrapping_sub(1))) as u32;
602+
(filePos[pos - 1] as size_t).wrapping_add(fileSizes[pos.wrapping_sub(1)]) as u32;
603603
}
604604

605605
if notificationLevel >= 2 {
@@ -1285,8 +1285,8 @@ unsafe fn ZDICT_trainFromBuffer_unsafe_legacy(
12851285
samplesSizes: &[usize],
12861286
params: ZDICT_legacy_params_t,
12871287
) -> size_t {
1288-
let nbSamples = samplesSizes.len() as u32;
1289-
let dictListSize = Ord::max(Ord::max(10000, nbSamples), (maxDictSize / 16) as u32);
1288+
let nbSamples = samplesSizes.len();
1289+
let dictListSize = Ord::max(Ord::max(10000, nbSamples), maxDictSize / 16);
12901290
let mut dictList = vec![DictItem::default(); dictListSize as size_t];
12911291
let selectivity = if params.selectivityLevel == 0 {
12921292
g_selectivity_default
@@ -1322,7 +1322,7 @@ unsafe fn ZDICT_trainFromBuffer_unsafe_legacy(
13221322
&mut dictList,
13231323
samples,
13241324
samplesBuffSize,
1325-
samplesSizes.as_ptr(),
1325+
samplesSizes,
13261326
nbSamples,
13271327
minRep,
13281328
notificationLevel,
@@ -1443,8 +1443,6 @@ unsafe fn ZDICT_trainFromBuffer_unsafe_legacy(
14431443
);
14441444
}
14451445

1446-
1447-
14481446
ZDICT_addEntropyTablesFromBuffer_advanced(
14491447
dictBuffer,
14501448
dictContentSize_0 as size_t,

0 commit comments

Comments
 (0)