TnbDcSelectAssistant.h

[詳解]

#pragma once
#include "TnbDef.h"
 
 
 
//TNB Library
namespace TNB
{
 
 
 
class CDcSelectAssistant : CCopyImpossible
{
public:
 
    CDcSelectAssistant(HDC dc) : m_dc(dc) 
        , m_textColor(CLR_INVALID), m_backColor(CLR_INVALID), m_backMode(-1)
        , m_font(NULL), m_brush(NULL), m_pen(NULL), m_bitmap(NULL)
    {
    }
 
    ~CDcSelectAssistant(void)
    {
        Restore();
    }
 
    void Restore(void)
    {
        if ( m_textColor != CLR_INVALID )
        {
            ::SetTextColor(m_dc, m_textColor);
            m_textColor = CLR_INVALID;
        }
        if ( m_backColor != CLR_INVALID )
        {
            ::SetBkColor(m_dc,m_backColor);
            m_backColor = CLR_INVALID;
        }
        if ( m_backMode >= 0 )
        {
            ::SetBkMode(m_dc, m_backMode);
            m_backMode = -1;
        }
        if ( m_font != NULL )
        {
            ::SelectObject(m_dc, m_font);
            m_font = NULL;
        }
        if ( m_brush != NULL )
        {
            ::SelectObject(m_dc, m_brush);
            m_brush = NULL;
        }
        if ( m_pen != NULL )
        {
            ::SelectObject(m_dc, m_pen);
            m_pen = NULL;
        }
        if ( m_bitmap != NULL )
        {
            ::SelectObject(m_dc, m_bitmap);
            m_bitmap = NULL;
        }
    }
 
    void SetTextColor(COLORREF c)
    {
        COLORREF r = ::SetTextColor(m_dc, c);
        if ( m_textColor == CLR_INVALID )
        {
            m_textColor = r;
        }
    }
 
    void SetBkColor(COLORREF c)
    {
        COLORREF r = ::SetBkColor(m_dc, c);
        if ( m_backColor == CLR_INVALID )
        {
            m_backColor = r;
        }
    }
 
    void SetBkMode(int mode)
    {
        int r = ::SetBkMode(m_dc, mode);
        if ( m_backMode < 0 )
        {
            m_backMode = r;
        }
    }
 
    void SelectBrush(HBRUSH brush)
    {
        HGDIOBJ r = ::SelectObject(m_dc, brush);
        if ( m_brush == NULL )
        {
            m_brush = static_cast<HBRUSH>(r);
        }
    }
 
    void SelectPen(HPEN pen)
    {
        HGDIOBJ r = ::SelectObject(m_dc, pen);
        if ( m_pen == NULL )
        {
            m_pen = static_cast<HPEN>(r);
        }
    }
 
    void SelectBitmap(HBITMAP bitmap)
    {
        HGDIOBJ r = ::SelectObject(m_dc, bitmap);
        if ( m_bitmap == NULL )
        {
            m_bitmap = static_cast<HBITMAP>(r);
        }
    }
 
    void SelectFont(HFONT font)
    {
        HGDIOBJ r = ::SelectObject(m_dc, font);
        if ( m_font == NULL )
        {
            m_font = static_cast<HFONT>(r);
        }
    }
 
    #ifdef __AFX_H__
 
    void SelectFont(CFont* pFont)
    {
        if ( pFont != NULL )
        {
            SelectFont(static_cast<HFONT>(pFont->GetSafeHandle()));
        }
    }
 
    #endif
 
private:
    HDC         m_dc;           
    COLORREF    m_textColor;    
    COLORREF    m_backColor;    
    int         m_backMode;     
    HFONT       m_font;         
    HBRUSH      m_brush;        
    HPEN        m_pen;          
    HBITMAP     m_bitmap;       
};
 
 
 
};//TNB