#pragma once
|
|
#include "cxcore.h"
|
#include "cv.h"
|
#include "highgui.h"
|
|
enum ImageBandType { ImageBand_Red=0, ImageBand_Green, ImageBand_Blue, ImageBand_Gray, ImageBand_Color, ImageBand_Count };
|
|
//struct _IplImage;
|
|
class AFX_EXT_CLASS CImageBuffer
|
{
|
public:
|
CImageBuffer(void);
|
CImageBuffer(int nWidth, int nHeight, int nDepth=8, int nChannels=1);
|
virtual ~CImageBuffer(void);
|
|
public: // virtual functions
|
virtual BOOL CreateImage(int nWidth, int nHeight, int nDepth=8, int nChannels=1);
|
virtual BOOL CreateImageHeader(int nWidth, int nHeight, int nDepth=8, int nChannels=1);
|
virtual BOOL GetImageExist(void) const;
|
virtual BOOL GetAllocatedData(void) const;
|
|
virtual void ReleaseImage(void);
|
virtual int GetHeight(void) const;
|
virtual int GetWidth(void) const;
|
virtual int GetWidthStep(void) const;
|
virtual int GetDepth(void) const;
|
virtual int GetChannels(void) const;
|
virtual int GetBpp(void) const;
|
virtual int GetImageSize(void) const;
|
|
virtual char* GetImageBuffer(void);
|
virtual char* GetImageBufferOrigin(void);
|
virtual const char* GetImageBuffer(void) const;
|
virtual const char* GetImageBufferOrigin(void) const;
|
virtual int GetPixelValue(int x, int y) const;
|
virtual int SetPixelValue(int x, int y, int nR, int nG, int nB);
|
virtual int SetPixelValue(int x, int y, int nR, int nG, int nB, int nA);
|
|
virtual BOOL LoadImage(const CString& strFilename);
|
virtual BOOL SaveImage(const CString& strFilename, int nQiality=95) const;
|
|
public: // common functions
|
// clear
|
BOOL ClearImage(int nValue);
|
BOOL ClearImage(int nValue, int nA);
|
|
// copy func
|
BOOL CopyImageTo(CImageBuffer *pToImage) const;
|
BOOL CopyImageTo(IplImage *pToImage) const;
|
BOOL CopyImageFrom(const CImageBuffer *pFromImage);
|
BOOL CopyImageFrom(const IplImage *pFromImage);
|
|
// show func
|
void ShowImage(HDC hDC, const CRect& dstRect);
|
void ShowImage(HDC hDC, const CRect& srcRect, const CRect& dstRect);
|
void ShowImage(HDC hDC, int nSx, int nSy, int nWidth, int nHeight, int nFromX, int nFromY);
|
|
// get image data
|
BOOL GetSubImage(int nSx, int nSy, int nWidth, int nHeight, CImageBuffer *pGetImage);
|
BOOL GetSubImage(int nSx, int nSy, int nWidth, int nHeight, int nChannels, BYTE *pGetBuffer);
|
BOOL GetSubImage(const CRect& rtRect, CImageBuffer *pGetImage);
|
BOOL GetSubImage(const CRect& rtRect, int nChannels, BYTE *pGetBuffer);
|
|
BOOL GetBandImage(int nColorBand, CImageBuffer* pGetImage) const;
|
BOOL GetBandImage(int nColorBand, int nSx, int nSy, int nWidth, int nHeight, CImageBuffer* pGetImage);
|
BOOL GetBandImage(int nColorBand, int nSx, int nSy, int nWidth, int nHeight, BYTE* pGetBuffer);
|
BOOL GetBandImage(int nColorBand, const CRect& rtRect, CImageBuffer* pGetImage);
|
BOOL GetBandImage(int nColorBand, const CRect& rtRect, BYTE* pGetBuffer);
|
|
// set image data
|
BOOL SetSubImage(int nSx, int nSy, int nWidth, int nHeight, CImageBuffer* pSetImage);
|
BOOL SetSubImage(int nSx, int nSy, int nWidth, int nHeight, int nBpp, BYTE *pSetBuffer);
|
BOOL SetSubImage(const CRect& rtRect, CImageBuffer *pSetImage);
|
BOOL SetSubImage(const CRect& rtRect, int nChannels, BYTE *pSetBuffer);
|
|
// draw function
|
BOOL DrawLine(CPoint ptPos1, CPoint ptPos2, DWORD nColor, int nThickness=1, int nLineType=8, int nShift=0 );
|
BOOL DrawRectangle(CPoint ptPos1, CPoint ptPos2, DWORD nColor, int nThickness=1, int nLineType=8, int nShift=0 );
|
BOOL DrawCircle(CPoint ptCenter, int nRadius, DWORD nColor, int nThickness=1, int nLineType=8, int nShift=0 );
|
BOOL DrawEllipse(CPoint ptCenter, CPoint ptSize, double dAngle, DWORD nColor, int nThickness=1, int nLineType=8, int nShift=0);
|
BOOL DrawText(CPoint ptPoint, DWORD nColor, const CString& strText);
|
|
// buffer access
|
IplImage* GetIplImage(void);
|
BOOL SetImageBufferPtr(char* pPtr);
|
|
BOOL DIBtoIplImage(HBITMAP& hbmp, HDC hdc=NULL);
|
BOOL DCtoIplImage(CDC* pDC, const CRect& rect);
|
|
// 4Ch Image..
|
CvScalar GetPixelScalar(int x, int y);
|
|
protected:
|
static void FillBitmapInfo(BITMAPINFO* bmi, int width, int height, int bpp, int origin);
|
|
BOOL m_bAllocatedData;
|
|
public:
|
IplImage *m_pImageData;
|
};
|
