TrayIcon

一直以为右下角气泡提醒是自己做的一个对话框.....才知道是TRAYICON的东东.

找到一篇好文章

http://www.codeproject.com/Articles/3147/WTL-Tray-Icon-Template

 

在此基础上少加改动,把气泡添上了..呵呵.

 

#pragma once

template <typename T>
class ATL_NO_VTABLE CTrayIconImpl
{
    UINT WM_TRAYICON;
    NOTIFYICONDATA m_nid;
    bool m_bInstalled;
public:
    CTrayIconImpl() : m_bInstalled( false )
    {
        memset( &m_nid, 0, sizeof( NOTIFYICONDATA ) );
        m_nid.cbSize = sizeof( NOTIFYICONDATA );

        WM_TRAYICON = ::RegisterWindowMessage( TEXT( "WM_TRAYICON" ) );
    }

    ~CTrayIconImpl()
    {
        if ( m_bInstalled )
        {
            m_nid.uFlags = 0;
            Shell_NotifyIcon( NIM_DELETE, &m_nid );
        }
    }

    bool InstallTrayIcon( HICON hIcon, LPCTSTR lpszTip = NULL, UINT uID = -1 )
    {
        T* pT = static_cast<T*>( this );

        if ( m_bInstalled )
        {
            return false;
        }

        m_nid.hWnd = pT->m_hWnd;
        m_nid.uID = uID;
        m_nid.hIcon = hIcon;
        m_nid.uCallbackMessage = WM_TRAYICON;

        if ( lpszTip != NULL )
        {
            m_nid.uFlags = NIF_MESSAGE | NIF_ICON | NIF_TIP;
            lstrcpy( m_nid.szTip, lpszTip );
        }
        else
        {
            m_nid.uFlags = NIF_MESSAGE | NIF_ICON;
        }

        m_bInstalled = Shell_NotifyIcon( NIM_ADD, &m_nid ) ? true : false;

        return m_bInstalled;
    }


    bool RemoveTrayIcon()
    {
        if ( m_bInstalled )
        {
            m_nid.uFlags = 0;

            if ( Shell_NotifyIcon( NIM_DELETE, &m_nid ) )
            {
                m_bInstalled = false;
                return true;
            }
        }

        return false;
    }

    bool SetTrayTip( LPCTSTR lpszTip )
    {
        if ( lpszTip != NULL )
        {
            m_nid.uFlags = NIF_TIP;
            lstrcpy( m_nid.szTip, lpszTip );
            return Shell_NotifyIcon( NIM_MODIFY, &m_nid ) ? true : false;
        }

        return false;
    }

    bool SetTrayInfo( LPCTSTR lpszInfo, LPCTSTR lpszTitle, UINT uTimeout = 3000, DWORD dwInfoFlags = NIIF_INFO )
    {
        if ( lpszInfo != NULL )
        {
            m_nid.uFlags = NIF_INFO;
            m_nid.uTimeout = uTimeout;
            m_nid.dwInfoFlags = dwInfoFlags;
            lstrcpy( m_nid.szInfo, lpszInfo );
            lstrcpy( m_nid.szInfoTitle, lpszTitle ? lpszTitle : TEXT( "" ) );
            return Shell_NotifyIcon( NIM_MODIFY, &m_nid ) ? true : false;
        }

        return false;
    }

    BEGIN_MSG_MAP( CTrayIconImpl<T> )
        MESSAGE_HANDLER( WM_TRAYICON, OnTrayIcon )
        MESSAGE_HANDLER( WM_SYSCOMMAND, OnSysCommand )
    END_MSG_MAP()

    void PrepareTrayMenu( HMENU hMenu )
    {
        ATLASSERT( FALSE );

        /*
        CMenuHandle menu(hMenu);
        menu.EnableMenuItem(ID_TEST_DISABLED, MF_GRAYED);
        menu.CheckMenuItem(ID_TEST_CHECKED, MF_CHECKED);
        */
    }

    LRESULT OnTrayIcon( UINT /*uMsg*/, WPARAM wParam, LPARAM lParam, BOOL& /*bHandled*/ )
    {
        POINT pt;
        HMENU hMenu, hPopup;

        T* pT = static_cast<T*>( this );

        if ( wParam != m_nid.uID )
        {
            return 0;
        }

        switch ( LOWORD( lParam ) )
        {
            case WM_RBUTTONUP:
                ::GetCursorPos( &pt );

                hMenu = ::LoadMenu( ModuleHelper::GetResourceInstance(), MAKEINTRESOURCE( m_nid.uID ) );

                if ( !hMenu )return 0;

                hPopup = ::GetSubMenu( hMenu, 0 );
                pT->PrepareTrayMenu( hPopup );

                ::SetForegroundWindow( pT->m_hWnd );
                ::TrackPopupMenu( hPopup, TPM_RIGHTALIGN, pt.x, pt.y, 0, pT->m_hWnd, NULL );
                // BUGFIX: See "PRB: Menus for Notification Icons Do Not Work Correctly"
                pT->PostMessage( WM_NULL );

                ::DestroyMenu( hMenu );
                break;

            case WM_LBUTTONDBLCLK:
                if ( pT->IsIconic() )
                {
                    pT->ShowWindow( SW_RESTORE );
                }
                else
                {
                    if ( !pT->IsWindowVisible() )
                    {
                        pT->ShowWindow( SW_SHOW );
                        ::BringWindowToTop( pT->m_hWnd );
                        ::SetForegroundWindow( pT->m_hWnd );
                    }
                    else
                    {
                        pT->ShowWindow( SW_HIDE );
                    }
                }

                break;

            case NIN_BALLOONUSERCLICK:
                if ( pT->IsIconic() )
                {
                    pT->ShowWindow( SW_RESTORE );
                }
                else
                {
                    if ( !pT->IsWindowVisible() )
                    {
                        pT->ShowWindow( SW_SHOW );
                        ::BringWindowToTop( pT->m_hWnd );
                    }

                    ::SetForegroundWindow( pT->m_hWnd );
                }

                break;
        }

        return 0;
    }

    LRESULT OnSysCommand( UINT /*uMsg*/, WPARAM wParam, LPARAM /*lParam*/, BOOL& bHandled )
    {
        T* pT = static_cast<T*>( this );

        if ( wParam == SC_MINIMIZE )
        {
            pT->ShowWindow( SW_HIDE );
            bHandled = TRUE;
        }
        else
        {
            bHandled = FALSE;
        }

        return 0;
    }
};