Skip to content

Commit de835ab

Browse files
michielp1807folkertdev
authored andcommitted
huf.rs: use more sensible constant types
1 parent 963fcf7 commit de835ab

File tree

5 files changed

+22
-23
lines changed

5 files changed

+22
-23
lines changed

lib/common/huf.rs

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -3,15 +3,15 @@ use libc::size_t;
33
use crate::lib::common::fse::FSE_DECOMPRESS_WKSP_SIZE_U32;
44
use crate::lib::zstd::ZSTD_btultra;
55

6-
pub(crate) const HUF_BLOCKSIZE_MAX: core::ffi::c_int = 128 * 1024;
6+
pub(crate) const HUF_BLOCKSIZE_MAX: usize = 128 * 1024;
77

8-
pub(crate) const HUF_WORKSPACE_SIZE: core::ffi::c_int = ((8) << 10) + 512;
8+
pub(crate) const HUF_WORKSPACE_SIZE: usize = (8 << 10) + 512;
99

1010
/// Max runtime value of tableLog (due to static allocation); can be modified up to HUF_TABLELOG_ABSOLUTEMAX.
1111
pub(crate) const HUF_TABLELOG_MAX: usize = 12;
1212
/// Default tableLog value when none specified
13-
pub(crate) const HUF_TABLELOG_DEFAULT: core::ffi::c_int = 11;
14-
pub(crate) const HUF_SYMBOLVALUE_MAX: core::ffi::c_int = 255;
13+
pub(crate) const HUF_TABLELOG_DEFAULT: u32 = 11;
14+
pub(crate) const HUF_SYMBOLVALUE_MAX: u32 = 255;
1515

1616
pub(crate) const HUF_CTABLE_WORKSPACE_SIZE_U32: usize =
1717
(4 * (HUF_SYMBOLVALUE_MAX as usize + 1)) + 192;
@@ -23,7 +23,7 @@ const _: () = assert!(
2323
"HUF_TABLELOG_MAX is too large !"
2424
);
2525

26-
pub(crate) const HUF_CTABLEBOUND: core::ffi::c_int = 129;
26+
pub(crate) const HUF_CTABLEBOUND: usize = 129;
2727

2828
pub(crate) type HUF_CElt = size_t;
2929

