using System;
using System.Collections.Generic;

namespace BizHawk.Client.Common
{
	public class NESGameGenieDecoder
	{
		public int Address { get; private set; }
		public int Value { get; private set; }
		public int? Compare { get; private set; }

		private readonly string _code = String.Empty;

		public NESGameGenieDecoder(string code)
		{
			_code = code;
			Decode();
		}

		private readonly Dictionary<char, int> GameGenieTable = new Dictionary<char, int>
			{
				{ 'A', 0 }, //0000
				{ 'P', 1 }, //0001
				{ 'Z', 2 }, //0010
				{ 'L', 3 }, //0011
				{ 'G', 4 }, //0100
				{ 'I', 5 }, //0101
				{ 'T', 6 }, //0110
				{ 'Y', 7 }, //0111
				{ 'E', 8 }, //1000
				{ 'O', 9 }, //1001
				{ 'X', 10}, //1010
				{ 'U', 11}, //1011
				{ 'K', 12}, //1100
				{ 'S', 13}, //1101
				{ 'V', 14}, //1110
				{ 'N', 15}, //1111
			};

		public void Decode()
		{
			//char 3 bit 3 denotes the code length.
			if (_code.Length == 6)
			{
				//Char # |   1   |   2   |   3   |   4   |   5   |   6   |
				//Bit  # |3|2|1|0|3|2|1|0|3|2|1|0|3|2|1|0|3|2|1|0|3|2|1|0|
				//maps to|1|6|7|8|H|2|3|4|-|I|J|K|L|A|B|C|D|M|N|O|5|E|F|G|
				Value = 0;
				Address = 0x8000;
				int x;

				GameGenieTable.TryGetValue(_code[0], out x);
				Value |= (x & 0x07);
				Value |= (x & 0x08) << 4;

				GameGenieTable.TryGetValue(_code[1], out x);
				Value |= (x & 0x07) << 4;
				Address |= (x & 0x08) << 4;

				GameGenieTable.TryGetValue(_code[2], out x);
				Address |= (x & 0x07) << 4;

				GameGenieTable.TryGetValue(_code[3], out x);
				Address |= (x & 0x07) << 12;
				Address |= (x & 0x08);

				GameGenieTable.TryGetValue(_code[4], out x);
				Address |= (x & 0x07);
				Address |= (x & 0x08) << 8;

				GameGenieTable.TryGetValue(_code[5], out x);
				Address |= (x & 0x07) << 8;
				Value |= (x & 0x08);
			}
			else if (_code.Length == 8)
			{
				//Char # |   1   |   2   |   3   |   4   |   5   |   6   |   7   |   8   |
				//Bit  # |3|2|1|0|3|2|1|0|3|2|1|0|3|2|1|0|3|2|1|0|3|2|1|0|3|2|1|0|3|2|1|0|
				//maps to|1|6|7|8|H|2|3|4|-|I|J|K|L|A|B|C|D|M|N|O|%|E|F|G|!|^|&|*|5|@|#|$|
				Value = 0;
				Address = 0x8000;
				Compare = 0;
				int x;

				GameGenieTable.TryGetValue(_code[0], out x);
				Value |= (x & 0x07);
				Value |= (x & 0x08) << 4;

				GameGenieTable.TryGetValue(_code[1], out x);
				Value |= (x & 0x07) << 4;
				Address |= (x & 0x08) << 4;

				GameGenieTable.TryGetValue(_code[2], out x);
				Address |= (x & 0x07) << 4;

				GameGenieTable.TryGetValue(_code[3], out x);
				Address |= (x & 0x07) << 12;
				Address |= (x & 0x08);

				GameGenieTable.TryGetValue(_code[4], out x);
				Address |= (x & 0x07);
				Address |= (x & 0x08) << 8;

				GameGenieTable.TryGetValue(_code[5], out x);
				Address |= (x & 0x07) << 8;
				Compare |= (x & 0x08);

				GameGenieTable.TryGetValue(_code[6], out x);
				Compare |= (x & 0x07);
				Compare |= (x & 0x08) << 4;

				GameGenieTable.TryGetValue(_code[7], out x);
				Compare |= (x & 0x07) << 4;
				Value |= (x & 0x08);
			}
		}
	}

	public class NESGameGenieEncoder
	{
		private readonly char[] letters = { 'A', 'P', 'Z', 'L', 'G', 'I', 'T', 'Y', 'E', 'O', 'X', 'U', 'K', 'S', 'V', 'N' };

		private int _address;
		private int _value;
		private int? _compare;

		public string GameGenieCode { get; private set; }

		public NESGameGenieEncoder(int address, int value, int? compare)
		{
			_address = address;
			if (_address >= 0x8000)
			{
				_address -= 0x8000;
			}
			_value = value;
			_compare = compare;

			GameGenieCode = String.Empty;
		}

		public string Encode()
		{
			byte[] num = { 0, 0, 0, 0, 0, 0, 0, 0 };
			num[0] = (byte)((_value & 7) + ((_value >> 4) & 8));
			num[1] = (byte)(((_value >> 4) & 7) + ((_address >> 4) & 8));
			num[2] = (byte)(((_address >> 4) & 7));
			num[3] = (byte)((_address >> 12) + (_address & 8));
			num[4] = (byte)((_address & 7) + ((_address >> 8) & 8));
			num[5] = (byte)(((_address >> 8) & 7));

			if (_compare.HasValue)
			{
				num[2] += 8;
				num[5] += (byte)(_compare & 8);
				num[6] = (byte)((_compare & 7) + ((_compare >> 4) & 8));
				num[7] = (byte)(((_compare >> 4) & 7) + (_value & 8));
				for (int i = 0; i < 8; i++)
				{
					GameGenieCode += letters[num[i]];
				}
			}
			else
			{
				num[5] += (byte)(_value & 8);
				for (int i = 0; i < 6; i++)
				{
					GameGenieCode += letters[num[i]];
				}
			}

			return GameGenieCode;
		}
	}
}