add support for external symbols. The mc instprinter can now handle
[oota-llvm.git] / lib / Target / ARM / AsmPrinter / ARMMCInstLower.cpp
1 //===-- ARMMCInstLower.cpp - Convert ARM MachineInstr to an MCInst --------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This file contains code to lower ARM MachineInstrs to their corresponding
11 // MCInst records.
12 //
13 //===----------------------------------------------------------------------===//
14
15 #include "ARMMCInstLower.h"
16 //#include "llvm/CodeGen/MachineModuleInfoImpls.h"
17 #include "llvm/CodeGen/AsmPrinter.h"
18 #include "llvm/CodeGen/MachineBasicBlock.h"
19 #include "llvm/MC/MCAsmInfo.h"
20 #include "llvm/MC/MCContext.h"
21 #include "llvm/MC/MCExpr.h"
22 #include "llvm/MC/MCInst.h"
23 //#include "llvm/MC/MCStreamer.h"
24 #include "llvm/Support/raw_ostream.h"
25 #include "llvm/Support/Mangler.h"
26 #include "llvm/ADT/SmallString.h"
27 using namespace llvm;
28
29
30 #if 0
31 const ARMSubtarget &ARMMCInstLower::getSubtarget() const {
32   return AsmPrinter.getSubtarget();
33 }
34
35 MachineModuleInfoMachO &ARMMCInstLower::getMachOMMI() const {
36   assert(getSubtarget().isTargetDarwin() &&"Can only get MachO info on darwin");
37   return AsmPrinter.MMI->getObjFileInfo<MachineModuleInfoMachO>(); 
38 }
39 #endif
40
41 MCSymbol *ARMMCInstLower::
42 GetGlobalAddressSymbol(const MachineOperand &MO) const {
43   const GlobalValue *GV = MO.getGlobal();
44   
45   SmallString<128> Name;
46   Mang.getNameWithPrefix(Name, GV, false);
47   
48   // FIXME: HANDLE PLT references how??
49   switch (MO.getTargetFlags()) {
50   default: assert(0 && "Unknown target flag on GV operand");
51   case 0: break;
52   }
53   
54   return Ctx.GetOrCreateSymbol(Name.str());
55 }
56
57 MCSymbol *ARMMCInstLower::
58 GetExternalSymbolSymbol(const MachineOperand &MO) const {
59   SmallString<128> Name;
60   Name += Printer.MAI->getGlobalPrefix();
61   Name += MO.getSymbolName();
62   
63   // FIXME: HANDLE PLT references how??
64   switch (MO.getTargetFlags()) {
65   default: assert(0 && "Unknown target flag on GV operand");
66   case 0: break;
67   }
68   
69   return Ctx.GetOrCreateSymbol(Name.str());
70 }
71
72
73
74 MCSymbol *ARMMCInstLower::
75 GetJumpTableSymbol(const MachineOperand &MO) const {
76   SmallString<256> Name;
77   raw_svector_ostream(Name) << Printer.MAI->getPrivateGlobalPrefix() << "JTI"
78     << Printer.getFunctionNumber() << '_' << MO.getIndex();
79   
80 #if 0
81   switch (MO.getTargetFlags()) {
82     default: llvm_unreachable("Unknown target flag on GV operand");
83   }
84 #endif
85   
86   // Create a symbol for the name.
87   return Ctx.GetOrCreateSymbol(Name.str());
88 }
89
90 MCSymbol *ARMMCInstLower::
91 GetConstantPoolIndexSymbol(const MachineOperand &MO) const {
92   SmallString<256> Name;
93   raw_svector_ostream(Name) << Printer.MAI->getPrivateGlobalPrefix() << "CPI"
94     << Printer.getFunctionNumber() << '_' << MO.getIndex();
95   
96 #if 0
97   switch (MO.getTargetFlags()) {
98   default: llvm_unreachable("Unknown target flag on GV operand");
99   }
100 #endif
101   
102   // Create a symbol for the name.
103   return Ctx.GetOrCreateSymbol(Name.str());
104 }
105   
106 MCOperand ARMMCInstLower::
107 LowerSymbolOperand(const MachineOperand &MO, MCSymbol *Sym) const {
108   // FIXME: We would like an efficient form for this, so we don't have to do a
109   // lot of extra uniquing.
110   const MCExpr *Expr = MCSymbolRefExpr::Create(Sym, Ctx);
111   
112 #if 0
113   switch (MO.getTargetFlags()) {
114   default: llvm_unreachable("Unknown target flag on GV operand");
115   }
116 #endif
117   
118   if (!MO.isJTI() && MO.getOffset())
119     Expr = MCBinaryExpr::CreateAdd(Expr,
120                                    MCConstantExpr::Create(MO.getOffset(), Ctx),
121                                    Ctx);
122   return MCOperand::CreateExpr(Expr);
123 }
124
125
126 void ARMMCInstLower::Lower(const MachineInstr *MI, MCInst &OutMI) const {
127   OutMI.setOpcode(MI->getOpcode());
128   
129   for (unsigned i = 0, e = MI->getNumOperands(); i != e; ++i) {
130     const MachineOperand &MO = MI->getOperand(i);
131     
132     MCOperand MCOp;
133     switch (MO.getType()) {
134     default:
135       MI->dump();
136       assert(0 && "unknown operand type");
137     case MachineOperand::MO_Register:
138       // Ignore all implicit register operands.
139       if (MO.isImplicit()) continue;
140       MCOp = MCOperand::CreateReg(MO.getReg());
141       break;
142     case MachineOperand::MO_Immediate:
143       MCOp = MCOperand::CreateImm(MO.getImm());
144       break;
145     case MachineOperand::MO_MachineBasicBlock:
146       MCOp = MCOperand::CreateExpr(MCSymbolRefExpr::Create(
147                        Printer.GetMBBSymbol(MO.getMBB()->getNumber()), Ctx));
148       break;
149     case MachineOperand::MO_GlobalAddress:
150       MCOp = LowerSymbolOperand(MO, GetGlobalAddressSymbol(MO));
151       break;
152     case MachineOperand::MO_ExternalSymbol:
153       MCOp = LowerSymbolOperand(MO, GetExternalSymbolSymbol(MO));
154       break;
155     case MachineOperand::MO_JumpTableIndex:
156       MCOp = LowerSymbolOperand(MO, GetJumpTableSymbol(MO));
157       break;
158     case MachineOperand::MO_ConstantPoolIndex:
159       MCOp = LowerSymbolOperand(MO, GetConstantPoolIndexSymbol(MO));
160       break;
161     }
162     
163     OutMI.addOperand(MCOp);
164   }
165   
166 }