1 files changed, 81 insertions, 4 deletions

diff --git a/src/text/Messages.cpp b/src/text/Messages.cpp
index 81339ae0..f55e6bd7 100644
--- a/src/text/Messages.cpp
+++ b/src/text/Messages.cpp
@@ -11,6 +11,8 @@
 
 #include "Font.h"
 
+#include "Pad.h"
+
 tMessage CMessages::BriefMessages[NUMBRIEFMESSAGES];
 tPreviousBrief CMessages::PreviousBriefs[NUMPREVIOUSBRIEFS];
 tBigMessage CMessages::BIGMessages[NUMBIGMESSAGES];
@@ -441,6 +443,64 @@ CMessages::InsertStringInString(wchar *str1, wchar *str2)
 		str1[i++] = '\0';
 }
 
+int
+CMessages::GetTokenPadKeyString(const wchar *in, wchar *out)
+{
+	wchar str[256];
+	memset(str, 0, sizeof(str));
+	str[0] = 'C';
+
+	// TODO: there was a switch here but that's stupid
+	str[1] = CPad::GetPad(0)->Mode + 48;
+
+	while (*in != '~') in++;
+	in++;
+
+	int i = 1;
+	while (*in != '~')
+		str[1+i++] = *(in++);
+
+	wchar *text = TheText.Get(UnicodeToAscii(str));
+	if (!text) return i;
+	while (text[0] != '\0')
+	{
+		if (text[0] == '~')
+		{
+			switch (text[1])
+			{
+			case 'L':
+				*(out++) = 'M';
+				break;
+			case 'N':
+				*(out++) = 'O';
+				break;
+			case 'O':
+				*(out++) = 227;
+				break;
+			case 'R':
+				*(out++) = 'S';
+				break;
+			case 'S':
+				*(out++) = 225;
+				break;
+			case 'T':
+				*(out++) = 224;
+				break;
+			case 'X':
+				*(out++) = 226;
+				break;
+			default:
+				break;
+			}
+			text += 3;
+		}
+		else {
+			*(out++) = *(text++);
+		}
+	}
+	return i;
+}
+
 void
 CMessages::InsertPlayerControlKeysInString(wchar *str)
 {
@@ -450,7 +510,7 @@ CMessages::InsertPlayerControlKeysInString(wchar *str)
 
 	if (!str) return;
 	uint16 strSize = GetWideStringLength(str);
-	memset(keybuf, 0, 256*sizeof(wchar));
+	memset(keybuf, 0, 256*sizeof(wchar)); // not memset? :O
 
 	wchar *_outstr = outstr;
 	for (i = 0; i < strSize;) {
@@ -460,9 +520,16 @@ CMessages::InsertPlayerControlKeysInString(wchar *str)
 #else
 		if (str[i] == '~' && str[i + 1] == 'k' && str[i + 2] == '~') {
 #endif
+			memset(keybuf, 0, 256 * sizeof(wchar));
 			i += 4;
-			bool done = false;
-			for (int32 cont = 0; cont < MAX_CONTROLLERACTIONS && !done; cont++) {
+			i += GetTokenPadKeyString(&str[i], keybuf) + 1;
+			uint16 keybuf_size = GetWideStringLength(keybuf);
+			for (uint16 j = 0; j < keybuf_size; j++) {
+				*(_outstr++) = keybuf[j];
+				keybuf[j] = '\0';
+			}
+
+			/*for (int32 cont = 0; cont < MAX_CONTROLLERACTIONS && !done; cont++) {
 				uint16 contSize = GetWideStringLength(ControlsManager.m_aActionNames[cont]);
 				if (contSize != 0) {
 					if (WideStringCompare(&str[i], ControlsManager.m_aActionNames[cont], contSize)) {
@@ -476,7 +543,7 @@ CMessages::InsertPlayerControlKeysInString(wchar *str)
 						i += contSize + 1;
 					}
 				}
-			}
+			}*/
 		} else {
 			*(_outstr++) = str[i++];
 		}
@@ -813,3 +880,13 @@ CMessages::ClearAllMessagesDisplayedByGame()
 	CHud::GetRidOfAllHudMessages();
 	CUserDisplay::Pager.ClearMessages();
 }
+
+void
+CMessages::ClearThisBigPrintNow(uint32 id)
+{
+	if (BIGMessages[id].m_Stack[0].m_pText)
+		ClearThisBigPrint(BIGMessages[id].m_Stack[0].m_pText);
+	CHud::m_BigMessage[id][0] = '\0';
+	BigMessageInUse[id] = 0.0f;
+}
+