diff --git a/llvm/include/llvm/Support/HashBuilder.h b/llvm/include/llvm/Support/HashBuilder.h --- a/llvm/include/llvm/Support/HashBuilder.h +++ b/llvm/include/llvm/Support/HashBuilder.h @@ -16,6 +16,7 @@ #define LLVM_SUPPORT_HASHBUILDER_H #include "llvm/ADT/ArrayRef.h" +#include "llvm/ADT/Hashing.h" #include "llvm/ADT/STLExtras.h" #include "llvm/ADT/StringRef.h" #include "llvm/Support/Endian.h" @@ -405,6 +406,33 @@ HashBuilderImpl; + +namespace hashbuilder_detail { +class HashCodeHasher { +public: + HashCodeHasher() : Code(0) {} + void update(ArrayRef Data) { + hash_code DataCode = hash_value(Data); + Code = hash_combine(Code, DataCode); + } + hash_code Code; +}; + +using HashCodeHashBuilder = HashBuilder; +} // namespace hashbuilder_detail + +/// Provide a default implementation of `hash_value` when `addHash(const T &)` +/// is supported. +template +std::enable_if_t< + is_detected::value, + hash_code> +hash_value(const T &Value) { + hashbuilder_detail::HashCodeHashBuilder HBuilder; + HBuilder.add(Value); + return HBuilder.getHasher().Code; +} } // end namespace llvm #endif // LLVM_SUPPORT_HASHBUILDER_H diff --git a/llvm/unittests/ADT/HashingTest.cpp b/llvm/unittests/ADT/HashingTest.cpp --- a/llvm/unittests/ADT/HashingTest.cpp +++ b/llvm/unittests/ADT/HashingTest.cpp @@ -12,6 +12,7 @@ #include "llvm/ADT/Hashing.h" #include "llvm/Support/DataTypes.h" +#include "llvm/Support/HashBuilder.h" #include "gtest/gtest.h" #include #include @@ -402,4 +403,36 @@ #undef CHECK_SAME } +struct StructWithHashBuilderSupport { + char C; + int I; + template + friend void addHash(llvm::HashBuilderImpl &HBuilder, + const StructWithHashBuilderSupport &Value) { + HBuilder.add(Value.C, Value.I); + } +}; + +TEST(HashingTest, HashWithHashBuilder) { + StructWithHashBuilderSupport S{'c', 1}; + EXPECT_NE(static_cast(llvm::hash_value(S)), static_cast(0)); } + +struct StructWithHashBuilderAndHashValueSupport { + char C; + int I; + template + friend void addHash(llvm::HashBuilderImpl &HBuilder, + const StructWithHashBuilderAndHashValueSupport &Value) {} + friend hash_code + hash_value(const StructWithHashBuilderAndHashValueSupport &Value) { + return 0xbeef; + } +}; + +TEST(HashingTest, HashBuilderAndHashValue) { + StructWithHashBuilderAndHashValueSupport S{'c', 1}; + EXPECT_EQ(static_cast(hash_value(S)), static_cast(0xbeef)); +} + +} // namespace