blob: 5d73c21ebdd4b01842c4d7a32aecaeca6fd53518 [file] [log] [blame]
khenaidoo7d3c5582021-08-11 18:09:44 -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
Akash Reddy Kankanalac28f0e22025-06-16 11:00:55 +053016// 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 {
khenaidoo7d3c5582021-08-11 18:09:44 -040021 switch mls {
Akash Reddy Kankanalac28f0e22025-06-16 11:00:55 +053022 case 3:
23 return (uint32(u<<8) * prime3bytes) >> (32 - length)
khenaidoo7d3c5582021-08-11 18:09:44 -040024 case 5:
Akash Reddy Kankanalac28f0e22025-06-16 11:00:55 +053025 return uint32(((u << (64 - 40)) * prime5bytes) >> (64 - length))
khenaidoo7d3c5582021-08-11 18:09:44 -040026 case 6:
Akash Reddy Kankanalac28f0e22025-06-16 11:00:55 +053027 return uint32(((u << (64 - 48)) * prime6bytes) >> (64 - length))
khenaidoo7d3c5582021-08-11 18:09:44 -040028 case 7:
Akash Reddy Kankanalac28f0e22025-06-16 11:00:55 +053029 return uint32(((u << (64 - 56)) * prime7bytes) >> (64 - length))
khenaidoo7d3c5582021-08-11 18:09:44 -040030 case 8:
Akash Reddy Kankanalac28f0e22025-06-16 11:00:55 +053031 return uint32((u * prime8bytes) >> (64 - length))
khenaidoo7d3c5582021-08-11 18:09:44 -040032 default:
Akash Reddy Kankanalac28f0e22025-06-16 11:00:55 +053033 return (uint32(u) * prime4bytes) >> (32 - length)
khenaidoo7d3c5582021-08-11 18:09:44 -040034 }
35}