//===-- TargetData.cpp - Data size & alignment routines --------------------==//
//
+// The LLVM Compiler Infrastructure
+//
+// This file was developed by the LLVM research group and is distributed under
+// the University of Illinois Open Source License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+//
// This file defines target properties related to datatype size/offset/alignment
-// information. It uses lazy annotations to cache information about how
-// structure types are laid out and used.
+// information.
//
// This structure should be created once, filled in if the defaults are not
// correct and then passed around by const&. None of the members functions
//===----------------------------------------------------------------------===//
#include "llvm/Target/TargetData.h"
+#include "llvm/Module.h"
#include "llvm/DerivedTypes.h"
#include "llvm/Constants.h"
+#include "llvm/Support/GetElementPtrTypeIterator.h"
+#include "llvm/Support/MathExtras.h"
+#include <algorithm>
+using namespace llvm;
+
+// Handle the Pass registration stuff necessary to use TargetData's.
+namespace {
+ // Register the default SparcV9 implementation...
+ RegisterPass<TargetData> X("targetdata", "Target Data Layout");
+}
static inline void getTypeInfo(const Type *Ty, const TargetData *TD,
- uint64_t &Size, unsigned char &Alignment);
+ uint64_t &Size, unsigned char &Alignment);
//===----------------------------------------------------------------------===//
-// Support for StructLayout Annotation
+// Support for StructLayout
//===----------------------------------------------------------------------===//
-StructLayout::StructLayout(const StructType *ST, const TargetData &TD)
- : Annotation(TD.getStructLayoutAID()) {
+StructLayout::StructLayout(const StructType *ST, const TargetData &TD) {
StructAlignment = 0;
StructSize = 0;
// Loop over each of the elements, placing them in memory...
- for (StructType::ElementTypes::const_iterator
- TI = ST->getElementTypes().begin(),
- TE = ST->getElementTypes().end(); TI != TE; ++TI) {
+ for (StructType::element_iterator TI = ST->element_begin(),
+ TE = ST->element_end(); TI != TE; ++TI) {
const Type *Ty = *TI;
unsigned char A;
unsigned TyAlign;
getTypeInfo(Ty, &TD, TySize, A);
TyAlign = A;
- // Add padding if neccesary to make the data element aligned properly...
+ // Add padding if necessary to make the data element aligned properly...
if (StructSize % TyAlign != 0)
StructSize = (StructSize/TyAlign + 1) * TyAlign; // Add padding...
StructSize += TySize; // Consume space for this data item
}
+ // Empty structures have alignment of 1 byte.
+ if (StructAlignment == 0) StructAlignment = 1;
+
// Add padding to the end of the struct so that it could be put in an array
// and all array elements would be aligned correctly.
if (StructSize % StructAlignment != 0)
StructSize = (StructSize/StructAlignment + 1) * StructAlignment;
-
- if (StructSize == 0) {
- StructSize = 1; // Empty struct is 1 byte
- StructAlignment = 1;
- }
}
-Annotation *TargetData::TypeAnFactory(AnnotationID AID, const Annotable *T,
- void *D) {
- const TargetData &TD = *(const TargetData*)D;
- assert(AID == TD.AID && "Target data annotation ID mismatch!");
- const Type *Ty = cast<const Type>((const Value *)T);
- assert(isa<StructType>(Ty) &&
- "Can only create StructLayout annotation on structs!");
- return new StructLayout((const StructType *)Ty, TD);
+
+/// getElementContainingOffset - Given a valid offset into the structure,
+/// return the structure index that contains it.
+unsigned StructLayout::getElementContainingOffset(uint64_t Offset) const {
+ std::vector<uint64_t>::const_iterator SI =
+ std::upper_bound(MemberOffsets.begin(), MemberOffsets.end(),
+ Offset);
+ assert(SI != MemberOffsets.begin() && "Offset not in structure type!");
+ --SI;
+ assert(*SI <= Offset && "upper_bound didn't work");
+ assert((SI == MemberOffsets.begin() || *(SI-1) < Offset) &&
+ (SI+1 == MemberOffsets.end() || *(SI+1) > Offset) &&
+ "Upper bound didn't work!");
+ return SI-MemberOffsets.begin();
}
//===----------------------------------------------------------------------===//
//===----------------------------------------------------------------------===//
TargetData::TargetData(const std::string &TargetName,
- unsigned char IntRegSize, unsigned char PtrSize,
- unsigned char PtrAl, unsigned char DoubleAl,
- unsigned char FloatAl, unsigned char LongAl,
- unsigned char IntAl, unsigned char ShortAl,
- unsigned char ByteAl)
- : AID(AnnotationManager::getID("TargetData::" + TargetName)) {
- AnnotationManager::registerAnnotationFactory(AID, TypeAnFactory, this);
-
- IntegerRegSize = IntRegSize;
+ bool isLittleEndian, unsigned char PtrSize,
+ unsigned char PtrAl, unsigned char DoubleAl,
+ unsigned char FloatAl, unsigned char LongAl,
+ unsigned char IntAl, unsigned char ShortAl,
+ unsigned char ByteAl, unsigned char BoolAl) {
+
+ // If this assert triggers, a pass "required" TargetData information, but the
+ // top level tool did not provide one for it. We do not want to default
+ // construct, or else we might end up using a bad endianness or pointer size!
+ //
+ assert(!TargetName.empty() &&
+ "ERROR: Tool did not specify a target data to use!");
+
+ LittleEndian = isLittleEndian;
PointerSize = PtrSize;
PointerAlignment = PtrAl;
DoubleAlignment = DoubleAl;
IntAlignment = IntAl;
ShortAlignment = ShortAl;
ByteAlignment = ByteAl;
+ BoolAlignment = BoolAl;
}
+TargetData::TargetData(const std::string &ToolName, const Module *M) {
+ LittleEndian = M->getEndianness() != Module::BigEndian;
+ PointerSize = M->getPointerSize() != Module::Pointer64 ? 4 : 8;
+ PointerAlignment = PointerSize;
+ DoubleAlignment = PointerSize;
+ FloatAlignment = 4;
+ LongAlignment = PointerSize;
+ IntAlignment = 4;
+ ShortAlignment = 2;
+ ByteAlignment = 1;
+ BoolAlignment = 1;
+}
+
+/// Layouts - The lazy cache of structure layout information maintained by
+/// TargetData.
+///
+static std::map<std::pair<const TargetData*,const StructType*>,
+ StructLayout> *Layouts = 0;
+
+
TargetData::~TargetData() {
- AnnotationManager::registerAnnotationFactory(AID, 0); // Deregister factory
+ if (Layouts) {
+ // Remove any layouts for this TD.
+ std::map<std::pair<const TargetData*,
+ const StructType*>, StructLayout>::iterator
+ I = Layouts->lower_bound(std::make_pair(this, (const StructType*)0));
+ while (I != Layouts->end() && I->first.first == this)
+ Layouts->erase(I++);
+ if (Layouts->empty()) {
+ delete Layouts;
+ Layouts = 0;
+ }
+ }
}
+const StructLayout *TargetData::getStructLayout(const StructType *Ty) const {
+ if (Layouts == 0)
+ Layouts = new std::map<std::pair<const TargetData*,const StructType*>,
+ StructLayout>();
+ std::map<std::pair<const TargetData*,const StructType*>,
+ StructLayout>::iterator
+ I = Layouts->lower_bound(std::make_pair(this, Ty));
+ if (I != Layouts->end() && I->first.first == this && I->first.second == Ty)
+ return &I->second;
+ else {
+ return &Layouts->insert(I, std::make_pair(std::make_pair(this, Ty),
+ StructLayout(Ty, *this)))->second;
+ }
+}
+
+/// InvalidateStructLayoutInfo - TargetData speculatively caches StructLayout
+/// objects. If a TargetData object is alive when types are being refined and
+/// removed, this method must be called whenever a StructType is removed to
+/// avoid a dangling pointer in this cache.
+void TargetData::InvalidateStructLayoutInfo(const StructType *Ty) const {
+ if (!Layouts) return; // No cache.
+
+ std::map<std::pair<const TargetData*,const StructType*>,
+ StructLayout>::iterator I = Layouts->find(std::make_pair(this, Ty));
+ if (I != Layouts->end())
+ Layouts->erase(I);
+}
+
+
+
static inline void getTypeInfo(const Type *Ty, const TargetData *TD,
- uint64_t &Size, unsigned char &Alignment) {
+ uint64_t &Size, unsigned char &Alignment) {
assert(Ty->isSized() && "Cannot getTypeInfo() on a type that is unsized!");
- switch (Ty->getPrimitiveID()) {
+ switch (Ty->getTypeID()) {
+ case Type::BoolTyID: Size = 1; Alignment = TD->getBoolAlignment(); return;
case Type::VoidTyID:
- case Type::BoolTyID:
case Type::UByteTyID:
case Type::SByteTyID: Size = 1; Alignment = TD->getByteAlignment(); return;
case Type::UShortTyID:
Size = TD->getPointerSize(); Alignment = TD->getPointerAlignment();
return;
case Type::ArrayTyID: {
- const ArrayType *ATy = (const ArrayType *)Ty;
+ const ArrayType *ATy = cast<ArrayType>(Ty);
getTypeInfo(ATy->getElementType(), TD, Size, Alignment);
- Size *= ATy->getNumElements();
+ unsigned AlignedSize = (Size + Alignment - 1)/Alignment*Alignment;
+ Size = AlignedSize*ATy->getNumElements();
+ return;
+ }
+ case Type::PackedTyID: {
+ const PackedType *PTy = cast<PackedType>(Ty);
+ getTypeInfo(PTy->getElementType(), TD, Size, Alignment);
+ unsigned AlignedSize = (Size + Alignment - 1)/Alignment*Alignment;
+ Size = AlignedSize*PTy->getNumElements();
return;
}
case Type::StructTyID: {
// Get the layout annotation... which is lazily created on demand.
- const StructLayout *Layout = TD->getStructLayout((const StructType*)Ty);
+ const StructLayout *Layout = TD->getStructLayout(cast<StructType>(Ty));
Size = Layout->StructSize; Alignment = Layout->StructAlignment;
return;
}
-
- case Type::TypeTyID:
+
default:
assert(0 && "Bad type for getTypeInfo!!!");
return;
return Align;
}
+unsigned char TargetData::getTypeAlignmentShift(const Type *Ty) const {
+ unsigned Align = getTypeAlignment(Ty);
+ assert(!(Align & (Align-1)) && "Alignment is not a power of two!");
+ return Log2_32(Align);
+}
+
+/// getIntPtrType - Return an unsigned integer type that is the same size or
+/// greater to the host pointer size.
+const Type *TargetData::getIntPtrType() const {
+ switch (getPointerSize()) {
+ default: assert(0 && "Unknown pointer size!");
+ case 2: return Type::UShortTy;
+ case 4: return Type::UIntTy;
+ case 8: return Type::ULongTy;
+ }
+}
+
+
uint64_t TargetData::getIndexedOffset(const Type *ptrTy,
- const std::vector<Value*> &Idx) const {
+ const std::vector<Value*> &Idx) const {
const Type *Ty = ptrTy;
assert(isa<PointerType>(Ty) && "Illegal argument for getIndexedOffset()");
uint64_t Result = 0;
- for (unsigned CurIDX = 0; CurIDX < Idx.size(); ++CurIDX) {
- if (Idx[CurIDX]->getType() == Type::UIntTy) {
- // Update Ty to refer to current element
- Ty = cast<SequentialType>(Ty)->getElementType();
-
- // Get the array index and the size of each array element.
- // Both must be known constants, or this will fail.
- // Also, the product needs to be sign-extended from 32 to 64 bits.
- uint64_t elementSize = this->getTypeSize(Ty);
- uint64_t arrayIdx = cast<ConstantUInt>(Idx[CurIDX])->getValue();
- Result += (uint64_t) (int) (arrayIdx * elementSize); // sign-extend
-
- } else if (const StructType *STy = dyn_cast<const StructType>(Ty)) {
- assert(Idx[CurIDX]->getType() == Type::UByteTy && "Illegal struct idx");
+ generic_gep_type_iterator<std::vector<Value*>::const_iterator>
+ TI = gep_type_begin(ptrTy, Idx.begin(), Idx.end());
+ for (unsigned CurIDX = 0; CurIDX != Idx.size(); ++CurIDX, ++TI) {
+ if (const StructType *STy = dyn_cast<StructType>(*TI)) {
+ assert(Idx[CurIDX]->getType() == Type::UIntTy && "Illegal struct idx");
unsigned FieldNo = cast<ConstantUInt>(Idx[CurIDX])->getValue();
// Get structure layout information...
Result += Layout->MemberOffsets[FieldNo];
// Update Ty to refer to current element
- Ty = STy->getElementTypes()[FieldNo];
-
- } else if (isa<const ArrayType>(Ty)) {
- assert(0 && "Loading from arrays not implemented yet!");
+ Ty = STy->getElementType(FieldNo);
} else {
- assert(0 && "Indexing type that is not struct or array?");
- return 0; // Load directly through ptr
+ // Update Ty to refer to current element
+ Ty = cast<SequentialType>(Ty)->getElementType();
+
+ // Get the array index and the size of each array element.
+ int64_t arrayIdx = cast<ConstantInt>(Idx[CurIDX])->getRawValue();
+ Result += arrayIdx * (int64_t)getTypeSize(Ty);
}
}
return Result;
}
+