// This is a part of the Microsoft Foundation Classes C++ library.
// Copyright (C) Microsoft Corporation
// All rights reserved.
//
// This source code is only intended as a supplement to the
// Microsoft Foundation Classes Reference and related
// electronic documentation provided with the library.
// See these sources for detailed information regarding the
// Microsoft Foundation Classes product.

#include "stdafx.h"
#include "afxmenuimages.h"
#include "afxcontrolbarutil.h"
#include "afxmenubutton.h"
#include "afxcontextmenumanager.h"
#include "afxpopupmenu.h"
#include "afxtagmanager.h"
#include "afxctrlcontainer.h"
#include "afxdialogex.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif

static const int nImageHorzMargin = 10;

/////////////////////////////////////////////////////////////////////////////
// CMFCMenuButton

IMPLEMENT_DYNAMIC(CMFCMenuButton, CMFCButton)

CMFCMenuButton::CMFCMenuButton()
{
	m_bRightArrow = FALSE;
	m_hMenu = NULL;
	m_nMenuResult = 0;
	m_bMenuIsActive = FALSE;
	m_bStayPressed = FALSE;
	m_bOSMenu = TRUE;
	m_bDefaultClick = FALSE;
	m_bClickOnMenu = FALSE;
}

CMFCMenuButton::~CMFCMenuButton()
{
}

BEGIN_MESSAGE_MAP(CMFCMenuButton, CMFCButton)
	ON_WM_KEYDOWN()
	ON_WM_LBUTTONDOWN()
	ON_WM_GETDLGCODE()
	ON_WM_LBUTTONUP()
	ON_WM_KILLFOCUS()
	ON_WM_LBUTTONDBLCLK()
	ON_MESSAGE(WM_MFC_INITCTRL, &CMFCMenuButton::OnInitControl)
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CMFCMenuButton message handlers

CSize CMFCMenuButton::SizeToContent(BOOL bCalcOnly)
{
	CSize size = CMFCButton::SizeToContent(FALSE);
	size.cx += CMenuImages::Size().cx;

	if (!bCalcOnly)
	{
		SetWindowPos(NULL, -1, -1, size.cx, size.cy, SWP_NOMOVE | SWP_NOACTIVATE | SWP_NOZORDER);
	}

	return size;
}

void CMFCMenuButton::OnDraw(CDC* pDC, const CRect& rect, UINT uiState)
{
	ASSERT_VALID(pDC);

	CSize sizeArrow = CMenuImages::Size();

	CRect rectParent = rect;
	rectParent.right -= sizeArrow.cx + nImageHorzMargin;

	CMFCButton::OnDraw(pDC, rectParent, uiState);

	CRect rectArrow = rect;
	rectArrow.left = rectParent.right;

	CMenuImages::Draw(pDC, m_bRightArrow ? CMenuImages::IdArrowRightLarge : CMenuImages::IdArrowDownLarge,
		rectArrow, (uiState & ODS_DISABLED) ? CMenuImages::ImageGray : CMenuImages::ImageBlack);

	if (m_bDefaultClick)
	{
		//----------------
		// Draw separator:
		//----------------
		CRect rectSeparator = rectArrow;
		rectSeparator.right = rectSeparator.left + 2;
		rectSeparator.DeflateRect(0, 2);

		if (!m_bWinXPTheme || m_bDontUseWinXPTheme)
		{
			rectSeparator.left += m_sizePushOffset.cx;
			rectSeparator.top += m_sizePushOffset.cy;
		}

		pDC->Draw3dRect(rectSeparator, GetGlobalData()->clrBtnDkShadow, GetGlobalData()->clrBtnHilite);
	}
}

void CMFCMenuButton::OnShowMenu()
{
	if (m_hMenu == NULL || m_bMenuIsActive)
	{
		return;
	}

	CRect rectWindow;
	GetWindowRect(rectWindow);

	int x, y;

	if (m_bRightArrow)
	{
		x = rectWindow.right;
		y = rectWindow.top;
	}
	else
	{
		x = rectWindow.left;
		y = rectWindow.bottom;
	}

	if (m_bStayPressed)
	{
		m_bPushed = TRUE;
		m_bHighlighted = TRUE;
	}

	m_bMenuIsActive = TRUE;
	Invalidate();

	if (!m_bOSMenu && afxContextMenuManager != NULL)
	{
		m_nMenuResult = afxContextMenuManager->TrackPopupMenu(m_hMenu, x, y, this);
		SetFocus();
	}
	else
	{
		m_nMenuResult = ::TrackPopupMenu(m_hMenu, TPM_LEFTALIGN | TPM_LEFTBUTTON | TPM_NONOTIFY | TPM_RETURNCMD, x, y, 0, GetSafeHwnd(), NULL);
	}

	CWnd* pParent = GetParent();

#ifdef _DEBUG
	if ((pParent->IsKindOf(RUNTIME_CLASS(CDialog))) && (!pParent->IsKindOf(RUNTIME_CLASS(CDialogEx))))
	{
		TRACE(_T("CMFCMenuButton parent is CDialog, should be CDialogEx for popup menu handling to work correctly.\n"));
	}
#endif

	if (m_nMenuResult != 0)
	{
		//-------------------------------------------------------
		// Trigger mouse up event(to button click notification):
		//-------------------------------------------------------
		if (pParent != NULL)
		{
			pParent->SendMessage( WM_COMMAND, MAKEWPARAM(GetDlgCtrlID(), BN_CLICKED), (LPARAM) m_hWnd);
		}
	}

	m_bPushed = FALSE;
	m_bHighlighted = FALSE;
	m_bMenuIsActive = FALSE;

	Invalidate();
	UpdateWindow();

	if (m_bCaptured)
	{
		ReleaseCapture();
		m_bCaptured = FALSE;
	}
}

