重载的CzButton类

zButton.h

#ifndef _CZBUTTON_H_
#define _CZBUTTON_H_
///////////////////////////////////////////////////////////////////////////// 
// CzButton window 
 
class CzButton : public CButton 
{ 
// Construction 
public: 
CzButton(); 
 
// Attributes 
public: 
 
// Operations 
public: 
 
// Overrides 
// ClassWizard generated virtual function overrides 
//{{AFX_VIRTUAL(CzButton) 
public: 
virtual void DrawItem(LPDRAWITEMSTRUCT lpDrawItemStruct); 
protected: 
virtual void PreSubclassWindow(); 
//}}AFX_VIRTUAL 
 
// Implementation 
public: 
virtual ~CzButton(); 
 
// Generated message map functions 
protected: 
//{{AFX_MSG(CzButton) 
//}}AFX_MSG 
DECLARE_MESSAGE_MAP() 
protected: 
    bool m_button_down; 
short m_FocusRectMargin; //dotted margin offset 
CBitmap m_bNormal,m_bDown,m_bDisabled,m_bMask,m_bOver,m_bFocus; //skin bitmaps 
short m_DrawMode; //0=normal; 1=stretch; 2=tiled; 
BOOL m_Border; //0=flat; 1=3D; 
bool m_Checked; //radio & check buttons 
    bool m_tracking; 
COLORREF m_TextColor; //button text color 
void FillWithBitmap(CDC* dc, HBITMAP hbmp, RECT r); 
void DrawBitmap(CDC* dc, HBITMAP hbmp, RECT r, int DrawMode); 
int GetBitmapWidth (HBITMAP hBitmap); 
int GetBitmapHeight (HBITMAP hBitmap); 
HRGN hClipRgn; //clipping region 
HRGN CreateRgnFromBitmap(HBITMAP hBmp, COLORREF color); 
 
 
 
 
public: 
void SetPosition(CRect RC, int pos); 
void SetSkin(UINT normal,UINT down, UINT over=0, UINT disabled=0, UINT focus=0,UINT mask=0, 
short drawmode=1,short border=1,short margin=4); 
COLORREF SetTextColor(COLORREF new_color); 
 
  
 
afx_msg void OnDestroy();
}; 
 
 
#endif //


// zButton.cpp : implementation file 
// 

#include "stdafx.h" 
#include "zButton.h" 
 

 
CzButton::CzButton() 
{ 
m_DrawMode=1; // normal drawing mode 
m_FocusRectMargin=0; // disable focus dotted rect 
hClipRgn=NULL; // no clipping region 
m_TextColor=GetSysColor(COLOR_BTNTEXT); // default button text color 
m_button_down = m_tracking = m_Checked = false; 
} 
 
CzButton::~CzButton() 
{ 
if (hClipRgn) DeleteObject(hClipRgn); // free clip region 
} 
 
 
BEGIN_MESSAGE_MAP(CzButton, CButton) 
//{{AFX_MSG_MAP(CzButton) 
// ON_WM_LBUTTONDOWN() 
// ON_WM_LBUTTONUP() 
// ON_WM_ERASEBKGND() 
//}}AFX_MSG_MAP 
ON_WM_DESTROY()
END_MESSAGE_MAP() 
 
///////////////////////////////////////////////////////////////////////////// 
// CzButton message handlers 
 
void CzButton::PreSubclassWindow()  
{ 
// TODO: Add your specialized code here and/or call the base class 
ModifyStyle( 0, BS_OWNERDRAW );  
CButton::PreSubclassWindow(); 
} 
 
