[cfe-commits] r44681 - in /cfe/trunk: Driver/RewriteTest.cpp test/Sema/protocol-rewrite-1.m

[Fariborz Jahanian]

Author: fjahanian
Date: Fri Dec  7 12:47:10 2007
New Revision: 44681

URL: http://llvm.org/viewvc/llvm-project?rev=44681&view=rev
Log:
Patch for rewriting of @protocol.

Added:
    cfe/trunk/test/Sema/protocol-rewrite-1.m
Modified:
    cfe/trunk/Driver/RewriteTest.cpp

Modified: cfe/trunk/Driver/RewriteTest.cpp
URL: http://llvm.org/viewvc/llvm-project/cfe/trunk/Driver/RewriteTest.cpp?rev=44681&r1=44680&r2=44681&view=diff

==============================================================================
--- cfe/trunk/Driver/RewriteTest.cpp (original)
+++ cfe/trunk/Driver/RewriteTest.cpp Fri Dec  7 12:47:10 2007
@@ -50,6 +50,7 @@
     FunctionDecl *GetMetaClassFunctionDecl;
     FunctionDecl *SelGetUidFunctionDecl;
     FunctionDecl *CFStringFunctionDecl;
+    FunctionDecl *GetProtocolFunctionDecl;
       
     // ObjC string constant support.
     FileVarDecl *ConstantStringClassReference;
@@ -73,6 +74,7 @@
       GetMetaClassFunctionDecl = 0;
       SelGetUidFunctionDecl = 0;
       CFStringFunctionDecl = 0;
+      GetProtocolFunctionDecl = 0;
       ConstantStringClassReference = 0;
       NSStringRecord = 0;
       CurMethodDecl = 0;
@@ -89,10 +91,15 @@
       // declaring objc_selector outside the parameter list removes a silly
       // scope related warning...
       const char *s = "struct objc_selector; struct objc_class;\n"
-					  "#ifndef OBJC_SUPER\n"
-					  "struct objc_super { struct objc_object *o; struct objc_object *superClass; };\n"
-					  "#define OBJC_SUPER\n"
-					  "#endif\n"
+                      "#ifndef OBJC_SUPER\n"
+                      "struct objc_super { struct objc_object *o; "
+                      "struct objc_object *superClass; };\n"
+                      "#define OBJC_SUPER\n"
+                      "#endif\n"
+                      "#ifndef _REWRITER_typedef_Protocol\n"
+                      "typedef struct objc_object Protocol;\n"
+                      "#define _REWRITER_typedef_Protocol\n"
+                      "#endif\n"
                       "extern struct objc_object *objc_msgSend"
                       "(struct objc_object *, struct objc_selector *, ...);\n"
                       "extern struct objc_object *objc_msgSendSuper"
@@ -113,6 +120,7 @@
                       "extern struct objc_object *objc_exception_extract(void *);\n"
                       "extern int objc_exception_match"
                       "(struct objc_class *, struct objc_object *, ...);\n"
