textprogressctrl.cpp

VisualC高级编程技术精粹.rar

// TextProgressCtrl.cpp : implementation file
//

#include "stdafx.h"
#include "TextProgressCtrl.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif


#ifndef _MEMDC_H_

class CMemDC : public CDC
{
public:

    // constructor sets up the memory DC
    CMemDC(CDC* pDC) : CDC()
    {
        ASSERT(pDC != NULL);

        m_pDC = pDC;
        m_pOldBitmap = NULL;
        m_bMemDC = !pDC->IsPrinting();
              
        if (m_bMemDC)    // Create a Memory DC
        {
            pDC->GetClipBox(&m_rect);
            CreateCompatibleDC(pDC);
            m_bitmap.CreateCompatibleBitmap(pDC, m_rect.Width(), m_rect.Height());
            m_pOldBitmap = SelectObject(&m_bitmap);
            SetWindowOrg(m_rect.left, m_rect.top);
        }
        else        // Make a copy of the relevent parts of the current DC for printing
        {
            m_bPrinting = pDC->m_bPrinting;
            m_hDC       = pDC->m_hDC;
            m_hAttribDC = pDC->m_hAttribDC;
        }
    }
    
    // Destructor copies the contents of the mem DC to the original DC
    ~CMemDC()
    {
        if (m_bMemDC) 
        {    
            // Copy the offscreen bitmap onto the screen.
            m_pDC->BitBlt(m_rect.left, m_rect.top, m_rect.Width(), m_rect.Height(),
                          this, m_rect.left, m_rect.top, SRCCOPY);

            //Swap back the original bitmap.
            SelectObject(m_pOldBitmap);
        } else {
            // All we need to do is replace the DC with an illegal value,
            // this keeps us from accidently deleting the handles associated with
            // the CDC that was passed to the constructor.
            m_hDC = m_hAttribDC = NULL;
        }
    }

    // Allow usage as a pointer
    CMemDC* operator->() {return this;}
        
    // Allow usage as a pointer
    operator CMemDC*() {return this;}

private:
    CBitmap  m_bitmap;      // Offscreen bitmap
    CBitmap* m_pOldBitmap;  // bitmap originally found in CMemDC
    CDC*     m_pDC;         // Saves CDC passed in constructor
    CRect    m_rect;        // Rectangle of drawing area.
    BOOL     m_bMemDC;      // TRUE if CDC really is a Memory DC.
};

#endif



/////////////////////////////////////////////////////////////////////////////
// CTextProgressCtrl

CTextProgressCtrl::CTextProgressCtrl()
{
    m_nPos      = 0;
    m_nStepSize = 1;
    m_nMax      = 100;
    m_nMin      = 0;
    m_bShowText = TRUE;
    m_strText.Empty();
    m_crBarClr  = CLR_DEFAULT;
    m_crBgClr   = CLR_DEFAULT;
    m_crTextClr = CLR_DEFAULT;

    m_nBarWidth = -1;
}

CTextProgressCtrl::~CTextProgressCtrl()
{
}

BEGIN_MESSAGE_MAP(CTextProgressCtrl, CProgressCtrl)
    //{{AFX_MSG_MAP(CTextProgressCtrl)
    ON_WM_ERASEBKGND()
    ON_WM_PAINT()
    ON_WM_SIZE()
    //}}AFX_MSG_MAP
    ON_MESSAGE(WM_SETTEXT, OnSetText)
    ON_MESSAGE(WM_GETTEXT, OnGetText)
    ON_MESSAGE(PBM_DELTAPOS, OnOffsetPos)
    ON_MESSAGE(PBM_GETPOS, OnGetPos)
    ON_MESSAGE(PBM_GETRANGE, OnGetRange)
    ON_MESSAGE(PBM_SETBARCOLOR, OnSetBarColor)
    ON_MESSAGE(PBM_SETBKCOLOR, OnSetBkColor)
    ON_MESSAGE(PBM_SETPOS, OnSetPos)
    ON_MESSAGE(PBM_SETRANGE, OnSetRange)
    ON_MESSAGE(PBM_SETRANGE32, OnSetRange32)
    ON_MESSAGE(PBM_SETSTEP, OnSetStep)
    ON_MESSAGE(PBM_STEPIT, OnStepIt)
    ON_MESSAGE(PBM_SETTEXTCOLOR, OnSetTextColor)
    ON_MESSAGE(PBM_SETSHOWTEXT, OnSetShowText)
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CTextProgressCtrl message handlers

