From 5d24f21a427f0790e51075c45de29511af4e7859 Mon Sep 17 00:00:00 2001
From: ugetab <ugetab@users.sf.net>
Date: Thu, 6 May 2010 23:47:57 +0000
Subject: [PATCH] Got Add Cheat button working in Ram Search and Ram Watch.

---
 src/drivers/win/ram_search.cpp | 21 +++++++++++++++++++++
 src/drivers/win/ramwatch.cpp   | 20 ++++++++++++++++++++
 2 files changed, 41 insertions(+)

diff --git a/src/drivers/win/ram_search.cpp b/src/drivers/win/ram_search.cpp
index 188a3c8d..dba1c972 100644
--- a/src/drivers/win/ram_search.cpp
+++ b/src/drivers/win/ram_search.cpp
@@ -33,6 +33,7 @@
 #include "resource.h"
 #include "ram_search.h"
 #include "ramwatch.h"
+#include "cheat.h"
 #include <assert.h>
 #include <commctrl.h>
 #include <list>
@@ -1698,6 +1699,7 @@ LRESULT CALLBACK RamSearchProc(HWND hDlg, UINT uMsg, WPARAM wParam, LPARAM lPara
 					if(watchItemIndex >= 0)
 					{
 						unsigned long address = CALL_WITH_T_SIZE_TYPES_1(GetHardwareAddressFromItemIndex, rs_type_size,rs_t=='s',noMisalign, watchItemIndex);
+						unsigned long curvalue = CALL_WITH_T_SIZE_TYPES_1(GetCurValueFromItemIndex, rs_type_size,rs_t=='s',noMisalign, watchItemIndex);
 
 						int sizeType = -1;
 						if(rs_type_size == 'b')
@@ -1716,6 +1718,25 @@ LRESULT CALLBACK RamSearchProc(HWND hDlg, UINT uMsg, WPARAM wParam, LPARAM lPara
 							numberType = 2;
 
 						// TODO: open add-cheat dialog
+
+						switch (sizeType) {
+						case 0: {
+								FCEUI_AddCheat("",address,curvalue,-1,1);
+								break; }
+						case 1: {
+								FCEUI_AddCheat("",address,curvalue & 0xFF,-1,1);
+								FCEUI_AddCheat("",address + 1,(curvalue & 0xFF00) / 0x100,-1,1);
+								break; }
+						case 2: {
+								FCEUI_AddCheat("",address,curvalue & 0xFF,-1,1);
+								FCEUI_AddCheat("",address + 1,(curvalue & 0xFF00) / 0x100,-1,1);
+								FCEUI_AddCheat("",address + 2,(curvalue & 0xFF0000) / 0x10000,-1,1);
+								FCEUI_AddCheat("",address + 3,(curvalue & 0xFF000000) / 0x1000000,-1,1);
+								break; }
+						}
+
+						UpdateCheatsAdded();
+
 					}
 				}	{rv = true; break;}
 				case IDC_C_RESET:
diff --git a/src/drivers/win/ramwatch.cpp b/src/drivers/win/ramwatch.cpp
index effa0ee0..ac22a92a 100644
--- a/src/drivers/win/ramwatch.cpp
+++ b/src/drivers/win/ramwatch.cpp
@@ -7,6 +7,7 @@ using namespace std;
 #include "resource.h"
 #include "ram_search.h"
 #include "ramwatch.h"
+#include "cheat.h"
 #include <assert.h>
 #include <windows.h>
 #include <commctrl.h>
@@ -1139,6 +1140,25 @@ LRESULT CALLBACK RamWatchProc(HWND hDlg, UINT uMsg, WPARAM wParam, LPARAM lParam
 							numberType = 2;
 
 						// TODO: open add-cheat dialog
+						// TODO: add add button
+
+						switch (sizeType) {
+						case 0: {
+								FCEUI_AddCheat("",address,rswatches[watchIndex].CurValue,-1,1);
+								break; }
+						case 1: {
+								FCEUI_AddCheat("",address,rswatches[watchIndex].CurValue & 0xFF,-1,1);
+								FCEUI_AddCheat("",address + 1,(rswatches[watchIndex].CurValue & 0xFF00) / 0x100,-1,1);
+								break; }
+						case 2: {
+								FCEUI_AddCheat("",address,rswatches[watchIndex].CurValue & 0xFF,-1,1);
+								FCEUI_AddCheat("",address + 1,(rswatches[watchIndex].CurValue & 0xFF00) / 0x100,-1,1);
+								FCEUI_AddCheat("",address + 2,(rswatches[watchIndex].CurValue & 0xFF0000) / 0x10000,-1,1);
+								FCEUI_AddCheat("",address + 3,(rswatches[watchIndex].CurValue & 0xFF000000) / 0x1000000,-1,1);
+								break; }
+						}
+
+						UpdateCheatsAdded();
 					}
 				}
 				break;