blob: 5d73c21ebdd4b01842c4d7a32aecaeca6fd53518 [file] [log] [blame]
khenaidoo26721882021-08-11 17:42:52 -04001// Copyright 2019+ Klaus Post. All rights reserved.
2// License information can be found in the LICENSE file.
3// Based on work by Yann Collet, released under BSD License.
4
5package zstd
6
7const (
8 prime3bytes = 506832829
9 prime4bytes = 2654435761
10 prime5bytes = 889523592379
11 prime6bytes = 227718039650203
12 prime7bytes = 58295818150454627
13 prime8bytes = 0xcf1bbcdcb7a56463
14)
15
Abhay Kumar40252eb2025-10-13 13:25:53 +000016// hashLen returns a hash of the lowest mls bytes of with length output bits.
17// mls must be >=3 and <=8. Any other value will return hash for 4 bytes.
18// length should always be < 32.
19// Preferably length and mls should be a constant for inlining.
20func hashLen(u uint64, length, mls uint8) uint32 {
khenaidoo26721882021-08-11 17:42:52 -040021 switch mls {
Abhay Kumar40252eb2025-10-13 13:25:53 +000022 case 3:
23 return (uint32(u<<8) * prime3bytes) >> (32 - length)
khenaidoo26721882021-08-11 17:42:52 -040024 case 5:
Abhay Kumar40252eb2025-10-13 13:25:53 +000025 return uint32(((u << (64 - 40)) * prime5bytes) >> (64 - length))
khenaidoo26721882021-08-11 17:42:52 -040026 case 6:
Abhay Kumar40252eb2025-10-13 13:25:53 +000027 return uint32(((u << (64 - 48)) * prime6bytes) >> (64 - length))
khenaidoo26721882021-08-11 17:42:52 -040028 case 7:
Abhay Kumar40252eb2025-10-13 13:25:53 +000029 return uint32(((u << (64 - 56)) * prime7bytes) >> (64 - length))
khenaidoo26721882021-08-11 17:42:52 -040030 case 8:
Abhay Kumar40252eb2025-10-13 13:25:53 +000031 return uint32((u * prime8bytes) >> (64 - length))
khenaidoo26721882021-08-11 17:42:52 -040032 default:
Abhay Kumar40252eb2025-10-13 13:25:53 +000033 return (uint32(u) * prime4bytes) >> (32 - length)
khenaidoo26721882021-08-11 17:42:52 -040034 }
35}