camle.java

// COMS22201: Skeleton of CAMLE compiler */

import java.io.*;
import java.lang.reflect.Array;
import org.antlr.runtime.*;
import org.antlr.runtime.tree.*;
import java.util.HashMap;


class camle {
  public static void main(String[] args)
  {
    System.out.println("CAMLE - Compiler to Abstract Machine for Language Engineering");
    String opt = "", inFile = "", outFile = "";
    int pos;
    if (Array.getLength(args) == 1 && args[0].charAt(0) != '-') {
      opt = "";
      inFile = args[0];
    }
    else if (Array.getLength(args) == 2 && args[0].charAt(0) == '-' &&
             args[1].charAt(0) != '-') {
      opt = args[0];
      inFile = args[1];
    }
    else {
      System.out.println("Usage: antlr3 camle [option] filename");
      System.out.println("");
      System.out.println("Options:");
      System.out.println("  -lex");
      System.out.println("  -syn");
      System.out.println("  -irt");
      System.exit(1);
    }
    outFile = inFile;
    if ((pos = outFile.lastIndexOf('.')) != -1)
      outFile = outFile.substring(0, pos);
    outFile = outFile+".ass";

    try {
      CharStream cs = new ANTLRFileStream(inFile);			
      Lex lexO = new Lex(cs);
      if (opt.equals("-lex")) {
        Token T;
        T = lexO.nextToken();
        while (T.getType() != -1) {
          System.out.println(T.getType()+" \""+T.getText()+"\"");
          T = lexO.nextToken();
        }
        System.exit(0);
      }
      CommonTokenStream tokens = new CommonTokenStream(lexO);
      Syn synO = new Syn(tokens);
      Syn.program_return parserResult = synO.program();//start rule
      CommonTree parserTree = (CommonTree) parserResult.getTree();
      if (opt.equals("-syn")) {
        System.out.println(parserTree.toStringTree());
        System.exit(0);
      }
      CommonTreeNodeStream ast = new CommonTreeNodeStream(parserTree);
      IRTree newIrt = Irt.convert(parserTree);
      if (opt.equals("-irt")) {
        System.out.println(newIrt);
        Memory.dumpData(System.out);
        System.exit(0);
      }
      PrintStream o = new PrintStream(new FileOutputStream(outFile));
      Cg.program(newIrt, o);
      Liveness_Analysis test1 = new Liveness_Analysis(outFile);
      Interference_graph graph = new Interference_graph(test1);
      Register_Allocation register_allocation = new Register_Allocation(graph);
      register_allocation.construct_graph();
      register_allocation.graph_coloring();
      HashMap<String, String> match_table = new HashMap<>();
      for (String original : register_allocation.result.keySet()){
          String replace_with = "R"+register_allocation.result.get(original);
          match_table.put(original, replace_with);
      }
      FileReplace fileReplace = new FileReplace(outFile, match_table);
      fileReplace.replaceFile();
    }
    catch(Exception e) {
      System.err.println("exception: "+e);
      e.printStackTrace();
    }
  }
}