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

///////////////////////////////////////////////////////////////////////////// 
// File name:      STScreenBuffer.h 
// Author:         Vassili Philippov (vasja@spbteam.com) 
// Created:        June 2001 
// Last changed:   07 August 2001 
// Version:        2.0 
// Description:    Class for pixel working with images. One could get access 
//                 to image bits using this library or create custom image. 
 
#if !defined(AFX_STSCREENBUFFER_H__22D62F5D_32E2_4785_B3D9_2341C11F84A3__INCLUDED_) 
#define AFX_STSCREENBUFFER_H__22D62F5D_32E2_4785_B3D9_2341C11F84A3__INCLUDED_ 
 
#if _MSC_VER > 1000 
#pragma once 
#endif // _MSC_VER > 1000 
 
 
///////////////////////////////////////////////////////////////////////////// 
// BGRColor structure encapsulates color information about one point. Color 
// order is Blue, Green, Red (not RGB). 
struct DIBINFO : public BITMAPINFO 
{ 
	RGBQUAD	 arColors[255];    // Color table info - adds an extra 255 entries to palette 
	 
	operator LPBITMAPINFO()          { return (LPBITMAPINFO) this; } 
	operator LPBITMAPINFOHEADER()    { return &bmiHeader;          } 
	RGBQUAD* ColorTable()            { return bmiColors;           } 
}; 
 
struct BGRColor 
{ 
	BGRColor() {} 
	BGRColor(byte R, byte G, byte B) : m_R(R), m_G(G), m_B(B) {} 
	byte m_B; 
	byte m_G; 
	byte m_R; 
}; 
 
 
///////////////////////////////////////////////////////////////////////////// 
// CSTScreenBuffer class provides fast drawing methods and offscreen buffer. 
 
class CSTScreenBuffer 
{ 
public: 
	// Creates uninitialized buffer. Call Create or CreateRGB to 
	// initialize the buffer. 
	CSTScreenBuffer(); 
 
	virtual ~CSTScreenBuffer(); 
 
	// Creates buffer with the given size.  
	void Create(int nWidth, int nHeight); 
 
	// Creates buffer with the given size and fills it with  
	// the given color 
	void Create(int nWidth, int nHeight, COLORREF clr); 
 
	// Creates buffer with the same width andheight as 
	// the given bitmap and that contains the same picture. 
	void Create(HBITMAP hBitmap); 
 
	// Creates buffer that will contain picture from the given 
 	// area of the given device context. 
	void Create(CDC *pDC, CRect rect); 
 
	// Creates buffer with the given size and uses the given 
	// array as a source of pixels color information. The given 
	// array should contain 3 bytes per pixel (RGB). To the array 
	// size should be 3*nWidth*nHeight 
	void CreateRGB(void *pData, int nWidth, int nHeight); 
 
	// Draws buffer into the given device context starting from 
	// the given point (top left corner) 
	BOOL Draw(CDC* pDC, CPoint ptDest); 
 
	// Sets color of the given point 
	inline void SetPoint(int nX, int nY, byte R, byte G, byte B) { 
		m_pBuffer[m_nCorrectedWidth*(m_nHeight-nY-1)+nX] = BGRColor(R,G,B); 
	} 
 
	// Returns color of the given point 
	inline BGRColor GetPoint(int nX, int nY) { 
		return m_pBuffer[m_nCorrectedWidth*(m_nHeight-nY-1)+nX]; 
	} 
 
	// Returns array that contains points color information. Each 
	// point is represented by 3 bytes in Blue, Green, Red order. 
	// Array contains CorrectedWidth*Height elements (Note! not  
	// Width*Height). 
	BGRColor *GetPointsArray() { 
		return m_pBuffer; 
	} 
 
	// Returns real width of the screen buffer. It could be slightly more then 
	// requested width. This paramater is important only when you work with 
	// points array directly (using GetPointsArray function). 
	int GetCorrectedWidth() { 
		return m_nCorrectedWidth; 
	} 
 
	// Returns screen buffer width. 
	int GetWidth() { 
		return m_nWidth; 
	} 
 
	// Returns screen buffer height. 
	int GetHeight() { 
		return m_nHeight; 
	} 
 
	// Returns handle of the encapsulated bitmap. 
	HBITMAP GetHBitmap() { 
		return m_hBitmap; 
	} 
 
	// Returns screen buffer device context. After drawing this 
	// device context should be closed (call ReleaseDC method).  
	CDC *GetDC(); 
 
	// Closes bitmap device context previously opened by GetDC function. 
	void ReleaseDC(); 
 
public: 
	// Creates bitmap with the given sizes and the given array of colors. 
	static HBITMAP CreateBitmapByRGBArray(void *pData, int nWidth, int nHeight); 
 
	// Returns minimum width that is greater then the given width and 
	// that is acceptable as image width (not all numbers are acceptable) 
	static int CorrectedWidth(int nWidth); 
 
protected: 
	// Creates internal bitmap and image buffer. Assignes width and 
	// height properties 
	BOOL CreateBitmap(int nWidth, int nHeight); 
 
	int m_nWidth; 
	int m_nHeight; 
	int m_nCorrectedWidth; 
	BGRColor *m_pBuffer; 
	HBITMAP m_hBitmap; 
 
	// Members related to device context 
	CDC *m_pDC; 
	HBITMAP m_hSaveBitmap; 
}; 
 
#endif // !defined(AFX_STSCREENBUFFER_H__22D62F5D_32E2_4785_B3D9_2341C11F84A3__INCLUDED_)