WTL::右鍵彈出菜單和托盤圖標

 　WTL嵌入類的架構能夠使整個程序很清晰，系統托盤氣泡提示、上下文菜單處理能夠分別獨立出來，如下是實現代碼，很清晰就不註釋了。基本上這兩個類不多須要修改，所以我把它們放到了系統包含文件搜索路徑中。

//ContextMenu.h
#pragma once
template <class T>
class CContextMenu
{
public:
             BOOL CreateContextMenu(UINT ID_Menu)
             {
                           T* pT = static_cast<T*>(this);
                           CMenu menu;
                           menu.LoadMenu(ID_Menu);
                           CMenu SubMenu(menu.GetSubMenu(0));
                           POINT pos;
                           GetCursorPos(&pos);
                           SubMenu.TrackPopupMenu(TPM_LEFTALIGN | TPM_RIGHTBUTTON, pos.x, pos.y, pT->m_hWnd);
                           return TRUE;
             }
};

//ShellIcon.h
#pragma once
#define WM_ICON WM_USER + 180
template <class T, class MenuT, int MenuID>
class CShellIcon : public MenuT
{
private:
             NOTIFYICONDATA m_data;
             UINT m_msgTaskbarRestart;
             CString m_appName;
public:
             CShellIcon()
             {
                           m_appName.LoadString(IDS_APPNAME);
                           m_msgTaskbarRestart = RegisterWindowMessage(TEXT("TaskbarCreated"));
             }
             ~CShellIcon()
             {
                           Shell_NotifyIcon(NIM_DELETE, &m_data);
             }
             BOOL CreateShellIcon()
             {
                           T* pT = static_cast<T*>(this);
                           SecureZeroMemory(&m_data, sizeof(m_data));
                           m_data.cbSize = sizeof(m_data);
                           m_data.hIcon = LoadIcon(_Module.get_m_hInst(), MAKEINTRESOURCE(IDR_MAINFRAME));
                           m_data.hWnd = pT->m_hWnd;
                           m_data.uID = IDR_MAINFRAME;
                           m_data.uFlags = NIF_ICON | NIF_MESSAGE | NIF_INFO | NIF_TIP;
                           m_data.uCallbackMessage = WM_ICON;
                           m_data.dwInfoFlags = NIIF_USER;
                           strcpy_s(m_data.szInfoTitle, m_appName);
                           strcpy_s(m_data.szTip, m_appName);
                           return Shell_NotifyIcon(NIM_ADD, &m_data);
             }
             void ModifyToolTips(LPCTSTR info)
             {
                           strcpy_s(m_data.szInfo, info);
             }
             BOOL DispalyToolTips()
             {
                           return Shell_NotifyIcon(NIM_MODIFY, &m_data);
             }
             BOOL BalloonToolTips(LPCTSTR info)
             {
                           ModifyToolTips(LPCTSTR info);
                           return DispalyToolTips();
             }
             BEGIN_MSG_MAP(CShellIcon)
                           MESSAGE_HANDLER(WM_ICON, OnIcon)
                           MESSAGE_HANDLER(m_msgTaskbarRestart, OnRestart)
                           MESSAGE_HANDLER(WM_SIZE, OnSize)
                           CHAIN_MSG_MAP(MenuT)
             END_MSG_MAP()
             LRESULT OnIcon(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
             {
                           T* pT = static_cast<T*>(this); char t;
                           if (wParam != IDR_MAINFRAME) return 1;
                           switch(lParam)
                           {
                           case WM_RBUTTONUP:
                                         t = *m_data.szInfo;
                                         *m_data.szInfo = '\0';
                                         Shell_NotifyIcon(NIM_MODIFY, &m_data);
                                         pT->CreateContextMenu(MenuID);
                                         *m_data.szInfo = t;
                                         break;
                           case WM_LBUTTONUP:
                                         pT->ShowWindow(SW_SHOW);
                                         OpenIcon(pT->m_hWnd);
                                         break;
//去掉下面的註釋能夠使鼠標懸停在圖標上時出現氣泡提示，我的不太喜歡
                           case WM_MOUSEMOVE:
             //                       DispalyToolTips(); break;
                           default:
                                         ;
                           }
                           return 0;
             }
//處理Explorer外殼崩潰後任務欄重建，你會發現不少程序都沒處理，結果就是Explorer一崩潰
//圖標就找不到了，若是最小化到任務欄更慘，還得用任務管理器關閉，這個函數我沒機會測試，但願有做用
             LRESULT OnRestart(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
             {
                           T* pT = static_cast<T*>(this);
                           SecureZeroMemory(&m_data, sizeof(m_data));
                           m_data.cbSize = sizeof(m_data);
                           m_data.hWnd = pT->m_hWnd;
                           m_data.uID = IDR_MAINFRAME;
                           Shell_NotifyIcon(NIM_DELETE, &m_data);
                           CreateShellIcon();
                           return 0;
             }
//最小化到系統托盤
             LRESULT OnSize(UINT , WPARAM , LPARAM , BOOL& )
             {
                           T* pT = static_cast<T*>(this);
                           if (pT->IsIconic()) pT->ShowWindow(SW_HIDE);
                           return 0;
             }
};

//MyIconMenu.h
#pragma once
#include "stdafx.h"
#include <ContextMenu.h>
template <class T>
class CMyIconMenu : public CContextMenu<T>
{
public:
             BEGIN_MSG_MAP(CContextMenu)
                           COMMAND_ID_HANDLER(ID_RESUME, OnResume)
                           COMMAND_ID_HANDLER(ID_QUIT, OnQuit)
             END_MSG_MAP()
             LRESULT OnResume(WORD , WORD wID, HWND , BOOL& )
             {
                           T* pT = static_cast<T*>(this);
                           pT->ShowWindow(SW_SHOW);
                           OpenIcon(pT->m_hWnd);
                           return 0;
             }

             LRESULT OnQuit(WORD , WORD wID, HWND , BOOL& )
             {
                           PostQuitMessage(0);
                           return 0;
             }
};

 

　　CContextMenu的使用

　　若是你的窗口須要上下文菜單，只須要從CContextMenu派生一個類，例如：template <class T> class CMyContextMenu : public CContextMenu<T>

　　而後完成消息映射，例如CMyIconMenu所作的。在你的窗口類（假設爲CMyWindowClass）的繼承列表裏面添加public CMyContextMenu<CMyWindowClass>，在消息映射表中添加CHAIN_MSG_MAP(CMyContextMenu<CMyWindowClass>)

　　接下來只要在須要顯示上下文菜單的地方，調用CreatContextMenu(UINT ID_Menu)就能夠了。

　　CShellIcon的使用

　　首先要把stdafx.h中IE的版本改爲5或者6，例如：

#define _WIN32_IE           0x0600

　　而後加入CString支持，可能每一個項目都幾乎用到CString，你們不妨直接修改模板文件，讓嚮導直接把CString加進去算了。

　　從CContextMenu派生一個類來完成菜單映射，例如CMyIconMenu所做的。

　　如下用模式對話框來演示如何給其加入托盤功能，藍色的部分爲添加的部分。

#pragma once
#include "MyIconMenu.h"
#include <ShellIcon.h>
class CMainDlg : public CDialogImpl<CMainDlg>, public CShellIcon<CMainDlg, CMyIconMenu<CMainDlg>, IDR_ICONMENU>
{
public:
             enum { IDD = IDD_MAINDLG };
             typedef CShellIcon<CMainDlg, CMyIconMenu<CMainDlg>, IDR_ICONMENU> CMyShellIcon;
             BEGIN_MSG_MAP(CMainDlg)
                           MESSAGE_HANDLER(WM_INITDIALOG, OnInitDialog)
                           COMMAND_ID_HANDLER(ID_APP_ABOUT, OnAppAbout)
                           COMMAND_ID_HANDLER(IDOK, OnOK)
                           COMMAND_ID_HANDLER(IDCANCEL, OnCancel)
                           CHAIN_MSG_MAP(CMyShellIcon)
             END_MSG_MAP()
             LRESULT OnInitDialog(UINT , WPARAM , LPARAM , BOOL& )
             {
                           // center the dialog on the screen
                           CenterWindow();
                           // set icons
                           HICON hIcon = (HICON)::LoadImage(_Module.GetResourceInstance(), MAKEINTRESOURCE(IDR_MAINFRAME), 
                                         IMAGE_ICON, ::GetSystemMetrics(SM_CXICON), ::GetSystemMetrics(SM_CYICON), LR_DEFAULTCOLOR);
                           SetIcon(hIcon, TRUE);
                           HICON hIconSmall = (HICON)::LoadImage(_Module.GetResourceInstance(), MAKEINTRESOURCE(IDR_MAINFRAME), 
                                         IMAGE_ICON, ::GetSystemMetrics(SM_CXSMICON), ::GetSystemMetrics(SM_CYSMICON), LR_DEFAULTCOLOR);
                           SetIcon(hIconSmall, FALSE);
                           CreateShellIcon();
                           return TRUE;
             }
……………………
};

　　在想彈出氣泡的時候調用BOOL BalloonToolTips(LPCTSTR info)