自绘Tab控件

自绘tab按钮效果图如下：

 

使用例子：

MyTabControl *tabControl = NULL;

tabControl = new MyTabControl();
tabControl->Create(this, CRect(0,0,125,27), L"花生", ID_BTN_MYTAB);
tabControl->SetTabIndex(index);

tabControl->SetShowText(L"花生");

tabControl->SetTabState(TAB_STATE_DOWN);

tabControl->SetToolTipText(L"花生");

头文件：

#pragma once
#include "LhsButton.h"

// MyTabControl

#define ID_BTN_MYTAB_CLOSE  2100            //关闭按钮的id
#define MYWM_BTN_TAB_CLOSE WM_USER+2001     //关闭按钮单击响应
#define MYWM_BTN_TAB_CLICK WM_USER+2002     //tab按钮单击

//tab按钮的状态
enum TabState
{
    TAB_STATE_NOR           = 0,
    TAB_STATE_DOWN          = 1,
};
class MyTabControl : public CWnd
{
    DECLARE_DYNAMIC(MyTabControl)

public:
    MyTabControl();
    virtual ~MyTabControl();

     bool Create(CWnd* pParent,CRect rc,CString text,DWORD id = 0,DWORD style = WS_VISIBLE|WS_CHILD);

    DECLARE_MESSAGE_MAP()

public:
    void SetShowText(CString strShowText);

protected:
    CString szClassName;
    bool m_isMouseHover;        //鼠标是否悬浮
    bool m_isMouseClicked;      //鼠标是否单击
    CString m_strShowText;      //要显示的文字
    CString m_strTabText;       //绘制在tab按钮下的文字

    Image*                      m_pImgNor;          //正常时的图片
    Image*                      m_pImgHot;          //鼠标悬浮时的图片
    Image*                      m_pImgDown;         //单击按下时的图片

    void PostClickEvent();
    afx_msg int OnCreate(LPCREATESTRUCT lpCreateStruct);
    afx_msg void OnMouseMove(UINT nFlags, CPoint point);
    afx_msg void OnMouseHover(UINT nFlags, CPoint point);
    afx_msg void OnMouseLeave();
    afx_msg void OnLButtonDown(UINT nFlags, CPoint point);
    afx_msg void OnLButtonUp(UINT nFlags, CPoint point);
    afx_msg BOOL OnEraseBkgnd(CDC* pDC);
    afx_msg void OnPaint();
    afx_msg void OnSize(UINT nType, int cx, int cy);
    afx_msg LRESULT OnBtnClose(WPARAM wParam, LPARAM lParam);
    virtual BOOL PreTranslateMessage(MSG* pMsg);
public:
    void SetTabState(TabState state){m_tabState = state; Invalidate();} //设置tab状态
    TabState GetTabState(){return m_tabState;}
    void SetToolTipText(CString spText, BOOL bActivate = TRUE);
    void SetTabIndex(int idx){m_nIndex = idx;}
    int GetTabIndex(){return m_nIndex;}
private:
    DWORD GetComfortSize(HDC hdc, DWORD dwWidth, CString &strText);
private:
    TabState m_tabState;                            //tab的状态
    CLhsButton m_btnClose;                          //关闭按钮
    CToolTipCtrl*   m_pToolTip;
    CString m_tooltext;
    bool m_bCenterAlign;                            //是否居中对齐
    int m_nIndex;                                   //tab按钮的索引

public:
    afx_msg HBRUSH OnCtlColor(CDC* pDC, CWnd* pWnd, UINT nCtlColor);
};

// MyTabControl.cpp : 实现文件
//

#include "stdafx.h"
#include "../Lander_mini.h"
#include "MyTabControl.h"
#include "memdc.h"
#include "../Utility.h"
// MyTabControl

IMPLEMENT_DYNAMIC(MyTabControl, CWnd)

