Skip to content
GitLab
Select Git revision
Find file BlameHistoryPermalink
Forked from 啊对对队 / proj122-Qos
Source project has a limited visibility.
Main.java 4.05 KiB
1 
package cn.edu.nju.software;
2
3 
import java.io.IOException;
4 
import java.util.Arrays;
5
6 
import cn.edu.nju.software.backend.RiscModule;
7 
import cn.edu.nju.software.pass.PassManager;
8 
import org.antlr.v4.runtime.CharStream;
9 
import org.antlr.v4.runtime.CharStreams;
10 
import org.antlr.v4.runtime.CommonTokenStream;
11 
import org.antlr.v4.runtime.tree.ParseTree;
12 
import cn.edu.nju.software.frontend.lexer.LexerErrorListener;
13 
import cn.edu.nju.software.frontend.lexer.SysYLexer;
14 
import cn.edu.nju.software.frontend.parser.ParserErrorListener;
15 
import cn.edu.nju.software.frontend.parser.SysYParser;
16 
import cn.edu.nju.software.frontend.semantic.SysYSemanticVisitor;
17 
import cn.edu.nju.software.ir.generator.IRVisitor;
18 
import cn.edu.nju.software.ir.module.ModuleRef;
19
20 
public class Main {
21 
    private static String input;
22 
    private static String output;
23
24 
    private static boolean optimized = false;
25 
    private static boolean emitLLVM = false;
26 
    private static boolean emitAssembly = false;
27
28 
    public static void main(String... args) {
29 
        Thread t = new Thread(() -> execute(args));
30 
        t.start();
31 
        try {
32 
            t.join();
33 
        } catch (Exception ignored) {}
34
35 
    }
36
37 
    private static void execute(String... args) {
38 
        parseArgs(args);
39 
        if (input == null || output == null) {
40 
            throw new RuntimeException("required input and output path");
41 
        }
42
43 
        CharStream inputStream;
44 
        try {
45 
            inputStream = CharStreams.fromFileName(input);
46 
        } catch (IOException e) {
47 
            return;
48 
        }
49
50 
        // lexer
51 
        SysYLexer sysYLexer = new SysYLexer(inputStream);
52 
        LexerErrorListener lexerErrorListener = new LexerErrorListener();
53 
        sysYLexer.removeErrorListeners();
54 
        sysYLexer.addErrorListener(lexerErrorListener);
55
56 
        // parser
57 
        CommonTokenStream tokens = new CommonTokenStream(sysYLexer);
58 
        SysYParser sysYParser = new SysYParser(tokens);
59 
        ParserErrorListener parserErrorListener = new ParserErrorListener();
60 
        sysYParser.removeErrorListeners();
61 
        sysYParser.addErrorListener(parserErrorListener);
62 
        ParseTree tree = sysYParser.program();
63 
        if (!parserErrorListener.noParseError()) {
64 
            return;
65 
        }
66
67 
        // semantic
68 
        SysYSemanticVisitor semanticVisitor = new SysYSemanticVisitor();
69 
        semanticVisitor.visit(tree);
70 
        if (!semanticVisitor.noSemanticError()) {
7172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129
return; } // generate llvm ir IRVisitor irVisitor = new IRVisitor(); irVisitor.visit(tree); // irVisitor.dumpModuleToConsole(); ModuleRef module = irVisitor.getModule(); if(module == null){ assert false; } if(optimized){ PassManager passManager=new PassManager(module); //TODO:调试完成后删除这句 //passManager.setDbgFlag(); passManager.runPass(); } if (emitLLVM) { module.dumpToFile(output); } if(emitAssembly){ RiscModule riscModule = new RiscModule(module); riscModule.codeGen(); riscModule.dumpToFile(output); } } 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", "-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; } } }); } }