Add TranslationUnit::getSelectorMap().

llvm-svn: 77542

2 files changed, 8 insertions, 1 deletions

diff --git a/clang/include/clang/Index/TranslationUnit.h b/clang/include/clang/Index/TranslationUnit.h
index 55417d4eb6c..bf9e78f7289 100644
--- a/clang/include/clang/Index/TranslationUnit.h
+++ b/clang/include/clang/Index/TranslationUnit.h
@@ -19,6 +19,7 @@ namespace clang {
 
 namespace idx {
   class DeclReferenceMap;
+  class SelectorMap;
 
 /// \brief Abstract interface for a translation unit.
 class TranslationUnit {
@@ -26,6 +27,7 @@ public:
   virtual ~TranslationUnit();
   virtual ASTContext &getASTContext() = 0;
   virtual DeclReferenceMap &getDeclReferenceMap() = 0;
+  virtual SelectorMap &getSelectorMap() = 0;
 };
 
 } // namespace idx
diff --git a/clang/tools/index-test/index-test.cpp b/clang/tools/index-test/index-test.cpp
index a0d60f7b492..037221fcef5 100644
--- a/clang/tools/index-test/index-test.cpp
+++ b/clang/tools/index-test/index-test.cpp
@@ -38,6 +38,7 @@
 #include "clang/Index/TranslationUnit.h"
 #include "clang/Index/ASTLocation.h"
 #include "clang/Index/DeclReferenceMap.h"
+#include "clang/Index/SelectorMap.h"
 #include "clang/Index/Handlers.h"
 #include "clang/Index/Analyzer.h"
 #include "clang/Index/Utils.h"
@@ -58,14 +59,18 @@ using namespace idx;
 class TUnit : public TranslationUnit {
 public:
   TUnit(ASTUnit *ast, const std::string &filename)
-    : AST(ast), Filename(filename), DeclRefMap(ast->getASTContext()) { }
+    : AST(ast), Filename(filename),
+      DeclRefMap(ast->getASTContext()),
+      SelMap(ast->getASTContext()) { }
   
   virtual ASTContext &getASTContext() { return AST->getASTContext(); }
   virtual DeclReferenceMap &getDeclReferenceMap() { return DeclRefMap; }
+  virtual SelectorMap &getSelectorMap() { return SelMap; }
   
   llvm::OwningPtr<ASTUnit> AST;
   std::string Filename;
   DeclReferenceMap DeclRefMap;
+  SelectorMap SelMap;
 };
 
 static llvm::cl::list<ParsedSourceLocation>