Update utilities.h

Fix and optimize mixDiscreteIdAndValue() and mixStringIdAndValue() functions
This commit is contained in:
Leon Banik 2023-04-01 17:57:31 +02:00 committed by GitHub
parent ec83d01dca
commit ca2cc4c12f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 17 additions and 14 deletions

View File

@ -1,18 +1,21 @@
#pragma once #pragma once
#ifdef __cplusplus
#include <cstdint>
#include <cstring>
namespace twml { namespace twml {
inline int64_t mixDiscreteIdAndValue(int64_t key, int64_t value) { inline int64_t mixDiscreteIdAndValue(int64_t key, int64_t value) {
key ^= ((17LL + value) * 2654435761LL); key ^= ((17LL + value) * 2654435761LL);
return key; return key;
} }
inline int64_t mixStringIdAndValue(int64_t key, int32_t str_len, const uint8_t *str) { inline int64_t mixStringIdAndValue(int64_t key, int32_t str_len, const uint8_t* str) {
int32_t hash = 0; const uint8_t* end = str + str_len;
for (int32_t i = 0; i < str_len; i++) { while (str != end) {
hash = (31 * hash) + (int32_t)str[i]; key = (key * 31) + *str++;
} }
return key ^ hash; return key;
} }
}
#endif } // namespace twml