Start refactoring PIC16 TargetObjectFile code. Eventually, all the stuff from
[oota-llvm.git] / lib / Target / PIC16 / PIC16TargetObjectFile.h
1 //===-- PIC16TargetObjectFile.h - PIC16 Object Info -------------*- C++ -*-===//
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 #ifndef LLVM_TARGET_PIC16_TARGETOBJECTFILE_H
11 #define LLVM_TARGET_PIC16_TARGETOBJECTFILE_H
12
13 #include "llvm/Target/TargetLoweringObjectFile.h"
14 #include "llvm/ADT/StringMap.h"
15 #include <vector>
16 #include <string>
17
18 namespace llvm {
19   class GlobalVariable;
20   class Module;
21   class PIC16TargetMachine;
22   class MCSectionPIC16;
23   
24   enum { DataBankSize = 80 };
25
26   /// PIC16 Splits the global data into mulitple udata and idata sections.
27   /// Each udata and idata section needs to contain a list of globals that
28   /// they contain, in order to avoid scanning over all the global values 
29   /// again and printing only those that match the current section. 
30   /// Keeping values inside the sections make printing a section much easier.
31   ///
32   /// FIXME: MOVE ALL THIS STUFF TO MCSectionPIC16.
33   ///
34   struct PIC16Section {
35     const MCSectionPIC16 *S_; // Connection to actual Section.
36     unsigned Size;  // Total size of the objects contained.
37     bool SectionPrinted;
38     std::vector<const GlobalVariable*> Items;
39     
40     PIC16Section(const MCSectionPIC16 *s) {
41       S_ = s;
42       Size = 0;
43       SectionPrinted = false;
44     }
45     bool isPrinted() const { return SectionPrinted; }
46     void setPrintedStatus(bool status) { SectionPrinted = status; } 
47   };
48   
49   class PIC16TargetObjectFile : public TargetLoweringObjectFile {
50     /// SectionsByName - Bindings of names to allocated sections.
51     mutable StringMap<MCSectionPIC16*> SectionsByName;
52
53     const TargetMachine *TM;
54     
55     const MCSectionPIC16 *getPIC16Section(const char *Name,
56                                           SectionKind K, 
57                                           int Address = -1) const;
58   public:
59     mutable std::vector<PIC16Section*> BSSSections;
60     mutable std::vector<PIC16Section*> IDATASections;
61     mutable std::vector<PIC16Section*> AutosSections;
62     mutable std::vector<PIC16Section*> ROSections;
63     mutable PIC16Section *ExternalVarDecls;
64     mutable PIC16Section *ExternalVarDefs;
65
66     PIC16TargetObjectFile();
67     ~PIC16TargetObjectFile();
68     
69     void Initialize(MCContext &Ctx, const TargetMachine &TM);
70
71     
72     virtual const MCSection *
73     getExplicitSectionGlobal(const GlobalValue *GV, SectionKind Kind, 
74                              Mangler *Mang, const TargetMachine &TM) const;
75     
76     virtual const MCSection *SelectSectionForGlobal(const GlobalValue *GV,
77                                                     SectionKind Kind,
78                                                     Mangler *Mang,
79                                                     const TargetMachine&) const;
80
81     const MCSection *getSectionForFunction(const std::string &FnName) const;
82     const MCSection *getSectionForFunctionFrame(const std::string &FnName)const;
83     
84     
85   private:
86     std::string getSectionNameForSym(const std::string &Sym) const;
87
88     const MCSection *getBSSSectionForGlobal(const GlobalVariable *GV) const;
89     const MCSection *getIDATASectionForGlobal(const GlobalVariable *GV) const;
90     const MCSection *getSectionForAuto(const GlobalVariable *GV) const;
91     const MCSection *CreateBSSSectionForGlobal(const GlobalVariable *GV,
92                                                std::string Addr = "") const;
93     const MCSection *CreateIDATASectionForGlobal(const GlobalVariable *GV,
94                                                  std::string Addr = "") const;
95     const MCSection *getROSectionForGlobal(const GlobalVariable *GV) const;
96     const MCSection *CreateROSectionForGlobal(const GlobalVariable *GV,
97                                               std::string Addr = "") const;
98     const MCSection *CreateSectionForGlobal(const GlobalVariable *GV,
99                                             Mangler *Mang,
100                                             const std::string &Addr = "") const;
101   public:
102     void SetSectionForGVs(Module &M);
103     const std::vector<PIC16Section*> &getBSSSections() const {
104       return BSSSections;
105     }
106     const std::vector<PIC16Section*> &getIDATASections() const {
107       return IDATASections;
108     }
109     const std::vector<PIC16Section*> &getAutosSections() const {
110       return AutosSections;
111     }
112     const std::vector<PIC16Section*> &getROSections() const {
113       return ROSections;
114     }
115     
116   };
117 } // end namespace llvm
118
119 #endif