More cool stuff for the dag combiner. We can now finally handle things
[oota-llvm.git] / lib / CodeGen / AsmPrinter.cpp
index 9d99bdc4cde1824453994816edc39de2347d8042..d907d67b5a2839daa9a27d554721bace32dcc6f7 100644 (file)
@@ -15,6 +15,7 @@
 #include "llvm/Constants.h"
 #include "llvm/Instruction.h"
 #include "llvm/Support/Mangler.h"
+#include "llvm/Support/MathExtras.h"
 #include "llvm/Target/TargetMachine.h"
 using namespace llvm;
 
@@ -41,17 +42,11 @@ void AsmPrinter::emitAlignment(unsigned NumBits) const {
 
 /// emitZeros - Emit a block of zeros.
 ///
-void AsmPrinter::emitZeros(unsigned NumZeros) const {
+void AsmPrinter::emitZeros(uint64_t NumZeros) const {
   if (NumZeros) {
     if (ZeroDirective)
       O << ZeroDirective << NumZeros << "\n";
     else {
-      if (NumZeros >= 8 && Data64bitsDirective) {
-        for (; NumZeros >= 8; NumZeros -= 8)
-          O << Data64bitsDirective << "0\n";
-      }
-      for (; NumZeros >= 4; NumZeros -= 4)
-        O << Data32bitsDirective << "0\n";
       for (; NumZeros; --NumZeros)
         O << Data8bitsDirective << "0\n";
     }
@@ -61,7 +56,7 @@ void AsmPrinter::emitZeros(unsigned NumZeros) const {
 // Print out the specified constant, without a storage class.  Only the
 // constants valid in constant expressions can occur here.
 void AsmPrinter::emitConstantValueOnly(const Constant *CV) {
-  if (CV->isNullValue())
+  if (CV->isNullValue() || isa<UndefValue>(CV))
     O << "0";
   else if (const ConstantBool *CB = dyn_cast<ConstantBool>(CV)) {
     assert(CB == ConstantBool::True);
@@ -70,24 +65,33 @@ void AsmPrinter::emitConstantValueOnly(const Constant *CV) {
     if (((CI->getValue() << 32) >> 32) == CI->getValue())
       O << CI->getValue();
     else
-      O << (unsigned long long)CI->getValue();
+      O << (uint64_t)CI->getValue();
   else if (const ConstantUInt *CI = dyn_cast<ConstantUInt>(CV))
     O << CI->getValue();
-  else if (isa<GlobalValue>((Value*)CV))
-    // This is a constant address for a global variable or function.  Use the
-    // name of the variable or function as the address value.
-    O << Mang->getValueName(CV);
-  else if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(CV)) {
+  else if (isa<GlobalValue>((Value*)CV)) {
+    // This is a constant address for a global variable or function. Use the
+    // name of the variable or function as the address value, possibly
+    // decorating it with GlobalVarAddrPrefix/Suffix or
+    // FunctionAddrPrefix/Suffix (these all default to "" )
+    if (isa<Function>((Value*)CV))
+      O << FunctionAddrPrefix << Mang->getValueName(CV) << FunctionAddrSuffix;
+    else
+      O << GlobalVarAddrPrefix << Mang->getValueName(CV) << GlobalVarAddrSuffix;
+  } else if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(CV)) {
     const TargetData &TD = TM.getTargetData();
     switch(CE->getOpcode()) {
     case Instruction::GetElementPtr: {
       // generate a symbolic expression for the byte address
       const Constant *ptrVal = CE->getOperand(0);
       std::vector<Value*> idxVec(CE->op_begin()+1, CE->op_end());
-      if (unsigned Offset = TD.getIndexedOffset(ptrVal->getType(), idxVec)) {
-        O << "(";
+      if (int64_t Offset = TD.getIndexedOffset(ptrVal->getType(), idxVec)) {
+        if (Offset)
+          O << "(";
         emitConstantValueOnly(ptrVal);
-        O << ") + " << Offset;
+        if (Offset > 0)
+          O << ") + " << Offset;
+        else if (Offset < 0)
+          O << ") - " << -Offset;
       } else {
         emitConstantValueOnly(ptrVal);
       }
@@ -145,7 +149,8 @@ static void printAsCString(std::ostream &O, const ConstantArray *CVA) {
 
   O << "\"";
   for (unsigned i = 0; i != CVA->getNumOperands(); ++i) {
-    unsigned char C = cast<ConstantInt>(CVA->getOperand(i))->getRawValue();
+    unsigned char C =
+        (unsigned char)cast<ConstantInt>(CVA->getOperand(i))->getRawValue();
 
     if (C == '"') {
       O << "\\\"";
@@ -174,10 +179,10 @@ static void printAsCString(std::ostream &O, const ConstantArray *CVA) {
 
 /// emitGlobalConstant - Print a general LLVM constant to the .s file.
 ///
-void AsmPrinter::emitGlobalConstant(const Constant *CV) {  
+void AsmPrinter::emitGlobalConstant(const Constant *CV) {
   const TargetData &TD = TM.getTargetData();
 
-  if (CV->isNullValue()) {
+  if (CV->isNullValue() || isa<UndefValue>(CV)) {
     emitZeros(TD.getTypeSize(CV->getType()));
     return;
   } else if (const ConstantArray *CVA = dyn_cast<ConstantArray>(CV)) {
@@ -193,13 +198,13 @@ void AsmPrinter::emitGlobalConstant(const Constant *CV) {
   } else if (const ConstantStruct *CVS = dyn_cast<ConstantStruct>(CV)) {
     // Print the fields in successive locations. Pad to align if needed!
     const StructLayout *cvsLayout = TD.getStructLayout(CVS->getType());
-    unsigned sizeSoFar = 0;
+    uint64_t sizeSoFar = 0;
     for (unsigned i = 0, e = CVS->getNumOperands(); i != e; ++i) {
       const Constant* field = CVS->getOperand(i);
 
       // Check if padding is needed and insert one or more 0s.
-      unsigned fieldSize = TD.getTypeSize(field->getType());
-      unsigned padSize = ((i == e-1? cvsLayout->StructSize
+      uint64_t fieldSize = TD.getTypeSize(field->getType());
+      uint64_t padSize = ((i == e-1? cvsLayout->StructSize
                            : cvsLayout->MemberOffsets[i+1])
                           - cvsLayout->MemberOffsets[i]) - fieldSize;
       sizeSoFar += fieldSize + padSize;
@@ -218,61 +223,49 @@ void AsmPrinter::emitGlobalConstant(const Constant *CV) {
     // precision...
     double Val = CFP->getValue();
     if (CFP->getType() == Type::DoubleTy) {
-      union DU {                            // Abide by C TBAA rules
-        double FVal;
-        uint64_t UVal;
-      } U;
-      U.FVal = Val;
-
       if (Data64bitsDirective)
-        O << Data64bitsDirective << U.UVal << "\t" << CommentChar
+        O << Data64bitsDirective << DoubleToBits(Val) << "\t" << CommentString
           << " double value: " << Val << "\n";
       else if (TD.isBigEndian()) {
-        O << Data32bitsDirective << unsigned(U.UVal >> 32)
-          << "\t" << CommentChar << " double most significant word "
+        O << Data32bitsDirective << unsigned(DoubleToBits(Val) >> 32)
+          << "\t" << CommentString << " double most significant word "
           << Val << "\n";
-        O << Data32bitsDirective << unsigned(U.UVal)
-          << "\t" << CommentChar << " double least significant word "
+        O << Data32bitsDirective << unsigned(DoubleToBits(Val))
+          << "\t" << CommentString << " double least significant word "
           << Val << "\n";
       } else {
-        O << Data32bitsDirective << unsigned(U.UVal)
-          << "\t" << CommentChar << " double least significant word " << Val
+        O << Data32bitsDirective << unsigned(DoubleToBits(Val))
+          << "\t" << CommentString << " double least significant word " << Val
           << "\n";
-        O << Data32bitsDirective << unsigned(U.UVal >> 32)
-          << "\t" << CommentChar << " double most significant word " << Val
+        O << Data32bitsDirective << unsigned(DoubleToBits(Val) >> 32)
+          << "\t" << CommentString << " double most significant word " << Val
           << "\n";
       }
       return;
     } else {
-      union FU {                            // Abide by C TBAA rules
-        float FVal;
-        int32_t UVal;
-      } U;
-      U.FVal = Val;
-      
-      O << Data32bitsDirective << U.UVal << "\t" << CommentChar
+      O << Data32bitsDirective << FloatToBits(Val) << "\t" << CommentString
         << " float " << Val << "\n";
       return;
     }
   } else if (CV->getType() == Type::ULongTy || CV->getType() == Type::LongTy) {
     if (const ConstantInt *CI = dyn_cast<ConstantInt>(CV)) {
       uint64_t Val = CI->getRawValue();
-        
+
       if (Data64bitsDirective)
         O << Data64bitsDirective << Val << "\n";
       else if (TD.isBigEndian()) {
         O << Data32bitsDirective << unsigned(Val >> 32)
-          << "\t" << CommentChar << " Double-word most significant word "
+          << "\t" << CommentString << " Double-word most significant word "
           << Val << "\n";
         O << Data32bitsDirective << unsigned(Val)
-          << "\t" << CommentChar << " Double-word least significant word "
+          << "\t" << CommentString << " Double-word least significant word "
           << Val << "\n";
       } else {
         O << Data32bitsDirective << unsigned(Val)
-          << "\t" << CommentChar << " Double-word least significant word "
+          << "\t" << CommentString << " Double-word least significant word "
           << Val << "\n";
         O << Data32bitsDirective << unsigned(Val >> 32)
-          << "\t" << CommentChar << " Double-word most significant word "
+          << "\t" << CommentString << " Double-word most significant word "
           << Val << "\n";
       }
       return;
@@ -281,19 +274,26 @@ void AsmPrinter::emitGlobalConstant(const Constant *CV) {
 
   const Type *type = CV->getType();
   switch (type->getTypeID()) {
+  case Type::BoolTyID:
   case Type::UByteTyID: case Type::SByteTyID:
     O << Data8bitsDirective;
     break;
   case Type::UShortTyID: case Type::ShortTyID:
     O << Data16bitsDirective;
     break;
-  case Type::BoolTyID: 
   case Type::PointerTyID:
+    if (TD.getPointerSize() == 8) {
+      O << Data64bitsDirective;
+      break;
+    }
+    //Fall through for pointer size == int size
   case Type::UIntTyID: case Type::IntTyID:
     O << Data32bitsDirective;
     break;
-  case Type::ULongTyID: case Type::LongTyID:    
-    assert (0 && "Should have already output double-word constant.");
+  case Type::ULongTyID: case Type::LongTyID:
+    assert(Data64bitsDirective &&"Target cannot handle 64-bit constant exprs!");
+    O << Data64bitsDirective;
+    break;
   case Type::FloatTyID: case Type::DoubleTyID:
     assert (0 && "Should have already output floating point constant.");
   default: