#include <cstdlib>

#include <iostream>

#include <wchar.h>

#include <locale.h>

#include <cstring>

#include <chrono>

#include <thread>

using namespace std;

typedef uint16_t address_t;

typedef uint8_t byte_t;

typedef uint8_t command_t;

typedef uint8_t sleep_t;

typedef uint8_t color_t;

typedef uint8_t led_t;

// TODO: strong typedefs http://www.boost.org/doc/libs/1_61_0/libs/serialization/doc/strong_typedef.html ?

const address_t MEMORY_SIZE = 1024;

/**

 * Skip to the given address.

 * parameter: address_t

 */

const command_t CMD_GOTO = 0x70;

/**

 * Compare values on two addresses and go to one of given three addresses.

 * parameter: address_t a

 * parameter: address_t b

 * parameter: address_t GOTO target when a == b

 * parameter: address_t GOTO target when a > b

 * parameter: address_t GOTO target when a < b

 */

const command_t CMD_GOTO_COMPARE = 0x80;

/**

 * Wait given time in ms.

 * parameter: sleep_t

 */

const command_t CMD_SLEEP = 0xFF;

/**

 * Set RGB LED color.

 * parameter: led_t LED number

 * parameter: color_t red

 * parameter: color_t green

 * parameter: color_t blue

 */

const command_t CMD_COLOR = 0xAA;

/**

 * Stop program.

 */

const command_t CMD_END = 0xED;

/**

 * Increase value at given address

 * parameter: address_t

 */

const command_t CMD_INCREMENT = 0x11;

/**

 * Decrease value at given address

 * parameter: address_t

 */

const command_t CMD_DECREMENT = 0x12;

/**

 * Placeholder for unsupported command.

 * Just for testing.

 */

const command_t CMD_INVALID = 0x1;

// TODO: more commands, better numbers

/**

 * Reads data on given position in memory and increments the index (position).

 */

template<typename T> T read(byte_t * memory, address_t &index) {

	// TODO: for addresses: map higher memory to static hardcoded areas or peripherals

	// TODO: sizeof (byte_t) != 1 ?

	T * value = reinterpret_cast<T*> (memory + index);

	index += sizeof (*value) / sizeof (byte_t);

	return *value;

}

/**

 * Writes data to given position in memory and increments the index (position).

 */

template<typename T> void write(byte_t * memory, address_t &index, const T value) {

	// TODO: sizeof (byte_t) != 1 ?

	// T * m = (T*) (memory + index);

	T * m = reinterpret_cast<T*> (memory + index);

	*m = value;

	index += sizeof (value) / sizeof (byte_t);

}

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

	setlocale(LC_ALL, "");

	byte_t * memory = (byte_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);

	}

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

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

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

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

		switch (command) {

			case CMD_GOTO:

			{

				i = read<address_t>(memory, i);

				wprintf(L"GOTO %*d\n", 5, i);

				break;

			}

			case CMD_SLEEP:

			{

				sleep_t delay = read<sleep_t>(memory, i);

				wprintf(L"SLEEP %*d ms\n", 4, delay);

				this_thread::sleep_for(chrono::milliseconds(delay));

				break;

			}

			case CMD_COLOR:

			{

				led_t led = read<led_t>(memory, i);

				color_t r = read<color_t>(memory, i);

				color_t g = read<color_t>(memory, i);

				color_t b = read<color_t>(memory, i);

				wprintf(L"COLOR  %02X %02X %02X → %d\n", r, g, b, led);

				break;

			}

			case CMD_INCREMENT:

			case CMD_DECREMENT:

			{

				address_t address = read<address_t>(memory, i);

				address_t address_r = address;

				address_t address_w = address_r;

				byte_t value = read<byte_t>(memory, address_r);

				value = command == CMD_INCREMENT ? value + 1 : value - 1;

				write<byte_t>(memory, address_w, value);

				wprintf(L"%sCREMENT %*d → %02X\n", (command == CMD_INCREMENT ? "IN" : "DE"), 5, address, value);

				break;

			}

			case CMD_GOTO_COMPARE:

			{

				address_t aa = read<address_t>(memory, i);

				address_t ab = read<address_t>(memory, i);

				address_t eq = read<address_t>(memory, i);

				address_t gt = read<address_t>(memory, i);

				address_t lt = read<address_t>(memory, i);

				byte_t a = read<byte_t>(memory, aa);

				byte_t b = read<byte_t>(memory, ab);

				if (a == b) i = eq;

				else if (a > b) i = gt;

				else i = lt;

				wprintf(L"GOTO COMPARE  a = %02X, b = %02X, eq = %d, gt = %d, lt = %d → %d\n", a, b, eq, gt, lt, i);

				break;

			}

			case CMD_END:

			{

				wprintf(L"END\n");

				i = MEMORY_SIZE;

				break;

			}

			default:

			{

				wprintf(L"invalid command\n");

			}

		}

	}

	free(memory);

	memory = nullptr;

	wprintf(L"all done\n");

	return 0;

}