khenaidoo | 7d3c558 | 2021-08-11 18:09:44 -0400 | [diff] [blame] | 1 | // 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 | |
| 5 | package zstd |
| 6 | |
| 7 | const ( |
| 8 | prime3bytes = 506832829 |
| 9 | prime4bytes = 2654435761 |
| 10 | prime5bytes = 889523592379 |
| 11 | prime6bytes = 227718039650203 |
| 12 | prime7bytes = 58295818150454627 |
| 13 | prime8bytes = 0xcf1bbcdcb7a56463 |
| 14 | ) |
| 15 | |
Akash Reddy Kankanala | c28f0e2 | 2025-06-16 11:00:55 +0530 | [diff] [blame^] | 16 | // 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. |
| 20 | func hashLen(u uint64, length, mls uint8) uint32 { |
khenaidoo | 7d3c558 | 2021-08-11 18:09:44 -0400 | [diff] [blame] | 21 | switch mls { |
Akash Reddy Kankanala | c28f0e2 | 2025-06-16 11:00:55 +0530 | [diff] [blame^] | 22 | case 3: |
| 23 | return (uint32(u<<8) * prime3bytes) >> (32 - length) |
khenaidoo | 7d3c558 | 2021-08-11 18:09:44 -0400 | [diff] [blame] | 24 | case 5: |
Akash Reddy Kankanala | c28f0e2 | 2025-06-16 11:00:55 +0530 | [diff] [blame^] | 25 | return uint32(((u << (64 - 40)) * prime5bytes) >> (64 - length)) |
khenaidoo | 7d3c558 | 2021-08-11 18:09:44 -0400 | [diff] [blame] | 26 | case 6: |
Akash Reddy Kankanala | c28f0e2 | 2025-06-16 11:00:55 +0530 | [diff] [blame^] | 27 | return uint32(((u << (64 - 48)) * prime6bytes) >> (64 - length)) |
khenaidoo | 7d3c558 | 2021-08-11 18:09:44 -0400 | [diff] [blame] | 28 | case 7: |
Akash Reddy Kankanala | c28f0e2 | 2025-06-16 11:00:55 +0530 | [diff] [blame^] | 29 | return uint32(((u << (64 - 56)) * prime7bytes) >> (64 - length)) |
khenaidoo | 7d3c558 | 2021-08-11 18:09:44 -0400 | [diff] [blame] | 30 | case 8: |
Akash Reddy Kankanala | c28f0e2 | 2025-06-16 11:00:55 +0530 | [diff] [blame^] | 31 | return uint32((u * prime8bytes) >> (64 - length)) |
khenaidoo | 7d3c558 | 2021-08-11 18:09:44 -0400 | [diff] [blame] | 32 | default: |
Akash Reddy Kankanala | c28f0e2 | 2025-06-16 11:00:55 +0530 | [diff] [blame^] | 33 | return (uint32(u) * prime4bytes) >> (32 - length) |
khenaidoo | 7d3c558 | 2021-08-11 18:09:44 -0400 | [diff] [blame] | 34 | } |
| 35 | } |