MFC中托盘（TRAYICON）实现

写在最前面的

将MFC中托盘功能模块抽象成一个类，把整个程序的功能模块细分了。这个想法在网上已经是俯拾即是了，但仍旧不能一下子明白其中的东西，特别是将其抽象之后。 在使用这个类的时候，需要注意：托盘菜单的ID要和图标资源的ID一样，否则会出错。

具体实现代码 

添加新的类，选择父类是CCmdTarget，下面的代码中有足够的提醒：

TrayIcon.h  

#pragma once

//  TrayIcon.h : 头文件
//
//     继承自CCmdTarget才能接收消息，详见《深入浅出 MFC》
class CTrayIcon :  public CCmdTarget 
{ 
    DECLARE_DYNAMIC(CTrayIcon)

public: 
    CTrayIcon(UINT uID); 
     ~CTrayIcon(); 
     
     //  Call this to receive tray notifications 
     void SetNotificationWnd(CWnd* pNotifyWnd, UINT uCbMsg);

     //  SetIcon functions. To remove icon, call SetIcon(0) 
     // 只安装图盘图标 
    BOOL SetIcon(UINT uID,LPWSTR lpTip=NULL);

     // 安装托盘图标和设置提示信息 
    BOOL SetIcon(HICON hIcon, LPWSTR lpTip);

     // 安装托盘图标和设置提示信息 
    BOOL SetIcon(LPCTSTR lpResName, LPWSTR lpTip) 
    { 
         return SetIcon(lpResName ?  
            AfxGetApp()->LoadIcon(lpResName) : NULL, lpTip); 
    }

     /*  
    BOOL SetStandardIcon(LPCTSTR lpszIconName, LPCSTR lpTip) 
    { 
        return SetIcon(::LoadIcon(NULL, lpszIconName), lpTip); 
    } 
     */

     virtual LRESULT OnTrayNotification(WPARAM uID, LPARAM lEvent);

     /*
    BOOL ShowBalloonTip();
    笔者没有实现泡泡其实，如果有想法的可以留言
     */
protected: 
    NOTIFYICONDATA m_nid; 
    DECLARE_MESSAGE_MAP() 
};

 

TrayIcon.cpp 

//  TrayIcon.cpp : 实现文件 
//
#include  " stdafx.h " 
#include  " TrayIconShell.h " 
#include  " TrayIcon.h "

/*  
注意：托盘菜单的ID要和图标资源的ID一样，否则会出错！
*/
IMPLEMENT_DYNAMIC(CTrayIcon, CCmdTarget)

CTrayIcon::CTrayIcon(UINT uID) 
{ 
    ::memset(&m_nid, 0, sizeof(m_nid)); 
    m_nid.uID = uID; 
    m_nid.cbSize =  sizeof(NOTIFYICONDATA); 
}

CTrayIcon::~CTrayIcon() 
{ 
     this->SetIcon( 0); 
}

void CTrayIcon::SetNotificationWnd(CWnd* pNotifyWnd, UINT uCbMsg) 
{ 
     // 断言，正确才中断 
    ASSERT(pNotifyWnd==NULL 
        ||::IsWindow(pNotifyWnd->GetSafeHwnd()));

    m_nid.hWnd = pNotifyWnd->m_hWnd;

    ASSERT(uCbMsg ==  0||uCbMsg>=WM_USER);

    m_nid.uCallbackMessage = uCbMsg; 
}

BOOL CTrayIcon::SetIcon(UINT uID,LPWSTR lpTip) 
{ 
    HICON hIcon = NULL; 
     if(uID) 
    { 
        m_nid.uID = uID; // 更改当前对象的uID 
        hIcon = AfxGetApp()->LoadIconW(uID); 
    } 
     return SetIcon(hIcon,lpTip); 
}

