go through PIC16TargetObjectFile to make sections instead of
[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 <vector>
15
16 namespace llvm {
17   class GlobalVariable;
18   class Module;
19   class PIC16TargetMachine;
20   
21   enum { DataBankSize = 80 };
22
23   /// PIC16 Splits the global data into mulitple udata and idata sections.
24   /// Each udata and idata section needs to contain a list of globals that
25   /// they contain, in order to avoid scanning over all the global values 
26   /// again and printing only those that match the current section. 
27   /// Keeping values inside the sections make printing a section much easier.
28   ///
29   /// FIXME: Reimplement by inheriting from MCSection.
30   ///
31   struct PIC16Section {
32     const MCSection *S_; // Connection to actual Section.
33     unsigned Size;  // Total size of the objects contained.
34     bool SectionPrinted;
35     std::vector<const GlobalVariable*> Items;
36     
37     PIC16Section(const MCSection *s) {
38       S_ = s;
39       Size = 0;
40       SectionPrinted = false;
41     }
42     bool isPrinted() const { return SectionPrinted; }
43     void setPrintedStatus(bool status) { SectionPrinted = status; } 
44   };
45   
46   class PIC16TargetObjectFile : public TargetLoweringObjectFile {
47     const TargetMachine *TM;
48   public:
49     mutable std::vector<PIC16Section*> BSSSections;
50     mutable std::vector<PIC16Section*> IDATASections;
51     mutable std::vector<PIC16Section*> AutosSections;
52     mutable std::vector<PIC16Section*> ROSections;
53     mutable PIC16Section *ExternalVarDecls;
54     mutable PIC16Section *ExternalVarDefs;
55
56     PIC16TargetObjectFile();
57     ~PIC16TargetObjectFile();
58     
59     void Initialize(MCContext &Ctx, const TargetMachine &TM);
60
61     
62     /// getSpecialCasedSectionGlobals - Allow the target to completely override
63     /// section assignment of a global.
64     virtual const MCSection *
65     getSpecialCasedSectionGlobals(const GlobalValue *GV, Mangler *Mang,
66                                   SectionKind Kind) const;
67     
68     virtual const MCSection *SelectSectionForGlobal(const GlobalValue *GV,
69                                                     SectionKind Kind,
70                                                     Mangler *Mang,
71                                                     const TargetMachine&) const;
72
73     const MCSection *getSectionForFunction(const std::string &FnName) const;
74     const MCSection *getSectionForFunctionFrame(const std::string &FnName)const;
75     
76     
77   private:
78     std::string getSectionNameForSym(const std::string &Sym) const;
79
80     const MCSection *getBSSSectionForGlobal(const GlobalVariable *GV) const;
81     const MCSection *getIDATASectionForGlobal(const GlobalVariable *GV) const;
82     const MCSection *getSectionForAuto(const GlobalVariable *GV) const;
83     const MCSection *CreateBSSSectionForGlobal(const GlobalVariable *GV,
84                                                std::string Addr = "") const;
85     const MCSection *CreateIDATASectionForGlobal(const GlobalVariable *GV,
86                                                  std::string Addr = "") const;
87     const MCSection *getROSectionForGlobal(const GlobalVariable *GV) const;
88     const MCSection *CreateROSectionForGlobal(const GlobalVariable *GV,
89                                               std::string Addr = "") const;
90     const MCSection *CreateSectionForGlobal(const GlobalVariable *GV,
91                                             Mangler *Mang,
92                                             const std::string &Addr = "") const;
93   public:
94     void SetSectionForGVs(Module &M);
95     const std::vector<PIC16Section*> &getBSSSections() const {
96       return BSSSections;
97     }
98     const std::vector<PIC16Section*> &getIDATASections() const {
99       return IDATASections;
100     }
101     const std::vector<PIC16Section*> &getAutosSections() const {
102       return AutosSections;
103     }
104     const std::vector<PIC16Section*> &getROSections() const {
105       return ROSections;
106     }
107     
108   };
109 } // end namespace llvm
110
111 #endif