The way we are trying to figure out banksel immediate operand may yield different...
[oota-llvm.git] / lib / Target / PIC16 / PIC16AsmPrinter.cpp
1 //===-- PIC16AsmPrinter.cpp - PIC16 LLVM assembly writer ------------------===//
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 a printer that converts from our internal representation
11 // of machine-dependent LLVM code to PIC16 assembly language.
12 //
13 //===----------------------------------------------------------------------===//
14
15 #include "PIC16AsmPrinter.h"
16 #include "PIC16TargetAsmInfo.h"
17 #include "llvm/DerivedTypes.h"
18 #include "llvm/Function.h"
19 #include "llvm/Module.h"
20 #include "llvm/CodeGen/DwarfWriter.h"
21 #include "llvm/CodeGen/MachineFrameInfo.h"
22 #include "llvm/Support/raw_ostream.h"
23 #include "llvm/Support/Mangler.h"
24 #include "llvm/CodeGen/DwarfWriter.h"
25 #include "llvm/CodeGen/MachineModuleInfo.h"
26
27 using namespace llvm;
28
29 #include "PIC16GenAsmWriter.inc"
30
31 inline static bool isLocalToFunc (std::string &FuncName, std::string &VarName) {
32   if (VarName.find(FuncName + ".auto.") != std::string::npos 
33       || VarName.find(FuncName + ".arg.") != std::string::npos)
34     return true;
35
36   return false;
37 }
38
39 inline static bool isLocalName (std::string &Name) {
40   if (Name.find(".auto.") != std::string::npos 
41       || Name.find(".arg.") != std::string::npos) 
42     return true;
43
44   return false;
45 }
46
47 bool PIC16AsmPrinter::printMachineInstruction(const MachineInstr *MI) {
48   std::string NewBank = "";
49   unsigned Operands = MI->getNumOperands();
50   if (Operands > 1) {
51     // Global address or external symbol should be second operand from last
52     // if we want to print banksel for it.
53     unsigned BankSelVar = Operands - 2;
54     // In cases where an instruction has a def or use defined in td file,
55     // that def or use becomes a machine instruction operand.
56     // eg. addfw_1 instruction defines STATUS register. So the machine
57     // instruction for it has MO_Register Operand as its last operand.
58     while ((MI->getOperand(BankSelVar + 1).getType() ==
59            MachineOperand::MO_Register) && (BankSelVar > 0))
60      BankSelVar--;
61     const MachineOperand &Op = MI->getOperand(BankSelVar);
62     unsigned OpType = Op.getType();
63     if (OpType == MachineOperand::MO_GlobalAddress ||
64         OpType == MachineOperand::MO_ExternalSymbol) {
65       if (OpType == MachineOperand::MO_GlobalAddress ) 
66         NewBank = Op.getGlobal()->getSection(); 
67       else {
68         // External Symbol is generated for temp data. Temp data in in
69         // fdata.<functionname>.# section.
70         NewBank = "fdata." + CurrentFnName +".#";
71       }
72       // Operand after global address or external symbol should be  banksel.
73       // Value 1 for this operand means we need to generate banksel else do not
74       // generate banksel.
75       const MachineOperand &BS = MI->getOperand(BankSelVar+1);
76       // If Section names are same then the variables are in same section.
77       // This is not true for external variables as section names for global
78       // variables in all files are same at this time. For eg. initialized 
79       // data in put in idata.# section in all files. 
80       if ((BS.getType() == MachineOperand::MO_Immediate 
81            && (int)BS.getImm() == 1) 
82           && ((Op.isGlobal() && Op.getGlobal()->hasExternalLinkage()) ||
83            (NewBank.compare(CurBank) != 0))) { 
84         O << "\tbanksel ";
85         printOperand(MI, BankSelVar);
86         O << "\n";
87         CurBank = NewBank;
88       }
89     }
90   }
91   printInstruction(MI);
92   return true;
93 }
94
95 /// runOnMachineFunction - This uses the printInstruction()
96 /// method to print assembly for each instruction.
97 ///
98 bool PIC16AsmPrinter::runOnMachineFunction(MachineFunction &MF) {
99   this->MF = &MF;
100
101   // This calls the base class function required to be called at beginning
102   // of runOnMachineFunction.
103   SetupMachineFunction(MF);
104
105   // Get the mangled name.
106   const Function *F = MF.getFunction();
107   CurrentFnName = Mang->getValueName(F);
108
109   // Emit the function variables.
110   emitFunctionData(MF);
111   std::string codeSection;
112   codeSection = "code." + CurrentFnName + ".# " + "CODE";
113   const Section *fCodeSection = TAI->getNamedSection(codeSection.c_str(),
114                                                SectionFlags::Code);
115   O <<  "\n";
116   SwitchToSection (fCodeSection);
117
118   // Emit the frame address of the function at the beginning of code.
119   O << CurrentFnName << ":\n";
120   O << "    retlw  low(" << CurrentFnName << ".frame)\n";
121   O << "    retlw  high(" << CurrentFnName << ".frame)\n"; 
122
123
124   // Print out code for the function.
125   for (MachineFunction::const_iterator I = MF.begin(), E = MF.end();
126        I != E; ++I) {
127     // Print a label for the basic block.
128     if (I != MF.begin()) {
129       printBasicBlockLabel(I, true);
130       O << '\n';
131     }
132     CurBank = "";
133     
134     // For emitting line directives, we need to keep track of the current
135     // source line. When it changes then only emit the line directive.
136     unsigned CurLine = 0;
137     for (MachineBasicBlock::const_iterator II = I->begin(), E = I->end();
138          II != E; ++II) {
139       // Emit the line directive if source line changed.
140       const DebugLoc DL = II->getDebugLoc();
141       if (!DL.isUnknown()) {
142         unsigned line = MF.getDebugLocTuple(DL).Line;
143         if (line != CurLine) {
144           O << "\t.line " << line << "\n";
145           CurLine = line;
146         }
147       }
148       // Print the assembly for the instruction.
149       printMachineInstruction(II);
150     }
151   }
152   return false;  // we didn't modify anything.
153 }
154
155 /// createPIC16CodePrinterPass - Returns a pass that prints the PIC16
156 /// assembly code for a MachineFunction to the given output stream,
157 /// using the given target machine description.  This should work
158 /// regardless of whether the function is in SSA form.
159 ///
160 FunctionPass *llvm::createPIC16CodePrinterPass(raw_ostream &o,
161                                                PIC16TargetMachine &tm,
162                                                bool fast, bool verbose) {
163   return new PIC16AsmPrinter(o, tm, tm.getTargetAsmInfo(), fast, verbose);
164 }
165
166 void PIC16AsmPrinter::printOperand(const MachineInstr *MI, int opNum) {
167   const MachineOperand &MO = MI->getOperand(opNum);
168
169   switch (MO.getType()) {
170     case MachineOperand::MO_Register:
171       if (TargetRegisterInfo::isPhysicalRegister(MO.getReg()))
172         O << TM.getRegisterInfo()->get(MO.getReg()).AsmName;
173       else
174         assert(0 && "not implemented");
175         return;
176
177     case MachineOperand::MO_Immediate:
178       O << (int)MO.getImm();
179       return;
180
181     case MachineOperand::MO_GlobalAddress:
182       O << Mang->getValueName(MO.getGlobal());
183       break;
184
185     case MachineOperand::MO_ExternalSymbol:
186       O << MO.getSymbolName();
187       break;
188
189     case MachineOperand::MO_MachineBasicBlock:
190       printBasicBlockLabel(MO.getMBB());
191       return;
192
193     default:
194       assert(0 && " Operand type not supported.");
195   }
196 }
197
198 void PIC16AsmPrinter::printCCOperand(const MachineInstr *MI, int opNum) {
199   int CC = (int)MI->getOperand(opNum).getImm();
200   O << PIC16CondCodeToString((PIC16CC::CondCodes)CC);
201 }
202
203
204 bool PIC16AsmPrinter::doInitialization (Module &M) {
205   bool Result = AsmPrinter::doInitialization(M);
206   // FIXME:: This is temporary solution to generate the include file.
207   // The processor should be passed to llc as in input and the header file
208   // should be generated accordingly.
209   O << "\t#include P16F1937.INC\n";
210   MachineModuleInfo *MMI = getAnalysisIfAvailable<MachineModuleInfo>();
211   assert(MMI);
212   DwarfWriter *DW = getAnalysisIfAvailable<DwarfWriter>();
213   assert(DW && "Dwarf Writer is not available");
214   DW->BeginModule(&M, MMI, O, this, TAI);
215
216   EmitExternsAndGlobals (M);
217   EmitInitData (M);
218   EmitUnInitData(M);
219   EmitRomData(M);
220   return Result;
221 }
222
223 void PIC16AsmPrinter::EmitExternsAndGlobals (Module &M) {
224  // Emit declarations for external functions.
225   O << "section.0" <<"\n";
226   for (Module::iterator I = M.begin(), E = M.end(); I != E; I++) {
227     std::string Name = Mang->getValueName(I);
228     if (Name.compare("abort") == 0)
229       continue;
230     if (I->isDeclaration()) {
231       O << "\textern " <<Name << "\n";
232       O << "\textern " << Name << ".retval\n";
233       O << "\textern " << Name << ".args\n";
234     }
235     else if (I->hasExternalLinkage()) {
236       O << "\tglobal " << Name << "\n";
237       O << "\tglobal " << Name << ".retval\n";
238       O << "\tglobal " << Name << ".args\n";
239     }
240   }
241
242   // Emit header file to include declaration of library functions
243   O << "\t#include C16IntrinsicCalls.INC\n";
244
245   // Emit declarations for external globals.
246   for (Module::const_global_iterator I = M.global_begin(), E = M.global_end();
247        I != E; I++) {
248     // Any variables reaching here with ".auto." in its name is a local scope
249     // variable and should not be printed in global data section.
250     std::string Name = Mang->getValueName(I);
251     if (isLocalName (Name))
252       continue;
253
254     if (I->isDeclaration())
255       O << "\textern "<< Name << "\n";
256     else if (I->hasCommonLinkage() || I->hasExternalLinkage())
257       O << "\tglobal "<< Name << "\n";
258   }
259 }
260
261 void PIC16AsmPrinter::EmitInitData (Module &M) {
262   SwitchToSection(TAI->getDataSection());
263   for (Module::global_iterator I = M.global_begin(), E = M.global_end();
264        I != E; ++I) {
265     if (!I->hasInitializer())   // External global require no code.
266       continue;
267
268     Constant *C = I->getInitializer();
269     const PointerType *PtrTy = I->getType();
270     int AddrSpace = PtrTy->getAddressSpace();
271
272     if ((!C->isNullValue()) && (AddrSpace == PIC16ISD::RAM_SPACE)) {
273     
274       if (EmitSpecialLLVMGlobal(I)) 
275         continue;
276
277       // Any variables reaching here with "." in its name is a local scope
278       // variable and should not be printed in global data section.
279       std::string Name = Mang->getValueName(I);
280       if (isLocalName(Name))
281         continue;
282
283       I->setSection(TAI->getDataSection()->getName());
284       O << Name;
285       EmitGlobalConstant(C, AddrSpace);
286     }
287   }
288 }
289
290 void PIC16AsmPrinter::EmitRomData (Module &M)
291 {
292   SwitchToSection(TAI->getReadOnlySection());
293   IsRomData = true;
294   for (Module::global_iterator I = M.global_begin(), E = M.global_end();
295        I != E; ++I) {
296     if (!I->hasInitializer())   // External global require no code.
297       continue;
298
299     Constant *C = I->getInitializer();
300     const PointerType *PtrTy = I->getType();
301     int AddrSpace = PtrTy->getAddressSpace();
302     if ((!C->isNullValue()) && (AddrSpace == PIC16ISD::ROM_SPACE)) {
303
304       if (EmitSpecialLLVMGlobal(I))
305         continue;
306
307       // Any variables reaching here with "." in its name is a local scope
308       // variable and should not be printed in global data section.
309       std::string name = Mang->getValueName(I);
310       if (name.find(".") != std::string::npos)
311         continue;
312
313       I->setSection(TAI->getReadOnlySection()->getName());
314       O << name;
315       EmitGlobalConstant(C, AddrSpace);
316       O << "\n";
317     }
318   }
319   IsRomData = false;
320 }
321
322 void PIC16AsmPrinter::EmitUnInitData (Module &M)
323 {
324   SwitchToSection(TAI->getBSSSection_());
325   const TargetData *TD = TM.getTargetData();
326
327   for (Module::global_iterator I = M.global_begin(), E = M.global_end();
328        I != E; ++I) {
329     if (!I->hasInitializer())   // External global require no code.
330       continue;
331
332     Constant *C = I->getInitializer();
333     if (C->isNullValue()) {
334
335       if (EmitSpecialLLVMGlobal(I))
336         continue;
337
338       // Any variables reaching here with "." in its name is a local scope
339       // variable and should not be printed in global data section.
340       std::string name = Mang->getValueName(I);
341       if (name.find(".") != std::string::npos)
342         continue;
343
344       I->setSection(TAI->getBSSSection_()->getName());
345
346       const Type *Ty = C->getType();
347       unsigned Size = TD->getTypePaddedSize(Ty);
348
349       O << name << " " <<"RES"<< " " << Size ;
350       O << "\n";
351     }
352   }
353 }
354
355 bool PIC16AsmPrinter::doFinalization(Module &M) {
356   O << "\t" << "END\n";
357   bool Result = AsmPrinter::doFinalization(M);
358   return Result;
359 }
360
361 void PIC16AsmPrinter::emitFunctionData(MachineFunction &MF) {
362   const Function *F = MF.getFunction();
363   std::string FuncName = Mang->getValueName(F);
364   Module *M = const_cast<Module *>(F->getParent());
365   const TargetData *TD = TM.getTargetData();
366   unsigned FrameSize = 0;
367   // Emit the data section name.
368   O << "\n"; 
369   std::string SectionName = "fdata." + CurrentFnName + ".# " + "UDATA";
370
371   const Section *fDataSection = TAI->getNamedSection(SectionName.c_str(),
372                                                SectionFlags::Writeable);
373   SwitchToSection(fDataSection);
374   
375
376   // Emit function frame label
377   O << CurrentFnName << ".frame:\n";
378
379   const Type *RetType = F->getReturnType();
380   unsigned RetSize = 0; 
381   if (RetType->getTypeID() != Type::VoidTyID) 
382     RetSize = TD->getTypePaddedSize(RetType);
383   
384   //Emit function return value space
385   if(RetSize > 0)
386      O << CurrentFnName << ".retval    RES  " << RetSize << "\n";
387   else
388      O << CurrentFnName << ".retval:\n";
389    
390   // Emit variable to hold the space for function arguments 
391   unsigned ArgSize = 0;
392   for (Function::const_arg_iterator argi = F->arg_begin(),
393            arge = F->arg_end(); argi != arge ; ++argi) {
394     const Type *Ty = argi->getType();
395     ArgSize += TD->getTypePaddedSize(Ty);
396    }
397   O << CurrentFnName << ".args      RES  " << ArgSize << "\n";
398
399   // Emit the function variables. 
400    
401   // In PIC16 all the function arguments and local variables are global.
402   // Therefore to get the variable belonging to this function entire
403   // global list will be traversed and variables belonging to this function
404   // will be emitted in the current data section.
405   for (Module::global_iterator I = M->global_begin(), E = M->global_end();
406        I != E; ++I) {
407     std::string VarName = Mang->getValueName(I);
408     
409     // The variables of a function are of form FuncName.* . If this variable
410     // does not belong to this function then continue. 
411     // Static local varilabes of a function does not have .auto. in their
412     // name. They are not printed as part of function data but module
413     // level global data.
414     if (! isLocalToFunc(FuncName, VarName))
415      continue;
416
417     I->setSection("fdata." + CurrentFnName + ".#");
418     Constant *C = I->getInitializer();
419     const Type *Ty = C->getType();
420     unsigned Size = TD->getTypePaddedSize(Ty);
421     FrameSize += Size; 
422     // Emit memory reserve directive.
423     O << VarName << "  RES  " << Size << "\n";
424   }
425
426   int TempSize = PTLI->GetTmpSize();
427   if (TempSize > 0 )
428     O << CurrentFnName << ".tmp       RES  " << TempSize <<"\n";
429 }