src/Utils/MiscUI/BufferDC.cpp

   1 #include "stdafx.h"
   2 #include "BufferDC.h"
   3
   4 IMPLEMENT_DYNAMIC(CBufferDC, CPaintDC)
   5
   6 CBufferDC::CBufferDC(CWnd* pWnd) : CPaintDC(pWnd)
   7 {
   8         if (pWnd != NULL && CPaintDC::m_hDC != NULL)
   9         {
  10                 m_hOutputDC    = CPaintDC::m_hDC;
  11                 m_hAttributeDC = CPaintDC::m_hAttribDC;
  12
  13                 pWnd->GetClientRect(&m_ClientRect);
  14
  15                 m_hMemoryDC = ::CreateCompatibleDC(m_hOutputDC);
  16
  17                 m_hPaintBitmap =
  18                         ::CreateCompatibleBitmap(
  19                                         m_hOutputDC,
  20                                         m_ClientRect.right  - m_ClientRect.left,
  21                                         m_ClientRect.bottom - m_ClientRect.top);
  22
  23                 m_hOldBitmap = (HBITMAP)::SelectObject(m_hMemoryDC, m_hPaintBitmap);
  24
  25                 CPaintDC::m_hDC       = m_hMemoryDC;
  26                 CPaintDC::m_hAttribDC = m_hMemoryDC;
  27         }
  28
  29         m_bBoundsUpdated = FALSE;
  30 }
  31
  32 CBufferDC::~CBufferDC(void)
  33 {
  34         Flush();
  35
  36         ::SelectObject(m_hMemoryDC, m_hOldBitmap);
  37         ::DeleteObject(m_hPaintBitmap);
  38
  39         CPaintDC::m_hDC           = m_hOutputDC;
  40         CPaintDC::m_hAttribDC = m_hAttributeDC;
  41
  42         ::DeleteDC(m_hMemoryDC);
  43 }
  44
  45 void CBufferDC::Flush()
  46 {
  47         ::BitBlt(
  48                 m_hOutputDC,
  49                 m_ClientRect.left, m_ClientRect.top,
  50                 m_ClientRect.right  - m_ClientRect.left,
  51                 m_ClientRect.bottom - m_ClientRect.top,
  52                 m_hMemoryDC,
  53                 0, 0,
  54                 SRCCOPY);
  55 }
  56
  57 UINT CBufferDC::SetBoundsRect( LPCRECT lpRectBounds, UINT flags )
  58 {
  59         if (lpRectBounds != NULL)
  60         {
  61                 if (m_ClientRect.right  - m_ClientRect.left > lpRectBounds->right  - lpRectBounds->left ||
  62                         m_ClientRect.bottom - m_ClientRect.top  > lpRectBounds->bottom - lpRectBounds->top)
  63                 {
  64                         lpRectBounds = &m_ClientRect;
  65                 }
  66
  67                 HBITMAP bmp =
  68                         ::CreateCompatibleBitmap(
  69                                         m_hOutputDC,
  70                                         lpRectBounds->right - lpRectBounds->left,
  71                                         lpRectBounds->bottom - lpRectBounds->top);
  72
  73                 HDC tmpDC  = ::CreateCompatibleDC(m_hOutputDC);
  74
  75                 HBITMAP oldBmp = (HBITMAP)::SelectObject(tmpDC, bmp);
  76
  77                 ::BitBlt(
  78                         tmpDC,
  79                         m_ClientRect.left, m_ClientRect.top,
  80                         m_ClientRect.right  - m_ClientRect.left,
  81                         m_ClientRect.bottom - m_ClientRect.top,
  82                         m_hMemoryDC,
  83                         0, 0,
  84                         SRCCOPY);
  85
  86                 ::SelectObject(tmpDC, oldBmp);
  87                 ::DeleteDC(tmpDC);
  88
  89                 HBITMAP old = (HBITMAP)::SelectObject(m_hMemoryDC, bmp);
  90
  91                 if (old != NULL && old != m_hPaintBitmap)
  92                 {
  93                         ::DeleteObject(old);
  94                 }
  95
  96                 if (m_hPaintBitmap != NULL)
  97                 {
  98                         ::DeleteObject(m_hPaintBitmap);
  99                 }
 100
 101                 m_hPaintBitmap = bmp;
 102
 103                 m_ClientRect = *lpRectBounds;
 104                 m_bBoundsUpdated = TRUE;
 105         }
 106
 107         return CPaintDC::SetBoundsRect(lpRectBounds, flags);
 108 }
 109
 110 BOOL CBufferDC::RestoreDC( int nSavedDC )
 111 {
 112         BOOL ret = CPaintDC::RestoreDC(nSavedDC);
 113
 114         if (m_bBoundsUpdated)
 115         {
 116                 SelectObject(m_hPaintBitmap);
 117         }
 118
 119         return ret;
 120 }