lib/Target/Alpha/AlphaAsmPrinter.cpp

   1 //===-- AlphaAsmPrinter.cpp - Alpha LLVM assembly writer ------------------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file was developed by the LLVM research group and is distributed under
   6 // the University of Illinois Open Source License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // This file contains a printer that converts from our internal representation
  11 // of machine-dependent LLVM code to GAS-format Alpha assembly language.
  12 //
  13 //===----------------------------------------------------------------------===//
  14
  15 #include "Alpha.h"
  16 #include "AlphaInstrInfo.h"
  17 #include "AlphaTargetMachine.h"
  18 #include "llvm/Module.h"
  19 #include "llvm/Type.h"
  20 #include "llvm/Assembly/Writer.h"
  21 #include "llvm/CodeGen/MachineConstantPool.h"
  22 #include "llvm/CodeGen/ValueTypes.h"
  23 #include "llvm/CodeGen/AsmPrinter.h"
  24 #include "llvm/Target/TargetMachine.h"
  25 #include "llvm/Support/Mangler.h"
  26 #include "llvm/ADT/Statistic.h"
  27
  28 using namespace llvm;
  29
  30 namespace {
  31   Statistic<> EmittedInsts("asm-printer", "Number of machine instrs printed");
  32
  33   struct AlphaAsmPrinter : public AsmPrinter {
  34
  35     /// Unique incrementer for label values for referencing Global values.
  36     ///
  37     unsigned LabelNumber;
  38
  39      AlphaAsmPrinter(std::ostream &o, TargetMachine &tm)
  40        : AsmPrinter(o, tm), LabelNumber(0) {
  41       AlignmentIsInBytes = false;
  42       PrivateGlobalPrefix = "$";
  43     }
  44
  45     /// We name each basic block in a Function with a unique number, so
  46     /// that we can consistently refer to them later. This is cleared
  47     /// at the beginning of each call to runOnMachineFunction().
  48     ///
  49     typedef std::map<const Value *, unsigned> ValueMapTy;
  50     ValueMapTy NumberForBB;
  51     std::string CurSection;
  52
  53     virtual const char *getPassName() const {
  54       return "Alpha Assembly Printer";
  55     }
  56     bool printInstruction(const MachineInstr *MI);
  57     void printOp(const MachineOperand &MO, bool IsCallOp = false);
  58     void printConstantPool(MachineConstantPool *MCP);
  59     void printOperand(const MachineInstr *MI, int opNum, MVT::ValueType VT);
  60     void printBaseOffsetPair (const MachineInstr *MI, int i, bool brackets=true);
  61     void printMachineInstruction(const MachineInstr *MI);
  62     bool runOnMachineFunction(MachineFunction &F);
  63     bool doInitialization(Module &M);
  64     bool doFinalization(Module &M);
  65   };
  66 } // end of anonymous namespace
  67
  68 /// createAlphaCodePrinterPass - Returns a pass that prints the Alpha
  69 /// assembly code for a MachineFunction to the given output stream,
  70 /// using the given target machine description.  This should work
  71 /// regardless of whether the function is in SSA form.
  72 ///
  73 FunctionPass *llvm::createAlphaCodePrinterPass (std::ostream &o,
  74                                                   TargetMachine &tm) {
  75   return new AlphaAsmPrinter(o, tm);
  76 }
  77
  78 #include "AlphaGenAsmWriter.inc"
  79
  80 void AlphaAsmPrinter::printOperand(const MachineInstr *MI, int opNum, MVT::ValueType VT)
  81 {
  82   const MachineOperand &MO = MI->getOperand(opNum);
  83   if (MO.getType() == MachineOperand::MO_MachineRegister) {
  84     assert(MRegisterInfo::isPhysicalRegister(MO.getReg())&&"Not physreg??");
  85     O << TM.getRegisterInfo()->get(MO.getReg()).Name;
  86   } else if (MO.isImmediate()) {
  87     O << MO.getImmedValue();
  88   } else {
  89     printOp(MO);
  90   }
  91 }
  92
  93
  94 void AlphaAsmPrinter::printOp(const MachineOperand &MO, bool IsCallOp) {
  95   const MRegisterInfo &RI = *TM.getRegisterInfo();
  96   int new_symbol;
  97
  98   switch (MO.getType()) {
  99   case MachineOperand::MO_VirtualRegister:
 100     if (Value *V = MO.getVRegValueOrNull()) {
 101       O << "<" << V->getName() << ">";
 102       return;
 103     }
 104     // FALLTHROUGH
 105   case MachineOperand::MO_MachineRegister:
 106   case MachineOperand::MO_CCRegister:
 107     O << RI.get(MO.getReg()).Name;
 108     return;
 109
 110   case MachineOperand::MO_SignExtendedImmed:
 111   case MachineOperand::MO_UnextendedImmed:
 112     std::cerr << "printOp() does not handle immediate values\n";
 113     abort();
 114     return;
 115
 116   case MachineOperand::MO_PCRelativeDisp:
 117     std::cerr << "Shouldn't use addPCDisp() when building Alpha MachineInstrs";
 118     abort();
 119     return;
 120
 121   case MachineOperand::MO_MachineBasicBlock: {
 122     MachineBasicBlock *MBBOp = MO.getMachineBasicBlock();
 123     O << PrivateGlobalPrefix << "LBB"
 124       << Mang->getValueName(MBBOp->getParent()->getFunction())
 125       << "_" << MBBOp->getNumber() << "\t" << CommentString << " "
 126       << MBBOp->getBasicBlock()->getName();
 127     return;
 128   }
 129
 130   case MachineOperand::MO_ConstantPoolIndex:
 131     O << PrivateGlobalPrefix << "CPI" << CurrentFnName << "_"
 132       << MO.getConstantPoolIndex();
 133     return;
 134
 135   case MachineOperand::MO_ExternalSymbol:
 136     O << MO.getSymbolName();
 137     return;
 138
 139   case MachineOperand::MO_GlobalAddress:
 140     //Abuse PCrel to specify pcrel calls
 141     //calls are the only thing that use this flag
 142     if (MO.isPCRelative())
 143       O << PrivateGlobalPrefix << Mang->getValueName(MO.getGlobal()) << "..ng";
 144     else
 145       O << Mang->getValueName(MO.getGlobal());
 146     return;
 147
 148   default:
 149     O << "<unknown operand type: " << MO.getType() << ">";
 150     return;
 151   }
 152 }
 153
 154 /// printMachineInstruction -- Print out a single Alpha MI to
 155 /// the current output stream.
 156 ///
 157 void AlphaAsmPrinter::printMachineInstruction(const MachineInstr *MI) {
 158   ++EmittedInsts;
 159   if (printInstruction(MI))
 160     return; // Printer was automatically generated
 161
 162   assert(0 && "Unhandled instruction in asm writer!");
 163   abort();
 164   return;
 165 }
 166
 167
 168 /// runOnMachineFunction - This uses the printMachineInstruction()
 169 /// method to print assembly for each instruction.
 170 ///
 171 bool AlphaAsmPrinter::runOnMachineFunction(MachineFunction &MF) {
 172   SetupMachineFunction(MF);
 173   O << "\n\n";
 174
 175   // Print out constants referenced by the function
 176   printConstantPool(MF.getConstantPool());
 177
 178   // Print out labels for the function.
 179   SwitchSection("\t.section .text", MF.getFunction());
 180   EmitAlignment(4);
 181   O << "\t.globl " << CurrentFnName << "\n";
 182   O << "\t.ent " << CurrentFnName << "\n";
 183
 184   O << CurrentFnName << ":\n";
 185
 186   // Print out code for the function.
 187   for (MachineFunction::const_iterator I = MF.begin(), E = MF.end();
 188        I != E; ++I) {
 189     // Print a label for the basic block.
 190     O << PrivateGlobalPrefix << "LBB" << CurrentFnName << "_" << I->getNumber()
 191       << ":\t" << CommentString << " " << I->getBasicBlock()->getName() << "\n";
 192     for (MachineBasicBlock::const_iterator II = I->begin(), E = I->end();
 193          II != E; ++II) {
 194       // Print the assembly for the instruction.
 195       O << "\t";
 196       printMachineInstruction(II);
 197     }
 198   }
 199   ++LabelNumber;
 200
 201   O << "\t.end " << CurrentFnName << "\n";
 202
 203   // We didn't modify anything.
 204   return false;
 205 }
 206
 207
 208 /// printConstantPool - Print to the current output stream assembly
 209 /// representations of the constants in the constant pool MCP. This is
 210 /// used to print out constants which have been "spilled to memory" by
 211 /// the code generator.
 212 ///
 213 void AlphaAsmPrinter::printConstantPool(MachineConstantPool *MCP) {
 214   const std::vector<Constant*> &CP = MCP->getConstants();
 215   const TargetData &TD = TM.getTargetData();
 216
 217   if (CP.empty()) return;
 218
 219   SwitchSection("\t.section .rodata", 0);
 220   for (unsigned i = 0, e = CP.size(); i != e; ++i) {
 221     EmitAlignment(TD.getTypeAlignmentShift(CP[i]->getType()));
 222     O << PrivateGlobalPrefix << "CPI" << CurrentFnName << "_" << i
 223       << ":\t\t\t\t\t" << CommentString << *CP[i] << "\n";
 224     EmitGlobalConstant(CP[i]);
 225   }
 226 }
 227
 228 bool AlphaAsmPrinter::doInitialization(Module &M)
 229 {
 230   AsmPrinter::doInitialization(M);
 231   if(TM.getSubtarget<AlphaSubtarget>().hasF2I()
 232      || TM.getSubtarget<AlphaSubtarget>().hasCT())
 233     O << "\t.arch ev6\n";
 234   else
 235     O << "\t.arch ev56\n";
 236   O << "\t.set noat\n";
 237   return false;
 238 }
 239
 240 bool AlphaAsmPrinter::doFinalization(Module &M) {
 241   const TargetData &TD = TM.getTargetData();
 242
 243   for (Module::const_global_iterator I = M.global_begin(), E = M.global_end(); I != E; ++I)
 244     if (I->hasInitializer()) {   // External global require no code
 245       O << "\n\n";
 246       std::string name = Mang->getValueName(I);
 247       Constant *C = I->getInitializer();
 248       unsigned Size = TD.getTypeSize(C->getType());
 249       unsigned Align = TD.getTypeAlignmentShift(C->getType());
 250
 251       if (C->isNullValue() &&
 252           (I->hasLinkOnceLinkage() || I->hasInternalLinkage() ||
 253            I->hasWeakLinkage() /* FIXME: Verify correct */)) {
 254         SwitchSection("\t.section .data", I);
 255         if (I->hasInternalLinkage())
 256           O << "\t.local " << name << "\n";
 257
 258         O << "\t.comm " << name << "," << TD.getTypeSize(C->getType())
 259           << "," << (1 << Align);
 260         O << "\t\t# ";
 261         WriteAsOperand(O, I, true, true, &M);
 262         O << "\n";
 263       } else {
 264         switch (I->getLinkage()) {
 265         case GlobalValue::LinkOnceLinkage:
 266         case GlobalValue::WeakLinkage:   // FIXME: Verify correct for weak.
 267           // Nonnull linkonce -> weak
 268           O << "\t.weak " << name << "\n";
 269           O << "\t.section\t.llvm.linkonce.d." << name << ",\"aw\",@progbits\n";
 270           SwitchSection("", I);
 271           break;
 272         case GlobalValue::AppendingLinkage:
 273           // FIXME: appending linkage variables should go into a section of
 274           // their name or something.  For now, just emit them as external.
 275         case GlobalValue::ExternalLinkage:
 276           // If external or appending, declare as a global symbol
 277           O << "\t.globl " << name << "\n";
 278           // FALL THROUGH
 279         case GlobalValue::InternalLinkage:
 280           SwitchSection(C->isNullValue() ? "\t.section .bss" :
 281                         "\t.section .data", I);
 282           break;
 283         case GlobalValue::GhostLinkage:
 284           std::cerr << "GhostLinkage cannot appear in AlphaAsmPrinter!\n";
 285           abort();
 286         }
 287
 288         EmitAlignment(Align);
 289         O << "\t.type " << name << ",@object\n";
 290         O << "\t.size " << name << "," << Size << "\n";
 291         O << name << ":\t\t\t\t# ";
 292         WriteAsOperand(O, I, true, true, &M);
 293         O << " = ";
 294         WriteAsOperand(O, C, false, false, &M);
 295         O << "\n";
 296         EmitGlobalConstant(C);
 297       }
 298     }
 299
 300   AsmPrinter::doFinalization(M);
 301   return false;
 302 }