Add support for having different alignment for objects on call frames.
[oota-llvm.git] / include / llvm / Target / TargetData.h
1 //===-- llvm/Target/TargetData.h - Data size & alignment info ---*- 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 defines target properties related to datatype size/offset/alignment
11 // information.  It uses lazy annotations to cache information about how
12 // structure types are laid out and used.
13 //
14 // This structure should be created once, filled in if the defaults are not
15 // correct and then passed around by const&.  None of the members functions
16 // require modification to the object.
17 //
18 //===----------------------------------------------------------------------===//
19
20 #ifndef LLVM_TARGET_TARGETDATA_H
21 #define LLVM_TARGET_TARGETDATA_H
22
23 #include "llvm/Pass.h"
24 #include "llvm/Support/DataTypes.h"
25 #include "llvm/ADT/SmallVector.h"
26 #include <string>
27
28 namespace llvm {
29
30 class Value;
31 class Type;
32 class StructType;
33 class StructLayout;
34 class GlobalVariable;
35
36 /// Enum used to categorize the alignment types stored by TargetAlignElem
37 enum AlignTypeEnum {
38   INTEGER_ALIGN = 'i',               ///< Integer type alignment
39   VECTOR_ALIGN = 'v',                ///< Vector type alignment
40   FLOAT_ALIGN = 'f',                 ///< Floating point type alignment
41   AGGREGATE_ALIGN = 'a',             ///< Aggregate alignment
42   STACK_ALIGN = 's'                  ///< Stack objects alignment
43 };
44 /// Target alignment element.
45 ///
46 /// Stores the alignment data associated with a given alignment type (pointer,
47 /// integer, vector, float) and type bit width.
48 ///
49 /// @note The unusual order of elements in the structure attempts to reduce
50 /// padding and make the structure slightly more cache friendly.
51 struct TargetAlignElem {
52   AlignTypeEnum       AlignType : 8;  //< Alignment type (AlignTypeEnum)
53   unsigned char       ABIAlign;       //< ABI alignment for this type/bitw
54   unsigned char       PrefAlign;      //< Pref. alignment for this type/bitw
55   uint32_t            TypeBitWidth;   //< Type bit width
56
57   /// Initializer
58   static TargetAlignElem get(AlignTypeEnum align_type, unsigned char abi_align,
59                              unsigned char pref_align, uint32_t bit_width);
60   /// Equality predicate
61   bool operator==(const TargetAlignElem &rhs) const;
62   /// output stream operator
63   std::ostream &dump(std::ostream &os) const;
64 };
65
66 class TargetData : public ImmutablePass {
67 private:
68   bool          LittleEndian;          ///< Defaults to false
69   unsigned char PointerMemSize;        ///< Pointer size in bytes
70   unsigned char PointerABIAlign;       ///< Pointer ABI alignment
71   unsigned char PointerPrefAlign;      ///< Pointer preferred alignment
72
73   //! Where the primitive type alignment data is stored.
74   /*!
75    @sa init().
76    @note Could support multiple size pointer alignments, e.g., 32-bit pointers
77    vs. 64-bit pointers by extending TargetAlignment, but for now, we don't.
78    */
79   SmallVector<TargetAlignElem, 16> Alignments;
80   //! Alignment iterator shorthand
81   typedef SmallVector<TargetAlignElem, 16>::iterator align_iterator;
82   //! Constant alignment iterator shorthand
83   typedef SmallVector<TargetAlignElem, 16>::const_iterator align_const_iterator;
84   //! Invalid alignment.
85   /*!
86     This member is a signal that a requested alignment type and bit width were
87     not found in the SmallVector.
88    */
89   static const TargetAlignElem InvalidAlignmentElem;
90
91   //! Set/initialize target alignments
92   void setAlignment(AlignTypeEnum align_type, unsigned char abi_align,
93                     unsigned char pref_align, uint32_t bit_width);
94   unsigned getAlignmentInfo(AlignTypeEnum align_type, uint32_t bit_width,
95                             bool ABIAlign) const;
96   //! Internal helper method that returns requested alignment for type.
97   unsigned char getAlignment(const Type *Ty, bool abi_or_pref) const;
98
99   /// Valid alignment predicate.
100   ///
101   /// Predicate that tests a TargetAlignElem reference returned by get() against
102   /// InvalidAlignmentElem.
103   inline bool validAlignment(const TargetAlignElem &align) const {
104     return (&align != &InvalidAlignmentElem);
105   }
106
107 public:
108   /// Default ctor.
109   ///
110   /// @note This has to exist, because this is a pass, but it should never be
111   /// used.
112   TargetData() : ImmutablePass(intptr_t(&ID)) {
113     assert(0 && "ERROR: Bad TargetData ctor used.  "
114            "Tool did not specify a TargetData to use?");
115     abort();
116   }
117     
118   /// Constructs a TargetData from a specification string. See init().
119   explicit TargetData(const std::string &TargetDescription) 
120     : ImmutablePass(intptr_t(&ID)) {
121     init(TargetDescription);
122   }
123
124   /// Initialize target data from properties stored in the module.
125   explicit TargetData(const Module *M);
126
127   TargetData(const TargetData &TD) : 
128     ImmutablePass(intptr_t(&ID)),
129     LittleEndian(TD.isLittleEndian()),
130     PointerMemSize(TD.PointerMemSize),
131     PointerABIAlign(TD.PointerABIAlign),
132     PointerPrefAlign(TD.PointerPrefAlign),
133     Alignments(TD.Alignments)
134   { }
135
136   ~TargetData();  // Not virtual, do not subclass this class
137
138   //! Parse a target data layout string and initialize TargetData alignments.
139   void init(const std::string &TargetDescription);
140   
141   /// Target endianness...
142   bool          isLittleEndian()       const { return     LittleEndian; }
143   bool          isBigEndian()          const { return    !LittleEndian; }
144
145   /// getStringRepresentation - Return the string representation of the
146   /// TargetData.  This representation is in the same format accepted by the
147   /// string constructor above.
148   std::string getStringRepresentation() const;
149   /// Target pointer alignment
150   unsigned char getPointerABIAlignment() const { return PointerABIAlign; }
151   /// Return target's alignment for stack-based pointers
152   unsigned char getPointerPrefAlignment() const { return PointerPrefAlign; }
153   /// Target pointer size
154   unsigned char getPointerSize()         const { return PointerMemSize; }
155   /// Target pointer size, in bits
156   unsigned char getPointerSizeInBits()   const { return 8*PointerMemSize; }
157
158   /// getTypeSize - Return the number of bytes necessary to hold the specified
159   /// type.
160   uint64_t getTypeSize(const Type *Ty) const;
161
162   /// getTypeSizeInBits - Return the number of bits necessary to hold the
163   /// specified type.
164   uint64_t getTypeSizeInBits(const Type* Ty) const;
165
166   /// getABITypeAlignment - Return the minimum ABI-required alignment for the
167   /// specified type.
168   unsigned char getABITypeAlignment(const Type *Ty) const;
169
170   /// getCallFrameTypeAlignment - Return the minimum ABI-required alignment
171   /// for the specified type when it is part of a call frame.
172   unsigned char getCallFrameTypeAlignment(const Type *Ty) const;
173
174
175   /// getPrefTypeAlignment - Return the preferred stack/global alignment for
176   /// the specified type.
177   unsigned char getPrefTypeAlignment(const Type *Ty) const;
178
179   /// getPreferredTypeAlignmentShift - Return the preferred alignment for the
180   /// specified type, returned as log2 of the value (a shift amount).
181   ///
182   unsigned char getPreferredTypeAlignmentShift(const Type *Ty) const;
183
184   /// getIntPtrType - Return an unsigned integer type that is the same size or
185   /// greater to the host pointer size.
186   ///
187   const Type *getIntPtrType() const;
188
189   /// getIndexedOffset - return the offset from the beginning of the type for the
190   /// specified indices.  This is used to implement getelementptr.
191   ///
192   uint64_t getIndexedOffset(const Type *Ty,
193                             Value* const* Indices, unsigned NumIndices) const;
194   
195   /// getStructLayout - Return a StructLayout object, indicating the alignment
196   /// of the struct, its size, and the offsets of its fields.  Note that this
197   /// information is lazily cached.
198   const StructLayout *getStructLayout(const StructType *Ty) const;
199   
200   /// InvalidateStructLayoutInfo - TargetData speculatively caches StructLayout
201   /// objects.  If a TargetData object is alive when types are being refined and
202   /// removed, this method must be called whenever a StructType is removed to
203   /// avoid a dangling pointer in this cache.
204   void InvalidateStructLayoutInfo(const StructType *Ty) const;
205
206   /// getPreferredAlignmentLog - Return the preferred alignment of the
207   /// specified global, returned in log form.  This includes an explicitly
208   /// requested alignment (if the global has one).
209   unsigned getPreferredAlignmentLog(const GlobalVariable *GV) const;
210
211   static char ID; // Pass identification, replacement for typeid
212 };
213
214 /// StructLayout - used to lazily calculate structure layout information for a
215 /// target machine, based on the TargetData structure.
216 ///
217 class StructLayout {
218   uint64_t StructSize;
219   unsigned StructAlignment;
220   unsigned NumElements;
221   uint64_t MemberOffsets[1];  // variable sized array!
222 public:
223
224   uint64_t getSizeInBytes() const {
225     return StructSize;
226   }
227   
228   unsigned getAlignment() const {
229     return StructAlignment;
230   }
231     
232   /// getElementContainingOffset - Given a valid offset into the structure,
233   /// return the structure index that contains it.
234   ///
235   unsigned getElementContainingOffset(uint64_t Offset) const;
236
237   uint64_t getElementOffset(unsigned Idx) const {
238     assert(Idx < NumElements && "Invalid element idx!");
239     return MemberOffsets[Idx];
240   }
241   
242 private:
243   friend class TargetData;   // Only TargetData can create this class
244   StructLayout(const StructType *ST, const TargetData &TD);
245 };
246
247 } // End llvm namespace
248
249 #endif