lib/compress/huf_compress.rs

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -261,7 +261,7 @@ pub unsafe fn HUF_writeCTable_wksp(
261261
if workspaceSize < ::core::mem::size_of::<HUF_WriteCTableWksp>() {
262262
return Error::GENERIC.to_error_code();
263263
}
264-
if maxSymbolValue > HUF_SYMBOLVALUE_MAX as core::ffi::c_uint {
264+
if maxSymbolValue > HUF_SYMBOLVALUE_MAX {
265265
return Error::maxSymbolValue_tooLarge.to_error_code();
266266
}
267267
*((*wksp).bitsToWeight).as_mut_ptr() = 0;
@@ -672,7 +672,7 @@ unsafe fn HUF_sort(
672672
n = n.wrapping_add(1);
673673
}
674674
}
675-
pub const STARTNODE: core::ffi::c_int = HUF_SYMBOLVALUE_MAX + 1;
675+
pub const STARTNODE: core::ffi::c_int = HUF_SYMBOLVALUE_MAX as i32 + 1;
676676
unsafe fn HUF_buildTree(huffNode: *mut nodeElt, maxSymbolValue: u32) -> core::ffi::c_int {
677677
let huffNode0 = huffNode.sub(1);
678678
let mut nonNullRank: core::ffi::c_int = 0;
@@ -817,9 +817,9 @@ pub unsafe fn HUF_buildCTable_wksp(
817817
return Error::workSpace_tooSmall.to_error_code();
818818
}
819819
if maxNbBits == 0 {
820-
maxNbBits = HUF_TABLELOG_DEFAULT as u32;
820+
maxNbBits = HUF_TABLELOG_DEFAULT;
821821
}
822-
if maxSymbolValue > HUF_SYMBOLVALUE_MAX as u32 {
822+
if maxSymbolValue > HUF_SYMBOLVALUE_MAX {
823823
return Error::maxSymbolValue_tooLarge.to_error_code();
824824
}
825825
ptr::write_bytes(
@@ -879,7 +879,7 @@ pub unsafe fn HUF_validateCTable(
879879
(bad == 0) as core::ffi::c_int
880880
}
881881
pub fn HUF_compressBound(size: size_t) -> size_t {
882-
(HUF_CTABLEBOUND as size_t).wrapping_add(size.wrapping_add(size >> 8).wrapping_add(8))
882+
HUF_CTABLEBOUND.wrapping_add(size.wrapping_add(size >> 8).wrapping_add(8))
883883
}
884884
pub const HUF_BITS_IN_CONTAINER: size_t = (::core::mem::size_of::<size_t>()).wrapping_mul(8);
885885
unsafe fn HUF_initCStream(
@@ -1418,20 +1418,20 @@ unsafe fn HUF_compress_internal(
14181418
if dstSize == 0 {
14191419
return 0;
14201420
}
1421-
if srcSize > HUF_BLOCKSIZE_MAX as size_t {
1421+
if srcSize > HUF_BLOCKSIZE_MAX {
14221422
return Error::srcSize_wrong.to_error_code();
14231423
}
14241424
if huffLog > HUF_TABLELOG_MAX as core::ffi::c_uint {
14251425
return Error::tableLog_tooLarge.to_error_code();
14261426
}
1427-
if maxSymbolValue > HUF_SYMBOLVALUE_MAX as core::ffi::c_uint {
1427+
if maxSymbolValue > HUF_SYMBOLVALUE_MAX {
14281428
return Error::maxSymbolValue_tooLarge.to_error_code();
14291429
}
14301430
if maxSymbolValue == 0 {
1431-
maxSymbolValue = HUF_SYMBOLVALUE_MAX as core::ffi::c_uint;
1431+
maxSymbolValue = HUF_SYMBOLVALUE_MAX;
14321432
}
14331433
if huffLog == 0 {
1434-
huffLog = HUF_TABLELOG_DEFAULT as core::ffi::c_uint;
1434+
huffLog = HUF_TABLELOG_DEFAULT;
14351435
}
14361436
if flags & HUF_flags_preferRepeat as core::ffi::c_int != 0
14371437
&& !repeat.is_null()

lib/compress/zstd_compress.rs

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -5658,7 +5658,7 @@ unsafe fn ZSTD_buildBlockEntropyStats_literals(
56585658
.wrapping_mul(::core::mem::size_of::<core::ffi::c_uint>());
56595659
let nodeWksp = countWkspStart.add(countWkspSize);
56605660
let nodeWkspSize = wkspEnd.offset_from_unsigned(nodeWksp);
5661-
let mut maxSymbolValue = HUF_SYMBOLVALUE_MAX as core::ffi::c_uint;
5661+
let mut maxSymbolValue = HUF_SYMBOLVALUE_MAX;
56625662
let mut huffLog = LitHufLog;
56635663
let mut repeat = (*prevHuf).repeatMode;
56645664
libc::memcpy(
@@ -5904,7 +5904,7 @@ unsafe fn ZSTD_estimateBlockSize_literal(
59045904
writeEntropy: core::ffi::c_int,
59055905
) -> size_t {
59065906
let countWksp = workspace as *mut core::ffi::c_uint;
5907-
let mut maxSymbolValue = HUF_SYMBOLVALUE_MAX as core::ffi::c_uint;
5907+
let mut maxSymbolValue = HUF_SYMBOLVALUE_MAX;
59085908
let literalSectionHeaderSize =
59095909
(3 + (litSize >= ((1) << 10) as size_t) as core::ffi::c_int
59105910
+ (litSize >= (16 * ((1) << 10)) as size_t) as core::ffi::c_int) as size_t;
@@ -8494,7 +8494,7 @@ pub unsafe extern "C" fn ZSTD_estimateCDictSize_advanced(
84948494
dictLoadMethod: ZSTD_dictLoadMethod_e,
84958495
) -> size_t {
84968496
(ZSTD_cwksp_alloc_size(::core::mem::size_of::<ZSTD_CDict>()))
8497-
.wrapping_add(ZSTD_cwksp_alloc_size(HUF_WORKSPACE_SIZE as size_t))
8497+
.wrapping_add(ZSTD_cwksp_alloc_size(HUF_WORKSPACE_SIZE))
84988498
.wrapping_add(ZSTD_sizeof_matchState(
84998499
&cParams,
85008500
ZSTD_resolveRowMatchFinderMode(ZSTD_ParamSwitch_e::ZSTD_ps_auto, &cParams),
@@ -8569,8 +8569,7 @@ unsafe fn ZSTD_initCDict_internal(
85698569
(*cdict).dictContentSize = dictSize;
85708570
(*cdict).dictContentType = dictContentType;
85718571
(*cdict).entropyWorkspace =
8572-
ZSTD_cwksp_reserve_object(&mut (*cdict).workspace, HUF_WORKSPACE_SIZE as size_t)
8573-
as *mut u32;
8572+
ZSTD_cwksp_reserve_object(&mut (*cdict).workspace, HUF_WORKSPACE_SIZE) as *mut u32;
85748573
ZSTD_reset_compressedBlockState(&mut (*cdict).cBlockState);
85758574
let err_code = ZSTD_reset_matchState(
85768575
&mut (*cdict).matchState,
@@ -8615,7 +8614,7 @@ unsafe fn ZSTD_createCDict_advanced_internal(
86158614
customMem: ZSTD_customMem,
86168615
) -> *mut ZSTD_CDict {
86178616
let workspaceSize = (ZSTD_cwksp_alloc_size(::core::mem::size_of::<ZSTD_CDict>()))
8618-
.wrapping_add(ZSTD_cwksp_alloc_size(HUF_WORKSPACE_SIZE as size_t))
8617+
.wrapping_add(ZSTD_cwksp_alloc_size(HUF_WORKSPACE_SIZE))
86198618
.wrapping_add(ZSTD_sizeof_matchState(
86208619
&cParams,
86218620
useRowMatchFinder,
@@ -8896,7 +8895,7 @@ pub unsafe extern "C" fn ZSTD_initStaticCDict(
88968895
::core::mem::size_of::<*mut core::ffi::c_void>(),
88978896
))
88988897
})
8899-
.wrapping_add(ZSTD_cwksp_alloc_size(HUF_WORKSPACE_SIZE as size_t))
8898+
.wrapping_add(ZSTD_cwksp_alloc_size(HUF_WORKSPACE_SIZE))
89008899
.wrapping_add(matchStateSize);
89018900
let mut cdict = core::ptr::null_mut::<ZSTD_CDict>();
89028901
let mut params = ZSTD_CCtx_params_s {

lib/compress/zstd_compress_literals.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -250,7 +250,7 @@ pub unsafe fn ZSTD_compressLiterals(
250250
dstCapacity.wrapping_sub(lhSize),
251251
src,
252252
srcSize,
253-
HUF_SYMBOLVALUE_MAX as core::ffi::c_uint,
253+
HUF_SYMBOLVALUE_MAX,
254254
LitHufLog,
255255
entropyWorkspace,
256256
entropyWorkspaceSize,

lib/dictBuilder/zdict.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,7 @@ pub unsafe extern "C" fn ZDICT_getDictHeaderSize(
115115
let mut bs = Box::<ZSTD_compressedBlockState_t>::new_uninit();
116116
unsafe { ZSTD_reset_compressedBlockState(bs.as_mut_ptr()) };
117117

118-
let mut wksp = Box::<[u32]>::new_uninit_slice(HUF_WORKSPACE_SIZE as size_t / 4);
118+
let mut wksp = Box::<[u32]>::new_uninit_slice(HUF_WORKSPACE_SIZE / 4);
119119

120120
ZSTD_loadCEntropy(
121121
bs.as_mut_ptr(),

0 commit comments

Comments
 (0)