Use the FunctionNumber provided by the AsmPrinter class
[oota-llvm.git] / lib / Target / PowerPC / PPCAsmPrinter.cpp
index cf0d5dacd11763f6c7c0d6d0bb317a2e859a638e..4812b7e94627dcb01fa1d44d9708882617af6870 100644 (file)
@@ -44,15 +44,11 @@ namespace {
   Statistic<> EmittedInsts("asm-printer", "Number of machine instrs printed");
 
   class PPCAsmPrinter : public AsmPrinter {
-  public:
+public:
     std::set<std::string> FnStubs, GVStubs, LinkOnceStubs;
     
     PPCAsmPrinter(std::ostream &O, TargetMachine &TM)
-      : AsmPrinter(O, TM), FunctionNumber(0) {}
-
-    /// Unique incrementer for label values for referencing Global values.
-    ///
-    unsigned FunctionNumber;
+      : AsmPrinter(O, TM) {}
 
     virtual const char *getPassName() const {
       return "PowerPC Assembly Printer";
@@ -62,6 +58,8 @@ namespace {
       return static_cast<PPCTargetMachine&>(TM);
     }
 
+    void printConstantPool(MachineConstantPool *MCP);
+
     unsigned enumRegToMachineReg(unsigned enumReg) {
       switch (enumReg) {
       default: assert(0 && "Unhandled register!"); break;
@@ -161,8 +159,8 @@ namespace {
     void printPICLabel(const MachineInstr *MI, unsigned OpNo,
                        MVT::ValueType VT) {
       // FIXME: should probably be converted to cout.width and cout.fill
-      O << "\"L0000" << FunctionNumber << "$pb\"\n";
-      O << "\"L0000" << FunctionNumber << "$pb\":";
+      O << "\"L0000" << getFunctionNumber() << "$pb\"\n";
+      O << "\"L0000" << getFunctionNumber() << "$pb\":";
     }
     void printSymbolHi(const MachineInstr *MI, unsigned OpNo,
                        MVT::ValueType VT) {
@@ -172,7 +170,7 @@ namespace {
         O << "ha16(";
         printOp(MI->getOperand(OpNo));
         if (PICEnabled)
-          O << "-\"L0000" << FunctionNumber << "$pb\")";
+          O << "-\"L0000" << getFunctionNumber() << "$pb\")";
         else
           O << ')';
       }
@@ -185,7 +183,7 @@ namespace {
         O << "lo16(";
         printOp(MI->getOperand(OpNo));
         if (PICEnabled)
-          O << "-\"L0000" << FunctionNumber << "$pb\")";
+          O << "-\"L0000" << getFunctionNumber() << "$pb\")";
         else
           O << ')';
       }
@@ -197,7 +195,6 @@ namespace {
       O << (0x80 >> RegNo);
     }
 
-    virtual void printConstantPool(MachineConstantPool *MCP) = 0;
     virtual bool runOnMachineFunction(MachineFunction &F) = 0;
     virtual bool doFinalization(Module &M) = 0;
   };
@@ -221,7 +218,6 @@ namespace {
       return "Darwin PPC Assembly Printer";
     }
 
-    void printConstantPool(MachineConstantPool *MCP);
     bool runOnMachineFunction(MachineFunction &F);
     bool doInitialization(Module &M);
     bool doFinalization(Module &M);
@@ -247,7 +243,6 @@ namespace {
       return "AIX PPC Assembly Printer";
     }
 
-    void printConstantPool(MachineConstantPool *MCP);
     bool runOnMachineFunction(MachineFunction &F);
     bool doInitialization(Module &M);
     bool doFinalization(Module &M);
@@ -302,13 +297,13 @@ void PPCAsmPrinter::printOp(const MachineOperand &MO) {
 
   case MachineOperand::MO_MachineBasicBlock: {
     MachineBasicBlock *MBBOp = MO.getMachineBasicBlock();
-    O << "LBB" << FunctionNumber << "_" << MBBOp->getNumber() << "\t; "
-      << MBBOp->getBasicBlock()->getName();
+    O << PrivateGlobalPrefix << "BB" << getFunctionNumber() << "_"
+      << MBBOp->getNumber() << "\t; " << MBBOp->getBasicBlock()->getName();
     return;
   }
 
   case MachineOperand::MO_ConstantPoolIndex:
-    O << PrivateGlobalPrefix << "CPI" << FunctionNumber
+    O << PrivateGlobalPrefix << "CPI" << getFunctionNumber()
       << '_' << MO.getConstantPoolIndex();
     return;
 
@@ -378,11 +373,37 @@ void PPCAsmPrinter::printMachineInstruction(const MachineInstr *MI) {
   return;
 }
 
+/// printConstantPool - Print to the current output stream assembly
+/// representations of the constants in the constant pool MCP. This is
+/// used to print out constants which have been "spilled to memory" by
+/// the code generator.
+///
+void PPCAsmPrinter::printConstantPool(MachineConstantPool *MCP) {
+  const std::vector<Constant*> &CP = MCP->getConstants();
+  const TargetData &TD = TM.getTargetData();
+  
+  if (CP.empty()) return;
+  
+  SwitchSection(".const", 0);
+  for (unsigned i = 0, e = CP.size(); i != e; ++i) {
+    // FIXME: force doubles to be naturally aligned.  We should handle this
+    // more correctly in the future.
+    unsigned Alignment = TD.getTypeAlignmentShift(CP[i]->getType());
+    if (CP[i]->getType() == Type::DoubleTy && Alignment < 3) Alignment = 3;
+    
+    EmitAlignment(Alignment);
+    O << PrivateGlobalPrefix << "CPI" << getFunctionNumber() << '_' << i
+      << ":\t\t\t\t\t" << CommentString << *CP[i] << '\n';
+    EmitGlobalConstant(CP[i]);
+  }
+}
+
+
 /// runOnMachineFunction - This uses the printMachineInstruction()
 /// method to print assembly for each instruction.
 ///
 bool DarwinAsmPrinter::runOnMachineFunction(MachineFunction &MF) {
-  setupMachineFunction(MF);
+  SetupMachineFunction(MF);
   O << "\n\n";
 
   // Print out constants referenced by the function
@@ -391,7 +412,7 @@ bool DarwinAsmPrinter::runOnMachineFunction(MachineFunction &MF) {
   // Print out labels for the function.
   const Function *F = MF.getFunction();
   SwitchSection(".text", F);
-  emitAlignment(4, F);
+  EmitAlignment(4, F);
   if (!F->hasInternalLinkage())
     O << "\t.globl\t" << CurrentFnName << "\n";
   O << CurrentFnName << ":\n";
@@ -401,7 +422,8 @@ bool DarwinAsmPrinter::runOnMachineFunction(MachineFunction &MF) {
        I != E; ++I) {
     // Print a label for the basic block.
     if (I != MF.begin()) {
-      O << "LBB" << FunctionNumber << '_' << I->getNumber() << ":\t";
+      O << PrivateGlobalPrefix << "BB" << getFunctionNumber() << '_'
+        << I->getNumber() << ":\t";
       if (!I->getBasicBlock()->getName().empty())
         O << CommentString << " " << I->getBasicBlock()->getName();
       O << "\n";
@@ -413,36 +435,11 @@ bool DarwinAsmPrinter::runOnMachineFunction(MachineFunction &MF) {
       printMachineInstruction(II);
     }
   }
-  ++FunctionNumber;
 
   // We didn't modify anything.
   return false;
 }
 
-/// printConstantPool - Print to the current output stream assembly
-/// representations of the constants in the constant pool MCP. This is
-/// used to print out constants which have been "spilled to memory" by
-/// the code generator.
-///
-void DarwinAsmPrinter::printConstantPool(MachineConstantPool *MCP) {
-  const std::vector<Constant*> &CP = MCP->getConstants();
-  const TargetData &TD = TM.getTargetData();
-
-  if (CP.empty()) return;
-
-  SwitchSection(".const", 0);
-  for (unsigned i = 0, e = CP.size(); i != e; ++i) {
-    // FIXME: force doubles to be naturally aligned.  We should handle this
-    // more correctly in the future.
-    if (CP[i]->getType() == Type::DoubleTy)
-      emitAlignment(3);
-    else
-      emitAlignment(TD.getTypeAlignmentShift(CP[i]->getType()));
-    O << PrivateGlobalPrefix << "CPI" << FunctionNumber << '_' << i
-      << ":\t\t\t\t\t" << CommentString << *CP[i] << '\n';
-    emitGlobalConstant(CP[i]);
-  }
-}
 
 bool DarwinAsmPrinter::doInitialization(Module &M) {
   if (TM.getSubtarget<PPCSubtarget>().isGigaProcessor())
@@ -506,9 +503,9 @@ bool DarwinAsmPrinter::doFinalization(Module &M) {
           abort();
         }
 
-        emitAlignment(Align, I);
+        EmitAlignment(Align, I);
         O << name << ":\t\t\t\t; '" << I->getName() << "'\n";
-        emitGlobalConstant(C);
+        EmitGlobalConstant(C);
       }
     }
 
@@ -519,7 +516,7 @@ bool DarwinAsmPrinter::doFinalization(Module &M) {
     if (PICEnabled) {
     O << ".data\n";
     O << ".section __TEXT,__picsymbolstub1,symbol_stubs,pure_instructions,32\n";
-    emitAlignment(2);
+    EmitAlignment(2);
     O << "L" << *i << "$stub:\n";
     O << "\t.indirect_symbol " << *i << "\n";
     O << "\tmflr r0\n";
@@ -538,7 +535,7 @@ bool DarwinAsmPrinter::doFinalization(Module &M) {
     O << "\t.long dyld_stub_binding_helper\n";
     } else {
     O << "\t.section __TEXT,__symbol_stub1,symbol_stubs,pure_instructions,16\n";
-    emitAlignment(4);
+    EmitAlignment(4);
     O << "L" << *i << "$stub:\n";
     O << "\t.indirect_symbol " << *i << "\n";
     O << "\tlis r11,ha16(L" << *i << "$lazy_ptr)\n";
@@ -588,7 +585,7 @@ bool DarwinAsmPrinter::doFinalization(Module &M) {
 /// method to print assembly for each instruction.
 ///
 bool AIXAsmPrinter::runOnMachineFunction(MachineFunction &MF) {
-  CurrentFnName = MF.getFunction()->getName();
+  SetupMachineFunction(MF);
 
   // Print out constants referenced by the function
   printConstantPool(MF.getConstantPool());
@@ -608,8 +605,9 @@ bool AIXAsmPrinter::runOnMachineFunction(MachineFunction &MF) {
   for (MachineFunction::const_iterator I = MF.begin(), E = MF.end();
        I != E; ++I) {
     // Print a label for the basic block.
-    O << "LBB" << CurrentFnName << '_' << I->getNumber() << ":\t# "
-      << I->getBasicBlock()->getName() << '\n';
+    O << PrivateGlobalPrefix << "BB" << getFunctionNumber() << '_'
+      << I->getNumber()
+      << ":\t" << CommentString << I->getBasicBlock()->getName() << '\n';
     for (MachineBasicBlock::const_iterator II = I->begin(), E = I->end();
       II != E; ++II) {
       // Print the assembly for the instruction.
@@ -617,7 +615,6 @@ bool AIXAsmPrinter::runOnMachineFunction(MachineFunction &MF) {
       printMachineInstruction(II);
     }
   }
-  ++FunctionNumber;
 
   O << "LT.." << CurrentFnName << ":\n"
     << "\t.long 0\n"
@@ -631,27 +628,6 @@ bool AIXAsmPrinter::runOnMachineFunction(MachineFunction &MF) {
   return false;
 }
 
-/// printConstantPool - Print to the current output stream assembly
-/// representations of the constants in the constant pool MCP. This is
-/// used to print out constants which have been "spilled to memory" by
-/// the code generator.
-///
-void AIXAsmPrinter::printConstantPool(MachineConstantPool *MCP) {
-  const std::vector<Constant*> &CP = MCP->getConstants();
-  const TargetData &TD = TM.getTargetData();
-
-  if (CP.empty()) return;
-
-  for (unsigned i = 0, e = CP.size(); i != e; ++i) {
-    SwitchSection(".const", 0);
-    O << "\t.align " << (unsigned)TD.getTypeAlignment(CP[i]->getType())
-      << "\n";
-    O << PrivateGlobalPrefix << "CPI" << FunctionNumber << '_' << i
-      << ":\t\t\t\t\t;" << *CP[i] << '\n';
-    emitGlobalConstant(CP[i]);
-  }
-}
-
 bool AIXAsmPrinter::doInitialization(Module &M) {
   SwitchSection("", 0);
   const TargetData &TD = TM.getTargetData();
@@ -676,7 +652,7 @@ bool AIXAsmPrinter::doInitialization(Module &M) {
       O << "\t.csect _global.rw_c[RW],3\n";
     }
     O << Name << ":\n";
-    emitGlobalConstant(C);
+    EmitGlobalConstant(C);
   }
 
   // Output labels for globals
@@ -688,14 +664,15 @@ bool AIXAsmPrinter::doInitialization(Module &M) {
     if (GV->isExternal() && GV->use_begin() == GV->use_end())
       continue;
 
+    IncrementFunctionNumber();
     std::string Name = GV->getName();
-    std::string Label = "LC.." + utostr(FunctionNumber++);
+    std::string Label = "LC.." + utostr(getFunctionNumber());
     GVToLabelMap[GV] = Label;
     O << Label << ":\n"
       << "\t.tc " << Name << "[TC]," << Name;
     if (GV->isExternal()) O << "[RW]";
     O << '\n';
-  }
+   }
 
   AsmPrinter::doInitialization(M);
   return false; // success
@@ -716,7 +693,7 @@ bool AIXAsmPrinter::doFinalization(Module &M) {
       O << "\t.comm " << Name << "," << TD.getTypeSize(I->getType())
         << "," << Log2_32((unsigned)TD.getTypeAlignment(I->getType()));
     }
-    O << "\t\t# ";
+    O << "\t\t" << CommentString << " ";
     WriteAsOperand(O, I, false, true, &M);
     O << "\n";
   }