Enable visiting template instantiations.

anikau31 · Dec 20, 2019 · 830ba42 · 830ba42
1 parent 96d96e0
commit 830ba42
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/src/FindConstructor.cpp b/src/FindConstructor.cpp
@@ -19,7 +19,7 @@ FindConstructor::~FindConstructor() {
   constructor_stmt_ = nullptr;
 }
 
-bool FindConstructor::shouldVisitTemplateInstantiations() { return true; }
+bool FindConstructor::shouldVisitTemplateInstantiations() const { return true; }
 
 bool FindConstructor::VisitCXXMethodDecl(CXXMethodDecl *method_declaration) {
   os_ << "@@ CXXMethodDecl\n";

diff --git a/src/FindConstructor.h b/src/FindConstructor.h
@@ -20,7 +20,7 @@ class FindConstructor : public RecursiveASTVisitor<FindConstructor> {
   FindConstructor(CXXRecordDecl *, llvm::raw_ostream &);
   virtual ~FindConstructor();
   virtual bool VisitCXXMethodDecl(CXXMethodDecl *);
-  bool shouldVisitTemplateInstantiations();
+  bool shouldVisitTemplateInstantiations() const;
 
   Stmt *returnConstructorStmt() const;
   void dump() const;

diff --git a/src/SystemCClang.cpp b/src/SystemCClang.cpp
@@ -171,6 +171,7 @@ bool SystemCConsumer::fire() {
       //
       vector<EntryFunctionContainer *> _entryFunctionContainerVector;
       FindConstructor constructor{add_module_decl->getModuleClassDecl(), os_};
+      constructor.dump();
       add_module_decl->addConstructor(constructor.returnConstructorStmt());