用于WTL工具栏的辅助类CToolBarHelper

#pragma once

template <typename T>
class ATL_NO_VTABLE CToolBarHelper
{
    HFONT m_nonClientFont;

    CSimpleMap<UINT, UINT> m_mapMenu;

public:
    CToolBarHelper() : m_nonClientFont( NULL ) {}

    ~CToolBarHelper()
    {
        if ( m_nonClientFont )
        {
            ::DeleteObject( m_nonClientFont );
        }
    }

    BEGIN_MSG_MAP( CToolBarHelper<T> )
        COMMAND_CODE_HANDLER( CBN_SELCHANGE, OnSelChangeToolBarCombo )
        NOTIFY_CODE_HANDLER( TBN_DROPDOWN, OnToolBarDropDown )
    END_MSG_MAP()

    BOOL ModifyToolBarStyle( HWND hWndToolBar )
    {
        CToolBarCtrl tbCtrl;

        tbCtrl = hWndToolBar;
        ATLASSERT( tbCtrl );

        return tbCtrl.ModifyStyle( 0, TBSTYLE_LIST );
    }

    void AddToolBarDropDownMenu( HWND hWndToolBar, UINT nButtonID, UINT nMenuID )
    {
        DWORD tbStyleEx;
        TBBUTTONINFO tbBtnInfo;
        CToolBarCtrl tbCtrl;

        tbCtrl = hWndToolBar;
        ATLASSERT( tbCtrl );

        tbStyleEx = tbCtrl.GetExtendedStyle() | TBSTYLE_EX_DRAWDDARROWS;
        tbCtrl.SetExtendedStyle( tbStyleEx );

        memset( &tbBtnInfo, 0, sizeof( TBBUTTONINFO ) );
        tbBtnInfo.cbSize = sizeof( TBBUTTONINFO );
        tbBtnInfo.dwMask = TBIF_STYLE;

        if ( tbCtrl.GetButtonInfo( nButtonID, &tbBtnInfo ) != -1 )
        {
            tbBtnInfo.fsStyle |= TBSTYLE_DROPDOWN;
            tbCtrl.SetButtonInfo( nButtonID, &tbBtnInfo );
            m_mapMenu.Add( nButtonID, nMenuID );
        }
    }

    LRESULT OnToolBarDropDown( int /*idCtrl*/, LPNMHDR pnmh, BOOL& /*bHandled*/ )
    {
        UINT nMenuID;
        RECT rect;
        POINT pt;
        HMENU hMenu, hPopup;
        CToolBarCtrl tbCtrl;
        T* pT;

        nMenuID = m_mapMenu.Lookup( ( ( LPNMTOOLBARW )pnmh )->iItem );

        if ( nMenuID )
        {
            tbCtrl = pnmh->hwndFrom;
            tbCtrl.GetItemRect( tbCtrl.CommandToIndex( ( ( LPNMTOOLBARW )pnmh )->iItem ), &rect );
            pt.x = rect.left;
            pt.y = rect.bottom;
            tbCtrl.MapWindowPoints( HWND_DESKTOP, &pt, 1 );
            hMenu = ::LoadMenu( ModuleHelper::GetResourceInstance(), MAKEINTRESOURCE( nMenuID ) );
            if ( hMenu )
            {
                hPopup = ::GetSubMenu( hMenu, 0 );
                if ( hPopup )
                {
                    pT = static_cast<T*>( this );
                    pT->PrepareToolBarMenu( hPopup );
                    pT->m_CmdBar.TrackPopupMenu( hPopup, \
                                                 TPM_RIGHTBUTTON | TPM_VERTICAL, pt.x, pt.y );
                }

                ::DestroyMenu( hMenu );
            }
        }

        return 0;
    }

    void PrepareToolBarMenu( HMENU /*hMenu*/ )
    {
        //ATLASSERT( FALSE );
    }

    void AddToolBarButtonText( HWND hWndToolBar, UINT nID, LPCTSTR lpsz )
    {
        UINT nIndex;
        DWORD tbStyleEx;
        TBBUTTON tbButton;
        CToolBarCtrl tbCtrl;

        tbCtrl = hWndToolBar;
        ATLASSERT( tbCtrl );

        tbStyleEx = tbCtrl.GetExtendedStyle() | TBSTYLE_EX_MIXEDBUTTONS;
        tbCtrl.SetExtendedStyle( tbStyleEx );

        nIndex = tbCtrl.CommandToIndex( nID );
        ATLASSERT( nIndex != -1 );

        memset( &tbButton, 0, sizeof( TBBUTTON ) );
        if ( tbCtrl.GetButton( nIndex, &tbButton ) )
        {
            tbButton.iString = tbCtrl.AddStrings( lpsz );
            tbButton.fsStyle |= TBSTYLE_AUTOSIZE | BTNS_SHOWTEXT;

            if ( tbCtrl.DeleteButton( nIndex ) )
            {
                tbCtrl.InsertButton( nIndex, &tbButton );
            }
        }
    }


