FindSCMain.cpp

Go to the documentation of this file.

#include "FindSCMain.h"
 
#include "clang/AST/DeclCXX.h"
 
using namespace systemc_clang;
 
FindSCMain::FindSCMain(clang::TranslationUnitDecl *tuDecl,
                       llvm::raw_ostream &os)
    : os_{os}, sc_main_function_declaration_{nullptr} {
  assert(!(tuDecl == nullptr));
  TraverseDecl(tuDecl);
}
 
FindSCMain::~FindSCMain() { sc_main_function_declaration_ = nullptr; }
 
bool FindSCMain::VisitFunctionDecl(clang::FunctionDecl *function_declaration) {
  if ((function_declaration->getNameInfo().getAsString() != "sc_main") ||
      (!function_declaration->hasBody()) || (function_declaration->isMain())) {
    return true;
  }
 
  sc_main_function_declaration_ = function_declaration;
  return true;
}
 
clang::FunctionDecl *FindSCMain::getSCMainFunctionDecl() const {
  assert(sc_main_function_declaration_ != nullptr);
 
  return sc_main_function_declaration_;
}
 
bool FindSCMain::isSCMainFound() const {
  return (sc_main_function_declaration_ != nullptr);
}