Add support for putting globals in a particular section
[oota-llvm.git] / include / llvm / GlobalValue.h
1 //===-- llvm/GlobalValue.h - Class to represent a global value --*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file was developed by the LLVM research group and is distributed under
6 // the University of Illinois Open Source License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This file is a common base class of all globally definable objects.  As such,
11 // it is subclassed by GlobalVariable and by Function.  This is used because you
12 // can do certain things with these global objects that you can't do to anything
13 // else.  For example, use the address of one as a constant.
14 //
15 //===----------------------------------------------------------------------===//
16
17 #ifndef LLVM_GLOBALVALUE_H
18 #define LLVM_GLOBALVALUE_H
19
20 #include "llvm/Constant.h"
21
22 namespace llvm {
23
24 class PointerType;
25 class Module;
26
27 class GlobalValue : public Constant {
28   GlobalValue(const GlobalValue &);             // do not implement
29 public:
30   enum LinkageTypes {
31     ExternalLinkage,   /// Externally visible function
32     LinkOnceLinkage,   /// Keep one copy of named function when linking (inline)
33     WeakLinkage,       /// Keep one copy of named function when linking (weak)
34     AppendingLinkage,  /// Special purpose, only applies to global arrays
35     InternalLinkage,   /// Rename collisions when linking (static functions)
36     GhostLinkage       /// Stand-in functions for streaming fns from BC files
37   };
38 protected:
39   GlobalValue(const Type *Ty, ValueTy vty, Use *Ops, unsigned NumOps,
40               LinkageTypes linkage, const std::string &name = "")
41     : Constant(Ty, vty, Ops, NumOps, name), 
42       Parent(0), Linkage(linkage), Alignment(0) { }
43
44   Module *Parent;
45   LinkageTypes Linkage;   // The linkage of this global
46   unsigned Alignment;     // Alignment of this symbol, must be power of two
47   std::string Section;    // Section to emit this into, empty mean default
48 public:
49   ~GlobalValue() {
50     removeDeadConstantUsers();   // remove any dead constants using this.
51   }
52
53   unsigned getAlignment() const { return Alignment; }
54   void setAlignment(unsigned Align) {
55     assert((Align & (Align-1)) == 0 && "Alignment is not a power of 2!");
56     Alignment = Align;
57   }
58   
59   bool hasSection() const { return !Section.empty(); }
60   const std::string &getSection() const { return Section; }
61   void setSection(const std::string &S) { Section = S; }
62   
63   /// If the usage is empty (except transitively dead constants), then this
64   /// global value can can be safely deleted since the destructor will
65   /// delete the dead constants as well.
66   /// @brief Determine if the usage of this global value is empty except
67   /// for transitively dead constants.
68   bool use_empty_except_constants();
69
70   /// getType - Global values are always pointers.
71   inline const PointerType *getType() const {
72     return reinterpret_cast<const PointerType*>(User::getType());
73   }
74
75   bool hasExternalLinkage()  const { return Linkage == ExternalLinkage; }
76   bool hasLinkOnceLinkage()  const { return Linkage == LinkOnceLinkage; }
77   bool hasWeakLinkage()      const { return Linkage == WeakLinkage; }
78   bool hasAppendingLinkage() const { return Linkage == AppendingLinkage; }
79   bool hasInternalLinkage()  const { return Linkage == InternalLinkage; }
80   void setLinkage(LinkageTypes LT) { Linkage = LT; }
81   LinkageTypes getLinkage() const { return Linkage; }
82
83   /// hasNotBeenReadFromBytecode - If a module provider is being used to lazily
84   /// stream in functions from disk, this method can be used to check to see if
85   /// the function has been read in yet or not.  Unless you are working on the
86   /// JIT or something else that streams stuff in lazily, you don't need to
87   /// worry about this.
88   bool hasNotBeenReadFromBytecode() const { return Linkage == GhostLinkage; }
89
90   /// Override from Constant class. No GlobalValue's are null values so this
91   /// always returns false.
92   virtual bool isNullValue() const { return false; }
93
94   /// Override from Constant class.
95   virtual void destroyConstant();
96
97   /// isExternal - Return true if the primary definition of this global value is
98   /// outside of the current translation unit...
99   virtual bool isExternal() const = 0;
100
101   /// getParent - Get the module that this global value is contained inside
102   /// of...
103   inline Module *getParent() { return Parent; }
104   inline const Module *getParent() const { return Parent; }
105
106   /// removeDeadConstantUsers - If there are any dead constant users dangling
107   /// off of this global value, remove them.  This method is useful for clients
108   /// that want to check to see if a global is unused, but don't want to deal
109   /// with potentially dead constants hanging off of the globals.
110   ///
111   /// This method tries to make the global dead.  If it detects a user that
112   /// would prevent it from becoming completely dead, it gives up early,
113   /// potentially leaving some dead constant users around.
114   void removeDeadConstantUsers();
115
116   // Methods for support type inquiry through isa, cast, and dyn_cast:
117   static inline bool classof(const GlobalValue *) { return true; }
118   static inline bool classof(const Value *V) {
119     return V->getValueType() == Value::FunctionVal ||
120            V->getValueType() == Value::GlobalVariableVal;
121   }
122 };
123
124 } // End llvm namespace
125
126 #endif