////////////////////////////////////////////////////////////////
// Microsoft Systems Journal -- November 1999
// If this code works, it was written by Paul DiLascia.
// If not, I don't know who wrote it.
// Compiles with Visual C++ 6.0, runs on Windows 98 and probably NT too.
//
#include "stdafx.h"
#include "MenuInit.h"

#ifdef _DEBUG
#	define new DEBUG_NEW
#	undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

//////////////////
// This static function can be sed to initialize any menu
// when you get WM_INITMENUPOPUP
//
void CPopupMenuInit::Init(CCmdTarget* pTarg, CMenu* pMenu, BOOL bAutoMenuEnable)
{
	CCmdUI state;
	state.m_pMenu = pMenu;
	_ASSERTE(state.m_pOther == NULL);
	_ASSERTE(state.m_pParentMenu == NULL);

	CWnd* pWnd = DYNAMIC_DOWNCAST(CWnd, pTarg);
	if (pWnd)
	{
		// determine if menu is popup in top-level menu and set m_pOther to
		//  it if so (m_pParentMenu == NULL indicates that it is secondary popup)
		HMENU hParentMenu;
		if (AfxGetThreadState()->m_hTrackingMenu == pMenu->m_hMenu)
			state.m_pParentMenu = pMenu; // parent == child for tracking popup
		else if ((hParentMenu = ::GetMenu(*pWnd)) != NULL)
		{
			CWnd* pParent = pWnd->GetTopLevelParent();
			// child windows don't have menus -- need to go to the top!
			if (pParent != NULL && (hParentMenu = ::GetMenu(pParent->m_hWnd)) != NULL)
			{
				int nIndexMax = ::GetMenuItemCount(hParentMenu);
				for (int nIndex = 0; nIndex < nIndexMax; nIndex++)
				{
					if (::GetSubMenu(hParentMenu, nIndex) == pMenu->m_hMenu)
					{
						// when popup is found, m_pParentMenu is containing menu
						state.m_pParentMenu = CMenu::FromHandle(hParentMenu);
						break;
					}
				}
			}
		}
	}

	state.m_nIndexMax = pMenu->GetMenuItemCount();
	for (state.m_nIndex = 0; state.m_nIndex < state.m_nIndexMax; state.m_nIndex++)
	{
		state.m_nID = pMenu->GetMenuItemID(state.m_nIndex);
		if (state.m_nID == 0)
			continue; // menu separator or invalid cmd - ignore it

		_ASSERTE(state.m_pOther == NULL);
		_ASSERTE(state.m_pMenu != NULL);
		if (state.m_nID == (UINT)-1)
		{
			// possibly a popup menu, route to first item of that popup
			state.m_pSubMenu = pMenu->GetSubMenu(state.m_nIndex);
			if (state.m_pSubMenu == NULL || (state.m_nID = state.m_pSubMenu->GetMenuItemID(0)) == 0
				|| state.m_nID == (UINT)-1)
			{
				continue; // first item of popup can't be routed to
			}
			state.DoUpdate(pTarg, FALSE); // popups are never auto disabled
		}
		else
		{
			// normal menu item
			// Auto enable/disable if 'bAutoMenuEnable'
			// and command is _not_ a system command.
			state.m_pSubMenu = NULL;
			state.DoUpdate(pTarg, bAutoMenuEnable && state.m_nID < 0xF000);
		}

		// adjust for menu deletions and additions
		UINT nCount = pMenu->GetMenuItemCount();
		if (nCount < state.m_nIndexMax)
		{
			state.m_nIndex -= (state.m_nIndexMax - nCount);
			while (state.m_nIndex < nCount && pMenu->GetMenuItemID(state.m_nIndex) == state.m_nID)
			{
				state.m_nIndex++;
			}
		}
		state.m_nIndexMax = nCount;
	}
}