make SectionForGlobal non-virtual, add a hook for pic16 to do its "address=" hack.
[oota-llvm.git] / lib / Target / PIC16 / PIC16TargetAsmInfo.cpp
1 //===-- PIC16TargetAsmInfo.cpp - PIC16 asm properties ---------------------===//
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 the declarations of the PIC16TargetAsmInfo properties.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #include "PIC16TargetAsmInfo.h"
15 #include "PIC16TargetMachine.h"
16 #include "llvm/GlobalValue.h"
17 #include "llvm/GlobalVariable.h"
18 #include "llvm/DerivedTypes.h"
19
20 using namespace llvm;
21
22 PIC16TargetAsmInfo::
23 PIC16TargetAsmInfo(const PIC16TargetMachine &TM) 
24   : TargetAsmInfo(TM) {
25   CommentString = ";";
26   GlobalPrefix = PAN::getTagName(PAN::PREFIX_SYMBOL);
27   GlobalDirective = "\tglobal\t";
28   ExternDirective = "\textern\t";
29
30   Data8bitsDirective = " db ";
31   Data16bitsDirective = " dw ";
32   Data32bitsDirective = " dl ";
33   Data64bitsDirective = NULL;
34   RomData8bitsDirective = " dw ";
35   RomData16bitsDirective = " rom_di ";
36   RomData32bitsDirective = " rom_dl ";
37   ZeroDirective = NULL;
38   AsciiDirective = " dt ";
39   AscizDirective = NULL;
40   BSSSection_  = getNamedSection("udata.# UDATA",
41                               SectionFlags::Writeable | SectionFlags::BSS);
42   ReadOnlySection = getNamedSection("romdata.# ROMDATA", SectionFlags::None);
43   DataSection = getNamedSection("idata.# IDATA", SectionFlags::Writeable);
44   SwitchToSectionDirective = "";
45   // Need because otherwise a .text symbol is emitted by DwarfWriter
46   // in BeginModule, and gpasm cribbs for that .text symbol.
47   TextSection = getUnnamedSection("", SectionFlags::Code);
48   PIC16Section *ROSection = new PIC16Section(getReadOnlySection());
49   ROSections.push_back(ROSection);
50   ExternalVarDecls = new PIC16Section(getNamedSection("ExternalVarDecls"));
51   ExternalVarDefs = new PIC16Section(getNamedSection("ExternalVarDefs"));
52   // Set it to false because we weed to generate c file name and not bc file
53   // name.
54   HasSingleParameterDotFile = false;
55 }
56
57 const char *PIC16TargetAsmInfo::getRomDirective(unsigned Size) const {
58   switch (Size) {
59   case  8: return RomData8bitsDirective;
60   case 16: return RomData16bitsDirective;
61   case 32: return RomData32bitsDirective;
62   default: return NULL;
63   }
64 }
65
66
67 const char *PIC16TargetAsmInfo::
68 getDataASDirective(unsigned Size, unsigned AS) const {
69   if (AS == PIC16ISD::ROM_SPACE)
70     return getRomDirective(Size);
71   return NULL;
72 }
73
74 const Section *
75 PIC16TargetAsmInfo::getBSSSectionForGlobal(const GlobalVariable *GV) const {
76   assert(GV->hasInitializer() && "This global doesn't need space");
77   Constant *C = GV->getInitializer();
78   assert(C->isNullValue() && "Unitialized globals has non-zero initializer");
79
80   // Find how much space this global needs.
81   const TargetData *TD = TM.getTargetData();
82   const Type *Ty = C->getType(); 
83   unsigned ValSize = TD->getTypeAllocSize(Ty);
84  
85   // Go through all BSS Sections and assign this variable
86   // to the first available section having enough space.
87   PIC16Section *FoundBSS = NULL;
88   for (unsigned i = 0; i < BSSSections.size(); i++) {
89     if (DataBankSize - BSSSections[i]->Size >= ValSize) {
90       FoundBSS = BSSSections[i];
91       break;
92     }
93   }
94
95   // No BSS section spacious enough was found. Crate a new one.
96   if (!FoundBSS) {
97     std::string name = PAN::getUdataSectionName(BSSSections.size());
98     const Section *NewSection = getNamedSection(name.c_str());
99
100     FoundBSS = new PIC16Section(NewSection);
101
102     // Add this newly created BSS section to the list of BSSSections.
103     BSSSections.push_back(FoundBSS);
104   }
105   
106   // Insert the GV into this BSS.
107   FoundBSS->Items.push_back(GV);
108   FoundBSS->Size += ValSize;
109   return FoundBSS->S_;
110
111
112 const Section *
113 PIC16TargetAsmInfo::getIDATASectionForGlobal(const GlobalVariable *GV) const {
114   assert(GV->hasInitializer() && "This global doesn't need space");
115   Constant *C = GV->getInitializer();
116   assert(!C->isNullValue() && "initialized globals has zero initializer");
117   assert(GV->getType()->getAddressSpace() == PIC16ISD::RAM_SPACE &&
118          "can split initialized RAM data only");
119
120   // Find how much space this global needs.
121   const TargetData *TD = TM.getTargetData();
122   const Type *Ty = C->getType(); 
123   unsigned ValSize = TD->getTypeAllocSize(Ty);
124  
125   // Go through all IDATA Sections and assign this variable
126   // to the first available section having enough space.
127   PIC16Section *FoundIDATA = NULL;
128   for (unsigned i = 0; i < IDATASections.size(); i++) {
129     if (DataBankSize - IDATASections[i]->Size >= ValSize) {
130       FoundIDATA = IDATASections[i]; 
131       break;
132     }
133   }
134
135   // No IDATA section spacious enough was found. Crate a new one.
136   if (!FoundIDATA) {
137     std::string name = PAN::getIdataSectionName(IDATASections.size());
138     const Section *NewSection = getNamedSection(name.c_str());
139
140     FoundIDATA = new PIC16Section(NewSection);
141
142     // Add this newly created IDATA section to the list of IDATASections.
143     IDATASections.push_back(FoundIDATA);
144   }
145   
146   // Insert the GV into this IDATA.
147   FoundIDATA->Items.push_back(GV);
148   FoundIDATA->Size += ValSize;
149   return FoundIDATA->S_;
150
151
152 // Get the section for an automatic variable of a function.
153 // For PIC16 they are globals only with mangled names.
154 const Section *
155 PIC16TargetAsmInfo::getSectionForAuto(const GlobalVariable *GV) const {
156
157   const std::string name = PAN::getSectionNameForSym(GV->getName());
158
159   // Go through all Auto Sections and assign this variable
160   // to the appropriate section.
161   PIC16Section *FoundAutoSec = NULL;
162   for (unsigned i = 0; i < AutosSections.size(); i++) {
163     if (AutosSections[i]->S_->getName() == name) {
164       FoundAutoSec = AutosSections[i];
165       break;
166     }
167   }
168
169   // No Auto section was found. Crate a new one.
170   if (!FoundAutoSec) {
171     const Section *NewSection = getNamedSection(name.c_str());
172
173     FoundAutoSec = new PIC16Section(NewSection);
174
175     // Add this newly created autos section to the list of AutosSections.
176     AutosSections.push_back(FoundAutoSec);
177   }
178
179   // Insert the auto into this section.
180   FoundAutoSec->Items.push_back(GV);
181
182   return FoundAutoSec->S_;
183 }
184
185
186 // Override default implementation to put the true globals into
187 // multiple data sections if required.
188 const Section*
189 PIC16TargetAsmInfo::SelectSectionForGlobal(const GlobalValue *GV1) const {
190   // We select the section based on the initializer here, so it really
191   // has to be a GlobalVariable.
192   const GlobalVariable *GV = dyn_cast<GlobalVariable>(GV1); 
193   if (!GV)
194     return TargetAsmInfo::SelectSectionForGlobal(GV1);
195
196   // Record Exteranl Var Decls.
197   if (GV->isDeclaration()) {
198     ExternalVarDecls->Items.push_back(GV);
199     return ExternalVarDecls->S_;
200   }
201     
202   assert(GV->hasInitializer() && "A def without initializer?");
203
204   // First, if this is an automatic variable for a function, get the section
205   // name for it and return.
206   std::string name = GV->getName();
207   if (PAN::isLocalName(name))
208     return getSectionForAuto(GV);
209
210   // Record Exteranl Var Defs.
211   if (GV->hasExternalLinkage() || GV->hasCommonLinkage())
212     ExternalVarDefs->Items.push_back(GV);
213
214   // See if this is an uninitialized global.
215   const Constant *C = GV->getInitializer();
216   if (C->isNullValue()) 
217     return getBSSSectionForGlobal(GV); 
218
219   // If this is initialized data in RAM. Put it in the correct IDATA section.
220   if (GV->getType()->getAddressSpace() == PIC16ISD::RAM_SPACE) 
221     return getIDATASectionForGlobal(GV);
222
223   // This is initialized data in rom, put it in the readonly section.
224   if (GV->getType()->getAddressSpace() == PIC16ISD::ROM_SPACE) 
225     return getROSectionForGlobal(GV);
226
227   // Else let the default implementation take care of it.
228   return TargetAsmInfo::SelectSectionForGlobal(GV);
229 }
230
231 PIC16TargetAsmInfo::~PIC16TargetAsmInfo() {
232   for (unsigned i = 0; i < BSSSections.size(); i++)
233     delete BSSSections[i]; 
234   for (unsigned i = 0; i < IDATASections.size(); i++)
235     delete IDATASections[i]; 
236   for (unsigned i = 0; i < AutosSections.size(); i++)
237     delete AutosSections[i]; 
238   for (unsigned i = 0; i < ROSections.size(); i++)
239     delete ROSections[i];
240   delete ExternalVarDecls;
241   delete ExternalVarDefs;
242 }
243
244
245 /// getSpecialCasedSectionGlobals - Allow the target to completely override
246 /// section assignment of a global.
247 const Section *
248 PIC16TargetAsmInfo::getSpecialCasedSectionGlobals(const GlobalValue *GV,
249                                                   SectionKind::Kind Kind) const{
250   // If GV has a sectin name or section address create that section now.
251   if (GV->hasSection()) {
252     if (const GlobalVariable *GVar = cast<GlobalVariable>(GV)) {
253       std::string SectName = GVar->getSection();
254       // If address for a variable is specified, get the address and create
255       // section.
256       std::string AddrStr = "Address=";
257       if (SectName.compare(0, AddrStr.length(), AddrStr) == 0) {
258         std::string SectAddr = SectName.substr(AddrStr.length());
259         return CreateSectionForGlobal(GVar, SectAddr);
260       }
261     }
262   }
263
264   return 0;
265 }
266
267 // Create a new section for global variable. If Addr is given then create
268 // section at that address else create by name.
269 const Section *
270 PIC16TargetAsmInfo::CreateSectionForGlobal(const GlobalVariable *GV,
271                                            const std::string &Addr) const {
272   // See if this is an uninitialized global.
273   const Constant *C = GV->getInitializer();
274   if (C->isNullValue())
275     return CreateBSSSectionForGlobal(GV, Addr);
276
277   // If this is initialized data in RAM. Put it in the correct IDATA section.
278   if (GV->getType()->getAddressSpace() == PIC16ISD::RAM_SPACE)
279     return CreateIDATASectionForGlobal(GV, Addr);
280
281   // This is initialized data in rom, put it in the readonly section.
282   if (GV->getType()->getAddressSpace() == PIC16ISD::ROM_SPACE) 
283     return CreateROSectionForGlobal(GV, Addr);
284
285   // Else let the default implementation take care of it.
286   return TargetAsmInfo::SectionForGlobal(GV);
287 }
288
289 // Create uninitialized section for a variable.
290 const Section *
291 PIC16TargetAsmInfo::CreateBSSSectionForGlobal(const GlobalVariable *GV,
292                                               std::string Addr) const {
293   assert(GV->hasInitializer() && "This global doesn't need space");
294   assert(GV->getInitializer()->isNullValue() &&
295          "Unitialized global has non-zero initializer");
296   std::string Name;
297   // If address is given then create a section at that address else create a
298   // section by section name specified in GV.
299   PIC16Section *FoundBSS = NULL;
300   if (Addr.empty()) { 
301     Name = GV->getSection() + " UDATA";
302     for (unsigned i = 0; i < BSSSections.size(); i++) {
303       if (BSSSections[i]->S_->getName() == Name) {
304         FoundBSS = BSSSections[i];
305         break;
306       }
307     }
308   } else {
309     std::string Prefix = GV->getNameStr() + "." + Addr + ".";
310     Name = PAN::getUdataSectionName(BSSSections.size(), Prefix) + " " + Addr;
311   }
312   
313   PIC16Section *NewBSS = FoundBSS;
314   if (NewBSS == NULL) {
315     const Section *NewSection = getNamedSection(Name.c_str());
316     NewBSS = new PIC16Section(NewSection);
317     BSSSections.push_back(NewBSS);
318   }
319
320   // Insert the GV into this BSS.
321   NewBSS->Items.push_back(GV);
322
323   // We do not want to put any  GV without explicit section into this section
324   // so set its size to DatabankSize.
325   NewBSS->Size = DataBankSize;
326   return NewBSS->S_;
327 }
328
329 // Get rom section for a variable. Currently there can be only one rom section
330 // unless a variable explicitly requests a section.
331 const Section *
332 PIC16TargetAsmInfo::getROSectionForGlobal(const GlobalVariable *GV) const {
333   ROSections[0]->Items.push_back(GV);
334   return ROSections[0]->S_;
335 }
336
337 // Create initialized data section for a variable.
338 const Section *
339 PIC16TargetAsmInfo::CreateIDATASectionForGlobal(const GlobalVariable *GV,
340                                                 std::string Addr) const {
341   assert(GV->hasInitializer() && "This global doesn't need space");
342   assert(!GV->getInitializer()->isNullValue() &&
343          "initialized global has zero initializer");
344   assert(GV->getType()->getAddressSpace() == PIC16ISD::RAM_SPACE &&
345          "can be used for initialized RAM data only");
346
347   std::string Name;
348   // If address is given then create a section at that address else create a
349   // section by section name specified in GV.
350   PIC16Section *FoundIDATASec = NULL;
351   if (Addr.empty()) {
352     Name = GV->getSection() + " IDATA";
353     for (unsigned i = 0; i < IDATASections.size(); i++) {
354       if (IDATASections[i]->S_->getName() == Name) {
355         FoundIDATASec = IDATASections[i];
356         break;
357       }
358     }
359   } else {
360     std::string Prefix = GV->getNameStr() + "." + Addr + ".";
361     Name = PAN::getIdataSectionName(IDATASections.size(), Prefix) + " " + Addr;
362   }
363
364   PIC16Section *NewIDATASec = FoundIDATASec;
365   if (NewIDATASec == NULL) {
366     const Section *NewSection = getNamedSection(Name.c_str());
367     NewIDATASec = new PIC16Section(NewSection);
368     IDATASections.push_back(NewIDATASec);
369   }
370   // Insert the GV into this IDATA Section.
371   NewIDATASec->Items.push_back(GV);
372   // We do not want to put any  GV without explicit section into this section 
373   // so set its size to DatabankSize.
374   NewIDATASec->Size = DataBankSize;
375   return NewIDATASec->S_;
376 }
377
378 // Create a section in rom for a variable.
379 const Section *
380 PIC16TargetAsmInfo::CreateROSectionForGlobal(const GlobalVariable *GV,
381                                              std::string Addr) const {
382   assert(GV->getType()->getAddressSpace() == PIC16ISD::ROM_SPACE &&
383          "can be used for ROM data only");
384
385   std::string Name;
386   // If address is given then create a section at that address else create a
387   // section by section name specified in GV.
388   PIC16Section *FoundROSec = NULL;
389   if (Addr.empty()) {
390     Name = GV->getSection() + " ROMDATA";
391     for (unsigned i = 1; i < ROSections.size(); i++) {
392       if (ROSections[i]->S_->getName() == Name) {
393         FoundROSec = ROSections[i];
394         break;
395       }
396     }
397   } else {
398     std::string Prefix = GV->getNameStr() + "." + Addr + ".";
399     Name = PAN::getRomdataSectionName(ROSections.size(), Prefix) + " " + Addr;
400   }
401
402   PIC16Section *NewRomSec = FoundROSec;
403   if (NewRomSec == NULL) {
404     const Section *NewSection = getNamedSection(Name.c_str());
405     NewRomSec = new PIC16Section(NewSection);
406     ROSections.push_back(NewRomSec);
407   }
408
409   // Insert the GV into this ROM Section.
410   NewRomSec->Items.push_back(GV);
411   return NewRomSec->S_;
412 }
413