对程序添加系统托盘图标支持-一个简单的C++封装

实现的关键是调用Windows Shell 的API Shell_NotifyIcon，该函数有两个参数，第一个参数指名操作类型，第二个参数是一个NOTIFYICONDATA结构，保存了操作相关的信息，比如图标，工具提示文本等等。关于该函数的详细情况可以在MSDN上查阅得知，这里不多叙述。

下面贴出类的实现代码，后面会有使用方法的介绍（这个Blog居然没有C++的Language Filter,太不爽了，暂时用Java的），

头文件

#ifnde f __CSYSTEMTRAY_H
#define __CSYSTEMTRAY_H

#include  " shellapi.h "

#define WM_TRAY_MSG WM_USER  +   0x01                    // 自定义消息，用以接收来自托盘的反馈
class  CSystemTray
... {
public:
    CSystemTray(void);
public:
    ~CSystemTray(void);
    void Destroy();
    BOOL Create(HWND hParent, HICON hIcon, LPCTSTR szToolTip);  //hParent 是接收WM_TRAY_MSG消息的窗口
    BOOL ModifyIcon(HICON hIcon);
    BOOL ModifyToolTipText(LPCTSTR szToolTip);
    BOOL ModifyBalloonText(LPCTSTR szTitle, LPCTSTR szInfo);
    BOOL ModifyBalloonIcon(HICON hIcon);
    BOOL SetFocus();
    //BOOL AnimateIcon(HICON *pIconArray, UINT nSize, DWORD dwTimeInter);
private:
    void Wait(int nSeconds);
    HWND m_hParent;
    HICON m_hIconTray, m_hIconBalloon;
    NOTIFYICONDATA m_nidSystemTray;
} ;

#endif

 

 实现：

 

#include  " SystemTray.h "

CSystemTray::CSystemTray( void ):m_hParent(NULL), m_hIconTray(NULL), m_hIconBalloon(NULL)
... {
    ::ZeroMemory(&m_nidSystemTray, sizeof(NOTIFYICONDATA));
}

CSystemTray:: ~ CSystemTray( void )
... {
    Destroy();
}

BOOL CSystemTray::Create(HWND hParent, HICON hIcon, LPCTSTR szToolTip)
... {
    m_hParent = hParent;
    

    
    m_nidSystemTray.cbSize = sizeof(NOTIFYICONDATA);
    m_nidSystemTray.hWnd = m_hParent;
    m_nidSystemTray.hIcon = m_hIconTray = hIcon;
    
    m_nidSystemTray.uFlags = NIF_ICON | NIF_MESSAGE;
    m_nidSystemTray.uCallbackMessage = WM_TRAY_MSG;
    if (szToolTip)
    ...{
        m_nidSystemTray.uFlags |= NIF_TIP;
        _tcscpy(m_nidSystemTray.szTip, szToolTip);
    }
    return Shell_NotifyIcon(NIM_ADD, &m_nidSystemTray);
}

BOOL CSystemTray::ModifyIcon(HICON hIcon)
... {
    ASSERT(m_nidSystemTray.hWnd != NULL);
    m_nidSystemTray.hIcon = hIcon;
    m_nidSystemTray.uFlags = NIF_ICON;
    return Shell_NotifyIcon(NIM_MODIFY, &m_nidSystemTray);
}

BOOL CSystemTray::ModifyToolTipText(LPCTSTR szToolTip)
... {
    ASSERT(m_nidSystemTray.hWnd != NULL);
    _tcscpy(m_nidSystemTray.szTip, szToolTip);
    m_nidSystemTray.uFlags = NIF_TIP;
    return Shell_NotifyIcon(NIM_MODIFY, &m_nidSystemTray);
}

