BizHawk/BizHawk.MultiClient/NEStools/NESGameGenie.cs

using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Data;
using System.Drawing;
using System.Linq;
using System.Text;
using System.Windows.Forms;
using System.Globalization;
using BizHawk;

namespace BizHawk.MultiClient
{
	public partial class NESGameGenie : Form
	{
		int address = -1;
		int value = -1;
		int compare = -1;
		Dictionary<char, int> GameGenieTable = new Dictionary<char, int>();

		public NESGameGenie()
		{
			InitializeComponent();
			Closing += (o, e) => SaveConfigSettings();
		}

		private void NESGameGenie_Load(object sender, EventArgs e)
		{
			GameGenieTable.Add('A', 0);     //0000
			GameGenieTable.Add('P', 1);     //0001
			GameGenieTable.Add('Z', 2);     //0010
			GameGenieTable.Add('L', 3);     //0011
			GameGenieTable.Add('G', 4);     //0100
			GameGenieTable.Add('I', 5);     //0101
			GameGenieTable.Add('T', 6);     //0110
			GameGenieTable.Add('Y', 7);     //0111
			GameGenieTable.Add('E', 8);     //1000
			GameGenieTable.Add('O', 9);     //1001
			GameGenieTable.Add('X', 10);    //1010
			GameGenieTable.Add('U', 11);    //1011
			GameGenieTable.Add('K', 12);    //1100
			GameGenieTable.Add('S', 13);    //1101
			GameGenieTable.Add('V', 14);    //1110
			GameGenieTable.Add('N', 15);    //1111

			AddCheat.Enabled = false;

			if (Global.Config.NESGGSaveWindowPosition && Global.Config.NESGGWndx >= 0 && Global.Config.NESGGWndy >= 0)
				Location = new Point(Global.Config.NESGGWndx, Global.Config.NESGGWndy);
		}

		private void SaveConfigSettings()
		{
			Global.Config.NESGGWndx = this.Location.X;
			Global.Config.NESGGWndy = this.Location.Y;
		}

		private void GameGenieCode_KeyPress(object sender, KeyPressEventArgs e)
		{
			//Make uppercase
			if (e.KeyChar > 97 && e.KeyChar < 123)
				e.KeyChar -= (char)32;

			if (!(GameGenieTable.ContainsKey(e.KeyChar)))
			{
				if (!(e.KeyChar == (char)Keys.Back)) //Allow backspace
					e.Handled = true;
			}
			else
			{
				Encoding.Checked = false;
			}
		}

		private int GetBit(int value, int bit)
		{
			return (value >> bit) & 1;
		}

