/**

  * RGB assembler

  * Copyright © 2017 František Kučera (frantovo.cz)

  *

  * This program 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.

  *

  * This program 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 this program. If not, see <http://www.gnu.org/licenses/>.

  */

 #include <cstdlib>

 #include <iostream>

 #include <wchar.h>

 #include <locale.h>

 #include <cstring>

 #include <unordered_map>

 #include "types.h"

 #include "memory.h"

 #include "Command.h"

 #include "commands.h"

 #include "commands/Goto.h"

 #include "commands/GotoCompare.h"

 #include "commands/Sleep.h"

 #include "commands/End.h"

 #include "commands/Color.h"

 #include "commands/IncrementDecrement.h"

 using namespace std;

 int main(int argc, char* argv[]) {

 	setlocale(LC_ALL, "");

 	// TODO: wrap in a class with read(), write() and setAddress() methods

 	octet_t * memory = (octet_t*) malloc(MEMORY_SIZE);

 	// Sample program / data:

 	// TODO: load bytes from file, stdin, serial port, network…

 	{

 		address_t a = 0;

 		write<command_t>(memory, a, CMD_SLEEP);

 		write<sleep_t>(memory, a, 255);

 		write<command_t>(memory, a, CMD_SLEEP);

 		write<sleep_t>(memory, a, 10);

 		write<command_t>(memory, a, CMD_SLEEP);

 		write<sleep_t>(memory, a, 255);

 		write<command_t>(memory, a, CMD_GOTO);

 		write<address_t>(memory, a, a + sizeof (address_t) + 2 * sizeof (command_t));

 		write<command_t>(memory, a, CMD_INVALID);

 		write<command_t>(memory, a, CMD_INVALID);

 		write<command_t>(memory, a, CMD_SLEEP);

 		write<sleep_t>(memory, a, 255);

 		write<command_t>(memory, a, CMD_COLOR);

 		write<led_t>(memory, a, 23);

 		write<color_t>(memory, a, 0);

 		write<color_t>(memory, a, 200);

 		write<color_t>(memory, a, 255);

 		write<command_t>(memory, a, CMD_INCREMENT);

 		write<address_t>(memory, a, 0);

 		write<command_t>(memory, a, CMD_DECREMENT);

 		write<address_t>(memory, a, 0);

 		write<command_t>(memory, a, CMD_GOTO_COMPARE);

 		write<address_t>(memory, a, 0);

 		write<address_t>(memory, a, 0 + sizeof (command_t) + sizeof (sleep_t));

 		write<address_t>(memory, a, a - 3 * sizeof (address_t) - 2 * sizeof (command_t));

 		write<address_t>(memory, a, 0);

 		write<address_t>(memory, a, a + sizeof (address_t));

 		write<command_t>(memory, a, CMD_END);

 	}

 	// Supported commands

 	// TODO: dynamic reconfiguration

 	unordered_map<command_t, shared_ptr < Command>> commands = {

 		{CMD_GOTO, make_shared<commands::Goto>()},

 		{CMD_GOTO_COMPARE, make_shared<commands::GotoCompare>()},

 		{CMD_SLEEP, make_shared<commands::Sleep>()},

 		{CMD_END, make_shared<commands::End>()},

 		{CMD_COLOR, make_shared<commands::Color>()},

 		{CMD_INCREMENT, make_shared <commands::IncrementDecrement>(true, 1)},

 		{CMD_DECREMENT, make_shared <commands::IncrementDecrement>(false, 1)},

 	};

 	// Main loop / interpreter:

 	for (address_t i = 0; i < MEMORY_SIZE;) {

 		wprintf(L"command %*d = ", 4, i);

 		command_t commandCode = read<command_t>(memory, i);

 		wprintf(L"%02X  ", commandCode);

 		shared_ptr<Command> command = commands[commandCode];

 		if (command) {

 			command->process(memory, i);

 		} else {

 			wprintf(L"invalid command\n");

 		}

 	}

 	free(memory);

 	memory = nullptr;

 	wprintf(L"all done\n");

 	return 0;

 }