dj2ll.cpp

#include "dj2ll.hpp"
#include "test.hpp"
#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <map>
#include <string>

ASTree *wholeProgram;
ASTree *mainExprs;
int numMainBlockLocals;
VarDecl *mainBlockST;
int numClasses;
ClassDecl *classesST;
std::string inputFile;
int instanceOfSeen;
int printNatSeen;
int readNatSeen;

bool findCLIOption(char **begin, char **end, const std::string &flag) {
  return std::find(begin, end, flag) != end;
}

void runClang() {
  auto outputFile = trimFromLastOccurrence(inputFile, "/");
  std::string command = "clang " + inputFile + ".o" + " -o " + outputFile;
  std::system(command.c_str());
  std::string rmCommand = "rm " + inputFile + ".o";
  std::system(rmCommand.c_str());
}

void dj2ll(std::map<std::string, bool> compilerFlags, std::string fileName,
           char **argv) {
  std::string extension = fileName.substr(fileName.size() - 3, fileName.size());
  inputFile = fileName.substr(0, fileName.size() - 3);
  if (extension != ".dj") {
    printf("ERROR: %s must be called on files ending with \".dj\"\n", argv[0]);
    exit(-1);
  }

  yyin = fopen(fileName.c_str(), "r");
  if (yyin == nullptr) {
    printf("ERROR: could not open file %s\n", fileName.c_str());
    exit(-1);
  }
  yyparse();
  fclose(yyin);

  setupSymbolTables(pgmAST);
  typecheckProgram();

  auto LLProgram = translateAST(wholeProgram);
  if (compilerFlags["verbose"]) {
    LLProgram.print();
  }

  if (compilerFlags["codegen"]) {
    LLProgram.runOptimizations = compilerFlags["optimizations"];
    LLProgram.emitLLVM = compilerFlags["emitLLVM"];
    symbolTable ST; /*throwaway*/
    LLProgram.codeGen(ST);
  }
  runClang();
}