taskbaricon.h

这是一本学习 window编程的很好的参考教材

#ifndef __TASKBARICON_H__
#define __TASKBARICON_H__

/////////////////////////////////////////////////////////////////////////////
// Task Bar Icon class
//
// Written by Bjarke Viksoe ([e-mail])
// Copyright (c) 2001 Bjarke Viksoe.
//
// This code may be used in compiled form in any way you desire. This
// file may be redistributed by any means PROVIDING it is 
// not sold for profit without the authors written consent, and 
// providing that this notice and the authors name is included. 
//
// This file is provided "as is" with no expressed or implied warranty.
// The author accepts no liability if it causes any damage to you or your
// computer whatsoever. It's free, so don't hassle me about it.
//
// Beware of bugs.
//

#pragma once

#ifndef __cplusplus
   #error WTL requires C++ compilation (use a .cpp suffix)
#endif

#ifndef __ATLWIN_H__
   #error TaskBarIcon.h requires atlwin.h to be included first
#endif

#if (_WIN32_IE < 0x0400)
   #error TaskBarIcon.h requires _WIN32_IE >= 0x0400
#endif


#ifndef WM_TASKBARICON
#define WM_TASKBARICON WM_APP+423
#endif

#define TASKBAR_MESSAGE_HANDLER(ti, msg, func) \
   if(uMsg==ti.m_nid.uCallbackMessage && wParam==ti.m_nid.uID && lParam==msg) \
   { \
      bHandled = TRUE; \
      lResult = func(lParam, bHandled); \
      if(bHandled) \
         return TRUE; \
   }

typedef HRESULT (WINAPI* DLLGETVERSION) (DLLVERSIONINFO *pdvi);
class CTaskBarIcon
{
public:
   NOTIFYICONDATA m_nid; 
   /*HMENU m_hMenu;*/
   UINT m_iTaskbarRestartMsg;
	CMDICommandBarXPCtrl m_CmdBar;
   CTaskBarIcon() : /*m_hMenu(NULL),*/ m_iTaskbarRestartMsg(0)
   {
      ::ZeroMemory(&m_nid, sizeof(m_nid));
      m_nid.cbSize = sizeof(m_nid); 
      m_nid.uCallbackMessage = WM_TASKBARICON;
      m_iTaskbarRestartMsg = ::RegisterWindowMessage(TEXT("TaskbarCreated"));
	  
	
   }
   ~CTaskBarIcon()
   {
      Uninstall();
   }

   BEGIN_MSG_MAP(CTaskBarIcon)
      MESSAGE_HANDLER(m_iTaskbarRestartMsg, OnRestart)
      TASKBAR_MESSAGE_HANDLER((*this), WM_RBUTTONDOWN, OnContextMenu)
   END_MSG_MAP()

