/**

 * 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, "");

	Memory memory;

	// Sample program / data:

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

	{

		memory.write<command_t>(CMD_SLEEP);

		memory.write<sleep_t>(255);

		memory.write<command_t>(CMD_SLEEP);

		memory.write<sleep_t>(10);

		memory.write<command_t>(CMD_SLEEP);

		memory.write<sleep_t>(255);

		memory.write<command_t>(CMD_GOTO);

		memory.write<address_t>(memory.getAddress() + sizeof (address_t) + 2 * sizeof (command_t));

		memory.write<command_t>(CMD_INVALID);

		memory.write<command_t>(CMD_INVALID);

		memory.write<command_t>(CMD_SLEEP);

		memory.write<sleep_t>(255);

		memory.write<command_t>(CMD_COLOR);

		memory.write<led_t>(23);

		memory.write<color_t>(0);

		memory.write<color_t>(200);

		memory.write<color_t>(255);

		memory.write<command_t>(CMD_INCREMENT);

		memory.write<address_t>(0);

		memory.write<command_t>(CMD_DECREMENT);

		memory.write<address_t>(0);

		memory.write<command_t>(CMD_GOTO_COMPARE);

		memory.write<address_t>(0);

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

		memory.write<address_t>(memory.getAddress() - 3 * sizeof (address_t) - 2 * sizeof (command_t));

		memory.write<address_t>(0);

		memory.write<address_t>(memory.getAddress() + sizeof (address_t));

		memory.write<command_t>(CMD_END);

		memory.setAddressToBeginning();

	}

	// 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:

	while (memory.isNotOver()) {

		wprintf(L"command %*d = ", 4, memory.getAddress());

		command_t commandCode = memory.read<command_t>();

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

		shared_ptr<Command> command = commands[commandCode];

		if (command) {

			command->process(memory);

		} else {

			wprintf(L"invalid command\n");

		}

	}

	wprintf(L"all done\n");

	return 0;

}