收藏 托盘

头文件：

/
// TrayIcon.h : header file
//

#ifndef _INCLUDED_TRAYICON_H_
#define _INCLUDED_TRAYICON_H_

/
// CTrayIcon window

class CTrayIcon : public CObject
{
// Construction/destruction
public:
 CTrayIcon();
 CTrayIcon(CWnd* pWnd, UINT uCallbackMessage, LPCTSTR szTip, HICON icon, UINT uID);
 virtual ~CTrayIcon();

// Operations
public:
 BOOL Enabled() { return m_bEnabled; }
 BOOL Visible() { return !m_bHidden; }

 //Create the tray icon
 int Create(CWnd* pWnd, UINT uCallbackMessage, LPCTSTR szTip, HICON icon, UINT uID);

 //Change or retrieve the Tooltip text
 BOOL    SetTooltipText(LPCTSTR pszTooltipText);
 BOOL    SetTooltipText(UINT nID);
 CString GetTooltipText() const;

 //Change or retrieve the icon displayed
 BOOL  SetIcon(HICON hIcon);
 BOOL  SetIcon(LPCTSTR lpIconName);
 BOOL  SetIcon(UINT nIDResource);
 BOOL  SetStandardIcon(LPCTSTR lpIconName);
 BOOL  SetStandardIcon(UINT nIDResource);
 HICON GetIcon() const;
 void  HideIcon();
 void  ShowIcon();
 void  RemoveIcon();
 void  MoveToRight();

 //Change or retrieve the window to send notification messages to
 BOOL  SetNotificationWnd(CWnd* pNotifyWnd);
 CWnd* GetNotificationWnd() const;

 //Default handler for tray notification message
 virtual LRESULT OnTrayNotification(WPARAM uID, LPARAM lEvent);

// Overrides
 // ClassWizard generated virtual function overrides
 //{{AFX_VIRTUAL(CTrayIcon)
 //}}AFX_VIRTUAL

// Implementation
protected:
 BOOL   m_bEnabled; // does O/S support tray icon?
 BOOL   m_bHidden; // Has the icon been hidden?
 NOTIFYICONDATA m_tnd;

 DECLARE_DYNAMIC(CTrayIcon)
};

#endif

/

cpp文件：

/
// TrayIcon.cpp : implementation file
//
// This is a conglomeration of ideas from the MSJ "Webster" application,
// sniffing round the online docs, and from other implementations such
// as PJ Naughter's "CTrayIconifyIcon" (http://indigo.ie/~pjn/ntray.html)
// especially the "CTrayIcon::OnTrayNotification" member function.
//
// This class is a light wrapper around the windows system tray stuff. It
// adds an icon to the system tray with the specified ToolTip text and
// callback notification value, which is sent back to the Parent window.
//
// The tray icon can be instantiated using either the constructor or by
// declaring the object and creating (and displaying) it later on in the
// program. eg.
//
//  CTrayIcon m_TrayIcon; // Member variable of some class
//  
//  ...
//  // in some member function maybe...
//  m_TrayIcon.Create(pParentWnd, WM_MY_NOTIFY, "Click here",
//        hIcon, nTrayIconID);
//
// Clobbered together by Chris Maunder.
//
//
/

#include "stdafx.h"
#include "TrayIcon.h"

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

IMPLEMENT_DYNAMIC(CTrayIcon, CObject)

/
// CTrayIcon construction/creation/destruction

CTrayIcon::CTrayIcon()
{
 memset(&m_tnd, 0, sizeof(m_tnd));
 m_bEnabled = FALSE;
 m_bHidden  = FALSE;
}

CTrayIcon::CTrayIcon(CWnd* pWnd, UINT uCallbackMessage, LPCTSTR szToolTip,
      HICON icon, UINT uID)
{
 Create(pWnd, uCallbackMessage, szToolTip, icon, uID);
 m_bHidden = FALSE;
}

