commit 2b8ee6c2994f738e5162ff46b638974870f51662
author Steve Naroff <snaroff@apple.com> Tue Sep 01 15:55:40 2009 +0000
committer Steve Naroff <snaroff@apple.com> Tue Sep 01 15:55:40 2009 +0000
tree e190491b93274ab2bdf97ba48caf1bc6413ead67
parent c4c65f2b696cf7a29e31f98fad6b00493f724217

Add explicit "blind" client data to callback function (since we aren't using blocks).



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

tools/CIndex/CIndex.cpp

diff --git a/tools/CIndex/CIndex.cpp b/tools/CIndex/CIndex.cpp
index f8d223a..9577391 100644
--- a/tools/CIndex/CIndex.cpp
+++ b/tools/CIndex/CIndex.cpp
@@ -28,9 +28,16 @@
 class TUVisitor : public DeclVisitor<TUVisitor> {
   CXTranslationUnit TUnit;
   CXTranslationUnitIterator Callback;
+  CXClientData CData;
+  
+  void Call(enum CXCursorKind CK, NamedDecl *ND) {
+    CXCursor C = { CK, ND };
+    Callback(TUnit, C, CData);
+  }
 public:
-  TUVisitor(CXTranslationUnit CTU, CXTranslationUnitIterator cback) : 
-    TUnit(CTU), Callback(cback) {}
+  TUVisitor(CXTranslationUnit CTU, 
+            CXTranslationUnitIterator cback, CXClientData D) : 
+    TUnit(CTU), Callback(cback), CData(D) {}
   
   void VisitTranslationUnitDecl(TranslationUnitDecl *D) {
     VisitDeclContext(dyn_cast<DeclContext>(D));
@@ -40,29 +47,23 @@
            I = DC->decls_begin(), E = DC->decls_end(); I != E; ++I)
       Visit(*I);
   }
-  void VisitTypedefDecl(TypedefDecl *ND) {
-    CXCursor C = { CXCursor_TypedefDecl, ND };
-    Callback(TUnit, C);
+  void VisitTypedefDecl(TypedefDecl *ND) { 
+    Call(CXCursor_TypedefDecl, ND); 
   }
   void VisitTagDecl(TagDecl *ND) {
-    CXCursor C = { ND->isEnum() ? CXCursor_EnumDecl : CXCursor_RecordDecl, ND };
-    Callback(TUnit, C);
+    Call(ND->isEnum() ? CXCursor_EnumDecl : CXCursor_RecordDecl, ND);
   }
   void VisitFunctionDecl(FunctionDecl *ND) {
-    CXCursor C = { CXCursor_FunctionDecl, ND };
-    Callback(TUnit, C);
+    Call(CXCursor_FunctionDecl, ND);
   }
   void VisitObjCInterfaceDecl(ObjCInterfaceDecl *ND) {
-    CXCursor C = { CXCursor_ObjCInterfaceDecl, ND };
-    Callback(TUnit, C);
+    Call(CXCursor_ObjCInterfaceDecl, ND);
   }
   void VisitObjCCategoryDecl(ObjCCategoryDecl *ND) {
-    CXCursor C = { CXCursor_ObjCCategoryDecl, ND };
-    Callback(TUnit, C);
+    Call(CXCursor_ObjCCategoryDecl, ND);
   }
   void VisitObjCProtocolDecl(ObjCProtocolDecl *ND) {
-    CXCursor C = { CXCursor_ObjCProtocolDecl, ND };
-    Callback(TUnit, C);
+    Call(CXCursor_ObjCProtocolDecl, ND);
   }
 };
 
@@ -105,13 +106,14 @@
 
 
 void clang_loadTranslationUnit(
-  CXTranslationUnit CTUnit, CXTranslationUnitIterator callback)
+  CXTranslationUnit CTUnit, CXTranslationUnitIterator callback,
+  CXClientData CData)
 {
   assert(CTUnit && "Passed null CXTranslationUnit");
   ASTUnit *CXXUnit = static_cast<ASTUnit *>(CTUnit);
   ASTContext &Ctx = CXXUnit->getASTContext();
   
-  TUVisitor DVisit(CTUnit, callback);
+  TUVisitor DVisit(CTUnit, callback, CData);
   DVisit.Visit(Ctx.getTranslationUnitDecl());
 }
 

tools/c-index-test/c-index-test.c

diff --git a/tools/c-index-test/c-index-test.c b/tools/c-index-test/c-index-test.c
index 1ef8e92..30700d8 100644
--- a/tools/c-index-test/c-index-test.c
+++ b/tools/c-index-test/c-index-test.c
@@ -1,14 +1,18 @@
+/* c-index-test.c */
 
 #include "clang-c/Index.h"
 #include <stdio.h>
 
-static void PrintDecls(CXTranslationUnit Unit, CXCursor Cursor) {
+static void PrintDecls(CXTranslationUnit Unit, CXCursor Cursor,
+                       CXClientData Filter) {
   if (clang_isDeclaration(Cursor.kind)) {
-    printf("%s => %s", clang_getKindSpelling(Cursor.kind),
-                       clang_getDeclSpelling(Cursor.decl));
-    printf(" (%s,%d:%d)\n", clang_getCursorSource(Cursor),
-                            clang_getCursorLine(Cursor),
-                            clang_getCursorColumn(Cursor));
+    if (Cursor.kind == *(enum CXCursorKind *)Filter) {
+      printf("%s => %s", clang_getKindSpelling(Cursor.kind),
+                         clang_getDeclSpelling(Cursor.decl));
+      printf(" (%s,%d:%d)\n", clang_getCursorSource(Cursor),
+                              clang_getCursorLine(Cursor),
+                              clang_getCursorColumn(Cursor));
+    }
   }
 }
 
@@ -18,6 +22,9 @@
 int main(int argc, char **argv) {
   CXIndex Idx = clang_createIndex();
   CXTranslationUnit TU = clang_createTranslationUnit(Idx, argv[1]);
-  clang_loadTranslationUnit(TU, PrintDecls);
+  
+  /* Use client data to only print ObjC interfaces */
+  enum CXCursorKind filterData = CXCursor_ObjCInterfaceDecl;
+  clang_loadTranslationUnit(TU, PrintDecls, &filterData);
   return 1;
 }