MyTabControl::MyTabControl()
{
    m_isMouseHover = false;
    m_isMouseClicked = false;
    // 注册控件类
    szClassName = AfxRegisterWndClass(0);
    m_pImgNor = NULL;
    m_pImgHot = NULL;
    m_pImgDown = NULL;
    m_tabState = TAB_STATE_NOR;
    m_pToolTip = NULL;
    m_bCenterAlign = false;
    m_nIndex = 0;
}

MyTabControl::~MyTabControl()
{
    SAFE_RELEASE(m_pToolTip);
    SAFE_RELEASE(m_pImgNor);
    SAFE_RELEASE(m_pImgHot);
    SAFE_RELEASE(m_pImgDown);
}


BEGIN_MESSAGE_MAP(MyTabControl, CWnd)
    ON_WM_MOUSEMOVE()
    ON_WM_MOUSEHOVER()  // 此消息系统并不会给我们发送
    ON_WM_MOUSELEAVE()
    ON_WM_LBUTTONDOWN()
    ON_WM_LBUTTONUP()
    ON_WM_PAINT()
    ON_WM_SIZE()
    ON_WM_ERASEBKGND()
    ON_WM_CREATE()
    ON_MESSAGE(MYWM_BTN_CLICK, &MyTabControl::OnBtnClose)
    ON_WM_CTLCOLOR()
END_MESSAGE_MAP()

// MyTabControl 消息处理程序
bool MyTabControl::Create(CWnd* pParent,CRect rc,CString text,DWORD id /* = 0 */,DWORD style /* = WS_VISIBLE|WS_CHILD */)
{
    // 动态创建控件
    BOOL ret = CWnd::CreateEx(0, szClassName, text, style, rc, pParent, id);
    return ret ? true : false;
}

int MyTabControl::OnCreate(LPCREATESTRUCT lpCreateStruct)
{

    //::SetWindowLong(m_hWnd, GWL_EXSTYLE, GetWindowLong(m_hWnd, GWL_EXSTYLE) ^ WS_EX_LAYERED);
    m_pImgNor = CQYUtility::LoadImgFromRes(_T("PNG"), IDB_PNG_TAB_NOR);
    m_pImgHot = CQYUtility::LoadImgFromRes(_T("PNG"), IDB_PNG_TAB_HOT);
    m_pImgDown = CQYUtility::LoadImgFromRes(_T("PNG"), IDB_PNG_TAB_DOWN);

    RECT rc = {0, 0, 20, 20};
    m_btnClose.LoadBtnImg(_T("PNG"),IDB_PNG_TAB_CLOSE_NOR, IDB_PNG_TAB_CLOSE_HOT,IDB_PNG_TAB_CLOSE_HOT);
    m_btnClose.Create(this, rc, L"",ID_BTN_MYTAB_CLOSE);
    m_btnClose.SetToolTipText(_T("关闭"));
    EnableToolTips(TRUE);
    return 0;
}

void MyTabControl::PostClickEvent()
{
    if (m_tabState == TAB_STATE_DOWN)
        return;     //选中的tab不响应单击了
    // 该函数用来向父窗口发送 单击 消息
    CWnd* parent = GetParent();
    if(parent)
        parent->SendMessage(MYWM_BTN_TAB_CLICK,m_nIndex,0);
}

void MyTabControl::OnMouseHover(UINT nFlags, CPoint point)
{
    // 鼠标进入
    Invalidate();
}

void MyTabControl::OnMouseMove(UINT nFlags, CPoint point)
{
    // 只处理鼠标第一次进入时的情况
    if(!m_isMouseHover)
    {
        m_isMouseHover = true;

        TRACKMOUSEEVENT evt = { sizeof(evt), TME_LEAVE|TME_HOVER, m_hWnd, 0 };
        TrackMouseEvent(&evt);

        OnMouseHover(0,CPoint());
    }
}

void MyTabControl::OnMouseLeave()
{
    // 鼠标离开
    m_isMouseHover = false;
    m_isMouseClicked = false;
    Invalidate();
}

