commit 4e1a72b37b2e2e3bce0d07362cc981843799f7cc
author Argyrios Kyrtzidis <akyrtzi@gmail.com> Tue Jun 30 02:34:53 2009 +0000
committer Argyrios Kyrtzidis <akyrtzi@gmail.com> Tue Jun 30 02:34:53 2009 +0000
tree 979e427f331fc5f34c9360af533e78bdc320e535
parent b4b64ca752f5c0e5b95ed64a94ddaeb72e0b8f5f

Decl::getTranslationUnitDecl() should return itself when the Decl is a TranslationUnitDecl.

llvm-svn: 74502

clang/lib/AST/DeclBase.cpp

diff --git a/clang/lib/AST/DeclBase.cpp b/clang/lib/AST/DeclBase.cpp
index d872eae..3d92331 100644
--- a/clang/lib/AST/DeclBase.cpp
+++ b/clang/lib/AST/DeclBase.cpp
@@ -158,6 +158,9 @@
 }
 
 TranslationUnitDecl *Decl::getTranslationUnitDecl() {
+  if (TranslationUnitDecl *TUD = dyn_cast<TranslationUnitDecl>(this))
+    return TUD;
+
   DeclContext *DC = getDeclContext();
   assert(DC && "This decl is not contained in a translation unit!");