利用Visual C++實(shí)現(xiàn)系統(tǒng)托盤程序

自從微軟公司推出Windows 95操作系統(tǒng)以來，系統(tǒng)托盤應(yīng)用作為一種極具吸引力的用戶界面設(shè)計(jì)深受廣大用戶的喜愛。使用系統(tǒng)托盤作為用戶界面的Windows應(yīng)用程序數(shù)不勝數(shù)，比如"金山詞霸"、"Winamp"、"RealPlayer"等等。

這些程序運(yùn)行時(shí)不顯示運(yùn)行窗口，只在任務(wù)欄上顯示一個(gè)圖標(biāo)，表示程序正在運(yùn)行，用戶可以通過鼠標(biāo)與應(yīng)用程序交互，程序開發(fā)人員有時(shí)也需要編制一些僅在后臺(tái)運(yùn)行的類似程序，為了不干擾前臺(tái)程序的運(yùn)行界面和不顯示不必要的窗口，應(yīng)使程序運(yùn)行時(shí)的主窗口不可見。同時(shí)將一個(gè)圖標(biāo)顯示在任務(wù)欄右端靜態(tài)通告區(qū)中并響應(yīng)用戶的鼠標(biāo)動(dòng)作。本實(shí)例就介紹Visual C++開發(fā)這類程序的設(shè)計(jì)方法，該程序編譯運(yùn)行后，如果雙擊托盤圖標(biāo)，程序會(huì)彈出一個(gè)消息列表窗口，只要鼠標(biāo)在托盤圖標(biāo)上移動(dòng)或點(diǎn)擊（無論是左右鍵的單擊或雙擊），產(chǎn)生的消息都會(huì)顯示在這個(gè)窗口里；當(dāng)鼠標(biāo)光標(biāo)移到托盤圖標(biāo)上時(shí)，在圖標(biāo)附近會(huì)顯示提示信息；單擊右鍵時(shí)彈出上下文菜單，這個(gè)菜單中應(yīng)包含打開屬性頁的命令或者打開與圖標(biāo)相關(guān)的其它窗口的命令，另外，該程序還可以動(dòng)態(tài)的改變托盤的圖標(biāo)。參照這個(gè)例子，相信讀者能輕松自如地在自己的程序中應(yīng)用系統(tǒng)托盤。

一、實(shí)現(xiàn)方法

為了實(shí)現(xiàn)拖盤程序，首先要使程序的主窗口不可見，這點(diǎn)實(shí)現(xiàn)起來十分容易，只要調(diào)用ShowWindow(SW_HIDE)就可以了，本實(shí)例采用的就是這種方法，還有一種思路是通過分別設(shè)置主邊框窗口的風(fēng)格和擴(kuò)展風(fēng)格來隱藏主框架：

BOOL CMainFrame::PreCreateWindow(CREATESTRUCT& cs)
{
cs.style =WS_POPUP;//使主窗口不可見；
cs.dwExStyle =WS_EX_TOOLWINDOW;//不顯示任務(wù)按鈕；
return CFrameWnd::PreCreateWindow(cs);
}

在任務(wù)條上顯示圖標(biāo)是利用系統(tǒng)API函數(shù)Shell_NotifyIcon（）來將一個(gè)圖標(biāo)顯示在任務(wù)欄的通告區(qū)中。該函數(shù)的原型為：

BOOL Shell_NotifyIcon(DWORD dwMessage, PNOTIFYICONDATA pnid);

該函數(shù)的第一個(gè)參數(shù)dwMessage類型為DWORD，表示要進(jìn)行的動(dòng)作，它可以是下面的值之一：

NIM_ADD： 添加一個(gè)圖標(biāo)到任務(wù)欄。

NIM_MODIFY： 修改狀態(tài)欄區(qū)域的圖標(biāo)。

NIM_DELETE： 刪除狀態(tài)欄區(qū)域的圖標(biāo)。

