Details | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line |
|---|---|---|---|
| 14 | pmbaty | 1 | //===- llvm/ADT/PointerIntPair.h - Pair for pointer and int -----*- 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 the PointerIntPair class. |
||
| 11 | /// |
||
| 12 | //===----------------------------------------------------------------------===// |
||
| 13 | |||
| 14 | #ifndef LLVM_ADT_POINTERINTPAIR_H |
||
| 15 | #define LLVM_ADT_POINTERINTPAIR_H |
||
| 16 | |||
| 17 | #include "llvm/Support/Compiler.h" |
||
| 18 | #include "llvm/Support/PointerLikeTypeTraits.h" |
||
| 19 | #include "llvm/Support/type_traits.h" |
||
| 20 | #include <cassert> |
||
| 21 | #include <cstdint> |
||
| 22 | #include <limits> |
||
| 23 | |||
| 24 | namespace llvm { |
||
| 25 | |||
| 26 | template <typename T, typename Enable> struct DenseMapInfo; |
||
| 27 | template <typename PointerT, unsigned IntBits, typename PtrTraits> |
||
| 28 | struct PointerIntPairInfo; |
||
| 29 | |||
| 30 | /// PointerIntPair - This class implements a pair of a pointer and small |
||
| 31 | /// integer. It is designed to represent this in the space required by one |
||
| 32 | /// pointer by bitmangling the integer into the low part of the pointer. This |
||
| 33 | /// can only be done for small integers: typically up to 3 bits, but it depends |
||
| 34 | /// on the number of bits available according to PointerLikeTypeTraits for the |
||
| 35 | /// type. |
||
| 36 | /// |
||
| 37 | /// Note that PointerIntPair always puts the IntVal part in the highest bits |
||
| 38 | /// possible. For example, PointerIntPair<void*, 1, bool> will put the bit for |
||
| 39 | /// the bool into bit #2, not bit #0, which allows the low two bits to be used |
||
| 40 | /// for something else. For example, this allows: |
||
| 41 | /// PointerIntPair<PointerIntPair<void*, 1, bool>, 1, bool> |
||
| 42 | /// ... and the two bools will land in different bits. |
||
| 43 | template <typename PointerTy, unsigned IntBits, typename IntType = unsigned, |
||
| 44 | typename PtrTraits = PointerLikeTypeTraits<PointerTy>, |
||
| 45 | typename Info = PointerIntPairInfo<PointerTy, IntBits, PtrTraits>> |
||
| 46 | class PointerIntPair { |
||
| 47 | // Used by MSVC visualizer and generally helpful for debugging/visualizing. |
||
| 48 | using InfoTy = Info; |
||
| 49 | intptr_t Value = 0; |
||
| 50 | |||
| 51 | public: |
||
| 52 | constexpr PointerIntPair() = default; |
||
| 53 | |||
| 54 | PointerIntPair(PointerTy PtrVal, IntType IntVal) { |
||
| 55 | setPointerAndInt(PtrVal, IntVal); |
||
| 56 | } |
||
| 57 | |||
| 58 | explicit PointerIntPair(PointerTy PtrVal) { initWithPointer(PtrVal); } |
||
| 59 | |||
| 60 | PointerTy getPointer() const { return Info::getPointer(Value); } |
||
| 61 | |||
| 62 | IntType getInt() const { return (IntType)Info::getInt(Value); } |
||
| 63 | |||
| 64 | void setPointer(PointerTy PtrVal) & { |
||
| 65 | Value = Info::updatePointer(Value, PtrVal); |
||
| 66 | } |
||
| 67 | |||
| 68 | void setInt(IntType IntVal) & { |
||
| 69 | Value = Info::updateInt(Value, static_cast<intptr_t>(IntVal)); |
||
| 70 | } |
||
| 71 | |||
| 72 | void initWithPointer(PointerTy PtrVal) & { |
||
| 73 | Value = Info::updatePointer(0, PtrVal); |
||
| 74 | } |
||
| 75 | |||
| 76 | void setPointerAndInt(PointerTy PtrVal, IntType IntVal) & { |
||
| 77 | Value = Info::updateInt(Info::updatePointer(0, PtrVal), |
||
| 78 | static_cast<intptr_t>(IntVal)); |
||
| 79 | } |
||
| 80 | |||
| 81 | PointerTy const *getAddrOfPointer() const { |
||
| 82 | return const_cast<PointerIntPair *>(this)->getAddrOfPointer(); |
||
| 83 | } |
||
| 84 | |||
| 85 | PointerTy *getAddrOfPointer() { |
||
| 86 | assert(Value == reinterpret_cast<intptr_t>(getPointer()) && |
||
| 87 | "Can only return the address if IntBits is cleared and " |
||
| 88 | "PtrTraits doesn't change the pointer"); |
||
| 89 | return reinterpret_cast<PointerTy *>(&Value); |
||
| 90 | } |
||
| 91 | |||
| 92 | void *getOpaqueValue() const { return reinterpret_cast<void *>(Value); } |
||
| 93 | |||
| 94 | void setFromOpaqueValue(void *Val) & { |
||
| 95 | Value = reinterpret_cast<intptr_t>(Val); |
||
| 96 | } |
||
| 97 | |||
| 98 | static PointerIntPair getFromOpaqueValue(void *V) { |
||
| 99 | PointerIntPair P; |
||
| 100 | P.setFromOpaqueValue(V); |
||
| 101 | return P; |
||
| 102 | } |
||
| 103 | |||
| 104 | // Allow PointerIntPairs to be created from const void * if and only if the |
||
| 105 | // pointer type could be created from a const void *. |
||
| 106 | static PointerIntPair getFromOpaqueValue(const void *V) { |
||
| 107 | (void)PtrTraits::getFromVoidPointer(V); |
||
| 108 | return getFromOpaqueValue(const_cast<void *>(V)); |
||
| 109 | } |
||
| 110 | |||
| 111 | bool operator==(const PointerIntPair &RHS) const { |
||
| 112 | return Value == RHS.Value; |
||
| 113 | } |
||
| 114 | |||
| 115 | bool operator!=(const PointerIntPair &RHS) const { |
||
| 116 | return Value != RHS.Value; |
||
| 117 | } |
||
| 118 | |||
| 119 | bool operator<(const PointerIntPair &RHS) const { return Value < RHS.Value; } |
||
| 120 | bool operator>(const PointerIntPair &RHS) const { return Value > RHS.Value; } |
||
| 121 | |||
| 122 | bool operator<=(const PointerIntPair &RHS) const { |
||
| 123 | return Value <= RHS.Value; |
||
| 124 | } |
||
| 125 | |||
| 126 | bool operator>=(const PointerIntPair &RHS) const { |
||
| 127 | return Value >= RHS.Value; |
||
| 128 | } |
||
| 129 | }; |
||
| 130 | |||
| 131 | template <typename PointerT, unsigned IntBits, typename PtrTraits> |
||
| 132 | struct PointerIntPairInfo { |
||
| 133 | static_assert(PtrTraits::NumLowBitsAvailable < |
||
| 134 | std::numeric_limits<uintptr_t>::digits, |
||
| 135 | "cannot use a pointer type that has all bits free"); |
||
| 136 | static_assert(IntBits <= PtrTraits::NumLowBitsAvailable, |
||
| 137 | "PointerIntPair with integer size too large for pointer"); |
||
| 138 | enum MaskAndShiftConstants : uintptr_t { |
||
| 139 | /// PointerBitMask - The bits that come from the pointer. |
||
| 140 | PointerBitMask = |
||
| 141 | ~(uintptr_t)(((intptr_t)1 << PtrTraits::NumLowBitsAvailable) - 1), |
||
| 142 | |||
| 143 | /// IntShift - The number of low bits that we reserve for other uses, and |
||
| 144 | /// keep zero. |
||
| 145 | IntShift = (uintptr_t)PtrTraits::NumLowBitsAvailable - IntBits, |
||
| 146 | |||
| 147 | /// IntMask - This is the unshifted mask for valid bits of the int type. |
||
| 148 | IntMask = (uintptr_t)(((intptr_t)1 << IntBits) - 1), |
||
| 149 | |||
| 150 | // ShiftedIntMask - This is the bits for the integer shifted in place. |
||
| 151 | ShiftedIntMask = (uintptr_t)(IntMask << IntShift) |
||
| 152 | }; |
||
| 153 | |||
| 154 | static PointerT getPointer(intptr_t Value) { |
||
| 155 | return PtrTraits::getFromVoidPointer( |
||
| 156 | reinterpret_cast<void *>(Value & PointerBitMask)); |
||
| 157 | } |
||
| 158 | |||
| 159 | static intptr_t getInt(intptr_t Value) { |
||
| 160 | return (Value >> IntShift) & IntMask; |
||
| 161 | } |
||
| 162 | |||
| 163 | static intptr_t updatePointer(intptr_t OrigValue, PointerT Ptr) { |
||
| 164 | intptr_t PtrWord = |
||
| 165 | reinterpret_cast<intptr_t>(PtrTraits::getAsVoidPointer(Ptr)); |
||
| 166 | assert((PtrWord & ~PointerBitMask) == 0 && |
||
| 167 | "Pointer is not sufficiently aligned"); |
||
| 168 | // Preserve all low bits, just update the pointer. |
||
| 169 | return PtrWord | (OrigValue & ~PointerBitMask); |
||
| 170 | } |
||
| 171 | |||
| 172 | static intptr_t updateInt(intptr_t OrigValue, intptr_t Int) { |
||
| 173 | intptr_t IntWord = static_cast<intptr_t>(Int); |
||
| 174 | assert((IntWord & ~IntMask) == 0 && "Integer too large for field"); |
||
| 175 | |||
| 176 | // Preserve all bits other than the ones we are updating. |
||
| 177 | return (OrigValue & ~ShiftedIntMask) | IntWord << IntShift; |
||
| 178 | } |
||
| 179 | }; |
||
| 180 | |||
| 181 | // Provide specialization of DenseMapInfo for PointerIntPair. |
||
| 182 | template <typename PointerTy, unsigned IntBits, typename IntType> |
||
| 183 | struct DenseMapInfo<PointerIntPair<PointerTy, IntBits, IntType>, void> { |
||
| 184 | using Ty = PointerIntPair<PointerTy, IntBits, IntType>; |
||
| 185 | |||
| 186 | static Ty getEmptyKey() { |
||
| 187 | uintptr_t Val = static_cast<uintptr_t>(-1); |
||
| 188 | Val <<= PointerLikeTypeTraits<Ty>::NumLowBitsAvailable; |
||
| 189 | return Ty::getFromOpaqueValue(reinterpret_cast<void *>(Val)); |
||
| 190 | } |
||
| 191 | |||
| 192 | static Ty getTombstoneKey() { |
||
| 193 | uintptr_t Val = static_cast<uintptr_t>(-2); |
||
| 194 | Val <<= PointerLikeTypeTraits<PointerTy>::NumLowBitsAvailable; |
||
| 195 | return Ty::getFromOpaqueValue(reinterpret_cast<void *>(Val)); |
||
| 196 | } |
||
| 197 | |||
| 198 | static unsigned getHashValue(Ty V) { |
||
| 199 | uintptr_t IV = reinterpret_cast<uintptr_t>(V.getOpaqueValue()); |
||
| 200 | return unsigned(IV) ^ unsigned(IV >> 9); |
||
| 201 | } |
||
| 202 | |||
| 203 | static bool isEqual(const Ty &LHS, const Ty &RHS) { return LHS == RHS; } |
||
| 204 | }; |
||
| 205 | |||
| 206 | // Teach SmallPtrSet that PointerIntPair is "basically a pointer". |
||
| 207 | template <typename PointerTy, unsigned IntBits, typename IntType, |
||
| 208 | typename PtrTraits> |
||
| 209 | struct PointerLikeTypeTraits< |
||
| 210 | PointerIntPair<PointerTy, IntBits, IntType, PtrTraits>> { |
||
| 211 | static inline void * |
||
| 212 | getAsVoidPointer(const PointerIntPair<PointerTy, IntBits, IntType> &P) { |
||
| 213 | return P.getOpaqueValue(); |
||
| 214 | } |
||
| 215 | |||
| 216 | static inline PointerIntPair<PointerTy, IntBits, IntType> |
||
| 217 | getFromVoidPointer(void *P) { |
||
| 218 | return PointerIntPair<PointerTy, IntBits, IntType>::getFromOpaqueValue(P); |
||
| 219 | } |
||
| 220 | |||
| 221 | static inline PointerIntPair<PointerTy, IntBits, IntType> |
||
| 222 | getFromVoidPointer(const void *P) { |
||
| 223 | return PointerIntPair<PointerTy, IntBits, IntType>::getFromOpaqueValue(P); |
||
| 224 | } |
||
| 225 | |||
| 226 | static constexpr int NumLowBitsAvailable = |
||
| 227 | PtrTraits::NumLowBitsAvailable - IntBits; |
||
| 228 | }; |
||
| 229 | |||
| 230 | // Allow structured bindings on PointerIntPair. |
||
| 231 | template <std::size_t I, typename PointerTy, unsigned IntBits, typename IntType, |
||
| 232 | typename PtrTraits, typename Info> |
||
| 233 | decltype(auto) |
||
| 234 | get(const PointerIntPair<PointerTy, IntBits, IntType, PtrTraits, Info> &Pair) { |
||
| 235 | static_assert(I < 2); |
||
| 236 | if constexpr (I == 0) |
||
| 237 | return Pair.getPointer(); |
||
| 238 | else |
||
| 239 | return Pair.getInt(); |
||
| 240 | } |
||
| 241 | |||
| 242 | } // end namespace llvm |
||
| 243 | |||
| 244 | namespace std { |
||
| 245 | template <typename PointerTy, unsigned IntBits, typename IntType, |
||
| 246 | typename PtrTraits, typename Info> |
||
| 247 | struct tuple_size< |
||
| 248 | llvm::PointerIntPair<PointerTy, IntBits, IntType, PtrTraits, Info>> |
||
| 249 | : std::integral_constant<std::size_t, 2> {}; |
||
| 250 | |||
| 251 | template <std::size_t I, typename PointerTy, unsigned IntBits, typename IntType, |
||
| 252 | typename PtrTraits, typename Info> |
||
| 253 | struct tuple_element< |
||
| 254 | I, llvm::PointerIntPair<PointerTy, IntBits, IntType, PtrTraits, Info>> |
||
| 255 | : std::conditional<I == 0, PointerTy, IntType> {}; |
||
| 256 | } // namespace std |
||
| 257 | |||
| 258 | #endif // LLVM_ADT_POINTERINTPAIR_H |