Skip to content

Commit d6b12dc

Browse files
committed
fn COVER_selectDict: pass samplesBuffer and samplesSizes as slices
1 parent 8c9dd82 commit d6b12dc

File tree

2 files changed

+16
-16
lines changed

2 files changed

+16
-16
lines changed

lib/dictBuilder/cover.rs

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -882,8 +882,8 @@ pub unsafe extern "C" fn ZDICT_trainFromBuffer_cover(
882882

883883
pub(super) unsafe fn COVER_checkTotalCompressedSize(
884884
parameters: ZDICT_cover_params_t,
885-
samplesSizes: *const size_t,
886-
samples: *const u8,
885+
samplesSizes: &[size_t],
886+
samples: &[u8],
887887
offsets: *mut size_t,
888888
nbTrainSamples: size_t,
889889
nbSamples: size_t,
@@ -902,8 +902,8 @@ pub(super) unsafe fn COVER_checkTotalCompressedSize(
902902
0
903903
};
904904
while i < nbSamples {
905-
maxSampleSize = if *samplesSizes.add(i) > maxSampleSize {
906-
*samplesSizes.add(i)
905+
maxSampleSize = if samplesSizes[i] > maxSampleSize {
906+
samplesSizes[i]
907907
} else {
908908
maxSampleSize
909909
};
@@ -929,8 +929,8 @@ pub(super) unsafe fn COVER_checkTotalCompressedSize(
929929
cctx,
930930
dst.as_mut_ptr().cast::<core::ffi::c_void>(),
931931
dstCapacity,
932-
samples.add(*offsets.add(i)) as *const core::ffi::c_void,
933-
*samplesSizes.add(i),
932+
samples[*offsets.add(i)..].as_ptr() as *const core::ffi::c_void,
933+
samplesSizes[i],
934934
cdict,
935935
);
936936
if ERR_isError(size) {
@@ -1026,8 +1026,8 @@ pub(super) unsafe fn COVER_selectDict(
10261026
customDictContent: &[u8],
10271027
dictBufferCapacity: size_t,
10281028
mut dictContentSize: size_t,
1029-
samplesBuffer: *const u8,
1030-
samplesSizes: *const size_t,
1029+
samplesBuffer: &[u8],
1030+
samplesSizes: &[size_t],
10311031
nbFinalizeSamples: core::ffi::c_uint,
10321032
nbCheckSamples: size_t,
10331033
nbSamples: size_t,
@@ -1047,8 +1047,8 @@ pub(super) unsafe fn COVER_selectDict(
10471047
dictBufferCapacity,
10481048
customDictContent.as_ptr() as *const core::ffi::c_void,
10491049
dictContentSize,
1050-
samplesBuffer as *const core::ffi::c_void,
1051-
samplesSizes,
1050+
samplesBuffer.as_ptr() as *const core::ffi::c_void,
1051+
samplesSizes.as_ptr(),
10521052
nbFinalizeSamples,
10531053
params.zParams,
10541054
);
@@ -1088,8 +1088,8 @@ pub(super) unsafe fn COVER_selectDict(
10881088
.as_ptr()
10891089
.cast(),
10901090
dictContentSize,
1091-
samplesBuffer as *const core::ffi::c_void,
1092-
samplesSizes,
1091+
samplesBuffer.as_ptr() as *const core::ffi::c_void,
1092+
samplesSizes.as_ptr(),
10931093
nbFinalizeSamples,
10941094
params.zParams,
10951095
);
@@ -1152,8 +1152,8 @@ unsafe fn COVER_tryParameters(opaque: *mut core::ffi::c_void) {
11521152
&dict[tail..],
11531153
dictBufferCapacity,
11541154
dictBufferCapacity.wrapping_sub(tail),
1155-
(*ctx).samples.as_ptr(),
1156-
(*ctx).samplesSizes.as_ptr(),
1155+
(*ctx).samples,
1156+
(*ctx).samplesSizes,
11571157
(*ctx).nbTrainSamples as core::ffi::c_uint,
11581158
(*ctx).nbTrainSamples,
11591159
(*ctx).nbSamples,

lib/dictBuilder/fastcover.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -456,8 +456,8 @@ unsafe fn FASTCOVER_tryParameters(opaque: *mut core::ffi::c_void) {
456456
&dict[tail..],
457457
dictBufferCapacity,
458458
dictBufferCapacity.wrapping_sub(tail),
459-
ctx.samples.as_ptr(),
460-
ctx.samplesSizes.as_ptr(),
459+
ctx.samples,
460+
ctx.samplesSizes,
461461
nbFinalizeSamples,
462462
ctx.nbTrainSamples,
463463
ctx.nbSamples,

0 commit comments

Comments
 (0)