NIM_SETFOCUS： 將焦點(diǎn)返回到任務(wù)欄通知區(qū)域。當(dāng)完成用戶界面操作時(shí)，任務(wù)欄圖標(biāo)必須用此消息。例如，如果任務(wù)欄圖標(biāo)正顯示上下文菜單，但用戶按下"ESCAPE"鍵取消操作，這時(shí)就必須用此消息將焦點(diǎn)返回到任務(wù)欄通知區(qū)域。

NIM_SETVERSION：指示任務(wù)欄按照相應(yīng)的動(dòng)態(tài)庫版本工作。

第二個(gè)參數(shù)pnid是NOTIFYICONDATA結(jié)構(gòu)的地址，其內(nèi)容視dwMessage的值而定。這個(gè)結(jié)構(gòu)在SHELLAPI.H文件中定義如下：

typedef struct _NOTIFYICONDATA {
DWORD cbSize; // 結(jié)構(gòu)大小（sizeof struct），必須設(shè)置
HWND hWnd; // 發(fā)送通知消息的窗口句柄
UINT uID; // 圖標(biāo)ID ( 由回調(diào)函數(shù)的WPARAM 指定)
UINT uFlags;
UINT uCallbackMessage; // 消息被發(fā)送到此窗口過程
HICON hIcon; // 任務(wù)欄圖標(biāo)句柄
CHAR szTip[64]; // 提示文本
} NOTIFYICONDATA;

該結(jié)構(gòu)中uFlags的值分別為：

#define NIF_MESSAGE 0x1 // 表示uCallbackMessage 有效
#define NIF_ICON 0x2 // 表示hIcon 有效
#define NIF_TIP 0x4 // 表示szTip 有效

在該結(jié)構(gòu)的成員中，cbSize為該結(jié)構(gòu)所占的字節(jié)數(shù)，hWnd為接受該圖標(biāo)所發(fā)出的消息的窗口的句柄（鼠標(biāo)在任務(wù)欄上程序圖標(biāo)上動(dòng)作時(shí)圖標(biāo)將發(fā)出消息，這個(gè)消息用戶要自己定義），uID為被顯示圖標(biāo)的ID，uFlags指明其余的幾個(gè)成員（hIcon、uCallBackMessage和szTip）的值是否有效，uCallbackMessage為一個(gè)用戶自定義的消息，當(dāng)用戶在該圖標(biāo)上作用一些鼠標(biāo)動(dòng)作時(shí)，圖標(biāo)將向應(yīng)用程序的主框架窗口（hWnd成員中指定的窗口）發(fā)出該消息，為了使程序的主框架得到該通知消息，需要設(shè)置NOTIFYICONDATA 結(jié)構(gòu)的flag成員的值為NIF_MESSAGE。hIcon為將在任務(wù)欄上顯示的圖標(biāo)句柄，szTip鼠標(biāo)停留在該圖標(biāo)上時(shí)顯示的提示字符串。

