add argv parsing

Author: Liu Yafei

CMakeLists.txt

@@ -1,23 +1,25 @@
 cmake_minimum_required(VERSION 3.12)
-project(myccpiler)
+project(mycc)
 
 set(CMAKE_CXX_STANDARD 14)
 #set (CMAKE_C_COMPILER clang)
 #set (CMAKE_CXX_COMPILER clang++)
 
 find_package(LLVM REQUIRED CONFIG)
+find_package(Boost REQUIRED COMPONENTS program_options)
 
-add_executable(myccpiler
+add_executable(mycc
         main.cpp
         src/lex/lex.cpp
         src/parser/parser.cpp
         src/sema/ast.cpp src/sema/symbol_tables.cpp src/sema/types.cpp includes/sema/qualifiedType.h src/sema/qualifiedType.cpp src/sema/value.cpp)
-target_include_directories(myccpiler PUBLIC includes)
+target_include_directories(mycc PUBLIC includes)
 
 llvm_map_components_to_libnames(llvm_libs support core irreader mc mcjit native interpreter)
-target_include_directories(myccpiler PUBLIC ${LLVM_INCLUDE_DIRS})
-target_compile_definitions(myccpiler PUBLIC ${LLVM_DEFINITIONS})
-target_link_libraries(myccpiler ${llvm_libs})
+target_include_directories(mycc PUBLIC ${LLVM_INCLUDE_DIRS})
+target_compile_definitions(mycc PUBLIC ${LLVM_DEFINITIONS})
+target_link_libraries(mycc ${llvm_libs})
+target_link_libraries(mycc Boost::program_options)
 
 add_executable(first_sets src/tools/main.cpp src/tools/first_set_generator.cpp)
-target_include_directories(first_sets PUBLIC includes)
+target_include_directories(first_sets PUBLIC includes)

main.cpp

@@ -10,47 +10,100 @@
 #include "llvm/Support/TargetSelect.h"
 #include "llvm/IR/Verifier.h"
 #include "llvm/Support/Error.h"
+#include <boost/program_options.hpp>
+
+namespace po = boost::program_options;
+
+int main(int argc, char *argv[]) {
+  std::string inputName;
+  bool printAST = false;
+  bool printIR = false;
+  bool execute = false;
+  unsigned int ac = 0;
+  std::string av{};
+
+  po::options_description desc("Allowed options");
+  desc.add_options()
+      ("help", "produce help message")
+      ("print-ast", "print the abstract syntax tree")
+      ("print-ir", "print the generated IR code")
+      ("execute", "execute main function with parameters")
+      ("input-file", po::value<std::vector<std::string> >(), "input file");
+  po::positional_options_description p;
+  p.add("input-file", -1);
+
+  po::variables_map vm;
+  po::store(po::command_line_parser(argc, argv).options(desc).positional(p).run(), vm);
+  po::notify(vm);
+
+  if (vm.empty()) {
+    std::cout << "no input files" << std::endl;
+    return 0;
+  }
+
+  if (vm.count("help")) {
+    std::cout << "Usage: mycc [options] file..." << std::endl;
+    std::cout << desc << std::endl;
+    return 0;
+  }
+
+  if (vm.count("input-file")) {
+    auto v = vm["input-file"].as<std::vector<std::string>>();
+    if (v.size() > 1) {
+      std::cout << "multiple input files detected, currently do not support";
+    }
+    inputName = v[0];
+  }
+
+  if (vm.count("print-ast")) {
+    printAST = true;
+  }
+
+  if (vm.count("print-ir")) {
+    printIR = true;
+  }
+
+  if (vm.count("execute")) {
+    execute = true;
+  }
 
-int main() {
   llvm::InitializeNativeTarget();
   llvm::InitializeNativeTargetAsmPrinter();
   llvm::InitializeNativeTargetAsmParser();
 
-
-//  try {
   std::ifstream testFile;
-  testFile.open("test.c");
+  testFile.open(inputName.c_str());
   auto parser = Parser(testFile);
   auto tr = parser.parseTranslationUnit();
-//    tr->print(0);
+  if (printAST) {
+    tr->print(0);
+  }
   tr->codegen();
-  AST::getModule()->print(llvm::outs(), nullptr);
-//  } catch (const SemaException &e) {
-//    std::cerr << e.what() << std::endl;
-//  } catch (const ParserException &e) {
-//    std::cerr << e.what() << std::endl;
-//  }
+  if (printIR) {
+    AST::getModule()->print(llvm::outs(), nullptr);
+  }
 
   if (llvm::verifyModule(*AST::getModule(), &llvm::errs())) {
     return 0;
-  } else {
-    std::cout << "Verified success\n";
-  }
-
-  std::string error;
-  auto engine = llvm::EngineBuilder(AST::takeModule())
-      .setErrorStr(&error)
-      .setOptLevel(llvm::CodeGenOpt::Aggressive)
-      .setEngineKind(llvm::EngineKind::JIT)
-      .create();
-  if (!engine) {
-    std::cerr << "EE Error: " << error << '\n';
-    return 1;
-  }
-
-  engine->finalizeObject();
-  typedef void (*Function)();
-  Function f = reinterpret_cast<Function>(
-      engine->getPointerToNamedFunction("main"));
-  f();
+  }
+
+  if (execute) {
+    std::string error;
+    auto engine = llvm::EngineBuilder(AST::takeModule())
+        .setErrorStr(&error)
+        .setOptLevel(llvm::CodeGenOpt::Aggressive)
+        .setEngineKind(llvm::EngineKind::JIT)
+        .create();
+    if (!engine) {
+      std::cerr << "EE Error: " << error << '\n';
+      return 1;
+    }
+
+    engine->finalizeObject();
+    typedef int (*Function)();
+    Function f = reinterpret_cast<Function>(
+        engine->getPointerToNamedFunction("main"));
+    f();
+  }
+  return 0;
 }