MFC中托盘（TRAYICON）实现

最新推荐文章于 2021-10-27 11:00:57 发布

weixin_30325071

最新推荐文章于 2021-10-27 11:00:57 发布

阅读量157

点赞数

文章标签： c/c++

原文链接：http://www.cnblogs.com/daoluanxiaozi/archive/2011/11/12/2246733.html

版权

写在最前面的

将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;
}