Код:
#include "hud.h"
#include "cl_util.h"
#include "parsemsg.h"
#include <string.h>
#include <stdio.h>
#include "vgui_TeamFortressViewport.h"
#define MAX_MENU_STRING 512
char g_szMenuString[MAX_MENU_STRING];
char g_szPrelocalisedMenuString[MAX_MENU_STRING];
int KB_ConvertString(char *in, char **ppout);
DECLARE_MESSAGE(m_Menu, ShowMenu);
int CHudMenu::Init(void)
{
gHUD.AddHudElem(this);
HOOK_MESSAGE(ShowMenu);
InitHUDData();
return 1;
}
void CHudMenu::InitHUDData(void)
{
m_fMenuDisplayed = 0;
m_bitsValidSlots = 0;
m_iFlags &= ~HUD_ACTIVE;
Reset();
}
void CHudMenu::Reset(void)
{
g_szPrelocalisedMenuString[0] = 0;
m_fWaitingForMore = FALSE;
}
int CHudMenu::VidInit(void)
{
return 1;
}
static int menu_r, menu_g, menu_b, menu_x, menu_ralign;
static inline const char* ParseEscapeToken(const char* token)
{
int HcR, HcG, HcB;
gHUD.GetHudColor(0, 0, HcR, HcG, HcB);
if (*token != '^')
return token;
token++;
switch (*token)
{
case '\0':
return token;
case '0':
//^0 Hud_Color
menu_r = HcR;
menu_g = HcG;
menu_b = HcB;
break;
case '1':
menu_r = 255;
menu_g = 0;
menu_b = 0;
break;
case '2':
menu_r = 0;
menu_g = 255;
menu_b = 0;
break;
case '3':
menu_r = 255;
menu_g = 255;
menu_b = 0;
break;
case '4':
menu_r = 0;
menu_g = 0;
menu_b = 255;
break;
case '5':
menu_r = 0;
menu_g = 255;
menu_b = 255;
break;
case '6':
menu_r = 255;
menu_g = 0;
menu_b = 255;
break;
case '7':
menu_r = 80;
menu_g = 80;
menu_b = 80;
break;
case '8':
menu_r = 255;
menu_g = 255;
menu_b = 255;
break;
case '9':
menu_r = 255;
menu_g = 127;
menu_b = 39;
break;
}
return ++token;
}
static inline const char* ParseEscapeTokenCaracter2(const char* token)
{
int HcR, HcG, HcB;
gHUD.GetHudColor(0, 0, HcR, HcG, HcB);
if (*token != '<')
return token;
token++;
switch (*token)
{
case '\0':
return token;
case '0':
//^0 Hud_Color
menu_r = HcR;
menu_g = HcG;
menu_b = HcB;
break;
case '1':
menu_r = 255;
menu_g = 0;
menu_b = 0;
break;
case '2':
menu_r = 0;
menu_g = 255;
menu_b = 0;
break;
case '3':
menu_r = 255;
menu_g = 255;
menu_b = 0;
break;
case '4':
menu_r = 0;
menu_g = 0;
menu_b = 255;
break;
case '5':
menu_r = 0;
menu_g = 255;
menu_b = 255;
break;
case '6':
menu_r = 255;
menu_g = 0;
menu_b = 255;
break;
case '7':
menu_r = 80;
menu_g = 80;
menu_b = 80;
break;
case '8':
menu_r = 255;
menu_g = 255;
menu_b = 255;
break;
case '9':
menu_r = 255;
menu_g = 127;
menu_b = 39;
break;
}
return ++token;
}
static inline const char* ParseEscapeTokenCaracter3(const char* token)
{
if (*token != '$')
return token;
token++;
switch (*token)
{
case '\0':
return token;
case '0':
menu_r = 208;
menu_g = 55;
menu_b = 85;
break;
case '1':
menu_r = 113;
menu_g = 143;
menu_b = 83;
break;
case '2':
menu_r = 14;
menu_g = 120;
menu_b = 220;
break;
case '3':
menu_r = 40;
menu_g = 40;
menu_b = 40;
break;
case '4':
menu_r = 126;
menu_g = 51;
menu_b = 196;
break;
case '5':
menu_r = 118;
menu_g = 224;
menu_b = 12;
break;
case '6':
menu_r = 0;
menu_g = 144;
menu_b = 255;
break;
case '7':
menu_r = 37;
menu_g = 137;
menu_b = 83;
break;
case '8':
menu_r = 125;
menu_g = 5;
menu_b = 4;
break;
case '9':
menu_r = 246;
menu_g = 39;
menu_b = 5;
break;
}
return ++token;
}
static inline const char* ParseEscapeTokenMenu(const char* tokenMenu)
{
if (*tokenMenu != '\\')
return tokenMenu;
tokenMenu++;
switch (*tokenMenu)
{
case '\0':
return tokenMenu;
case 'a':
menu_r = 98;
menu_g = 137;
menu_b = 182;
break;
case 'b':
menu_r = 200;
menu_g = 90;
menu_b = 70;
break;
case 'w':
menu_r = 255;
menu_g = 255;
menu_b = 255;
break;
case 'd':
menu_r = 80;
menu_g = 80;
menu_b = 80;
break;
case 'y':
menu_r = 255;
menu_g = 255;
menu_b = 0;
break;
case 'r':
menu_r = 255;
menu_g = 0;
menu_b = 0;
break;
case 'R':
menu_x = ScreenWidth / 3;
menu_ralign = TRUE;
break;
}
return ++tokenMenu;
}
int CHudMenu::Draw(float flTime)
{
// check for if menu is set to disappear
if (m_flShutoffTime > 0)
{
if (m_flShutoffTime <= gHUD.m_flTime)
{ // times up, shutoff
m_fMenuDisplayed = 0;
m_iFlags &= ~HUD_ACTIVE;
return 1;
}
}
// don't draw the menu if the scoreboard is being shown
/*if (gViewPort && gViewPort->IsScoreBoardVisible())
return 1;*/
// draw the menu, along the left-hand side of the screen
// count the number of newlines
int nlc = 0;
int i;
for (i = 0; i < MAX_MENU_STRING && g_szMenuString[i] != '\0'; i++)
{
if (g_szMenuString[i] == '\n')
nlc++;
}
// center it
int y = (ScreenHeight / 2) - ((nlc / 2) * 12) - 40; // make sure it is above the say text
int R, G, B;
gHUD.GetHudColor(0, 0, R, G, B);
menu_r = R;
menu_g = G;
menu_b = B;
menu_x = 20;
menu_ralign = FALSE;
const char* sptr = g_szMenuString;
while (*sptr != '\0')
{
if (*sptr == '^')
{
sptr = ParseEscapeToken(sptr);
}
if (*sptr == '<')
{
sptr = ParseEscapeTokenCaracter2(sptr);
}
if (*sptr == '$')
{
sptr = ParseEscapeTokenCaracter3(sptr);
}
if (*sptr == '\\')
{
sptr = ParseEscapeTokenMenu(sptr);
}
else if (*sptr == '\n')
{
menu_ralign = FALSE;
menu_x = 20;
y += (12);
sptr++;
}
else
{
char menubuf[80];
const char *ptr = sptr;
while (*sptr != '\0' && *sptr != '\n' && *sptr != '^' && *sptr != '<' && *sptr != '$' && *sptr != '\\')
{
sptr++;
}
strncpy(menubuf, ptr, min((sptr - ptr), (int)sizeof(menubuf)));
menubuf[min((sptr - ptr), (int)(sizeof(menubuf)-1))] = '\0';
if (menu_ralign)
{
// IMPORTANT: Right-to-left rendered text does not parse escape tokens!
menu_x = gHUD.DrawHudStringReverse(menu_x, y, 0, menubuf, menu_r, menu_g, menu_b);
}
else
{
menu_x = gHUD.DrawHudString(menu_x, y, 320, menubuf, menu_r, menu_g, menu_b);
}
}
}
return 1;
}
// selects an item from the menu
void CHudMenu::SelectMenuItem(int menu_item)
{
// if menu_item is in a valid slot, send a menuselect command to the server
if ((menu_item > 0) && (m_bitsValidSlots & (1 << (menu_item - 1))))
{
char szbuf[32];
sprintf(szbuf, "menuselect %d\n", menu_item);
ClientCmd(szbuf);
// remove the menu
m_fMenuDisplayed = 0;
m_iFlags &= ~HUD_ACTIVE;
}
}
// Message handler for ShowMenu message
// takes four values:
// short: a bitfield of keys that are valid input
// char : the duration, in seconds, the menu should stay up. -1 means is stays until something is chosen.
// byte : a boolean, TRUE if there is more string yet to be received before displaying the menu, FALSE if it's the last string
// string: menu string to display
// if this message is never received, then scores will simply be the combined totals of the players.
int CHudMenu::MsgFunc_ShowMenu(const char *pszName, int iSize, void *pbuf)
{
char *temp = NULL;
BEGIN_READ(pbuf, iSize);
m_bitsValidSlots = READ_SHORT();
int DisplayTime = READ_CHAR();
int NeedMore = READ_BYTE();
if (DisplayTime > 0)
m_flShutoffTime = DisplayTime + gHUD.m_flTime;
else
m_flShutoffTime = -1;
if (m_bitsValidSlots)
{
if (!m_fWaitingForMore) // this is the start of a new menu
{
strncpy(g_szPrelocalisedMenuString, READ_STRING(), MAX_MENU_STRING);
}
else
{ // append to the current menu string
strncat(g_szPrelocalisedMenuString, READ_STRING(), MAX_MENU_STRING - strlen(g_szPrelocalisedMenuString));
}
g_szPrelocalisedMenuString[MAX_MENU_STRING - 1] = 0; // ensure null termination (strncat/strncpy does not)
if (!NeedMore)
{ // we have the whole string, so we can localise it now
strcpy(g_szMenuString, gHUD.m_TextMessage.BufferedLocaliseTextString(g_szPrelocalisedMenuString));
// Swap in characters
if (KB_ConvertString(g_szMenuString, &temp))
{
strcpy(g_szMenuString, temp);
free(temp);
}
}
m_fMenuDisplayed = 1;
m_iFlags |= HUD_ACTIVE;
}
else
{
m_fMenuDisplayed = 0; // no valid slots means that the menu should be turned off
m_iFlags &= ~HUD_ACTIVE;
}
m_fWaitingForMore = NeedMore;
return 1;
}