首页 > 代码库 > 系统托盘图标
系统托盘图标
系统托盘图标
系统托盘图标的创建主要使用 Shell_NotifyIcon 函数来实现，实现过程比较简单。
注意：
要创建系统托盘图标，必须要有窗口；要创建窗口，必须要有窗口过程函数。
一个窗口可以有多个系统托盘图标；图标彼此间用 hWnd 和 uID 区分；
窗口过程函数接收到事件后， wParam 表示 uID，lParam 表示 message；
以下是封装的托盘类代码：
// notifyicon.h
#ifndef _NOTIFYICON_H_#define _NOTIFYICON_H_#include <windows.h>#include <shellapi.h>#include <map>typedef VOID (*NotifyIconCallback) (UINT uID, UINT message);class NotifyIcon{public: static UINT AddIcon(HICON hIcon, WCHAR* szTip); static VOID DelIcon(UINT uID); static VOID ModIcon(UINT uID, HICON hIcon, WCHAR* szTip); static UINT Attach(NotifyIconCallback pCallback); static VOID Detach(UINT uCookie);private: static LRESULT CALLBACK WndProc(HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam);private: static HWND m_hWnd; static UINT m_uCookieCount; static std::map<UINT,NotifyIconCallback> m_mapCallback; static UINT m_uIDCount; static std::map<UINT,NOTIFYICONDATA> m_mapNotifyData;};#endif
// notifyicon.cpp
#include "notifyicon.h"HWND NotifyIcon::m_hWnd = NULL;UINT NotifyIcon::m_uCookieCount = 0;std::map<UINT,NotifyIconCallback> NotifyIcon::m_mapCallback;UINT NotifyIcon::m_uIDCount = 0;std::map<UINT,NOTIFYICONDATA> NotifyIcon::m_mapNotifyData;UINT NotifyIcon::AddIcon( HICON hIcon, WCHAR* szTip ){ // 创建窗口 LPCTSTR pszClassName = TEXT("NotifyIcon_ClassName"); LPCTSTR pszWndName = TEXT("NotifyIcon_WndName"); HINSTANCE hInstance = GetModuleHandle(NULL); WNDCLASS wndClass; if (GetClassInfo(hInstance, pszClassName, &wndClass) == 0) { wndClass.style = CS_HREDRAW | CS_VREDRAW; wndClass.lpfnWndProc = NotifyIcon::WndProc; wndClass.cbClsExtra = 0; wndClass.cbWndExtra = 0; wndClass.hInstance = hInstance; wndClass.hIcon = NULL; wndClass.hCursor = NULL; wndClass.hbrBackground = NULL; wndClass.lpszClassName = pszClassName; wndClass.lpszMenuName = NULL; if (RegisterClass(&wndClass) == NULL) { return NULL; } } if (FindWindow(pszClassName, pszWndName) == NULL) { m_hWnd = CreateWindowEx(WS_EX_TOOLWINDOW, pszClassName, pszWndName, WS_POPUP, CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT, NULL, NULL, hInstance, NULL); if (m_hWnd == NULL) { return NULL; } } UINT uID = m_uIDCount; m_uIDCount++; NOTIFYICONDATA nd; nd.cbSize = sizeof(nd); nd.hWnd = m_hWnd; nd.uID = uID; nd.uFlags = NIF_ICON | NIF_MESSAGE | NIF_TIP; nd.uCallbackMessage = WM_USER; nd.hIcon = hIcon; if (szTip != NULL && lstrlen(szTip) < 127) { lstrcpy(nd.szTip, szTip); } if(Shell_NotifyIcon(NIM_ADD, &nd) == TRUE) { m_mapNotifyData.insert(std::make_pair(uID, nd)); return uID; } return NULL;}VOID NotifyIcon::DelIcon( UINT uID ){ if (m_hWnd == NULL) { return; } std::map<UINT, NOTIFYICONDATA>::iterator iter = m_mapNotifyData.find(uID); if (iter != m_mapNotifyData.end()) { Shell_NotifyIcon(NIM_DELETE, &(iter->second)); m_mapNotifyData.erase(iter); }}VOID NotifyIcon::ModIcon( UINT uID, HICON hIcon, WCHAR* szTip ){ if (m_hWnd == NULL) { return; } std::map<UINT, NOTIFYICONDATA>::iterator iter = m_mapNotifyData.find(uID); if (iter != m_mapNotifyData.end()) { if (hIcon != NULL) { iter->second.hIcon = hIcon; } if (szTip && lstrlen(szTip) < 127) { lstrcpy(iter->second.szTip, szTip); } else { lstrcpy(iter->second.szTip, L""); } Shell_NotifyIcon(NIM_MODIFY, &(iter->second)); }}UINT NotifyIcon::Attach( NotifyIconCallback pCallback ){ if (pCallback == NULL) { return NULL; } UINT uCookie = m_uCookieCount; m_uCookieCount ++; m_mapCallback.insert(std::make_pair(uCookie, pCallback)); return uCookie;}VOID NotifyIcon::Detach( UINT uCookie ){ std::map<UINT, NotifyIconCallback>::iterator iter = m_mapCallback.find(uCookie); if (iter != m_mapCallback.end()) { m_mapCallback.erase(iter); }}LRESULT CALLBACK NotifyIcon::WndProc( HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam ){ switch(message) { case WM_CREATE: break; case WM_DESTROY: break; case WM_USER: for (std::map<UINT, NotifyIconCallback>::iterator iter = m_mapCallback.begin(); iter != m_mapCallback.end(); ++iter) { NotifyIconCallback pCallback = iter->second; pCallback(wParam, lParam); // wParam --> uID, lParam --> message } break; default: break; } return DefWindowProc(hwnd, message, wParam, lParam);}
// main.cpp 测试
#include <windows.h>#include "notifyicon.h"void callbackFunc(UINT uID, UINT message);int main(){ HINSTANCE hInstance = GetModuleHandle(NULL); HICON hIcon = LoadIcon(hInstance, IDI_APPLICATION); WCHAR szTip[128] = L"哈哈哈"; UINT uID1 = NotifyIcon::AddIcon(hIcon, szTip); UINT uID2 = NotifyIcon::AddIcon(hIcon, szTip); UINT uID3 = NotifyIcon::AddIcon(hIcon, szTip); NotifyIcon::ModIcon(uID3, hIcon, NULL); NotifyIcon::DelIcon(uID2); UINT uCookie = NotifyIcon::Attach(callbackFunc); NotifyIcon::Detach(uCookie); MSG msg; while (GetMessage(&msg, NULL, 0, 0)) { TranslateMessage(&msg); DispatchMessage(&msg); } return 0;}void callbackFunc(UINT uID, UINT message){ if (message == WM_LBUTTONDOWN || message == WM_LBUTTONUP || message == WM_LBUTTONDBLCLK || message == WM_RBUTTONDOWN || message == WM_RBUTTONUP || message == WM_RBUTTONDBLCLK) { WCHAR szDbg[256]; ZeroMemory(szDbg,256); wsprintf(szDbg, L"[Dongyu] uID = %d, message = %x", uID, message); OutputDebugString(szDbg); }}
参考文章：
http://www.nowamagic.net/librarys/veda/detail/227
声明：以上内容来自用户投稿及互联网公开渠道收集整理发布，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任，若内容有误或涉及侵权可进行投诉： 投诉/举报 工作人员会在5个工作日内联系你，一经查实，本站将立刻删除涉嫌侵权内容。