commit b391e6e2a8faa67ecdb1ad5d1b86e01ec4f2aa05
author Fariborz Jahanian <fjahanian@apple.com> Wed Oct 17 16:58:11 2007 +0000
committer Fariborz Jahanian <fjahanian@apple.com> Wed Oct 17 16:58:11 2007 +0000
tree bd337adfbb798a73ac9879132f7db9c40a193a9f
parent ff08d2c2d4e1f348d27dd8c33682b42adb23e853

Implementation of AST for @protocol expression.


git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@43075 91177308-0d34-0410-b5e6-96231b3b80d8

AST/ASTContext.cpp

diff --git a/AST/ASTContext.cpp b/AST/ASTContext.cpp
index 4c06e7c..cf617d1 100644
--- a/AST/ASTContext.cpp
+++ b/AST/ASTContext.cpp
@@ -870,6 +870,17 @@
   SelStructType = rec;
 }
 
+void ASTContext::setObjcProtoType(TypedefDecl *TD)
+{
+  assert(ObjcProtoType.isNull() && "'Protocol' type already set!");
+  
+  // typedef struct Protocol Protocol;
+  ObjcProtoType = TD->getUnderlyingType();
+  // Protocol * type
+  ObjcProtoType = getPointerType(ObjcProtoType);  
+  ProtoStructType = TD->getUnderlyingType()->getAsStructureType();
+}
+
 void ASTContext::setObjcConstantStringInterface(ObjcInterfaceDecl *Decl) {
   assert(ObjcConstantStringType.isNull() && 
          "'NSConstantString' type already set!");

AST/Expr.cpp

diff --git a/AST/Expr.cpp b/AST/Expr.cpp
index 1b989a4..d691332 100644
--- a/AST/Expr.cpp
+++ b/AST/Expr.cpp
@@ -1084,6 +1084,10 @@
 Stmt::child_iterator ObjCSelectorExpr::child_begin() { return NULL; }
 Stmt::child_iterator ObjCSelectorExpr::child_end() { return NULL; }
 
+// ObjCProtocolExpr
+Stmt::child_iterator ObjCProtocolExpr::child_begin() { return NULL; }
+Stmt::child_iterator ObjCProtocolExpr::child_end() { return NULL; }
+
 // ObjCMessageExpr
 Stmt::child_iterator ObjCMessageExpr::child_begin() {
   return reinterpret_cast<Stmt**>(&SubExprs[0]);

AST/StmtDumper.cpp

diff --git a/AST/StmtDumper.cpp b/AST/StmtDumper.cpp
index 39d2eff..6288698 100644
--- a/AST/StmtDumper.cpp
+++ b/AST/StmtDumper.cpp
@@ -14,6 +14,7 @@
 
 #include "clang/AST/StmtVisitor.h"
 #include "clang/AST/Decl.h"
+#include "clang/AST/DeclObjc.h"
 #include "clang/AST/ExprCXX.h"
 #include "clang/Basic/IdentifierTable.h"
 #include "clang/Basic/SourceManager.h"
@@ -129,6 +130,7 @@
     // ObjC
     void VisitObjCEncodeExpr(ObjCEncodeExpr *Node);
     void VisitObjCSelectorExpr(ObjCSelectorExpr *Node);
+    void VisitObjCProtocolExpr(ObjCProtocolExpr *Node);
   };
 }
 
@@ -415,6 +417,12 @@
   fprintf(F, "%s", selector.getName().c_str());
 }
 
+void StmtDumper::VisitObjCProtocolExpr(ObjCProtocolExpr *Node) {
+  DumpExpr(Node);
+  
+  fprintf(F, " ");
+  fprintf(F, "%s", Node->getProtocol()->getName());
+}
 //===----------------------------------------------------------------------===//
 // Stmt method implementations
 //===----------------------------------------------------------------------===//

AST/StmtPrinter.cpp

diff --git a/AST/StmtPrinter.cpp b/AST/StmtPrinter.cpp
index a19ad37..90c5ef2 100644
--- a/AST/StmtPrinter.cpp
+++ b/AST/StmtPrinter.cpp
@@ -14,6 +14,7 @@
 
 #include "clang/AST/StmtVisitor.h"
 #include "clang/AST/Decl.h"
+#include "clang/AST/DeclObjc.h"
 #include "clang/AST/ExprCXX.h"
 #include "clang/AST/PrettyPrinter.h"
 #include "clang/Basic/IdentifierTable.h"
@@ -627,6 +628,12 @@
   OS << ")";
 }
 
+void StmtPrinter::VisitObjCProtocolExpr(ObjCProtocolExpr *Node) {
+  OS << "@protocol(";
+  OS << Node->getProtocol()->getName();
+  OS << ")";
+}
+
 void StmtPrinter::VisitObjCMessageExpr(ObjCMessageExpr *Mess) {
   OS << "[";
   Expr *receiver = Mess->getReceiver();