void CzButton::DrawItem(LPDRAWITEMSTRUCT lpDrawItemStruct)  
{ 
// TODO: Add your code to draw the specified item 
    ASSERT (lpDrawItemStruct); 
    //TRACE("* Captured: %08X\n", ::GetCapture()); 
 
    //Check if the button state in not in inconsistent mode... 
    POINT mouse_position; 
    if ((m_button_down) && (::GetCapture() == m_hWnd) && (::GetCursorPos(&mouse_position))){ 
if (::WindowFromPoint(mouse_position) == m_hWnd){ 
if ((GetState() & BST_PUSHED) != BST_PUSHED) { 
//TRACE("* Inconsistency up detected! Fixing.\n"); 
SetState(TRUE); 
return; 
} 
} else { 
if ((GetState() & BST_PUSHED) == BST_PUSHED) { 
//TRACE("* Inconsistency up detected! Fixing.\n"); 
SetState(FALSE); 
return; 
} 
} 
} 
 
    //TRACE("* Drawing: %08x\n", lpDrawItemStruct->itemState); 
CString sCaption; 
CDC *pDC = CDC::FromHandle(lpDrawItemStruct->hDC); // get device context 
RECT r=lpDrawItemStruct->rcItem; // context rectangle 
int cx = r.right  - r.left ; // get width 
int cy = r.bottom - r.top  ; // get height 
// get text box position 
RECT tr={r.left+m_FocusRectMargin+2,r.top,r.right-m_FocusRectMargin-2,r.bottom}; 
 
GetWindowText(sCaption); // get button text 
pDC->SetBkMode(TRANSPARENT); 
 
// Select the correct skin  
if (lpDrawItemStruct->itemState & ODS_DISABLED){ // DISABLED BUTTON 
if(m_bDisabled.m_hObject==NULL) 
 // no skin selected for disabled state -> standard button 
pDC->FillSolidRect(&r,GetSysColor(COLOR_BTNFACE)); 
else // paint the skin 
DrawBitmap(pDC,(HBITMAP)m_bDisabled,r,m_DrawMode); 
// if needed, draw the standard 3D rectangular border 
if (m_Border) pDC->DrawEdge(&r,EDGE_RAISED,BF_RECT); 
// paint the etched button text 
pDC->SetTextColor(GetSysColor(COLOR_3DHILIGHT)); 
pDC->DrawText(sCaption,&tr,DT_SINGLELINE|DT_VCENTER|DT_CENTER); 
pDC->SetTextColor(GetSysColor(COLOR_GRAYTEXT)); 
OffsetRect(&tr,-1,-1); 
pDC->DrawText(sCaption,&tr,DT_SINGLELINE|DT_VCENTER|DT_CENTER); 
} else { // SELECTED (DOWN) BUTTON 
if ((lpDrawItemStruct->itemState & ODS_SELECTED)||m_Checked){ 
if(m_bDown.m_hObject==NULL) 
 // no skin selected for selected state -> standard button 
pDC->FillSolidRect(&r,GetSysColor(COLOR_BTNFACE)); 
            else { // paint the skin 
DrawBitmap(pDC,(HBITMAP)m_bDown,r,m_DrawMode); 
            } 
OffsetRect(&tr,1,1);  //shift text 
// if needed, draw the standard 3D rectangular border 
if (m_Border) pDC->DrawEdge(&r,EDGE_SUNKEN,BF_RECT); 
} else { // DEFAULT BUTTON 
if(m_bNormal.m_hObject==NULL) 
 // no skin selected for normal state -> standard button 
pDC->FillSolidRect(&r,GetSysColor(COLOR_BTNFACE)); 
else // paint the skin 
                if ((m_tracking)&&(m_bOver.m_hObject!=NULL)){ 
DrawBitmap(pDC,(HBITMAP)m_bOver,r,m_DrawMode); 
} else { 
if ((lpDrawItemStruct->itemState & ODS_FOCUS)&&(m_bFocus.m_hObject!=NULL)){ 
DrawBitmap(pDC,(HBITMAP)m_bFocus,r,m_DrawMode); 
} else { 
DrawBitmap(pDC,(HBITMAP)m_bNormal,r,m_DrawMode); 
} 
} 
// if needed, draw the standard 3D rectangular border 
if (m_Border) pDC->DrawEdge(&r,EDGE_RAISED,BF_RECT); 
} 
// paint the focus rect 
if ((lpDrawItemStruct->itemState & ODS_FOCUS)&&(m_FocusRectMargin>0)){ 
r.left   += m_FocusRectMargin ; 
r.top    += m_FocusRectMargin ; 
r.right  -= m_FocusRectMargin ; 
r.bottom -= m_FocusRectMargin ; 
DrawFocusRect (lpDrawItemStruct->hDC, &r) ; 
} 
// paint the enabled button text 
pDC->SetTextColor(m_TextColor); 
pDC->DrawText(sCaption,&tr,DT_SINGLELINE|DT_VCENTER|DT_CENTER); 
} 
} 
 
