VS2010+Opencv+MFC讀取圖像和視頻顯示在Picture控件

#pragma once
 
#ifndef CVVIMAGE_CLASS_DEF 
#define CVVIMAGE_CLASS_DEF 
#include "opencv.hpp"
 
class CvvImage
 
{
 
public:
 
 CvvImage();
 
 virtual ~CvvImage();
 
 
 
 virtual bool Create( int width, int height, int bits_per_pixel, int image_origin = 0 ); 
 virtual bool Load( const char* filename, int desired_color = 1 );
 virtual bool LoadRect( const char* filename, 
 int desired_color, CvRect r );
 
#if defined WIN32 || defined _WIN32 
 virtual bool LoadRect( const char* filename, 
 int desired_color, RECT r )
 
 {
 
 return LoadRect( filename, desired_color,
  cvRect( r.left, r.top, r.right - r.left, r.bottom - r.top ));
 
 }
 
#endif
 
 
 
 virtual bool Save( const char* filename ); 
 virtual void CopyOf( CvvImage& image, int desired_color = -1 ); 
 virtual void CopyOf( IplImage* img, int desired_color = -1 ); 
 IplImage* GetImage() { return m_img; };
 
 virtual void Destroy(void);
 
 int Width() { return !m_img ? 0 : !m_img->roi ? m_img->width : m_img->roi->width; }; 
 int Height() { return !m_img ? 0 : !m_img->roi ? m_img->height : m_img->roi->height;}; 
 int Bpp() { return m_img ? (m_img->depth & 255)*m_img->nChannels : 0; };
 
 virtual void Fill( int color ); 
 virtual void Show( const char* window );
 
#if defined WIN32 || defined _WIN32
 
 virtual void Show( HDC dc, int x, int y, int width, int height,
 int from_x = 0, int from_y = 0 );
 
 
 
 virtual void DrawToHDC( HDC hDCDst, RECT* pDstRect ); 
#endif 
protected: 
 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( r.left > r.right )
 {
 t = r.left;
 r.left = r.right;
 r.right = t;
 }
 if( r.top > r.bottom )
 {
 t = r.top;
 r.top = r.bottom;
 r.bottom = t;
 }
 