LRESULT CTextProgressCtrl::OnSetText(UINT, LPCTSTR szText)
{
    LRESULT result = Default();

    if ( (!szText && m_strText.GetLength()) ||
         (szText && (m_strText != szText))   )
    {
        m_strText = szText;
        Invalidate();
    }

    return result;
}

LRESULT CTextProgressCtrl::OnGetText(UINT cchTextMax, LPTSTR szText)
{
    if (!_tcsncpy(szText, m_strText, cchTextMax))
        return 0;
    else 
        return min(cchTextMax, (UINT) m_strText.GetLength());
}

BOOL CTextProgressCtrl::OnEraseBkgnd(CDC* /*pDC*/) 
{    
     return TRUE;
}

void CTextProgressCtrl::OnSize(UINT nType, int cx, int cy) 
{
    CProgressCtrl::OnSize(nType, cx, cy);
    
    m_nBarWidth    = -1;   // Force update if SetPos called
}

void CTextProgressCtrl::OnPaint() 
{
    if (m_nMin >= m_nMax) 
        return;

    COLORREF crBarColour, crBgColour;

    crBarColour = (m_crBarClr == CLR_DEFAULT)? ::GetSysColor(COLOR_HIGHLIGHT) : m_crBarClr;
    crBgColour = (m_crBgClr == CLR_DEFAULT)? ::GetSysColor(COLOR_WINDOW) : m_crBgClr;

    CRect LeftRect, RightRect, ClientRect;
    GetClientRect(ClientRect);

    double Fraction = (double)(m_nPos - m_nMin) / ((double)(m_nMax - m_nMin));

    CPaintDC PaintDC(this); // device context for painting
    CMemDC dc(&PaintDC);
    //CPaintDC dc(this);    // device context for painting (if not double buffering)

    dc.SelectObject(GetParent()->GetFont()); // pja July 1, 2001

    // Draw Bar
    LeftRect = RightRect = ClientRect;

#ifdef PBS_VERTICAL
    DWORD dwStyle = GetStyle();
    if (dwStyle & PBS_VERTICAL)
    {
        LeftRect.top = LeftRect.bottom - (int)((LeftRect.bottom - LeftRect.top)*Fraction);
        RightRect.bottom = LeftRect.top;
    }
    else
#endif
    {
        LeftRect.right = LeftRect.left + (int)((LeftRect.right - LeftRect.left)*Fraction);
        RightRect.left = LeftRect.right;
    }
    dc.FillSolidRect(LeftRect, crBarColour);
    dc.FillSolidRect(RightRect, crBgColour);

    // Draw Text if not vertical
    if (m_bShowText
#ifdef PBS_VERTICAL
        && (dwStyle & PBS_VERTICAL) == 0
#endif
        )
    {
        CString str;
        if (m_strText.GetLength())
            str = m_strText;
        else
            str.Format("%d%%", (int)(Fraction*100.0));

        dc.SetBkMode(TRANSPARENT);

        DWORD dwTextStyle = DT_CENTER | DT_VCENTER | DT_SINGLELINE;

        // If we are drawing vertical, then create a new verticla font
        // based on the current font (only works with TrueType fonts)
        CFont font, *pOldFont = NULL;
#ifdef PBS_VERTICAL
        if (dwStyle & PBS_VERTICAL)
        {
            LOGFONT lf;
            GetFont()->GetLogFont(&lf);
            lf.lfEscapement = lf.lfOrientation = 900;
            font.CreateFontIndirect(&lf);
            pOldFont = (CFont*) dc.SelectObject(&font);

            dwTextStyle = DT_VCENTER|DT_CENTER|DT_SINGLELINE;
        }
#endif

        CRgn rgn;
        rgn.CreateRectRgn(LeftRect.left, LeftRect.top, LeftRect.right, LeftRect.bottom);
        dc.SelectClipRgn(&rgn);
        dc.SetTextColor(m_crTextClr == CLR_DEFAULT ? crBgColour : m_crTextClr);
        dc.DrawText(str, ClientRect, dwTextStyle);

        rgn.DeleteObject();
        rgn.CreateRectRgn(RightRect.left, RightRect.top, RightRect.right, RightRect.bottom);
        dc.SelectClipRgn(&rgn);
        dc.SetTextColor(m_crTextClr == CLR_DEFAULT ? crBarColour : m_crTextClr);
        dc.DrawText(str, ClientRect, dwTextStyle);

        if (pOldFont)
        {
            dc.SelectObject(pOldFont);
            font.DeleteObject();
        }
    }
}