   BOOL Install(HWND hWnd, UINT iID, HICON hIcon, HMENU hMenu, LPTSTR lpszTip)
   {
      ATLASSERT(::IsWindow(hWnd));
  /*    ATLASSERT(m_hMenu==NULL);*/
      ATLASSERT(m_nid.hIcon==NULL);
      m_nid.hWnd = hWnd;
      m_nid.uID = iID;
      m_nid.hIcon = hIcon; 
      ::lstrcpyn(m_nid.szTip, (lpszTip!=NULL ? lpszTip : _T("")), sizeof(m_nid.szTip)/sizeof(TCHAR)); 
     /* m_hMenu = hMenu;*/
      return AddTaskBarIcon();
   }
   BOOL Install(HWND hWnd, UINT iID, UINT nRes)
   {
      ATLASSERT(::IsWindow(hWnd));
      //ATLASSERT(m_hMenu==NULL);
      ATLASSERT(m_nid.hIcon==NULL);
      m_nid.hWnd = hWnd;
      m_nid.uID = iID;
      m_nid.hIcon = (HICON)::LoadImage(_Module.GetResourceInstance(), MAKEINTRESOURCE(nRes), IMAGE_ICON, ::GetSystemMetrics(SM_CXSMICON), ::GetSystemMetrics(SM_CYSMICON), LR_DEFAULTCOLOR);
      m_nid.szTip[0] = '\0';
      ::LoadString(_Module.GetResourceInstance(), nRes, m_nid.szTip, sizeof(m_nid.szTip)/sizeof(TCHAR));
      //m_hMenu = ::LoadMenu(_Module.GetResourceInstance(), MAKEINTRESOURCE(nRes));

	  HWND hWndCmdBar = m_CmdBar.Create(hWnd, CWindow::rcDefault, NULL, ATL_SIMPLE_CMDBAR_PANE_STYLE);
	  // Attach menu
	  CMenuHandle menu;
	  menu.LoadMenu(IDR_POPUP);
	  m_CmdBar.AttachMenu(menu);
	  // Load command bar images
	  m_CmdBar.LoadImages(IDR_MAINFRAME);
	  // Remove old menu
	 /* SetMenu(hWnd,NULL);*/
	  m_CmdBar.Prepare();
      return AddTaskBarIcon();
   }
   BOOL Uninstall()
   {
      BOOL res = TRUE;
      if( m_nid.hWnd!=NULL ) res = DeleteTaskBarIcon();
      m_nid.hWnd=NULL;
      if( m_nid.hIcon ) ::DestroyIcon(m_nid.hIcon);
      m_nid.hIcon = NULL;
      //if( m_hMenu ) ::DestroyMenu(m_hMenu);
      //m_hMenu = NULL;
      return res;
   }
   BOOL IsInstalled() const
   {
      return m_nid.hWnd!=NULL;
   }
   void SetIcon(HICON hIcon) 
   { 
      ATLASSERT(m_nid.hIcon==NULL);
      m_nid.hIcon = hIcon; 
   }
  /* void SetMenu(HMENU hMenu) 
   { 
      ATLASSERT(m_hMenu==NULL);
      ATLASSERT(::IsMenu(hMenu));
      m_hMenu = hMenu; 
   }*/
   BOOL AddTaskBarIcon()
   {
      ATLASSERT(::IsWindow(m_nid.hWnd));
      m_nid.uFlags = NIF_MESSAGE | NIF_ICON | NIF_TIP; 
      BOOL res = ::Shell_NotifyIcon(NIM_ADD, &m_nid); 
      return res;
   }
   BOOL ChangeIcon(HICON hIcon,BOOL bDestroy=FALSE)
   {
      // NOTE: The class takes ownership of the icon!
      ATLASSERT(::IsWindow(m_nid.hWnd));
      if( m_nid.hIcon && bDestroy) ::DestroyIcon(m_nid.hIcon);
      m_nid.uFlags = NIF_ICON; 
      m_nid.hIcon = hIcon;
      BOOL res = ::Shell_NotifyIcon(NIM_MODIFY, &m_nid); 
      return res;
   }
   BOOL ShowBalloon(TCHAR * szBalloonTitle,char* fmt, ...)
   {
	   DLLVERSIONINFO ver;
	   int shell32_ver;
	   ZeroMemory((void*)&ver,sizeof DLLVERSIONINFO);
	   ver.cbSize = sizeof(DLLVERSIONINFO);
	   DLLGETVERSION DllGetVersion;
	   HMODULE hMod=LoadLibrary("Shell32.dll");
	   if( hMod==NULL)return FALSE;
	   DllGetVersion=(DLLGETVERSION)GetProcAddress(hMod,"DllGetVersion");
	   if(DllGetVersion==NULL){
		   FreeLibrary(hMod);
		   return FALSE;
	   }
	   if(DllGetVersion(&ver) == NOERROR) {
		   shell32_ver = (ver.dwMajorVersion * 100) + ver.dwMinorVersion;
		   if(shell32_ver < 500){
			   FreeLibrary(hMod);
			   return FALSE;
		   }
		   FreeLibrary(hMod);

		   char szBuffer[256];
		   va_list argptr;
		   va_start(argptr,fmt);
		   vsprintf(szBuffer,fmt,argptr);
		   va_end(argptr);

		   m_nid.uFlags  = m_nid.uFlags| NIF_INFO|NIF_ICON; 
		 
		   strcpy(m_nid.szInfo  ,szBuffer);
		   strcpy(m_nid.szInfoTitle,szBalloonTitle);

		   m_nid.dwInfoFlags = NIIF_INFO; 
		

		   m_nid.uTimeout = 100;



		   //    Shell_NotifyIcon(NIM_SETVERSION,&niData);
		   Shell_NotifyIcon(NIM_MODIFY,&m_nid);
		 
		   return TRUE;
	   }
	   else{
		   FreeLibrary(hMod);
		   return FALSE;

	   }
   }
   BOOL DeleteTaskBarIcon() 
   { 
       return ::Shell_NotifyIcon(NIM_DELETE, &m_nid); 
   } 

   // Message handlers

   LRESULT OnRestart(UINT /*uMsg*/, WPARAM /*wParam*/, LPARAM /*lParam*/, BOOL& bHandled)
   {
      AddTaskBarIcon();
      bHandled = FALSE;
      return 0;
   }
   LRESULT OnContextMenu(LPARAM /*uMsg*/, BOOL& bHandled)
   {
      //if( !::IsMenu(m_hMenu) ) {
      //   bHandled = FALSE;
      //   return 0;
      //}
      //HMENU hSubMenu = ::GetSubMenu(m_hMenu, 0);   
      //ATLASSERT(::IsMenu(hSubMenu));
      //// Make first menu-item the default (bold font)      
      //::SetMenuDefaultItem(hSubMenu, 0, TRUE); 
      // Display the menu at the current mouse location.
      POINT pt;
      ::GetCursorPos(&pt);      
      ::SetForegroundWindow(m_nid.hWnd);
	 CMenuHandle menu = m_CmdBar.m_hMenu;
	   CMenuHandle submenu = menu.GetSubMenu(0);
	  m_CmdBar.TrackPopupMenu(submenu, TPM_LEFTALIGN | TPM_RIGHTBUTTON, pt.x, pt.y);
    //  ::PostMessage(m_nid.hWnd, WM_NULL, 0,0); // Fixes Win95 bug
      return 0;
   }
};

#endif // __TASKBARICON_H__