Details | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line |
|---|---|---|---|
| 14 | pmbaty | 1 | //===- llvm/ADT/DenseMapInfo.h - Type traits for DenseMap -------*- C++ -*-===// |
| 2 | // |
||
| 3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
||
| 4 | // See https://llvm.org/LICENSE.txt for license information. |
||
| 5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
||
| 6 | // |
||
| 7 | //===----------------------------------------------------------------------===// |
||
| 8 | /// |
||
| 9 | /// \file |
||
| 10 | /// This file defines DenseMapInfo traits for DenseMap. |
||
| 11 | /// |
||
| 12 | //===----------------------------------------------------------------------===// |
||
| 13 | |||
| 14 | #ifndef LLVM_ADT_DENSEMAPINFO_H |
||
| 15 | #define LLVM_ADT_DENSEMAPINFO_H |
||
| 16 | |||
| 17 | #include <cassert> |
||
| 18 | #include <cstddef> |
||
| 19 | #include <cstdint> |
||
| 20 | #include <tuple> |
||
| 21 | #include <type_traits> |
||
| 22 | #include <utility> |
||
| 23 | #include <variant> |
||
| 24 | |||
| 25 | namespace llvm { |
||
| 26 | |||
| 27 | namespace detail { |
||
| 28 | |||
| 29 | /// Simplistic combination of 32-bit hash values into 32-bit hash values. |
||
| 30 | static inline unsigned combineHashValue(unsigned a, unsigned b) { |
||
| 31 | uint64_t key = (uint64_t)a << 32 | (uint64_t)b; |
||
| 32 | key += ~(key << 32); |
||
| 33 | key ^= (key >> 22); |
||
| 34 | key += ~(key << 13); |
||
| 35 | key ^= (key >> 8); |
||
| 36 | key += (key << 3); |
||
| 37 | key ^= (key >> 15); |
||
| 38 | key += ~(key << 27); |
||
| 39 | key ^= (key >> 31); |
||
| 40 | return (unsigned)key; |
||
| 41 | } |
||
| 42 | |||
| 43 | } // end namespace detail |
||
| 44 | |||
| 45 | /// An information struct used to provide DenseMap with the various necessary |
||
| 46 | /// components for a given value type `T`. `Enable` is an optional additional |
||
| 47 | /// parameter that is used to support SFINAE (generally using std::enable_if_t) |
||
| 48 | /// in derived DenseMapInfo specializations; in non-SFINAE use cases this should |
||
| 49 | /// just be `void`. |
||
| 50 | template<typename T, typename Enable = void> |
||
| 51 | struct DenseMapInfo { |
||
| 52 | //static inline T getEmptyKey(); |
||
| 53 | //static inline T getTombstoneKey(); |
||
| 54 | //static unsigned getHashValue(const T &Val); |
||
| 55 | //static bool isEqual(const T &LHS, const T &RHS); |
||
| 56 | }; |
||
| 57 | |||
| 58 | // Provide DenseMapInfo for all pointers. Come up with sentinel pointer values |
||
| 59 | // that are aligned to alignof(T) bytes, but try to avoid requiring T to be |
||
| 60 | // complete. This allows clients to instantiate DenseMap<T*, ...> with forward |
||
| 61 | // declared key types. Assume that no pointer key type requires more than 4096 |
||
| 62 | // bytes of alignment. |
||
| 63 | template<typename T> |
||
| 64 | struct DenseMapInfo<T*> { |
||
| 65 | // The following should hold, but it would require T to be complete: |
||
| 66 | // static_assert(alignof(T) <= (1 << Log2MaxAlign), |
||
| 67 | // "DenseMap does not support pointer keys requiring more than " |
||
| 68 | // "Log2MaxAlign bits of alignment"); |
||
| 69 | static constexpr uintptr_t Log2MaxAlign = 12; |
||
| 70 | |||
| 71 | static inline T* getEmptyKey() { |
||
| 72 | uintptr_t Val = static_cast<uintptr_t>(-1); |
||
| 73 | Val <<= Log2MaxAlign; |
||
| 74 | return reinterpret_cast<T*>(Val); |
||
| 75 | } |
||
| 76 | |||
| 77 | static inline T* getTombstoneKey() { |
||
| 78 | uintptr_t Val = static_cast<uintptr_t>(-2); |
||
| 79 | Val <<= Log2MaxAlign; |
||
| 80 | return reinterpret_cast<T*>(Val); |
||
| 81 | } |
||
| 82 | |||
| 83 | static unsigned getHashValue(const T *PtrVal) { |
||
| 84 | return (unsigned((uintptr_t)PtrVal) >> 4) ^ |
||
| 85 | (unsigned((uintptr_t)PtrVal) >> 9); |
||
| 86 | } |
||
| 87 | |||
| 88 | static bool isEqual(const T *LHS, const T *RHS) { return LHS == RHS; } |
||
| 89 | }; |
||
| 90 | |||
| 91 | // Provide DenseMapInfo for chars. |
||
| 92 | template<> struct DenseMapInfo<char> { |
||
| 93 | static inline char getEmptyKey() { return ~0; } |
||
| 94 | static inline char getTombstoneKey() { return ~0 - 1; } |
||
| 95 | static unsigned getHashValue(const char& Val) { return Val * 37U; } |
||
| 96 | |||
| 97 | static bool isEqual(const char &LHS, const char &RHS) { |
||
| 98 | return LHS == RHS; |
||
| 99 | } |
||
| 100 | }; |
||
| 101 | |||
| 102 | // Provide DenseMapInfo for unsigned chars. |
||
| 103 | template <> struct DenseMapInfo<unsigned char> { |
||
| 104 | static inline unsigned char getEmptyKey() { return ~0; } |
||
| 105 | static inline unsigned char getTombstoneKey() { return ~0 - 1; } |
||
| 106 | static unsigned getHashValue(const unsigned char &Val) { return Val * 37U; } |
||
| 107 | |||
| 108 | static bool isEqual(const unsigned char &LHS, const unsigned char &RHS) { |
||
| 109 | return LHS == RHS; |
||
| 110 | } |
||
| 111 | }; |
||
| 112 | |||
| 113 | // Provide DenseMapInfo for unsigned shorts. |
||
| 114 | template <> struct DenseMapInfo<unsigned short> { |
||
| 115 | static inline unsigned short getEmptyKey() { return 0xFFFF; } |
||
| 116 | static inline unsigned short getTombstoneKey() { return 0xFFFF - 1; } |
||
| 117 | static unsigned getHashValue(const unsigned short &Val) { return Val * 37U; } |
||
| 118 | |||
| 119 | static bool isEqual(const unsigned short &LHS, const unsigned short &RHS) { |
||
| 120 | return LHS == RHS; |
||
| 121 | } |
||
| 122 | }; |
||
| 123 | |||
| 124 | // Provide DenseMapInfo for unsigned ints. |
||
| 125 | template<> struct DenseMapInfo<unsigned> { |
||
| 126 | static inline unsigned getEmptyKey() { return ~0U; } |
||
| 127 | static inline unsigned getTombstoneKey() { return ~0U - 1; } |
||
| 128 | static unsigned getHashValue(const unsigned& Val) { return Val * 37U; } |
||
| 129 | |||
| 130 | static bool isEqual(const unsigned& LHS, const unsigned& RHS) { |
||
| 131 | return LHS == RHS; |
||
| 132 | } |
||
| 133 | }; |
||
| 134 | |||
| 135 | // Provide DenseMapInfo for unsigned longs. |
||
| 136 | template<> struct DenseMapInfo<unsigned long> { |
||
| 137 | static inline unsigned long getEmptyKey() { return ~0UL; } |
||
| 138 | static inline unsigned long getTombstoneKey() { return ~0UL - 1L; } |
||
| 139 | |||
| 140 | static unsigned getHashValue(const unsigned long& Val) { |
||
| 141 | return (unsigned)(Val * 37UL); |
||
| 142 | } |
||
| 143 | |||
| 144 | static bool isEqual(const unsigned long& LHS, const unsigned long& RHS) { |
||
| 145 | return LHS == RHS; |
||
| 146 | } |
||
| 147 | }; |
||
| 148 | |||
| 149 | // Provide DenseMapInfo for unsigned long longs. |
||
| 150 | template<> struct DenseMapInfo<unsigned long long> { |
||
| 151 | static inline unsigned long long getEmptyKey() { return ~0ULL; } |
||
| 152 | static inline unsigned long long getTombstoneKey() { return ~0ULL - 1ULL; } |
||
| 153 | |||
| 154 | static unsigned getHashValue(const unsigned long long& Val) { |
||
| 155 | return (unsigned)(Val * 37ULL); |
||
| 156 | } |
||
| 157 | |||
| 158 | static bool isEqual(const unsigned long long& LHS, |
||
| 159 | const unsigned long long& RHS) { |
||
| 160 | return LHS == RHS; |
||
| 161 | } |
||
| 162 | }; |
||
| 163 | |||
| 164 | // Provide DenseMapInfo for shorts. |
||
| 165 | template <> struct DenseMapInfo<short> { |
||
| 166 | static inline short getEmptyKey() { return 0x7FFF; } |
||
| 167 | static inline short getTombstoneKey() { return -0x7FFF - 1; } |
||
| 168 | static unsigned getHashValue(const short &Val) { return Val * 37U; } |
||
| 169 | static bool isEqual(const short &LHS, const short &RHS) { return LHS == RHS; } |
||
| 170 | }; |
||
| 171 | |||
| 172 | // Provide DenseMapInfo for ints. |
||
| 173 | template<> struct DenseMapInfo<int> { |
||
| 174 | static inline int getEmptyKey() { return 0x7fffffff; } |
||
| 175 | static inline int getTombstoneKey() { return -0x7fffffff - 1; } |
||
| 176 | static unsigned getHashValue(const int& Val) { return (unsigned)(Val * 37U); } |
||
| 177 | |||
| 178 | static bool isEqual(const int& LHS, const int& RHS) { |
||
| 179 | return LHS == RHS; |
||
| 180 | } |
||
| 181 | }; |
||
| 182 | |||
| 183 | // Provide DenseMapInfo for longs. |
||
| 184 | template<> struct DenseMapInfo<long> { |
||
| 185 | static inline long getEmptyKey() { |
||
| 186 | return (1UL << (sizeof(long) * 8 - 1)) - 1UL; |
||
| 187 | } |
||
| 188 | |||
| 189 | static inline long getTombstoneKey() { return getEmptyKey() - 1L; } |
||
| 190 | |||
| 191 | static unsigned getHashValue(const long& Val) { |
||
| 192 | return (unsigned)(Val * 37UL); |
||
| 193 | } |
||
| 194 | |||
| 195 | static bool isEqual(const long& LHS, const long& RHS) { |
||
| 196 | return LHS == RHS; |
||
| 197 | } |
||
| 198 | }; |
||
| 199 | |||
| 200 | // Provide DenseMapInfo for long longs. |
||
| 201 | template<> struct DenseMapInfo<long long> { |
||
| 202 | static inline long long getEmptyKey() { return 0x7fffffffffffffffLL; } |
||
| 203 | static inline long long getTombstoneKey() { return -0x7fffffffffffffffLL-1; } |
||
| 204 | |||
| 205 | static unsigned getHashValue(const long long& Val) { |
||
| 206 | return (unsigned)(Val * 37ULL); |
||
| 207 | } |
||
| 208 | |||
| 209 | static bool isEqual(const long long& LHS, |
||
| 210 | const long long& RHS) { |
||
| 211 | return LHS == RHS; |
||
| 212 | } |
||
| 213 | }; |
||
| 214 | |||
| 215 | // Provide DenseMapInfo for all pairs whose members have info. |
||
| 216 | template<typename T, typename U> |
||
| 217 | struct DenseMapInfo<std::pair<T, U>> { |
||
| 218 | using Pair = std::pair<T, U>; |
||
| 219 | using FirstInfo = DenseMapInfo<T>; |
||
| 220 | using SecondInfo = DenseMapInfo<U>; |
||
| 221 | |||
| 222 | static inline Pair getEmptyKey() { |
||
| 223 | return std::make_pair(FirstInfo::getEmptyKey(), |
||
| 224 | SecondInfo::getEmptyKey()); |
||
| 225 | } |
||
| 226 | |||
| 227 | static inline Pair getTombstoneKey() { |
||
| 228 | return std::make_pair(FirstInfo::getTombstoneKey(), |
||
| 229 | SecondInfo::getTombstoneKey()); |
||
| 230 | } |
||
| 231 | |||
| 232 | static unsigned getHashValue(const Pair& PairVal) { |
||
| 233 | return detail::combineHashValue(FirstInfo::getHashValue(PairVal.first), |
||
| 234 | SecondInfo::getHashValue(PairVal.second)); |
||
| 235 | } |
||
| 236 | |||
| 237 | static bool isEqual(const Pair &LHS, const Pair &RHS) { |
||
| 238 | return FirstInfo::isEqual(LHS.first, RHS.first) && |
||
| 239 | SecondInfo::isEqual(LHS.second, RHS.second); |
||
| 240 | } |
||
| 241 | }; |
||
| 242 | |||
| 243 | // Provide DenseMapInfo for all tuples whose members have info. |
||
| 244 | template <typename... Ts> struct DenseMapInfo<std::tuple<Ts...>> { |
||
| 245 | using Tuple = std::tuple<Ts...>; |
||
| 246 | |||
| 247 | static inline Tuple getEmptyKey() { |
||
| 248 | return Tuple(DenseMapInfo<Ts>::getEmptyKey()...); |
||
| 249 | } |
||
| 250 | |||
| 251 | static inline Tuple getTombstoneKey() { |
||
| 252 | return Tuple(DenseMapInfo<Ts>::getTombstoneKey()...); |
||
| 253 | } |
||
| 254 | |||
| 255 | template <unsigned I> |
||
| 256 | static unsigned getHashValueImpl(const Tuple &values, std::false_type) { |
||
| 257 | using EltType = std::tuple_element_t<I, Tuple>; |
||
| 258 | std::integral_constant<bool, I + 1 == sizeof...(Ts)> atEnd; |
||
| 259 | return detail::combineHashValue( |
||
| 260 | DenseMapInfo<EltType>::getHashValue(std::get<I>(values)), |
||
| 261 | getHashValueImpl<I + 1>(values, atEnd)); |
||
| 262 | } |
||
| 263 | |||
| 264 | template <unsigned I> |
||
| 265 | static unsigned getHashValueImpl(const Tuple &, std::true_type) { |
||
| 266 | return 0; |
||
| 267 | } |
||
| 268 | |||
| 269 | static unsigned getHashValue(const std::tuple<Ts...> &values) { |
||
| 270 | std::integral_constant<bool, 0 == sizeof...(Ts)> atEnd; |
||
| 271 | return getHashValueImpl<0>(values, atEnd); |
||
| 272 | } |
||
| 273 | |||
| 274 | template <unsigned I> |
||
| 275 | static bool isEqualImpl(const Tuple &lhs, const Tuple &rhs, std::false_type) { |
||
| 276 | using EltType = std::tuple_element_t<I, Tuple>; |
||
| 277 | std::integral_constant<bool, I + 1 == sizeof...(Ts)> atEnd; |
||
| 278 | return DenseMapInfo<EltType>::isEqual(std::get<I>(lhs), std::get<I>(rhs)) && |
||
| 279 | isEqualImpl<I + 1>(lhs, rhs, atEnd); |
||
| 280 | } |
||
| 281 | |||
| 282 | template <unsigned I> |
||
| 283 | static bool isEqualImpl(const Tuple &, const Tuple &, std::true_type) { |
||
| 284 | return true; |
||
| 285 | } |
||
| 286 | |||
| 287 | static bool isEqual(const Tuple &lhs, const Tuple &rhs) { |
||
| 288 | std::integral_constant<bool, 0 == sizeof...(Ts)> atEnd; |
||
| 289 | return isEqualImpl<0>(lhs, rhs, atEnd); |
||
| 290 | } |
||
| 291 | }; |
||
| 292 | |||
| 293 | // Provide DenseMapInfo for variants whose all alternatives have DenseMapInfo. |
||
| 294 | template <typename... Ts> struct DenseMapInfo<std::variant<Ts...>> { |
||
| 295 | using Variant = std::variant<Ts...>; |
||
| 296 | using FirstT = std::variant_alternative_t<0, Variant>; |
||
| 297 | |||
| 298 | static inline Variant getEmptyKey() { |
||
| 299 | return Variant(std::in_place_index<0>, DenseMapInfo<FirstT>::getEmptyKey()); |
||
| 300 | } |
||
| 301 | |||
| 302 | static inline Variant getTombstoneKey() { |
||
| 303 | return Variant(std::in_place_index<0>, |
||
| 304 | DenseMapInfo<FirstT>::getTombstoneKey()); |
||
| 305 | } |
||
| 306 | |||
| 307 | static unsigned getHashValue(const Variant &Val) { |
||
| 308 | return std::visit( |
||
| 309 | [&Val](auto &&Alternative) { |
||
| 310 | using T = std::decay_t<decltype(Alternative)>; |
||
| 311 | // Include index in hash to make sure same value as different |
||
| 312 | // alternatives don't collide. |
||
| 313 | return detail::combineHashValue( |
||
| 314 | DenseMapInfo<size_t>::getHashValue(Val.index()), |
||
| 315 | DenseMapInfo<T>::getHashValue(Alternative)); |
||
| 316 | }, |
||
| 317 | Val); |
||
| 318 | } |
||
| 319 | |||
| 320 | static bool isEqual(const Variant &LHS, const Variant &RHS) { |
||
| 321 | return LHS == RHS; |
||
| 322 | } |
||
| 323 | }; |
||
| 324 | } // end namespace llvm |
||
| 325 | |||
| 326 | #endif // LLVM_ADT_DENSEMAPINFO_H |