系统托盘

 #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);
}