盡管Shell_NotifyIcon函數(shù)簡單實(shí)用，但它畢竟是個(gè)Win32 API，為此本實(shí)例將它封裝在了一個(gè)C++類中，這個(gè)類叫做CTrayIcon，有了它，托盤編程會(huì)更加輕松自如，因?yàn)樗[藏了NOTIFYICONDATA、消息代碼、標(biāo)志以及一些繁瑣的細(xì)節(jié)。

二、編程步驟

1、 啟動(dòng)Visual C++6.0，生成一個(gè)單文檔的應(yīng)用程序TrayTest，取消文檔視圖支持；

2、 在CMainFrame類中添加自定義消息#define WM_MY_TRAY_NOTIFICATION WM_USER+0，并在該類中為此自定義消息手動(dòng)添加消息映射ON_MESSAGE(WM_MY_TRAY_NOTIFICATION, OnTrayNotification)和消息響應(yīng)函數(shù)afx_msg LRESULT OnTrayNotification(WPARAM wp, LPARAM lp)；

3、 設(shè)計(jì)二個(gè)圖標(biāo)添加到項(xiàng)目中，其ID標(biāo)志分別為"IDI_MYICON"、"IDI_MYICON2"，作為托盤顯示時(shí)的圖標(biāo)；

4、 在CMainFrame類中添加下述變量： CTrayIcon m_trayIcon（用來操作圖標(biāo)的類對(duì)象）、CEdit m_wndEdit（編輯框用來顯示所跟蹤到的鼠標(biāo)消息）、int m_iWhichIcon（決定當(dāng)前托盤使用哪個(gè)圖標(biāo)）、BOOL m_bShutdown（是否關(guān)閉當(dāng)前拖盤程序標(biāo)志）、BOOL m_bShowTrayNotifications（是否顯示托盤消息標(biāo)志）；

5、 為程序的IDR_MAINFRAME添加處理菜單項(xiàng)和托盤的上下文菜單IDI_TRAYICON（具體的菜單項(xiàng)的標(biāo)題和ID標(biāo)志符參見代碼部分），然后使用Class Wizard為各個(gè)菜單項(xiàng)添加處理函數(shù)；

6、 添加代碼，編譯運(yùn)行程序。

///////////////////////////////////////////////CTrayIcon類的頭文件；
#ifndef _TRAYICON_H
#define _TRAYICON_H
class CTrayIcon : public CCmdTarget {
　protected:
DECLARE_DYNAMIC(CTrayIcon)
NOTIFYICONDATA m_nid; // struct for Shell_NotifyIcon args
　public:
CTrayIcon(UINT uID);
~CTrayIcon();
// Call this to receive tray notifications
void SetNotificationWnd(CWnd* pNotifyWnd, UINT uCbMsg);
BOOL SetIcon(UINT uID); // main variant you want to use
BOOL SetIcon(HICON hicon, LPCSTR lpTip);
BOOL SetIcon(LPCTSTR lpResName, LPCSTR lpTip)
{ return SetIcon(lpResName ?
AfxGetApp()->LoadIcon(lpResName) : NULL, lpTip); }
BOOL SetStandardIcon(LPCTSTR lpszIconName, LPCSTR lpTip)
{ return SetIcon(::LoadIcon(NULL, lpszIconName), lpTip); }
virtual LRESULT OnTrayNotification(WPARAM uID, LPARAM lEvent);
};
#endif
///////////////////////////////////////////////////CTrayIcon類的.CPP文件
#include "stdafx.h"
#include "trayicon.h"
#include // for AfxLoadString
IMPLEMENT_DYNAMIC(CTrayIcon, CCmdTarget)
CTrayIcon::CTrayIcon(UINT uID)
{
　memset(&m_nid, 0 , sizeof(m_nid)); // Initialize NOTIFYICONDATA
　m_nid.cbSize = sizeof(m_nid);
　m_nid.uID = uID; // never changes after construction
　AfxLoadString(uID, m_nid.szTip, sizeof(m_nid.szTip));
　// Use resource string as tip if there is one
}

CTrayIcon::~CTrayIcon()
{
　SetIcon(0); // remove icon from system tray
}

void CTrayIcon::SetNotificationWnd(CWnd* pNotifyWnd, UINT uCbMsg)
{
　// Set notification window. It must created already.
　ASSERT(pNotifyWnd==NULL::IsWindow(pNotifyWnd->GetSafeHwnd()));
　m_nid.hWnd = pNotifyWnd->GetSafeHwnd();
　ASSERT(uCbMsg==0uCbMsg>=WM_USER);
　m_nid.uCallbackMessage = uCbMsg;
}

BOOL CTrayIcon::SetIcon(UINT uID)
{
　// Sets both the icon and tooltip from resource ID ，To remove the icon, call SetIcon(0)
　HICON hicon=NULL;
　if (uID) {
AfxLoadString(uID, m_nid.szTip, sizeof(m_nid.szTip));
hicon = AfxGetApp()->LoadIcon(uID);
　}
　return SetIcon(hicon, NULL);
}

BOOL CTrayIcon::SetIcon(HICON hicon, LPCSTR lpTip)
{
　// Common SetIcon for all overloads.
　UINT msg;
　m_nid.uFlags = 0;
　if (hicon) {
// Set the icon
msg = m_nid.hIcon ? NIM_MODIFY : NIM_ADD;
m_nid.hIcon = hicon; // Add or replace icon in system tray
m_nid.uFlags= NIF_ICON;
　} else {
if (m_nid.hIcon==NULL) // remove icon from tray
return TRUE; // already deleted
msg = NIM_DELETE;
　}
　if (lpTip) // Use the tip, if any
strncpy(m_nid.szTip, lpTip, sizeof(m_nid.szTip));
　if (m_nid.szTip[0])
m_nid.uFlags= NIF_TIP;
　if (m_nid.uCallbackMessage && m_nid.hWnd) // Use callback if any
m_nid.uFlags= NIF_MESSAGE;
BOOL bRet = Shell_NotifyIcon(msg, &m_nid); // Do it
　if (msg==NIM_DELETE!bRet)
m_nid.hIcon = NULL; // failed
　return bRet;
}
LRESULT CTrayIcon::OnTrayNotification(WPARAM wID, LPARAM lEvent)
{
　if (wID!=m_nid.uID(lEvent!=WM_RBUTTONUP && lEvent!=WM_LBUTTONDBLCLK))
return 0;
　CMenu menu;//裝載上下文菜單；
　if (!menu.LoadMenu(m_nid.uID))
return 0;
　CMenu* pSubMenu = menu.GetSubMenu(0);
　if (!pSubMenu)
return 0;
　if (lEvent==WM_RBUTTONUP) {//設(shè)置第一個(gè)菜單項(xiàng)為默認(rèn)菜單項(xiàng)目
::SetMenuDefaultItem(pSubMenu->m_hMenu, 0, TRUE);
//將當(dāng)前菜單作為上下文菜單；
CPoint mouse;
GetCursorPos(&mouse);
::SetForegroundWindow(m_nid.hWnd);
::TrackPopupMenu(pSubMenu->m_hMenu, 0, mouse.x, mouse.y, 0,m_nid.hWnd, NULL);
　} else // double click: execute first menu item
::SendMessage(m_nid.hWnd, WM_COMMAND, pSubMenu->GetMenuItemID(0), 0);
　return 1;
}

///////////////////////////////// MainFrm.h : interface of the CMainFrame class
#if !defined(AFX_MAINFRM_H__9ED70A69_C975_4F20_9D4E_B2877E3575D0__INCLUDED_)
#define AFX_MAINFRM_H__9ED70A69_C975_4F20_9D4E_B2877E3575D0__INCLUDED_
#if _MSC_VER >1000
#pragma once
#endif // _MSC_VER >1000

#include "trayicon.h"
class CMainFrame : public CFrameWnd
{
　public:
CMainFrame();
　protected:
DECLARE_DYNAMIC(CMainFrame)
// Attributes
　public:
// Overrides
// ClassWizard generated virtual function overrides
//{{AFX_VIRTUAL(CMainFrame)
//}}AFX_VIRTUAL
// Implementation
　public:
virtual ~CMainFrame();
#ifdef _DEBUG
virtual void AssertValid() const;
virtual void Dump(CDumpContext& dc) const;
#endif
　protected: // control bar embedded members
CStatusBar m_wndStatusBar;
CTrayIcon m_trayIcon; // my tray icon
CEdit m_wndEdit; // to display tray notifications
int m_iWhichIcon; // which HICON to use
BOOL m_bShutdown; // OK to terminate TRAYTEST
BOOL m_bShowTrayNotifications; // display info in main window
// Generated message map functions
　protected:
//{{AFX_MSG(CMainFrame)
afx_msg LRESULT OnTrayNotification(WPARAM wp, LPARAM lp);
afx_msg int OnCreate(LPCREATESTRUCT lpCreateStruct);
afx_msg void OnToggleIcon();
afx_msg void OnViewClear();
afx_msg void OnViewNotifications();
afx_msg void OnUpdateViewClear(CCmdUI* pCmdUI);
afx_msg void OnUpdateViewNotifications(CCmdUI* pCmdUI);
afx_msg void OnClose();
afx_msg void OnAppOpen();
afx_msg void OnAppSuspend();
// NOTE - the ClassWizard will add and remove member functions here.
// DO NOT EDIT what you see in these blocks of generated code!
//}}AFX_MSG
　DECLARE_MESSAGE_MAP()
};

///////////////////////////////////////////////CMainFrm.cpp
#include "stdafx.h"
#include "TrayTest.h"
#include "MainFrm.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
LRESULT CTrayIcon::OnTrayNotification(WPARAM wID, LPARAM lEvent)
{
　if (wID!=m_nid.uID(lEvent!=WM_RBUTTONUP && lEvent!=WM_LBUTTONDBLCLK))
return 0;
　CMenu menu;//裝載上下文菜單；
　if (!menu.LoadMenu(m_nid.uID))
return 0;
　CMenu* pSubMenu = menu.GetSubMenu(0);
　if (!pSubMenu)
return 0;
　if (lEvent==WM_RBUTTONUP) {//設(shè)置第一個(gè)菜單項(xiàng)為默認(rèn)菜單項(xiàng)目
::SetMenuDefaultItem(pSubMenu->m_hMenu, 0, TRUE);
//將當(dāng)前菜單作為上下文菜單；
CPoint mouse;
GetCursorPos(&mouse);
::SetForegroundWindow(m_nid.hWnd);
::TrackPopupMenu(pSubMenu->m_hMenu, 0, mouse.x, mouse.y, 0,m_nid.hWnd, NULL);
　} else // double click: execute first menu item
::SendMessage(m_nid.hWnd, WM_COMMAND, pSubMenu->GetMenuItemID(0), 0);
　return 1;
}

///////////////////////////////// MainFrm.h : interface of the CMainFrame class
#if !defined(AFX_MAINFRM_H__9ED70A69_C975_4F20_9D4E_B2877E3575D0__INCLUDED_)
#define AFX_MAINFRM_H__9ED70A69_C975_4F20_9D4E_B2877E3575D0__INCLUDED_
#if _MSC_VER >1000
#pragma once
#endif // _MSC_VER >1000

#include "trayicon.h"
class CMainFrame : public CFrameWnd
{
　public:
CMainFrame();
　protected:
DECLARE_DYNAMIC(CMainFrame)
// Attributes
　public:
// Overrides
// ClassWizard generated virtual function overrides
//{{AFX_VIRTUAL(CMainFrame)
//}}AFX_VIRTUAL
// Implementation
　public:
virtual ~CMainFrame();
#ifdef _DEBUG
virtual void AssertValid() const;
virtual void Dump(CDumpContext& dc) const;
#endif
　protected: // control bar embedded members
CStatusBar m_wndStatusBar;
CTrayIcon m_trayIcon; // my tray icon
CEdit m_wndEdit; // to display tray notifications
int m_iWhichIcon; // which HICON to use
BOOL m_bShutdown; // OK to terminate TRAYTEST
BOOL m_bShowTrayNotifications; // display info in main window
// Generated message map functions
　protected:
//{{AFX_MSG(CMainFrame)
afx_msg LRESULT OnTrayNotification(WPARAM wp, LPARAM lp);
afx_msg int OnCreate(LPCREATESTRUCT lpCreateStruct);
afx_msg void OnToggleIcon();
afx_msg void OnViewClear();
afx_msg void OnViewNotifications();
afx_msg void OnUpdateViewClear(CCmdUI* pCmdUI);
afx_msg void OnUpdateViewNotifications(CCmdUI* pCmdUI);
afx_msg void OnClose();
afx_msg void OnAppOpen();
afx_msg void OnAppSuspend();
// NOTE - the ClassWizard will add and remove member functions here.
// DO NOT EDIT what you see in these blocks of generated code!
//}}AFX_MSG
　DECLARE_MESSAGE_MAP()
};

///////////////////////////////////////////////CMainFrm.cpp
#include "stdafx.h"
#include "TrayTest.h"
#include "MainFrm.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

歡迎轉(zhuǎn)載,信息來自維庫電子市場網(wǎng)（www.dzsc.com）