Merge branch 'lab2-IRGen'

src/sysyc.cpp

@@ -52,14 +52,14 @@ void parseArgs(int argc, char **argv) {
 int main(int argc, char **argv) {
   parseArgs(argc, argv);
 
-  // open the input file
+  // 打开输入文件
   ifstream fin(argInputFile);
   if (not fin) {
     cerr << "Failed to open file " << argv[1];
     return EXIT_FAILURE;
   }
 
-  // parse sysy source to AST
+  // 解析 SysY 源码为 AST
   ANTLRInputStream input(fin);
   SysYLexer lexer(&input);
   CommonTokenStream tokens(&lexer);
@@ -70,7 +70,7 @@ int main(int argc, char **argv) {
     return EXIT_SUCCESS;
   }
 
-  // pretty format the input file
+  // 格式化输入文件
   if (argFormat) {
     ASTPrinter printer;
     printer.visitCompUnit(moduleAST);
@@ -78,26 +78,20 @@ int main(int argc, char **argv) {
   }
 
   // visit AST to generate IR
-
+  SysYIRGenerator generator;
+  generator.visitCompUnit(moduleAST);
+  auto moduleIR = generator.get();
   if (argStopAfter == "ir") {
-    SysYIRGenerator generator;
-    generator.visitCompUnit(moduleAST);
-    auto moduleIR = generator.get();
     moduleIR->print(cout);
     return EXIT_SUCCESS;
-  } else if (argStopAfter == "llvmir") {
-    LLVMIRGenerator llvmirGenerator;
-    llvmirGenerator.generateIR(moduleAST); // 使用公共接口生成 IR
-    cout << llvmirGenerator.getIR();
-    return EXIT_SUCCESS;
   }
 
-  // // generate assembly
-  // CodeGen codegen(moduleIR);
-  // string asmCode = codegen.code_gen();
-  // cout << asmCode << endl;
-  // if (argStopAfter == "asm")
-  //   return EXIT_SUCCESS;
+  // generate assembly
+  CodeGen codegen(moduleIR);
+  string asmCode = codegen.code_gen();
+  cout << asmCode << endl;
+  if (argStopAfter == "asm")
+    return EXIT_SUCCESS;
 
   return EXIT_SUCCESS;
 }