1 //===-- Type.cpp - Implement the Type class -------------------------------===//
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 implements the Type class for the VMCore library.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #include "LLVMContextImpl.h"
15 #include "llvm/Module.h"
16 #include <algorithm>
17 #include <cstdarg>
18 #include "llvm/ADT/SmallString.h"
19 using namespace llvm;
20
21 //===----------------------------------------------------------------------===//
22 // Type Class Implementation
23 //===----------------------------------------------------------------------===//
24
getPrimitiveType(LLVMContext & C,TypeID IDNumber)25 Type *Type::getPrimitiveType(LLVMContext &C, TypeID IDNumber) {
26 switch (IDNumber) {
27 case VoidTyID : return getVoidTy(C);
28 case HalfTyID : return getHalfTy(C);
29 case FloatTyID : return getFloatTy(C);
30 case DoubleTyID : return getDoubleTy(C);
31 case X86_FP80TyID : return getX86_FP80Ty(C);
32 case FP128TyID : return getFP128Ty(C);
33 case PPC_FP128TyID : return getPPC_FP128Ty(C);
34 case LabelTyID : return getLabelTy(C);
35 case MetadataTyID : return getMetadataTy(C);
36 case X86_MMXTyID : return getX86_MMXTy(C);
37 default:
38 return 0;
39 }
40 }
41
42 /// getScalarType - If this is a vector type, return the element type,
43 /// otherwise return this.
getScalarType()44 Type *Type::getScalarType() {
45 if (VectorType *VTy = dyn_cast<VectorType>(this))
46 return VTy->getElementType();
47 return this;
48 }
49
50 /// isIntegerTy - Return true if this is an IntegerType of the specified width.
isIntegerTy(unsigned Bitwidth) const51 bool Type::isIntegerTy(unsigned Bitwidth) const {
52 return isIntegerTy() && cast<IntegerType>(this)->getBitWidth() == Bitwidth;
53 }
54
55 /// isIntOrIntVectorTy - Return true if this is an integer type or a vector of
56 /// integer types.
57 ///
isIntOrIntVectorTy() const58 bool Type::isIntOrIntVectorTy() const {
59 if (isIntegerTy())
60 return true;
61 if (getTypeID() != Type::VectorTyID) return false;
62
63 return cast<VectorType>(this)->getElementType()->isIntegerTy();
64 }
65
66 /// isFPOrFPVectorTy - Return true if this is a FP type or a vector of FP types.
67 ///
isFPOrFPVectorTy() const68 bool Type::isFPOrFPVectorTy() const {
69 if (getTypeID() == Type::HalfTyID || getTypeID() == Type::FloatTyID ||
70 getTypeID() == Type::DoubleTyID ||
71 getTypeID() == Type::FP128TyID || getTypeID() == Type::X86_FP80TyID ||
72 getTypeID() == Type::PPC_FP128TyID)
73 return true;
74 if (getTypeID() != Type::VectorTyID) return false;
75
76 return cast<VectorType>(this)->getElementType()->isFloatingPointTy();
77 }
78
79 // canLosslesslyBitCastTo - Return true if this type can be converted to
80 // 'Ty' without any reinterpretation of bits. For example, i8* to i32*.
81 //
canLosslesslyBitCastTo(Type * Ty) const82 bool Type::canLosslesslyBitCastTo(Type *Ty) const {
83 // Identity cast means no change so return true
84 if (this == Ty)
85 return true;
86
87 // They are not convertible unless they are at least first class types
88 if (!this->isFirstClassType() || !Ty->isFirstClassType())
89 return false;
90
91 // Vector -> Vector conversions are always lossless if the two vector types
92 // have the same size, otherwise not. Also, 64-bit vector types can be
93 // converted to x86mmx.
94 if (const VectorType *thisPTy = dyn_cast<VectorType>(this)) {
95 if (const VectorType *thatPTy = dyn_cast<VectorType>(Ty))
96 return thisPTy->getBitWidth() == thatPTy->getBitWidth();
97 if (Ty->getTypeID() == Type::X86_MMXTyID &&
98 thisPTy->getBitWidth() == 64)
99 return true;
100 }
101
102 if (this->getTypeID() == Type::X86_MMXTyID)
103 if (const VectorType *thatPTy = dyn_cast<VectorType>(Ty))
104 if (thatPTy->getBitWidth() == 64)
105 return true;
106
107 // At this point we have only various mismatches of the first class types
108 // remaining and ptr->ptr. Just select the lossless conversions. Everything
109 // else is not lossless.
110 if (this->isPointerTy())
111 return Ty->isPointerTy();
112 return false; // Other types have no identity values
113 }
114
isEmptyTy() const115 bool Type::isEmptyTy() const {
116 const ArrayType *ATy = dyn_cast<ArrayType>(this);
117 if (ATy) {
118 unsigned NumElements = ATy->getNumElements();
119 return NumElements == 0 || ATy->getElementType()->isEmptyTy();
120 }
121
122 const StructType *STy = dyn_cast<StructType>(this);
123 if (STy) {
124 unsigned NumElements = STy->getNumElements();
125 for (unsigned i = 0; i < NumElements; ++i)
126 if (!STy->getElementType(i)->isEmptyTy())
127 return false;
128 return true;
129 }
130
131 return false;
132 }
133
getPrimitiveSizeInBits() const134 unsigned Type::getPrimitiveSizeInBits() const {
135 switch (getTypeID()) {
136 case Type::HalfTyID: return 16;
137 case Type::FloatTyID: return 32;
138 case Type::DoubleTyID: return 64;
139 case Type::X86_FP80TyID: return 80;
140 case Type::FP128TyID: return 128;
141 case Type::PPC_FP128TyID: return 128;
142 case Type::X86_MMXTyID: return 64;
143 case Type::IntegerTyID: return cast<IntegerType>(this)->getBitWidth();
144 case Type::VectorTyID: return cast<VectorType>(this)->getBitWidth();
145 default: return 0;
146 }
147 }
148
149 /// getScalarSizeInBits - If this is a vector type, return the
150 /// getPrimitiveSizeInBits value for the element type. Otherwise return the
151 /// getPrimitiveSizeInBits value for this type.
getScalarSizeInBits()152 unsigned Type::getScalarSizeInBits() {
153 return getScalarType()->getPrimitiveSizeInBits();
154 }
155
156 /// getFPMantissaWidth - Return the width of the mantissa of this type. This
157 /// is only valid on floating point types. If the FP type does not
158 /// have a stable mantissa (e.g. ppc long double), this method returns -1.
getFPMantissaWidth() const159 int Type::getFPMantissaWidth() const {
160 if (const VectorType *VTy = dyn_cast<VectorType>(this))
161 return VTy->getElementType()->getFPMantissaWidth();
162 assert(isFloatingPointTy() && "Not a floating point type!");
163 if (getTypeID() == HalfTyID) return 11;
164 if (getTypeID() == FloatTyID) return 24;
165 if (getTypeID() == DoubleTyID) return 53;
166 if (getTypeID() == X86_FP80TyID) return 64;
167 if (getTypeID() == FP128TyID) return 113;
168 assert(getTypeID() == PPC_FP128TyID && "unknown fp type");
169 return -1;
170 }
171
172 /// isSizedDerivedType - Derived types like structures and arrays are sized
173 /// iff all of the members of the type are sized as well. Since asking for
174 /// their size is relatively uncommon, move this operation out of line.
isSizedDerivedType() const175 bool Type::isSizedDerivedType() const {
176 if (this->isIntegerTy())
177 return true;
178
179 if (const ArrayType *ATy = dyn_cast<ArrayType>(this))
180 return ATy->getElementType()->isSized();
181
182 if (const VectorType *VTy = dyn_cast<VectorType>(this))
183 return VTy->getElementType()->isSized();
184
185 if (!this->isStructTy())
186 return false;
187
188 return cast<StructType>(this)->isSized();
189 }
190
191 //===----------------------------------------------------------------------===//
192 // Subclass Helper Methods
193 //===----------------------------------------------------------------------===//
194
getIntegerBitWidth() const195 unsigned Type::getIntegerBitWidth() const {
196 return cast<IntegerType>(this)->getBitWidth();
197 }
198
isFunctionVarArg() const199 bool Type::isFunctionVarArg() const {
200 return cast<FunctionType>(this)->isVarArg();
201 }
202
getFunctionParamType(unsigned i) const203 Type *Type::getFunctionParamType(unsigned i) const {
204 return cast<FunctionType>(this)->getParamType(i);
205 }
206
getFunctionNumParams() const207 unsigned Type::getFunctionNumParams() const {
208 return cast<FunctionType>(this)->getNumParams();
209 }
210
getStructName() const211 StringRef Type::getStructName() const {
212 return cast<StructType>(this)->getName();
213 }
214
getStructNumElements() const215 unsigned Type::getStructNumElements() const {
216 return cast<StructType>(this)->getNumElements();
217 }
218
getStructElementType(unsigned N) const219 Type *Type::getStructElementType(unsigned N) const {
220 return cast<StructType>(this)->getElementType(N);
221 }
222
223
224
getSequentialElementType() const225 Type *Type::getSequentialElementType() const {
226 return cast<SequentialType>(this)->getElementType();
227 }
228
getArrayNumElements() const229 uint64_t Type::getArrayNumElements() const {
230 return cast<ArrayType>(this)->getNumElements();
231 }
232
getVectorNumElements() const233 unsigned Type::getVectorNumElements() const {
234 return cast<VectorType>(this)->getNumElements();
235 }
236
getPointerAddressSpace() const237 unsigned Type::getPointerAddressSpace() const {
238 return cast<PointerType>(this)->getAddressSpace();
239 }
240
241
242
243
244 //===----------------------------------------------------------------------===//
245 // Primitive 'Type' data
246 //===----------------------------------------------------------------------===//
247
getVoidTy(LLVMContext & C)248 Type *Type::getVoidTy(LLVMContext &C) { return &C.pImpl->VoidTy; }
getLabelTy(LLVMContext & C)249 Type *Type::getLabelTy(LLVMContext &C) { return &C.pImpl->LabelTy; }
getHalfTy(LLVMContext & C)250 Type *Type::getHalfTy(LLVMContext &C) { return &C.pImpl->HalfTy; }
getFloatTy(LLVMContext & C)251 Type *Type::getFloatTy(LLVMContext &C) { return &C.pImpl->FloatTy; }
getDoubleTy(LLVMContext & C)252 Type *Type::getDoubleTy(LLVMContext &C) { return &C.pImpl->DoubleTy; }
getMetadataTy(LLVMContext & C)253 Type *Type::getMetadataTy(LLVMContext &C) { return &C.pImpl->MetadataTy; }
getX86_FP80Ty(LLVMContext & C)254 Type *Type::getX86_FP80Ty(LLVMContext &C) { return &C.pImpl->X86_FP80Ty; }
getFP128Ty(LLVMContext & C)255 Type *Type::getFP128Ty(LLVMContext &C) { return &C.pImpl->FP128Ty; }
getPPC_FP128Ty(LLVMContext & C)256 Type *Type::getPPC_FP128Ty(LLVMContext &C) { return &C.pImpl->PPC_FP128Ty; }
getX86_MMXTy(LLVMContext & C)257 Type *Type::getX86_MMXTy(LLVMContext &C) { return &C.pImpl->X86_MMXTy; }
258
getInt1Ty(LLVMContext & C)259 IntegerType *Type::getInt1Ty(LLVMContext &C) { return &C.pImpl->Int1Ty; }
getInt8Ty(LLVMContext & C)260 IntegerType *Type::getInt8Ty(LLVMContext &C) { return &C.pImpl->Int8Ty; }
getInt16Ty(LLVMContext & C)261 IntegerType *Type::getInt16Ty(LLVMContext &C) { return &C.pImpl->Int16Ty; }
getInt32Ty(LLVMContext & C)262 IntegerType *Type::getInt32Ty(LLVMContext &C) { return &C.pImpl->Int32Ty; }
getInt64Ty(LLVMContext & C)263 IntegerType *Type::getInt64Ty(LLVMContext &C) { return &C.pImpl->Int64Ty; }
264
getIntNTy(LLVMContext & C,unsigned N)265 IntegerType *Type::getIntNTy(LLVMContext &C, unsigned N) {
266 return IntegerType::get(C, N);
267 }
268
getHalfPtrTy(LLVMContext & C,unsigned AS)269 PointerType *Type::getHalfPtrTy(LLVMContext &C, unsigned AS) {
270 return getHalfTy(C)->getPointerTo(AS);
271 }
272
getFloatPtrTy(LLVMContext & C,unsigned AS)273 PointerType *Type::getFloatPtrTy(LLVMContext &C, unsigned AS) {
274 return getFloatTy(C)->getPointerTo(AS);
275 }
276
getDoublePtrTy(LLVMContext & C,unsigned AS)277 PointerType *Type::getDoublePtrTy(LLVMContext &C, unsigned AS) {
278 return getDoubleTy(C)->getPointerTo(AS);
279 }
280
getX86_FP80PtrTy(LLVMContext & C,unsigned AS)281 PointerType *Type::getX86_FP80PtrTy(LLVMContext &C, unsigned AS) {
282 return getX86_FP80Ty(C)->getPointerTo(AS);
283 }
284
getFP128PtrTy(LLVMContext & C,unsigned AS)285 PointerType *Type::getFP128PtrTy(LLVMContext &C, unsigned AS) {
286 return getFP128Ty(C)->getPointerTo(AS);
287 }
288
getPPC_FP128PtrTy(LLVMContext & C,unsigned AS)289 PointerType *Type::getPPC_FP128PtrTy(LLVMContext &C, unsigned AS) {
290 return getPPC_FP128Ty(C)->getPointerTo(AS);
291 }
292
getX86_MMXPtrTy(LLVMContext & C,unsigned AS)293 PointerType *Type::getX86_MMXPtrTy(LLVMContext &C, unsigned AS) {
294 return getX86_MMXTy(C)->getPointerTo(AS);
295 }
296
getIntNPtrTy(LLVMContext & C,unsigned N,unsigned AS)297 PointerType *Type::getIntNPtrTy(LLVMContext &C, unsigned N, unsigned AS) {
298 return getIntNTy(C, N)->getPointerTo(AS);
299 }
300
getInt1PtrTy(LLVMContext & C,unsigned AS)301 PointerType *Type::getInt1PtrTy(LLVMContext &C, unsigned AS) {
302 return getInt1Ty(C)->getPointerTo(AS);
303 }
304
getInt8PtrTy(LLVMContext & C,unsigned AS)305 PointerType *Type::getInt8PtrTy(LLVMContext &C, unsigned AS) {
306 return getInt8Ty(C)->getPointerTo(AS);
307 }
308
getInt16PtrTy(LLVMContext & C,unsigned AS)309 PointerType *Type::getInt16PtrTy(LLVMContext &C, unsigned AS) {
310 return getInt16Ty(C)->getPointerTo(AS);
311 }
312
getInt32PtrTy(LLVMContext & C,unsigned AS)313 PointerType *Type::getInt32PtrTy(LLVMContext &C, unsigned AS) {
314 return getInt32Ty(C)->getPointerTo(AS);
315 }
316
getInt64PtrTy(LLVMContext & C,unsigned AS)317 PointerType *Type::getInt64PtrTy(LLVMContext &C, unsigned AS) {
318 return getInt64Ty(C)->getPointerTo(AS);
319 }
320
321
322 //===----------------------------------------------------------------------===//
323 // IntegerType Implementation
324 //===----------------------------------------------------------------------===//
325
get(LLVMContext & C,unsigned NumBits)326 IntegerType *IntegerType::get(LLVMContext &C, unsigned NumBits) {
327 assert(NumBits >= MIN_INT_BITS && "bitwidth too small");
328 assert(NumBits <= MAX_INT_BITS && "bitwidth too large");
329
330 // Check for the built-in integer types
331 switch (NumBits) {
332 case 1: return cast<IntegerType>(Type::getInt1Ty(C));
333 case 8: return cast<IntegerType>(Type::getInt8Ty(C));
334 case 16: return cast<IntegerType>(Type::getInt16Ty(C));
335 case 32: return cast<IntegerType>(Type::getInt32Ty(C));
336 case 64: return cast<IntegerType>(Type::getInt64Ty(C));
337 default:
338 break;
339 }
340
341 IntegerType *&Entry = C.pImpl->IntegerTypes[NumBits];
342
343 if (Entry == 0)
344 Entry = new (C.pImpl->TypeAllocator) IntegerType(C, NumBits);
345
346 return Entry;
347 }
348
isPowerOf2ByteWidth() const349 bool IntegerType::isPowerOf2ByteWidth() const {
350 unsigned BitWidth = getBitWidth();
351 return (BitWidth > 7) && isPowerOf2_32(BitWidth);
352 }
353
getMask() const354 APInt IntegerType::getMask() const {
355 return APInt::getAllOnesValue(getBitWidth());
356 }
357
358 //===----------------------------------------------------------------------===//
359 // FunctionType Implementation
360 //===----------------------------------------------------------------------===//
361
FunctionType(Type * Result,ArrayRef<Type * > Params,bool IsVarArgs)362 FunctionType::FunctionType(Type *Result, ArrayRef<Type*> Params,
363 bool IsVarArgs)
364 : Type(Result->getContext(), FunctionTyID) {
365 Type **SubTys = reinterpret_cast<Type**>(this+1);
366 assert(isValidReturnType(Result) && "invalid return type for function");
367 setSubclassData(IsVarArgs);
368
369 SubTys[0] = const_cast<Type*>(Result);
370
371 for (unsigned i = 0, e = Params.size(); i != e; ++i) {
372 assert(isValidArgumentType(Params[i]) &&
373 "Not a valid type for function argument!");
374 SubTys[i+1] = Params[i];
375 }
376
377 ContainedTys = SubTys;
378 NumContainedTys = Params.size() + 1; // + 1 for result type
379 }
380
381 // FunctionType::get - The factory function for the FunctionType class.
get(Type * ReturnType,ArrayRef<Type * > Params,bool isVarArg)382 FunctionType *FunctionType::get(Type *ReturnType,
383 ArrayRef<Type*> Params, bool isVarArg) {
384 LLVMContextImpl *pImpl = ReturnType->getContext().pImpl;
385 FunctionTypeKeyInfo::KeyTy Key(ReturnType, Params, isVarArg);
386 LLVMContextImpl::FunctionTypeMap::iterator I =
387 pImpl->FunctionTypes.find_as(Key);
388 FunctionType *FT;
389
390 if (I == pImpl->FunctionTypes.end()) {
391 FT = (FunctionType*) pImpl->TypeAllocator.
392 Allocate(sizeof(FunctionType) + sizeof(Type*) * (Params.size() + 1),
393 AlignOf<FunctionType>::Alignment);
394 new (FT) FunctionType(ReturnType, Params, isVarArg);
395 pImpl->FunctionTypes[FT] = true;
396 } else {
397 FT = I->first;
398 }
399
400 return FT;
401 }
402
403
get(Type * Result,bool isVarArg)404 FunctionType *FunctionType::get(Type *Result, bool isVarArg) {
405 return get(Result, ArrayRef<Type *>(), isVarArg);
406 }
407
408
409 /// isValidReturnType - Return true if the specified type is valid as a return
410 /// type.
isValidReturnType(Type * RetTy)411 bool FunctionType::isValidReturnType(Type *RetTy) {
412 return !RetTy->isFunctionTy() && !RetTy->isLabelTy() &&
413 !RetTy->isMetadataTy();
414 }
415
416 /// isValidArgumentType - Return true if the specified type is valid as an
417 /// argument type.
isValidArgumentType(Type * ArgTy)418 bool FunctionType::isValidArgumentType(Type *ArgTy) {
419 return ArgTy->isFirstClassType();
420 }
421
422 //===----------------------------------------------------------------------===//
423 // StructType Implementation
424 //===----------------------------------------------------------------------===//
425
426 // Primitive Constructors.
427
get(LLVMContext & Context,ArrayRef<Type * > ETypes,bool isPacked)428 StructType *StructType::get(LLVMContext &Context, ArrayRef<Type*> ETypes,
429 bool isPacked) {
430 LLVMContextImpl *pImpl = Context.pImpl;
431 AnonStructTypeKeyInfo::KeyTy Key(ETypes, isPacked);
432 LLVMContextImpl::StructTypeMap::iterator I =
433 pImpl->AnonStructTypes.find_as(Key);
434 StructType *ST;
435
436 if (I == pImpl->AnonStructTypes.end()) {
437 // Value not found. Create a new type!
438 ST = new (Context.pImpl->TypeAllocator) StructType(Context);
439 ST->setSubclassData(SCDB_IsLiteral); // Literal struct.
440 ST->setBody(ETypes, isPacked);
441 Context.pImpl->AnonStructTypes[ST] = true;
442 } else {
443 ST = I->first;
444 }
445
446 return ST;
447 }
448
setBody(ArrayRef<Type * > Elements,bool isPacked)449 void StructType::setBody(ArrayRef<Type*> Elements, bool isPacked) {
450 assert(isOpaque() && "Struct body already set!");
451
452 setSubclassData(getSubclassData() | SCDB_HasBody);
453 if (isPacked)
454 setSubclassData(getSubclassData() | SCDB_Packed);
455
456 unsigned NumElements = Elements.size();
457 Type **Elts = getContext().pImpl->TypeAllocator.Allocate<Type*>(NumElements);
458 memcpy(Elts, Elements.data(), sizeof(Elements[0]) * NumElements);
459
460 ContainedTys = Elts;
461 NumContainedTys = NumElements;
462 }
463
setName(StringRef Name)464 void StructType::setName(StringRef Name) {
465 if (Name == getName()) return;
466
467 StringMap<StructType *> &SymbolTable = getContext().pImpl->NamedStructTypes;
468 typedef StringMap<StructType *>::MapEntryTy EntryTy;
469
470 // If this struct already had a name, remove its symbol table entry. Don't
471 // delete the data yet because it may be part of the new name.
472 if (SymbolTableEntry)
473 SymbolTable.remove((EntryTy *)SymbolTableEntry);
474
475 // If this is just removing the name, we're done.
476 if (Name.empty()) {
477 if (SymbolTableEntry) {
478 // Delete the old string data.
479 ((EntryTy *)SymbolTableEntry)->Destroy(SymbolTable.getAllocator());
480 SymbolTableEntry = 0;
481 }
482 return;
483 }
484
485 // Look up the entry for the name.
486 EntryTy *Entry = &getContext().pImpl->NamedStructTypes.GetOrCreateValue(Name);
487
488 // While we have a name collision, try a random rename.
489 if (Entry->getValue()) {
490 SmallString<64> TempStr(Name);
491 TempStr.push_back('.');
492 raw_svector_ostream TmpStream(TempStr);
493 unsigned NameSize = Name.size();
494
495 do {
496 TempStr.resize(NameSize + 1);
497 TmpStream.resync();
498 TmpStream << getContext().pImpl->NamedStructTypesUniqueID++;
499
500 Entry = &getContext().pImpl->
501 NamedStructTypes.GetOrCreateValue(TmpStream.str());
502 } while (Entry->getValue());
503 }
504
505 // Okay, we found an entry that isn't used. It's us!
506 Entry->setValue(this);
507
508 // Delete the old string data.
509 if (SymbolTableEntry)
510 ((EntryTy *)SymbolTableEntry)->Destroy(SymbolTable.getAllocator());
511 SymbolTableEntry = Entry;
512 }
513
514 //===----------------------------------------------------------------------===//
515 // StructType Helper functions.
516
create(LLVMContext & Context,StringRef Name)517 StructType *StructType::create(LLVMContext &Context, StringRef Name) {
518 StructType *ST = new (Context.pImpl->TypeAllocator) StructType(Context);
519 if (!Name.empty())
520 ST->setName(Name);
521 return ST;
522 }
523
get(LLVMContext & Context,bool isPacked)524 StructType *StructType::get(LLVMContext &Context, bool isPacked) {
525 return get(Context, llvm::ArrayRef<Type*>(), isPacked);
526 }
527
get(Type * type,...)528 StructType *StructType::get(Type *type, ...) {
529 assert(type != 0 && "Cannot create a struct type with no elements with this");
530 LLVMContext &Ctx = type->getContext();
531 va_list ap;
532 SmallVector<llvm::Type*, 8> StructFields;
533 va_start(ap, type);
534 while (type) {
535 StructFields.push_back(type);
536 type = va_arg(ap, llvm::Type*);
537 }
538 return llvm::StructType::get(Ctx, StructFields);
539 }
540
create(LLVMContext & Context,ArrayRef<Type * > Elements,StringRef Name,bool isPacked)541 StructType *StructType::create(LLVMContext &Context, ArrayRef<Type*> Elements,
542 StringRef Name, bool isPacked) {
543 StructType *ST = create(Context, Name);
544 ST->setBody(Elements, isPacked);
545 return ST;
546 }
547
create(LLVMContext & Context,ArrayRef<Type * > Elements)548 StructType *StructType::create(LLVMContext &Context, ArrayRef<Type*> Elements) {
549 return create(Context, Elements, StringRef());
550 }
551
create(LLVMContext & Context)552 StructType *StructType::create(LLVMContext &Context) {
553 return create(Context, StringRef());
554 }
555
556
create(ArrayRef<Type * > Elements,StringRef Name,bool isPacked)557 StructType *StructType::create(ArrayRef<Type*> Elements, StringRef Name,
558 bool isPacked) {
559 assert(!Elements.empty() &&
560 "This method may not be invoked with an empty list");
561 return create(Elements[0]->getContext(), Elements, Name, isPacked);
562 }
563
create(ArrayRef<Type * > Elements)564 StructType *StructType::create(ArrayRef<Type*> Elements) {
565 assert(!Elements.empty() &&
566 "This method may not be invoked with an empty list");
567 return create(Elements[0]->getContext(), Elements, StringRef());
568 }
569
create(StringRef Name,Type * type,...)570 StructType *StructType::create(StringRef Name, Type *type, ...) {
571 assert(type != 0 && "Cannot create a struct type with no elements with this");
572 LLVMContext &Ctx = type->getContext();
573 va_list ap;
574 SmallVector<llvm::Type*, 8> StructFields;
575 va_start(ap, type);
576 while (type) {
577 StructFields.push_back(type);
578 type = va_arg(ap, llvm::Type*);
579 }
580 return llvm::StructType::create(Ctx, StructFields, Name);
581 }
582
isSized() const583 bool StructType::isSized() const {
584 if ((getSubclassData() & SCDB_IsSized) != 0)
585 return true;
586 if (isOpaque())
587 return false;
588
589 // Okay, our struct is sized if all of the elements are, but if one of the
590 // elements is opaque, the struct isn't sized *yet*, but may become sized in
591 // the future, so just bail out without caching.
592 for (element_iterator I = element_begin(), E = element_end(); I != E; ++I)
593 if (!(*I)->isSized())
594 return false;
595
596 // Here we cheat a bit and cast away const-ness. The goal is to memoize when
597 // we find a sized type, as types can only move from opaque to sized, not the
598 // other way.
599 const_cast<StructType*>(this)->setSubclassData(
600 getSubclassData() | SCDB_IsSized);
601 return true;
602 }
603
getName() const604 StringRef StructType::getName() const {
605 assert(!isLiteral() && "Literal structs never have names");
606 if (SymbolTableEntry == 0) return StringRef();
607
608 return ((StringMapEntry<StructType*> *)SymbolTableEntry)->getKey();
609 }
610
setBody(Type * type,...)611 void StructType::setBody(Type *type, ...) {
612 assert(type != 0 && "Cannot create a struct type with no elements with this");
613 va_list ap;
614 SmallVector<llvm::Type*, 8> StructFields;
615 va_start(ap, type);
616 while (type) {
617 StructFields.push_back(type);
618 type = va_arg(ap, llvm::Type*);
619 }
620 setBody(StructFields);
621 }
622
isValidElementType(Type * ElemTy)623 bool StructType::isValidElementType(Type *ElemTy) {
624 return !ElemTy->isVoidTy() && !ElemTy->isLabelTy() &&
625 !ElemTy->isMetadataTy() && !ElemTy->isFunctionTy();
626 }
627
628 /// isLayoutIdentical - Return true if this is layout identical to the
629 /// specified struct.
isLayoutIdentical(StructType * Other) const630 bool StructType::isLayoutIdentical(StructType *Other) const {
631 if (this == Other) return true;
632
633 if (isPacked() != Other->isPacked() ||
634 getNumElements() != Other->getNumElements())
635 return false;
636
637 return std::equal(element_begin(), element_end(), Other->element_begin());
638 }
639
640
641 /// getTypeByName - Return the type with the specified name, or null if there
642 /// is none by that name.
getTypeByName(StringRef Name) const643 StructType *Module::getTypeByName(StringRef Name) const {
644 StringMap<StructType*>::iterator I =
645 getContext().pImpl->NamedStructTypes.find(Name);
646 if (I != getContext().pImpl->NamedStructTypes.end())
647 return I->second;
648 return 0;
649 }
650
651
652 //===----------------------------------------------------------------------===//
653 // CompositeType Implementation
654 //===----------------------------------------------------------------------===//
655
getTypeAtIndex(const Value * V)656 Type *CompositeType::getTypeAtIndex(const Value *V) {
657 if (StructType *STy = dyn_cast<StructType>(this)) {
658 unsigned Idx = (unsigned)cast<ConstantInt>(V)->getZExtValue();
659 assert(indexValid(Idx) && "Invalid structure index!");
660 return STy->getElementType(Idx);
661 }
662
663 return cast<SequentialType>(this)->getElementType();
664 }
getTypeAtIndex(unsigned Idx)665 Type *CompositeType::getTypeAtIndex(unsigned Idx) {
666 if (StructType *STy = dyn_cast<StructType>(this)) {
667 assert(indexValid(Idx) && "Invalid structure index!");
668 return STy->getElementType(Idx);
669 }
670
671 return cast<SequentialType>(this)->getElementType();
672 }
indexValid(const Value * V) const673 bool CompositeType::indexValid(const Value *V) const {
674 if (const StructType *STy = dyn_cast<StructType>(this)) {
675 // Structure indexes require 32-bit integer constants.
676 if (V->getType()->isIntegerTy(32))
677 if (const ConstantInt *CU = dyn_cast<ConstantInt>(V))
678 return CU->getZExtValue() < STy->getNumElements();
679 return false;
680 }
681
682 // Sequential types can be indexed by any integer.
683 return V->getType()->isIntegerTy();
684 }
685
indexValid(unsigned Idx) const686 bool CompositeType::indexValid(unsigned Idx) const {
687 if (const StructType *STy = dyn_cast<StructType>(this))
688 return Idx < STy->getNumElements();
689 // Sequential types can be indexed by any integer.
690 return true;
691 }
692
693
694 //===----------------------------------------------------------------------===//
695 // ArrayType Implementation
696 //===----------------------------------------------------------------------===//
697
ArrayType(Type * ElType,uint64_t NumEl)698 ArrayType::ArrayType(Type *ElType, uint64_t NumEl)
699 : SequentialType(ArrayTyID, ElType) {
700 NumElements = NumEl;
701 }
702
703
get(Type * elementType,uint64_t NumElements)704 ArrayType *ArrayType::get(Type *elementType, uint64_t NumElements) {
705 Type *ElementType = const_cast<Type*>(elementType);
706 assert(isValidElementType(ElementType) && "Invalid type for array element!");
707
708 LLVMContextImpl *pImpl = ElementType->getContext().pImpl;
709 ArrayType *&Entry =
710 pImpl->ArrayTypes[std::make_pair(ElementType, NumElements)];
711
712 if (Entry == 0)
713 Entry = new (pImpl->TypeAllocator) ArrayType(ElementType, NumElements);
714 return Entry;
715 }
716
isValidElementType(Type * ElemTy)717 bool ArrayType::isValidElementType(Type *ElemTy) {
718 return !ElemTy->isVoidTy() && !ElemTy->isLabelTy() &&
719 !ElemTy->isMetadataTy() && !ElemTy->isFunctionTy();
720 }
721
722 //===----------------------------------------------------------------------===//
723 // VectorType Implementation
724 //===----------------------------------------------------------------------===//
725
VectorType(Type * ElType,unsigned NumEl)726 VectorType::VectorType(Type *ElType, unsigned NumEl)
727 : SequentialType(VectorTyID, ElType) {
728 NumElements = NumEl;
729 }
730
get(Type * elementType,unsigned NumElements)731 VectorType *VectorType::get(Type *elementType, unsigned NumElements) {
732 Type *ElementType = const_cast<Type*>(elementType);
733 assert(NumElements > 0 && "#Elements of a VectorType must be greater than 0");
734 assert(isValidElementType(ElementType) &&
735 "Elements of a VectorType must be a primitive type");
736
737 LLVMContextImpl *pImpl = ElementType->getContext().pImpl;
738 VectorType *&Entry = ElementType->getContext().pImpl
739 ->VectorTypes[std::make_pair(ElementType, NumElements)];
740
741 if (Entry == 0)
742 Entry = new (pImpl->TypeAllocator) VectorType(ElementType, NumElements);
743 return Entry;
744 }
745
isValidElementType(Type * ElemTy)746 bool VectorType::isValidElementType(Type *ElemTy) {
747 if (PointerType *PTy = dyn_cast<PointerType>(ElemTy))
748 ElemTy = PTy->getElementType();
749 return ElemTy->isIntegerTy() || ElemTy->isFloatingPointTy();
750 }
751
752 //===----------------------------------------------------------------------===//
753 // PointerType Implementation
754 //===----------------------------------------------------------------------===//
755
get(Type * EltTy,unsigned AddressSpace)756 PointerType *PointerType::get(Type *EltTy, unsigned AddressSpace) {
757 assert(EltTy && "Can't get a pointer to <null> type!");
758 assert(isValidElementType(EltTy) && "Invalid type for pointer element!");
759
760 LLVMContextImpl *CImpl = EltTy->getContext().pImpl;
761
762 // Since AddressSpace #0 is the common case, we special case it.
763 PointerType *&Entry = AddressSpace == 0 ? CImpl->PointerTypes[EltTy]
764 : CImpl->ASPointerTypes[std::make_pair(EltTy, AddressSpace)];
765
766 if (Entry == 0)
767 Entry = new (CImpl->TypeAllocator) PointerType(EltTy, AddressSpace);
768 return Entry;
769 }
770
771
PointerType(Type * E,unsigned AddrSpace)772 PointerType::PointerType(Type *E, unsigned AddrSpace)
773 : SequentialType(PointerTyID, E) {
774 #ifndef NDEBUG
775 const unsigned oldNCT = NumContainedTys;
776 #endif
777 setSubclassData(AddrSpace);
778 // Check for miscompile. PR11652.
779 assert(oldNCT == NumContainedTys && "bitfield written out of bounds?");
780 }
781
getPointerTo(unsigned addrs)782 PointerType *Type::getPointerTo(unsigned addrs) {
783 return PointerType::get(this, addrs);
784 }
785
isValidElementType(Type * ElemTy)786 bool PointerType::isValidElementType(Type *ElemTy) {
787 return !ElemTy->isVoidTy() && !ElemTy->isLabelTy() &&
788 !ElemTy->isMetadataTy();
789 }
790