Owen Anderson | 3fb4aab | 2009-08-23 04:24:24 +0000 | [diff] [blame] | 1 | //===-- ConstantsContext.h - Constants-related Context Interals -----------===// |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 2 | // |
| 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
| 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
| 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | // |
| 10 | // This file defines various helper methods and classes used by |
| 11 | // LLVMContextImpl for creating and managing constants. |
| 12 | // |
| 13 | //===----------------------------------------------------------------------===// |
| 14 | |
Benjamin Kramer | a7c40ef | 2014-08-13 16:26:38 +0000 | [diff] [blame] | 15 | #ifndef LLVM_LIB_IR_CONSTANTSCONTEXT_H |
| 16 | #define LLVM_LIB_IR_CONSTANTSCONTEXT_H |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 17 | |
Talin | 46e9b44 | 2012-02-05 20:54:10 +0000 | [diff] [blame] | 18 | #include "llvm/ADT/DenseMap.h" |
Jay Foad | cc5fd3e | 2012-03-06 10:43:52 +0000 | [diff] [blame] | 19 | #include "llvm/ADT/Hashing.h" |
Chandler Carruth | 9fb823b | 2013-01-02 11:36:10 +0000 | [diff] [blame] | 20 | #include "llvm/IR/InlineAsm.h" |
| 21 | #include "llvm/IR/Instructions.h" |
| 22 | #include "llvm/IR/Operator.h" |
David Greene | 338a903 | 2010-01-05 01:34:26 +0000 | [diff] [blame] | 23 | #include "llvm/Support/Debug.h" |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 24 | #include "llvm/Support/ErrorHandling.h" |
Chris Lattner | 34822f6 | 2009-08-23 04:44:11 +0000 | [diff] [blame] | 25 | #include "llvm/Support/raw_ostream.h" |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 26 | #include <map> |
Benjamin Kramer | 7499657 | 2014-04-29 23:37:02 +0000 | [diff] [blame] | 27 | #include <tuple> |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 28 | |
Chandler Carruth | e96dd89 | 2014-04-21 22:55:11 +0000 | [diff] [blame] | 29 | #define DEBUG_TYPE "ir" |
| 30 | |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 31 | namespace llvm { |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 32 | |
| 33 | /// UnaryConstantExpr - This class is private to Constants.cpp, and is used |
| 34 | /// behind the scenes to implement unary constant exprs. |
Benjamin Kramer | 079b96e | 2013-09-11 18:05:11 +0000 | [diff] [blame] | 35 | class UnaryConstantExpr : public ConstantExpr { |
Craig Topper | f398d7c | 2014-03-05 06:35:38 +0000 | [diff] [blame] | 36 | void anchor() override; |
Craig Topper | a60c0f1 | 2012-09-15 17:09:36 +0000 | [diff] [blame] | 37 | void *operator new(size_t, unsigned) LLVM_DELETED_FUNCTION; |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 38 | public: |
| 39 | // allocate space for exactly one operand |
| 40 | void *operator new(size_t s) { |
| 41 | return User::operator new(s, 1); |
| 42 | } |
Chris Lattner | 229907c | 2011-07-18 04:54:35 +0000 | [diff] [blame] | 43 | UnaryConstantExpr(unsigned Opcode, Constant *C, Type *Ty) |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 44 | : ConstantExpr(Ty, Opcode, &Op<0>(), 1) { |
| 45 | Op<0>() = C; |
| 46 | } |
| 47 | DECLARE_TRANSPARENT_OPERAND_ACCESSORS(Value); |
| 48 | }; |
| 49 | |
| 50 | /// BinaryConstantExpr - This class is private to Constants.cpp, and is used |
| 51 | /// behind the scenes to implement binary constant exprs. |
Benjamin Kramer | 079b96e | 2013-09-11 18:05:11 +0000 | [diff] [blame] | 52 | class BinaryConstantExpr : public ConstantExpr { |
Craig Topper | f398d7c | 2014-03-05 06:35:38 +0000 | [diff] [blame] | 53 | void anchor() override; |
Craig Topper | a60c0f1 | 2012-09-15 17:09:36 +0000 | [diff] [blame] | 54 | void *operator new(size_t, unsigned) LLVM_DELETED_FUNCTION; |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 55 | public: |
| 56 | // allocate space for exactly two operands |
| 57 | void *operator new(size_t s) { |
| 58 | return User::operator new(s, 2); |
| 59 | } |
Dan Gohman | 1b84908 | 2009-09-07 23:54:19 +0000 | [diff] [blame] | 60 | BinaryConstantExpr(unsigned Opcode, Constant *C1, Constant *C2, |
| 61 | unsigned Flags) |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 62 | : ConstantExpr(C1->getType(), Opcode, &Op<0>(), 2) { |
| 63 | Op<0>() = C1; |
| 64 | Op<1>() = C2; |
Dan Gohman | 1b84908 | 2009-09-07 23:54:19 +0000 | [diff] [blame] | 65 | SubclassOptionalData = Flags; |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 66 | } |
| 67 | /// Transparently provide more efficient getOperand methods. |
| 68 | DECLARE_TRANSPARENT_OPERAND_ACCESSORS(Value); |
| 69 | }; |
| 70 | |
| 71 | /// SelectConstantExpr - This class is private to Constants.cpp, and is used |
| 72 | /// behind the scenes to implement select constant exprs. |
Benjamin Kramer | 079b96e | 2013-09-11 18:05:11 +0000 | [diff] [blame] | 73 | class SelectConstantExpr : public ConstantExpr { |
Craig Topper | f398d7c | 2014-03-05 06:35:38 +0000 | [diff] [blame] | 74 | void anchor() override; |
Craig Topper | a60c0f1 | 2012-09-15 17:09:36 +0000 | [diff] [blame] | 75 | void *operator new(size_t, unsigned) LLVM_DELETED_FUNCTION; |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 76 | public: |
| 77 | // allocate space for exactly three operands |
| 78 | void *operator new(size_t s) { |
| 79 | return User::operator new(s, 3); |
| 80 | } |
| 81 | SelectConstantExpr(Constant *C1, Constant *C2, Constant *C3) |
| 82 | : ConstantExpr(C2->getType(), Instruction::Select, &Op<0>(), 3) { |
| 83 | Op<0>() = C1; |
| 84 | Op<1>() = C2; |
| 85 | Op<2>() = C3; |
| 86 | } |
| 87 | /// Transparently provide more efficient getOperand methods. |
| 88 | DECLARE_TRANSPARENT_OPERAND_ACCESSORS(Value); |
| 89 | }; |
| 90 | |
| 91 | /// ExtractElementConstantExpr - This class is private to |
| 92 | /// Constants.cpp, and is used behind the scenes to implement |
| 93 | /// extractelement constant exprs. |
Benjamin Kramer | 079b96e | 2013-09-11 18:05:11 +0000 | [diff] [blame] | 94 | class ExtractElementConstantExpr : public ConstantExpr { |
Craig Topper | f398d7c | 2014-03-05 06:35:38 +0000 | [diff] [blame] | 95 | void anchor() override; |
Craig Topper | a60c0f1 | 2012-09-15 17:09:36 +0000 | [diff] [blame] | 96 | void *operator new(size_t, unsigned) LLVM_DELETED_FUNCTION; |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 97 | public: |
| 98 | // allocate space for exactly two operands |
| 99 | void *operator new(size_t s) { |
| 100 | return User::operator new(s, 2); |
| 101 | } |
| 102 | ExtractElementConstantExpr(Constant *C1, Constant *C2) |
| 103 | : ConstantExpr(cast<VectorType>(C1->getType())->getElementType(), |
| 104 | Instruction::ExtractElement, &Op<0>(), 2) { |
| 105 | Op<0>() = C1; |
| 106 | Op<1>() = C2; |
| 107 | } |
| 108 | /// Transparently provide more efficient getOperand methods. |
| 109 | DECLARE_TRANSPARENT_OPERAND_ACCESSORS(Value); |
| 110 | }; |
| 111 | |
| 112 | /// InsertElementConstantExpr - This class is private to |
| 113 | /// Constants.cpp, and is used behind the scenes to implement |
| 114 | /// insertelement constant exprs. |
Benjamin Kramer | 079b96e | 2013-09-11 18:05:11 +0000 | [diff] [blame] | 115 | class InsertElementConstantExpr : public ConstantExpr { |
Craig Topper | f398d7c | 2014-03-05 06:35:38 +0000 | [diff] [blame] | 116 | void anchor() override; |
Craig Topper | a60c0f1 | 2012-09-15 17:09:36 +0000 | [diff] [blame] | 117 | void *operator new(size_t, unsigned) LLVM_DELETED_FUNCTION; |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 118 | public: |
| 119 | // allocate space for exactly three operands |
| 120 | void *operator new(size_t s) { |
| 121 | return User::operator new(s, 3); |
| 122 | } |
| 123 | InsertElementConstantExpr(Constant *C1, Constant *C2, Constant *C3) |
| 124 | : ConstantExpr(C1->getType(), Instruction::InsertElement, |
| 125 | &Op<0>(), 3) { |
| 126 | Op<0>() = C1; |
| 127 | Op<1>() = C2; |
| 128 | Op<2>() = C3; |
| 129 | } |
| 130 | /// Transparently provide more efficient getOperand methods. |
| 131 | DECLARE_TRANSPARENT_OPERAND_ACCESSORS(Value); |
| 132 | }; |
| 133 | |
| 134 | /// ShuffleVectorConstantExpr - This class is private to |
| 135 | /// Constants.cpp, and is used behind the scenes to implement |
| 136 | /// shufflevector constant exprs. |
Benjamin Kramer | 079b96e | 2013-09-11 18:05:11 +0000 | [diff] [blame] | 137 | class ShuffleVectorConstantExpr : public ConstantExpr { |
Craig Topper | f398d7c | 2014-03-05 06:35:38 +0000 | [diff] [blame] | 138 | void anchor() override; |
Craig Topper | a60c0f1 | 2012-09-15 17:09:36 +0000 | [diff] [blame] | 139 | void *operator new(size_t, unsigned) LLVM_DELETED_FUNCTION; |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 140 | public: |
| 141 | // allocate space for exactly three operands |
| 142 | void *operator new(size_t s) { |
| 143 | return User::operator new(s, 3); |
| 144 | } |
| 145 | ShuffleVectorConstantExpr(Constant *C1, Constant *C2, Constant *C3) |
| 146 | : ConstantExpr(VectorType::get( |
| 147 | cast<VectorType>(C1->getType())->getElementType(), |
| 148 | cast<VectorType>(C3->getType())->getNumElements()), |
| 149 | Instruction::ShuffleVector, |
| 150 | &Op<0>(), 3) { |
| 151 | Op<0>() = C1; |
| 152 | Op<1>() = C2; |
| 153 | Op<2>() = C3; |
| 154 | } |
| 155 | /// Transparently provide more efficient getOperand methods. |
| 156 | DECLARE_TRANSPARENT_OPERAND_ACCESSORS(Value); |
| 157 | }; |
| 158 | |
| 159 | /// ExtractValueConstantExpr - This class is private to |
| 160 | /// Constants.cpp, and is used behind the scenes to implement |
| 161 | /// extractvalue constant exprs. |
Benjamin Kramer | 079b96e | 2013-09-11 18:05:11 +0000 | [diff] [blame] | 162 | class ExtractValueConstantExpr : public ConstantExpr { |
Craig Topper | f398d7c | 2014-03-05 06:35:38 +0000 | [diff] [blame] | 163 | void anchor() override; |
Craig Topper | a60c0f1 | 2012-09-15 17:09:36 +0000 | [diff] [blame] | 164 | void *operator new(size_t, unsigned) LLVM_DELETED_FUNCTION; |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 165 | public: |
| 166 | // allocate space for exactly one operand |
| 167 | void *operator new(size_t s) { |
| 168 | return User::operator new(s, 1); |
| 169 | } |
Duncan P. N. Exon Smith | 1cd4aeb | 2014-08-19 00:23:17 +0000 | [diff] [blame] | 170 | ExtractValueConstantExpr(Constant *Agg, ArrayRef<unsigned> IdxList, |
Chris Lattner | 229907c | 2011-07-18 04:54:35 +0000 | [diff] [blame] | 171 | Type *DestTy) |
Duncan P. N. Exon Smith | 1cd4aeb | 2014-08-19 00:23:17 +0000 | [diff] [blame] | 172 | : ConstantExpr(DestTy, Instruction::ExtractValue, &Op<0>(), 1), |
| 173 | Indices(IdxList.begin(), IdxList.end()) { |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 174 | Op<0>() = Agg; |
| 175 | } |
| 176 | |
| 177 | /// Indices - These identify which value to extract. |
| 178 | const SmallVector<unsigned, 4> Indices; |
| 179 | |
| 180 | /// Transparently provide more efficient getOperand methods. |
| 181 | DECLARE_TRANSPARENT_OPERAND_ACCESSORS(Value); |
| 182 | }; |
| 183 | |
| 184 | /// InsertValueConstantExpr - This class is private to |
| 185 | /// Constants.cpp, and is used behind the scenes to implement |
| 186 | /// insertvalue constant exprs. |
Benjamin Kramer | 079b96e | 2013-09-11 18:05:11 +0000 | [diff] [blame] | 187 | class InsertValueConstantExpr : public ConstantExpr { |
Craig Topper | f398d7c | 2014-03-05 06:35:38 +0000 | [diff] [blame] | 188 | void anchor() override; |
Craig Topper | a60c0f1 | 2012-09-15 17:09:36 +0000 | [diff] [blame] | 189 | void *operator new(size_t, unsigned) LLVM_DELETED_FUNCTION; |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 190 | public: |
| 191 | // allocate space for exactly one operand |
| 192 | void *operator new(size_t s) { |
| 193 | return User::operator new(s, 2); |
| 194 | } |
| 195 | InsertValueConstantExpr(Constant *Agg, Constant *Val, |
Duncan P. N. Exon Smith | 1cd4aeb | 2014-08-19 00:23:17 +0000 | [diff] [blame] | 196 | ArrayRef<unsigned> IdxList, Type *DestTy) |
| 197 | : ConstantExpr(DestTy, Instruction::InsertValue, &Op<0>(), 2), |
| 198 | Indices(IdxList.begin(), IdxList.end()) { |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 199 | Op<0>() = Agg; |
| 200 | Op<1>() = Val; |
| 201 | } |
| 202 | |
| 203 | /// Indices - These identify the position for the insertion. |
| 204 | const SmallVector<unsigned, 4> Indices; |
| 205 | |
| 206 | /// Transparently provide more efficient getOperand methods. |
| 207 | DECLARE_TRANSPARENT_OPERAND_ACCESSORS(Value); |
| 208 | }; |
| 209 | |
| 210 | |
| 211 | /// GetElementPtrConstantExpr - This class is private to Constants.cpp, and is |
| 212 | /// used behind the scenes to implement getelementpr constant exprs. |
Benjamin Kramer | 079b96e | 2013-09-11 18:05:11 +0000 | [diff] [blame] | 213 | class GetElementPtrConstantExpr : public ConstantExpr { |
Craig Topper | f398d7c | 2014-03-05 06:35:38 +0000 | [diff] [blame] | 214 | void anchor() override; |
Chris Lattner | a474bb2 | 2012-01-26 20:40:56 +0000 | [diff] [blame] | 215 | GetElementPtrConstantExpr(Constant *C, ArrayRef<Constant*> IdxList, |
Chris Lattner | 229907c | 2011-07-18 04:54:35 +0000 | [diff] [blame] | 216 | Type *DestTy); |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 217 | public: |
| 218 | static GetElementPtrConstantExpr *Create(Constant *C, |
Chris Lattner | a474bb2 | 2012-01-26 20:40:56 +0000 | [diff] [blame] | 219 | ArrayRef<Constant*> IdxList, |
Chris Lattner | 229907c | 2011-07-18 04:54:35 +0000 | [diff] [blame] | 220 | Type *DestTy, |
Dan Gohman | 1b84908 | 2009-09-07 23:54:19 +0000 | [diff] [blame] | 221 | unsigned Flags) { |
| 222 | GetElementPtrConstantExpr *Result = |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 223 | new(IdxList.size() + 1) GetElementPtrConstantExpr(C, IdxList, DestTy); |
Dan Gohman | 1b84908 | 2009-09-07 23:54:19 +0000 | [diff] [blame] | 224 | Result->SubclassOptionalData = Flags; |
| 225 | return Result; |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 226 | } |
| 227 | /// Transparently provide more efficient getOperand methods. |
| 228 | DECLARE_TRANSPARENT_OPERAND_ACCESSORS(Value); |
| 229 | }; |
| 230 | |
| 231 | // CompareConstantExpr - This class is private to Constants.cpp, and is used |
| 232 | // behind the scenes to implement ICmp and FCmp constant expressions. This is |
| 233 | // needed in order to store the predicate value for these instructions. |
Benjamin Kramer | 079b96e | 2013-09-11 18:05:11 +0000 | [diff] [blame] | 234 | class CompareConstantExpr : public ConstantExpr { |
Craig Topper | f398d7c | 2014-03-05 06:35:38 +0000 | [diff] [blame] | 235 | void anchor() override; |
Craig Topper | a60c0f1 | 2012-09-15 17:09:36 +0000 | [diff] [blame] | 236 | void *operator new(size_t, unsigned) LLVM_DELETED_FUNCTION; |
David Blaikie | a379b181 | 2011-12-20 02:50:00 +0000 | [diff] [blame] | 237 | public: |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 238 | // allocate space for exactly two operands |
| 239 | void *operator new(size_t s) { |
| 240 | return User::operator new(s, 2); |
| 241 | } |
| 242 | unsigned short predicate; |
Chris Lattner | 229907c | 2011-07-18 04:54:35 +0000 | [diff] [blame] | 243 | CompareConstantExpr(Type *ty, Instruction::OtherOps opc, |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 244 | unsigned short pred, Constant* LHS, Constant* RHS) |
| 245 | : ConstantExpr(ty, opc, &Op<0>(), 2), predicate(pred) { |
| 246 | Op<0>() = LHS; |
| 247 | Op<1>() = RHS; |
| 248 | } |
| 249 | /// Transparently provide more efficient getOperand methods. |
| 250 | DECLARE_TRANSPARENT_OPERAND_ACCESSORS(Value); |
| 251 | }; |
| 252 | |
| 253 | template <> |
Jay Foad | c8adf5f | 2011-01-11 15:07:38 +0000 | [diff] [blame] | 254 | struct OperandTraits<UnaryConstantExpr> : |
| 255 | public FixedNumOperandTraits<UnaryConstantExpr, 1> { |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 256 | }; |
| 257 | DEFINE_TRANSPARENT_OPERAND_ACCESSORS(UnaryConstantExpr, Value) |
| 258 | |
| 259 | template <> |
Jay Foad | c8adf5f | 2011-01-11 15:07:38 +0000 | [diff] [blame] | 260 | struct OperandTraits<BinaryConstantExpr> : |
| 261 | public FixedNumOperandTraits<BinaryConstantExpr, 2> { |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 262 | }; |
| 263 | DEFINE_TRANSPARENT_OPERAND_ACCESSORS(BinaryConstantExpr, Value) |
| 264 | |
| 265 | template <> |
Jay Foad | c8adf5f | 2011-01-11 15:07:38 +0000 | [diff] [blame] | 266 | struct OperandTraits<SelectConstantExpr> : |
| 267 | public FixedNumOperandTraits<SelectConstantExpr, 3> { |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 268 | }; |
| 269 | DEFINE_TRANSPARENT_OPERAND_ACCESSORS(SelectConstantExpr, Value) |
| 270 | |
| 271 | template <> |
Jay Foad | c8adf5f | 2011-01-11 15:07:38 +0000 | [diff] [blame] | 272 | struct OperandTraits<ExtractElementConstantExpr> : |
| 273 | public FixedNumOperandTraits<ExtractElementConstantExpr, 2> { |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 274 | }; |
| 275 | DEFINE_TRANSPARENT_OPERAND_ACCESSORS(ExtractElementConstantExpr, Value) |
| 276 | |
| 277 | template <> |
Jay Foad | c8adf5f | 2011-01-11 15:07:38 +0000 | [diff] [blame] | 278 | struct OperandTraits<InsertElementConstantExpr> : |
| 279 | public FixedNumOperandTraits<InsertElementConstantExpr, 3> { |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 280 | }; |
| 281 | DEFINE_TRANSPARENT_OPERAND_ACCESSORS(InsertElementConstantExpr, Value) |
| 282 | |
| 283 | template <> |
Jay Foad | c8adf5f | 2011-01-11 15:07:38 +0000 | [diff] [blame] | 284 | struct OperandTraits<ShuffleVectorConstantExpr> : |
| 285 | public FixedNumOperandTraits<ShuffleVectorConstantExpr, 3> { |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 286 | }; |
| 287 | DEFINE_TRANSPARENT_OPERAND_ACCESSORS(ShuffleVectorConstantExpr, Value) |
| 288 | |
| 289 | template <> |
Jay Foad | c8adf5f | 2011-01-11 15:07:38 +0000 | [diff] [blame] | 290 | struct OperandTraits<ExtractValueConstantExpr> : |
| 291 | public FixedNumOperandTraits<ExtractValueConstantExpr, 1> { |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 292 | }; |
| 293 | DEFINE_TRANSPARENT_OPERAND_ACCESSORS(ExtractValueConstantExpr, Value) |
| 294 | |
| 295 | template <> |
Jay Foad | c8adf5f | 2011-01-11 15:07:38 +0000 | [diff] [blame] | 296 | struct OperandTraits<InsertValueConstantExpr> : |
| 297 | public FixedNumOperandTraits<InsertValueConstantExpr, 2> { |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 298 | }; |
| 299 | DEFINE_TRANSPARENT_OPERAND_ACCESSORS(InsertValueConstantExpr, Value) |
| 300 | |
| 301 | template <> |
Jay Foad | c8adf5f | 2011-01-11 15:07:38 +0000 | [diff] [blame] | 302 | struct OperandTraits<GetElementPtrConstantExpr> : |
| 303 | public VariadicOperandTraits<GetElementPtrConstantExpr, 1> { |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 304 | }; |
| 305 | |
| 306 | DEFINE_TRANSPARENT_OPERAND_ACCESSORS(GetElementPtrConstantExpr, Value) |
| 307 | |
| 308 | |
| 309 | template <> |
Jay Foad | c8adf5f | 2011-01-11 15:07:38 +0000 | [diff] [blame] | 310 | struct OperandTraits<CompareConstantExpr> : |
| 311 | public FixedNumOperandTraits<CompareConstantExpr, 2> { |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 312 | }; |
| 313 | DEFINE_TRANSPARENT_OPERAND_ACCESSORS(CompareConstantExpr, Value) |
| 314 | |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 315 | template <class ConstantClass> struct ConstantAggrKeyType; |
| 316 | struct InlineAsmKeyType; |
| 317 | struct ConstantExprKeyType; |
| 318 | |
| 319 | template <class ConstantClass> struct ConstantInfo; |
| 320 | template <> struct ConstantInfo<ConstantExpr> { |
| 321 | typedef ConstantExprKeyType ValType; |
| 322 | typedef Type TypeClass; |
| 323 | }; |
| 324 | template <> struct ConstantInfo<InlineAsm> { |
| 325 | typedef InlineAsmKeyType ValType; |
| 326 | typedef PointerType TypeClass; |
| 327 | }; |
| 328 | template <> struct ConstantInfo<ConstantArray> { |
| 329 | typedef ConstantAggrKeyType<ConstantArray> ValType; |
| 330 | typedef ArrayType TypeClass; |
| 331 | }; |
| 332 | template <> struct ConstantInfo<ConstantStruct> { |
| 333 | typedef ConstantAggrKeyType<ConstantStruct> ValType; |
| 334 | typedef StructType TypeClass; |
| 335 | }; |
| 336 | template <> struct ConstantInfo<ConstantVector> { |
| 337 | typedef ConstantAggrKeyType<ConstantVector> ValType; |
| 338 | typedef VectorType TypeClass; |
| 339 | }; |
| 340 | |
| 341 | template <class ConstantClass> struct ConstantAggrKeyType { |
| 342 | ArrayRef<Constant *> Operands; |
| 343 | ConstantAggrKeyType(ArrayRef<Constant *> Operands) : Operands(Operands) {} |
Duncan P. N. Exon Smith | 909620a | 2014-08-19 19:13:30 +0000 | [diff] [blame^] | 344 | ConstantAggrKeyType(ArrayRef<Constant *> Operands, const ConstantClass *) |
| 345 | : Operands(Operands) {} |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 346 | ConstantAggrKeyType(const ConstantClass *C, |
| 347 | SmallVectorImpl<Constant *> &Storage) { |
| 348 | assert(Storage.empty() && "Expected empty storage"); |
| 349 | for (unsigned I = 0, E = C->getNumOperands(); I != E; ++I) |
| 350 | Storage.push_back(C->getOperand(I)); |
| 351 | Operands = Storage; |
Duncan P. N. Exon Smith | 344ebf7 | 2014-08-19 01:02:18 +0000 | [diff] [blame] | 352 | } |
| 353 | |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 354 | bool operator==(const ConstantAggrKeyType &X) const { |
| 355 | return Operands == X.Operands; |
| 356 | } |
| 357 | bool operator==(const ConstantClass *C) const { |
| 358 | if (Operands.size() != C->getNumOperands()) |
| 359 | return false; |
| 360 | for (unsigned I = 0, E = Operands.size(); I != E; ++I) |
| 361 | if (Operands[I] != C->getOperand(I)) |
| 362 | return false; |
| 363 | return true; |
| 364 | } |
| 365 | unsigned getHash() const { |
| 366 | return hash_combine_range(Operands.begin(), Operands.end()); |
| 367 | } |
| 368 | |
| 369 | typedef typename ConstantInfo<ConstantClass>::TypeClass TypeClass; |
| 370 | ConstantClass *create(TypeClass *Ty) const { |
| 371 | return new (Operands.size()) ConstantClass(Ty, Operands); |
Duncan P. N. Exon Smith | 344ebf7 | 2014-08-19 01:02:18 +0000 | [diff] [blame] | 372 | } |
| 373 | }; |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 374 | |
Benjamin Kramer | 079b96e | 2013-09-11 18:05:11 +0000 | [diff] [blame] | 375 | struct InlineAsmKeyType { |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 376 | StringRef AsmString; |
| 377 | StringRef Constraints; |
| 378 | bool HasSideEffects; |
| 379 | bool IsAlignStack; |
| 380 | InlineAsm::AsmDialect AsmDialect; |
| 381 | |
| 382 | InlineAsmKeyType(StringRef AsmString, StringRef Constraints, |
| 383 | bool HasSideEffects, bool IsAlignStack, |
| 384 | InlineAsm::AsmDialect AsmDialect) |
| 385 | : AsmString(AsmString), Constraints(Constraints), |
| 386 | HasSideEffects(HasSideEffects), IsAlignStack(IsAlignStack), |
| 387 | AsmDialect(AsmDialect) {} |
| 388 | InlineAsmKeyType(const InlineAsm *Asm, SmallVectorImpl<Constant *> &) |
| 389 | : AsmString(Asm->getAsmString()), Constraints(Asm->getConstraintString()), |
| 390 | HasSideEffects(Asm->hasSideEffects()), |
| 391 | IsAlignStack(Asm->isAlignStack()), AsmDialect(Asm->getDialect()) {} |
| 392 | |
| 393 | bool operator==(const InlineAsmKeyType &X) const { |
| 394 | return HasSideEffects == X.HasSideEffects && |
| 395 | IsAlignStack == X.IsAlignStack && AsmDialect == X.AsmDialect && |
| 396 | AsmString == X.AsmString && Constraints == X.Constraints; |
Jeffrey Yasskin | ade270e | 2010-03-21 20:37:19 +0000 | [diff] [blame] | 397 | } |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 398 | bool operator==(const InlineAsm *Asm) const { |
| 399 | return HasSideEffects == Asm->hasSideEffects() && |
| 400 | IsAlignStack == Asm->isAlignStack() && |
| 401 | AsmDialect == Asm->getDialect() && |
| 402 | AsmString == Asm->getAsmString() && |
| 403 | Constraints == Asm->getConstraintString(); |
| 404 | } |
| 405 | unsigned getHash() const { |
| 406 | return hash_combine(AsmString, Constraints, HasSideEffects, IsAlignStack, |
| 407 | AsmDialect); |
Jeffrey Yasskin | ade270e | 2010-03-21 20:37:19 +0000 | [diff] [blame] | 408 | } |
| 409 | |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 410 | typedef ConstantInfo<InlineAsm>::TypeClass TypeClass; |
| 411 | InlineAsm *create(TypeClass *Ty) const { |
| 412 | return new InlineAsm(Ty, AsmString, Constraints, HasSideEffects, |
| 413 | IsAlignStack, AsmDialect); |
Duncan P. N. Exon Smith | 8d12558 | 2014-08-19 00:42:32 +0000 | [diff] [blame] | 414 | } |
| 415 | }; |
| 416 | |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 417 | struct ConstantExprKeyType { |
| 418 | uint8_t Opcode; |
| 419 | uint8_t SubclassOptionalData; |
| 420 | uint16_t SubclassData; |
| 421 | ArrayRef<Constant *> Ops; |
| 422 | ArrayRef<unsigned> Indexes; |
Duncan P. N. Exon Smith | 8d12558 | 2014-08-19 00:42:32 +0000 | [diff] [blame] | 423 | |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 424 | ConstantExprKeyType(unsigned Opcode, ArrayRef<Constant *> Ops, |
| 425 | unsigned short SubclassData = 0, |
| 426 | unsigned short SubclassOptionalData = 0, |
| 427 | ArrayRef<unsigned> Indexes = None) |
| 428 | : Opcode(Opcode), SubclassOptionalData(SubclassOptionalData), |
| 429 | SubclassData(SubclassData), Ops(Ops), Indexes(Indexes) {} |
Duncan P. N. Exon Smith | 909620a | 2014-08-19 19:13:30 +0000 | [diff] [blame^] | 430 | ConstantExprKeyType(ArrayRef<Constant *> Operands, const ConstantExpr *CE) |
| 431 | : Opcode(CE->getOpcode()), |
| 432 | SubclassOptionalData(CE->getRawSubclassOptionalData()), |
| 433 | SubclassData(CE->isCompare() ? CE->getPredicate() : 0), Ops(Operands), |
| 434 | Indexes(CE->hasIndices() ? CE->getIndices() : ArrayRef<unsigned>()) {} |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 435 | ConstantExprKeyType(const ConstantExpr *CE, |
| 436 | SmallVectorImpl<Constant *> &Storage) |
| 437 | : Opcode(CE->getOpcode()), |
| 438 | SubclassOptionalData(CE->getRawSubclassOptionalData()), |
| 439 | SubclassData(CE->isCompare() ? CE->getPredicate() : 0), |
| 440 | Indexes(CE->hasIndices() ? CE->getIndices() : ArrayRef<unsigned>()) { |
| 441 | assert(Storage.empty() && "Expected empty storage"); |
| 442 | for (unsigned I = 0, E = CE->getNumOperands(); I != E; ++I) |
| 443 | Storage.push_back(CE->getOperand(I)); |
| 444 | Ops = Storage; |
Duncan P. N. Exon Smith | 8d12558 | 2014-08-19 00:42:32 +0000 | [diff] [blame] | 445 | } |
| 446 | |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 447 | bool operator==(const ConstantExprKeyType &X) const { |
| 448 | return Opcode == X.Opcode && SubclassData == X.SubclassData && |
| 449 | SubclassOptionalData == X.SubclassOptionalData && Ops == X.Ops && |
| 450 | Indexes == X.Indexes; |
Duncan P. N. Exon Smith | 8d12558 | 2014-08-19 00:42:32 +0000 | [diff] [blame] | 451 | } |
| 452 | |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 453 | bool operator==(const ConstantExpr *CE) const { |
| 454 | if (Opcode != CE->getOpcode()) |
| 455 | return false; |
| 456 | if (SubclassOptionalData != CE->getRawSubclassOptionalData()) |
| 457 | return false; |
| 458 | if (Ops.size() != CE->getNumOperands()) |
| 459 | return false; |
| 460 | if (SubclassData != (CE->isCompare() ? CE->getPredicate() : 0)) |
| 461 | return false; |
| 462 | for (unsigned I = 0, E = Ops.size(); I != E; ++I) |
| 463 | if (Ops[I] != CE->getOperand(I)) |
| 464 | return false; |
| 465 | if (Indexes != (CE->hasIndices() ? CE->getIndices() : ArrayRef<unsigned>())) |
| 466 | return false; |
| 467 | return true; |
Duncan P. N. Exon Smith | 8d12558 | 2014-08-19 00:42:32 +0000 | [diff] [blame] | 468 | } |
| 469 | |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 470 | unsigned getHash() const { |
| 471 | return hash_combine(Opcode, SubclassOptionalData, SubclassData, |
| 472 | hash_combine_range(Ops.begin(), Ops.end()), |
| 473 | hash_combine_range(Indexes.begin(), Indexes.end())); |
Aaron Ballman | e4b91dc | 2014-08-19 14:59:02 +0000 | [diff] [blame] | 474 | } |
Aaron Ballman | e4b91dc | 2014-08-19 14:59:02 +0000 | [diff] [blame] | 475 | |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 476 | typedef ConstantInfo<ConstantExpr>::TypeClass TypeClass; |
| 477 | ConstantExpr *create(TypeClass *Ty) const { |
| 478 | switch (Opcode) { |
| 479 | default: |
| 480 | if (Instruction::isCast(Opcode)) |
| 481 | return new UnaryConstantExpr(Opcode, Ops[0], Ty); |
| 482 | if ((Opcode >= Instruction::BinaryOpsBegin && |
| 483 | Opcode < Instruction::BinaryOpsEnd)) |
| 484 | return new BinaryConstantExpr(Opcode, Ops[0], Ops[1], |
| 485 | SubclassOptionalData); |
| 486 | llvm_unreachable("Invalid ConstantExpr!"); |
| 487 | case Instruction::Select: |
| 488 | return new SelectConstantExpr(Ops[0], Ops[1], Ops[2]); |
| 489 | case Instruction::ExtractElement: |
| 490 | return new ExtractElementConstantExpr(Ops[0], Ops[1]); |
| 491 | case Instruction::InsertElement: |
| 492 | return new InsertElementConstantExpr(Ops[0], Ops[1], Ops[2]); |
| 493 | case Instruction::ShuffleVector: |
| 494 | return new ShuffleVectorConstantExpr(Ops[0], Ops[1], Ops[2]); |
| 495 | case Instruction::InsertValue: |
| 496 | return new InsertValueConstantExpr(Ops[0], Ops[1], Indexes, Ty); |
| 497 | case Instruction::ExtractValue: |
| 498 | return new ExtractValueConstantExpr(Ops[0], Indexes, Ty); |
| 499 | case Instruction::GetElementPtr: |
| 500 | return GetElementPtrConstantExpr::Create(Ops[0], Ops.slice(1), Ty, |
| 501 | SubclassOptionalData); |
| 502 | case Instruction::ICmp: |
| 503 | return new CompareConstantExpr(Ty, Instruction::ICmp, SubclassData, |
| 504 | Ops[0], Ops[1]); |
| 505 | case Instruction::FCmp: |
| 506 | return new CompareConstantExpr(Ty, Instruction::FCmp, SubclassData, |
| 507 | Ops[0], Ops[1]); |
Aaron Ballman | e4b91dc | 2014-08-19 14:59:02 +0000 | [diff] [blame] | 508 | } |
Aaron Ballman | e4b91dc | 2014-08-19 14:59:02 +0000 | [diff] [blame] | 509 | } |
| 510 | }; |
| 511 | |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 512 | template <class ConstantClass> class ConstantUniqueMap { |
Aaron Ballman | e4b91dc | 2014-08-19 14:59:02 +0000 | [diff] [blame] | 513 | public: |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 514 | typedef typename ConstantInfo<ConstantClass>::ValType ValType; |
| 515 | typedef typename ConstantInfo<ConstantClass>::TypeClass TypeClass; |
| 516 | typedef std::pair<TypeClass *, ValType> LookupKey; |
Aaron Ballman | e4b91dc | 2014-08-19 14:59:02 +0000 | [diff] [blame] | 517 | |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 518 | private: |
Duncan P. N. Exon Smith | 8d12558 | 2014-08-19 00:42:32 +0000 | [diff] [blame] | 519 | struct MapInfo { |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 520 | typedef DenseMapInfo<ConstantClass *> ConstantClassInfo; |
| 521 | static inline ConstantClass *getEmptyKey() { |
Duncan P. N. Exon Smith | 8d12558 | 2014-08-19 00:42:32 +0000 | [diff] [blame] | 522 | return ConstantClassInfo::getEmptyKey(); |
| 523 | } |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 524 | static inline ConstantClass *getTombstoneKey() { |
Duncan P. N. Exon Smith | 8d12558 | 2014-08-19 00:42:32 +0000 | [diff] [blame] | 525 | return ConstantClassInfo::getTombstoneKey(); |
| 526 | } |
| 527 | static unsigned getHashValue(const ConstantClass *CP) { |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 528 | SmallVector<Constant *, 8> Storage; |
| 529 | return getHashValue(LookupKey(CP->getType(), ValType(CP, Storage))); |
Duncan P. N. Exon Smith | 8d12558 | 2014-08-19 00:42:32 +0000 | [diff] [blame] | 530 | } |
| 531 | static bool isEqual(const ConstantClass *LHS, const ConstantClass *RHS) { |
| 532 | return LHS == RHS; |
| 533 | } |
| 534 | static unsigned getHashValue(const LookupKey &Val) { |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 535 | return hash_combine(Val.first, Val.second.getHash()); |
Duncan P. N. Exon Smith | 8d12558 | 2014-08-19 00:42:32 +0000 | [diff] [blame] | 536 | } |
| 537 | static bool isEqual(const LookupKey &LHS, const ConstantClass *RHS) { |
| 538 | if (RHS == getEmptyKey() || RHS == getTombstoneKey()) |
| 539 | return false; |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 540 | if (LHS.first != RHS->getType()) |
Duncan P. N. Exon Smith | 8d12558 | 2014-08-19 00:42:32 +0000 | [diff] [blame] | 541 | return false; |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 542 | return LHS.second == RHS; |
Duncan P. N. Exon Smith | 8d12558 | 2014-08-19 00:42:32 +0000 | [diff] [blame] | 543 | } |
| 544 | }; |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 545 | |
Duncan P. N. Exon Smith | 8d12558 | 2014-08-19 00:42:32 +0000 | [diff] [blame] | 546 | public: |
| 547 | typedef DenseMap<ConstantClass *, char, MapInfo> MapTy; |
| 548 | |
| 549 | private: |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 550 | MapTy Map; |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 551 | |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 552 | public: |
Devang Patel | c5aa8c6 | 2009-08-11 06:31:57 +0000 | [diff] [blame] | 553 | typename MapTy::iterator map_begin() { return Map.begin(); } |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 554 | typename MapTy::iterator map_end() { return Map.end(); } |
Torok Edwin | d18e668 | 2009-08-31 16:14:59 +0000 | [diff] [blame] | 555 | |
| 556 | void freeConstants() { |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 557 | for (auto &I : Map) |
Jeffrey Yasskin | a6eedc3 | 2010-03-22 05:23:37 +0000 | [diff] [blame] | 558 | // Asserts that use_empty(). |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 559 | delete I.first; |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 560 | } |
Dan Gohman | e4532f3 | 2009-09-15 15:58:07 +0000 | [diff] [blame] | 561 | |
Duncan P. N. Exon Smith | 8d12558 | 2014-08-19 00:42:32 +0000 | [diff] [blame] | 562 | private: |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 563 | ConstantClass *create(TypeClass *Ty, ValType V) { |
| 564 | ConstantClass *Result = V.create(Ty); |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 565 | |
| 566 | assert(Result->getType() == Ty && "Type specified is not correct!"); |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 567 | insert(Result); |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 568 | |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 569 | return Result; |
| 570 | } |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 571 | |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 572 | public: |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 573 | /// Return the specified constant from the map, creating it if necessary. |
| 574 | ConstantClass *getOrCreate(TypeClass *Ty, ValType V) { |
Aaron Ballman | e4b91dc | 2014-08-19 14:59:02 +0000 | [diff] [blame] | 575 | LookupKey Lookup(Ty, V); |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 576 | ConstantClass *Result = nullptr; |
Aaron Ballman | e4b91dc | 2014-08-19 14:59:02 +0000 | [diff] [blame] | 577 | |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 578 | auto I = find(Lookup); |
| 579 | if (I == Map.end()) |
| 580 | Result = create(Ty, V); |
| 581 | else |
Duncan P. N. Exon Smith | 8d12558 | 2014-08-19 00:42:32 +0000 | [diff] [blame] | 582 | Result = I->first; |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 583 | assert(Result && "Unexpected nullptr"); |
Duncan P. N. Exon Smith | 8d12558 | 2014-08-19 00:42:32 +0000 | [diff] [blame] | 584 | |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 585 | return Result; |
| 586 | } |
| 587 | |
Duncan P. N. Exon Smith | 8d12558 | 2014-08-19 00:42:32 +0000 | [diff] [blame] | 588 | /// Find the constant by lookup key. |
| 589 | typename MapTy::iterator find(LookupKey Lookup) { |
| 590 | return Map.find_as(Lookup); |
| 591 | } |
| 592 | |
| 593 | /// Insert the constant into its proper slot. |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 594 | void insert(ConstantClass *CP) { Map[CP] = '\0'; } |
Duncan P. N. Exon Smith | 8d12558 | 2014-08-19 00:42:32 +0000 | [diff] [blame] | 595 | |
| 596 | /// Remove this constant from the map |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 597 | void remove(ConstantClass *CP) { |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 598 | typename MapTy::iterator I = Map.find(CP); |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 599 | assert(I != Map.end() && "Constant not found in constant table!"); |
Duncan P. N. Exon Smith | 8d12558 | 2014-08-19 00:42:32 +0000 | [diff] [blame] | 600 | assert(I->first == CP && "Didn't find correct element?"); |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 601 | Map.erase(I); |
| 602 | } |
| 603 | |
Duncan P. N. Exon Smith | 909620a | 2014-08-19 19:13:30 +0000 | [diff] [blame^] | 604 | ConstantClass *replaceOperandsInPlace(ArrayRef<Constant *> Operands, |
| 605 | ConstantClass *CP, Value *From, |
| 606 | Constant *To, unsigned NumUpdated = 0, |
| 607 | unsigned OperandNo = ~0u) { |
| 608 | LookupKey Lookup(CP->getType(), ValType(Operands, CP)); |
| 609 | auto I = find(Lookup); |
| 610 | if (I != Map.end()) |
| 611 | return I->first; |
| 612 | |
| 613 | // Update to the new value. Optimize for the case when we have a single |
| 614 | // operand that we're changing, but handle bulk updates efficiently. |
| 615 | remove(CP); |
| 616 | if (NumUpdated == 1) { |
| 617 | assert(OperandNo < CP->getNumOperands() && "Invalid index"); |
| 618 | assert(CP->getOperand(OperandNo) != To && "I didn't contain From!"); |
| 619 | CP->setOperand(OperandNo, To); |
| 620 | } else { |
| 621 | for (unsigned I = 0, E = CP->getNumOperands(); I != E; ++I) |
| 622 | if (CP->getOperand(I) == From) |
| 623 | CP->setOperand(I, To); |
| 624 | } |
| 625 | insert(CP); |
| 626 | return nullptr; |
| 627 | } |
| 628 | |
Duncan P. N. Exon Smith | 317c139 | 2014-08-19 16:39:58 +0000 | [diff] [blame] | 629 | void dump() const { DEBUG(dbgs() << "Constant.cpp: ConstantUniqueMap\n"); } |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 630 | }; |
| 631 | |
Duncan P. N. Exon Smith | 25fb110 | 2014-08-19 00:21:04 +0000 | [diff] [blame] | 632 | } // end namespace llvm |
Owen Anderson | 9b67698 | 2009-08-04 22:55:26 +0000 | [diff] [blame] | 633 | |
| 634 | #endif |