|
15 | 15 | // specific language governing permissions and limitations |
16 | 16 | // under the License. |
17 | 17 |
|
| 18 | +use std::mem; |
| 19 | + |
18 | 20 | #[cfg(target_feature = "neon")] |
19 | 21 | use std::arch::aarch64::*; |
20 | 22 |
|
@@ -187,3 +189,198 @@ mod tests { |
187 | 189 | assert!(!is_latin_standard(¬_latin_str)); |
188 | 190 | } |
189 | 191 | } |
| 192 | + |
| 193 | +fn fmix64(mut k: u64) -> u64 { |
| 194 | + k ^= k >> 33; |
| 195 | + k = k.wrapping_mul(0xff51afd7ed558ccdu64); |
| 196 | + k ^= k >> 33; |
| 197 | + k = k.wrapping_mul(0xc4ceb9fe1a85ec53u64); |
| 198 | + k ^= k >> 33; |
| 199 | + |
| 200 | + k |
| 201 | +} |
| 202 | + |
| 203 | +pub fn murmurhash3_x64_128(bytes: &[u8], seed: u64) -> (u64, u64) { |
| 204 | + let c1 = 0x87c37b91114253d5u64; |
| 205 | + let c2 = 0x4cf5ad432745937fu64; |
| 206 | + let read_size = 16; |
| 207 | + let len = bytes.len() as u64; |
| 208 | + let block_count = len / read_size; |
| 209 | + |
| 210 | + let (mut h1, mut h2) = (seed, seed); |
| 211 | + |
| 212 | + for i in 0..block_count as usize { |
| 213 | + let b64: &[u64] = unsafe { mem::transmute(bytes) }; |
| 214 | + let (mut k1, mut k2) = (b64[i * 2], b64[i * 2 + 1]); |
| 215 | + |
| 216 | + k1 = k1.wrapping_mul(c1); |
| 217 | + k1 = k1.rotate_left(31); |
| 218 | + k1 = k1.wrapping_mul(c2); |
| 219 | + h1 ^= k1; |
| 220 | + |
| 221 | + h1 = h1.rotate_left(27); |
| 222 | + h1 = h1.wrapping_add(h2); |
| 223 | + h1 = h1.wrapping_mul(5); |
| 224 | + h1 = h1.wrapping_add(0x52dce729); |
| 225 | + |
| 226 | + k2 = k2.wrapping_mul(c2); |
| 227 | + k2 = k2.rotate_left(33); |
| 228 | + k2 = k2.wrapping_mul(c1); |
| 229 | + h2 ^= k2; |
| 230 | + |
| 231 | + h2 = h2.rotate_left(31); |
| 232 | + h2 = h2.wrapping_add(h1); |
| 233 | + h2 = h2.wrapping_mul(5); |
| 234 | + h2 = h2.wrapping_add(0x38495ab5); |
| 235 | + } |
| 236 | + let (mut k1, mut k2) = (0u64, 0u64); |
| 237 | + |
| 238 | + if len & 15 == 15 { |
| 239 | + k2 ^= (bytes[(block_count * read_size) as usize + 14] as u64) << 48; |
| 240 | + } |
| 241 | + if len & 15 >= 14 { |
| 242 | + k2 ^= (bytes[(block_count * read_size) as usize + 13] as u64) << 40; |
| 243 | + } |
| 244 | + if len & 15 >= 13 { |
| 245 | + k2 ^= (bytes[(block_count * read_size) as usize + 12] as u64) << 32; |
| 246 | + } |
| 247 | + if len & 15 >= 12 { |
| 248 | + k2 ^= (bytes[(block_count * read_size) as usize + 11] as u64) << 24; |
| 249 | + } |
| 250 | + if len & 15 >= 11 { |
| 251 | + k2 ^= (bytes[(block_count * read_size) as usize + 10] as u64) << 16; |
| 252 | + } |
| 253 | + if len & 15 >= 10 { |
| 254 | + k2 ^= (bytes[(block_count * read_size) as usize + 9] as u64) << 8; |
| 255 | + } |
| 256 | + if len & 15 >= 9 { |
| 257 | + k2 ^= bytes[(block_count * read_size) as usize + 8] as u64; |
| 258 | + k2 = k2.wrapping_mul(c2); |
| 259 | + k2 = k2.rotate_left(33); |
| 260 | + k2 = k2.wrapping_mul(c1); |
| 261 | + h2 ^= k2; |
| 262 | + } |
| 263 | + |
| 264 | + if len & 15 >= 8 { |
| 265 | + k1 ^= (bytes[(block_count * read_size) as usize + 7] as u64) << 56; |
| 266 | + } |
| 267 | + if len & 15 >= 7 { |
| 268 | + k1 ^= (bytes[(block_count * read_size) as usize + 6] as u64) << 48; |
| 269 | + } |
| 270 | + if len & 15 >= 6 { |
| 271 | + k1 ^= (bytes[(block_count * read_size) as usize + 5] as u64) << 40; |
| 272 | + } |
| 273 | + if len & 15 >= 5 { |
| 274 | + k1 ^= (bytes[(block_count * read_size) as usize + 4] as u64) << 32; |
| 275 | + } |
| 276 | + if len & 15 >= 4 { |
| 277 | + k1 ^= (bytes[(block_count * read_size) as usize + 3] as u64) << 24; |
| 278 | + } |
| 279 | + if len & 15 >= 3 { |
| 280 | + k1 ^= (bytes[(block_count * read_size) as usize + 2] as u64) << 16; |
| 281 | + } |
| 282 | + if len & 15 >= 2 { |
| 283 | + k1 ^= (bytes[(block_count * read_size) as usize + 1] as u64) << 8; |
| 284 | + } |
| 285 | + if len & 15 >= 1 { |
| 286 | + k1 ^= bytes[(block_count * read_size) as usize] as u64; |
| 287 | + k1 = k1.wrapping_mul(c1); |
| 288 | + k1 = k1.rotate_left(31); |
| 289 | + k1 = k1.wrapping_mul(c2); |
| 290 | + h1 ^= k1; |
| 291 | + } |
| 292 | + |
| 293 | + h1 ^= bytes.len() as u64; |
| 294 | + h2 ^= bytes.len() as u64; |
| 295 | + |
| 296 | + h1 = h1.wrapping_add(h2); |
| 297 | + h2 = h2.wrapping_add(h1); |
| 298 | + |
| 299 | + h1 = fmix64(h1); |
| 300 | + h2 = fmix64(h2); |
| 301 | + |
| 302 | + h1 = h1.wrapping_add(h2); |
| 303 | + h2 = h2.wrapping_add(h1); |
| 304 | + |
| 305 | + (h1, h2) |
| 306 | +} |
| 307 | + |
| 308 | +#[cfg(test)] |
| 309 | +mod test_hash { |
| 310 | + use super::murmurhash3_x64_128; |
| 311 | + |
| 312 | + #[test] |
| 313 | + fn test_empty_string() { |
| 314 | + assert!(murmurhash3_x64_128("".as_bytes(), 0) == (0, 0)); |
| 315 | + } |
| 316 | + |
| 317 | + #[test] |
| 318 | + fn test_tail_lengths() { |
| 319 | + assert!( |
| 320 | + murmurhash3_x64_128("1".as_bytes(), 0) == (8213365047359667313, 10676604921780958775) |
| 321 | + ); |
| 322 | + assert!( |
| 323 | + murmurhash3_x64_128("12".as_bytes(), 0) == (5355690773644049813, 9855895140584599837) |
| 324 | + ); |
| 325 | + assert!( |
| 326 | + murmurhash3_x64_128("123".as_bytes(), 0) == (10978418110857903978, 4791445053355511657) |
| 327 | + ); |
| 328 | + assert!( |
| 329 | + murmurhash3_x64_128("1234".as_bytes(), 0) == (619023178690193332, 3755592904005385637) |
| 330 | + ); |
| 331 | + assert!( |
| 332 | + murmurhash3_x64_128("12345".as_bytes(), 0) |
| 333 | + == (2375712675693977547, 17382870096830835188) |
| 334 | + ); |
| 335 | + assert!( |
| 336 | + murmurhash3_x64_128("123456".as_bytes(), 0) |
| 337 | + == (16435832985690558678, 5882968373513761278) |
| 338 | + ); |
| 339 | + assert!( |
| 340 | + murmurhash3_x64_128("1234567".as_bytes(), 0) |
| 341 | + == (3232113351312417698, 4025181827808483669) |
| 342 | + ); |
| 343 | + assert!( |
| 344 | + murmurhash3_x64_128("12345678".as_bytes(), 0) |
| 345 | + == (4272337174398058908, 10464973996478965079) |
| 346 | + ); |
| 347 | + assert!( |
| 348 | + murmurhash3_x64_128("123456789".as_bytes(), 0) |
| 349 | + == (4360720697772133540, 11094893415607738629) |
| 350 | + ); |
| 351 | + assert!( |
| 352 | + murmurhash3_x64_128("123456789a".as_bytes(), 0) |
| 353 | + == (12594836289594257748, 2662019112679848245) |
| 354 | + ); |
| 355 | + assert!( |
| 356 | + murmurhash3_x64_128("123456789ab".as_bytes(), 0) |
| 357 | + == (6978636991469537545, 12243090730442643750) |
| 358 | + ); |
| 359 | + assert!( |
| 360 | + murmurhash3_x64_128("123456789abc".as_bytes(), 0) |
| 361 | + == (211890993682310078, 16480638721813329343) |
| 362 | + ); |
| 363 | + assert!( |
| 364 | + murmurhash3_x64_128("123456789abcd".as_bytes(), 0) |
| 365 | + == (12459781455342427559, 3193214493011213179) |
| 366 | + ); |
| 367 | + assert!( |
| 368 | + murmurhash3_x64_128("123456789abcde".as_bytes(), 0) |
| 369 | + == (12538342858731408721, 9820739847336455216) |
| 370 | + ); |
| 371 | + assert!( |
| 372 | + murmurhash3_x64_128("123456789abcdef".as_bytes(), 0) |
| 373 | + == (9165946068217512774, 2451472574052603025) |
| 374 | + ); |
| 375 | + assert!( |
| 376 | + murmurhash3_x64_128("123456789abcdef1".as_bytes(), 0) |
| 377 | + == (9259082041050667785, 12459473952842597282) |
| 378 | + ); |
| 379 | + } |
| 380 | + |
| 381 | + #[test] |
| 382 | + fn test_large_data() { |
| 383 | + assert!(murmurhash3_x64_128("Lorem ipsum dolor sit amet, consectetur adipiscing elit. Etiam at consequat massa. Cras eleifend pellentesque ex, at dignissim libero maximus ut. Sed eget nulla felis".as_bytes(), 0) |
| 384 | + == (9455322759164802692, 17863277201603478371)); |
| 385 | + } |
| 386 | +} |
0 commit comments