vc 托盘封装

#ifndef _INCLUDED_TRAYICON_H_
#define _INCLUDED_TRAYICON_H_
class CTrayIcon : public CObject
{
// Construction/destruction
public:
    CTrayIcon();
    CTrayIcon(CWnd* pWnd, UINT uCallbackMessage, LPCTSTR szTip, HICON icon, UINT uID);
    virtual ~CTrayIcon(); 
public:
    BOOL Enabled() { return m_bEnabled; }
    BOOL Visible() { return !m_bHidden; } 
    int Create(CWnd* pWnd, UINT uCallbackMessage, LPCTSTR szTip, HICON icon, UINT uID);
    BOOL    SetTooltipText(LPCTSTR pszTooltipText);
    BOOL    SetTooltipText(UINT nID);
    CString GetTooltipText() const; 
    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(); 
    BOOL  SetNotificationWnd(CWnd* pNotifyWnd);
    CWnd* GetNotificationWnd() const;
    virtual LRESULT OnTrayNotification(WPARAM uID, LPARAM lEvent);
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

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