MFC自定义button实现颜色控制

#pragma once
#include "stdafx.h"
// CustomButton dialog
class CCustomButton : public CButton
{
  DECLARE_DYNAMIC(CCustomButton)
public:
  CCustomButton();
  virtual ~CCustomButton();
  void SetButtonBgColor(COLORREF color);
  void SetButtonTextColor(COLORREF color);
private:
  COLORREF m_bgColor;
  COLORREF m_textColor;
  BOOL m_bPressed;
  BOOL m_bTrackingMouse;
protected:
  DECLARE_MESSAGE_MAP()
  afx_msg void DrawItem(LPDRAWITEMSTRUCT lpDrawItemStruct);
  afx_msg void OnLButtonDown(UINT nFlags, CPoint point);
  afx_msg void OnLButtonUp(UINT nFlags, CPoint point);
  afx_msg void OnMouseMove(UINT nFlags, CPoint point);
  afx_msg void OnMouseHover(UINT nFlags, CPoint point);
  afx_msg void OnMouseLeave();
};

// CustomButton.cpp : implementation file
//
#include "stdafx.h"
#include "CustomButton.h"
#include "afxdialogex.h"
// CustomButton dialog
IMPLEMENT_DYNAMIC(CCustomButton, CButton)
CCustomButton::CCustomButton()
{
  //m_bgColor = RGB(52, 52, 52);
  //m_textColor = RGB(153, 153, 153);
  //m_bgColor = RGB(0, 255, 0);
  //m_textColor = RGB(255, 0, 0);
  m_bPressed = FALSE;
  m_bTrackingMouse = true;
}
CCustomButton::~CCustomButton()
{
}
BEGIN_MESSAGE_MAP(CCustomButton, CButton)
  ON_WM_DRAWITEM()
  ON_WM_MOUSEHOVER()
  ON_WM_MOUSELEAVE()
  ON_WM_LBUTTONDOWN()
  ON_WM_LBUTTONUP()
  ON_WM_MOUSEMOVE()
END_MESSAGE_MAP()
// CCustomButton 消息处理程序
void CCustomButton::DrawItem(LPDRAWITEMSTRUCT lpDrawItemStruct)
{
  CRect rect;
  GetClientRect(rect);
  CDC dc;
  dc.Attach(lpDrawItemStruct->hDC);
  //  UINT state = lpDrawItemStruct->itemState;
    //CRect focusRect(rect);
    //focusRect.DeflateRect(4, 4, 4, 4);
  if (m_bTrackingMouse)
  {
    m_bgColor = RGB(0, 255, 0);
    m_textColor = RGB(255, 0, 0);
    CPen pen(PS_DOT, 1, m_bgColor);
    CBrush brush;
    brush.CreateStockObject(NULL_BRUSH);
    dc.SelectObject(&brush);
    dc.SelectObject(&pen);
    dc.FillSolidRect(rect, m_bgColor);
    dc.Rectangle(rect);
    CString strText;
    GetWindowText(strText);
    //m_bgColor = RGB(52, 52, 52);
    //m_textColor = RGB(153, 153, 153);
    dc.SetBkColor(m_bgColor);
    dc.SetTextColor(m_textColor);
    dc.DrawText(strText, rect, DT_CENTER | DT_SINGLELINE | DT_VCENTER);
  }
  else
  {
    m_bgColor = RGB(255, 0, 0);
    m_textColor = RGB(0, 255, 0);
    CPen pen(PS_DOT, 1, m_bgColor);
    CBrush brush;
    brush.CreateStockObject(NULL_BRUSH);
    dc.SelectObject(&brush);
    dc.SelectObject(&pen);
    dc.FillSolidRect(rect, m_bgColor);
    dc.Rectangle(rect);
    CString strText;
    GetWindowText(strText);
    //m_bgColor = RGB(40, 53, 69);
    //m_textColor = RGB(46, 189, 255);
    dc.SetBkColor(m_bgColor);
    dc.SetTextColor(m_textColor);
    dc.DrawText(strText, rect, DT_CENTER | DT_SINGLELINE | DT_VCENTER);
    //dc.DrawText(strText, rect);
    //CBrush brush(RGB(255, 0, 0));
    //dc.FillRect(&(lpDrawItemStruct->rcItem), &brush);
    //DrawText(lpDrawItemStruct->hDC, strText, strText.GetLength(),
    //  &lpDrawItemStruct->rcItem, DT_SINGLELINE | DT_VCENTER | DT_CENTER);
//    SetBkMode(lpDrawItemStruct->hDC, TRANSPARENT);
  }
  // button字体
  //CString strText;
  //GetWindowText(strText);
  //dc.SetBkMode(TRANSPARENT);
  //dc.SetTextColor(m_textColor);
  ///*if () {
  //  dc.SetTextColor(RGB(172, 168, 153));
  //}*/
  //dc.DrawText(strText, rect, DT_CENTER | DT_SINGLELINE | DT_VCENTER);
  dc.Detach();
}
//按钮被按下
void CCustomButton::OnLButtonDown(UINT nFlags, CPoint point)
{
  m_bPressed = TRUE;
  CButton::OnLButtonDown(nFlags, point);
}
//按钮被释放
void CCustomButton::OnLButtonUp(UINT nFlags, CPoint point)
{
  m_bPressed = FALSE;
  CButton::OnLButtonDown(nFlags, point);
}
//设置按钮背景的颜色 
void CCustomButton::SetButtonBgColor(COLORREF color)
{
  m_bgColor = color;
}
//设置按钮字体的颜色
void CCustomButton::SetButtonTextColor(COLORREF color)
{
  m_textColor = color;
}
void CCustomButton::OnMouseMove(UINT nFlags, CPoint point)
{
  if (m_bTrackingMouse)
  {
    TRACKMOUSEEVENT tme;
    tme.cbSize = sizeof(TRACKMOUSEEVENT);
    tme.dwFlags = TME_LEAVE | TME_HOVER;
    tme.hwndTrack = GetSafeHwnd();
    tme.dwHoverTime = 10;     // 影响OnMouseHover函数的响应时间，设置小点。
    if (::TrackMouseEvent(&tme))
    {
      m_bTrackingMouse = false;
    }
  }
  CButton::OnMouseMove(nFlags, point);
}
void CCustomButton::OnMouseHover(UINT nFlags, CPoint point)
{
  //HDC hdc;
  //hdc = ::GetDC(this->GetParent()->GetSafeHwnd());
  //HPEN hpen = ::CreatePen(PS_SOLID, 1, RGB(255, 0, 0));//创建画笔
  //HPEN oHpen;
  //oHpen = (HPEN)::SelectObject(hdc, hpen);//把新的画笔填充到DC
  Invalidate();
  //CButton* btn = (CButton*)this->GetParent()->GetDlgItem(IDC_BUTTON1)->SetFaceColor(RGB(0, 255, 0));
  m_bTrackingMouse = false;
  CButton::OnMouseHover(nFlags, point);
}
void CCustomButton::OnMouseLeave()
{
  /*CRect rt;
  GetClientRect(&rt);
  InvalidateRect(rt);
  */
  Invalidate();
  m_bTrackingMouse = true;
  CButton::OnMouseLeave();
}