The TargetData is not used for the isPowerOfTwo determination. It has never
[oota-llvm.git] / include / llvm / Operator.h
index beea034bcaa4253db5c26a30192024d2fa6af968..b2bb2e6b314da7aa88878b4939f3b2fdd9313ea1 100644 (file)
 #define LLVM_OPERATOR_H
 
 #include "llvm/Constants.h"
+#include "llvm/DataLayout.h"
 #include "llvm/DerivedTypes.h"
 #include "llvm/Instruction.h"
 #include "llvm/Type.h"
+#include "llvm/Support/GetElementPtrTypeIterator.h"
 
 namespace llvm {
 
@@ -165,27 +167,12 @@ public:
 };
 
 /// Convenience struct for specifying and reasoning about fast-math flags.
-struct FastMathFlags {
-  bool UnsafeAlgebra   : 1;
-  bool NoNaNs          : 1;
-  bool NoInfs          : 1;
-  bool NoSignedZeros   : 1;
-  bool AllowReciprocal : 1;
-
-  FastMathFlags() : UnsafeAlgebra(false), NoNaNs(false), NoInfs(false),
-                    NoSignedZeros(false), AllowReciprocal(false)
-  { }
-
-  bool any() {
-    return UnsafeAlgebra || NoNaNs || NoInfs || NoSignedZeros ||
-      AllowReciprocal;
-  }
-};
-
+class FastMathFlags {
+private:
+  friend class FPMathOperator;
+  unsigned Flags;
+  FastMathFlags(unsigned F) : Flags(F) { }
 
-/// FPMathOperator - Utility class for floating point operations which can have
-/// information about relaxed accuracy requirements attached to them.
-class FPMathOperator : public Operator {
 public:
   enum {
     UnsafeAlgebra   = (1 << 0),
@@ -195,12 +182,47 @@ public:
     AllowReciprocal = (1 << 4)
   };
 
+  FastMathFlags() : Flags(0)
+  { }
+
+  /// Whether any flag is set
+  bool any() { return Flags != 0; }
+
+  /// Set all the flags to false
+  void clear() { Flags = 0; }
+
+  /// Flag queries
+  bool noNaNs()          { return 0 != (Flags & NoNaNs); }
+  bool noInfs()          { return 0 != (Flags & NoInfs); }
+  bool noSignedZeros()   { return 0 != (Flags & NoSignedZeros); }
+  bool allowReciprocal() { return 0 != (Flags & AllowReciprocal); }
+  bool unsafeAlgebra()   { return 0 != (Flags & UnsafeAlgebra); }
+
+  /// Flag setters
+  void setNoNaNs()          { Flags |= NoNaNs; }
+  void setNoInfs()          { Flags |= NoInfs; }
+  void setNoSignedZeros()   { Flags |= NoSignedZeros; }
+  void setAllowReciprocal() { Flags |= AllowReciprocal; }
+  void setUnsafeAlgebra() {
+    Flags |= UnsafeAlgebra;
+    setNoNaNs();
+    setNoInfs();
+    setNoSignedZeros();
+    setAllowReciprocal();
+  }
+};
+
+
+/// FPMathOperator - Utility class for floating point operations which can have
+/// information about relaxed accuracy requirements attached to them.
+class FPMathOperator : public Operator {
 private:
   friend class Instruction;
 
   void setHasUnsafeAlgebra(bool B) {
     SubclassOptionalData =
-      (SubclassOptionalData & ~UnsafeAlgebra) | (B * UnsafeAlgebra);
+      (SubclassOptionalData & ~FastMathFlags::UnsafeAlgebra) |
+      (B * FastMathFlags::UnsafeAlgebra);
 
     // Unsafe algebra implies all the others
     if (B) {
@@ -212,79 +234,66 @@ private:
   }
   void setHasNoNaNs(bool B) {
     SubclassOptionalData =
-      (SubclassOptionalData & ~NoNaNs) | (B * NoNaNs);
+      (SubclassOptionalData & ~FastMathFlags::NoNaNs) |
+      (B * FastMathFlags::NoNaNs);
   }
   void setHasNoInfs(bool B) {
     SubclassOptionalData =
-      (SubclassOptionalData & ~NoInfs) | (B * NoInfs);
+      (SubclassOptionalData & ~FastMathFlags::NoInfs) |
+      (B * FastMathFlags::NoInfs);
   }
   void setHasNoSignedZeros(bool B) {
     SubclassOptionalData =
-      (SubclassOptionalData & ~NoSignedZeros) | (B * NoSignedZeros);
+      (SubclassOptionalData & ~FastMathFlags::NoSignedZeros) |
+      (B * FastMathFlags::NoSignedZeros);
   }
   void setHasAllowReciprocal(bool B) {
     SubclassOptionalData =
-      (SubclassOptionalData & ~AllowReciprocal) | (B * AllowReciprocal);
+      (SubclassOptionalData & ~FastMathFlags::AllowReciprocal) |
+      (B * FastMathFlags::AllowReciprocal);
   }
 
   /// Convenience function for setting all the fast-math flags
   void setFastMathFlags(FastMathFlags FMF) {
-    if (FMF.UnsafeAlgebra) {
-      // Set all the bits to true
-      setHasUnsafeAlgebra(true);
-      return;
-    }
-
-    setHasUnsafeAlgebra(FMF.UnsafeAlgebra);
-    setHasNoNaNs(FMF.NoNaNs);
-    setHasNoInfs(FMF.NoInfs);
-    setHasNoSignedZeros(FMF.NoSignedZeros);
-    setHasAllowReciprocal(FMF.AllowReciprocal);
+    SubclassOptionalData |= FMF.Flags;
   }
 
 public:
   /// Test whether this operation is permitted to be
   /// algebraically transformed, aka the 'A' fast-math property.
   bool hasUnsafeAlgebra() const {
-    return (SubclassOptionalData & UnsafeAlgebra) != 0;
+    return (SubclassOptionalData & FastMathFlags::UnsafeAlgebra) != 0;
   }
 
   /// Test whether this operation's arguments and results are to be
   /// treated as non-NaN, aka the 'N' fast-math property.
   bool hasNoNaNs() const {
-    return (SubclassOptionalData & NoNaNs) != 0;
+    return (SubclassOptionalData & FastMathFlags::NoNaNs) != 0;
   }
 
   /// Test whether this operation's arguments and results are to be
   /// treated as NoN-Inf, aka the 'I' fast-math property.
   bool hasNoInfs() const {
-    return (SubclassOptionalData & NoInfs) != 0;
+    return (SubclassOptionalData & FastMathFlags::NoInfs) != 0;
   }
 
   /// Test whether this operation can treat the sign of zero
   /// as insignificant, aka the 'S' fast-math property.
   bool hasNoSignedZeros() const {
-    return (SubclassOptionalData & NoSignedZeros) != 0;
+    return (SubclassOptionalData & FastMathFlags::NoSignedZeros) != 0;
   }
 
   /// Test whether this operation is permitted to use
   /// reciprocal instead of division, aka the 'R' fast-math property.
   bool hasAllowReciprocal() const {
-    return (SubclassOptionalData & AllowReciprocal) != 0;
+    return (SubclassOptionalData & FastMathFlags::AllowReciprocal) != 0;
   }
 
   /// Convenience function for getting all the fast-math flags
   FastMathFlags getFastMathFlags() const {
-    FastMathFlags FMF;
-    FMF.UnsafeAlgebra   = hasUnsafeAlgebra();
-    FMF.NoNaNs          = hasNoNaNs();
-    FMF.NoInfs          = hasNoInfs();
-    FMF.NoSignedZeros   = hasNoSignedZeros();
-    FMF.AllowReciprocal = hasAllowReciprocal();
-    return FMF;
+    return FastMathFlags(SubclassOptionalData);
   }
 
-
   /// \brief Get the maximum error permitted by this operation in ULPs.  An
   /// accuracy of 0.0 means that the operation should be performed with the
   /// default precision.
@@ -423,6 +432,45 @@ public:
     }
     return true;
   }
+
+  /// \brief Accumulate the constant address offset of this GEP if possible.
+  ///
+  /// This routine accepts an APInt into which it will accumulate the constant
+  /// offset of this GEP if the GEP is in fact constant. If the GEP is not
+  /// all-constant, it returns false and the value of the offset APInt is
+  /// undefined (it is *not* preserved!). The APInt passed into this routine
+  /// must be at least as wide as the IntPtr type for the address space of
+  /// the base GEP pointer.
+  bool accumulateConstantOffset(const DataLayout &DL, APInt &Offset) const {
+    assert(Offset.getBitWidth() ==
+           DL.getPointerSizeInBits(getPointerAddressSpace()) &&
+           "The offset must have exactly as many bits as our pointer.");
+
+    for (gep_type_iterator GTI = gep_type_begin(this), GTE = gep_type_end(this);
+         GTI != GTE; ++GTI) {
+      ConstantInt *OpC = dyn_cast<ConstantInt>(GTI.getOperand());
+      if (!OpC)
+        return false;
+      if (OpC->isZero())
+        continue;
+
+      // Handle a struct index, which adds its field offset to the pointer.
+      if (StructType *STy = dyn_cast<StructType>(*GTI)) {
+        unsigned ElementIdx = OpC->getZExtValue();
+        const StructLayout *SL = DL.getStructLayout(STy);
+        Offset += APInt(Offset.getBitWidth(),
+                        SL->getElementOffset(ElementIdx));
+        continue;
+      }
+
+      // For array or vector indices, scale the index by the size of the type.
+      APInt Index = OpC->getValue().sextOrTrunc(Offset.getBitWidth());
+      Offset += Index * APInt(Offset.getBitWidth(),
+                              DL.getTypeAllocSize(GTI.getIndexedType()));
+    }
+    return true;
+  }
+
 };
 
 } // End llvm namespace