void MyTabControl::OnLButtonDown(UINT nFlags, CPoint point)
{
    // 鼠标按下
    m_isMouseClicked = true;
    Invalidate();
}

void MyTabControl::OnLButtonUp(UINT nFlags, CPoint point)
{
    // 鼠标松开
    if(m_isMouseClicked)
    {
        m_isMouseClicked = false;
        Invalidate();
        PostClickEvent();
    }
}

BOOL MyTabControl::OnEraseBkgnd(CDC* pDC)
{
//  return CWnd::OnEraseBkgnd(pDC);
    return TRUE;    // 阻止擦除背景，防止闪烁
}

void MyTabControl::OnPaint()
{
    CPaintDC dc(this);
    CRect rc;
    GetClientRect(&rc);

    // 采用双缓存，防止闪烁
    CMemDC memdc(&dc,&rc,TRUE);
    Graphics graphic(memdc);
    if (!m_pImgNor || !m_pImgHot || !m_pImgDown)
    {//没有提供按钮图片就刷下背景
        // 刷背景
        COLORREF bkgnd = RGB(100,0,0);
        if(m_isMouseHover)
        {
            if(m_isMouseClicked)
                bkgnd = RGB(250,0,0);
            else
                bkgnd = RGB(180,0,0);
        }
        memdc.FillSolidRect(&rc,bkgnd);
    }

    if (m_isMouseClicked || m_tabState == TAB_STATE_DOWN)
    {//单击一定画单击状态
        graphic.DrawImage(m_pImgDown, 0, 0, m_pImgDown->GetWidth(), m_pImgDown->GetHeight());
    }
    else if (m_isMouseHover && !m_isMouseClicked)
    {
        //悬浮,但是没单击
        graphic.DrawImage(m_pImgHot, 0, 0, m_pImgHot->GetWidth(), m_pImgHot->GetHeight());
    }
    else
    {
        graphic.DrawImage(m_pImgNor, 0, 0, m_pImgNor->GetWidth(), m_pImgNor->GetHeight());
    }

    if (!m_strTabText.IsEmpty())
    {
        // 设置文字字体
        CFont font;
        font.CreatePointFont(100,L"宋体"); // 11号字体，该参数与实际字体号有10倍的关系
        CFont* poldFont = memdc.SelectObject(&font);
        // 设置文字属性
        memdc.SetBkMode(TRANSPARENT);
        if (m_tabState == TAB_STATE_DOWN)
            memdc.SetTextColor(RGB(136,76,25));
        else
            memdc.SetTextColor(RGB(255,245,190));
        // 绘制文本
        DWORD style = DT_SINGLELINE | DT_VCENTER ;   // 文本格式：单行+水平居中+垂直居中
        //if (m_bCenterAlign)
        //  style |= DT_CENTER;
        CRect fontRect(rc);
        fontRect.left += 13;
        fontRect.right -= 22;
        memdc.DrawText(m_strTabText, -1, &fontRect, style);    // 更多文本显示格式可参考百度百科DrawText说明
        memdc->SelectObject(poldFont);
    }

    // 使绘制生效
    graphic.ReleaseHDC(memdc);
}


BOOL MyTabControl::PreTranslateMessage(MSG* pMsg)
{
    if (m_pToolTip)
    {
        if (::IsWindow(m_pToolTip->m_hWnd))
        {
            m_pToolTip->RelayEvent(pMsg);
        }
    }

    return CWnd::PreTranslateMessage(pMsg);
}


void MyTabControl::OnSize(UINT nType, int cx, int cy)
{
    CRect rect;
    GetClientRect(&rect);
    m_btnClose.SetWindowPos(NULL, rect.right - 22, rect.top + 2, 20, 20, SWP_NOZORDER);
}

LRESULT MyTabControl::OnBtnClose(WPARAM wParam, LPARAM lParam)
{
    // 该函数用来向父窗口发送 单击 消息
    CWnd* parent = GetParent();
    if(parent != NULL)
    {
        parent->SendMessage(MYWM_BTN_TAB_CLOSE, m_nIndex, 0);
    }
    return 0;
}