BOOL CTrayIcon::Create(CWnd* pWnd, UINT uCallbackMessage, LPCTSTR szToolTip,
        HICON icon, UINT uID)
{
 // this is only for Windows 95 (or higher)
 VERIFY(m_bEnabled = ( GetVersion() & 0xff ) >= 4);
 if (!m_bEnabled) return FALSE;

 //Make sure Notification window is valid
 VERIFY(m_bEnabled = (pWnd && ::IsWindow(pWnd->GetSafeHwnd())));
 if (!m_bEnabled) return FALSE;
 
 //Make sure we avoid conflict with other messages
 ASSERT(uCallbackMessage >= WM_USER);

 //Tray only supports tooltip text up to 64 characters
 ASSERT(_tcslen(szToolTip) <= 64);

 // load up the NOTIFYICONDATA structure
 m_tnd.cbSize = sizeof(NOTIFYICONDATA);
 m_tnd.hWnd  = pWnd->GetSafeHwnd();
 m_tnd.uID  = uID;
 m_tnd.hIcon  = icon;
 m_tnd.uFlags = NIF_MESSAGE | NIF_ICON | NIF_TIP;
 m_tnd.uCallbackMessage = uCallbackMessage;
 strcpy (m_tnd.szTip, szToolTip);

 // Set the tray icon
 VERIFY(m_bEnabled = Shell_NotifyIcon(NIM_ADD, &m_tnd));
 return m_bEnabled;
}

CTrayIcon::~CTrayIcon()
{
 RemoveIcon();
}

/
// CTrayIcon icon manipulation

void CTrayIcon::MoveToRight()
{
 HideIcon();
 ShowIcon();
}

void CTrayIcon::RemoveIcon()
{
 if (!m_bEnabled) return;

 m_tnd.uFlags = 0;
    Shell_NotifyIcon(NIM_DELETE, &m_tnd);
    m_bEnabled = FALSE;
}

void CTrayIcon::HideIcon()
{
 if (m_bEnabled && !m_bHidden) {
  m_tnd.uFlags = NIF_ICON;
  Shell_NotifyIcon (NIM_DELETE, &m_tnd);
  m_bHidden = TRUE;
 }
}

void CTrayIcon::ShowIcon()
{
 if (m_bEnabled && m_bHidden) {
  m_tnd.uFlags = NIF_MESSAGE | NIF_ICON | NIF_TIP;
  Shell_NotifyIcon(NIM_ADD, &m_tnd);
  m_bHidden = FALSE;
 }
}

BOOL CTrayIcon::SetIcon(HICON hIcon)
{
 if (!m_bEnabled) return FALSE;

 m_tnd.uFlags = NIF_ICON;
 m_tnd.hIcon = hIcon;

 return Shell_NotifyIcon(NIM_MODIFY, &m_tnd);
}

BOOL CTrayIcon::SetIcon(LPCTSTR lpszIconName)
{
 HICON hIcon = AfxGetApp()->LoadIcon(lpszIconName);

 return SetIcon(hIcon);
}

BOOL CTrayIcon::SetIcon(UINT nIDResource)
{
 HICON hIcon = AfxGetApp()->LoadIcon(nIDResource);

 return SetIcon(hIcon);
}

BOOL CTrayIcon::SetStandardIcon(LPCTSTR lpIconName)
{
 HICON hIcon = LoadIcon(NULL, lpIconName);

 return SetIcon(hIcon);
}

BOOL CTrayIcon::SetStandardIcon(UINT nIDResource)
{
 HICON hIcon = LoadIcon(NULL, MAKEINTRESOURCE(nIDResource));

 return SetIcon(hIcon);
}
 
HICON CTrayIcon::GetIcon() const
{
 HICON hIcon = NULL;
 if (m_bEnabled)
  hIcon = m_tnd.hIcon;

 return hIcon;
}

/
// CTrayIcon tooltip text manipulation

BOOL CTrayIcon::SetTooltipText(LPCTSTR pszTip)
{
 if (!m_bEnabled) return FALSE;

 m_tnd.uFlags = NIF_TIP;
 _tcscpy(m_tnd.szTip, pszTip);

 return Shell_NotifyIcon(NIM_MODIFY, &m_tnd);
}