    HWND CreateToolBarComboBox( HWND hWndToolBar, UINT nID, UINT nWidth = 16, UINT nHeight = 16,
                                DWORD dwComboStyle = CBS_DROPDOWNLIST )
    {
        POINT siz;
        UINT nIndex;
        UINT cx, cy;
        RECT rect;
        DWORD dwStyle;
        CComboBox cbxCtrl;
        TBBUTTONINFO tbBtnInfo;
        CToolBarCtrl tbCtrl;
        T* pT;

        if ( !GetFontSize( &siz ) )
        {
            return NULL;
        }

        cx = ( nWidth + 8 ) * siz.x;
        cy = nHeight * siz.y;

        memset( &tbBtnInfo, 0, sizeof( TBBUTTONINFO ) );
        tbBtnInfo.cbSize = sizeof( TBBUTTONINFO );
        tbBtnInfo.dwMask = ( TBIF_SIZE | TBIF_STATE | TBIF_STYLE );
        //tbBtnInfo.fsState = 0;
        tbBtnInfo.fsStyle = BTNS_SHOWTEXT;
        tbBtnInfo.cx = ( WORD )cx;



        tbCtrl = hWndToolBar;
        ATLASSERT( tbCtrl );

        if ( tbCtrl.SetButtonInfo( nID, &tbBtnInfo ) )
        {
            nIndex = tbCtrl.CommandToIndex( nID );
            ATLASSERT( nIndex != -1 );
            tbCtrl.GetItemRect( nIndex, &rect );
            rect.bottom = cy;

            pT = static_cast<T*>( this );

            dwStyle = WS_CHILD | WS_VISIBLE | WS_VSCROLL | WS_TABSTOP | dwComboStyle;
            cbxCtrl.Create( pT->m_hWnd, rect, NULL, dwStyle, 0, nID );
            cbxCtrl.SetFont( m_nonClientFont );
            cbxCtrl.SetParent( tbCtrl );

            MoveToolBarComboBox( tbCtrl, cbxCtrl );

            return cbxCtrl;
        }

        return NULL;
    }

    LRESULT OnSelChangeToolBarCombo( WORD /*wNotifyCode*/, WORD wID, HWND hWndCtl, BOOL& /*bHandled*/ )
    {
        T* pT;

        pT = static_cast<T*>( this );
        pT->ToolBarComboboxAction( wID , hWndCtl );
        pT->SetFocus();

        return 0;
    }

    void ToolBarComboboxAction( WORD /*wID*/, HWND /*hWndCtl*/ )
    {
        //ATLASSERT( FALSE );
    }

private:

    BOOL GetFontSize( LPPOINT lpPoint )
    {
        HDC hDC;
        BOOL bRet;
        HFONT hOldFont;
        TEXTMETRIC textMetric;
        NONCLIENTMETRICS nonClientMetrics;
        T* pT;

        if ( !m_nonClientFont )
        {
            memset( &nonClientMetrics, 0, sizeof( NONCLIENTMETRICS ) );
            nonClientMetrics.cbSize = sizeof( NONCLIENTMETRICS );
            if ( !::SystemParametersInfo( SPI_GETNONCLIENTMETRICS, sizeof( NONCLIENTMETRICS ), &nonClientMetrics, 0 ) || \
                    !( m_nonClientFont = ::CreateFontIndirect( &nonClientMetrics.lfMenuFont ) ) )
            {
                return FALSE;
            }
        }

        pT = static_cast<T*>( this );

        hDC = ::GetDC( pT->m_hWnd );
        hOldFont = ( HFONT )::SelectObject( hDC, m_nonClientFont );
        memset( &textMetric, 0, sizeof( TEXTMETRIC ) );
        bRet = ::GetTextMetrics( hDC, &textMetric );
        ::SelectObject( hDC, hOldFont );
        ::ReleaseDC( pT->m_hWnd, hDC );

        if ( !bRet )
        {
            return FALSE;
        }

        lpPoint->x = textMetric.tmAveCharWidth;
        lpPoint->y = textMetric.tmHeight + textMetric.tmExternalLeading;

        return TRUE;
    }

    void MoveToolBarComboBox( HWND hWndToolBar, HWND hWndComboBox )
    {
        int nDiff;
        RECT rtToolBar, rtComboBox;
        CToolBarCtrl tbCtrl;
        CComboBox cbxCtrl;

        tbCtrl = hWndToolBar;
        cbxCtrl = hWndComboBox;

        tbCtrl.GetClientRect( &rtToolBar );
        cbxCtrl.GetWindowRect( &rtComboBox );

        nDiff = ( rtToolBar.bottom - rtToolBar.top ) - ( rtComboBox.bottom - rtComboBox.top );

        if ( nDiff > 1 )
        {
            tbCtrl.ScreenToClient( &rtComboBox );
            cbxCtrl.MoveWindow( rtComboBox.left, nDiff / 2, \
                                rtComboBox.right - rtComboBox.left, rtComboBox.bottom - rtComboBox.top );
        }
    }
};

 在原来基础上有一些小改动

 WTL Toolbar Helper