#include "CombineInstr.h"
#include "ConstSpread.h"
#include "LoopInvariant.h"
#include "SimplifyJump.h"
#include "ast.h"
#include "lirstmtbuilder.h"
#include "define.h"
#include "genIR.h"
#include "DeleteDeadCode.h"
#include "opt.h"
#include <fstream>
#include <iostream>
#include <ostream>
#include <unistd.h>

extern unique_ptr<CompUnitAST> root;
extern int yyparse();
extern FILE *yyin;

int main(int argc, char **argv) {
  // Assert the number of arguments
  assert(argc >= 2);

  // TODO: advanced argument parser
  char *filename = nullptr;
  int print_ir = false;
  int print_asm = false;

  std::string output = "-";

  int opt;
  bool isO2 = false;
  while ((opt = getopt(argc, argv, "Sco:O::")) != -1) {
    switch (opt) {
    case 'S':
      print_asm = true;
      print_ir = false;
      break;
    case 'c':
      print_ir = true;
      print_asm = false;
      break;
    case 'o':
      output = optarg;
      break;
    case 'O':
      isO2 = true;
      break;
    default:
      break;
    }
  }
  filename = argv[optind];

  yyin = fopen(filename, "r");
  if (yyin == nullptr) {
    std::cout << "yyin open" << filename << "failed" << std::endl;
    return -1;
  }

  // Frontend parser
  yyparse();

  // Generate IR from AST
  GenIR genIR;
  root->accept(genIR);
  std::unique_ptr<Module> m = genIR.getModule();

  // Run IR optimization
  // TODO
  if (isO2) {
    std::vector<Optimization *> Opt;
    Opt.push_back(new DeadCodeDeletion(m.get()));
    Opt.push_back(new ConstSpread(m.get()));
    Opt.push_back(new CombineInstr(m.get()));
    Opt.push_back(new DomainTree(m.get()));
    Opt.push_back(new SimplifyJump(m.get()));
    Opt.push_back(new LoopInvariant(m.get()));
    Opt.push_back(new SimplifyJump(m.get()));
    for (auto x : Opt)
      x->execute();
  }

  // Open output file
  std::ofstream fout;
  std::ostream *out;
  if (output == "-") {
    out = &std::cout;
  } else {
    fout.open(output);
    out = &fout;
  }

  // Print IR result
  const std::string IR = m->print();
  if (print_ir) {
    *out << IR << std::endl;
  }

  // Generate assembly file
  // TODO
  if (print_asm) {
    LIRStmtBuilder* lirbuilder;
    lirbuilder = new LIRStmtBuilder(m.get(), false, isO2);
    lirbuilder->build_lir();
    lirbuilder->build_asm();
    std::string ASM = lirbuilder->lirm_->print();
    *out << ASM << std::endl;
  }
  return 0;
}