系统托盘

最新推荐文章于 2025-08-11 20:47:16 发布

wanglc08

最新推荐文章于 2025-08-11 20:47:16 发布

阅读量446

点赞数 2

CC 4.0 BY-SA版权

分类专栏： c++ 文章标签： function destructor microsoft shell constructor insert

本文链接：https://blog.youkuaiyun.com/wanglc08/article/details/4862833

c++ 专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一个用于创建和管理Windows系统托盘图标的C++类CSystemTray。该类提供了一系列方法，如创建托盘图标、修改图标、设置通知窗口及处理托盘通知等。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#if !defined(AFX_SYSTEMTRAY_H__48272E0E_9FD1_4CFF_996B_CE36ED449FFF__INCLUDED_)
#define AFX_SYSTEMTRAY_H__48272E0E_9FD1_4CFF_996B_CE36ED449FFF__INCLUDED_

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000

#define WM_ICON_NOTIFY WM_USER+111

class CSystemTray : public CWnd
{
// Construction/destruction
public:
CSystemTray();
virtual ~CSystemTray();

DECLARE_DYNAMIC(CSystemTray)

// Operations
public:
BOOL Create(UINT uCallbackMessage, LPCTSTR lpszTip, HICON hIcon, UINT uID);

BOOL SetIcon(UINT nIDResource);

BOOL SetNotificationWnd(CWnd* pNotifyWnd);
virtual LRESULT OnTrayNotification(WPARAM uID, LPARAM lEvent);

// Overrides
// ClassWizard generated virtual function overrides
//{{AFX_VIRTUAL(CSystemTray)
protected:
virtual LRESULT WindowProc(UINT message, WPARAM wParam, LPARAM lParam);
//}}AFX_VIRTUAL

// Implementation
protected:
NOTIFYICONDATA m_NotifyIconData;

// Generated message map functions
protected:
//{{AFX_MSG(CSystemTray)
//}}AFX_MSG

DECLARE_MESSAGE_MAP()
};

/////////////////////////////////////////////////////////////////////////////

//{{AFX_INSERT_LOCATION}}
// Microsoft Visual C++ will insert additional declarations immediately before the previous line.

#endif // !defined(AFX_SYSTEMTRAY_H__48272E0E_9FD1_4CFF_996B_CE36ED449FFF__INCLUDED_)

#include "stdafx.h"
#include "SystemTray.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

IMPLEMENT_DYNAMIC(CSystemTray, CWnd)

/********************************************************************/
/*                 */
/* Function name : CSystemTray::CSystemTray       */
/* Description   : Constructor          */
/*                 */
/********************************************************************/
CSystemTray::CSystemTray()
{
    ZeroMemory(&m_NotifyIconData, sizeof(m_NotifyIconData));
}

/********************************************************************/
/*                 */
/* Function name : CSystemTray::~CSystemTray      */
/* Description   : Destructor          */
/*                 */
/********************************************************************/
CSystemTray::~CSystemTray()
{
// remove icon from system tray
    m_NotifyIconData.uFlags = 0;
    Shell_NotifyIcon(NIM_DELETE, &m_NotifyIconData);

// destroy dummy window
DestroyWindow();
}

BEGIN_MESSAGE_MAP(CSystemTray, CWnd)
//{{AFX_MSG_MAP(CSystemTray)
//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/********************************************************************/
/*                 */
/* Function name : CSystemTray::Create        */
/* Description   : Create systemtray icon       */
/*                 */
/********************************************************************/
BOOL CSystemTray::Create(UINT uCallbackMessage, LPCTSTR lpszToolTip, HICON hIcon, UINT uID)
{
// create dummy window
    if (!CWnd::CreateEx(0, AfxRegisterWndClass(0), "CSystemTray Dummy Window", WS_POPUP, 0,0,0,0, NULL, 0))
  return FALSE;

    // setup NOTIFYICONDATA
    m_NotifyIconData.cbSize = sizeof(NOTIFYICONDATA);
    m_NotifyIconData.hWnd   = m_hWnd;
    m_NotifyIconData.uID    = uID;
    m_NotifyIconData.hIcon = hIcon;
    m_NotifyIconData.uFlags = NIF_MESSAGE | NIF_ICON | NIF_TIP;
    m_NotifyIconData.uCallbackMessage = uCallbackMessage;
    lstrcpy(m_NotifyIconData.szTip, lpszToolTip);

// add tray icon
return Shell_NotifyIcon(NIM_ADD, &m_NotifyIconData);
}

/********************************************************************/
/*                 */
/* Function name : CSystemTray::SetIcon        */
/* Description   : Modify icon in the system tray     */
/*                 */
/********************************************************************/
BOOL CSystemTray::SetIcon(UINT nIDResource)
{
    HICON hIcon = AfxGetApp()->LoadIcon(nIDResource);
    m_NotifyIconData.uFlags = NIF_ICON;
    m_NotifyIconData.hIcon = hIcon;
    return Shell_NotifyIcon(NIM_MODIFY, &m_NotifyIconData);
}

/********************************************************************/
/*                 */
/* Function name : CSystemTray::SetNotificationWnd     */
/* Description   : Modify notification window      */
/*                 */
/********************************************************************/
BOOL CSystemTray::SetNotificationWnd(CWnd* pWnd)
{
    m_NotifyIconData.hWnd = pWnd->GetSafeHwnd();
    m_NotifyIconData.uFlags = 0;
    return Shell_NotifyIcon(NIM_MODIFY, &m_NotifyIconData);
}

/********************************************************************/
/*                 */
/* Function name : CSystemTray::OnTrayNotification     */
/* Description   : Handle notification messages      */
/*                 */
/********************************************************************/
LRESULT CSystemTray::OnTrayNotification(UINT wParam, LONG lParam)
{
    // is it for us ?
    if (wParam != m_NotifyIconData.uID)
        return 0L;

CWnd *pMainWnd = AfxGetMainWnd();

    switch(LOWORD(lParam))
{
  case WM_RBUTTONUP:
  {
   // right button click brings up a context menu
   CMenu menu, *pSubMenu;
   if (!menu.LoadMenu(m_NotifyIconData.uID))
    return 0;

   pSubMenu = menu.GetSubMenu(0);

if (pSubMenu == NULL)
return 0;

CPoint pos;
GetCursorPos(&pos);

// display popup menu
pSubMenu->TrackPopupMenu(0, pos.x, pos.y, pMainWnd, NULL);

   menu.DestroyMenu();
   break;
  }

  case WM_LBUTTONDBLCLK:
  {
   // double click brings up the main dialog
   pMainWnd->ShowWindow(SW_SHOW);
   pMainWnd->SetForegroundWindow();
   break;
  }
  default:
   break;
}
    return 1;
}

/********************************************************************/
/*                 */
/* Function name : CSystemTray::WindowProc       */
/* Description   : Dispatch messages through the window抯 message */
/*       map            */
/*                 */
/********************************************************************/
LRESULT CSystemTray::WindowProc(UINT uMsg, WPARAM wParam, LPARAM lParam)
{
    if (uMsg == m_NotifyIconData.uCallbackMessage)
{
        return OnTrayNotification(wParam, lParam);
}
return CWnd::WindowProc(uMsg, wParam, lParam);
}