Skip to content
GitLab
Select Git revision
  • wcy-final-submit
  • final-tmp
  • syq-submit
  • syq-backend-final
  • 6-main
  • 6-main-2
  • main default
  • wcy-temp
  • wzh-dev
  • ymh-final
  • arm-test-v3
  • tmp-arm-test-v2
  • tmp-arm-test
  • dev-ymhui-pass-backup810
  • 4-8-9
  • 5-86_long_code2-sy
  • w
  • wangchenyang-submit
  • dev-ymhui-pass
  • syq-backend-arm
Find file BlameHistoryPermalink
Main.java 3.48 KiB
1 
package cn.edu.nju.software;
2
3 
import cn.edu.nju.software.frontend.lexer.LexerErrorListener;
4 
import cn.edu.nju.software.frontend.lexer.SysYLexer;
5 
import cn.edu.nju.software.frontend.parser.ParserErrorListener;
6 
import cn.edu.nju.software.frontend.parser.SysYParser;
7 
import cn.edu.nju.software.frontend.semantic.SysYSemanticVisitor;
8 
import cn.edu.nju.software.ir.generator.IRVisitor;
9 
import cn.edu.nju.software.ir.module.ModuleRef;
10 
import org.antlr.v4.runtime.CharStream;
11 
import org.antlr.v4.runtime.CharStreams;
12 
import org.antlr.v4.runtime.CommonTokenStream;
13 
import org.antlr.v4.runtime.tree.ParseTree;
14
15 
import java.io.IOException;
16 
import java.util.Arrays;
17
18 
public class Main {
19 
    private static String input;
20 
    private static String output;
21
22 
    private static boolean optimized = false;
23 
    private static boolean emitLLVM = false;
24 
    private static boolean emitAssembly = false;
25
26 
    public static void main(String... args) {
27 
        Thread t = new Thread(() -> execute(args));
28 
        t.start();
29 
        try {
30 
            t.join();
31 
        } catch (Exception ignored) {}
32
33 
    }
34
35 
    private static void execute(String... args) {
36 
        parseArgs(args);
37 
        assert input != null && output != null;
38
39 
        CharStream inputStream;
40 
        try {
41 
            inputStream = CharStreams.fromFileName(input);
42 
        } catch (IOException e) {
43 
            return;
44 
        }
45
46 
        // lexer
47 
        SysYLexer sysYLexer = new SysYLexer(inputStream);
48 
        LexerErrorListener lexerErrorListener = new LexerErrorListener();
49 
        sysYLexer.removeErrorListeners();
50 
        sysYLexer.addErrorListener(lexerErrorListener);
51
52 
        // parser
53 
        CommonTokenStream tokens = new CommonTokenStream(sysYLexer);
54 
        SysYParser sysYParser = new SysYParser(tokens);
55 
        ParserErrorListener parserErrorListener = new ParserErrorListener();
56 
        sysYParser.removeErrorListeners();
57 
        sysYParser.addErrorListener(parserErrorListener);
58 
        ParseTree tree = sysYParser.program();
59 
        if (!parserErrorListener.noParseError()) {
60 
            return;
61 
        }
62
63 
        // semantic
64 
        SysYSemanticVisitor semanticVisitor = new SysYSemanticVisitor();
65 
        semanticVisitor.visit(tree);
66 
        if (!semanticVisitor.noSemanticError()) {
67 
            return;
68 
        }
69
70 
        // generate llvm ir
7172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112
IRVisitor irVisitor = new IRVisitor(); irVisitor.visit(tree); ModuleRef module = irVisitor.getModule(); if (emitLLVM) { module.dumpToFile(output); } irVisitor.dumpModuleToConsole(); // todo: emit assembly } private static void parseArgs(String... args) { Arrays.stream(args).forEach(arg -> { int index; switch (arg) { case "-o": index = Arrays.asList(args).indexOf(arg); output = args[index + 1]; break; case "-O0": optimized = false; break; case "-O1": case "-O2": optimized = true; break; case "--emit-llvm": emitLLVM = true; break; case "-S": emitAssembly = true; break; default: index = Arrays.asList(args).indexOf(arg); if (0 == index || !"-o".equals(args[index - 1]) ) { input = arg; } } }); } }