1 //===-- CGBlocks.h - state for LLVM CodeGen for blocks ----------*- C++ -*-===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 // This is the internal state used for llvm translation for block literals.
12 //===----------------------------------------------------------------------===//
14 #ifndef CLANG_CODEGEN_CGBLOCKS_H
15 #define CLANG_CODEGEN_CGBLOCKS_H
17 #include "CodeGenTypes.h"
18 #include "clang/AST/Type.h"
19 #include "llvm/Module.h"
20 #include "llvm/ADT/SmallVector.h"
21 #include "clang/Basic/TargetInfo.h"
22 #include "clang/AST/CharUnits.h"
23 #include "clang/AST/Expr.h"
24 #include "clang/AST/ExprCXX.h"
25 #include "clang/AST/ExprObjC.h"
27 #include "CGBuilder.h"
51 BLOCK_HAS_COPY_DISPOSE = (1 << 25),
52 BLOCK_HAS_CXX_OBJ = (1 << 26),
53 BLOCK_IS_GLOBAL = (1 << 28),
54 BLOCK_USE_STRET = (1 << 29),
55 BLOCK_HAS_SIGNATURE = (1 << 30)
60 BlockFlags(uint32_t flags) : flags(flags) {}
62 BlockFlags() : flags(0) {}
63 BlockFlags(BlockFlag_t flag) : flags(flag) {}
65 uint32_t getBitMask() const { return flags; }
66 bool empty() const { return flags == 0; }
68 friend BlockFlags operator|(BlockFlags l, BlockFlags r) {
69 return BlockFlags(l.flags | r.flags);
71 friend BlockFlags &operator|=(BlockFlags &l, BlockFlags r) {
75 friend bool operator&(BlockFlags l, BlockFlags r) {
76 return (l.flags & r.flags);
79 inline BlockFlags operator|(BlockFlag_t l, BlockFlag_t r) {
80 return BlockFlags(l) | BlockFlags(r);
83 enum BlockFieldFlag_t {
84 BLOCK_FIELD_IS_OBJECT = 0x03, /* id, NSObject, __attribute__((NSObject)),
86 BLOCK_FIELD_IS_BLOCK = 0x07, /* a block variable */
88 BLOCK_FIELD_IS_BYREF = 0x08, /* the on stack structure holding the __block
90 BLOCK_FIELD_IS_WEAK = 0x10, /* declared __weak, only used in byref copy
93 BLOCK_BYREF_CALLER = 128, /* called from __block (byref) copy/dispose
95 BLOCK_BYREF_CURRENT_MAX = 256
98 class BlockFieldFlags {
101 BlockFieldFlags(uint32_t flags) : flags(flags) {}
103 BlockFieldFlags() : flags(0) {}
104 BlockFieldFlags(BlockFieldFlag_t flag) : flags(flag) {}
106 uint32_t getBitMask() const { return flags; }
107 bool empty() const { return flags == 0; }
109 /// Answers whether the flags indicate that this field is an object
110 /// or block pointer that requires _Block_object_assign/dispose.
111 bool isSpecialPointer() const { return flags & BLOCK_FIELD_IS_OBJECT; }
113 friend BlockFieldFlags operator|(BlockFieldFlags l, BlockFieldFlags r) {
114 return BlockFieldFlags(l.flags | r.flags);
116 friend BlockFieldFlags &operator|=(BlockFieldFlags &l, BlockFieldFlags r) {
120 friend bool operator&(BlockFieldFlags l, BlockFieldFlags r) {
121 return (l.flags & r.flags);
124 inline BlockFieldFlags operator|(BlockFieldFlag_t l, BlockFieldFlag_t r) {
125 return BlockFieldFlags(l) | BlockFieldFlags(r);
128 /// CGBlockInfo - Information to generate a block literal.
131 /// Name - The name of the block, kindof.
134 /// The field index of 'this' within the block, if there is one.
135 unsigned CXXThisIndex;
141 bool isIndex() const { return (Data & 1) != 0; }
142 bool isConstant() const { return !isIndex(); }
143 unsigned getIndex() const { assert(isIndex()); return Data >> 1; }
144 llvm::Value *getConstant() const {
145 assert(isConstant());
146 return reinterpret_cast<llvm::Value*>(Data);
149 static Capture makeIndex(unsigned index) {
151 v.Data = (index << 1) | 1;
155 static Capture makeConstant(llvm::Value *value) {
157 v.Data = reinterpret_cast<uintptr_t>(value);
162 /// The mapping of allocated indexes within the block.
163 llvm::DenseMap<const VarDecl*, Capture> Captures;
165 /// CanBeGlobal - True if the block can be global, i.e. it has
166 /// no non-constant captures.
167 bool CanBeGlobal : 1;
169 /// True if the block needs a custom copy or dispose function.
170 bool NeedsCopyDispose : 1;
172 /// HasCXXObject - True if the block's custom copy/dispose functions
173 /// need to be run even in GC mode.
174 bool HasCXXObject : 1;
176 const llvm::StructType *StructureType;
177 const BlockExpr *Block;
179 CharUnits BlockAlign;
181 const Capture &getCapture(const VarDecl *var) const {
182 llvm::DenseMap<const VarDecl*, Capture>::const_iterator
183 it = Captures.find(var);
184 assert(it != Captures.end() && "no entry for variable!");
188 const BlockDecl *getBlockDecl() const { return Block->getBlockDecl(); }
189 const BlockExpr *getBlockExpr() const { return Block; }
191 CGBlockInfo(const BlockExpr *blockExpr, const char *Name);
194 } // end namespace CodeGen
195 } // end namespace clang