void CzButton::DrawBitmap(CDC *dc, HBITMAP hbmp, RECT r, int DrawMode) 
{ 
// DrawMode: 0=Normal; 1=stretch; 2=tiled fill 
if(DrawMode==2){ 
FillWithBitmap(dc,hbmp,r); 
return; 
} 
if(!hbmp) return; //safe check 
 
int cx=r.right  - r.left; 
int cy=r.bottom - r.top; 
CDC dcBmp,dcMask; 
dcBmp.CreateCompatibleDC(dc); 
dcBmp.SelectObject(hbmp); 
 
if (m_bMask.m_hObject!=NULL){ 
dcMask.CreateCompatibleDC(dc); 
dcMask.SelectObject(m_bMask); 
 
CDC hdcMem; 
hdcMem.CreateCompatibleDC(dc); 
    CBitmap hBitmap; 
hBitmap.CreateCompatibleBitmap(dc,cx,cy); 
    hdcMem.SelectObject(hBitmap); 
 
        hdcMem.BitBlt(r.left,r.top,cx,cy,dc,0,0,SRCCOPY); 
if(!DrawMode){ 
hdcMem.BitBlt(r.left,r.top,cx,cy,&dcBmp,0,0,SRCINVERT); 
hdcMem.BitBlt(r.left,r.top,cx,cy,&dcMask,0,0,SRCAND); 
hdcMem.BitBlt(r.left,r.top,cx,cy,&dcBmp,0,0,SRCINVERT); 
} else { 
int bx=GetBitmapWidth(hbmp); 
int by=GetBitmapHeight(hbmp); 
hdcMem.StretchBlt(r.left,r.top,cx,cy,&dcBmp,0,0,bx,by,SRCINVERT); 
hdcMem.StretchBlt(r.left,r.top,cx,cy,&dcMask,0,0,bx,by,SRCAND); 
hdcMem.StretchBlt(r.left,r.top,cx,cy,&dcBmp,0,0,bx,by,SRCINVERT); 
} 
        dc->BitBlt(r.left,r.top,cx,cy,&hdcMem,0,0,SRCCOPY); 
 
hdcMem.DeleteDC(); 
hBitmap.DeleteObject(); 
 
DeleteDC(dcMask); 
} else { 
if(!DrawMode){ 
dc->BitBlt(r.left,r.top,cx,cy,&dcBmp,0,0,SRCCOPY); 
} else { 
int bx=GetBitmapWidth(hbmp); 
int by=GetBitmapHeight(hbmp); 
dc->StretchBlt(r.left,r.top,cx,cy,&dcBmp,0,0,bx,by,SRCCOPY); 
} 
} 
DeleteDC(dcBmp); 
 
} 
 
void CzButton::FillWithBitmap(CDC *dc, HBITMAP hbmp, RECT r) 
{ 
if(!hbmp) return; 
CDC memdc; 
memdc.CreateCompatibleDC(dc); 
memdc.SelectObject(hbmp); 
int w = r.right - r.left; 
int h = r.bottom - r.top; 
int x,y,z; 
int bx=GetBitmapWidth(hbmp); 
int by=GetBitmapHeight(hbmp); 
for (y = r.top ; y < h ; y += by){ 
if ((y+by)>h) by=h-y; 
z=bx; 
for (x = r.left ; x < w ; x += z){ 
if ((x+z)>w) z=w-x; 
dc->BitBlt(x, y, z, by, &memdc, 0, 0, SRCCOPY); 
} 
} 
DeleteDC(memdc); 
 
}