 return r;
}
CV_INLINE CvRect RectToCvRect( RECT sr );
CV_INLINE CvRect RectToCvRect( RECT sr )
{
 sr = NormalizeRect( sr );
 return cvRect( sr.left, sr.top, sr.right - sr.left, sr.bottom - sr.top );
}
CV_INLINE RECT CvRectToRect( CvRect sr );
CV_INLINE RECT CvRectToRect( CvRect sr )
{
 RECT dr;
 dr.left = sr.x;
 dr.top = sr.y;
 dr.right = sr.x + sr.width;
 dr.bottom = sr.y + sr.height;
 
 return dr;
}
CV_INLINE IplROI RectToROI( RECT r );
CV_INLINE IplROI RectToROI( RECT r )
{
 IplROI roi;
 r = NormalizeRect( r );
 roi.xOffset = r.left;
 roi.yOffset = r.top;
 roi.width = r.right - r.left;
 roi.height = r.bottom - r.top;
 roi.coi = 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();
 
 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 = image.GetImage();
 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( size.width, size.height,
  (!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( r.width < 0 || r.height < 0 ) return false;
 
 IplImage* img = cvLoadImage( filename, desired_color );
 if( !img )
 return false;
 if( r.width == 0 || r.height == 0 )
 {
 r.width = img->width;
 r.height = img->height;
 r.x = r.y = 0;
 }
 if( r.x > img->width || r.y > img->height ||
 r.x + r.width < 0 || r.y + r.height < 0 )
 {
 cvReleaseImage( &img );
 return false;
 }
 
 if( r.x < 0 )
 {
 r.width += r.x;
 r.x = 0;
 }
 if( r.y < 0 )
 {
 r.height += r.y;
 r.y = 0;
 }
 if( r.x + r.width > img->width )
 r.width = img->width - r.x;
 
 if( r.y + r.height > img->height )
 r.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( roi.width == dst.width && roi.height == dst.height )
 {
  Show( hDCDst, dst.x, dst.y, dst.width, dst.height, roi.x, roi.y );
  return;
 }
 if( roi.width > dst.width )
 {
  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, dst.width, dst.height,
  roi.x, roi.y, roi.width, roi.height,
  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) );
}

void CopencvtestDlg::OnBnClickedCancel()
{
 // TODO: 在此添加控件通知處理程序代碼
 CDC *pDC = GetDlgItem(IDC_STATIC)->GetDC();//根據(jù)ID獲得窗口指針再獲取與該窗口關(guān)聯(lián)的上下文指針
 HDC hdc= pDC->GetSafeHdc();      // 獲取設(shè)備上下文句柄
 CRect rect;
// 矩形類
 GetDlgItem(IDC_STATIC)->GetClientRect(&rect); //獲取box1客戶區(qū)
 CvvImage cimg;
 IplImage *src; // 定義IplImage指針變量src  
 src = cvLoadImage("D:\\me.bmp",-1); // 將src指向當(dāng)前工程文件目錄下的圖像me.bmp 
 cimg.CopyOf(src,src->nChannels);
 
 cimg.DrawToHDC(hdc,&rect);
//輸出圖像
 ReleaseDC( pDC );
 cimg.Destroy();
//銷毀
}

void CopencvtestDlg::OnBnClickedOk()
{
 // TODO: 在此添加控件通知處理程序代碼
 //IplImage *src; // 定義IplImage指針變量src  
 // src = cvLoadImage("D:\\me.bmp",-1); // 將src指向當(dāng)前工程文件目錄下的圖像me.bmp 
 // cvNamedWindow("me",0);//定義一個窗口名為lena的顯示窗口 
 // cvShowImage("me",src);//在lena窗口中，顯示src指針?biāo)赶虻膱D像 
 // cvWaitKey(0);//無限等待，即圖像總顯示 
 // cvDestroyWindow("me");//銷毀窗口lena 
 // cvReleaseImage(&src);//釋放IplImage指針src 
 
 CDC *pDC = GetDlgItem(IDC_STATIC)->GetDC();//根據(jù)ID獲得窗口指針再獲取與該窗口關(guān)聯(lián)的上下文指針
 HDC hdc= pDC->GetSafeHdc();      // 獲取設(shè)備上下文句柄
 CRect rect;
// 矩形類
 GetDlgItem(IDC_STATIC)->GetClientRect(&rect); //獲取box1客戶區(qū)
 
 
 CvCapture *capture = cvCreateFileCapture ("D:\\tree.avi"); //讀取視頻
 if(capture==NULL) {
  printf("NO capture"); //讀取不成功，則標(biāo)識
 //return 1;
 }; 
 double fps=cvGetCaptureProperty(capture, CV_CAP_PROP_FPS ); //讀取視頻的幀率
 int vfps = 1000 / fps;          //計(jì)算每幀播放的時間
 printf("%5.1f\t%5d\n",fps,vfps);        
 double frames=cvGetCaptureProperty(capture,CV_CAP_PROP_FRAME_COUNT);//讀取視頻中有多少幀
 printf("frames is %f\n",frames);
 //cvNamedWindow("example",CV_WINDOW_AUTOSIZE);     //定義窗口
 IplImage *frame;
 
 CvvImage cimg;
 
 while(1){ 
 frame = cvQueryFrame( capture );       //抓取幀
 cimg.CopyOf(frame,frame->nChannels);
 cimg.DrawToHDC(hdc,&rect);
 float ratio = cvGetCaptureProperty(capture, CV_CAP_PROP_POS_AVI_RATIO);  //讀取該幀在視頻中的相對位置
 printf("%f\n",ratio);
 if(!frame)break;
 //cvShowImage("IDC_STATIC",frame);       //顯示
 
 char c = cvWaitKey(vfps);
 if(c == 27 )break;
 }
 ReleaseDC( pDC );
 cvReleaseCapture(&capture);
 cvDestroyWindow("example");
}