void CMFCMenuButton::OnKeyDown(UINT nChar, UINT nRepCnt, UINT nFlags)
{
	if (nChar == VK_SPACE || nChar == VK_DOWN)
	{
		m_bClickOnMenu = TRUE;
		OnShowMenu();
		return;
	}

	CButton::OnKeyDown(nChar, nRepCnt, nFlags);
}

void CMFCMenuButton::OnLButtonDown(UINT nFlags, CPoint point)
{
	if (m_bMenuIsActive)
	{
		Default();
		return;
	}

	m_bClickOnMenu = TRUE;

	if (m_bDefaultClick)
	{
		CRect rectClient;
		GetClientRect(rectClient);

		CRect rectArrow = rectClient;
		rectArrow.left = rectArrow.right - CMenuImages::Size().cx - nImageHorzMargin;

		if (!rectArrow.PtInRect(point))
		{
			m_bClickOnMenu = FALSE;
			m_nMenuResult = 0;
			CMFCButton::OnLButtonDown(nFlags, point);
			return;
		}
	}

	SetFocus();
	OnShowMenu();
}

UINT CMFCMenuButton::OnGetDlgCode()
{
	return DLGC_WANTARROWS;
}

void CMFCMenuButton::OnLButtonUp(UINT nFlags, CPoint point)
{
	if (m_bStayPressed && m_bMenuIsActive && m_bPushed)
	{
		m_bClickiedInside = FALSE;

		CButton::OnLButtonUp(nFlags, point);

		if (m_bCaptured)
		{
			ReleaseCapture();
			m_bCaptured = FALSE;
		}
	}
	else if (!m_bClickOnMenu)
	{
		CMFCButton::OnLButtonUp(nFlags, point);
	}
}

void CMFCMenuButton::OnKillFocus(CWnd* pNewWnd)
{
	if (m_bStayPressed && m_bMenuIsActive && m_bPushed)
	{
		CButton::OnKillFocus(pNewWnd);

		if (m_bCaptured)
		{
			ReleaseCapture();
			m_bCaptured = FALSE;
		}

		m_bClickiedInside = FALSE;
		m_bHover = FALSE;
	}
	else
	{
		CMFCButton::OnKillFocus(pNewWnd);
	}
}

BOOL CMFCMenuButton::PreTranslateMessage(MSG* pMsg)
{
	if (pMsg->message == WM_KEYDOWN &&
		pMsg->wParam == VK_RETURN &&
		CMFCPopupMenu::GetActiveMenu() == NULL)
	{
		m_bClickOnMenu = TRUE;
		OnShowMenu();
		return TRUE;
	}

	return CMFCButton::PreTranslateMessage(pMsg);
}

void CMFCMenuButton::OnLButtonDblClk(UINT nFlags, CPoint point)
{
	if (!m_bMenuIsActive)
	{
		CMFCButton::OnLButtonDblClk(nFlags, point);
		m_bClickOnMenu = FALSE;
	}
	if (m_bCaptured)
	{
		ReleaseCapture ();
		m_bCaptured = FALSE;
	}
}

LRESULT CMFCMenuButton::OnInitControl(WPARAM wParam, LPARAM lParam)
{
	DWORD dwSize = (DWORD)wParam;
	BYTE* pbInitData = (BYTE*)lParam;

	CString strDst;
	CMFCControlContainer::UTF8ToString((LPSTR)pbInitData, strDst, dwSize);

	CTagManager tagManager(strDst);

	BOOL bOSMenu = FALSE;
	if (ReadBoolProp(tagManager, PS_MFCMenuButton_OSMenu, bOSMenu))
	{
		m_bOSMenu = bOSMenu;
	}

	BOOL bRightArrow = FALSE;
	if (ReadBoolProp(tagManager, PS_MFCMenuButton_RightArrow, bRightArrow))
	{
		m_bRightArrow = bRightArrow;
	}

	BOOL bStayPressed = FALSE;
	if (ReadBoolProp(tagManager, PS_MFCMenuButton_StayPressed, bStayPressed))
	{
		m_bStayPressed = bStayPressed;
	}

	BOOL bDefaultClick = FALSE;
	if (ReadBoolProp(tagManager, PS_MFCMenuButton_DefaultClick, bDefaultClick))
	{
		m_bDefaultClick = bDefaultClick;
	}

	BOOL bAutosize = FALSE;
	if (ReadBoolProp(tagManager, PS_MFCMenuButton_Autosize, bAutosize))
	{
		if (bAutosize)
		{
			SizeToContent();
		}
	}

	return 0;
}