BOOL CTrayIcon::SetIcon(HICON hIcon, LPWSTR lpTip) 
{ 
    UINT uMsg; 
    m_nid.uFlags =  0;

     if(hIcon) 
    { 
        uMsg = m_nid.hIcon?NIM_MODIFY:NIM_ADD; // 当前如果存在ICON，就修改 
        m_nid.hIcon = hIcon; 
        m_nid.uFlags = NIF_ICON; // 设置标志

         // 如果存在hIcon才进行设置提示 
         if(lpTip) 
        { 
            ::lstrcpyn(m_nid.szTip,(LPCWSTR)lpTip, sizeof(m_nid.szTip)); 
        } 
         if(m_nid.szTip[ 0]) 
            m_nid.uFlags |= NIF_TIP; // 设置标志 
    } 
     else // 删除图标 
    { 
         // 如果当前m_nid的hIcon为null，那么直接结束，啥都不用干啦 
         if(m_nid.hIcon==NULL)     return  true;
        uMsg = NIM_DELETE;     // 如果当前m_nid的hIcon不为null，删除图标 
        m_nid.hIcon = NULL; 
    }

     if(m_nid.uCallbackMessage && m_nid.hWnd) 
        m_nid.uFlags |= NIF_MESSAGE;

    BOOL bRet = ::Shell_NotifyIcon(uMsg,&m_nid);

     // 把失败和NIM_DELETE的操作放在这里 
     if(!bRet)    m_nid.hIcon = NULL;
     return bRet; 
} 

LRESULT CTrayIcon::OnTrayNotification(WPARAM uID, LPARAM lEvent) 
{ 
     if(uID != m_nid.uID ||         // 如果uID不属于此对象 
        (lEvent != WM_RBUTTONUP && lEvent != WM_LBUTTONDBLCLK)) 
         return  0;

    CMenu menu;  // 知道吗，MFC的封装我觉得在这里已经表现得淋漓尽致了。
     if(!menu.LoadMenuW(m_nid.uID)) 
         return  0;

    CMenu * pSubMenu = menu.GetSubMenu( 0);

     if(!pSubMenu) 
         return  0;

     if(lEvent == WM_RBUTTONUP) 
    { 
        CPoint mouse; 
        ::GetCursorPos(&mouse); 
        ::SetForegroundWindow(m_nid.hWnd); 
        ::TrackPopupMenu(pSubMenu->m_hMenu,TRUE,mouse.x,mouse.y, 0, 
            m_nid.hWnd,NULL); 
    } 
     if(lEvent == WM_LBUTTONDBLCLK) 
    { 
        ::SendMessage(m_nid.hWnd,WM_COMMAND, 
            pSubMenu->GetMenuItemID( 0), 0); 
    } 
     return  1; 
}

BEGIN_MESSAGE_MAP(CTrayIcon, CCmdTarget) 
END_MESSAGE_MAP()

使用方法 

由于这个类的父类是CCmdTarget，所以，可以接受命令消息（菜单的消息），我们可以自定义添加消息。 

在需要使用“托盘”的窗口类中声明一个CTrayIcon的对象，然后在OnCreate函数当中调用即可。

int CTrayIconShellDlg::OnCreate(LPCREATESTRUCT lpCreateStruct)
{
     if (CDialog::OnCreate(lpCreateStruct) == - 1)
         return - 1;

     //  TODO:  在此添加您专用的创建代码
    TCHAR tip[ 100] = TEXT( " 捣乱小子 ");
    TrayIcon.SetNotificationWnd( this,WM_NOTIFY_MSG);
    TrayIcon.SetIcon(IDR_MAINFRAME,tip);
     return  0;
}

 总结 

　　在这一次实验之后，我更加仰慕和崇尚面向对象（OO）的思想，如果编程思路清晰，数据设计得当，接口功能完善，以及接口与接口之间有良好的信息沟通，OO能让你在程序设计当中游刃有余，所以在平时的时候要养成面向对象的思考习惯。

　　在C++面向对象编程的过程当中，会遇到一些公共的接口，这些接口只完成某些固定的操作，与其他的接口的联系不是非常的紧密的时候，我们可以将此接口设计为static模式，这样就不会让每个对象都担负着那么多的方法了。

 

本文完 Monday, April 23, 2012（更新）

捣乱小子 http://www.cnblogs.com/daoluanxiaozi/