blob: 7f2198dfcb269d14ccae17156fed78978ddfd5ea [file] [log] [blame]
Owen Anderson3fb4aab2009-08-23 04:24:24 +00001//===-- ConstantsContext.h - Constants-related Context Interals -----------===//
Owen Anderson9b676982009-08-04 22:55:26 +00002//
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
15#ifndef LLVM_CONSTANTSCONTEXT_H
16#define LLVM_CONSTANTSCONTEXT_H
17
Talin46e9b442012-02-05 20:54:10 +000018#include "llvm/ADT/DenseMap.h"
Jay Foadcc5fd3e2012-03-06 10:43:52 +000019#include "llvm/ADT/Hashing.h"
Jeffrey Yasskinade270e2010-03-21 20:37:19 +000020#include "llvm/InlineAsm.h"
Owen Anderson9b676982009-08-04 22:55:26 +000021#include "llvm/Instructions.h"
22#include "llvm/Operator.h"
David Greene338a9032010-01-05 01:34:26 +000023#include "llvm/Support/Debug.h"
Owen Anderson9b676982009-08-04 22:55:26 +000024#include "llvm/Support/ErrorHandling.h"
Chris Lattner34822f62009-08-23 04:44:11 +000025#include "llvm/Support/raw_ostream.h"
Owen Anderson9b676982009-08-04 22:55:26 +000026#include <map>
27
28namespace llvm {
29template<class ValType>
30struct ConstantTraits;
31
32/// UnaryConstantExpr - This class is private to Constants.cpp, and is used
33/// behind the scenes to implement unary constant exprs.
34class UnaryConstantExpr : public ConstantExpr {
David Blaikiea379b1812011-12-20 02:50:00 +000035 virtual void anchor();
Owen Anderson9b676982009-08-04 22:55:26 +000036 void *operator new(size_t, unsigned); // DO NOT IMPLEMENT
37public:
38 // allocate space for exactly one operand
39 void *operator new(size_t s) {
40 return User::operator new(s, 1);
41 }
Chris Lattner229907c2011-07-18 04:54:35 +000042 UnaryConstantExpr(unsigned Opcode, Constant *C, Type *Ty)
Owen Anderson9b676982009-08-04 22:55:26 +000043 : ConstantExpr(Ty, Opcode, &Op<0>(), 1) {
44 Op<0>() = C;
45 }
46 DECLARE_TRANSPARENT_OPERAND_ACCESSORS(Value);
47};
48
49/// BinaryConstantExpr - This class is private to Constants.cpp, and is used
50/// behind the scenes to implement binary constant exprs.
51class BinaryConstantExpr : public ConstantExpr {
David Blaikiea379b1812011-12-20 02:50:00 +000052 virtual void anchor();
Owen Anderson9b676982009-08-04 22:55:26 +000053 void *operator new(size_t, unsigned); // DO NOT IMPLEMENT
54public:
55 // allocate space for exactly two operands
56 void *operator new(size_t s) {
57 return User::operator new(s, 2);
58 }
Dan Gohman1b849082009-09-07 23:54:19 +000059 BinaryConstantExpr(unsigned Opcode, Constant *C1, Constant *C2,
60 unsigned Flags)
Owen Anderson9b676982009-08-04 22:55:26 +000061 : ConstantExpr(C1->getType(), Opcode, &Op<0>(), 2) {
62 Op<0>() = C1;
63 Op<1>() = C2;
Dan Gohman1b849082009-09-07 23:54:19 +000064 SubclassOptionalData = Flags;
Owen Anderson9b676982009-08-04 22:55:26 +000065 }
66 /// Transparently provide more efficient getOperand methods.
67 DECLARE_TRANSPARENT_OPERAND_ACCESSORS(Value);
68};
69
70/// SelectConstantExpr - This class is private to Constants.cpp, and is used
71/// behind the scenes to implement select constant exprs.
72class SelectConstantExpr : public ConstantExpr {
David Blaikiea379b1812011-12-20 02:50:00 +000073 virtual void anchor();
Owen Anderson9b676982009-08-04 22:55:26 +000074 void *operator new(size_t, unsigned); // DO NOT IMPLEMENT
75public:
76 // allocate space for exactly three operands
77 void *operator new(size_t s) {
78 return User::operator new(s, 3);
79 }
80 SelectConstantExpr(Constant *C1, Constant *C2, Constant *C3)
81 : ConstantExpr(C2->getType(), Instruction::Select, &Op<0>(), 3) {
82 Op<0>() = C1;
83 Op<1>() = C2;
84 Op<2>() = C3;
85 }
86 /// Transparently provide more efficient getOperand methods.
87 DECLARE_TRANSPARENT_OPERAND_ACCESSORS(Value);
88};
89
90/// ExtractElementConstantExpr - This class is private to
91/// Constants.cpp, and is used behind the scenes to implement
92/// extractelement constant exprs.
93class ExtractElementConstantExpr : public ConstantExpr {
David Blaikiea379b1812011-12-20 02:50:00 +000094 virtual void anchor();
Owen Anderson9b676982009-08-04 22:55:26 +000095 void *operator new(size_t, unsigned); // DO NOT IMPLEMENT
96public:
97 // allocate space for exactly two operands
98 void *operator new(size_t s) {
99 return User::operator new(s, 2);
100 }
101 ExtractElementConstantExpr(Constant *C1, Constant *C2)
102 : ConstantExpr(cast<VectorType>(C1->getType())->getElementType(),
103 Instruction::ExtractElement, &Op<0>(), 2) {
104 Op<0>() = C1;
105 Op<1>() = C2;
106 }
107 /// Transparently provide more efficient getOperand methods.
108 DECLARE_TRANSPARENT_OPERAND_ACCESSORS(Value);
109};
110
111/// InsertElementConstantExpr - This class is private to
112/// Constants.cpp, and is used behind the scenes to implement
113/// insertelement constant exprs.
114class InsertElementConstantExpr : public ConstantExpr {
David Blaikiea379b1812011-12-20 02:50:00 +0000115 virtual void anchor();
Owen Anderson9b676982009-08-04 22:55:26 +0000116 void *operator new(size_t, unsigned); // DO NOT IMPLEMENT
117public:
118 // allocate space for exactly three operands
119 void *operator new(size_t s) {
120 return User::operator new(s, 3);
121 }
122 InsertElementConstantExpr(Constant *C1, Constant *C2, Constant *C3)
123 : ConstantExpr(C1->getType(), Instruction::InsertElement,
124 &Op<0>(), 3) {
125 Op<0>() = C1;
126 Op<1>() = C2;
127 Op<2>() = C3;
128 }
129 /// Transparently provide more efficient getOperand methods.
130 DECLARE_TRANSPARENT_OPERAND_ACCESSORS(Value);
131};
132
133/// ShuffleVectorConstantExpr - This class is private to
134/// Constants.cpp, and is used behind the scenes to implement
135/// shufflevector constant exprs.
136class ShuffleVectorConstantExpr : public ConstantExpr {
David Blaikiea379b1812011-12-20 02:50:00 +0000137 virtual void anchor();
Owen Anderson9b676982009-08-04 22:55:26 +0000138 void *operator new(size_t, unsigned); // DO NOT IMPLEMENT
139public:
140 // allocate space for exactly three operands
141 void *operator new(size_t s) {
142 return User::operator new(s, 3);
143 }
144 ShuffleVectorConstantExpr(Constant *C1, Constant *C2, Constant *C3)
145 : ConstantExpr(VectorType::get(
146 cast<VectorType>(C1->getType())->getElementType(),
147 cast<VectorType>(C3->getType())->getNumElements()),
148 Instruction::ShuffleVector,
149 &Op<0>(), 3) {
150 Op<0>() = C1;
151 Op<1>() = C2;
152 Op<2>() = C3;
153 }
154 /// Transparently provide more efficient getOperand methods.
155 DECLARE_TRANSPARENT_OPERAND_ACCESSORS(Value);
156};
157
158/// ExtractValueConstantExpr - This class is private to
159/// Constants.cpp, and is used behind the scenes to implement
160/// extractvalue constant exprs.
161class ExtractValueConstantExpr : public ConstantExpr {
David Blaikiea379b1812011-12-20 02:50:00 +0000162 virtual void anchor();
Owen Anderson9b676982009-08-04 22:55:26 +0000163 void *operator new(size_t, unsigned); // DO NOT IMPLEMENT
164public:
165 // allocate space for exactly one operand
166 void *operator new(size_t s) {
167 return User::operator new(s, 1);
168 }
169 ExtractValueConstantExpr(Constant *Agg,
170 const SmallVector<unsigned, 4> &IdxList,
Chris Lattner229907c2011-07-18 04:54:35 +0000171 Type *DestTy)
Owen Anderson9b676982009-08-04 22:55:26 +0000172 : ConstantExpr(DestTy, Instruction::ExtractValue, &Op<0>(), 1),
173 Indices(IdxList) {
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.
187class InsertValueConstantExpr : public ConstantExpr {
David Blaikiea379b1812011-12-20 02:50:00 +0000188 virtual void anchor();
Owen Anderson9b676982009-08-04 22:55:26 +0000189 void *operator new(size_t, unsigned); // DO NOT IMPLEMENT
190public:
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,
196 const SmallVector<unsigned, 4> &IdxList,
Chris Lattner229907c2011-07-18 04:54:35 +0000197 Type *DestTy)
Owen Anderson9b676982009-08-04 22:55:26 +0000198 : ConstantExpr(DestTy, Instruction::InsertValue, &Op<0>(), 2),
199 Indices(IdxList) {
200 Op<0>() = Agg;
201 Op<1>() = Val;
202 }
203
204 /// Indices - These identify the position for the insertion.
205 const SmallVector<unsigned, 4> Indices;
206
207 /// Transparently provide more efficient getOperand methods.
208 DECLARE_TRANSPARENT_OPERAND_ACCESSORS(Value);
209};
210
211
212/// GetElementPtrConstantExpr - This class is private to Constants.cpp, and is
213/// used behind the scenes to implement getelementpr constant exprs.
214class GetElementPtrConstantExpr : public ConstantExpr {
David Blaikiea379b1812011-12-20 02:50:00 +0000215 virtual void anchor();
Chris Lattnera474bb22012-01-26 20:40:56 +0000216 GetElementPtrConstantExpr(Constant *C, ArrayRef<Constant*> IdxList,
Chris Lattner229907c2011-07-18 04:54:35 +0000217 Type *DestTy);
Owen Anderson9b676982009-08-04 22:55:26 +0000218public:
219 static GetElementPtrConstantExpr *Create(Constant *C,
Chris Lattnera474bb22012-01-26 20:40:56 +0000220 ArrayRef<Constant*> IdxList,
Chris Lattner229907c2011-07-18 04:54:35 +0000221 Type *DestTy,
Dan Gohman1b849082009-09-07 23:54:19 +0000222 unsigned Flags) {
223 GetElementPtrConstantExpr *Result =
Owen Anderson9b676982009-08-04 22:55:26 +0000224 new(IdxList.size() + 1) GetElementPtrConstantExpr(C, IdxList, DestTy);
Dan Gohman1b849082009-09-07 23:54:19 +0000225 Result->SubclassOptionalData = Flags;
226 return Result;
Owen Anderson9b676982009-08-04 22:55:26 +0000227 }
228 /// Transparently provide more efficient getOperand methods.
229 DECLARE_TRANSPARENT_OPERAND_ACCESSORS(Value);
230};
231
232// CompareConstantExpr - This class is private to Constants.cpp, and is used
233// behind the scenes to implement ICmp and FCmp constant expressions. This is
234// needed in order to store the predicate value for these instructions.
David Blaikiea379b1812011-12-20 02:50:00 +0000235class CompareConstantExpr : public ConstantExpr {
236 virtual void anchor();
Owen Anderson9b676982009-08-04 22:55:26 +0000237 void *operator new(size_t, unsigned); // DO NOT IMPLEMENT
David Blaikiea379b1812011-12-20 02:50:00 +0000238public:
Owen Anderson9b676982009-08-04 22:55:26 +0000239 // allocate space for exactly two operands
240 void *operator new(size_t s) {
241 return User::operator new(s, 2);
242 }
243 unsigned short predicate;
Chris Lattner229907c2011-07-18 04:54:35 +0000244 CompareConstantExpr(Type *ty, Instruction::OtherOps opc,
Owen Anderson9b676982009-08-04 22:55:26 +0000245 unsigned short pred, Constant* LHS, Constant* RHS)
246 : ConstantExpr(ty, opc, &Op<0>(), 2), predicate(pred) {
247 Op<0>() = LHS;
248 Op<1>() = RHS;
249 }
250 /// Transparently provide more efficient getOperand methods.
251 DECLARE_TRANSPARENT_OPERAND_ACCESSORS(Value);
252};
253
254template <>
Jay Foadc8adf5f2011-01-11 15:07:38 +0000255struct OperandTraits<UnaryConstantExpr> :
256 public FixedNumOperandTraits<UnaryConstantExpr, 1> {
Owen Anderson9b676982009-08-04 22:55:26 +0000257};
258DEFINE_TRANSPARENT_OPERAND_ACCESSORS(UnaryConstantExpr, Value)
259
260template <>
Jay Foadc8adf5f2011-01-11 15:07:38 +0000261struct OperandTraits<BinaryConstantExpr> :
262 public FixedNumOperandTraits<BinaryConstantExpr, 2> {
Owen Anderson9b676982009-08-04 22:55:26 +0000263};
264DEFINE_TRANSPARENT_OPERAND_ACCESSORS(BinaryConstantExpr, Value)
265
266template <>
Jay Foadc8adf5f2011-01-11 15:07:38 +0000267struct OperandTraits<SelectConstantExpr> :
268 public FixedNumOperandTraits<SelectConstantExpr, 3> {
Owen Anderson9b676982009-08-04 22:55:26 +0000269};
270DEFINE_TRANSPARENT_OPERAND_ACCESSORS(SelectConstantExpr, Value)
271
272template <>
Jay Foadc8adf5f2011-01-11 15:07:38 +0000273struct OperandTraits<ExtractElementConstantExpr> :
274 public FixedNumOperandTraits<ExtractElementConstantExpr, 2> {
Owen Anderson9b676982009-08-04 22:55:26 +0000275};
276DEFINE_TRANSPARENT_OPERAND_ACCESSORS(ExtractElementConstantExpr, Value)
277
278template <>
Jay Foadc8adf5f2011-01-11 15:07:38 +0000279struct OperandTraits<InsertElementConstantExpr> :
280 public FixedNumOperandTraits<InsertElementConstantExpr, 3> {
Owen Anderson9b676982009-08-04 22:55:26 +0000281};
282DEFINE_TRANSPARENT_OPERAND_ACCESSORS(InsertElementConstantExpr, Value)
283
284template <>
Jay Foadc8adf5f2011-01-11 15:07:38 +0000285struct OperandTraits<ShuffleVectorConstantExpr> :
286 public FixedNumOperandTraits<ShuffleVectorConstantExpr, 3> {
Owen Anderson9b676982009-08-04 22:55:26 +0000287};
288DEFINE_TRANSPARENT_OPERAND_ACCESSORS(ShuffleVectorConstantExpr, Value)
289
290template <>
Jay Foadc8adf5f2011-01-11 15:07:38 +0000291struct OperandTraits<ExtractValueConstantExpr> :
292 public FixedNumOperandTraits<ExtractValueConstantExpr, 1> {
Owen Anderson9b676982009-08-04 22:55:26 +0000293};
294DEFINE_TRANSPARENT_OPERAND_ACCESSORS(ExtractValueConstantExpr, Value)
295
296template <>
Jay Foadc8adf5f2011-01-11 15:07:38 +0000297struct OperandTraits<InsertValueConstantExpr> :
298 public FixedNumOperandTraits<InsertValueConstantExpr, 2> {
Owen Anderson9b676982009-08-04 22:55:26 +0000299};
300DEFINE_TRANSPARENT_OPERAND_ACCESSORS(InsertValueConstantExpr, Value)
301
302template <>
Jay Foadc8adf5f2011-01-11 15:07:38 +0000303struct OperandTraits<GetElementPtrConstantExpr> :
304 public VariadicOperandTraits<GetElementPtrConstantExpr, 1> {
Owen Anderson9b676982009-08-04 22:55:26 +0000305};
306
307DEFINE_TRANSPARENT_OPERAND_ACCESSORS(GetElementPtrConstantExpr, Value)
308
309
310template <>
Jay Foadc8adf5f2011-01-11 15:07:38 +0000311struct OperandTraits<CompareConstantExpr> :
312 public FixedNumOperandTraits<CompareConstantExpr, 2> {
Owen Anderson9b676982009-08-04 22:55:26 +0000313};
314DEFINE_TRANSPARENT_OPERAND_ACCESSORS(CompareConstantExpr, Value)
315
316struct ExprMapKeyType {
Owen Anderson9b676982009-08-04 22:55:26 +0000317 ExprMapKeyType(unsigned opc,
Jay Foad0091fe82011-04-13 15:22:40 +0000318 ArrayRef<Constant*> ops,
Dan Gohman1b849082009-09-07 23:54:19 +0000319 unsigned short flags = 0,
320 unsigned short optionalflags = 0,
Jay Foad0091fe82011-04-13 15:22:40 +0000321 ArrayRef<unsigned> inds = ArrayRef<unsigned>())
Dan Gohman1b849082009-09-07 23:54:19 +0000322 : opcode(opc), subclassoptionaldata(optionalflags), subclassdata(flags),
Jay Foad0091fe82011-04-13 15:22:40 +0000323 operands(ops.begin(), ops.end()), indices(inds.begin(), inds.end()) {}
Dan Gohman1b849082009-09-07 23:54:19 +0000324 uint8_t opcode;
325 uint8_t subclassoptionaldata;
326 uint16_t subclassdata;
Owen Anderson9b676982009-08-04 22:55:26 +0000327 std::vector<Constant*> operands;
Jay Foad0091fe82011-04-13 15:22:40 +0000328 SmallVector<unsigned, 4> indices;
Owen Anderson9b676982009-08-04 22:55:26 +0000329 bool operator==(const ExprMapKeyType& that) const {
330 return this->opcode == that.opcode &&
Dan Gohman1b849082009-09-07 23:54:19 +0000331 this->subclassdata == that.subclassdata &&
332 this->subclassoptionaldata == that.subclassoptionaldata &&
Owen Anderson9b676982009-08-04 22:55:26 +0000333 this->operands == that.operands &&
334 this->indices == that.indices;
335 }
336 bool operator<(const ExprMapKeyType & that) const {
Dan Gohman1b849082009-09-07 23:54:19 +0000337 if (this->opcode != that.opcode) return this->opcode < that.opcode;
338 if (this->operands != that.operands) return this->operands < that.operands;
339 if (this->subclassdata != that.subclassdata)
340 return this->subclassdata < that.subclassdata;
341 if (this->subclassoptionaldata != that.subclassoptionaldata)
342 return this->subclassoptionaldata < that.subclassoptionaldata;
343 if (this->indices != that.indices) return this->indices < that.indices;
344 return false;
Owen Anderson9b676982009-08-04 22:55:26 +0000345 }
346
347 bool operator!=(const ExprMapKeyType& that) const {
348 return !(*this == that);
349 }
350};
351
Jeffrey Yasskinade270e2010-03-21 20:37:19 +0000352struct InlineAsmKeyType {
353 InlineAsmKeyType(StringRef AsmString,
354 StringRef Constraints, bool hasSideEffects,
355 bool isAlignStack)
356 : asm_string(AsmString), constraints(Constraints),
357 has_side_effects(hasSideEffects), is_align_stack(isAlignStack) {}
358 std::string asm_string;
359 std::string constraints;
360 bool has_side_effects;
361 bool is_align_stack;
362 bool operator==(const InlineAsmKeyType& that) const {
363 return this->asm_string == that.asm_string &&
364 this->constraints == that.constraints &&
365 this->has_side_effects == that.has_side_effects &&
366 this->is_align_stack == that.is_align_stack;
367 }
368 bool operator<(const InlineAsmKeyType& that) const {
369 if (this->asm_string != that.asm_string)
370 return this->asm_string < that.asm_string;
371 if (this->constraints != that.constraints)
372 return this->constraints < that.constraints;
373 if (this->has_side_effects != that.has_side_effects)
374 return this->has_side_effects < that.has_side_effects;
375 if (this->is_align_stack != that.is_align_stack)
376 return this->is_align_stack < that.is_align_stack;
377 return false;
378 }
379
380 bool operator!=(const InlineAsmKeyType& that) const {
381 return !(*this == that);
382 }
383};
384
Owen Anderson9b676982009-08-04 22:55:26 +0000385// The number of operands for each ConstantCreator::create method is
386// determined by the ConstantTraits template.
387// ConstantCreator - A class that is used to create constants by
Jeffrey Yasskinf6ee7be2009-10-27 23:45:55 +0000388// ConstantUniqueMap*. This class should be partially specialized if there is
Owen Anderson9b676982009-08-04 22:55:26 +0000389// something strange that needs to be done to interface to the ctor for the
390// constant.
391//
392template<typename T, typename Alloc>
393struct ConstantTraits< std::vector<T, Alloc> > {
394 static unsigned uses(const std::vector<T, Alloc>& v) {
395 return v.size();
396 }
397};
398
Chris Lattner392be582010-02-12 20:49:41 +0000399template<>
400struct ConstantTraits<Constant *> {
401 static unsigned uses(Constant * const & v) {
402 return 1;
403 }
404};
405
Owen Anderson9b676982009-08-04 22:55:26 +0000406template<class ConstantClass, class TypeClass, class ValType>
407struct ConstantCreator {
Chris Lattner229907c2011-07-18 04:54:35 +0000408 static ConstantClass *create(TypeClass *Ty, const ValType &V) {
Owen Anderson9b676982009-08-04 22:55:26 +0000409 return new(ConstantTraits<ValType>::uses(V)) ConstantClass(Ty, V);
410 }
411};
412
Talin46e9b442012-02-05 20:54:10 +0000413template<class ConstantClass, class TypeClass>
414struct ConstantArrayCreator {
415 static ConstantClass *create(TypeClass *Ty, ArrayRef<Constant*> V) {
416 return new(V.size()) ConstantClass(Ty, V);
417 }
418};
419
Dan Gohmane4532f32009-09-15 15:58:07 +0000420template<class ConstantClass>
421struct ConstantKeyData {
422 typedef void ValType;
423 static ValType getValType(ConstantClass *C) {
424 llvm_unreachable("Unknown Constant type!");
Owen Anderson9b676982009-08-04 22:55:26 +0000425 }
426};
427
428template<>
429struct ConstantCreator<ConstantExpr, Type, ExprMapKeyType> {
Chris Lattner229907c2011-07-18 04:54:35 +0000430 static ConstantExpr *create(Type *Ty, const ExprMapKeyType &V,
Owen Anderson9b676982009-08-04 22:55:26 +0000431 unsigned short pred = 0) {
432 if (Instruction::isCast(V.opcode))
433 return new UnaryConstantExpr(V.opcode, V.operands[0], Ty);
434 if ((V.opcode >= Instruction::BinaryOpsBegin &&
435 V.opcode < Instruction::BinaryOpsEnd))
Dan Gohman1b849082009-09-07 23:54:19 +0000436 return new BinaryConstantExpr(V.opcode, V.operands[0], V.operands[1],
437 V.subclassoptionaldata);
Owen Anderson9b676982009-08-04 22:55:26 +0000438 if (V.opcode == Instruction::Select)
439 return new SelectConstantExpr(V.operands[0], V.operands[1],
440 V.operands[2]);
441 if (V.opcode == Instruction::ExtractElement)
442 return new ExtractElementConstantExpr(V.operands[0], V.operands[1]);
443 if (V.opcode == Instruction::InsertElement)
444 return new InsertElementConstantExpr(V.operands[0], V.operands[1],
445 V.operands[2]);
446 if (V.opcode == Instruction::ShuffleVector)
447 return new ShuffleVectorConstantExpr(V.operands[0], V.operands[1],
448 V.operands[2]);
449 if (V.opcode == Instruction::InsertValue)
450 return new InsertValueConstantExpr(V.operands[0], V.operands[1],
451 V.indices, Ty);
452 if (V.opcode == Instruction::ExtractValue)
453 return new ExtractValueConstantExpr(V.operands[0], V.indices, Ty);
454 if (V.opcode == Instruction::GetElementPtr) {
455 std::vector<Constant*> IdxList(V.operands.begin()+1, V.operands.end());
Dan Gohman1b849082009-09-07 23:54:19 +0000456 return GetElementPtrConstantExpr::Create(V.operands[0], IdxList, Ty,
457 V.subclassoptionaldata);
Owen Anderson9b676982009-08-04 22:55:26 +0000458 }
459
460 // The compare instructions are weird. We have to encode the predicate
461 // value and it is combined with the instruction opcode by multiplying
462 // the opcode by one hundred. We must decode this to get the predicate.
463 if (V.opcode == Instruction::ICmp)
Dan Gohman1b849082009-09-07 23:54:19 +0000464 return new CompareConstantExpr(Ty, Instruction::ICmp, V.subclassdata,
Owen Anderson9b676982009-08-04 22:55:26 +0000465 V.operands[0], V.operands[1]);
466 if (V.opcode == Instruction::FCmp)
Dan Gohman1b849082009-09-07 23:54:19 +0000467 return new CompareConstantExpr(Ty, Instruction::FCmp, V.subclassdata,
Owen Anderson9b676982009-08-04 22:55:26 +0000468 V.operands[0], V.operands[1]);
469 llvm_unreachable("Invalid ConstantExpr!");
Owen Anderson9b676982009-08-04 22:55:26 +0000470 }
471};
472
473template<>
Dan Gohmane4532f32009-09-15 15:58:07 +0000474struct ConstantKeyData<ConstantExpr> {
475 typedef ExprMapKeyType ValType;
476 static ValType getValType(ConstantExpr *CE) {
477 std::vector<Constant*> Operands;
478 Operands.reserve(CE->getNumOperands());
479 for (unsigned i = 0, e = CE->getNumOperands(); i != e; ++i)
480 Operands.push_back(cast<Constant>(CE->getOperand(i)));
481 return ExprMapKeyType(CE->getOpcode(), Operands,
482 CE->isCompare() ? CE->getPredicate() : 0,
483 CE->getRawSubclassOptionalData(),
484 CE->hasIndices() ?
Jay Foad0091fe82011-04-13 15:22:40 +0000485 CE->getIndices() : ArrayRef<unsigned>());
Owen Anderson9b676982009-08-04 22:55:26 +0000486 }
487};
488
Jeffrey Yasskinade270e2010-03-21 20:37:19 +0000489template<>
490struct ConstantCreator<InlineAsm, PointerType, InlineAsmKeyType> {
Chris Lattner229907c2011-07-18 04:54:35 +0000491 static InlineAsm *create(PointerType *Ty, const InlineAsmKeyType &Key) {
Jeffrey Yasskinade270e2010-03-21 20:37:19 +0000492 return new InlineAsm(Ty, Key.asm_string, Key.constraints,
493 Key.has_side_effects, Key.is_align_stack);
494 }
495};
496
497template<>
498struct ConstantKeyData<InlineAsm> {
499 typedef InlineAsmKeyType ValType;
500 static ValType getValType(InlineAsm *Asm) {
501 return InlineAsmKeyType(Asm->getAsmString(), Asm->getConstraintString(),
502 Asm->hasSideEffects(), Asm->isAlignStack());
503 }
504};
505
Jay Foadc365eea2011-06-22 08:50:06 +0000506template<class ValType, class ValRefType, class TypeClass, class ConstantClass,
Owen Anderson9b676982009-08-04 22:55:26 +0000507 bool HasLargeKey = false /*true for arrays and structs*/ >
Chris Lattnerb1ed91f2011-07-09 17:41:24 +0000508class ConstantUniqueMap {
Owen Anderson9b676982009-08-04 22:55:26 +0000509public:
Chris Lattner229907c2011-07-18 04:54:35 +0000510 typedef std::pair<TypeClass*, ValType> MapKey;
Dan Gohmane4532f32009-09-15 15:58:07 +0000511 typedef std::map<MapKey, ConstantClass *> MapTy;
512 typedef std::map<ConstantClass *, typename MapTy::iterator> InverseMapTy;
Owen Anderson9b676982009-08-04 22:55:26 +0000513private:
514 /// Map - This is the main map from the element descriptor to the Constants.
515 /// This is the primary way we avoid creating two of the same shape
516 /// constant.
517 MapTy Map;
518
519 /// InverseMap - If "HasLargeKey" is true, this contains an inverse mapping
520 /// from the constants to their element in Map. This is important for
521 /// removal of constants from the array, which would otherwise have to scan
522 /// through the map with very large keys.
523 InverseMapTy InverseMap;
524
Owen Anderson9b676982009-08-04 22:55:26 +0000525public:
Devang Patelc5aa8c62009-08-11 06:31:57 +0000526 typename MapTy::iterator map_begin() { return Map.begin(); }
Owen Anderson9b676982009-08-04 22:55:26 +0000527 typename MapTy::iterator map_end() { return Map.end(); }
Torok Edwind18e6682009-08-31 16:14:59 +0000528
529 void freeConstants() {
530 for (typename MapTy::iterator I=Map.begin(), E=Map.end();
531 I != E; ++I) {
Jeffrey Yasskina6eedc32010-03-22 05:23:37 +0000532 // Asserts that use_empty().
533 delete I->second;
Torok Edwind18e6682009-08-31 16:14:59 +0000534 }
535 }
Owen Anderson9b676982009-08-04 22:55:26 +0000536
537 /// InsertOrGetItem - Return an iterator for the specified element.
538 /// If the element exists in the map, the returned iterator points to the
539 /// entry and Exists=true. If not, the iterator points to the newly
540 /// inserted entry and returns Exists=false. Newly inserted entries have
541 /// I->second == 0, and should be filled in.
Dan Gohmane4532f32009-09-15 15:58:07 +0000542 typename MapTy::iterator InsertOrGetItem(std::pair<MapKey, ConstantClass *>
Owen Anderson9b676982009-08-04 22:55:26 +0000543 &InsertVal,
544 bool &Exists) {
545 std::pair<typename MapTy::iterator, bool> IP = Map.insert(InsertVal);
546 Exists = !IP.second;
547 return IP.first;
548 }
549
550private:
551 typename MapTy::iterator FindExistingElement(ConstantClass *CP) {
552 if (HasLargeKey) {
553 typename InverseMapTy::iterator IMI = InverseMap.find(CP);
554 assert(IMI != InverseMap.end() && IMI->second != Map.end() &&
555 IMI->second->second == CP &&
556 "InverseMap corrupt!");
557 return IMI->second;
558 }
559
560 typename MapTy::iterator I =
Chris Lattner229907c2011-07-18 04:54:35 +0000561 Map.find(MapKey(static_cast<TypeClass*>(CP->getType()),
Dan Gohmane4532f32009-09-15 15:58:07 +0000562 ConstantKeyData<ConstantClass>::getValType(CP)));
Owen Anderson9b676982009-08-04 22:55:26 +0000563 if (I == Map.end() || I->second != CP) {
564 // FIXME: This should not use a linear scan. If this gets to be a
565 // performance problem, someone should look at this.
566 for (I = Map.begin(); I != Map.end() && I->second != CP; ++I)
567 /* empty */;
568 }
569 return I;
570 }
Dan Gohmane4532f32009-09-15 15:58:07 +0000571
Chris Lattner229907c2011-07-18 04:54:35 +0000572 ConstantClass *Create(TypeClass *Ty, ValRefType V,
Owen Anderson9b676982009-08-04 22:55:26 +0000573 typename MapTy::iterator I) {
574 ConstantClass* Result =
575 ConstantCreator<ConstantClass,TypeClass,ValType>::create(Ty, V);
576
577 assert(Result->getType() == Ty && "Type specified is not correct!");
578 I = Map.insert(I, std::make_pair(MapKey(Ty, V), Result));
579
580 if (HasLargeKey) // Remember the reverse mapping if needed.
581 InverseMap.insert(std::make_pair(Result, I));
582
Owen Anderson9b676982009-08-04 22:55:26 +0000583 return Result;
584 }
585public:
586
587 /// getOrCreate - Return the specified constant from the map, creating it if
588 /// necessary.
Chris Lattner229907c2011-07-18 04:54:35 +0000589 ConstantClass *getOrCreate(TypeClass *Ty, ValRefType V) {
Owen Anderson9b676982009-08-04 22:55:26 +0000590 MapKey Lookup(Ty, V);
591 ConstantClass* Result = 0;
592
593 typename MapTy::iterator I = Map.find(Lookup);
594 // Is it in the map?
595 if (I != Map.end())
Dan Gohmane4532f32009-09-15 15:58:07 +0000596 Result = I->second;
Owen Anderson9b676982009-08-04 22:55:26 +0000597
598 if (!Result) {
599 // If no preexisting value, create one now...
600 Result = Create(Ty, V, I);
601 }
602
603 return Result;
604 }
605
606 void remove(ConstantClass *CP) {
Owen Anderson9b676982009-08-04 22:55:26 +0000607 typename MapTy::iterator I = FindExistingElement(CP);
608 assert(I != Map.end() && "Constant not found in constant table!");
609 assert(I->second == CP && "Didn't find correct element?");
610
611 if (HasLargeKey) // Remember the reverse mapping if needed.
612 InverseMap.erase(CP);
Owen Anderson9b676982009-08-04 22:55:26 +0000613
614 Map.erase(I);
615 }
616
Owen Anderson9b676982009-08-04 22:55:26 +0000617 /// MoveConstantToNewSlot - If we are about to change C to be the element
618 /// specified by I, update our internal data structures to reflect this
619 /// fact.
Owen Anderson9b676982009-08-04 22:55:26 +0000620 void MoveConstantToNewSlot(ConstantClass *C, typename MapTy::iterator I) {
621 // First, remove the old location of the specified constant in the map.
622 typename MapTy::iterator OldI = FindExistingElement(C);
623 assert(OldI != Map.end() && "Constant not found in constant table!");
624 assert(OldI->second == C && "Didn't find correct element?");
625
Chris Lattnerb1ed91f2011-07-09 17:41:24 +0000626 // Remove the old entry from the map.
Owen Anderson9b676982009-08-04 22:55:26 +0000627 Map.erase(OldI);
628
629 // Update the inverse map so that we know that this constant is now
630 // located at descriptor I.
631 if (HasLargeKey) {
632 assert(I->second == C && "Bad inversemap entry!");
633 InverseMap[C] = I;
634 }
635 }
Owen Anderson9b676982009-08-04 22:55:26 +0000636
637 void dump() const {
David Greene338a9032010-01-05 01:34:26 +0000638 DEBUG(dbgs() << "Constant.cpp: ConstantUniqueMap\n");
Owen Anderson9b676982009-08-04 22:55:26 +0000639 }
640};
641
Talin46e9b442012-02-05 20:54:10 +0000642// Unique map for aggregate constants
643template<class TypeClass, class ConstantClass>
644class ConstantAggrUniqueMap {
645public:
646 typedef ArrayRef<Constant*> Operands;
647 typedef std::pair<TypeClass*, Operands> LookupKey;
648private:
649 struct MapInfo {
650 typedef DenseMapInfo<ConstantClass*> ConstantClassInfo;
651 typedef DenseMapInfo<Constant*> ConstantInfo;
652 typedef DenseMapInfo<TypeClass*> TypeClassInfo;
653 static inline ConstantClass* getEmptyKey() {
654 return ConstantClassInfo::getEmptyKey();
655 }
656 static inline ConstantClass* getTombstoneKey() {
657 return ConstantClassInfo::getTombstoneKey();
658 }
659 static unsigned getHashValue(const ConstantClass *CP) {
Jay Foadcc5fd3e2012-03-06 10:43:52 +0000660 hash_code code = hash_value(CP->getType());
661 for (unsigned I = 0, E = CP->getNumOperands(); I < E; ++I)
662 code = hash_combine(code, hash_value(CP->getOperand(I)));
663 return code;
Talin46e9b442012-02-05 20:54:10 +0000664 }
665 static bool isEqual(const ConstantClass *LHS, const ConstantClass *RHS) {
666 return LHS == RHS;
667 }
668 static unsigned getHashValue(const LookupKey &Val) {
Jay Foadcc5fd3e2012-03-06 10:43:52 +0000669 hash_code code = hash_value(Val.first);
Talin46e9b442012-02-05 20:54:10 +0000670 for (Operands::const_iterator
Jay Foadcc5fd3e2012-03-06 10:43:52 +0000671 I = Val.second.begin(), E = Val.second.end(); I != E; ++I)
672 code = hash_combine(code, hash_value(*I));
673 return code;
Talin46e9b442012-02-05 20:54:10 +0000674 }
675 static bool isEqual(const LookupKey &LHS, const ConstantClass *RHS) {
676 if (RHS == getEmptyKey() || RHS == getTombstoneKey())
677 return false;
678 if (LHS.first != RHS->getType()
679 || LHS.second.size() != RHS->getNumOperands())
680 return false;
681 for (unsigned I = 0, E = RHS->getNumOperands(); I < E; ++I) {
682 if (LHS.second[I] != RHS->getOperand(I))
683 return false;
684 }
685 return true;
686 }
687 };
688public:
689 typedef DenseMap<ConstantClass *, char, MapInfo> MapTy;
690
691private:
692 /// Map - This is the main map from the element descriptor to the Constants.
693 /// This is the primary way we avoid creating two of the same shape
694 /// constant.
695 MapTy Map;
696
697public:
698 typename MapTy::iterator map_begin() { return Map.begin(); }
699 typename MapTy::iterator map_end() { return Map.end(); }
700
701 void freeConstants() {
702 for (typename MapTy::iterator I=Map.begin(), E=Map.end();
703 I != E; ++I) {
704 // Asserts that use_empty().
705 delete I->first;
706 }
707 }
708
709private:
710 typename MapTy::iterator findExistingElement(ConstantClass *CP) {
711 return Map.find(CP);
712 }
713
714 ConstantClass *Create(TypeClass *Ty, Operands V, typename MapTy::iterator I) {
715 ConstantClass* Result =
716 ConstantArrayCreator<ConstantClass,TypeClass>::create(Ty, V);
717
718 assert(Result->getType() == Ty && "Type specified is not correct!");
719 Map[Result] = '\0';
720
721 return Result;
722 }
723public:
724
725 /// getOrCreate - Return the specified constant from the map, creating it if
726 /// necessary.
727 ConstantClass *getOrCreate(TypeClass *Ty, Operands V) {
728 LookupKey Lookup(Ty, V);
729 ConstantClass* Result = 0;
730
731 typename MapTy::iterator I = Map.find_as(Lookup);
732 // Is it in the map?
733 if (I != Map.end())
734 Result = I->first;
735
736 if (!Result) {
737 // If no preexisting value, create one now...
738 Result = Create(Ty, V, I);
739 }
740
741 return Result;
742 }
743
744 /// Find the constant by lookup key.
745 typename MapTy::iterator find(LookupKey Lookup) {
746 return Map.find_as(Lookup);
747 }
748
749 /// Insert the constant into its proper slot.
750 void insert(ConstantClass *CP) {
751 Map[CP] = '\0';
752 }
753
754 /// Remove this constant from the map
755 void remove(ConstantClass *CP) {
756 typename MapTy::iterator I = findExistingElement(CP);
757 assert(I != Map.end() && "Constant not found in constant table!");
758 assert(I->first == CP && "Didn't find correct element?");
759 Map.erase(I);
760 }
761
762 void dump() const {
763 DEBUG(dbgs() << "Constant.cpp: ConstantUniqueMap\n");
764 }
765};
766
Owen Anderson9b676982009-08-04 22:55:26 +0000767}
768
769#endif