blob: 5d73c21ebdd4b01842c4d7a32aecaeca6fd53518 [file] [log] [blame]
Scott Bakered4efab2020-01-13 19:12:25 -08001// 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 Kankanalac0014632025-05-21 17:12:20 +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 {
Scott Bakered4efab2020-01-13 19:12:25 -080021 switch mls {
Akash Reddy Kankanalac0014632025-05-21 17:12:20 +053022 case 3:
23 return (uint32(u<<8) * prime3bytes) >> (32 - length)
Scott Bakered4efab2020-01-13 19:12:25 -080024 case 5:
Akash Reddy Kankanalac0014632025-05-21 17:12:20 +053025 return uint32(((u << (64 - 40)) * prime5bytes) >> (64 - length))
Scott Bakered4efab2020-01-13 19:12:25 -080026 case 6:
Akash Reddy Kankanalac0014632025-05-21 17:12:20 +053027 return uint32(((u << (64 - 48)) * prime6bytes) >> (64 - length))
Scott Bakered4efab2020-01-13 19:12:25 -080028 case 7:
Akash Reddy Kankanalac0014632025-05-21 17:12:20 +053029 return uint32(((u << (64 - 56)) * prime7bytes) >> (64 - length))
Scott Bakered4efab2020-01-13 19:12:25 -080030 case 8:
Akash Reddy Kankanalac0014632025-05-21 17:12:20 +053031 return uint32((u * prime8bytes) >> (64 - length))
Scott Bakered4efab2020-01-13 19:12:25 -080032 default:
Akash Reddy Kankanalac0014632025-05-21 17:12:20 +053033 return (uint32(u) * prime4bytes) >> (32 - length)
Scott Bakered4efab2020-01-13 19:12:25 -080034 }
35}