blob: fffd011fd6b84f17a4cb5be0046537be116fdb8f [file] [log] [blame]
Anders Carlsson55085182007-08-21 17:43:55 +00001//===---- CGBuiltin.cpp - Emit LLVM Code for builtins ---------------------===//
2//
3// The LLVM Compiler Infrastructure
4//
Chris Lattner0bc735f2007-12-29 19:59:25 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Anders Carlsson55085182007-08-21 17:43:55 +00007//
8//===----------------------------------------------------------------------===//
9//
10// This contains code to emit Objective-C code as LLVM code.
11//
12//===----------------------------------------------------------------------===//
13
Ted Kremenek2979ec72008-04-09 15:51:31 +000014#include "CGObjCRuntime.h"
Anders Carlsson55085182007-08-21 17:43:55 +000015#include "CodeGenFunction.h"
16#include "CodeGenModule.h"
Daniel Dunbar85c59ed2008-08-29 08:11:39 +000017#include "clang/AST/ASTContext.h"
Daniel Dunbarc4a1dea2008-08-11 05:35:13 +000018#include "clang/AST/DeclObjC.h"
Daniel Dunbare66f4e32008-09-03 00:27:26 +000019#include "clang/Basic/Diagnostic.h"
Anders Carlsson3d8400d2008-08-30 19:51:14 +000020#include "llvm/ADT/STLExtras.h"
Daniel Dunbarc1cf4a52008-09-24 04:04:31 +000021#include "llvm/Target/TargetData.h"
Chris Lattner41110242008-06-17 18:05:57 +000022
Anders Carlsson55085182007-08-21 17:43:55 +000023using namespace clang;
24using namespace CodeGen;
25
Chris Lattner8fdf3282008-06-24 17:04:18 +000026/// Emits an instance of NSConstantString representing the object.
Daniel Dunbar71fcec92008-11-25 21:53:21 +000027llvm::Value *CodeGenFunction::EmitObjCStringLiteral(const ObjCStringLiteral *E)
28{
29 std::string String(E->getString()->getStrData(),
30 E->getString()->getByteLength());
Daniel Dunbarbbce49b2008-08-12 00:12:39 +000031 llvm::Constant *C = CGM.getObjCRuntime().GenerateConstantString(String);
Daniel Dunbared7c6182008-08-20 00:28:19 +000032 // FIXME: This bitcast should just be made an invariant on the Runtime.
Daniel Dunbarbbce49b2008-08-12 00:12:39 +000033 return llvm::ConstantExpr::getBitCast(C, ConvertType(E->getType()));
Chris Lattner8fdf3282008-06-24 17:04:18 +000034}
35
36/// Emit a selector.
37llvm::Value *CodeGenFunction::EmitObjCSelectorExpr(const ObjCSelectorExpr *E) {
38 // Untyped selector.
39 // Note that this implementation allows for non-constant strings to be passed
40 // as arguments to @selector(). Currently, the only thing preventing this
41 // behaviour is the type checking in the front end.
Daniel Dunbar208ff5e2008-08-11 18:12:00 +000042 return CGM.getObjCRuntime().GetSelector(Builder, E->getSelector());
Chris Lattner8fdf3282008-06-24 17:04:18 +000043}
44
Daniel Dunbared7c6182008-08-20 00:28:19 +000045llvm::Value *CodeGenFunction::EmitObjCProtocolExpr(const ObjCProtocolExpr *E) {
46 // FIXME: This should pass the Decl not the name.
47 return CGM.getObjCRuntime().GenerateProtocolRef(Builder, E->getProtocol());
48}
Chris Lattner8fdf3282008-06-24 17:04:18 +000049
50
Daniel Dunbar8f2926b2008-08-23 03:46:30 +000051RValue CodeGenFunction::EmitObjCMessageExpr(const ObjCMessageExpr *E) {
Chris Lattner8fdf3282008-06-24 17:04:18 +000052 // Only the lookup mechanism and first two arguments of the method
53 // implementation vary between runtimes. We can get the receiver and
54 // arguments in generic code.
55
Daniel Dunbar208ff5e2008-08-11 18:12:00 +000056 CGObjCRuntime &Runtime = CGM.getObjCRuntime();
Chris Lattner8fdf3282008-06-24 17:04:18 +000057 const Expr *ReceiverExpr = E->getReceiver();
58 bool isSuperMessage = false;
Daniel Dunbarf56f1912008-08-25 08:19:24 +000059 bool isClassMessage = false;
Chris Lattner8fdf3282008-06-24 17:04:18 +000060 // Find the receiver
61 llvm::Value *Receiver;
62 if (!ReceiverExpr) {
Daniel Dunbarddb2a3d2008-08-16 00:25:02 +000063 const ObjCInterfaceDecl *OID = E->getClassInfo().first;
64
65 // Very special case, super send in class method. The receiver is
66 // self (the class object) and the send uses super semantics.
67 if (!OID) {
Chris Lattner92e62b02008-11-20 04:42:34 +000068 assert(E->getClassName()->isStr("super") &&
Daniel Dunbarddb2a3d2008-08-16 00:25:02 +000069 "Unexpected missing class interface in message send.");
Daniel Dunbarddb2a3d2008-08-16 00:25:02 +000070 isSuperMessage = true;
Daniel Dunbarf56f1912008-08-25 08:19:24 +000071 Receiver = LoadObjCSelf();
72 } else {
73 Receiver = Runtime.GetClass(Builder, OID);
Chris Lattner8fdf3282008-06-24 17:04:18 +000074 }
Daniel Dunbarf56f1912008-08-25 08:19:24 +000075
76 isClassMessage = true;
Douglas Gregorcd9b46e2008-11-04 14:56:14 +000077 } else if (isa<ObjCSuperExpr>(E->getReceiver())) {
Chris Lattner8fdf3282008-06-24 17:04:18 +000078 isSuperMessage = true;
79 Receiver = LoadObjCSelf();
80 } else {
Daniel Dunbar2bedbf82008-08-12 05:28:47 +000081 Receiver = EmitScalarExpr(E->getReceiver());
Chris Lattner8fdf3282008-06-24 17:04:18 +000082 }
83
Daniel Dunbar19cd87e2008-08-30 03:02:31 +000084 CallArgList Args;
85 for (CallExpr::const_arg_iterator i = E->arg_begin(), e = E->arg_end();
86 i != e; ++i)
Daniel Dunbar46f45b92008-09-09 01:06:48 +000087 Args.push_back(std::make_pair(EmitAnyExprToTemp(*i), (*i)->getType()));
Daniel Dunbar19cd87e2008-08-30 03:02:31 +000088
Chris Lattner8fdf3282008-06-24 17:04:18 +000089 if (isSuperMessage) {
Chris Lattner9384c762008-06-26 04:42:20 +000090 // super is only valid in an Objective-C method
91 const ObjCMethodDecl *OMD = cast<ObjCMethodDecl>(CurFuncDecl);
Daniel Dunbar7f8ea5c2008-08-30 05:35:15 +000092 return Runtime.GenerateMessageSendSuper(*this, E->getType(),
93 E->getSelector(),
Daniel Dunbarf56f1912008-08-25 08:19:24 +000094 OMD->getClassInterface(),
95 Receiver,
Daniel Dunbar19cd87e2008-08-30 03:02:31 +000096 isClassMessage,
97 Args);
Chris Lattner8fdf3282008-06-24 17:04:18 +000098 }
Daniel Dunbar7f8ea5c2008-08-30 05:35:15 +000099 return Runtime.GenerateMessageSend(*this, E->getType(), E->getSelector(),
100 Receiver, isClassMessage, Args);
Anders Carlsson55085182007-08-21 17:43:55 +0000101}
102
Daniel Dunbaraf05bb92008-08-26 08:29:31 +0000103/// StartObjCMethod - Begin emission of an ObjCMethod. This generates
104/// the LLVM function and sets the other context used by
105/// CodeGenFunction.
Daniel Dunbaraf05bb92008-08-26 08:29:31 +0000106void CodeGenFunction::StartObjCMethod(const ObjCMethodDecl *OMD) {
Daniel Dunbar7c086512008-09-09 23:14:03 +0000107 FunctionArgList Args;
108 llvm::Function *Fn = CGM.getObjCRuntime().GenerateMethod(OMD);
Daniel Dunbarf80519b2008-09-04 23:41:35 +0000109
Daniel Dunbar7c086512008-09-09 23:14:03 +0000110 CGM.SetMethodAttributes(OMD, Fn);
Chris Lattner41110242008-06-17 18:05:57 +0000111
Daniel Dunbar7c086512008-09-09 23:14:03 +0000112 Args.push_back(std::make_pair(OMD->getSelfDecl(),
113 OMD->getSelfDecl()->getType()));
114 Args.push_back(std::make_pair(OMD->getCmdDecl(),
115 OMD->getCmdDecl()->getType()));
Chris Lattner41110242008-06-17 18:05:57 +0000116
Daniel Dunbar7c086512008-09-09 23:14:03 +0000117 for (unsigned i = 0, e = OMD->getNumParams(); i != e; ++i) {
118 ParmVarDecl *IPD = OMD->getParamDecl(i);
119 Args.push_back(std::make_pair(IPD, IPD->getType()));
Chris Lattner41110242008-06-17 18:05:57 +0000120 }
Chris Lattner41110242008-06-17 18:05:57 +0000121
Daniel Dunbar2284ac92008-10-18 18:22:23 +0000122 StartFunction(OMD, OMD->getResultType(), Fn, Args, OMD->getLocEnd());
Daniel Dunbaraf05bb92008-08-26 08:29:31 +0000123}
Daniel Dunbarb7ec2462008-08-16 03:19:19 +0000124
Daniel Dunbaraf05bb92008-08-26 08:29:31 +0000125/// Generate an Objective-C method. An Objective-C method is a C function with
126/// its pointer, name, and types registered in the class struture.
127void CodeGenFunction::GenerateObjCMethod(const ObjCMethodDecl *OMD) {
128 StartObjCMethod(OMD);
129 EmitStmt(OMD->getBody());
Daniel Dunbar2284ac92008-10-18 18:22:23 +0000130 FinishFunction(cast<CompoundStmt>(OMD->getBody())->getRBracLoc());
Daniel Dunbaraf05bb92008-08-26 08:29:31 +0000131}
132
133// FIXME: I wasn't sure about the synthesis approach. If we end up
134// generating an AST for the whole body we can just fall back to
135// having a GenerateFunction which takes the body Stmt.
136
137/// GenerateObjCGetter - Generate an Objective-C property getter
138/// function. The given Decl must be either an ObjCCategoryImplDecl
139/// or an ObjCImplementationDecl.
Fariborz Jahanianfef30b52008-12-09 20:23:04 +0000140void CodeGenFunction::GenerateObjCGetter(ObjCImplementationDecl *IMP,
141 const ObjCPropertyImplDecl *PID) {
Daniel Dunbarc1cf4a52008-09-24 04:04:31 +0000142 ObjCIvarDecl *Ivar = PID->getPropertyIvarDecl();
Daniel Dunbaraf05bb92008-08-26 08:29:31 +0000143 const ObjCPropertyDecl *PD = PID->getPropertyDecl();
144 ObjCMethodDecl *OMD = PD->getGetterMethodDecl();
145 assert(OMD && "Invalid call to generate getter (empty method)");
146 // FIXME: This is rather murky, we create this here since they will
147 // not have been created by Sema for us.
Fariborz Jahanianfef30b52008-12-09 20:23:04 +0000148 OMD->createImplicitParams(getContext(), IMP->getClassInterface());
Daniel Dunbaraf05bb92008-08-26 08:29:31 +0000149 StartObjCMethod(OMD);
150
Daniel Dunbarc1cf4a52008-09-24 04:04:31 +0000151 // Determine if we should use an objc_getProperty call for
Fariborz Jahanian447d7ae2008-12-08 23:56:17 +0000152 // this. Non-atomic properties are directly evaluated.
153 // atomic 'copy' and 'retain' properties are also directly
154 // evaluated in gc-only mode.
Daniel Dunbarc1cf4a52008-09-24 04:04:31 +0000155 if (CGM.getLangOptions().getGCMode() != LangOptions::GCOnly &&
Fariborz Jahanian447d7ae2008-12-08 23:56:17 +0000156 !(PD->getPropertyAttributes() & ObjCPropertyDecl::OBJC_PR_nonatomic) &&
157 (PD->getSetterKind() == ObjCPropertyDecl::Copy ||
158 PD->getSetterKind() == ObjCPropertyDecl::Retain)) {
Daniel Dunbarc1cf4a52008-09-24 04:04:31 +0000159 llvm::Value *GetPropertyFn =
160 CGM.getObjCRuntime().GetPropertyGetFunction();
161
162 if (!GetPropertyFn) {
163 CGM.ErrorUnsupported(PID, "Obj-C getter requiring atomic copy");
164 FinishFunction();
165 return;
166 }
167
168 // Return (ivar-type) objc_getProperty((id) self, _cmd, offset, true).
169 // FIXME: Can't this be simpler? This might even be worse than the
170 // corresponding gcc code.
171 CodeGenTypes &Types = CGM.getTypes();
172 ValueDecl *Cmd = OMD->getCmdDecl();
173 llvm::Value *CmdVal = Builder.CreateLoad(LocalDeclMap[Cmd], "cmd");
174 QualType IdTy = getContext().getObjCIdType();
175 llvm::Value *SelfAsId =
176 Builder.CreateBitCast(LoadObjCSelf(), Types.ConvertType(IdTy));
Fariborz Jahanianfef30b52008-12-09 20:23:04 +0000177 llvm::Value *Offset = EmitIvarOffset(IMP->getClassInterface(), Ivar);
Daniel Dunbarc1cf4a52008-09-24 04:04:31 +0000178 llvm::Value *True =
179 llvm::ConstantInt::get(Types.ConvertTypeForMem(getContext().BoolTy), 1);
180 CallArgList Args;
181 Args.push_back(std::make_pair(RValue::get(SelfAsId), IdTy));
182 Args.push_back(std::make_pair(RValue::get(CmdVal), Cmd->getType()));
183 Args.push_back(std::make_pair(RValue::get(Offset), getContext().LongTy));
184 Args.push_back(std::make_pair(RValue::get(True), getContext().BoolTy));
185 RValue RV = EmitCall(GetPropertyFn, PD->getType(), Args);
186 // We need to fix the type here. Ivars with copy & retain are
187 // always objects so we don't need to worry about complex or
188 // aggregates.
189 RV = RValue::get(Builder.CreateBitCast(RV.getScalarVal(),
190 Types.ConvertType(PD->getType())));
191 EmitReturnOfRValue(RV, PD->getType());
192 } else {
Fariborz Jahanianfd64bb62008-12-15 20:35:07 +0000193 FieldDecl *Field =
194 IMP->getClassInterface()->lookupFieldDeclForIvar(getContext(), Ivar);
195 LValue LV = EmitLValueForIvar(LoadObjCSelf(), Ivar, Field, 0);
Fariborz Jahanian6010bca2008-11-26 22:36:09 +0000196 if (hasAggregateLLVMType(Ivar->getType())) {
197 EmitAggregateCopy(ReturnValue, LV.getAddress(), Ivar->getType());
198 }
199 else
200 EmitReturnOfRValue(EmitLoadOfLValue(LV, Ivar->getType()),
201 PD->getType());
Daniel Dunbarc1cf4a52008-09-24 04:04:31 +0000202 }
Daniel Dunbaraf05bb92008-08-26 08:29:31 +0000203
204 FinishFunction();
205}
206
207/// GenerateObjCSetter - Generate an Objective-C property setter
208/// function. The given Decl must be either an ObjCCategoryImplDecl
209/// or an ObjCImplementationDecl.
Fariborz Jahanianfef30b52008-12-09 20:23:04 +0000210void CodeGenFunction::GenerateObjCSetter(ObjCImplementationDecl *IMP,
211 const ObjCPropertyImplDecl *PID) {
Daniel Dunbar86957eb2008-09-24 06:32:09 +0000212 ObjCIvarDecl *Ivar = PID->getPropertyIvarDecl();
Daniel Dunbaraf05bb92008-08-26 08:29:31 +0000213 const ObjCPropertyDecl *PD = PID->getPropertyDecl();
214 ObjCMethodDecl *OMD = PD->getSetterMethodDecl();
215 assert(OMD && "Invalid call to generate setter (empty method)");
216 // FIXME: This is rather murky, we create this here since they will
217 // not have been created by Sema for us.
Fariborz Jahanianfef30b52008-12-09 20:23:04 +0000218 OMD->createImplicitParams(getContext(), IMP->getClassInterface());
Daniel Dunbaraf05bb92008-08-26 08:29:31 +0000219 StartObjCMethod(OMD);
Daniel Dunbaraf05bb92008-08-26 08:29:31 +0000220
Daniel Dunbar86957eb2008-09-24 06:32:09 +0000221 bool IsCopy = PD->getSetterKind() == ObjCPropertyDecl::Copy;
222 bool IsAtomic =
223 !(PD->getPropertyAttributes() & ObjCPropertyDecl::OBJC_PR_nonatomic);
224
225 // Determine if we should use an objc_setProperty call for
226 // this. Properties with 'copy' semantics always use it, as do
227 // non-atomic properties with 'release' semantics as long as we are
228 // not in gc-only mode.
229 if (IsCopy ||
230 (CGM.getLangOptions().getGCMode() != LangOptions::GCOnly &&
231 PD->getSetterKind() == ObjCPropertyDecl::Retain)) {
232 llvm::Value *SetPropertyFn =
233 CGM.getObjCRuntime().GetPropertySetFunction();
234
235 if (!SetPropertyFn) {
236 CGM.ErrorUnsupported(PID, "Obj-C getter requiring atomic copy");
237 FinishFunction();
238 return;
239 }
240
241 // Emit objc_setProperty((id) self, _cmd, offset, arg,
242 // <is-atomic>, <is-copy>).
243 // FIXME: Can't this be simpler? This might even be worse than the
244 // corresponding gcc code.
245 CodeGenTypes &Types = CGM.getTypes();
246 ValueDecl *Cmd = OMD->getCmdDecl();
247 llvm::Value *CmdVal = Builder.CreateLoad(LocalDeclMap[Cmd], "cmd");
248 QualType IdTy = getContext().getObjCIdType();
249 llvm::Value *SelfAsId =
250 Builder.CreateBitCast(LoadObjCSelf(), Types.ConvertType(IdTy));
Fariborz Jahanianfef30b52008-12-09 20:23:04 +0000251 llvm::Value *Offset = EmitIvarOffset(IMP->getClassInterface(), Ivar);
Daniel Dunbar86957eb2008-09-24 06:32:09 +0000252 llvm::Value *Arg = LocalDeclMap[OMD->getParamDecl(0)];
253 llvm::Value *ArgAsId =
254 Builder.CreateBitCast(Builder.CreateLoad(Arg, "arg"),
255 Types.ConvertType(IdTy));
256 llvm::Value *True =
257 llvm::ConstantInt::get(Types.ConvertTypeForMem(getContext().BoolTy), 1);
258 llvm::Value *False =
259 llvm::ConstantInt::get(Types.ConvertTypeForMem(getContext().BoolTy), 0);
260 CallArgList Args;
261 Args.push_back(std::make_pair(RValue::get(SelfAsId), IdTy));
262 Args.push_back(std::make_pair(RValue::get(CmdVal), Cmd->getType()));
263 Args.push_back(std::make_pair(RValue::get(Offset), getContext().LongTy));
264 Args.push_back(std::make_pair(RValue::get(ArgAsId), IdTy));
265 Args.push_back(std::make_pair(RValue::get(IsAtomic ? True : False),
266 getContext().BoolTy));
267 Args.push_back(std::make_pair(RValue::get(IsCopy ? True : False),
268 getContext().BoolTy));
269 EmitCall(SetPropertyFn, PD->getType(), Args);
270 } else {
271 SourceLocation Loc = PD->getLocation();
272 ValueDecl *Self = OMD->getSelfDecl();
273 ObjCIvarDecl *Ivar = PID->getPropertyIvarDecl();
274 DeclRefExpr Base(Self, Self->getType(), Loc);
275 ParmVarDecl *ArgDecl = OMD->getParamDecl(0);
276 DeclRefExpr Arg(ArgDecl, ArgDecl->getType(), Loc);
Fariborz Jahanianaaa63a72008-12-13 22:20:28 +0000277 ObjCInterfaceDecl *OI = IMP->getClassInterface();
Fariborz Jahanianefc4c4b2008-12-18 17:29:46 +0000278 ObjCIvarRefExpr IvarRef(Ivar, Ivar->getType(), Loc, &Base,
Daniel Dunbar86957eb2008-09-24 06:32:09 +0000279 true, true);
Fariborz Jahanianefc4c4b2008-12-18 17:29:46 +0000280 getContext().setFieldDecl(OI, Ivar, &IvarRef);
Daniel Dunbar86957eb2008-09-24 06:32:09 +0000281 BinaryOperator Assign(&IvarRef, &Arg, BinaryOperator::Assign,
282 Ivar->getType(), Loc);
283 EmitStmt(&Assign);
284 }
Daniel Dunbaraf05bb92008-08-26 08:29:31 +0000285
286 FinishFunction();
Chris Lattner41110242008-06-17 18:05:57 +0000287}
288
Daniel Dunbarc1cf4a52008-09-24 04:04:31 +0000289llvm::Value *CodeGenFunction::LoadObjCSelf() {
Daniel Dunbarb7ec2462008-08-16 03:19:19 +0000290 const ObjCMethodDecl *OMD = cast<ObjCMethodDecl>(CurFuncDecl);
291 return Builder.CreateLoad(LocalDeclMap[OMD->getSelfDecl()], "self");
Chris Lattner41110242008-06-17 18:05:57 +0000292}
293
Fariborz Jahanian5daf5702008-11-22 18:39:36 +0000294RValue CodeGenFunction::EmitObjCPropertyGet(const Expr *Exp) {
Fariborz Jahanian43f44702008-11-22 22:30:21 +0000295 // FIXME: Split it into two separate routines.
Fariborz Jahanian5daf5702008-11-22 18:39:36 +0000296 if (const ObjCPropertyRefExpr *E = dyn_cast<ObjCPropertyRefExpr>(Exp)) {
297 Selector S = E->getProperty()->getGetterName();
Fariborz Jahanian5daf5702008-11-22 18:39:36 +0000298 return CGM.getObjCRuntime().
Fariborz Jahanian43f44702008-11-22 22:30:21 +0000299 GenerateMessageSend(*this, Exp->getType(), S,
300 EmitScalarExpr(E->getBase()),
301 false, CallArgList());
Fariborz Jahanian5daf5702008-11-22 18:39:36 +0000302 }
Fariborz Jahanian43f44702008-11-22 22:30:21 +0000303 else if (const ObjCKVCRefExpr *E = dyn_cast<ObjCKVCRefExpr>(Exp)) {
304 Selector S = E->getGetterMethod()->getSelector();
305 return CGM.getObjCRuntime().
306 GenerateMessageSend(*this, Exp->getType(), S,
307 EmitScalarExpr(E->getBase()),
308 false, CallArgList());
309 }
310 else
311 assert (0 && "bad expression node in EmitObjCPropertyGet");
Daniel Dunbar9c3fc702008-08-27 06:57:25 +0000312}
313
Fariborz Jahanian43f44702008-11-22 22:30:21 +0000314void CodeGenFunction::EmitObjCPropertySet(const Expr *Exp,
Daniel Dunbar85c59ed2008-08-29 08:11:39 +0000315 RValue Src) {
Fariborz Jahanian43f44702008-11-22 22:30:21 +0000316 // FIXME: Split it into two separate routines.
317 if (const ObjCPropertyRefExpr *E = dyn_cast<ObjCPropertyRefExpr>(Exp)) {
318 Selector S = E->getProperty()->getSetterName();
319 CallArgList Args;
320 Args.push_back(std::make_pair(Src, E->getType()));
321 CGM.getObjCRuntime().GenerateMessageSend(*this, getContext().VoidTy, S,
322 EmitScalarExpr(E->getBase()),
323 false, Args);
324 }
325 else if (const ObjCKVCRefExpr *E = dyn_cast<ObjCKVCRefExpr>(Exp)) {
326 Selector S = E->getSetterMethod()->getSelector();
327 CallArgList Args;
328 Args.push_back(std::make_pair(Src, E->getType()));
329 CGM.getObjCRuntime().GenerateMessageSend(*this, getContext().VoidTy, S,
330 EmitScalarExpr(E->getBase()),
331 false, Args);
332 }
333 else
334 assert (0 && "bad expression node in EmitObjCPropertySet");
Daniel Dunbar85c59ed2008-08-29 08:11:39 +0000335}
336
Anders Carlsson3d8400d2008-08-30 19:51:14 +0000337void CodeGenFunction::EmitObjCForCollectionStmt(const ObjCForCollectionStmt &S)
338{
Daniel Dunbarc1cf4a52008-09-24 04:04:31 +0000339 llvm::Function *EnumerationMutationFn =
340 CGM.getObjCRuntime().EnumerationMutationFunction();
Anders Carlssonf484c312008-08-31 02:33:12 +0000341 llvm::Value *DeclAddress;
342 QualType ElementTy;
343
Daniel Dunbarc1cf4a52008-09-24 04:04:31 +0000344 if (!EnumerationMutationFn) {
345 CGM.ErrorUnsupported(&S, "Obj-C fast enumeration for this runtime");
346 return;
347 }
348
Anders Carlssonf484c312008-08-31 02:33:12 +0000349 if (const DeclStmt *SD = dyn_cast<DeclStmt>(S.getElement())) {
350 EmitStmt(SD);
Daniel Dunbara448fb22008-11-11 23:11:34 +0000351 assert(HaveInsertPoint() && "DeclStmt destroyed insert point!");
Ted Kremenek39741ce2008-10-06 20:59:48 +0000352 const ScopedDecl* D = SD->getSolitaryDecl();
353 ElementTy = cast<ValueDecl>(D)->getType();
354 DeclAddress = LocalDeclMap[D];
Anders Carlssonf484c312008-08-31 02:33:12 +0000355 } else {
356 ElementTy = cast<Expr>(S.getElement())->getType();
357 DeclAddress = 0;
358 }
359
360 // Fast enumeration state.
361 QualType StateTy = getContext().getObjCFastEnumerationStateType();
362 llvm::AllocaInst *StatePtr = CreateTempAlloca(ConvertType(StateTy),
363 "state.ptr");
364 StatePtr->setAlignment(getContext().getTypeAlign(StateTy) >> 3);
Anders Carlsson2abd89c2008-08-31 04:05:03 +0000365 EmitMemSetToZero(StatePtr, StateTy);
Anders Carlssonf484c312008-08-31 02:33:12 +0000366
367 // Number of elements in the items array.
Anders Carlsson2abd89c2008-08-31 04:05:03 +0000368 static const unsigned NumItems = 16;
Anders Carlssonf484c312008-08-31 02:33:12 +0000369
370 // Get selector
371 llvm::SmallVector<IdentifierInfo*, 3> II;
372 II.push_back(&CGM.getContext().Idents.get("countByEnumeratingWithState"));
373 II.push_back(&CGM.getContext().Idents.get("objects"));
374 II.push_back(&CGM.getContext().Idents.get("count"));
375 Selector FastEnumSel = CGM.getContext().Selectors.getSelector(II.size(),
376 &II[0]);
377
378 QualType ItemsTy =
379 getContext().getConstantArrayType(getContext().getObjCIdType(),
380 llvm::APInt(32, NumItems),
381 ArrayType::Normal, 0);
382 llvm::Value *ItemsPtr = CreateTempAlloca(ConvertType(ItemsTy), "items.ptr");
383
384 llvm::Value *Collection = EmitScalarExpr(S.getCollection());
385
386 CallArgList Args;
Daniel Dunbar46f45b92008-09-09 01:06:48 +0000387 Args.push_back(std::make_pair(RValue::get(StatePtr),
Anders Carlssonf484c312008-08-31 02:33:12 +0000388 getContext().getPointerType(StateTy)));
389
Daniel Dunbar46f45b92008-09-09 01:06:48 +0000390 Args.push_back(std::make_pair(RValue::get(ItemsPtr),
Anders Carlssonf484c312008-08-31 02:33:12 +0000391 getContext().getPointerType(ItemsTy)));
392
393 const llvm::Type *UnsignedLongLTy = ConvertType(getContext().UnsignedLongTy);
394 llvm::Constant *Count = llvm::ConstantInt::get(UnsignedLongLTy, NumItems);
Daniel Dunbar46f45b92008-09-09 01:06:48 +0000395 Args.push_back(std::make_pair(RValue::get(Count),
396 getContext().UnsignedLongTy));
Anders Carlssonf484c312008-08-31 02:33:12 +0000397
398 RValue CountRV =
399 CGM.getObjCRuntime().GenerateMessageSend(*this,
400 getContext().UnsignedLongTy,
401 FastEnumSel,
402 Collection, false, Args);
403
404 llvm::Value *LimitPtr = CreateTempAlloca(UnsignedLongLTy, "limit.ptr");
405 Builder.CreateStore(CountRV.getScalarVal(), LimitPtr);
406
Daniel Dunbar55e87422008-11-11 02:29:29 +0000407 llvm::BasicBlock *NoElements = createBasicBlock("noelements");
408 llvm::BasicBlock *SetStartMutations = createBasicBlock("setstartmutations");
Anders Carlssonf484c312008-08-31 02:33:12 +0000409
410 llvm::Value *Limit = Builder.CreateLoad(LimitPtr);
411 llvm::Value *Zero = llvm::Constant::getNullValue(UnsignedLongLTy);
412
413 llvm::Value *IsZero = Builder.CreateICmpEQ(Limit, Zero, "iszero");
Anders Carlsson2abd89c2008-08-31 04:05:03 +0000414 Builder.CreateCondBr(IsZero, NoElements, SetStartMutations);
Anders Carlssonf484c312008-08-31 02:33:12 +0000415
Anders Carlsson2abd89c2008-08-31 04:05:03 +0000416 EmitBlock(SetStartMutations);
417
418 llvm::Value *StartMutationsPtr =
419 CreateTempAlloca(UnsignedLongLTy);
420
421 llvm::Value *StateMutationsPtrPtr =
422 Builder.CreateStructGEP(StatePtr, 2, "mutationsptr.ptr");
423 llvm::Value *StateMutationsPtr = Builder.CreateLoad(StateMutationsPtrPtr,
424 "mutationsptr");
425
426 llvm::Value *StateMutations = Builder.CreateLoad(StateMutationsPtr,
427 "mutations");
428
429 Builder.CreateStore(StateMutations, StartMutationsPtr);
430
Daniel Dunbar55e87422008-11-11 02:29:29 +0000431 llvm::BasicBlock *LoopStart = createBasicBlock("loopstart");
Anders Carlssonf484c312008-08-31 02:33:12 +0000432 EmitBlock(LoopStart);
433
Anders Carlssonf484c312008-08-31 02:33:12 +0000434 llvm::Value *CounterPtr = CreateTempAlloca(UnsignedLongLTy, "counter.ptr");
435 Builder.CreateStore(Zero, CounterPtr);
436
Daniel Dunbar55e87422008-11-11 02:29:29 +0000437 llvm::BasicBlock *LoopBody = createBasicBlock("loopbody");
Anders Carlssonf484c312008-08-31 02:33:12 +0000438 EmitBlock(LoopBody);
439
Anders Carlsson2abd89c2008-08-31 04:05:03 +0000440 StateMutationsPtr = Builder.CreateLoad(StateMutationsPtrPtr, "mutationsptr");
441 StateMutations = Builder.CreateLoad(StateMutationsPtr, "statemutations");
442
443 llvm::Value *StartMutations = Builder.CreateLoad(StartMutationsPtr,
444 "mutations");
445 llvm::Value *MutationsEqual = Builder.CreateICmpEQ(StateMutations,
446 StartMutations,
447 "tobool");
448
449
Daniel Dunbar55e87422008-11-11 02:29:29 +0000450 llvm::BasicBlock *WasMutated = createBasicBlock("wasmutated");
451 llvm::BasicBlock *WasNotMutated = createBasicBlock("wasnotmutated");
Anders Carlsson2abd89c2008-08-31 04:05:03 +0000452
453 Builder.CreateCondBr(MutationsEqual, WasNotMutated, WasMutated);
454
455 EmitBlock(WasMutated);
456 llvm::Value *V =
457 Builder.CreateBitCast(Collection,
458 ConvertType(getContext().getObjCIdType()),
459 "tmp");
Daniel Dunbarc1cf4a52008-09-24 04:04:31 +0000460 Builder.CreateCall(EnumerationMutationFn, V);
Anders Carlsson2abd89c2008-08-31 04:05:03 +0000461
462 EmitBlock(WasNotMutated);
463
Anders Carlssonf484c312008-08-31 02:33:12 +0000464 llvm::Value *StateItemsPtr =
465 Builder.CreateStructGEP(StatePtr, 1, "stateitems.ptr");
466
467 llvm::Value *Counter = Builder.CreateLoad(CounterPtr, "counter");
468
469 llvm::Value *EnumStateItems = Builder.CreateLoad(StateItemsPtr,
470 "stateitems");
471
472 llvm::Value *CurrentItemPtr =
473 Builder.CreateGEP(EnumStateItems, Counter, "currentitem.ptr");
474
475 llvm::Value *CurrentItem = Builder.CreateLoad(CurrentItemPtr, "currentitem");
476
477 // Cast the item to the right type.
478 CurrentItem = Builder.CreateBitCast(CurrentItem,
479 ConvertType(ElementTy), "tmp");
480
481 if (!DeclAddress) {
482 LValue LV = EmitLValue(cast<Expr>(S.getElement()));
483
484 // Set the value to null.
485 Builder.CreateStore(CurrentItem, LV.getAddress());
486 } else
487 Builder.CreateStore(CurrentItem, DeclAddress);
488
489 // Increment the counter.
490 Counter = Builder.CreateAdd(Counter,
491 llvm::ConstantInt::get(UnsignedLongLTy, 1));
492 Builder.CreateStore(Counter, CounterPtr);
493
Daniel Dunbar55e87422008-11-11 02:29:29 +0000494 llvm::BasicBlock *LoopEnd = createBasicBlock("loopend");
495 llvm::BasicBlock *AfterBody = createBasicBlock("afterbody");
Anders Carlssonf484c312008-08-31 02:33:12 +0000496
Anders Carlssone21269b2008-12-13 22:52:24 +0000497 BreakContinueStack.push_back(BreakContinue(LoopEnd, AfterBody,
498 ObjCEHStack.size()));
Anders Carlssonf484c312008-08-31 02:33:12 +0000499
500 EmitStmt(S.getBody());
501
502 BreakContinueStack.pop_back();
503
504 EmitBlock(AfterBody);
505
Daniel Dunbar55e87422008-11-11 02:29:29 +0000506 llvm::BasicBlock *FetchMore = createBasicBlock("fetchmore");
Fariborz Jahanianf0906c42009-01-06 18:56:31 +0000507
508 Counter = Builder.CreateLoad(CounterPtr);
509 Limit = Builder.CreateLoad(LimitPtr);
Anders Carlssonf484c312008-08-31 02:33:12 +0000510 llvm::Value *IsLess = Builder.CreateICmpULT(Counter, Limit, "isless");
Daniel Dunbarfe2b2c02008-09-04 21:54:37 +0000511 Builder.CreateCondBr(IsLess, LoopBody, FetchMore);
Anders Carlssonf484c312008-08-31 02:33:12 +0000512
513 // Fetch more elements.
514 EmitBlock(FetchMore);
515
516 CountRV =
517 CGM.getObjCRuntime().GenerateMessageSend(*this,
518 getContext().UnsignedLongTy,
519 FastEnumSel,
520 Collection, false, Args);
521 Builder.CreateStore(CountRV.getScalarVal(), LimitPtr);
522 Limit = Builder.CreateLoad(LimitPtr);
523
524 IsZero = Builder.CreateICmpEQ(Limit, Zero, "iszero");
525 Builder.CreateCondBr(IsZero, NoElements, LoopStart);
526
527 // No more elements.
528 EmitBlock(NoElements);
529
530 if (!DeclAddress) {
531 // If the element was not a declaration, set it to be null.
532
533 LValue LV = EmitLValue(cast<Expr>(S.getElement()));
534
535 // Set the value to null.
536 Builder.CreateStore(llvm::Constant::getNullValue(ConvertType(ElementTy)),
537 LV.getAddress());
538 }
539
540 EmitBlock(LoopEnd);
Anders Carlsson3d8400d2008-08-30 19:51:14 +0000541}
542
Anders Carlsson64d5d6c2008-09-09 10:04:29 +0000543void CodeGenFunction::EmitObjCAtTryStmt(const ObjCAtTryStmt &S)
544{
Fariborz Jahanianbd71be42008-11-21 00:49:24 +0000545 CGM.getObjCRuntime().EmitTryOrSynchronizedStmt(*this, S);
Anders Carlsson64d5d6c2008-09-09 10:04:29 +0000546}
547
548void CodeGenFunction::EmitObjCAtThrowStmt(const ObjCAtThrowStmt &S)
549{
550 CGM.getObjCRuntime().EmitThrowStmt(*this, S);
551}
552
Chris Lattner10cac6f2008-11-15 21:26:17 +0000553void CodeGenFunction::EmitObjCAtSynchronizedStmt(
554 const ObjCAtSynchronizedStmt &S)
555{
Fariborz Jahanianbd71be42008-11-21 00:49:24 +0000556 CGM.getObjCRuntime().EmitTryOrSynchronizedStmt(*this, S);
Chris Lattner10cac6f2008-11-15 21:26:17 +0000557}
558
Ted Kremenek2979ec72008-04-09 15:51:31 +0000559CGObjCRuntime::~CGObjCRuntime() {}