2024年2月21日发(作者:)

virtual void Fill( int color ); /* draw to highgui window */ virtual void Show( const char* window );#if defined WIN32 || defined _WIN32 /* draw part of image to the specified DC */ virtual void Show( HDC dc, int x, int y, int width, int height, int from_x = 0, int from_y = 0 ); /* draw the current image ROI to the specified rectangle of the destination DC */ virtual void DrawToHDC( HDC hDCDst, RECT* pDstRect );#endifprotected: IplImage* m_img;};typedef CvvImage CImage;#endif

//

#include "StdAfx.h"#include "CvvImage.h"//// Construction/Destruction//CV_INLINE RECT NormalizeRect( RECT r );CV_INLINE RECT NormalizeRect( RECT r ){ int t; if( > ) { t = ; = ; = t; } if( > ) { t = ; = ; = t; } return r;}CV_INLINE CvRect RectToCvRect( RECT sr );CV_INLINE CvRect RectToCvRect( RECT sr ){ sr = NormalizeRect( sr );

return cvRect( , , - , - );}CV_INLINE RECT CvRectToRect( CvRect sr );CV_INLINE RECT CvRectToRect( CvRect sr ){ RECT dr; = sr.x; = sr.y; = sr.x + ; = sr.y + ; return dr;}CV_INLINE IplROI RectToROI( RECT r );CV_INLINE IplROI RectToROI( RECT r ){ IplROI roi; r = NormalizeRect( r ); t = ; t = ; = - ; = - ; = 0; return roi;}void FillBitmapInfo( BITMAPINFO* bmi, int width, int height, int bpp, int origin ){ assert( bmi && width >= 0 && height >= 0 && (bpp == 8 || bpp == 24 || bpp == 32)); BITMAPINFOHEADER* bmih = &(bmi->bmiHeader); memset( bmih, 0, sizeof(*bmih)); bmih->biSize = sizeof(BITMAPINFOHEADER); bmih->biWidth = width; bmih->biHeight = origin ? abs(height) : -abs(height); bmih->biPlanes = 1; bmih->biBitCount = (unsigned short)bpp; bmih->biCompression = BI_RGB; if( bpp == 8 ) { RGBQUAD* palette = bmi->bmiColors; int i; for( i = 0; i < 256; i++ ) { palette[i].rgbBlue = palette[i].rgbGreen = palette[i].rgbRed = (BYTE)i; palette[i].rgbReserved = 0; } }}CvvImage::CvvImage(){ m_img = 0;}void CvvImage::Destroy(){ cvReleaseImage( &m_img );}

CvvImage::~CvvImage(){ Destroy();}bool CvvImage::Create( int w, int h, int bpp, int origin ){ const unsigned max_img_size = 10000; if( (bpp != 8 && bpp != 24 && bpp != 32) || (unsigned)w >= max_img_size || (unsigned)h >= max_img_size || (origin != IPL_ORIGIN_TL && origin != IPL_ORIGIN_BL)) { assert(0); // most probably, it is a programming error return false; } if( !m_img || Bpp() != bpp || m_img->width != w || m_img->height != h ) { if( m_img && m_img->nSize == sizeof(IplImage)) Destroy(); /* prepare IPL header */ m_img = cvCreateImage( cvSize( w, h ), IPL_DEPTH_8U, bpp/8 ); } if( m_img ) m_img->origin = origin == 0 ? IPL_ORIGIN_TL : IPL_ORIGIN_BL; return m_img != 0;}void CvvImage::CopyOf( CvvImage& image, int desired_color ){ IplImage* img = ge(); if( img ) { CopyOf( img, desired_color ); }}#define HG_IS_IMAGE(img) ((img) != 0 && ((const IplImage*)(img))->nSize == sizeof(IplImage) && ((IplImage*)img)->imageData != 0)void CvvImage::CopyOf( IplImage* img, int desired_color ){ if( HG_IS_IMAGE(img) ) { int color = desired_color; CvSize size = cvGetSize( img );

if( color < 0 ) color = img->nChannels > 1; if( Create( , , (!color ? 1 : img->nChannels > 1 ? img->nChannels : 3)*8, img->origin )) { cvConvertImage( img, m_img, 0 ); } }}

