Find defines

Author: kornilova203

bindgen/CMakeLists.txt

@@ -20,6 +20,14 @@ add_executable(bindgen
   visitor/TreeVisitor.h
   visitor/TreeVisitor.cpp
   visitor/TreeConsumer.h
+  defines/DefineFinderAction.h
+  defines/DefineFinderAction.cpp
+  defines/DefineFinder.cpp
+  defines/DefineFinder.h
+  defines/DefineFinderActionFactory.cpp 
+  defines/DefineFinderActionFactory.h
+  ir/Define.h
+  ir/Define.cpp
   TypeTranslator.h
   TypeTranslator.cpp
   HeaderManager.h

bindgen/Main.cpp

@@ -1,3 +1,4 @@
+#include "defines/DefineFinderActionFactory.h"
 #include "visitor/ScalaFrontendActionFactory.h"
 #include <clang/Tooling/CommonOptionsParser.h>
 
@@ -71,8 +72,11 @@ int main(int argc, char *argv[]) {
     locations.clear();
 
     IR ir(libName, linkName, objectName, Package.getValue());
-    ScalaFrontendActionFactory actionFactory(ir);
 
+    DefineFinderActionFactory defineFinderActionFactory(ir);
+    Tool.run(&defineFinderActionFactory);
+
+    ScalaFrontendActionFactory actionFactory(ir);
     int result = Tool.run(&actionFactory);
 
     auto printLoc = PrintHeadersLocation.getValue();

bindgen/defines/DefineFinder.cpp

@@ -0,0 +1,19 @@
+#include "DefineFinder.h"
+
+#include <clang/Basic/IdentifierTable.h>
+#include <clang/Lex/MacroInfo.h>
+
+DefineFinder::DefineFinder(IR &ir, const clang::CompilerInstance &compiler)
+    : ir(ir), compiler(compiler) {}
+
+void DefineFinder::MacroDefined(const clang::Token &MacroNameTok,
+                                const clang::MacroDirective *MD) {
+    std::string name = MacroNameTok.getIdentifierInfo()->getName();
+    clang::SourceManager &sm = compiler.getSourceManager();
+    if (sm.isWrittenInMainFile(MacroNameTok.getLocation())) {
+        llvm::errs() << name << "\n";
+        for (const auto &tok : MD->getMacroInfo()->tokens()) {
+            llvm::errs() << tok.getName() << "\n";
+        }
+    }
+}

bindgen/defines/DefineFinder.h

@@ -0,0 +1,20 @@
+#ifndef SCALA_NATIVE_BINDGEN_DEFINEFINDER_H
+#define SCALA_NATIVE_BINDGEN_DEFINEFINDER_H
+
+#include "../ir/IR.h"
+#include <clang/Frontend/CompilerInstance.h>
+#include <clang/Lex/PPCallbacks.h>
+
+class DefineFinder : public clang::PPCallbacks {
+  public:
+    DefineFinder(IR &ir, const clang::CompilerInstance &compiler);
+
+    void MacroDefined(const clang::Token &MacroNameTok,
+                      const clang::MacroDirective *MD) override;
+
+  private:
+    IR &ir;
+    const clang::CompilerInstance &compiler;
+};
+
+#endif // SCALA_NATIVE_BINDGEN_DEFINEFINDER_H

bindgen/defines/DefineFinderAction.cpp

@@ -0,0 +1,3 @@
+#include "DefineFinderAction.h"
+
+DefineFinderAction::DefineFinderAction(IR &ir) : ir(ir) {}

bindgen/defines/DefineFinderAction.h

@@ -0,0 +1,25 @@
+#ifndef SCALA_NATIVE_BINDGEN_DEFINEFINDERACTION_H
+#define SCALA_NATIVE_BINDGEN_DEFINEFINDERACTION_H
+
+#include "DefineFinder.h"
+#include <clang/Frontend/FrontendActions.h>
+#include <clang/Lex/Preprocessor.h>
+
+class DefineFinderAction : public clang::PreprocessOnlyAction {
+  public:
+    explicit DefineFinderAction(IR &ir);
+
+  protected:
+    void ExecuteAction() override {
+        getCompilerInstance().getPreprocessor().addPPCallbacks(
+            std::unique_ptr<clang::PPCallbacks>(
+                new DefineFinder(ir, getCompilerInstance())));
+
+        clang::PreprocessOnlyAction::ExecuteAction();
+    }
+
+  private:
+    IR &ir;
+};
+
+#endif // SCALA_NATIVE_BINDGEN_DEFINEFINDERACTION_H

bindgen/defines/DefineFinderActionFactory.cpp

@@ -0,0 +1,8 @@
+#include "DefineFinderActionFactory.h"
+#include "DefineFinderAction.h"
+
+DefineFinderActionFactory::DefineFinderActionFactory(IR &ir) : ir(ir) {}
+
+clang::FrontendAction *DefineFinderActionFactory::create() {
+    return new DefineFinderAction(ir);
+}

bindgen/defines/DefineFinderActionFactory.h

@@ -0,0 +1,17 @@
+#ifndef SCALA_NATIVE_BINDGEN_DEFINEFINDERACTIONFACTORY_H
+#define SCALA_NATIVE_BINDGEN_DEFINEFINDERACTIONFACTORY_H
+
+#include "../ir/IR.h"
+#include <clang/Tooling/Tooling.h>
+
+class DefineFinderActionFactory : public clang::tooling::FrontendActionFactory {
+  public:
+    explicit DefineFinderActionFactory(IR &ir);
+
+    clang::FrontendAction *create() override;
+
+  private:
+    IR &ir;
+};
+
+#endif // SCALA_NATIVE_BINDGEN_DEFINEFINDERACTIONFACTORY_H

bindgen/ir/Define.cpp

@@ -0,0 +1,4 @@
+#include "Define.h"
+
+Define::Define(const std::string &name, const std::string &type)
+    : TypeAndName(name, type) {}

bindgen/ir/Define.h

@@ -0,0 +1,11 @@
+#ifndef SCALA_NATIVE_BINDGEN_DEFINE_H
+#define SCALA_NATIVE_BINDGEN_DEFINE_H
+
+#include "TypeAndName.h"
+
+class Define : TypeAndName {
+  public:
+    Define(const std::string &name, const std::string &type);
+};
+
+#endif // SCALA_NATIVE_BINDGEN_DEFINE_H