Change all #include'd files to be :: rules instead of : rules
[oota-llvm.git] / lib / Transforms / TransformInternals.cpp
index 85868320763ea5b6e85d63c8f0d6f03f89b2593a..c8e5ecc5f2b76ffaa39c2c4267d2122763988393 100644 (file)
 #include "llvm/Function.h"
 #include "llvm/iOther.h"
 
-// TargetData Hack: Eventually we will have annotations given to us by the
-// backend so that we know stuff about type size and alignments.  For now
-// though, just use this, because it happens to match the model that GCC uses.
-//
-const TargetData TD("LevelRaise: Should be GCC though!");
-
-
-static const Type *getStructOffsetStep(const StructType *STy, unsigned &Offset,
-                                       std::vector<Value*> &Indices) {
+static const Type *getStructOffsetStep(const StructType *STy, uint64_t &Offset,
+                                       std::vector<Value*> &Indices,
+                                       const TargetData &TD) {
   assert(Offset < TD.getTypeSize(STy) && "Offset not in composite!");
   const StructLayout *SL = TD.getStructLayout(STy);
 
@@ -52,17 +46,18 @@ static const Type *getStructOffsetStep(const StructType *STy, unsigned &Offset,
 //
 const Type *getStructOffsetType(const Type *Ty, unsigned &Offset,
                                 std::vector<Value*> &Indices,
-                                bool StopEarly) {
+                                const TargetData &TD, bool StopEarly) {
   if (Offset == 0 && StopEarly && !Indices.empty())
     return Ty;    // Return the leaf type
 
-  unsigned ThisOffset;
+  uint64_t ThisOffset;
   const Type *NextType;
   if (const StructType *STy = dyn_cast<StructType>(Ty)) {
     ThisOffset = Offset;
-    NextType = getStructOffsetStep(STy, ThisOffset, Indices);
+    NextType = getStructOffsetStep(STy, ThisOffset, Indices, TD);
   } else if (const ArrayType *ATy = dyn_cast<ArrayType>(Ty)) {
-    assert(Offset < TD.getTypeSize(ATy) && "Offset not in composite!");
+    assert(Offset == 0 || Offset < TD.getTypeSize(ATy) &&
+           "Offset not in composite!");
 
     NextType = ATy->getElementType();
     unsigned ChildSize = TD.getTypeSize(NextType);
@@ -75,18 +70,19 @@ const Type *getStructOffsetType(const Type *Ty, unsigned &Offset,
 
   unsigned SubOffs = Offset - ThisOffset;
   const Type *LeafTy = getStructOffsetType(NextType, SubOffs,
-                                           Indices, StopEarly);
+                                           Indices, TD, StopEarly);
   Offset = ThisOffset + SubOffs;
   return LeafTy;
 }
 
-// ConvertableToGEP - This function returns true if the specified value V is
+// ConvertibleToGEP - This function returns true if the specified value V is
 // a valid index into a pointer of type Ty.  If it is valid, Idx is filled in
 // with the values that would be appropriate to make this a getelementptr
 // instruction.  The type returned is the root type that the GEP would point to
 //
-const Type *ConvertableToGEP(const Type *Ty, Value *OffsetVal,
+const Type *ConvertibleToGEP(const Type *Ty, Value *OffsetVal,
                              std::vector<Value*> &Indices,
+                             const TargetData &TD,
                              BasicBlock::iterator *BI) {
   const CompositeType *CompTy = dyn_cast<CompositeType>(Ty);
   if (CompTy == 0) return 0;
@@ -99,8 +95,8 @@ const Type *ConvertableToGEP(const Type *Ty, Value *OffsetVal,
   // Get the offset and scale values if they exists...
   // A scale of zero with Expr.Var != 0 means a scale of 1.
   //
-  int Offset = Expr.Offset ? getConstantValue(Expr.Offset) : 0;
-  int Scale  = Expr.Scale  ? getConstantValue(Expr.Scale)  : 0;
+  int64_t Offset = Expr.Offset ? getConstantValue(Expr.Offset) : 0;
+  int64_t Scale  = Expr.Scale  ? getConstantValue(Expr.Scale)  : 0;
 
   if (Expr.Var && Scale == 0) Scale = 1;   // Scale != 0 if Expr.Var != 0
  
@@ -115,15 +111,16 @@ const Type *ConvertableToGEP(const Type *Ty, Value *OffsetVal,
 
     if (const StructType *StructTy = dyn_cast<StructType>(CompTy)) {
       // Step into the appropriate element of the structure...
-      unsigned ActualOffset = (Offset < 0) ? 0 : (unsigned)Offset;
-      NextTy = getStructOffsetStep(StructTy, ActualOffset, Indices);
+      uint64_t ActualOffset = (Offset < 0) ? 0 : (uint64_t)Offset;
+      NextTy = getStructOffsetStep(StructTy, ActualOffset, Indices, TD);
       Offset -= ActualOffset;
     } else {
       const Type *ElTy = cast<SequentialType>(CompTy)->getElementType();
-      if (!ElTy->isSized())
+      if (!ElTy->isSized() || (isa<PointerType>(CompTy) && !Indices.empty()))
         return 0; // Type is unreasonable... escape!
       unsigned ElSize = TD.getTypeSize(ElTy);
-      int ElSizeS = (int)ElSize;
+      if (ElSize == 0) return 0;   // Avoid division by zero...
+      int64_t ElSizeS = ElSize;
 
       // See if the user is indexing into a different cell of this array...
       if (Scale && (Scale >= ElSizeS || -Scale >= ElSizeS)) {
@@ -131,12 +128,12 @@ const Type *ConvertableToGEP(const Type *Ty, Value *OffsetVal,
         // array by one.  In this case, we will have to insert math to munge
         // the index.
         //
-        int ScaleAmt = Scale/ElSizeS;
+        int64_t ScaleAmt = Scale/ElSizeS;
         if (Scale-ScaleAmt*ElSizeS)
           return 0;  // Didn't scale by a multiple of element size, bail out
         Scale = 0;   // Scale is consumed
 
-        int Index = Offset/ElSize;            // is zero unless Offset > ElSize
+        int64_t Index = Offset/ElSize;        // is zero unless Offset > ElSize
         Offset -= Index*ElSize;               // Consume part of the offset
 
         if (BI) {              // Generate code?
@@ -147,15 +144,14 @@ const Type *ConvertableToGEP(const Type *Ty, Value *OffsetVal,
 
           if (ScaleAmt && ScaleAmt != 1) {
             // If we have to scale up our index, do so now
-            Value *ScaleAmtVal = ConstantSInt::get(Type::LongTy,
-                                                   (unsigned)ScaleAmt);
+            Value *ScaleAmtVal = ConstantSInt::get(Type::LongTy, ScaleAmt);
             Expr.Var = BinaryOperator::create(Instruction::Mul, Expr.Var,
                                               ScaleAmtVal,
                                               Expr.Var->getName()+"-scale",*BI);
           }
 
           if (Index) {  // Add an offset to the index
-            Value *IndexAmt = ConstantSInt::get(Type::LongTy, (unsigned)Index);
+            Value *IndexAmt = ConstantSInt::get(Type::LongTy, Index);
             Expr.Var = BinaryOperator::create(Instruction::Add, Expr.Var,
                                               IndexAmt,
                                               Expr.Var->getName()+"-offset",
@@ -165,11 +161,11 @@ const Type *ConvertableToGEP(const Type *Ty, Value *OffsetVal,
 
         Indices.push_back(Expr.Var);
         Expr.Var = 0;
-      } else if (Offset >= (int)ElSize || -Offset >= (int)ElSize) {
+      } else if (Offset >= (int64_t)ElSize || -Offset >= (int64_t)ElSize) {
         // Calculate the index that we are entering into the array cell with
-        unsigned Index = Offset/ElSize;
+        uint64_t Index = Offset/ElSize;
         Indices.push_back(ConstantSInt::get(Type::LongTy, Index));
-        Offset -= (int)(Index*ElSize);            // Consume part of the offset
+        Offset -= (int64_t)(Index*ElSize);        // Consume part of the offset
 
       } else if (isa<ArrayType>(CompTy) || Indices.empty()) {
         // Must be indexing a small amount into the first cell of the array