cc.cpp

#include <stdio.h>
#include <stdlib.h>
#include <assert.h>
#include "ast.hpp"
#include "c.tab.hpp"
#include <iostream>
#include "optimize.hpp"


extern "C" int yylex();
int yyparse();
extern "C" FILE *yyin;


fstream file;
int register_counter = 0;
int label_counter = 0;
int loop_label_counter = 0;
unordered_map<string, pair<string, int>> string_constants; // (register name, string constant)
string func_ret_type;

static void usage()
{
  printf("Usage: cc <prog.c>\n");
}

int
main(int argc, char **argv)
{
  if (argc != 2) {
    usage();
    exit(1);
  }
  char const *filename = argv[1];
  yyin = fopen(filename, "r");
  assert(yyin);
  int ret = yyparse();
  // At this point ast_root gives me the AST.
  // if (ret==0) printAST(ast_root);

  // Lab 2 Part 2 /////////////////////////////////////

  // try{
  //   throw string("Hello");
  // } catch (string s) {
  //   cout << s << endl;
  // }

  // cout<<string("Hello")+' '+string("World")<<endl;

  // stringstream ss;
  // ss<<"declare "<<"void"<<" "<<"@empty"<<endl;
  // cout<<ss.str()<<endl;

  // string str = "\"hello world\"";
  // str.insert(str.size()-1,"\\0A\\00");
  // cout<<str<<endl;

  if(ret==0){
    unordered_map<string, ll_node*> symbol_table;
    try {
      check_scope(ast_root, symbol_table, 0);
    } 
    catch (string s){
      cout << s << endl;
      ret = 1;
    }

    // for(auto x: symbol_table){
    //   ll_node* list = x.second;
    //   cout<<x.first<<"\t";
    //   while(list){
    //     cout<<list->node->scope_level<<"\t";
    //     list=list->next;
    //   }
    //   cout<<endl;
    // }
  }

  ////////////////////////////////////////////////////


  // Lab 2 Part 4 ////////////////////////////////////
  if(ret == 0){
    // printAST(ast_root);
    optimize(ast_root);
    printAST(ast_root);
  }

  ////////////////////////////////////////////////////

    
  // Lab 2 Part 3 ////////////////////////////////////
  if(ret==0){
    file.open("cc.ll", ios::trunc | ios::out | ios::in);
    // file << "hello world";
    unordered_map<string, ll_node*> symtable;
    try {
      cgen(ast_root, symtable, 0);
    } catch (string s){
      cout << s << endl;
      ret = 1;
    }
    
    // print string constants here
    file<<endl;
    for(auto x: string_constants){
      string reg_name = x.first; string str = x.second.first; int sz = x.second.second;
      string type = "[ "+to_string(sz)+" x i8 ]";
      file<<reg_name<<" = "<<" private constant "<<type<<" c"<<str<<"\n";
    }

    file.close();
  }

  ////////////////////////////////////////////////////

  printf("retv = %d\n", ret);
  exit(0);
}