trayicon.cpp

仿QQ的聊天程序vc++/mfc来学习的简易聊天室代码功能较全

// TrayIcon.cpp : implementation file
//

#include "stdafx.h"
#include "TrayIcon.h"
#include "GfxPopupMenu.h"
#include "resource.h"

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

/////////////////////////////////////////////////////////////////////////////
// CTrayIcon
IMPLEMENT_DYNCREATE(CTrayIcon, CCmdTarget)
CTrayIcon::CTrayIcon()
{
	//初始化NOTIFYICONDATA结构变量
	memset(&m_nid, 0 , sizeof(m_nid));
	m_nid.cbSize = sizeof(m_nid);	
}

CTrayIcon::~CTrayIcon()
{
	SetIcon(0,NULL); // 从系统托盘中删除图标
}

// 设定通知窗口，该窗口必须已被创建
void CTrayIcon::SetNotificationWnd(CWnd* pNotifyWnd, UINT uCbMsg)
{
    ASSERT(pNotifyWnd==NULL || ::IsWindow(pNotifyWnd->GetSafeHwnd()));
	
    m_nid.hWnd = pNotifyWnd->GetSafeHwnd();
	
    ASSERT(uCbMsg==0 || uCbMsg>=WM_USER);
    m_nid.uCallbackMessage = uCbMsg;
}

BOOL CTrayIcon::SetIcon(UINT uID,LPCSTR lpTip)
{ 
	HICON hicon=NULL;
	if (uID)
	{
		if(lpTip!=NULL)
			strcpy(m_nid.szTip,lpTip);
		hicon = AfxGetApp()->LoadIcon(uID);
	}
	
	UINT msg;
	m_nid.uFlags = 0;
	
	// 设定图标
	if (hicon)
	{
		// 判断是要在系统托盘中增加还是要删除图标
		msg = m_nid.hIcon ? NIM_MODIFY : NIM_ADD;
		m_nid.hIcon = hicon;
		m_nid.uFlags = NIF_ICON|NIF_MESSAGE;
		if(lpTip!=NULL)
			m_nid.uFlags|=NIF_TIP;
	} 
	else 
	{ // 删除图标
		if (m_nid.hIcon==NULL)
			return TRUE;   //已被删除
		msg = NIM_DELETE;
	}	
	
	BOOL bRet = Shell_NotifyIcon(msg, &m_nid);
	if (msg==NIM_DELETE || !bRet)
		m_nid.hIcon = NULL; 
	return bRet;
}

// 缺省事件处理程序，该程序处理鼠标右击及双击事件。
LRESULT CTrayIcon::OnTrayNotification(WPARAM wID,LPARAM lEvent)
{
	if (wID!=m_nid.uID ||(lEvent!=WM_RBUTTONUP && lEvent!=WM_LBUTTONDBLCLK && lEvent!=WM_LBUTTONDOWN))
		return 0;	
	
	CMenu menu;
	CGfxPopupMenu cMenu;

	if(!menu.LoadMenu(IDR_MAINFRAME))
		return 0;	
	
	if (lEvent==WM_RBUTTONUP) {		
		
		cMenu.Attach(menu.GetSubMenu(0)->GetSafeHmenu());
		cMenu.LoadToolBarResource(IDR_TOOLBAR1);
		cMenu.RemapMenu(&cMenu);
		// 在鼠标的当前位置弹出菜单。
		CPoint mouse;
		GetCursorPos(&mouse);
		::SetForegroundWindow(m_nid.hWnd); 
		::TrackPopupMenu(cMenu.m_hMenu,
			0,
			mouse.x,
			mouse.y,
			0,
			m_nid.hWnd,
			NULL);		
	}
	else if(lEvent==WM_LBUTTONDOWN)
	{		
		cMenu.Attach(menu.GetSubMenu(1)->GetSafeHmenu());
		cMenu.LoadToolBarResource(IDR_TOOLBAR2);
		cMenu.RemapMenu(&cMenu);
		CPoint mouse;
		GetCursorPos(&mouse);
		::SetForegroundWindow(m_nid.hWnd); 
		::TrackPopupMenu(cMenu.m_hMenu,
			0,
			mouse.x,
			mouse.y,
			0,
			m_nid.hWnd,
			NULL);
	}
	else
	{
		::SendMessage(m_nid.hWnd, WM_HOTKEY,1001, 0);
	}
	
	return 1; // 表示事件已被处理
}

/////////////////////////////////////////////////////////////////////////////
// CTrayIcon message handlers