Subversion Repositories theia_gpu

/* This program is free software. It comes without any warranty, to
 * the extent permitted by applicable law. You can redistribute it
 * and/or modify it under the terms of the Do What The Fuck You Want
 * To Public License, Version 2, as published by Sam Hocevar. See
 * http://sam.zoy.org/wtfpl/COPYING for more details. */
 
#include "Compiler.h"
#include "Preprocessor.h"
#include <string>
#include <iostream>
#include <fstream>
#include <cstring>
 
Preprocessor PP;
 
int main(int argc, char * argv[]) {
	// make sure we received a filename
	if (argc < 2) {
		std::cerr << "Usage: ./theia_compile -i [FILENAME] [-f <hex32|hex64>]" << std::endl;
		return 255;
	}
 
 
std::cout << "---------------------------------------------------------------\n";
std::cout << "  \n";
std::cout << " _/_/_/_/_/  _/                  _/            \n";
std::cout << "   _/      _/_/_/      _/_/          _/_/_/   \n";
std::cout << "  _/      _/    _/  _/_/_/_/  _/  _/    _/    \n";
std::cout << " _/      _/    _/  _/        _/  _/    _/     \n";
std::cout << "_/      _/    _/    _/_/_/  _/    _/_/_/      \n";
std::cout << "\n";
std::cout << "\n";
std::cout << "---------------------------------------------------------------\n";
 
	char * inputFile, * outputMode; 
	TheiaCompiler * Compiler;
	bool OutputMode32 = false;
	bool FilePathDefined = false;
	bool StdIn = false;
	try {
 
	for (int i = 1; i < argc; i++)
	{
 
			if (!strcmp(argv[i],"-i"))
			{
				inputFile = argv[i+1];
				FilePathDefined = true;
				i++;
			} 
			else if (!strcmp(argv[i],"-stdin"))
			{
				StdIn = true;
			}
			else if (!strcmp(argv[i],"-hex32"))
			{
					OutputMode32 = true;
 
			}
			else
			{
				std::cout << "Error: Invalid option " <<  argv[i] << "\n";
				return 255;
			}
 
 
 
	}
	if (!FilePathDefined)
	{
		std::cout << "Error: Input file not defined\n ";
		return 255;
	}
 
	if (StdIn)
	{
		Compiler = new TheiaCompiler(std::cin);
	} else {
		PP.Execute(inputFile);
		Compiler = new TheiaCompiler( std::string(inputFile)+ ".preprocessed",OutputMode32);
 
 
	}
	/*
		// - means stdin, not a file named '-'
		if (strcmp(argv[1], "-") == 0) {
 
			Compiler = new TheiaCompiler(std::cin);
		} else {
			PP.Execute(argv[1]);
 
			Compiler = new TheiaCompiler( std::string(argv[1])+ ".preprocessed");
		}
		*/
	} catch (std::string error) {
		std::cerr << "ERROR: " << error << std::endl;
		return 255;
	}
	Compiler->Print();
 
	std::ofstream ofs;
	ofs.open("code.list");
	if (!ofs.good())
	{
		std::cout << "Error could not open file for write 'code.mem'\n";
		return 0;
	}	
 
	ofs << Compiler->GetHexCodeDump();
	ofs.close();
	ofs.open("code.mem");
	ofs << Compiler->PostProcess("code.list");
	ofs.close();
 
    delete Compiler;
	std::string Command = "rm " + std::string(inputFile)+ ".preprocessed";
	system(Command.c_str());
	std::cout << "Code successfully compiled!\n";
	return 0;
}