BOOL CSystemTray::ModifyBalloonText(LPCTSTR szTitle, LPCTSTR szInfo)
... {
    ASSERT(m_nidSystemTray.hWnd != NULL);
    _tcscpy(m_nidSystemTray.szInfoTitle, szTitle);
    _tcscpy(m_nidSystemTray.szInfo, szInfo);
    m_nidSystemTray.hIcon = m_hIconBalloon;
    m_nidSystemTray.uFlags = NIF_INFO | NIF_ICON;
    BOOL bRet = Shell_NotifyIcon(NIM_MODIFY, &m_nidSystemTray);

    if (bRet)
    ...{
        m_nidSystemTray.hIcon = m_hIconTray;
        m_nidSystemTray.uFlags = NIF_ICON;
     return Shell_NotifyIcon(NIM_MODIFY, &m_nidSystemTray);
    }
    else return FALSE;
}

BOOL CSystemTray::ModifyBalloonIcon(HICON hIcon)
... {
    m_nidSystemTray.dwInfoFlags = NIIF_USER;
    
    m_nidSystemTray.hIcon = m_hIconBalloon = hIcon;
    
    m_nidSystemTray.uFlags = NIF_INFO | NIF_ICON;
    BOOL bRet = FALSE;
    bRet = Shell_NotifyIcon(NIM_MODIFY, &m_nidSystemTray);
    if (bRet)
    ...{
        m_nidSystemTray.hIcon =m_hIconTray;
    
        m_nidSystemTray.uFlags = NIF_ICON;
    
        return Shell_NotifyIcon(NIM_MODIFY, &m_nidSystemTray);
    }
    else return FALSE;
}
BOOL CSystemTray::SetFocus()
... {
    ASSERT(m_nidSystemTray.hWnd != NULL);
    return Shell_NotifyIcon(NIM_SETFOCUS, &m_nidSystemTray);
}
void  CSystemTray::Destroy()
... {
    if(m_nidSystemTray.hWnd != NULL)
    ...{
        Shell_NotifyIcon(NIM_DELETE, &m_nidSystemTray);    
        ::ZeroMemory(&m_nidSystemTray, sizeof(NOTIFYICONDATA));
    }
    m_hParent = NULL;
}

 

首先创建一个对话框程序，将上述文件添加至工程。

在xxxDlg.h（xxx是你的工程名）文件中，添加#include "systemtray.h"，在类声明中定一个CSystemTray类型变量，比如m_stMyProg;

然后再对话框的实现文件OnInitDialog函数中添加以下代码

m_stMyProg.Create(m_hWnd,::LoadIcon(theApp.m_hInstance,MAKEINTRESOURCE(IDI_ACCEPT)), _T("This is a SystemTray example"));

其中IDI_ACCEPT是一个事先定义好的图标资源。

这样启动程序以后，我们就可以在托盘上看到相应的图标了

接下来在程序的任何位置调用ModifyIcon可以修改托盘的图标，调用ModifyToolTip可以修改托盘的工具提示(tooltip)，调用ModifyBalloonIcon可以修改气球提示的图标，相应的，如果想调用系统默认图标可以调用函数::LoadStandardIcon。需要注意的是，气球提示只有在每次修改文本时才会出现，因此如果想立即看到修改图标的效果，请调用一次ModifyBalloonText。OK！

前面的头文件中定义了一个WM_TRAY_MSG消息，当鼠标经过托盘图标时该消息就会得到响应，其中消息相关的lParam参数表示鼠标的操作，比如在托盘处单击了一下鼠标左键，lParam就是一个WM_LBUTTONDOWN消息，可以根据此消息来创建上下文菜单，等等。

首先要在对话框的头文件中添加一个消息相应函数的声明

afx_msg LRESULT OnTrayMsg(WPARAM  wParam, LPARAM  lParam);

再在MFC的消息映射中添加 ON_MESSAGE(WM_TRAY_MSG, OnTrayMsg)

最后在对话框的实现文件中添加对该函数的实现

 

LRESULT CxxxxDlg::OnTrayMsg(WPARAM wParam, LPARAM lParam)
... {
    switch (lParam)
        ...{
          case WM_LBUTTONDOWN:
              //do someting...
               break;
          case WM_RBUTTONDOWN:
              //then do another...
           ...


          default:
            break;
        }
    return 0;
}

 

根据接收到的不同鼠标消息对其进行响应。

这个类写的还不够完整，以后会添加关于托盘鼠标动画的支持，等等。目前先到这里

2007.10.9     by  赵科进(GodBird)