AK: Add ptr_hash to use int_hash or u64_hash depending on pointer size

akimdi · Feb 25, 2020 · 7e6ac54 · 7e6ac54
1 parent 32e6453
commit 7e6ac54
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/AK/HashFunctions.h b/AK/HashFunctions.h
@@ -50,3 +50,11 @@ inline unsigned u64_hash(u64 key)
  u32 last = key >> 32;
  return pair_int_hash(first, last);
 }
+
+inline unsigned ptr_hash(uintptr_t ptr)
+{
+ if constexpr(sizeof(ptr) == 8)
+ return u64_hash((u64)ptr);
+ else
+ return int_hash((u32)ptr);
+}
diff --git a/AK/OwnPtr.h b/AK/OwnPtr.h
@@ -204,7 +204,7 @@ inline void swap(OwnPtr<T>& a, OwnPtr<U>& b)
 template<typename T>
 struct Traits<OwnPtr<T>> : public GenericTraits<OwnPtr<T>> {
  using PeekType = const T*;
- static unsigned hash(const OwnPtr<T>& p) { return int_hash((u32)p.ptr()); }
+ static unsigned hash(const OwnPtr<T>& p) { return ptr_hash(p.ptr()); }
  static bool equals(const OwnPtr<T>& a, const OwnPtr<T>& b) { return a.ptr() == b.ptr(); }
 };
 

diff --git a/AK/RefPtr.h b/AK/RefPtr.h
@@ -278,7 +278,7 @@ inline const LogStream& operator<<(const LogStream& stream, const RefPtr<T>& val
 template<typename T>
 struct Traits<RefPtr<T>> : public GenericTraits<RefPtr<T>> {
  using PeekType = const T*;
- static unsigned hash(const RefPtr<T>& p) { return int_hash((u32)p.ptr()); }
+ static unsigned hash(const RefPtr<T>& p) { return ptr_hash(p.ptr()); }
  static bool equals(const RefPtr<T>& a, const RefPtr<T>& b) { return a.ptr() == b.ptr(); }
 };