src/Utils/MiscUI/MenuButton.cpp

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2011 - Sven Strickroth <email@cs-ware.de>
   4
   5 //based on:
   6 // Copyright (C) 2003-2006,2008 - Stefan Kueng
   7
   8 // This program is free software; you can redistribute it and/or
   9 // modify it under the terms of the GNU General Public License
  10 // as published by the Free Software Foundation; either version 2
  11 // of the License, or (at your option) any later version.
  12
  13 // This program is distributed in the hope that it will be useful,
  14 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  15 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16 // GNU General Public License for more details.
  17
  18 // You should have received a copy of the GNU General Public License
  19 // along with this program; if not, write to the Free Software Foundation,
  20 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  21 //
  22 #include "stdafx.h"
  23 #include "MenuButton.h"
  24
  25
  26 #ifdef _DEBUG
  27 #undef THIS_FILE
  28 static char THIS_FILE[]=__FILE__;
  29 #define new DEBUG_NEW
  30 #endif
  31
  32
  33 IMPLEMENT_DYNCREATE(CMenuButton, CMFCMenuButton)
  34
  35 CMenuButton::CMenuButton(void) : CMFCMenuButton()
  36         , m_nDefault(-1)
  37         , m_bMarkDefault(TRUE)
  38         , m_bRealMenuIsActive(false)
  39 {
  40         m_bOSMenu = TRUE;
  41         m_bDefaultClick = TRUE;
  42         m_bTransparent = TRUE;
  43         m_bMenuIsActive = TRUE;
  44
  45         m_btnMenu.CreatePopupMenu();
  46         m_hMenu = m_btnMenu.GetSafeHmenu();
  47 }
  48
  49 CMenuButton::~CMenuButton(void)
  50 {
  51 }
  52
  53 bool CMenuButton::SetCurrentEntry(INT_PTR entry)
  54 {
  55         if (entry < 0 || entry >= m_sEntries.GetCount())
  56                 return false;
  57
  58         m_nDefault = entry + 1;
  59         SetWindowText(m_sEntries[entry]);
  60         if (m_bMarkDefault)
  61                 m_btnMenu.SetDefaultItem((UINT)m_nDefault, FALSE);
  62
  63         return true;
  64 }
  65
  66 void CMenuButton::RemoveAll()
  67 {
  68         for (int index = 0; index < m_sEntries.GetCount(); index++)
  69                 m_btnMenu.RemoveMenu(0, MF_BYPOSITION);
  70         m_sEntries.RemoveAll();
  71         m_nDefault = m_nMenuResult = -1;
  72         m_bMenuIsActive = TRUE;
  73 }
  74
  75 INT_PTR CMenuButton::AddEntry(UINT iconId, const CString& sEntry)
  76 {
  77         INT_PTR ret = m_sEntries.Add(sEntry);
  78         m_btnMenu.AppendMenuIcon(m_sEntries.GetCount(), sEntry, iconId);
  79         if (m_sEntries.GetCount() == 2)
  80                 m_bMenuIsActive = FALSE;
  81
  82         if (ret == 0)
  83                 SetCurrentEntry(ret);
  84         return ret;
  85 }
  86
  87 INT_PTR CMenuButton::AddEntry(const CString& sEntry)
  88 {
  89         INT_PTR ret = m_sEntries.Add(sEntry);
  90         m_btnMenu.AppendMenuIcon(m_sEntries.GetCount(), sEntry);
  91         if (m_sEntries.GetCount() == 2)
  92                 m_bMenuIsActive = FALSE;
  93
  94         if (ret == 0)
  95                 SetCurrentEntry(ret);
  96         return ret;
  97 }
  98
  99 INT_PTR CMenuButton::AddEntries(const CStringArray& sEntries)
 100 {
 101         INT_PTR ret = -1;
 102         for (int index = 0; index < sEntries.GetCount(); index++)
 103         {
 104                 if (index == 0)
 105                         ret = AddEntry(sEntries[index]);
 106                 else
 107                         AddEntry(sEntries[index]);
 108         }
 109         return ret;
 110 }
 111
 112 BEGIN_MESSAGE_MAP(CMenuButton, CMFCMenuButton)
 113         ON_WM_DESTROY()
 114         ON_CONTROL_REFLECT_EX(BN_CLICKED, &CMenuButton::OnClicked)
 115 END_MESSAGE_MAP()
 116
 117
 118 BOOL CMenuButton::OnClicked()
 119 {
 120         SetCurrentEntry(GetCurrentEntry());
 121
 122         // let the parent handle the message the usual way
 123         return FALSE;
 124 }
 125
 126 BOOL CMenuButton::PreTranslateMessage(MSG* pMsg)
 127 {
 128         if(pMsg->message == WM_KEYDOWN)
 129         {
 130                 switch(pMsg->wParam)
 131                 {
 132                 case VK_F4:
 133                         OnShowMenu();
 134                         return TRUE;
 135                 }
 136         }
 137
 138         return CMFCMenuButton::PreTranslateMessage(pMsg);
 139 }
 140
 141 void CMenuButton::OnDestroy()
 142 {
 143         m_sEntries.RemoveAll();
 144
 145         CMFCMenuButton::OnDestroy();
 146 }
 147
 148 void CMenuButton::OnDraw(CDC* pDC, const CRect& rect, UINT uiState)
 149 {
 150         if (m_bMenuIsActive && !m_bRealMenuIsActive)
 151                 CMFCButton::OnDraw(pDC, rect, uiState);
 152         else
 153                 CMFCMenuButton::OnDraw(pDC, rect, uiState);
 154 }
 155
 156 void CMenuButton::OnShowMenu()
 157 {
 158         m_bRealMenuIsActive = true;
 159         CMFCMenuButton::OnShowMenu();
 160         m_bRealMenuIsActive = false;
 161 }