/////////////////////////////////////////////////////////////////////////////
// CTextProgressCtrl operations

void CTextProgressCtrl::OnSetShowText(WPARAM bShow, LPARAM)
{
    SetShowText((BOOL)bShow);
}

void CTextProgressCtrl::SetShowText(BOOL bShow)
{ 
    if (::IsWindow(m_hWnd) && m_bShowText != bShow)
        Invalidate();

    m_bShowText = bShow;
}

void CTextProgressCtrl::OnSetRange(WPARAM, LPARAM range)
{
    OnSetRange32(LOWORD(range), HIWORD(range));
}

void CTextProgressCtrl::OnSetRange32(WPARAM Lower, LPARAM Upper)
{
    m_nMax = (int)Upper;
    m_nMin = (int)Lower;
}

LRESULT CTextProgressCtrl::OnGetRange(WPARAM Type, LPARAM range)
{
    BOOL bType = (BOOL)Type;
    PPBRANGE pRange = (PPBRANGE)range;
    if (pRange)
    {
        pRange->iHigh = m_nMax;
        pRange->iLow = m_nMin;
    }
    return (LRESULT)(bType ? m_nMin : m_nMax);
}

LRESULT CTextProgressCtrl::OnGetPos(WPARAM, LPARAM)
{
    return (LRESULT)m_nPos;
}

LRESULT CTextProgressCtrl::OnSetPos(WPARAM newPos, LPARAM) 
{    
    if (!::IsWindow(m_hWnd))
        return -1;

    int nPos = (int)newPos;

    int nOldPos = m_nPos;
    m_nPos = nPos;

    CRect rect;
    GetClientRect(rect);

    double Fraction = (double)(m_nPos - m_nMin) / ((double)(m_nMax - m_nMin));
    int nBarWidth = (int) (Fraction * rect.Width());

    if (nBarWidth != m_nBarWidth)
    {
        m_nBarWidth = nBarWidth;
        RedrawWindow();
    }

    return (LRESULT)nOldPos;
}

LRESULT CTextProgressCtrl::OnStepIt(WPARAM, LPARAM) 
{    
   return (LRESULT)SetPos(m_nPos + m_nStepSize);
}

LRESULT CTextProgressCtrl::OnOffsetPos(WPARAM nPos, LPARAM)
{
    return (LRESULT)SetPos(m_nPos + (int)nPos);
}

LRESULT CTextProgressCtrl::OnSetStep(WPARAM nStep, LPARAM)
{
    int nOldStep = m_nStepSize;
    m_nStepSize = (int)nStep;
    return (LRESULT)nOldStep;
}

LRESULT CTextProgressCtrl::OnSetBarColor(WPARAM, LPARAM Colour)
{
    return (LRESULT)SetBarColour((COLORREF)Colour);
}

COLORREF CTextProgressCtrl::SetBarColour(COLORREF crBarClr /*= CLR_DEFAULT*/)
{
    if (::IsWindow(m_hWnd))
        Invalidate();

    COLORREF crOldBarClr = m_crBarClr;
    m_crBarClr = crBarClr;
    return crOldBarClr;
}

COLORREF CTextProgressCtrl::GetBarColour() const
{ 
    return m_crBarClr;
}

LRESULT CTextProgressCtrl::OnSetBkColor(WPARAM, LPARAM Colour)
{
    return (LRESULT)SetBgColour((COLORREF)Colour);
}

COLORREF CTextProgressCtrl::SetBgColour(COLORREF crBgClr /*= CLR_DEFAULT*/)
{
    if (::IsWindow(m_hWnd))
        Invalidate();

    COLORREF crOldBgClr = m_crBgClr;
    m_crBgClr = crBgClr;
    return crOldBgClr;
}

COLORREF CTextProgressCtrl::GetBgColour() const
{ 
    return m_crBgClr;
}

LRESULT CTextProgressCtrl::OnSetTextColor(WPARAM, LPARAM Colour)
{
    return (LRESULT)SetTextColour((COLORREF)Colour);
}

COLORREF CTextProgressCtrl::SetTextColour(COLORREF crTextClr /*= CLR_DEFAULT*/)
{
    if (::IsWindow(m_hWnd))
        Invalidate();

    COLORREF crOldTextClr = m_crTextClr;
    m_crTextClr = crTextClr;
    return crOldTextClr;
}

COLORREF CTextProgressCtrl::GetTextColour()
{
    return m_crTextClr;
}