		private void DecodeGameGenieCode(string code)
		{
			//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);

				SetProperties();

			}
			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);
				SetProperties();
			}
		}

		private void SetProperties()
		{
			if (address >= 0)
				AddressBox.Text = String.Format("{0:X4}", address);
			else
				AddressBox.Text = "";

			if (compare >= 0)
				CompareBox.Text = String.Format("{0:X2}", compare);
			else
				CompareBox.Text = "";

			if (value >= 0)
				ValueBox.Text = String.Format("{0:X2}", value);

		}

		private void ClearProperties()
		{
			address = -1;
			value = -1;
			compare = -1;
			AddressBox.Text = "";
			CompareBox.Text = "";
			ValueBox.Text = "";
			AddCheat.Enabled = false;
		}

		private void GameGenieCode_TextChanged(object sender, EventArgs e)
		{
			if (Encoding.Checked == false)
			{
				if (GameGenieCode.Text.Length == 6 || GameGenieCode.Text.Length == 8)
					DecodeGameGenieCode(GameGenieCode.Text);
				else
					ClearProperties();
			}
			TryEnableAddCheat();
		}

		private void Keypad_Click(object sender, EventArgs e)
		{
			if (GameGenieCode.Text.Length < 8)
			{
				if (sender == A) GameGenieCode.Text += "A";
				if (sender == P) GameGenieCode.Text += "P";
				if (sender == Z) GameGenieCode.Text += "Z";
				if (sender == L) GameGenieCode.Text += "L";
				if (sender == G) GameGenieCode.Text += "G";
				if (sender == I) GameGenieCode.Text += "I";
				if (sender == T) GameGenieCode.Text += "T";
				if (sender == Y) GameGenieCode.Text += "Y";
				if (sender == E) GameGenieCode.Text += "E";
				if (sender == O) GameGenieCode.Text += "O";
				if (sender == X) GameGenieCode.Text += "X";
				if (sender == U) GameGenieCode.Text += "U";
				if (sender == K) GameGenieCode.Text += "K";
				if (sender == S) GameGenieCode.Text += "S";
				if (sender == V) GameGenieCode.Text += "V";
				if (sender == N) GameGenieCode.Text += "N";
				Encoding.Checked = false;
			}
		}

		private void AddressBox_KeyPress(object sender, KeyPressEventArgs e)
		{
			if (!(e.KeyChar == (char)Keys.Back)) //Allow backspace
			{
				if (InputValidate.IsValidHexNumber(e.KeyChar))
				{
					Encoding.Checked = true;
				}
				else
					e.Handled = true;
			}
			else
				Encoding.Checked = true;
		}

		private void CompareBox_KeyPress(object sender, KeyPressEventArgs e)
		{
			if (!(e.KeyChar == (char)Keys.Back)) //Allow backspace
			{
				if (InputValidate.IsValidHexNumber(e.KeyChar))
				{

					Encoding.Checked = true;
				}
				else
					e.Handled = true;
			}
			else
				Encoding.Checked = true;
		}

		private void ValueBox_KeyPress(object sender, KeyPressEventArgs e)
		{
			if (!(e.KeyChar == (char)Keys.Back)) //Allow backspace
			{
				if (InputValidate.IsValidHexNumber(e.KeyChar))
				{

					Encoding.Checked = true;
				}
				else
					e.Handled = true;
			}
			else
				Encoding.Checked = true;
		}

		private void AddressBox_TextChanged(object sender, EventArgs e)
		{
			if (Encoding.Checked && AddressBox.Text.Length > 0)
			{
				int a = int.Parse(AddressBox.Text, NumberStyles.HexNumber); //TODO: try/catch just in case?
				if (ValueBox.Text.Length > 0)
				{
					address = a;
					EncodeGameGenie();
				}
			}
			TryEnableAddCheat();
		}

		private void CompareBox_TextChanged(object sender, EventArgs e)
		{
			if (Encoding.Checked)
			{
				if (CompareBox.Text.Length > 0)
				{
					int c = int.Parse(CompareBox.Text, NumberStyles.HexNumber);
					if (c > 0 && c < 0x100)
					{
						if (ValueBox.Text.Length > 0 && AddressBox.Text.Length > 0)
						{
							compare = c;
							EncodeGameGenie();
						}
					}
				}
				else
				{
					compare = -1;
					EncodeGameGenie();
				}
			}
			TryEnableAddCheat();
		}

		private void TryEnableAddCheat()
		{
			if (AddressBox.Text.Length > 0 && ValueBox.Text.Length > 0 && GameGenieCode.Text.Length > 0)
				AddCheat.Enabled = true;
			else
				AddCheat.Enabled = false;
		}

		private void ValueBox_TextChanged(object sender, EventArgs e)
		{
			if (Encoding.Checked && ValueBox.Text.Length > 0)
			{
				int v = int.Parse(ValueBox.Text, NumberStyles.HexNumber);
				if (v > 0 && v < 0x100)
				{
					if (AddressBox.Text.Length > 0)
					{
						value = v;
						EncodeGameGenie();
					}
				}
			}

			TryEnableAddCheat();
		}

		private void EncodeGameGenie()
		{
			char[] letters = { 'A', 'P', 'Z', 'L', 'G', 'I', 'T', 'Y', 'E', 'O', 'X', 'U', 'K', 'S', 'V', 'N' };
			if (address >= 0x8000)
				address -= 0x8000;
			GameGenieCode.Text = "";
			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 < 0 || CompareBox.Text.Length == 0)
			{
				num[5] += (byte)(value & 8);
				for (int x = 0; x < 6; x++)
					GameGenieCode.Text += letters[num[x]];
			}
			else
			{
				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 x = 0; x < 8; x++)
					GameGenieCode.Text += letters[num[x]];
			}

		}

		private void ClearButton_Click(object sender, EventArgs e)
		{
			ClearProperties();
			GameGenieCode.Text = "";
			Encoding.Checked = false;
		}

		private void AddCheat_Click(object sender, EventArgs e)
		{
			Cheat c = new Cheat();
			c.name = GameGenieCode.Text;
			c.address = int.Parse(AddressBox.Text, NumberStyles.HexNumber);
			c.value = byte.Parse(ValueBox.Text, NumberStyles.HexNumber);
			c.domain = Global.Emulator.MemoryDomains[1];
			c.Enable();

			Global.MainForm.Cheats1.AddCheat(c);
		}

		private void exitToolStripMenuItem_Click(object sender, EventArgs e)
		{
			this.Close();
		}

		private void saveWindowPositionToolStripMenuItem_Click(object sender, EventArgs e)
		{
			Global.Config.NESGGSaveWindowPosition ^= true;
		}

		private void autoloadToolStripMenuItem_Click(object sender, EventArgs e)
		{
			Global.Config.NESGGAutoload ^= true;
		}

		private void optionsToolStripMenuItem_DropDownOpened(object sender, EventArgs e)
		{
			autoloadToolStripMenuItem.Checked = Global.Config.NESGGAutoload;
			saveWindowPositionToolStripMenuItem.Checked = Global.Config.NESGGSaveWindowPosition;
		}
	}
}