BOOL CTrayIcon::SetTooltipText(UINT nID)
{
 CString strText;
 VERIFY(strText.LoadString(nID));

 return SetTooltipText(strText);
}

CString CTrayIcon::GetTooltipText() const
{
 CString strText;
 if (m_bEnabled)
  strText = m_tnd.szTip;

 return strText;
}

/
// CTrayIcon notification window stuff

BOOL CTrayIcon::SetNotificationWnd(CWnd* pWnd)
{
 if (!m_bEnabled) return FALSE;

 //Make sure Notification window is valid
 ASSERT(pWnd && ::IsWindow(pWnd->GetSafeHwnd()));

 m_tnd.hWnd = pWnd->GetSafeHwnd();
 m_tnd.uFlags = 0;

 return Shell_NotifyIcon(NIM_MODIFY, &m_tnd);
}

CWnd* CTrayIcon::GetNotificationWnd() const
{
 return CWnd::FromHandle(m_tnd.hWnd);
}

/
// CTrayIcon implentation of OnTrayNotification

LRESULT CTrayIcon::OnTrayNotification(UINT wParam, LONG lParam)
{
 //Return quickly if its not for this tray icon
 if (wParam != m_tnd.uID)
  return 0L;

 CMenu menu, *pSubMenu;

 // Clicking with right button brings up a context menu
 if (LOWORD(lParam) == WM_RBUTTONUP)
 { 
  if (!menu.LoadMenu(m_tnd.uID)) return 0;
  if (!(pSubMenu = menu.GetSubMenu(0))) return 0;

  // Make first menu item the default (bold font)
  ::SetMenuDefaultItem(pSubMenu->m_hMenu, 0, TRUE);

  //Display and track the popup menu
  CPoint pos;
  GetCursorPos(&pos);
  ::SetForegroundWindow(m_tnd.hWnd); 
  ::TrackPopupMenu(pSubMenu->m_hMenu, 0, pos.x, pos.y, 0, m_tnd.hWnd, NULL);

  //pSubMenu->TrackPopupMenu(TPM_LEFTALIGN | TPM_RIGHTBUTTON, pos.x, pos.y, this, NULL);
  menu.DestroyMenu();
 }
 else if (LOWORD(lParam) == WM_LBUTTONDBLCLK)
 {
  if (!menu.LoadMenu(m_tnd.uID)) return 0;
  if (!(pSubMenu = menu.GetSubMenu(0))) return 0;

  // double click received, the default action is to execute first menu item
  ::SetForegroundWindow(m_tnd.hWnd);
  ::SendMessage(m_tnd.hWnd, WM_COMMAND, pSubMenu->GetMenuItemID(0), 0);
  menu.DestroyMenu();
 }

 return 1;
}

使用方法：

afx_msg void CHeartClockDlg::OnNotifyIcon(WPARAM wParam, LPARAM lParam)
{
 if ((wParam == IDI_ICON_ROSE)&&(lParam == WM_LBUTTONDOWN))
  ShowWindow(SW_SHOWNORMAL);
 if ((wParam == IDI_ICON_ROSE)&&(lParam == WM_RBUTTONDOWN))
  {// 用户在托盘图标上单击鼠标右键，弹出上下文菜单隐藏/显示对话框。
    CMenu oMenu;
    if (oMenu.LoadMenu(IDR_MENU_NOTIFY))       
    {
     CMenu* pPopup = oMenu.GetSubMenu(0);
    //ASSERT(pPopup != NULL);
    CPoint oPoint;
    //if (IsWindowVisible())// 根据对话框窗口的显示/隐藏状态修改菜单名称
    //oMenu.ModifyMenu(IDC_SHOW,MF_STRING,IDC_SHOW,"隐藏(&H)");
    //else
    //oMenu.ModifyMenu(IDC_SHOW,MF_STRING,IDC_SHOW,"显示(&S)");
    // 确定鼠标位置以便在该位置附近显示菜单
    GetCursorPos( &oPoint );
    SetForegroundWindow();
    pPopup->TrackPopupMenu(
     TPM_LEFTALIGN | TPM_RIGHTBUTTON,
     oPoint.x, oPoint.y, this);
    }
  }
}