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

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

diff --git a/include/clang/Index/TranslationUnit.h b/include/clang/Index/TranslationUnit.h
index 06c42dc4e57227008a3e8f79f793459b9329310e..55417d4eb6c196df0a2912641c171f60f19b9021 100644 (file)
--- a/include/clang/Index/TranslationUnit.h
+++ b/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/tools/index-test/index-test.cpp b/tools/index-test/index-test.cpp
index 78ffd37b1ef0562cb2690d637290bb4769961918..5fc6269f7a8f0b043587fc59ca156d690e3ed449 100644 (file)
--- a/tools/index-test/index-test.cpp
+++ b/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>