bool CvvImage::Load( const char* filename, int desired_color ){ IplImage* img = cvLoadImage( filename, desired_color ); if( !img ) return false; CopyOf( img, desired_color ); cvReleaseImage( &img ); return true;}bool CvvImage::LoadRect( const char* filename, int desired_color, CvRect r ){ if( < 0 || < 0 ) return false; IplImage* img = cvLoadImage( filename, desired_color ); if( !img ) return false; if( == 0 || == 0 ) { = img->width; = img->height; r.x = r.y = 0; } if( r.x > img->width || r.y > img->height || r.x + < 0 || r.y + < 0 ) { cvReleaseImage( &img ); return false; } /* truncate r to source image */ if( r.x < 0 ) { += r.x; r.x = 0; } if( r.y < 0 ) { += r.y; r.y = 0; } if( r.x + > img->width ) = img->width - r.x; if( r.y + > img->height ) = img->height - r.y; cvSetImageROI( img, r ); CopyOf( img, desired_color ); cvReleaseImage( &img ); return true;}bool CvvImage::Save( const char* filename )

{ if( !m_img ) return false; cvSaveImage( filename, m_img ); return true;}void CvvImage::Show( const char* window ){ if( m_img ) cvShowImage( window, m_img );}void CvvImage::Show( HDC dc, int x, int y, int w, int h, int from_x, int from_y ){ if( m_img && m_img->depth == IPL_DEPTH_8U ) { uchar buffer[sizeof(BITMAPINFOHEADER) + 1024]; BITMAPINFO* bmi = (BITMAPINFO*)buffer; int bmp_w = m_img->width, bmp_h = m_img->height; FillBitmapInfo( bmi, bmp_w, bmp_h, Bpp(), m_img->origin ); from_x = MIN( MAX( from_x, 0 ), bmp_w - 1 ); from_y = MIN( MAX( from_y, 0 ), bmp_h - 1 ); int sw = MAX( MIN( bmp_w - from_x, w ), 0 ); int sh = MAX( MIN( bmp_h - from_y, h ), 0 ); SetDIBitsToDevice( dc, x, y, sw, sh, from_x, from_y, from_y, sh, m_img->imageData + from_y*m_img->widthStep, bmi, DIB_RGB_COLORS ); }}void CvvImage::DrawToHDC( HDC hDCDst, RECT* pDstRect )

{ if( pDstRect && m_img && m_img->depth == IPL_DEPTH_8U && m_img->imageData ) { uchar buffer[sizeof(BITMAPINFOHEADER) + 1024]; BITMAPINFO* bmi = (BITMAPINFO*)buffer; int bmp_w = m_img->width, bmp_h = m_img->height; CvRect roi = cvGetImageROI( m_img ); CvRect dst = RectToCvRect( *pDstRect ); if( == && == ) { Show( hDCDst, dst.x, dst.y, , , roi.x, roi.y ); return; } if( > ) { SetStretchBltMode( hDCDst, // handle to device context HALFTONE ); } else { SetStretchBltMode(

hDCDst, // handle to device context COLORONCOLOR ); } FillBitmapInfo( bmi, bmp_w, bmp_h, Bpp(), m_img->origin ); ::StretchDIBits( hDCDst, dst.x, dst.y, , , roi.x, roi.y, , , m_img->imageData, bmi, DIB_RGB_COLORS, SRCCOPY ); }}void CvvImage::Fill( int color ){ cvSet( m_img, cvScalar(color&255,(color>>8)&255,(color>>16)&255,(color>>24)&255) );}