clean up some corner cases
[oota-llvm.git] / lib / CodeGen / AsmPrinter.cpp
index 7773274eed415ed2a690c9ef316a4c584f002e5d..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;
 
@@ -31,15 +32,31 @@ bool AsmPrinter::doFinalization(Module &M) {
 void AsmPrinter::setupMachineFunction(MachineFunction &MF) {
   // What's my mangled name?
   CurrentFnName = Mang->getValueName((Value*)MF.getFunction());
-
 }
 
+// emitAlignment - Emit an alignment directive to the specified power of two.
+void AsmPrinter::emitAlignment(unsigned NumBits) const {
+  if (AlignmentIsInBytes) NumBits = 1 << NumBits;
+  O << AlignDirective << NumBits << "\n";
+}
 
+/// emitZeros - Emit a block of zeros.
+///
+void AsmPrinter::emitZeros(uint64_t NumZeros) const {
+  if (NumZeros) {
+    if (ZeroDirective)
+      O << ZeroDirective << NumZeros << "\n";
+    else {
+      for (; NumZeros; --NumZeros)
+        O << Data8bitsDirective << "0\n";
+    }
+  }
+}
 
 // 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);
@@ -48,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);
       }
@@ -123,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 << "\\\"";
@@ -152,11 +179,11 @@ 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()) {
-    O << ZeroDirective << TD.getTypeSize(CV->getType()) << "\n";
+  if (CV->isNullValue() || isa<UndefValue>(CV)) {
+    emitZeros(TD.getTypeSize(CV->getType()));
     return;
   } else if (const ConstantArray *CVA = dyn_cast<ConstantArray>(CV)) {
     if (CVA->isString()) {
@@ -171,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;
@@ -186,8 +213,7 @@ void AsmPrinter::emitGlobalConstant(const Constant *CV) {
       emitGlobalConstant(field);
 
       // Insert the field padding unless it's zero bytes...
-      if (padSize)
-        O << ZeroDirective << padSize << "\n";      
+      emitZeros(padSize);
     }
     assert(sizeSoFar == cvsLayout->StructSize &&
            "Layout of constant struct may be incorrect!");
@@ -197,69 +223,77 @@ 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 (TD.isBigEndian()) {
-        O << Data32bitsDirective << unsigned(U.UVal >> 32)
-          << "\t; double most significant word " << Val << "\n";
-        O << Data32bitsDirective << unsigned(U.UVal)
-          << "\t; double least significant word " << Val << "\n";
+      if (Data64bitsDirective)
+        O << Data64bitsDirective << DoubleToBits(Val) << "\t" << CommentString
+          << " double value: " << Val << "\n";
+      else if (TD.isBigEndian()) {
+        O << Data32bitsDirective << unsigned(DoubleToBits(Val) >> 32)
+          << "\t" << CommentString << " double most significant word "
+          << Val << "\n";
+        O << Data32bitsDirective << unsigned(DoubleToBits(Val))
+          << "\t" << CommentString << " double least significant word "
+          << Val << "\n";
       } else {
-        O << Data32bitsDirective << unsigned(U.UVal)
-          << "\t; double least significant word " << Val << "\n";
-        O << Data32bitsDirective << unsigned(U.UVal >> 32)
-          << "\t; double most significant word " << Val << "\n";
+        O << Data32bitsDirective << unsigned(DoubleToBits(Val))
+          << "\t" << CommentString << " double least significant word " << Val
+          << "\n";
+        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; float " << Val << "\n";
+      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 (TD.isBigEndian()) {
+
+      if (Data64bitsDirective)
+        O << Data64bitsDirective << Val << "\n";
+      else if (TD.isBigEndian()) {
         O << Data32bitsDirective << unsigned(Val >> 32)
-          << "\t; Double-word most significant word " << Val << "\n";
+          << "\t" << CommentString << " Double-word most significant word "
+          << Val << "\n";
         O << Data32bitsDirective << unsigned(Val)
-          << "\t; Double-word least significant word " << Val << "\n";
+          << "\t" << CommentString << " Double-word least significant word "
+          << Val << "\n";
       } else {
         O << Data32bitsDirective << unsigned(Val)
-          << "\t; Double-word least significant word " << Val << "\n";
+          << "\t" << CommentString << " Double-word least significant word "
+          << Val << "\n";
         O << Data32bitsDirective << unsigned(Val >> 32)
-          << "\t; Double-word most significant word " << Val << "\n";
+          << "\t" << CommentString << " Double-word most significant word "
+          << Val << "\n";
       }
       return;
     }
   }
 
   const Type *type = CV->getType();
-  O << "\t";
   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: