/*

  * Copyright (C) 2012 David Anderson

  *

  * This file is part of JITCraft.

  *

  * JITCraft is free software: you can redistribute it and/or modify it under

  * the terms of the GNU General Public License as published by the Free

  * Software Foundation, either version 3 of the License, or (at your option)

  * any later version.

  * 

  * Foobar is distributed in the hope that it will be useful, but WITHOUT ANY

  * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS

  * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.

  *

  * You should have received a copy of the GNU General Public License along with

  * JITCraft. If not, see http://www.gnu.org/licenses/.

  */

 #include <stdio.h>

 #include "Utility.h"

 #include "Scanner.h"

 #include "Parser.h"

 #include "SemanticAnalysis.h"

 #include "Zone.h"

 #include "BytecodeCompiler.h"

 #include "Code.h"

 #include "Interpreter.h"

 #include "Modules.h"

 #include "RootScopes.h"

 using namespace ke;

 class File

 {

     FILE *fp_;

   public:

     File(FILE *fp)

       : fp_(fp)

     {

     }

     ~File() {

         if (fp_)

             fclose(fp_);

     }

     operator FILE *() {

         return fp_;

     }

     operator bool () {

         return !!fp_;

     }

 };

 static bool

 Compile(const char *stream, size_t length)

 {

     if (!ZONE()->initialize())

         return false;

     RootScope roots(ZONE());

     CompileContext cc(ZONE());

     Parser parser(ZONE(), cc, stream, length);

     ParseTree *tree = parser.parse();

     if (!tree)

         return false;

     tree->dump(stdout);

     if (!Scope::Bind(tree))

         return false;

     SemanticAnalysis semantics(ZONE(), cc, tree);

     if (!semantics.analyze())

         return false;

     Scope::Allocate(tree);

     BytecodeCompiler compiler(ZONE(), tree, NULL);

     Module *module = compiler.compile();

     if (!module)

         return false;

     printf("\n ------------ \n\n");

     Disassemble(module->getFunction(1)->code(), stdout);

     //Disassemble(module->getFunction(1), stdout);

     printf("\n ------------ \n\n");

     Function *main = module->getFunction(1);

     Top top = ZONE()->stack().begin();

     LocalTop localTop(top);

     Frame *frame = ZONE()->stack().pushFrame(main);

     if (!frame)

         return false;

     LocalSlot result;

     if (!Interpret(ZONE(), frame, &result))

         return false;

     printf("Result: %d\n", result.i32);

     return true;

 }

 int main(int argc, char **argv)

 {

     if (argc < 2) {

         fprintf(stderr, "Usage: <file>\n");

         return 1;

     }

     AutoVec<char> bytes(NULL);

     size_t length;

     {

         File fp = fopen(argv[1], "rb");

         if (!fp) {

             fprintf(stderr, "Can't open file: \"%s\"\n", argv[1]);

             return 1;

         }

         if (fseek(fp, 0, SEEK_END)) {

             fprintf(stderr, "Error seeking file!\n");

             return 1;

         }

         long size = ftell(fp);

         if (size < 0) {

             fprintf(stderr, "Error getting file length!\n");

             return 1;

         }

         rewind(fp);

         length = size_t(size);

         bytes = new char[length + 1];

         if (fread(bytes, 1, length, fp) != length) {

             fprintf(stderr, "Error reading file!\n");

             return 1;

         }

     }

     Compile(bytes, length);

 }