/*
 * main.cc
 * 
 * This file contains some generic functions. It is part of
 * the assignment for the Compiler Construction course at the Leiden Institute
 * of Advanced Computer Science (LIACS), Leiden University.
 */

#include "ICGenerator.h"
#include "SymbolTable.h"
#include "debug.h"

/* Globals */
ICGenerator * icg;
IntermediateCode * icode;
bool bWarnings = true;
bool bOptimize1 = false;
bool extraDump = false;


/* Extern functions */
int yyparse();


/* Extern variables */
extern int lineno;
extern SyntaxTree * tree;
extern SymbolTable * symtab;
extern int gErrorCount;
extern int gWarningCount;
extern int msglevel;




void PrintError (char * msg) {
  fprintf (stderr, "Error on line %d: %s\n", lineno, msg);
  gErrorCount++;
}//PrintError


void PrintWarning (char * msg) {
  if (bWarnings) {
    fprintf (stderr, "Warning on line %d: %s\n", lineno, msg);
    gWarningCount++;
  }
}//PrintWarning

void usage(char * argv0) {
    fprintf (stderr, "Usage: %s [OPTION]\n", argv0);
    fprintf (stderr, "Defaults: %s -w -O0\n", argv0);
    fprintf (stderr, "-h  = Reading right now ;-)\n");
    fprintf (stderr, "-w  = Enable warnings\n");
    fprintf (stderr, "-O0 = No optimalisation\n");
    fprintf (stderr, "-O1 = Use otimalisation\n");
    fprintf (stderr, "-g = Debug, more flags higher level\n");
    fprintf (stderr, "-s = Dump symbolTable after ICG\n");
}

/* Parse the command line */
int ParseCommandline(int argc, char ** argv)
{
  for (int i = 1; i < argc; i++) {
    if (strcmp(argv[i], "-w") == 0) {
      bWarnings = false;
    }
    else if (strcmp(argv[i], "-O0") == 0) {
      bOptimize1 = false;
    }
    else if (strcmp(argv[i], "-O1") == 0) {
      bOptimize1 = true;
    }
    else if (strcmp(argv[i], "-h") == 0) {
        usage(argv[0]);
        return 1;
    }
    else if (strcmp(argv[i], "-g") == 0) {
        msglevel += 30;
    }
    else if (strcmp(argv[i], "-s") == 0) {
        extraDump = true;
    }
    else {
      fprintf(stderr, "Error: unrecognized command line option '%s'\n", argv[i]);
      usage(argv[0]);
      return 1;
    }
  }

  return 0;
}


void initialize()
{
  /* Create & initialize the abstract data structures */
  tree = new SyntaxTree;
  symtab = new SymbolTable;
  icg = new ICGenerator;
  icode = NULL;

  gErrorCount = 0;
  gWarningCount = 0;
}


void cleanup()
{
  /* Free objects */
  delete icg;
  delete tree;
  delete symtab;
  if (icode) delete icode;
}


int main(int argc, char ** argv)
{
  int parseResult;
  
  if (ParseCommandline(argc, argv) == 1) {
    return 1;
  }

  initialize();
  
  /* Start the parsing procedure */
  parseResult = yyparse();
  
  /* Dump the syntax tree & symbol table */
  tree->Dump (stdout);
  printf("\n");
  symtab->Dump (stdout);

  /* Generate intermediate code, if everything is still ok */
  if (!(parseResult || gErrorCount)) {
    if (bOptimize1) {
      icg->Preprocess(tree, symtab);
      tree->Dump (stdout);
    }
    icode = icg->GenerateIntermediateCode(tree, symtab);
    if (bOptimize1) {
      icg->Postprocess(icode, symtab);
    }
    icode->Dump (stdout);

    /* Dump Extra symbol table at the end, to view edited variables on the symbol table */
    if (extraDump) {
      printf("\n");
      symtab->Dump (stdout);
    }
  }

  printf("%d error(s), %d warning(s)\n", gErrorCount + parseResult, gWarningCount);
  
  cleanup();
  
  return (parseResult || gErrorCount);

}