+                      "extern Protocol *objc_getProtocol(const char *);\n"
                       "#include <objc/objc.h>\n";
 
       Rewrite.InsertText(SourceLocation::getFileLoc(mainFileID, 0), 
@@ -152,6 +160,7 @@
     Stmt *RewriteAtSelector(ObjCSelectorExpr *Exp);
     Stmt *RewriteMessageExpr(ObjCMessageExpr *Exp);
     Stmt *RewriteObjCStringLiteral(ObjCStringLiteral *Exp);
+    Stmt *RewriteObjCProtocolExpr(ObjCProtocolExpr *Exp);
     Stmt *RewriteObjcTryStmt(ObjcAtTryStmt *S);
     Stmt *RewriteObjcCatchStmt(ObjcAtCatchStmt *S);
     Stmt *RewriteObjcFinallyStmt(ObjcAtFinallyStmt *S);
@@ -167,6 +176,7 @@
     void SynthGetMetaClassFunctionDecl();
     void SynthCFStringFunctionDecl();
     void SynthSelGetUidFunctionDecl();
+    void SynthGetProtocolFunctionDecl();
       
     // Metadata emission.
     void RewriteObjcClassMetaData(ObjcImplementationDecl *IDecl,
@@ -708,6 +718,9 @@
 
   if (ObjcAtThrowStmt *StmtThrow = dyn_cast<ObjcAtThrowStmt>(S))
     return RewriteObjcThrowStmt(StmtThrow);
+  
+  if (ObjCProtocolExpr *ProtocolExp = dyn_cast<ObjCProtocolExpr>(S))
+    return RewriteObjCProtocolExpr(ProtocolExp);
 #if 0
   if (ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(S)) {
     CastExpr *Replacement = new CastExpr(ICE->getType(), ICE->getSubExpr(), SourceLocation());
@@ -1036,6 +1049,20 @@
                                            FunctionDecl::Extern, false, 0);
 }
 
+// SynthGetProtocolFunctionDecl - Protocol objc_getProtocol(const char *proto);
+void RewriteTest::SynthGetProtocolFunctionDecl() {
+  IdentifierInfo *SelGetProtoIdent = &Context->Idents.get("objc_getProtocol");
+  llvm::SmallVector<QualType, 16> ArgTys;
+  ArgTys.push_back(Context->getPointerType(
+    Context->CharTy.getQualifiedType(QualType::Const)));
+  QualType getFuncType = Context->getFunctionType(Context->getObjcProtoType(),
+                                                  &ArgTys[0], ArgTys.size(),
+                                                  false /*isVariadic*/);
+  GetProtocolFunctionDecl = new FunctionDecl(SourceLocation(), 
+                                             SelGetProtoIdent, getFuncType,
+                                             FunctionDecl::Extern, false, 0);
+}
+
 void RewriteTest::RewriteFunctionDecl(FunctionDecl *FD) {
   // declared in <objc/objc.h>
   if (strcmp(FD->getName(), "sel_registerName") == 0) {
@@ -1555,6 +1582,27 @@
   return CE;
 }
 
+/// RewriteObjCProtocolExpr - Rewrite a protocol expression into
+/// call to objc_getProtocol("proto-name").
+Stmt *RewriteTest::RewriteObjCProtocolExpr(ObjCProtocolExpr *Exp) {
+  if (!GetProtocolFunctionDecl)
+    SynthGetProtocolFunctionDecl();
+  // Create a call to objc_getProtocol("ProtocolName").
+  llvm::SmallVector<Expr*, 8> ProtoExprs;
+  QualType argType = Context->getPointerType(Context->CharTy);
+  ProtoExprs.push_back(new StringLiteral(Exp->getProtocol()->getName(),
+                                       strlen(Exp->getProtocol()->getName()),
+                                       false, argType, SourceLocation(),
+                                       SourceLocation()));
+  CallExpr *ProtoExp = SynthesizeCallToFunctionDecl(GetProtocolFunctionDecl,
+                                                    &ProtoExprs[0], 
+                                                    ProtoExprs.size());
+  Rewrite.ReplaceStmt(Exp, ProtoExp);
+  delete Exp;
+  return ProtoExp;
+  
+}
+
 /// SynthesizeObjcInternalStruct - Rewrite one internal struct corresponding to
 /// an objective-c class with ivars.
 void RewriteTest::SynthesizeObjcInternalStruct(ObjcInterfaceDecl *CDecl,

Added: cfe/trunk/test/Sema/protocol-rewrite-1.m
URL: http://llvm.org/viewvc/llvm-project/cfe/trunk/test/Sema/protocol-rewrite-1.m?rev=44681&view=auto

==============================================================================
--- cfe/trunk/test/Sema/protocol-rewrite-1.m (added)
+++ cfe/trunk/test/Sema/protocol-rewrite-1.m Fri Dec  7 12:47:10 2007
@@ -0,0 +1,48 @@
+// RUN: clang -rewrite-test %s
+
+typedef struct MyWidget {
+  int a;
+} MyWidget;
+
+MyWidget gWidget = { 17 };
+
+ at protocol MyProto
+- (MyWidget *)widget;
+ at end
+
+ at interface Foo 
+ at end
+
+ at interface Bar: Foo <MyProto>
+ at end
+
+ at interface Container 
++ (MyWidget *)elementForView:(Foo *)view;
+ at end
+
+ at implementation Foo
+ at end
+
+ at implementation Bar
+- (MyWidget *)widget {
+  return &gWidget;
+}
+ at end
+
+ at implementation Container
++ (MyWidget *)elementForView:(Foo *)view
+{
+  MyWidget *widget = (void*)0;
+  if (@protocol(MyProto)) {
+    widget = [(id <MyProto>)view widget];
+  }
+  return widget;
+}
+ at end
+
+int main(void) {
+  id view;
+  MyWidget *w = [Container elementForView: view];
+
+  return 0;
+}