Output: Factor out start decl addition

1 files changed, 11 insertions, 2 deletions

diff --git a/src/Output.cxx b/src/Output.cxx
index bda9a97..bb09490 100644
--- a/src/Output.cxx
+++ b/src/Output.cxx
@@ -146,6 +146,9 @@ class ASTVisitor: public ASTVisitorBase
   /** Allocate a dump node for a source file entry.  */
   unsigned int AddDumpFile(clang::FileEntry const* f);
 
+  /** Add a starting declaration for output.  */
+  void AddStartDecl(clang::Decl const* d);
+
   /** Queue leftover nodes that do not need complete output.  */
   void QueueIncompleteDumpNodes();
 
@@ -383,6 +386,12 @@ unsigned int ASTVisitor::AddDumpFile(clang::FileEntry const* f)
 }
 
 //----------------------------------------------------------------------------
+void ASTVisitor::AddStartDecl(clang::Decl const* d)
+{
+  this->AddDumpNode(d, true);
+}
+
+//----------------------------------------------------------------------------
 void ASTVisitor::QueueIncompleteDumpNodes()
 {
   // Queue declaration nodes that do not need complete output.
@@ -989,7 +998,7 @@ void ASTVisitor::LookupStart(clang::DeclContext const* dc,
   if(pos == name.npos) {
     for(clang::DeclContext::lookup_const_iterator i = r.begin(), e = r.end();
         i != e; ++i) {
-      this->AddDumpNode(*i, true);
+      this->AddStartDecl(*i);
     }
   } else {
     std::string rest = name.substr(pos+2);
@@ -1014,7 +1023,7 @@ void ASTVisitor::HandleTranslationUnit(clang::TranslationUnitDecl const* tu)
     }
   } else {
     // No start specified.  Use whole translation unit.
-    this->AddDumpNode(tu, true);
+    this->AddStartDecl(tu);
   }
 
   // Start dump with gccxml-compatible format.