Add getDeclReferenceMap() to the abstract interface of TranslationUnit class.

llvm-svn: 77530

2 files changed, 5 insertions, 1 deletions

diff --git a/clang/include/clang/Index/TranslationUnit.h b/clang/include/clang/Index/TranslationUnit.h
index 06c42dc4e57..55417d4eb6c 100644
--- a/clang/include/clang/Index/TranslationUnit.h
+++ b/clang/include/clang/Index/TranslationUnit.h
@@ -18,12 +18,14 @@ namespace clang {
   class ASTContext;
 
 namespace idx {
+  class DeclReferenceMap;
 
 /// \brief Abstract interface for a translation unit.
 class TranslationUnit {
 public:
   virtual ~TranslationUnit();
   virtual ASTContext &getASTContext() = 0;
+  virtual DeclReferenceMap &getDeclReferenceMap() = 0;
 };
 
 } // namespace idx
diff --git a/clang/tools/index-test/index-test.cpp b/clang/tools/index-test/index-test.cpp
index 78ffd37b1ef..5fc6269f7a8 100644
--- a/clang/tools/index-test/index-test.cpp
+++ b/clang/tools/index-test/index-test.cpp
@@ -57,12 +57,14 @@ using namespace idx;
 class TUnit : public TranslationUnit {
 public:
   TUnit(ASTUnit *ast, const std::string &filename)
-    : AST(ast), Filename(filename) { }
+    : AST(ast), Filename(filename), DeclRefMap(ast->getASTContext()) { }
   
   virtual ASTContext &getASTContext() { return AST->getASTContext(); }
+  virtual DeclReferenceMap &getDeclReferenceMap() { return DeclRefMap; }
   
   llvm::OwningPtr<ASTUnit> AST;
   std::string Filename;
+  DeclReferenceMap DeclRefMap;
 };
 
 static llvm::cl::list<ParsedSourceLocation>