2007-08-22 01:43:55 +08:00
|
|
|
//===---- CGBuiltin.cpp - Emit LLVM Code for builtins ---------------------===//
|
|
|
|
//
|
|
|
|
// The LLVM Compiler Infrastructure
|
|
|
|
//
|
2007-12-30 03:59:25 +08:00
|
|
|
// This file is distributed under the University of Illinois Open Source
|
|
|
|
// License. See LICENSE.TXT for details.
|
2007-08-22 01:43:55 +08:00
|
|
|
//
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
//
|
|
|
|
// This contains code to emit Objective-C code as LLVM code.
|
|
|
|
//
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
|
2008-04-09 23:51:31 +08:00
|
|
|
#include "CGObjCRuntime.h"
|
2007-08-22 01:43:55 +08:00
|
|
|
#include "CodeGenFunction.h"
|
|
|
|
#include "CodeGenModule.h"
|
2008-08-29 16:11:39 +08:00
|
|
|
#include "clang/AST/ASTContext.h"
|
2008-08-11 13:35:13 +08:00
|
|
|
#include "clang/AST/DeclObjC.h"
|
2008-09-03 08:27:26 +08:00
|
|
|
#include "clang/Basic/Diagnostic.h"
|
2008-08-31 03:51:14 +08:00
|
|
|
#include "llvm/ADT/STLExtras.h"
|
2008-09-24 12:04:31 +08:00
|
|
|
#include "llvm/Target/TargetData.h"
|
2008-06-18 02:05:57 +08:00
|
|
|
|
2007-08-22 01:43:55 +08:00
|
|
|
using namespace clang;
|
|
|
|
using namespace CodeGen;
|
|
|
|
|
2008-06-25 01:04:18 +08:00
|
|
|
/// Emits an instance of NSConstantString representing the object.
|
2008-08-12 08:12:39 +08:00
|
|
|
llvm::Value *CodeGenFunction::EmitObjCStringLiteral(const ObjCStringLiteral *E) {
|
|
|
|
std::string String(E->getString()->getStrData(), E->getString()->getByteLength());
|
|
|
|
llvm::Constant *C = CGM.getObjCRuntime().GenerateConstantString(String);
|
2008-08-20 08:28:19 +08:00
|
|
|
// FIXME: This bitcast should just be made an invariant on the Runtime.
|
2008-08-12 08:12:39 +08:00
|
|
|
return llvm::ConstantExpr::getBitCast(C, ConvertType(E->getType()));
|
2008-06-25 01:04:18 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
/// Emit a selector.
|
|
|
|
llvm::Value *CodeGenFunction::EmitObjCSelectorExpr(const ObjCSelectorExpr *E) {
|
|
|
|
// Untyped selector.
|
|
|
|
// Note that this implementation allows for non-constant strings to be passed
|
|
|
|
// as arguments to @selector(). Currently, the only thing preventing this
|
|
|
|
// behaviour is the type checking in the front end.
|
2008-08-12 02:12:00 +08:00
|
|
|
return CGM.getObjCRuntime().GetSelector(Builder, E->getSelector());
|
2008-06-25 01:04:18 +08:00
|
|
|
}
|
|
|
|
|
2008-08-20 08:28:19 +08:00
|
|
|
llvm::Value *CodeGenFunction::EmitObjCProtocolExpr(const ObjCProtocolExpr *E) {
|
|
|
|
// FIXME: This should pass the Decl not the name.
|
|
|
|
return CGM.getObjCRuntime().GenerateProtocolRef(Builder, E->getProtocol());
|
|
|
|
}
|
2008-06-25 01:04:18 +08:00
|
|
|
|
|
|
|
|
2008-08-23 11:46:30 +08:00
|
|
|
RValue CodeGenFunction::EmitObjCMessageExpr(const ObjCMessageExpr *E) {
|
2008-06-25 01:04:18 +08:00
|
|
|
// Only the lookup mechanism and first two arguments of the method
|
|
|
|
// implementation vary between runtimes. We can get the receiver and
|
|
|
|
// arguments in generic code.
|
|
|
|
|
2008-08-12 02:12:00 +08:00
|
|
|
CGObjCRuntime &Runtime = CGM.getObjCRuntime();
|
2008-06-25 01:04:18 +08:00
|
|
|
const Expr *ReceiverExpr = E->getReceiver();
|
|
|
|
bool isSuperMessage = false;
|
2008-08-25 16:19:24 +08:00
|
|
|
bool isClassMessage = false;
|
2008-06-25 01:04:18 +08:00
|
|
|
// Find the receiver
|
|
|
|
llvm::Value *Receiver;
|
|
|
|
if (!ReceiverExpr) {
|
2008-08-16 08:25:02 +08:00
|
|
|
const ObjCInterfaceDecl *OID = E->getClassInfo().first;
|
|
|
|
|
|
|
|
// Very special case, super send in class method. The receiver is
|
|
|
|
// self (the class object) and the send uses super semantics.
|
|
|
|
if (!OID) {
|
|
|
|
assert(!strcmp(E->getClassName()->getName(), "super") &&
|
|
|
|
"Unexpected missing class interface in message send.");
|
|
|
|
isSuperMessage = true;
|
2008-08-25 16:19:24 +08:00
|
|
|
Receiver = LoadObjCSelf();
|
|
|
|
} else {
|
|
|
|
Receiver = Runtime.GetClass(Builder, OID);
|
2008-06-25 01:04:18 +08:00
|
|
|
}
|
2008-08-25 16:19:24 +08:00
|
|
|
|
|
|
|
isClassMessage = true;
|
2008-11-04 22:56:14 +08:00
|
|
|
} else if (isa<ObjCSuperExpr>(E->getReceiver())) {
|
2008-06-25 01:04:18 +08:00
|
|
|
isSuperMessage = true;
|
|
|
|
Receiver = LoadObjCSelf();
|
|
|
|
} else {
|
2008-08-12 13:28:47 +08:00
|
|
|
Receiver = EmitScalarExpr(E->getReceiver());
|
2008-06-25 01:04:18 +08:00
|
|
|
}
|
|
|
|
|
2008-08-30 11:02:31 +08:00
|
|
|
CallArgList Args;
|
|
|
|
for (CallExpr::const_arg_iterator i = E->arg_begin(), e = E->arg_end();
|
|
|
|
i != e; ++i)
|
2008-09-09 09:06:48 +08:00
|
|
|
Args.push_back(std::make_pair(EmitAnyExprToTemp(*i), (*i)->getType()));
|
2008-08-30 11:02:31 +08:00
|
|
|
|
2008-06-25 01:04:18 +08:00
|
|
|
if (isSuperMessage) {
|
2008-06-26 12:42:20 +08:00
|
|
|
// super is only valid in an Objective-C method
|
|
|
|
const ObjCMethodDecl *OMD = cast<ObjCMethodDecl>(CurFuncDecl);
|
2008-08-30 13:35:15 +08:00
|
|
|
return Runtime.GenerateMessageSendSuper(*this, E->getType(),
|
|
|
|
E->getSelector(),
|
2008-08-25 16:19:24 +08:00
|
|
|
OMD->getClassInterface(),
|
|
|
|
Receiver,
|
2008-08-30 11:02:31 +08:00
|
|
|
isClassMessage,
|
|
|
|
Args);
|
2008-06-25 01:04:18 +08:00
|
|
|
}
|
2008-08-30 13:35:15 +08:00
|
|
|
return Runtime.GenerateMessageSend(*this, E->getType(), E->getSelector(),
|
|
|
|
Receiver, isClassMessage, Args);
|
2007-08-22 01:43:55 +08:00
|
|
|
}
|
|
|
|
|
2008-08-26 16:29:31 +08:00
|
|
|
/// StartObjCMethod - Begin emission of an ObjCMethod. This generates
|
|
|
|
/// the LLVM function and sets the other context used by
|
|
|
|
/// CodeGenFunction.
|
|
|
|
void CodeGenFunction::StartObjCMethod(const ObjCMethodDecl *OMD) {
|
2008-09-10 07:14:03 +08:00
|
|
|
FunctionArgList Args;
|
|
|
|
llvm::Function *Fn = CGM.getObjCRuntime().GenerateMethod(OMD);
|
2008-09-05 07:41:35 +08:00
|
|
|
|
2008-09-10 07:14:03 +08:00
|
|
|
CGM.SetMethodAttributes(OMD, Fn);
|
2008-06-18 02:05:57 +08:00
|
|
|
|
2008-09-10 07:14:03 +08:00
|
|
|
Args.push_back(std::make_pair(OMD->getSelfDecl(),
|
|
|
|
OMD->getSelfDecl()->getType()));
|
|
|
|
Args.push_back(std::make_pair(OMD->getCmdDecl(),
|
|
|
|
OMD->getCmdDecl()->getType()));
|
2008-06-18 02:05:57 +08:00
|
|
|
|
2008-09-10 07:14:03 +08:00
|
|
|
for (unsigned i = 0, e = OMD->getNumParams(); i != e; ++i) {
|
|
|
|
ParmVarDecl *IPD = OMD->getParamDecl(i);
|
|
|
|
Args.push_back(std::make_pair(IPD, IPD->getType()));
|
2008-06-18 02:05:57 +08:00
|
|
|
}
|
2008-08-16 11:19:19 +08:00
|
|
|
|
2008-10-19 02:22:23 +08:00
|
|
|
StartFunction(OMD, OMD->getResultType(), Fn, Args, OMD->getLocEnd());
|
2008-08-26 16:29:31 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
/// Generate an Objective-C method. An Objective-C method is a C function with
|
|
|
|
/// its pointer, name, and types registered in the class struture.
|
|
|
|
void CodeGenFunction::GenerateObjCMethod(const ObjCMethodDecl *OMD) {
|
|
|
|
StartObjCMethod(OMD);
|
|
|
|
EmitStmt(OMD->getBody());
|
2008-10-19 02:22:23 +08:00
|
|
|
FinishFunction(cast<CompoundStmt>(OMD->getBody())->getRBracLoc());
|
2008-08-26 16:29:31 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// FIXME: I wasn't sure about the synthesis approach. If we end up
|
|
|
|
// generating an AST for the whole body we can just fall back to
|
|
|
|
// having a GenerateFunction which takes the body Stmt.
|
|
|
|
|
|
|
|
/// GenerateObjCGetter - Generate an Objective-C property getter
|
|
|
|
/// function. The given Decl must be either an ObjCCategoryImplDecl
|
|
|
|
/// or an ObjCImplementationDecl.
|
|
|
|
void CodeGenFunction::GenerateObjCGetter(const ObjCPropertyImplDecl *PID) {
|
2008-09-24 12:04:31 +08:00
|
|
|
ObjCIvarDecl *Ivar = PID->getPropertyIvarDecl();
|
2008-08-26 16:29:31 +08:00
|
|
|
const ObjCPropertyDecl *PD = PID->getPropertyDecl();
|
|
|
|
ObjCMethodDecl *OMD = PD->getGetterMethodDecl();
|
|
|
|
assert(OMD && "Invalid call to generate getter (empty method)");
|
|
|
|
// FIXME: This is rather murky, we create this here since they will
|
|
|
|
// not have been created by Sema for us.
|
|
|
|
OMD->createImplicitParams(getContext());
|
|
|
|
StartObjCMethod(OMD);
|
|
|
|
|
2008-09-24 12:04:31 +08:00
|
|
|
// Determine if we should use an objc_getProperty call for
|
|
|
|
// this. Non-atomic and properties with assign semantics are
|
|
|
|
// directly evaluated, and in gc-only mode we don't need it at all.
|
|
|
|
if (CGM.getLangOptions().getGCMode() != LangOptions::GCOnly &&
|
|
|
|
PD->getSetterKind() != ObjCPropertyDecl::Assign &&
|
|
|
|
!(PD->getPropertyAttributes() & ObjCPropertyDecl::OBJC_PR_nonatomic)) {
|
|
|
|
llvm::Value *GetPropertyFn =
|
|
|
|
CGM.getObjCRuntime().GetPropertyGetFunction();
|
|
|
|
|
|
|
|
if (!GetPropertyFn) {
|
|
|
|
CGM.ErrorUnsupported(PID, "Obj-C getter requiring atomic copy");
|
|
|
|
FinishFunction();
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Return (ivar-type) objc_getProperty((id) self, _cmd, offset, true).
|
|
|
|
// FIXME: Can't this be simpler? This might even be worse than the
|
|
|
|
// corresponding gcc code.
|
|
|
|
CodeGenTypes &Types = CGM.getTypes();
|
|
|
|
ValueDecl *Cmd = OMD->getCmdDecl();
|
|
|
|
llvm::Value *CmdVal = Builder.CreateLoad(LocalDeclMap[Cmd], "cmd");
|
|
|
|
QualType IdTy = getContext().getObjCIdType();
|
|
|
|
llvm::Value *SelfAsId =
|
|
|
|
Builder.CreateBitCast(LoadObjCSelf(), Types.ConvertType(IdTy));
|
|
|
|
llvm::Value *Offset = EmitIvarOffset(OMD->getClassInterface(), Ivar);
|
|
|
|
llvm::Value *True =
|
|
|
|
llvm::ConstantInt::get(Types.ConvertTypeForMem(getContext().BoolTy), 1);
|
|
|
|
CallArgList Args;
|
|
|
|
Args.push_back(std::make_pair(RValue::get(SelfAsId), IdTy));
|
|
|
|
Args.push_back(std::make_pair(RValue::get(CmdVal), Cmd->getType()));
|
|
|
|
Args.push_back(std::make_pair(RValue::get(Offset), getContext().LongTy));
|
|
|
|
Args.push_back(std::make_pair(RValue::get(True), getContext().BoolTy));
|
|
|
|
RValue RV = EmitCall(GetPropertyFn, PD->getType(), Args);
|
|
|
|
// We need to fix the type here. Ivars with copy & retain are
|
|
|
|
// always objects so we don't need to worry about complex or
|
|
|
|
// aggregates.
|
|
|
|
RV = RValue::get(Builder.CreateBitCast(RV.getScalarVal(),
|
|
|
|
Types.ConvertType(PD->getType())));
|
|
|
|
EmitReturnOfRValue(RV, PD->getType());
|
|
|
|
} else {
|
|
|
|
EmitReturnOfRValue(EmitLoadOfLValue(EmitLValueForIvar(LoadObjCSelf(),
|
|
|
|
Ivar, 0),
|
|
|
|
Ivar->getType()),
|
|
|
|
PD->getType());
|
|
|
|
}
|
2008-08-26 16:29:31 +08:00
|
|
|
|
|
|
|
FinishFunction();
|
|
|
|
}
|
|
|
|
|
|
|
|
/// GenerateObjCSetter - Generate an Objective-C property setter
|
|
|
|
/// function. The given Decl must be either an ObjCCategoryImplDecl
|
|
|
|
/// or an ObjCImplementationDecl.
|
|
|
|
void CodeGenFunction::GenerateObjCSetter(const ObjCPropertyImplDecl *PID) {
|
2008-09-24 14:32:09 +08:00
|
|
|
ObjCIvarDecl *Ivar = PID->getPropertyIvarDecl();
|
2008-08-26 16:29:31 +08:00
|
|
|
const ObjCPropertyDecl *PD = PID->getPropertyDecl();
|
|
|
|
ObjCMethodDecl *OMD = PD->getSetterMethodDecl();
|
|
|
|
assert(OMD && "Invalid call to generate setter (empty method)");
|
|
|
|
// FIXME: This is rather murky, we create this here since they will
|
|
|
|
// not have been created by Sema for us.
|
|
|
|
OMD->createImplicitParams(getContext());
|
|
|
|
StartObjCMethod(OMD);
|
2008-08-16 11:19:19 +08:00
|
|
|
|
2008-09-24 14:32:09 +08:00
|
|
|
bool IsCopy = PD->getSetterKind() == ObjCPropertyDecl::Copy;
|
|
|
|
bool IsAtomic =
|
|
|
|
!(PD->getPropertyAttributes() & ObjCPropertyDecl::OBJC_PR_nonatomic);
|
|
|
|
|
|
|
|
// Determine if we should use an objc_setProperty call for
|
|
|
|
// this. Properties with 'copy' semantics always use it, as do
|
|
|
|
// non-atomic properties with 'release' semantics as long as we are
|
|
|
|
// not in gc-only mode.
|
|
|
|
if (IsCopy ||
|
|
|
|
(CGM.getLangOptions().getGCMode() != LangOptions::GCOnly &&
|
|
|
|
PD->getSetterKind() == ObjCPropertyDecl::Retain)) {
|
|
|
|
llvm::Value *SetPropertyFn =
|
|
|
|
CGM.getObjCRuntime().GetPropertySetFunction();
|
|
|
|
|
|
|
|
if (!SetPropertyFn) {
|
|
|
|
CGM.ErrorUnsupported(PID, "Obj-C getter requiring atomic copy");
|
|
|
|
FinishFunction();
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Emit objc_setProperty((id) self, _cmd, offset, arg,
|
|
|
|
// <is-atomic>, <is-copy>).
|
|
|
|
// FIXME: Can't this be simpler? This might even be worse than the
|
|
|
|
// corresponding gcc code.
|
|
|
|
CodeGenTypes &Types = CGM.getTypes();
|
|
|
|
ValueDecl *Cmd = OMD->getCmdDecl();
|
|
|
|
llvm::Value *CmdVal = Builder.CreateLoad(LocalDeclMap[Cmd], "cmd");
|
|
|
|
QualType IdTy = getContext().getObjCIdType();
|
|
|
|
llvm::Value *SelfAsId =
|
|
|
|
Builder.CreateBitCast(LoadObjCSelf(), Types.ConvertType(IdTy));
|
|
|
|
llvm::Value *Offset = EmitIvarOffset(OMD->getClassInterface(), Ivar);
|
|
|
|
llvm::Value *Arg = LocalDeclMap[OMD->getParamDecl(0)];
|
|
|
|
llvm::Value *ArgAsId =
|
|
|
|
Builder.CreateBitCast(Builder.CreateLoad(Arg, "arg"),
|
|
|
|
Types.ConvertType(IdTy));
|
|
|
|
llvm::Value *True =
|
|
|
|
llvm::ConstantInt::get(Types.ConvertTypeForMem(getContext().BoolTy), 1);
|
|
|
|
llvm::Value *False =
|
|
|
|
llvm::ConstantInt::get(Types.ConvertTypeForMem(getContext().BoolTy), 0);
|
|
|
|
CallArgList Args;
|
|
|
|
Args.push_back(std::make_pair(RValue::get(SelfAsId), IdTy));
|
|
|
|
Args.push_back(std::make_pair(RValue::get(CmdVal), Cmd->getType()));
|
|
|
|
Args.push_back(std::make_pair(RValue::get(Offset), getContext().LongTy));
|
|
|
|
Args.push_back(std::make_pair(RValue::get(ArgAsId), IdTy));
|
|
|
|
Args.push_back(std::make_pair(RValue::get(IsAtomic ? True : False),
|
|
|
|
getContext().BoolTy));
|
|
|
|
Args.push_back(std::make_pair(RValue::get(IsCopy ? True : False),
|
|
|
|
getContext().BoolTy));
|
|
|
|
EmitCall(SetPropertyFn, PD->getType(), Args);
|
|
|
|
} else {
|
|
|
|
SourceLocation Loc = PD->getLocation();
|
|
|
|
ValueDecl *Self = OMD->getSelfDecl();
|
|
|
|
ObjCIvarDecl *Ivar = PID->getPropertyIvarDecl();
|
|
|
|
DeclRefExpr Base(Self, Self->getType(), Loc);
|
|
|
|
ParmVarDecl *ArgDecl = OMD->getParamDecl(0);
|
|
|
|
DeclRefExpr Arg(ArgDecl, ArgDecl->getType(), Loc);
|
|
|
|
ObjCIvarRefExpr IvarRef(Ivar, Ivar->getType(), Loc, &Base,
|
|
|
|
true, true);
|
|
|
|
BinaryOperator Assign(&IvarRef, &Arg, BinaryOperator::Assign,
|
|
|
|
Ivar->getType(), Loc);
|
|
|
|
EmitStmt(&Assign);
|
|
|
|
}
|
2008-08-26 16:29:31 +08:00
|
|
|
|
|
|
|
FinishFunction();
|
2008-06-18 02:05:57 +08:00
|
|
|
}
|
|
|
|
|
2008-09-24 12:04:31 +08:00
|
|
|
llvm::Value *CodeGenFunction::LoadObjCSelf() {
|
2008-08-16 11:19:19 +08:00
|
|
|
const ObjCMethodDecl *OMD = cast<ObjCMethodDecl>(CurFuncDecl);
|
|
|
|
return Builder.CreateLoad(LocalDeclMap[OMD->getSelfDecl()], "self");
|
2008-06-18 02:05:57 +08:00
|
|
|
}
|
|
|
|
|
2008-08-27 14:57:25 +08:00
|
|
|
RValue CodeGenFunction::EmitObjCPropertyGet(const ObjCPropertyRefExpr *E) {
|
|
|
|
// Determine getter selector.
|
|
|
|
Selector S;
|
2008-09-03 08:27:26 +08:00
|
|
|
if (E->getKind() == ObjCPropertyRefExpr::MethodRef) {
|
|
|
|
S = E->getGetterMethod()->getSelector();
|
2008-08-27 14:57:25 +08:00
|
|
|
} else {
|
2008-09-03 08:27:26 +08:00
|
|
|
S = E->getProperty()->getGetterName();
|
2008-08-27 14:57:25 +08:00
|
|
|
}
|
|
|
|
|
2008-08-30 13:35:15 +08:00
|
|
|
return CGM.getObjCRuntime().
|
|
|
|
GenerateMessageSend(*this, E->getType(), S,
|
|
|
|
EmitScalarExpr(E->getBase()),
|
|
|
|
false, CallArgList());
|
2008-08-27 14:57:25 +08:00
|
|
|
}
|
|
|
|
|
2008-08-29 16:11:39 +08:00
|
|
|
void CodeGenFunction::EmitObjCPropertySet(const ObjCPropertyRefExpr *E,
|
|
|
|
RValue Src) {
|
2008-08-30 13:35:15 +08:00
|
|
|
Selector S;
|
2008-09-03 08:27:26 +08:00
|
|
|
if (E->getKind() == ObjCPropertyRefExpr::MethodRef) {
|
|
|
|
ObjCMethodDecl *Setter = E->getSetterMethod();
|
|
|
|
|
|
|
|
if (Setter) {
|
|
|
|
S = Setter->getSelector();
|
|
|
|
} else {
|
|
|
|
// FIXME: This should be diagnosed by sema.
|
|
|
|
SourceRange Range = E->getSourceRange();
|
|
|
|
CGM.getDiags().Report(getContext().getFullLoc(E->getLocStart()),
|
|
|
|
diag::err_typecheck_assign_const, 0, 0,
|
|
|
|
&Range, 1);
|
|
|
|
return;
|
|
|
|
}
|
2008-08-30 13:35:15 +08:00
|
|
|
} else {
|
2008-09-03 08:27:26 +08:00
|
|
|
S = E->getProperty()->getSetterName();
|
2008-08-30 13:35:15 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
CallArgList Args;
|
2008-09-09 09:06:48 +08:00
|
|
|
Args.push_back(std::make_pair(Src, E->getType()));
|
2008-08-30 13:35:15 +08:00
|
|
|
CGM.getObjCRuntime().GenerateMessageSend(*this, getContext().VoidTy, S,
|
|
|
|
EmitScalarExpr(E->getBase()),
|
|
|
|
false, Args);
|
2008-08-29 16:11:39 +08:00
|
|
|
}
|
|
|
|
|
2008-08-31 03:51:14 +08:00
|
|
|
void CodeGenFunction::EmitObjCForCollectionStmt(const ObjCForCollectionStmt &S)
|
|
|
|
{
|
2008-09-24 12:04:31 +08:00
|
|
|
llvm::Function *EnumerationMutationFn =
|
|
|
|
CGM.getObjCRuntime().EnumerationMutationFunction();
|
2008-08-31 10:33:12 +08:00
|
|
|
llvm::Value *DeclAddress;
|
|
|
|
QualType ElementTy;
|
|
|
|
|
2008-09-24 12:04:31 +08:00
|
|
|
if (!EnumerationMutationFn) {
|
|
|
|
CGM.ErrorUnsupported(&S, "Obj-C fast enumeration for this runtime");
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2008-08-31 10:33:12 +08:00
|
|
|
if (const DeclStmt *SD = dyn_cast<DeclStmt>(S.getElement())) {
|
|
|
|
EmitStmt(SD);
|
2008-11-12 07:11:34 +08:00
|
|
|
assert(HaveInsertPoint() && "DeclStmt destroyed insert point!");
|
2008-10-07 04:59:48 +08:00
|
|
|
const ScopedDecl* D = SD->getSolitaryDecl();
|
|
|
|
ElementTy = cast<ValueDecl>(D)->getType();
|
|
|
|
DeclAddress = LocalDeclMap[D];
|
2008-08-31 10:33:12 +08:00
|
|
|
} else {
|
|
|
|
ElementTy = cast<Expr>(S.getElement())->getType();
|
|
|
|
DeclAddress = 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Fast enumeration state.
|
|
|
|
QualType StateTy = getContext().getObjCFastEnumerationStateType();
|
|
|
|
llvm::AllocaInst *StatePtr = CreateTempAlloca(ConvertType(StateTy),
|
|
|
|
"state.ptr");
|
|
|
|
StatePtr->setAlignment(getContext().getTypeAlign(StateTy) >> 3);
|
2008-08-31 12:05:03 +08:00
|
|
|
EmitMemSetToZero(StatePtr, StateTy);
|
2008-08-31 10:33:12 +08:00
|
|
|
|
|
|
|
// Number of elements in the items array.
|
2008-08-31 12:05:03 +08:00
|
|
|
static const unsigned NumItems = 16;
|
2008-08-31 10:33:12 +08:00
|
|
|
|
|
|
|
// Get selector
|
|
|
|
llvm::SmallVector<IdentifierInfo*, 3> II;
|
|
|
|
II.push_back(&CGM.getContext().Idents.get("countByEnumeratingWithState"));
|
|
|
|
II.push_back(&CGM.getContext().Idents.get("objects"));
|
|
|
|
II.push_back(&CGM.getContext().Idents.get("count"));
|
|
|
|
Selector FastEnumSel = CGM.getContext().Selectors.getSelector(II.size(),
|
|
|
|
&II[0]);
|
|
|
|
|
|
|
|
QualType ItemsTy =
|
|
|
|
getContext().getConstantArrayType(getContext().getObjCIdType(),
|
|
|
|
llvm::APInt(32, NumItems),
|
|
|
|
ArrayType::Normal, 0);
|
|
|
|
llvm::Value *ItemsPtr = CreateTempAlloca(ConvertType(ItemsTy), "items.ptr");
|
|
|
|
|
|
|
|
llvm::Value *Collection = EmitScalarExpr(S.getCollection());
|
|
|
|
|
|
|
|
CallArgList Args;
|
2008-09-09 09:06:48 +08:00
|
|
|
Args.push_back(std::make_pair(RValue::get(StatePtr),
|
2008-08-31 10:33:12 +08:00
|
|
|
getContext().getPointerType(StateTy)));
|
|
|
|
|
2008-09-09 09:06:48 +08:00
|
|
|
Args.push_back(std::make_pair(RValue::get(ItemsPtr),
|
2008-08-31 10:33:12 +08:00
|
|
|
getContext().getPointerType(ItemsTy)));
|
|
|
|
|
|
|
|
const llvm::Type *UnsignedLongLTy = ConvertType(getContext().UnsignedLongTy);
|
|
|
|
llvm::Constant *Count = llvm::ConstantInt::get(UnsignedLongLTy, NumItems);
|
2008-09-09 09:06:48 +08:00
|
|
|
Args.push_back(std::make_pair(RValue::get(Count),
|
|
|
|
getContext().UnsignedLongTy));
|
2008-08-31 10:33:12 +08:00
|
|
|
|
|
|
|
RValue CountRV =
|
|
|
|
CGM.getObjCRuntime().GenerateMessageSend(*this,
|
|
|
|
getContext().UnsignedLongTy,
|
|
|
|
FastEnumSel,
|
|
|
|
Collection, false, Args);
|
|
|
|
|
|
|
|
llvm::Value *LimitPtr = CreateTempAlloca(UnsignedLongLTy, "limit.ptr");
|
|
|
|
Builder.CreateStore(CountRV.getScalarVal(), LimitPtr);
|
|
|
|
|
2008-11-11 10:29:29 +08:00
|
|
|
llvm::BasicBlock *NoElements = createBasicBlock("noelements");
|
|
|
|
llvm::BasicBlock *SetStartMutations = createBasicBlock("setstartmutations");
|
2008-08-31 10:33:12 +08:00
|
|
|
|
|
|
|
llvm::Value *Limit = Builder.CreateLoad(LimitPtr);
|
|
|
|
llvm::Value *Zero = llvm::Constant::getNullValue(UnsignedLongLTy);
|
|
|
|
|
|
|
|
llvm::Value *IsZero = Builder.CreateICmpEQ(Limit, Zero, "iszero");
|
2008-08-31 12:05:03 +08:00
|
|
|
Builder.CreateCondBr(IsZero, NoElements, SetStartMutations);
|
2008-08-31 10:33:12 +08:00
|
|
|
|
2008-08-31 12:05:03 +08:00
|
|
|
EmitBlock(SetStartMutations);
|
|
|
|
|
|
|
|
llvm::Value *StartMutationsPtr =
|
|
|
|
CreateTempAlloca(UnsignedLongLTy);
|
|
|
|
|
|
|
|
llvm::Value *StateMutationsPtrPtr =
|
|
|
|
Builder.CreateStructGEP(StatePtr, 2, "mutationsptr.ptr");
|
|
|
|
llvm::Value *StateMutationsPtr = Builder.CreateLoad(StateMutationsPtrPtr,
|
|
|
|
"mutationsptr");
|
|
|
|
|
|
|
|
llvm::Value *StateMutations = Builder.CreateLoad(StateMutationsPtr,
|
|
|
|
"mutations");
|
|
|
|
|
|
|
|
Builder.CreateStore(StateMutations, StartMutationsPtr);
|
|
|
|
|
2008-11-11 10:29:29 +08:00
|
|
|
llvm::BasicBlock *LoopStart = createBasicBlock("loopstart");
|
2008-08-31 10:33:12 +08:00
|
|
|
EmitBlock(LoopStart);
|
|
|
|
|
|
|
|
llvm::Value *CounterPtr = CreateTempAlloca(UnsignedLongLTy, "counter.ptr");
|
|
|
|
Builder.CreateStore(Zero, CounterPtr);
|
|
|
|
|
2008-11-11 10:29:29 +08:00
|
|
|
llvm::BasicBlock *LoopBody = createBasicBlock("loopbody");
|
2008-08-31 10:33:12 +08:00
|
|
|
EmitBlock(LoopBody);
|
|
|
|
|
2008-08-31 12:05:03 +08:00
|
|
|
StateMutationsPtr = Builder.CreateLoad(StateMutationsPtrPtr, "mutationsptr");
|
|
|
|
StateMutations = Builder.CreateLoad(StateMutationsPtr, "statemutations");
|
|
|
|
|
|
|
|
llvm::Value *StartMutations = Builder.CreateLoad(StartMutationsPtr,
|
|
|
|
"mutations");
|
|
|
|
llvm::Value *MutationsEqual = Builder.CreateICmpEQ(StateMutations,
|
|
|
|
StartMutations,
|
|
|
|
"tobool");
|
|
|
|
|
|
|
|
|
2008-11-11 10:29:29 +08:00
|
|
|
llvm::BasicBlock *WasMutated = createBasicBlock("wasmutated");
|
|
|
|
llvm::BasicBlock *WasNotMutated = createBasicBlock("wasnotmutated");
|
2008-08-31 12:05:03 +08:00
|
|
|
|
|
|
|
Builder.CreateCondBr(MutationsEqual, WasNotMutated, WasMutated);
|
|
|
|
|
|
|
|
EmitBlock(WasMutated);
|
|
|
|
llvm::Value *V =
|
|
|
|
Builder.CreateBitCast(Collection,
|
|
|
|
ConvertType(getContext().getObjCIdType()),
|
|
|
|
"tmp");
|
2008-09-24 12:04:31 +08:00
|
|
|
Builder.CreateCall(EnumerationMutationFn, V);
|
2008-08-31 12:05:03 +08:00
|
|
|
|
|
|
|
EmitBlock(WasNotMutated);
|
|
|
|
|
2008-08-31 10:33:12 +08:00
|
|
|
llvm::Value *StateItemsPtr =
|
|
|
|
Builder.CreateStructGEP(StatePtr, 1, "stateitems.ptr");
|
|
|
|
|
|
|
|
llvm::Value *Counter = Builder.CreateLoad(CounterPtr, "counter");
|
|
|
|
|
|
|
|
llvm::Value *EnumStateItems = Builder.CreateLoad(StateItemsPtr,
|
|
|
|
"stateitems");
|
|
|
|
|
|
|
|
llvm::Value *CurrentItemPtr =
|
|
|
|
Builder.CreateGEP(EnumStateItems, Counter, "currentitem.ptr");
|
|
|
|
|
|
|
|
llvm::Value *CurrentItem = Builder.CreateLoad(CurrentItemPtr, "currentitem");
|
|
|
|
|
|
|
|
// Cast the item to the right type.
|
|
|
|
CurrentItem = Builder.CreateBitCast(CurrentItem,
|
|
|
|
ConvertType(ElementTy), "tmp");
|
|
|
|
|
|
|
|
if (!DeclAddress) {
|
|
|
|
LValue LV = EmitLValue(cast<Expr>(S.getElement()));
|
|
|
|
|
|
|
|
// Set the value to null.
|
|
|
|
Builder.CreateStore(CurrentItem, LV.getAddress());
|
|
|
|
} else
|
|
|
|
Builder.CreateStore(CurrentItem, DeclAddress);
|
|
|
|
|
|
|
|
// Increment the counter.
|
|
|
|
Counter = Builder.CreateAdd(Counter,
|
|
|
|
llvm::ConstantInt::get(UnsignedLongLTy, 1));
|
|
|
|
Builder.CreateStore(Counter, CounterPtr);
|
|
|
|
|
2008-11-11 10:29:29 +08:00
|
|
|
llvm::BasicBlock *LoopEnd = createBasicBlock("loopend");
|
|
|
|
llvm::BasicBlock *AfterBody = createBasicBlock("afterbody");
|
2008-08-31 10:33:12 +08:00
|
|
|
|
|
|
|
BreakContinueStack.push_back(BreakContinue(LoopEnd, AfterBody));
|
|
|
|
|
|
|
|
EmitStmt(S.getBody());
|
|
|
|
|
|
|
|
BreakContinueStack.pop_back();
|
|
|
|
|
|
|
|
EmitBlock(AfterBody);
|
|
|
|
|
2008-11-11 10:29:29 +08:00
|
|
|
llvm::BasicBlock *FetchMore = createBasicBlock("fetchmore");
|
2008-08-31 10:33:12 +08:00
|
|
|
|
|
|
|
llvm::Value *IsLess = Builder.CreateICmpULT(Counter, Limit, "isless");
|
2008-09-05 05:54:37 +08:00
|
|
|
Builder.CreateCondBr(IsLess, LoopBody, FetchMore);
|
2008-08-31 10:33:12 +08:00
|
|
|
|
|
|
|
// Fetch more elements.
|
|
|
|
EmitBlock(FetchMore);
|
|
|
|
|
|
|
|
CountRV =
|
|
|
|
CGM.getObjCRuntime().GenerateMessageSend(*this,
|
|
|
|
getContext().UnsignedLongTy,
|
|
|
|
FastEnumSel,
|
|
|
|
Collection, false, Args);
|
|
|
|
Builder.CreateStore(CountRV.getScalarVal(), LimitPtr);
|
|
|
|
Limit = Builder.CreateLoad(LimitPtr);
|
|
|
|
|
|
|
|
IsZero = Builder.CreateICmpEQ(Limit, Zero, "iszero");
|
|
|
|
Builder.CreateCondBr(IsZero, NoElements, LoopStart);
|
|
|
|
|
|
|
|
// No more elements.
|
|
|
|
EmitBlock(NoElements);
|
|
|
|
|
|
|
|
if (!DeclAddress) {
|
|
|
|
// If the element was not a declaration, set it to be null.
|
|
|
|
|
|
|
|
LValue LV = EmitLValue(cast<Expr>(S.getElement()));
|
|
|
|
|
|
|
|
// Set the value to null.
|
|
|
|
Builder.CreateStore(llvm::Constant::getNullValue(ConvertType(ElementTy)),
|
|
|
|
LV.getAddress());
|
|
|
|
}
|
|
|
|
|
|
|
|
EmitBlock(LoopEnd);
|
2008-08-31 03:51:14 +08:00
|
|
|
}
|
|
|
|
|
2008-09-09 18:04:29 +08:00
|
|
|
void CodeGenFunction::EmitObjCAtTryStmt(const ObjCAtTryStmt &S)
|
|
|
|
{
|
|
|
|
CGM.getObjCRuntime().EmitTryStmt(*this, S);
|
|
|
|
}
|
|
|
|
|
|
|
|
void CodeGenFunction::EmitObjCAtThrowStmt(const ObjCAtThrowStmt &S)
|
|
|
|
{
|
|
|
|
CGM.getObjCRuntime().EmitThrowStmt(*this, S);
|
|
|
|
}
|
|
|
|
|
2008-11-16 05:26:17 +08:00
|
|
|
void CodeGenFunction::EmitObjCAtSynchronizedStmt(
|
|
|
|
const ObjCAtSynchronizedStmt &S)
|
|
|
|
{
|
|
|
|
CGM.getObjCRuntime().EmitSynchronizedStmt(*this, S);
|
|
|
|
}
|
|
|
|
|
2008-04-09 23:51:31 +08:00
|
|
|
CGObjCRuntime::~CGObjCRuntime() {}
|