www.pudn.com > ImageVidwer.rar > STScreenBuffer.cpp

#include "stdafx.h" 
#include "STScreenBuffer.h" 
 
#ifdef _DEBUG 
#undef THIS_FILE 
static char THIS_FILE[]=__FILE__; 
#define new DEBUG_NEW 
#endif 
 
////////////////////////////////////////////////////////////////////// 
// Construction/Destruction 
////////////////////////////////////////////////////////////////////// 
 
 
int CSTScreenBuffer::CorrectedWidth(int nWidth) 
{ 
	return ( ( nWidth + 3 ) / 4 ) * 4; 
} 
 
///////////////////////////////////////////////////////////////////////////// 
// CSTScreenBuffer 
 
CSTScreenBuffer::CSTScreenBuffer() 
	: m_hBitmap(NULL), 
	  m_pBuffer(NULL), 
	  m_pDC(NULL) 
{ 
} 
 
CSTScreenBuffer::~CSTScreenBuffer() 
{ 
	if (m_hBitmap!=NULL) { 
		ReleaseDC(); 
		::DeleteObject(m_hBitmap); 
	} 
} 
 
BOOL CSTScreenBuffer::CreateBitmap(int nWidth, int nHeight) 
{ 
	ASSERT(nWidth>0); 
	ASSERT(nHeight>0); 
 
	if (m_hBitmap!=NULL) DeleteObject(m_hBitmap); 
 
	m_nCorrectedWidth = CorrectedWidth(nWidth); 
	m_nWidth = nWidth; 
	m_nHeight = nHeight; 
 
	DIBINFO  dibInfo; 
 
	dibInfo.bmiHeader.biBitCount = 24; 
	dibInfo.bmiHeader.biClrImportant = 0; 
	dibInfo.bmiHeader.biClrUsed = 0; 
	dibInfo.bmiHeader.biCompression = 0; 
	dibInfo.bmiHeader.biHeight = m_nHeight; 
	dibInfo.bmiHeader.biPlanes = 1; 
	dibInfo.bmiHeader.biSize = 40; 
	dibInfo.bmiHeader.biSizeImage = m_nCorrectedWidth*m_nHeight*3; 
	dibInfo.bmiHeader.biWidth = m_nCorrectedWidth; 
	dibInfo.bmiHeader.biXPelsPerMeter = 3780; 
	dibInfo.bmiHeader.biYPelsPerMeter = 3780; 
	dibInfo.bmiColors[0].rgbBlue = 0; 
	dibInfo.bmiColors[0].rgbGreen = 0; 
	dibInfo.bmiColors[0].rgbRed = 0; 
	dibInfo.bmiColors[0].rgbReserved = 0; 
 
	HDC hDC = ::GetDC(NULL); 
	ASSERT(hDC); 
	m_hBitmap = CreateDIBSection(hDC, (const BITMAPINFO*)dibInfo, DIB_RGB_COLORS, (void**)&m_pBuffer, NULL, 0); 
	::ReleaseDC(NULL, hDC); 
	ASSERT(m_hBitmap); 
	ASSERT(m_pBuffer); 
 
	return TRUE; 
} 
 
void CSTScreenBuffer::Create(int nWidth, int nHeight) 
{ 
	ASSERT(nWidth>0); 
	ASSERT(nHeight>0); 
 
	CreateBitmap(nWidth, nHeight); 
} 
 
void CSTScreenBuffer::Create(int nWidth, int nHeight, COLORREF clr) 
{ 
	ASSERT(nWidth>0); 
	ASSERT(nHeight>0); 
 
	CreateBitmap(nWidth, nHeight); 
 
	BGRColor bgrColor = BGRColor(GetBValue(clr), GetGValue(clr), GetRValue(clr)); 
	int nPosition = 0; 
 
	for (int y=0; yBitBlt(0,0, rect.Width(), rect.Height(), pDC, rect.left, rect.top, SRCCOPY); 
} 
 
void CSTScreenBuffer::CreateRGB(void *pData, int nWidth, int nHeight) 
{ 
	ASSERT(pData); 
	ASSERT(nWidth>0); 
	ASSERT(nHeight>0); 
 
	CreateBitmap(nWidth, nHeight); 
 
	byte *pByteData = (byte*)pData; 
	int nPosition = 0; 
	int nDataPosition = 0; 
 
	for (int y=0; yBitBlt(ptDest.x, ptDest.y, m_nWidth, m_nHeight, &memDc, ptOrigin.x, ptOrigin.y, SRCCOPY); 
	::SelectObject(memDc.GetSafeHdc(), m_hOldBitmap); 
	memDc.DeleteDC(); 
 
	return bResult; 
} 
 
 
HBITMAP CSTScreenBuffer::CreateBitmapByRGBArray(void *pData, int nWidth, int nHeight) 
{ 
	HBITMAP hResult = NULL;	 
	CSTScreenBuffer sb; 
 
	sb.CreateRGB(pData, nWidth, nHeight); 
	hResult = sb.m_hBitmap; 
 
	sb.m_hBitmap = NULL; 
	sb.m_pBuffer = NULL; 
 
	return hResult; 
} 
 
CDC *CSTScreenBuffer::GetDC() 
{ 
	if (m_pDC) return m_pDC; 
 
	m_pDC = new CDC; 
	if (!m_pDC->CreateCompatibleDC(NULL)) { 
		delete m_pDC; 
		return NULL; 
	} 
 
	m_hSaveBitmap = (HBITMAP)m_pDC->SelectObject(GetHBitmap()); 
	return m_pDC; 
} 
 
void CSTScreenBuffer::ReleaseDC() 
{ 
	if (m_pDC) { 
		m_pDC->SelectObject(m_hSaveBitmap); 
		m_pDC->DeleteDC(); 
		delete m_pDC; 
		m_pDC = NULL; 
	} 
}