core/MenuStyle_Base.cpp
author Nicholas Hastings <skamonkey@gmail.com>
Sat Jun 02 10:07:25 2012 -0400 (2012-06-02)
changeset 3549 657ea0e26e7a
parent 3515 c14c14e06cc1
child 3564 48c4e173997f
permissions -rw-r--r--
Added missing CS:GO GameRulesProxy gamedata.
     1 /**
     2  * vim: set ts=4 :
     3  * =============================================================================
     4  * SourceMod
     5  * Copyright (C) 2004-2008 AlliedModders LLC.  All rights reserved.
     6  * =============================================================================
     7  *
     8  * This program is free software; you can redistribute it and/or modify it under
     9  * the terms of the GNU General Public License, version 3.0, as published by the
    10  * Free Software Foundation.
    11  * 
    12  * This program is distributed in the hope that it will be useful, but WITHOUT
    13  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
    14  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
    15  * details.
    16  *
    17  * You should have received a copy of the GNU General Public License along with
    18  * this program.  If not, see <http://www.gnu.org/licenses/>.
    19  *
    20  * As a special exception, AlliedModders LLC gives you permission to link the
    21  * code of this program (as well as its derivative works) to "Half-Life 2," the
    22  * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
    23  * by the Valve Corporation.  You must obey the GNU General Public License in
    24  * all respects for all other code used.  Additionally, AlliedModders LLC grants
    25  * this exception to all derivative works.  AlliedModders LLC defines further
    26  * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
    27  * or <http://www.sourcemod.net/license.php>.
    28  *
    29  * Version: $Id$
    30  */
    31 
    32 #include <stdarg.h>
    33 #include "sm_stringutil.h"
    34 #include "MenuStyle_Base.h"
    35 #include "PlayerManager.h"
    36 #include "MenuManager.h"
    37 #include "HandleSys.h"
    38 #include "CellRecipientFilter.h"
    39 #if defined MENU_DEBUG
    40 #include "Logger.h"
    41 #endif
    42 
    43 BaseMenuStyle::BaseMenuStyle() : m_WatchList(256), m_hHandle(BAD_HANDLE)
    44 {
    45 }
    46 
    47 Handle_t BaseMenuStyle::GetHandle()
    48 {
    49 	/* Don't create the handle until we need it */
    50 	if (m_hHandle == BAD_HANDLE)
    51 	{
    52 		m_hHandle = g_Menus.CreateStyleHandle(this);
    53 	}
    54 
    55 	return m_hHandle;
    56 }
    57 
    58 void BaseMenuStyle::AddClientToWatch(int client)
    59 {
    60 #if defined MENU_DEBUG
    61 	g_Logger.LogMessage("[SM_MENU] AddClientToWatch(%d)", client);
    62 #endif
    63 	m_WatchList.push_back(client);
    64 }
    65 
    66 void BaseMenuStyle::RemoveClientFromWatch(int client)
    67 {
    68 #if defined MENU_DEBUG
    69 	g_Logger.LogMessage("[SM_MENU] RemoveClientFromWatch(%d)", client);
    70 #endif
    71 	m_WatchList.remove(client);
    72 }
    73 
    74 void BaseMenuStyle::_CancelClientMenu(int client, MenuCancelReason reason, bool bAutoIgnore/* =false */)
    75 {
    76 #if defined MENU_DEBUG
    77 	g_Logger.LogMessage("[SM_MENU] _CancelClientMenu() (client %d) (bAutoIgnore %d) (reason %d)", client, bAutoIgnore, reason);
    78 #endif
    79 	CBaseMenuPlayer *player = GetMenuPlayer(client);
    80 	menu_states_t &states = player->states;
    81 
    82 	bool bOldIgnore = player->bAutoIgnore;
    83 	if (bAutoIgnore)
    84 	{
    85 		player->bAutoIgnore = true;
    86 	}
    87 
    88 	/* Save states */
    89 	IMenuHandler *mh = states.mh;
    90 	IBaseMenu *menu = states.menu;
    91 
    92 	/* Clear menu */
    93 	player->bInMenu = false;
    94 	if (player->menuHoldTime)
    95 	{
    96 		RemoveClientFromWatch(client);
    97 	}
    98 
    99 	/* Fire callbacks */
   100 	mh->OnMenuCancel(menu, client, reason);
   101 	
   102 	/* Only fire end if there's a valid menu */
   103 	if (menu)
   104 	{
   105 		mh->OnMenuEnd(menu, MenuEnd_Cancelled);
   106 	}
   107 
   108 	if (bAutoIgnore)
   109 	{
   110 		player->bAutoIgnore = bOldIgnore;
   111 	}
   112 }
   113 
   114 void BaseMenuStyle::CancelMenu(CBaseMenu *menu)
   115 {
   116 #if defined MENU_DEBUG
   117 	g_Logger.LogMessage("[SM_MENU] CancelMenu() (menu %p)", menu);
   118 #endif
   119 	int maxClients = g_Players.GetMaxClients();
   120 	for (int i=1; i<=maxClients; i++)
   121 	{
   122 		CBaseMenuPlayer *player = GetMenuPlayer(i);
   123 		if (player->bInMenu)
   124 		{
   125 			menu_states_t &states = player->states;
   126 			if (states.menu == menu)
   127 			{
   128 				_CancelClientMenu(i, MenuCancel_Interrupted);
   129 			}
   130 		}
   131 	}
   132 }
   133 
   134 bool BaseMenuStyle::CancelClientMenu(int client, bool autoIgnore)
   135 {
   136 #if defined MENU_DEBUG
   137 	g_Logger.LogMessage("[SM_MENU] CancelClientMenu() (client %d) (bAutoIgnore %d)", client, autoIgnore);
   138 #endif
   139 	if (client < 1 || client > g_Players.MaxClients())
   140 	{
   141 		return false;
   142 	}
   143 
   144 	CBaseMenuPlayer *player = GetMenuPlayer(client);
   145 	if (!player->bInMenu)
   146 	{
   147 		return false;
   148 	}
   149 
   150 	_CancelClientMenu(client, MenuCancel_Interrupted, autoIgnore);
   151 
   152 	return true;
   153 }
   154 
   155 MenuSource BaseMenuStyle::GetClientMenu(int client, void **object)
   156 {
   157 	if (client < 1 || client > g_Players.GetMaxClients())
   158 	{
   159 		return MenuSource_None;
   160 	}
   161 
   162 	CBaseMenuPlayer *player = GetMenuPlayer(client);
   163 
   164 	if (player->bInMenu)
   165 	{
   166 		IBaseMenu *menu;
   167 		if ((menu=player->states.menu) != NULL)
   168 		{
   169 			if (object)
   170 			{
   171 				*object = menu;
   172 			}
   173 			return MenuSource_BaseMenu;
   174 		}
   175 
   176 		return MenuSource_Display;
   177 	} else if (player->bInExternMenu) {
   178 		if (player->menuHoldTime != 0
   179 			&& (gpGlobals->curtime > player->menuStartTime + player->menuHoldTime))
   180 		{
   181 			player->bInExternMenu = false;
   182 			return MenuSource_None;
   183 		}
   184 		return MenuSource_External;
   185 	}
   186 
   187 	return MenuSource_None;
   188 }
   189 
   190 void BaseMenuStyle::OnClientDisconnected(int client)
   191 {
   192 #if defined MENU_DEBUG
   193 	g_Logger.LogMessage("[SM_MENU] OnClientDisconnected(%d)", client);
   194 #endif
   195 	CBaseMenuPlayer *player = GetMenuPlayer(client);
   196 	if (!player->bInMenu)
   197 	{
   198 		return;
   199 	}
   200 
   201 	_CancelClientMenu(client, MenuCancel_Disconnected, true);
   202 
   203 	player->bInMenu = false;
   204 	player->bInExternMenu = false;
   205 }
   206 
   207 static int do_lookup[256];
   208 void BaseMenuStyle::ProcessWatchList()
   209 {
   210 	if (!m_WatchList.size())
   211 	{
   212 		return;
   213 	}
   214 
   215 #if defined MENU_DEBUG
   216 	g_Logger.LogMessage("BaseMenuStyle::ProcessWatchList(%d,%d,%d,%d,%d,%p)", 
   217 		m_WatchList.m_Size,
   218 		m_WatchList.m_FirstLink,
   219 		m_WatchList.m_FreeNodes,
   220 		m_WatchList.m_LastLink,
   221 		m_WatchList.m_MaxSize,
   222 		m_WatchList.m_Nodes);
   223 #endif
   224 
   225 	unsigned int total = 0;
   226 	for (FastLink<int>::iterator iter=m_WatchList.begin(); iter!=m_WatchList.end(); ++iter)
   227 	{
   228 		do_lookup[total++] = (*iter);
   229 	}
   230 
   231 #if defined MENU_DEBUG
   232 	if (total)
   233 	{
   234 		g_Logger.LogMessage("[SM_MENU] ProcessWatchList() found %d clients", total);
   235 	}
   236 #endif
   237 
   238 	int client;
   239 	CBaseMenuPlayer *player;
   240 	float curTime = gpGlobals->curtime;
   241 	for (unsigned int i=0; i<total; i++)
   242 	{
   243 		client = do_lookup[i];
   244 		player = GetMenuPlayer(client);
   245 #if defined MENU_DEBUG
   246 		g_Logger.LogMessage("[SM_MENU] ProcessWatchList() (client %d) (bInMenu %d) (menuHoldTime %d) (curTime %f) (menuStartTime %f)",
   247 			client,
   248 			player->bInMenu,
   249 			player->menuHoldTime,
   250 			curTime,
   251 			player->menuStartTime);
   252 #endif
   253 		if (!player->bInMenu || !player->menuHoldTime)
   254 		{
   255 #if defined MENU_DEBUG
   256 			g_Logger.LogMessage("[SM_MENU] ProcessWatchList() removing client %d", client);
   257 #endif
   258 			m_WatchList.remove(client);
   259 			continue;
   260 		}
   261 		if (curTime > player->menuStartTime + player->menuHoldTime)
   262 		{
   263 			_CancelClientMenu(client, MenuCancel_Timeout, false);
   264 		}
   265 	}
   266 }
   267 
   268 void BaseMenuStyle::ClientPressedKey(int client, unsigned int key_press)
   269 {
   270 #if defined MENU_DEBUG
   271 	g_Logger.LogMessage("[SM_MENU] ClientPressedKey() (client %d) (key_press %d)", client, key_press);
   272 #endif
   273 	CBaseMenuPlayer *player = GetMenuPlayer(client);
   274 
   275 	/* First question: Are we in a menu? */
   276 	if (!player->bInMenu)
   277 	{
   278 		return;
   279 	}
   280 
   281 	bool cancel = false;
   282 	unsigned int item = 0;
   283 	MenuCancelReason reason = MenuCancel_Exit;
   284 	MenuEndReason end_reason = MenuEnd_Selected;
   285 	menu_states_t &states = player->states;
   286 
   287 	/* Save variables */
   288 	IMenuHandler *mh = states.mh;
   289 	IBaseMenu *menu = states.menu;
   290 
   291 	unsigned int item_on_page = states.item_on_page;
   292 
   293 	assert(mh != NULL);
   294 
   295 	if (menu == NULL)
   296 	{
   297 		item = key_press;
   298 	} else if (key_press < 1 || key_press > GetMaxPageItems()) {
   299 		cancel = true;
   300 	} else {
   301 		ItemSelection type = states.slots[key_press].type;
   302 
   303 		/* Check if we should play a sound about the type */
   304 		if (g_Menus.MenuSoundsEnabled() && 
   305 			(!menu || (menu->GetMenuOptionFlags() & MENUFLAG_NO_SOUND) != MENUFLAG_NO_SOUND))
   306 		{
   307 			CellRecipientFilter filter;
   308 			cell_t clients[1];
   309 
   310 			clients[0] = client;
   311 			filter.Initialize(clients, 1);
   312 
   313 			const char *sound = g_Menus.GetMenuSound(type);
   314 
   315 			if (sound != NULL)
   316 			{
   317 				edict_t *pEdict = PEntityOfEntIndex(client);
   318 				if (pEdict)
   319 				{
   320 					ICollideable *pCollideable = pEdict->GetCollideable();
   321 
   322 					if (pCollideable)
   323 					{
   324 						const Vector & pos = pCollideable->GetCollisionOrigin();
   325 						enginesound->EmitSound(filter, 
   326 							client, 
   327 							CHAN_AUTO, 
   328 #if SOURCE_ENGINE >= SE_PORTAL2
   329 							sound, 
   330 							-1, 
   331 #endif
   332 							sound, 
   333 							VOL_NORM, 
   334 							ATTN_NORM, 
   335 #if SOURCE_ENGINE >= SE_PORTAL2
   336 							0, 
   337 #endif
   338 							0, 
   339 							PITCH_NORM, 
   340 							&pos);
   341 					}
   342 				}
   343 			}
   344 		}
   345 
   346 		/* For navigational items, we're going to redisplay */
   347 		if (type == ItemSel_Back)
   348 		{
   349 			if (!RedoClientMenu(client, ItemOrder_Descending))
   350 			{
   351 				cancel = true;
   352 				reason = MenuCancel_NoDisplay;
   353 				end_reason = MenuEnd_Cancelled;
   354 			} else {
   355 				return;
   356 			}
   357 		} else if (type == ItemSel_Next) {
   358 			if (!RedoClientMenu(client, ItemOrder_Ascending))
   359 			{
   360 				cancel = true;						/* I like Saltines. */
   361 				reason = MenuCancel_NoDisplay;
   362 				end_reason = MenuEnd_Cancelled;
   363 			} else {
   364 				return;
   365 			}
   366 		} else if (type == ItemSel_Exit || type == ItemSel_None) {
   367 			cancel = true;
   368 			reason = MenuCancel_Exit;
   369 			end_reason = MenuEnd_Exit;
   370 		} else if (type == ItemSel_ExitBack) {
   371 			cancel = true;
   372 			reason = MenuCancel_ExitBack;
   373 			end_reason = MenuEnd_ExitBack;
   374 		} else {
   375 			item = states.slots[key_press].item;
   376 		}
   377 	}
   378 
   379 	/* Clear states */
   380 	player->bInMenu = false;
   381 	if (player->menuHoldTime)
   382 	{
   383 		RemoveClientFromWatch(client);
   384 	}
   385 
   386 	Handle_t hndl = menu ? menu->GetHandle() : BAD_HANDLE;
   387 	AutoHandleRooter ahr(hndl);
   388 
   389 	if (cancel)
   390 	{
   391 		mh->OnMenuCancel(menu, client, reason);
   392 	} else {
   393 		mh->OnMenuSelect(menu, client, item);
   394 		if (mh->GetMenuAPIVersion2() >= 13)
   395 		{
   396 			mh->OnMenuSelect2(menu, client, item, item_on_page);
   397 		}
   398 	}
   399 
   400 	/* Only fire end for valid menus */
   401 	if (menu)
   402 	{
   403 		mh->OnMenuEnd(menu, end_reason);
   404 	}
   405 }
   406 
   407 bool BaseMenuStyle::DoClientMenu(int client, IMenuPanel *menu, IMenuHandler *mh, unsigned int time)
   408 {
   409 #if defined MENU_DEBUG
   410 	g_Logger.LogMessage("[SM_MENU] DoClientMenu() (client %d) (panel %p) (mh %p) (time %d)",
   411 		client,
   412 		menu,
   413 		mh,
   414 		time);
   415 #endif
   416 	CPlayer *pPlayer = g_Players.GetPlayerByIndex(client);
   417 	if (!pPlayer || pPlayer->IsFakeClient() || !pPlayer->IsInGame())
   418 	{
   419 		return false;
   420 	}
   421 
   422 	CBaseMenuPlayer *player = GetMenuPlayer(client);
   423 	if (player->bAutoIgnore)
   424 	{
   425 		return false;
   426 	}
   427 
   428 	/* For the duration of this, we are going to totally ignore whether
   429 	* the player is already in a menu or not (except to cancel the old one).
   430 	* Instead, we are simply going to ignore any further menu displays, so
   431 	* this display can't be interrupted.
   432 	*/
   433 	player->bAutoIgnore = true;
   434 
   435 	/* Cancel any old menus */
   436 	menu_states_t &states = player->states;
   437 	if (player->bInMenu)
   438 	{
   439 		_CancelClientMenu(client, MenuCancel_Interrupted, true);
   440 	}
   441 
   442 	states.firstItem = 0;
   443 	states.lastItem = 0;
   444 	states.menu = NULL;
   445 	states.mh = mh;
   446 	states.apiVers = SMINTERFACE_MENUMANAGER_VERSION;
   447 	player->bInMenu = true;
   448 	player->bInExternMenu = false;
   449 	player->menuStartTime = gpGlobals->curtime;
   450 	player->menuHoldTime = time;
   451 
   452 	if (time)
   453 	{
   454 		AddClientToWatch(client);
   455 	}
   456 
   457 	/* Draw the display */
   458 	SendDisplay(client, menu);
   459 
   460 	/* We can be interrupted again! */
   461 	player->bAutoIgnore = false;
   462 
   463 	return true;
   464 }
   465 
   466 bool BaseMenuStyle::DoClientMenu(int client,
   467 								 CBaseMenu *menu,
   468 								 unsigned int first_item,
   469 								 IMenuHandler *mh,
   470 								 unsigned int time)
   471 {
   472 #if defined MENU_DEBUG
   473 	g_Logger.LogMessage("[SM_MENU] DoClientMenu() (client %d) (menu %p) (mh %p) (time %d)",
   474 		client,
   475 		menu,
   476 		mh,
   477 		time);
   478 #endif
   479 	mh->OnMenuStart(menu);
   480 
   481 	CPlayer *pPlayer = g_Players.GetPlayerByIndex(client);
   482 	if (!pPlayer || pPlayer->IsFakeClient() || !pPlayer->IsInGame())
   483 	{
   484 #if defined MENU_DEBUG
   485 		g_Logger.LogMessage("[SM_MENU] DoClientMenu(): Failed to display to client %d", client);
   486 #endif
   487 		mh->OnMenuCancel(menu, client, MenuCancel_NoDisplay);
   488 		mh->OnMenuEnd(menu, MenuEnd_Cancelled);
   489 		return false;
   490 	}
   491 
   492 	CBaseMenuPlayer *player = GetMenuPlayer(client);
   493 	if (player->bAutoIgnore)
   494 	{
   495 #if defined MENU_DEBUG
   496 		g_Logger.LogMessage("[SM_MENU] DoClientMenu(): Client %d is autoIgnoring", client);
   497 #endif
   498 		mh->OnMenuCancel(menu, client, MenuCancel_NoDisplay);
   499 		mh->OnMenuEnd(menu, MenuEnd_Cancelled);
   500 		return false;
   501 	}
   502 
   503 	/* For the duration of this, we are going to totally ignore whether
   504 	 * the player is already in a menu or not (except to cancel the old one).
   505 	 * Instead, we are simply going to ignore any further menu displays, so
   506 	 * this display can't be interrupted.
   507 	 */
   508 	player->bAutoIgnore = true;
   509 
   510 	/* Cancel any old menus */
   511 	menu_states_t &states = player->states;
   512 	if (player->bInMenu)
   513 	{
   514 #if defined MENU_DEBUG
   515 		g_Logger.LogMessage("[SM_MENU] DoClientMenu(): Cancelling old menu to client %d", client);
   516 #endif
   517 		_CancelClientMenu(client, MenuCancel_Interrupted, true);
   518 	}
   519 
   520 	states.firstItem = 0;
   521 	states.lastItem = first_item;
   522 	states.menu = menu;
   523 	states.mh = mh;
   524 	states.apiVers = SMINTERFACE_MENUMANAGER_VERSION;
   525 
   526 	IMenuPanel *display = g_Menus.RenderMenu(client, states, ItemOrder_Ascending);
   527 	if (!display)
   528 	{
   529 #if defined MENU_DEBUG
   530 		g_Logger.LogMessage("[SM_MENU] DoClientMenu(): Failed to render to client %d", client);
   531 #endif
   532 		player->bAutoIgnore = false;
   533 		player->bInMenu = false;
   534 		mh->OnMenuCancel(menu, client, MenuCancel_NoDisplay);
   535 		mh->OnMenuEnd(menu, MenuEnd_Cancelled);
   536 		return false;
   537 	}
   538 
   539 	/* Finally, set our states */
   540 	player->bInMenu = true;
   541 	player->bInExternMenu = false;
   542 	player->menuStartTime = gpGlobals->curtime;
   543 	player->menuHoldTime = time;
   544 
   545 	if (time)
   546 	{
   547 		AddClientToWatch(client);
   548 	}
   549 
   550 	/* Draw the display */
   551 	SendDisplay(client, display);
   552 
   553 	/* Free the display pointer */
   554 	display->DeleteThis();
   555 
   556 	/* We can be interrupted again! */
   557 	player->bAutoIgnore = false;
   558 
   559 #if defined MENU_DEBUG
   560 	g_Logger.LogMessage("[SM_MENU] DoClientMenu() finished successfully (client %d)", client);
   561 #endif
   562 
   563 	return true;
   564 }
   565 
   566 bool BaseMenuStyle::RedoClientMenu(int client, ItemOrder order)
   567 {
   568 #if defined MENU_DEBUG
   569 	g_Logger.LogMessage("[SM_MENU] RedoClientMenu() (client %d) (order %d)", client, order);
   570 #endif
   571 	CBaseMenuPlayer *player = GetMenuPlayer(client);
   572 	menu_states_t &states = player->states;
   573 
   574 	player->bAutoIgnore = true;
   575 	IMenuPanel *display = g_Menus.RenderMenu(client, states, order);
   576 	if (!display)
   577 	{
   578 #if defined MENU_DEBUG
   579 		g_Logger.LogMessage("[SM_MENU] RedoClientMenu(): Failed to render menu");
   580 #endif
   581 		if (player->menuHoldTime)
   582 		{
   583 			RemoveClientFromWatch(client);
   584 		}
   585 		player->bAutoIgnore = false;
   586 		return false;
   587 	}
   588 
   589 	SendDisplay(client, display);
   590 
   591 	display->DeleteThis();
   592 
   593 	player->bAutoIgnore = false;
   594 
   595 #if defined MENU_DEBUG
   596 	g_Logger.LogMessage("[SM_MENU] RedoClientMenu(): Succeeded to client %d", client);
   597 #endif
   598 
   599 	return true;
   600 }
   601 
   602 CBaseMenu::CBaseMenu(IMenuHandler *pHandler, IMenuStyle *pStyle, IdentityToken_t *pOwner) : 
   603 m_pStyle(pStyle), m_Strings(512), m_Pagination(7), m_bShouldDelete(false), m_bCancelling(false), 
   604 m_pOwner(pOwner ? pOwner : g_pCoreIdent), m_bDeleting(false), m_bWillFreeHandle(false), 
   605 m_hHandle(BAD_HANDLE), m_pHandler(pHandler), m_nFlags(MENUFLAG_BUTTON_EXIT)
   606 {
   607 }
   608 
   609 CBaseMenu::~CBaseMenu()
   610 {
   611 }
   612 
   613 Handle_t CBaseMenu::GetHandle()
   614 {
   615 	if (!m_hHandle)
   616 	{
   617 		m_hHandle = g_Menus.CreateMenuHandle(this, m_pOwner);
   618 	}
   619 
   620 	return m_hHandle;
   621 }
   622 
   623 bool CBaseMenu::AppendItem(const char *info, const ItemDrawInfo &draw)
   624 {
   625 	if (m_Pagination == (unsigned)MENU_NO_PAGINATION
   626 		&& m_items.size() >= m_pStyle->GetMaxPageItems())
   627 	{
   628 		return false;
   629 	}
   630 
   631 	CItem item;
   632 
   633 	item.infoString = m_Strings.AddString(info);
   634 	if (draw.display)
   635 	{
   636 		item.displayString = m_Strings.AddString(draw.display);
   637 	}
   638 	item.style = draw.style;
   639 
   640 	m_items.push_back(item);
   641 
   642 	return true;
   643 }
   644 
   645 bool CBaseMenu::InsertItem(unsigned int position, const char *info, const ItemDrawInfo &draw)
   646 {
   647 	if (m_Pagination == (unsigned)MENU_NO_PAGINATION
   648 		&& m_items.size() >= m_pStyle->GetMaxPageItems())
   649 	{
   650 		return false;
   651 	}
   652 
   653 	if (position >= m_items.size())
   654 	{
   655 		return false;
   656 	}
   657 
   658 	CItem item;
   659 	item.infoString = m_Strings.AddString(info);
   660 	if (draw.display)
   661 	{
   662 		item.displayString = m_Strings.AddString(draw.display);
   663 	}
   664 	item.style = draw.style;
   665 
   666 	CVector<CItem>::iterator iter = m_items.iterAt(position);
   667 	m_items.insert(iter, item);
   668 
   669 	return true;
   670 }
   671 
   672 bool CBaseMenu::RemoveItem(unsigned int position)
   673 {
   674 	if (position >= m_items.size())
   675 	{
   676 		return false;
   677 	}
   678 
   679 	m_items.erase(m_items.iterAt(position));
   680 
   681 	if (m_items.size() == 0)
   682 	{
   683 		m_Strings.Reset();
   684 	}
   685 
   686 	return true;
   687 }
   688 
   689 void CBaseMenu::RemoveAllItems()
   690 {
   691 	m_items.clear();
   692 	m_Strings.Reset();
   693 }
   694 
   695 const char *CBaseMenu::GetItemInfo(unsigned int position, ItemDrawInfo *draw/* =NULL */)
   696 {
   697 	if (position >= m_items.size())
   698 	{
   699 		return NULL;
   700 	}
   701 
   702 	if (draw)
   703 	{
   704 		draw->display = m_Strings.GetString(m_items[position].displayString);
   705 		draw->style = m_items[position].style;
   706 	}
   707 
   708 	return m_Strings.GetString(m_items[position].infoString);
   709 }
   710 
   711 unsigned int CBaseMenu::GetItemCount()
   712 {
   713 	return m_items.size();
   714 }
   715 
   716 bool CBaseMenu::SetPagination(unsigned int itemsPerPage)
   717 {
   718 	if (itemsPerPage > 7 || itemsPerPage == 1)
   719 	{
   720 		return false;
   721 	}
   722 
   723 	if (itemsPerPage == MENU_NO_PAGINATION
   724 		&& m_Pagination != MENU_NO_PAGINATION)
   725 	{
   726 		m_nFlags &= ~MENUFLAG_BUTTON_EXIT;
   727 	}
   728 
   729 	m_Pagination = itemsPerPage;
   730 
   731 	return true;
   732 }
   733 
   734 unsigned int CBaseMenu::GetPagination()
   735 {
   736 	return m_Pagination;
   737 }
   738 
   739 IMenuStyle *CBaseMenu::GetDrawStyle()
   740 {
   741 	return m_pStyle;
   742 }
   743 
   744 void CBaseMenu::SetDefaultTitle(const char *message)
   745 {
   746 	m_Title.assign(message);
   747 }
   748 
   749 const char *CBaseMenu::GetDefaultTitle()
   750 {
   751 	return m_Title.c_str();
   752 }
   753 
   754 void CBaseMenu::Cancel()
   755 {
   756 	if (m_bCancelling)
   757 	{
   758 		return;
   759 	}
   760 
   761 #if defined MENU_DEBUG
   762 	g_Logger.LogMessage("[SM_MENU] CBaseMenu::Cancel(%p) (m_bShouldDelete %d)",
   763 		this,
   764 		m_bShouldDelete);
   765 #endif
   766 
   767 	m_bCancelling = true;
   768 	Cancel_Finally();
   769 	m_bCancelling = false;
   770 
   771 	if (m_bShouldDelete)
   772 	{
   773 		InternalDelete();
   774 	}
   775 }
   776 
   777 void CBaseMenu::Destroy(bool releaseHandle)
   778 {
   779 	/* Check if we shouldn't be here */
   780 	if (m_bDeleting)
   781 	{
   782 		return;
   783 	}
   784 
   785 #if defined MENU_DEBUG
   786 	g_Logger.LogMessage("[SM_MENU] CBaseMenu::Destroy(%p) (release %d) (m_bCancelling %d) (m_bShouldDelete %d)",
   787 		this,
   788 		releaseHandle,
   789 		m_bCancelling,
   790 		m_bShouldDelete);
   791 #endif
   792 
   793 	/* Save the destruction hint about our handle */
   794 	m_bWillFreeHandle = releaseHandle;
   795 
   796 	/* Now actually do stuff */
   797 	if (!m_bCancelling || m_bShouldDelete)
   798 	{
   799 		Cancel();
   800 		InternalDelete();
   801 	} else {
   802 		m_bShouldDelete = true;
   803 	}
   804 }
   805 
   806 void CBaseMenu::InternalDelete()
   807 {
   808 	if (m_bWillFreeHandle && m_hHandle != BAD_HANDLE)
   809 	{
   810 		Handle_t hndl = m_hHandle;
   811 		HandleSecurity sec;
   812 
   813 		sec.pOwner = m_pOwner;
   814 		sec.pIdentity = g_pCoreIdent;
   815 
   816 		m_hHandle = BAD_HANDLE;
   817 		m_bDeleting = true;
   818 		g_HandleSys.FreeHandle(hndl, &sec);
   819 	}
   820 
   821 	m_pHandler->OnMenuDestroy(this);
   822 
   823 	delete this;
   824 }
   825 
   826 unsigned int CBaseMenu::GetMenuOptionFlags()
   827 {
   828 	return m_nFlags;
   829 }
   830 
   831 void CBaseMenu::SetMenuOptionFlags(unsigned int flags)
   832 {
   833 	m_nFlags = flags;
   834 }
   835 
   836 IMenuHandler *CBaseMenu::GetHandler()
   837 {
   838 	return m_pHandler;
   839 }
   840 
   841 unsigned int CBaseMenu::GetBaseMemUsage()
   842 {
   843 	return m_Title.size() 
   844 		+ m_Strings.GetMemTable()->GetMemUsage()
   845 		+ (m_items.size() * sizeof(CItem));
   846 }