void MyTabControl::SetToolTipText(CString spText, BOOL bActivate)
{
    if (m_pToolTip == NULL)
    {
        m_pToolTip = new CToolTipCtrl;
        // Create ToolTip control
        m_pToolTip->Create(this);
        m_pToolTip->Activate(TRUE);
    }

    m_tooltext = spText;

    // If there is no tooltip defined then add it
    if (m_pToolTip->GetToolCount() == 0)
    {
        CRect rectBtn;
        GetClientRect(rectBtn);
        m_pToolTip->AddTool(this, m_tooltext, rectBtn, 1);
    }

    // Set text for tooltip
    m_pToolTip->UpdateTipText(m_tooltext, this, 1);
    m_pToolTip->SetDelayTime(2000);
    m_pToolTip->Activate(bActivate);
}


DWORD MyTabControl::GetComfortSize(HDC hdc,DWORD dwWidth,CString &strText)
{
    //二分法查找
    DWORD dwComfortSize = 0;
    DWORD dwBeginSize = 0;
    DWORD dwEndSize = strText.GetLength();

    while(TRUE)
    {
        DWORD dwMiddleSize = (dwEndSize + dwBeginSize) / 2;
        if(dwMiddleSize == dwBeginSize || dwMiddleSize == dwEndSize)
        {
            //两个点之间已经没有数值可以检测，退出循环
            dwComfortSize = dwBeginSize;
            break;
        }

        SIZE sizeChk = {0};
        ::GetTextExtentPoint(hdc, strText.GetBuffer(), dwMiddleSize, &sizeChk);

        if(sizeChk.cx == dwWidth)
        {
            //数值刚好合适，跳出循环
            dwComfortSize = dwMiddleSize;
            break;
        }
        else if(static_cast<DWORD>(sizeChk.cx) > dwWidth)
        {
            //重新设置边界
            dwEndSize = dwMiddleSize;
        }
        else
        {
            //重新设置边界
            dwBeginSize = dwMiddleSize;
        }

    }

    return dwComfortSize;
}

void MyTabControl::SetShowText(CString strShowText)
{
    m_strShowText = strShowText;

    CRect rect;
    GetClientRect(&rect);
    SIZE sizeEllipsis = {0};
    ::GetTextExtentPoint(GetDC()->GetSafeHdc(), strShowText.GetBuffer(), lstrlen(strShowText.GetBuffer()), &sizeEllipsis);
    if (sizeEllipsis.cx  < rect.Width() - 40)
    {//显示区域够宽
        m_bCenterAlign = true;
        m_strTabText = strShowText;
        return;
    }
    memset(&sizeEllipsis,0, sizeof(sizeEllipsis));
    CString ellipsStr = _T("...");
    ::GetTextExtentPoint(GetDC()->GetSafeHdc(), ellipsStr.GetBuffer(), lstrlen(ellipsStr.GetBuffer()), &sizeEllipsis);
    DWORD dwMaxDisp = GetComfortSize(GetDC()->GetSafeHdc(), rect.Width() - sizeEllipsis.cx - 20, strShowText);
    if (dwMaxDisp >= strShowText.GetLength())
    {
        m_strTabText = strShowText;
    }
    else
    {
        m_strTabText = m_strShowText.Left(dwMaxDisp);
        m_strTabText += ellipsStr;
    }

}



HBRUSH MyTabControl::OnCtlColor(CDC* pDC, CWnd* pWnd, UINT nCtlColor)
{
    HBRUSH hbr = CWnd::OnCtlColor(pDC, pWnd, nCtlColor);

    // TODO:  在此更改 DC 的任何属性
//  m_btnClose.SetBkGnd(pDC);
    // TODO:  如果默认的不是所需画笔，则返回另一个画笔
    return hbr;
}

//内存DC类和按钮类见链接：点击打开链接

http://blog.csdn.net/huasonl88/article/details/43227291