OpenCV系列1

OpenCV系列1

先看图：
拉伸显示

居中显示

原始大小显示

显示视频：

实现图片显示，只需要两句话：

只需要包含一个h文件：

调整显示方式：

下面是CvImgCtrl.h的源代码，复制保存下来即可用，或者也可以下载demo，不要积分
控件采用的是双缓冲绘图，显示效率很高，不会闪烁，支持视频、图片的显示

/************************************************************************/
/*                              OpenCV MFC图像显示控件 v1.0
/*    该控件很容易集成到mfc中显示，具体用法见demo：
/*                                                                               Wish: 512690069
/*                                                                      2016年4月6日 15:28:58 
/************************************************************************/#pragma once#include <cv.h>
#include <highgui.h>
#include <afxwin.h>//自定义消息基础值
#define MsgBase                 (WM_USER + 0xA0)
#define MsgLBDown           (MsgBase + 0)                   //鼠标左键被按下的消息
#define MsgLBUp                 (MsgBase + 1)                   //左键抬起来
#define MsgRBDown           (MsgBase + 2)                   //右键按下
#define MsgMove             (MsgBase + 3)                   //鼠标移动#define MsgMBDown           (MsgBase + 4)                   //中键按下
#define MsgMBUp             (MsgBase + 5)                   //中键抬起#define MsgKeyDown          (MsgBase + 6)                   //按下某个键盘按键  
#define MsgDragFile             (MsgBase + 7)                   //拖动文件到控件上去的时候#define MsgVideoStart           (MsgBase + 8)                   //视频开始的时候
#define MsgVideoOver            (MsgBase + 9)                   //视频结束了//得到一帧视频画面了，具体看demo
//1、在dlg的头文件中写：afx_msg LRESULT onFrame(WPARAM wp, LPARAM lp);声明
//2、在dlg的BEGIN_MESSAGE_MAP里面写：ON_MESSAGE(MsgVideoFrame, onFrame)
//3、在dlg的cpp中写：
/*
LRESULT CdemoDlg::onFrame(WPARAM wp, LPARAM lp){Mat frame((IplImage*)lp, false);flip(frame, frame, -1);//这里就是在处理每一帧的图像了，类似的MsgVideoPreDisplay消息也是这么来的return 0;
}
*/
#define MsgVideoFrame           (MsgBase + 0x0A)            
#define MsgVideoPreDisplay  (MsgBase + 0x0B)        //视频一帧画面准备显示了，他会在MsgVideoFrame之后//控件被单击消息
#define MsgClick                            (MsgBase + 0x0C)//内部消息，内部使用，不需要理会
#define InnerMsgBase                    (MsgBase + 0xF0)                    //基本消息起点
#define InnerMsgVideoFrame      (InnerMsgBase + 0)              //接受到一帧画面//图像显示控件
class CvImgCtrl : public CWnd
{DECLARE_DYNAMIC(CvImgCtrl)public://视图缩放类型enum ResizeType{ResizeType_Raw,                 //原始大小，以左上角为准，图像多大就多大，不做缩放ResizeType_CenterResize,        //居中缩放，图像宽高等比例，画面居中ResizeType_Tensile              //拉伸，宽高拉伸到控件一样大};//视频播放的状态enum VideoState{VideoState_Stop,                    //视频停止了VideoState_Playing,             //视频播放中VideoState_Pause                    //视频暂停了};public:CvImgCtrl();virtual ~CvImgCtrl();//链接到对话框上的控件，链接过后，就被这个类接管了，他会负责将画面绘制到控件上去//idd：控件的id//parent：父控件的指针，比如在dialog里面，给this就行了BOOL linkDlgItem(UINT idd, CWnd* parent);//创建一个控件，如果需要动态创建，就调用这个创建函数，而不是CWnd的创建函数//pParent：父级的指针，一般给dialog的this指针就好了//rcPos：控件所在位置，可以直接给CRect()也行，然后在OnSize里边调整宽高，或者给指定值也行//autoDelete：如果CvImgCtrl* m_show，这种，指针，采用new的方式得到的控件，而不是实体变量，那么当这个参数给true的时候，控件销毁时//                    会自己回收内存（delete this），也就是你就不用手动的去delete他了//resizeType：缩放方式，当显示图像和窗口大小不一致时，该如何缩放，有3种方式，看枚举类型ResizeType的备注//defaultColor：默认颜色，即背景默认色，比如居中显示的时候，四周会留空，该是什么颜色，或者刚创建完毕时是什么颜色等bool create(CWnd* pParent, CRect rcPos, bool autoDelete = false, ResizeType resizeType = ResizeType_Raw, CvScalar defaultColor = cvScalarAll(255));//开始视频，这个控件内置了VideoCapture，所以这里直接可以调用openVideo来打开视频//file：视频文件路径，直接是VideoCapture.open，如果失败格式不支持，那就去看VideoCapture//waitTime：延迟时间，每一帧画面需要延迟的时间，单位毫秒，如果不延迟，画面会很快的，当然你可以改改成从VideoCapture中获取的bool openVideo(const char* file, unsigned int waitTime = 30);//开始视频摄像头，这个控件内置了VideoCapture，所以这里直接可以调用openVideo来打开视频//cameraIndex：摄像头索引，一样是VideoCapture的open函数，opencv默认-1会弹出选择视频对话框，0-n是你电脑n个摄像头的序号，他们有固定顺序的//waitTime：延迟时间，每一帧画面需要延迟的时间，单位毫秒，如果不延迟，画面会很快的，当然你可以改改成从VideoCapture中获取的bool openVideo(int cameraIndex = 0, unsigned int waitTime = 30);//暂停视频void pauseVideo(); //继续视频void resumeVideo();//关闭视频void closeVideo();//设置图像，设置后控件会刷新显示，会执行onUpdateImage，允许你设置任意图像，内部会自动调整//如果设置的图像是相同大小、相同通道的图像，则只会进行copy，所以效率是杠杠的void setImage(const cv::Mat img){setImage(&IplImage(img));}//设置图像ipl方式，设置后控件会刷新显示，会执行onUpdateImagevoid setImage(const IplImage* img);//获取当前图像cv::Mat getImageMat(){return cv::Mat(m_img, false);}//获取当前图像IplImage* getImage(){return m_img;}//更新显示图像，如果你通过getImageMat得到的Mat或者IplImage，你修改了他，那么请使用onUpdateImage来刷新画面应用到控件上void onUpateImage();//获取显示图像，显示图像即跟控件一样大小的那个图像，随着控件大小改变而改变IplImage* getDisplayImage(){return m_imgResize;}//获取显示图像Matcv::Mat getDisplayImageMat(){return cv::Mat(m_imgResize, false);}//更新显示图像，当你通过getDisplayImage获取到的图像，并修改了他，那么请使用这个函数来刷新应用到画面上void onUpdateDisplayImage();//设置缩放方式，共3种方式，见ResizeType的定义上写的备注，设置后立马生效并刷新void setResizeType(ResizeType rt){m_rt = rt; onUpateImage();}//设置数据，存一个int，以备特殊时候使用，比如要存什么id之类的，指针之类的void setData(int data){m_saveData = data;}int getData(){return m_saveData;}//获取视频捕获器cv::VideoCapture* getVideoCapture(){return &m_videoCapture;}//返回当前播放视频状态VideoState getVideoState(){return m_videoState;}private://设置图像，只是设置图像，不会显示到控件上去void _onlySetImage(const IplImage* img);//启动视频，device_or_file，如果给定-1或者0-1024以内的数字，则认为是camera//否则认为是文件路径，waittime则是帧延迟时间bool _startVideo(const void* device_or_file, unsigned int waitTime = 30);//重画image到displayimage（即resizeimage）void redrawResize();//清空显示的那个图像void voidResize();//通知父亲，默认wp是thisLRESULT notifyParent(UINT msg, LPARAM lp = 0);private:cv::VideoCapture m_videoCapture;VideoState m_videoState;int m_videoWaitTime;  unsigned int m_lastdownTime;int m_saveData;IplImage* m_img;IplImage* m_imgResize;BITMAPINFO m_bmpHeader;bool m_autoDelete;CvScalar m_defaultColor;ResizeType m_rt;CRITICAL_SECTION m_resizeCS;protected:DECLARE_MESSAGE_MAP()
public:afx_msg void OnSize(UINT nType, int cx, int cy);afx_msg void OnDestroy();afx_msg BOOL OnEraseBkgnd(CDC* pDC);afx_msg void OnPaint();afx_msg void OnNcDestroy();afx_msg void OnMouseMove(UINT nFlags, CPoint point);afx_msg void OnLButtonDown(UINT nFlags, CPoint point);afx_msg void OnLButtonUp(UINT nFlags, CPoint point);afx_msg void OnRButtonDown(UINT nFlags, CPoint point);afx_msg LRESULT OnNcHitTest(CPoint point);afx_msg void OnMButtonDown(UINT nFlags, CPoint point);afx_msg void OnMButtonUp(UINT nFlags, CPoint point);afx_msg void OnKeyDown(UINT nChar, UINT nRepCnt, UINT nFlags);afx_msg void OnDropFiles(HDROP hDropInfo);afx_msg LRESULT OnVideoFrame(WPARAM wp, LPARAM lp);
};#define BEGIN_MESSAGE_MAP_EX(class_name)                PTM_WARNING_DISABLE                                                 inline const AFX_MSGMAP* class_name::GetMessageMap() const  
{return GetThisMessageMap();}            inline const AFX_MSGMAP* PASCAL class_name::GetThisMessageMap()     
{typedef class_name ThisClass;                          typedef CWnd TheBaseClass;                          static const AFX_MSGMAP_ENTRY _messageEntries[] = {#define IMPLEMENT_DYNAMIC_EX(class_name, super_class)           
AFX_COMDAT const CRuntimeClass class_name::class##class_name = {    
#class_name, sizeof(class class_name), 0xFFFF, NULL,                        
((CRuntimeClass*)(&super_class::class##super_class)), NULL, NULL };     
inline CRuntimeClass* class_name::GetRuntimeClass() const 
{ return RUNTIME_CLASS(class_name); }IMPLEMENT_DYNAMIC_EX(CvImgCtrl, CWnd)inline static IplImage imgOfROI(const IplImage* img, CvRect roi, bool invalidROI=false)
{IplImage ret = *img;int imgWidth = img->roi != 0 && !invalidROI ? img->roi->width : img->width;int imgHeight = img->roi != 0 && !invalidROI ? img->roi->height : img->i = 0;if(roi.x >= imgWidth || roi.y >= imgHeight){ret.width = 0;ret.height = 0;ret.imageData = 0;return ret;}//区域裁剪if(roi.x < 0){roi.width += roi.x;roi.x = 0;}if(roi.y < 0) {roi.height += roi.y;roi.y = 0;}if(roi.x + roi.width > imgWidth)        roi.width = imgWidth - roi.x;if(roi.y + roi.height > imgHeight)      roi.height = imgHeight - roi.y;CV_Assert(roi.width > 0 && roi.height > 0);ret.width = roi.width;ret.height = roi.height;ret.imageSize = ret.height * ret.widthStep;ret.imageDataOrigin = 0;if (!invalidROI && img->roi != 0)ret.imageData = img->imageData + (img->roi->xOffset + roi.x) * img->nChannels * (img->depth & ~IPL_DEPTH_SIGN) / 8 + (img->roi->yOffset + roi.y) * img->widthStep;elseret.imageData = img->imageData + roi.x * img->nChannels * (img->depth & ~IPL_DEPTH_SIGN) / 8 + roi.y * img->widthStep;return ret;
}inline static void doEvent(unsigned int waitMS = -1){MSG msg;unsigned int time = GetTickCount();while(GetTickCount() - time < waitMS){while(GetTickCount() - time < waitMS && PeekMessage(&msg, 0, 0, 0, PM_REMOVE)){ssage == WM_QUIT)break;TranslateMessage(&msg);DispatchMessage(&msg);}}
}#define min(a,b) (((a) < (b)) ? (a) : (b))inline CvImgCtrl::CvImgCtrl(){memset(&m_bmpHeader, 0, sizeof(m_bmpHeader));m_bmpHeader.bmiHeader.biSize = sizeof(m_bmpHeader);m_bmpHeader.bmiHeader.biXPelsPerMeter = 72;m_bmpHeader.bmiHeader.biYPelsPerMeter = 72;m_bmpHeader.bmiHeader.biPlanes = 1;m_imgResize = 0;m_img = 0;m_rt = ResizeType_Tensile;m_autoDelete = false;m_videoState = VideoState_Stop;m_defaultColor = cvScalarAll(255);InitializeCriticalSection(&m_resizeCS);
}inline CvImgCtrl::~CvImgCtrl(){DeleteCriticalSection(&m_resizeCS);
}BEGIN_MESSAGE_MAP_EX(CvImgCtrl)ON_WM_SIZE()ON_WM_DESTROY()ON_WM_ERASEBKGND()ON_WM_PAINT()ON_WM_NCDESTROY()ON_WM_MOUSEMOVE()ON_WM_LBUTTONDOWN()ON_WM_LBUTTONUP()ON_WM_NCHITTEST()ON_WM_RBUTTONDOWN()ON_WM_MBUTTONDOWN()ON_WM_MBUTTONUP()ON_WM_KEYDOWN()ON_WM_DROPFILES()ON_MESSAGE(InnerMsgVideoFrame, OnVideoFrame)
END_MESSAGE_MAP()inline bool CvImgCtrl::openVideo(const char* file, unsigned int waitTime){return _startVideo(file, waitTime);
}inline void CvImgCtrl::pauseVideo(){if(m_videoState == VideoState_Playing)m_videoState = VideoState_Pause;
}inline bool CvImgCtrl::openVideo(int cameraIndex, unsigned int waitTime){return _startVideo((const void*)cameraIndex, waitTime);
}inline void CvImgCtrl::resumeVideo(){if(m_videoState == VideoState_Pause){m_videoState = VideoState_Playing;PostMessage(InnerMsgVideoFrame, 0, 0);}
}inline void CvImgCtrl::closeVideo(){m_videoState = VideoState_Stop;if(m_videoCapture.isOpened())lease();
}inline bool CvImgCtrl::_startVideo(const void* device_or_file, unsigned int waitTime){if(!::IsWindow(m_hWnd)) return false;int dof = (int)device_or_file;if(m_videoCapture.isOpened())lease();bool success = false;const int max_count_of_device_index = 1024;const char* file = (const char*)device_or_file;const int device_index = (int)device_or_file;m_videoWaitTime = waitTime;if(dof == -1 || dof >= 0 && dof <= max_count_of_device_index)success = m_videoCapture.open(device_index);elsesuccess = m_videoCapture.open(file);m_videoState = success ? VideoState_Playing : VideoState_Stop;if(success) {cv::Mat frame;m_videoCapture >> frame;if(!pty()){_onlySetImage(&IplImage(frame));notifyParent(MsgVideoStart, (LPARAM)device_or_file);notifyParent(MsgVideoFrame, (LPARAM)m_img);if (m_imgResize != 0){redrawResize();notifyParent(MsgVideoPreDisplay, (LPARAM)m_imgResize);this->onUpdateDisplayImage();}if(m_videoState == VideoState_Playing)PostMessage(InnerMsgVideoFrame, 0, 0);}else{success = false;}}m_videoState = success ? VideoState_Playing : VideoState_Stop;return success;
}inline LRESULT CvImgCtrl::notifyParent(UINT msg, LPARAM lp){return ::SendMessage(GetParent()->m_hWnd, msg, (WPARAM)this, lp);
}inline LRESULT CvImgCtrl::OnVideoFrame(WPARAM wp, LPARAM lp){if(m_videoState == VideoState_Playing){bool grab = ab();bool retrieve = grab && ieve(cv::Mat(m_img, false));if(retrieve){notifyParent(MsgVideoFrame, (LPARAM)m_img);if (m_imgResize != 0){redrawResize();notifyParent(MsgVideoPreDisplay, (LPARAM)m_imgResize);this->onUpdateDisplayImage();}if(m_videoState == VideoState_Playing){doEvent(m_videoWaitTime);PostMessage(InnerMsgVideoFrame, wp, lp);}}else{notifyParent(MsgVideoOver);m_videoState = VideoState_Stop;}}return 0;
}inline bool CvImgCtrl::create(CWnd* pParent, CRect rcPos, bool autoDelete /*= false*/,ResizeType rt /*= _ResizeType_Non*/, CvScalar defaultColor /*= cvScalarAll(255)*/){m_defaultColor = defaultColor;m_autoDelete = autoDelete;m_rt = rt;return CWnd::Create(_T("Static"), 0, WS_OVERLAPPED | WS_CHILD | WS_VISIBLE, rcPos, pParent, 0, 0) != 0;
}inline void CvImgCtrl::redrawResize(){if(m_imgResize == 0 || m_img == 0 || m_img->width * m_img->height == 0)return;EnterCriticalSection(&m_resizeCS);switch(m_rt)    {case ResizeType_Tensile:    {if(m_img->nChannels != 3){IplImage* re = cvCreateImage(cvGetSize(m_img), 8, 3);cvCvtColor(m_img, re, CV_GRAY2BGR);cvResize(re, m_imgResize);cvReleaseImage(&re);}elsecvResize(m_img, m_imgResize);}break;case ResizeType_Raw: {voidResize();CvRect roi = {0, 0, min(m_img->width, m_imgResize->width), min(m_img->height, m_imgResize->height)};if(m_img->nChannels != 3)cvCvtColor(&imgOfROI(m_img, roi), &imgOfROI(m_imgResize, roi), CV_GRAY2BGR);elsecvCopy(&imgOfROI(m_img, roi), &imgOfROI(m_imgResize, roi));}break;case ResizeType_CenterResize: {voidResize();int nw = m_imgResize->width;int nh = int(m_imgResize->width / (double)m_img->width * m_img->height);if (nh > m_imgResize->height)   {nh = m_imgResize->height;nw = int(m_imgResize->height / (double)m_img->height * m_img->width);}cvSetImageROI(m_imgResize, cvRect((m_imgResize->width - nw) / 2, (m_imgResize->height - nh) / 2, nw, nh));if(m_img->nChannels != 3){IplImage* re = cvCreateImage(cvGetSize(m_img), 8, 3);cvCvtColor(m_img, re, CV_GRAY2BGR);cvResize(re, m_imgResize);cvReleaseImage(&re);}elsecvResize(m_img, m_imgResize);cvResetImageROI(m_imgResize);}break;}LeaveCriticalSection(&m_resizeCS);
}inline void CvImgCtrl::onUpateImage(){if (m_imgResize != 0 && ::IsWindow(m_hWnd)){redrawResize();this->onUpdateDisplayImage();}
}inline void CvImgCtrl::voidResize(){cvSet(m_imgResize, m_defaultColor);
}inline BOOL CvImgCtrl::linkDlgItem(UINT idd, CWnd* parent){BOOL r = CWnd::SubclassDlgItem(idd, parent);if(r){CRect rc;GetWindowRect(&rc);OnSize(0, rc.Width(), rc.Height());}return r;
}inline void CvImgCtrl::OnSize(UINT nType, int cx, int cy){if (cx < 1 || cy < 1){CWnd::OnSize(nType, cx, cy);return;}EnterCriticalSection(&m_resizeCS);cvReleaseImage(&m_imgResize);m_imgResize = cvCreateImage(cvSize(cx, cy), 8, 3);LeaveCriticalSection(&m_resizeCS);if(m_img != 0)redrawResize();elsevoidResize();m_bmpHeader.bmiHeader.biBitCount = m_imgResize->nChannels * m_imgResize->depth;m_bmpHeader.bmiHeader.biHeight = -m_imgResize->height;m_bmpHeader.bmiHeader.biWidth = m_imgResize->width;m_bmpHeader.bmiHeader.biSizeImage = m_imgResize->imageSize;this->onUpdateDisplayImage();
}inline void CvImgCtrl::_onlySetImage(const IplImage* img){if (img == 0)   {if (m_img != 0)cvReleaseImage(&m_img);if (m_imgResize != 0)voidResize();return;}if (m_img == 0){m_img = (IplImage*)cvClone(img);}else{if (m_img->width != img->width || m_img->height != img->height || m_img->nChannels != img->nChannels || m_img->depth != img->depth){cvReleaseImage(&m_img);m_img = (IplImage*)cvClone(img);}elsememcpy(m_img->imageData, img->imageData, min(img->imageSize, m_img->imageSize));//cvCopy(img, m_img);}
}inline void CvImgCtrl::setImage(const IplImage* img){_onlySetImage(img);onUpateImage();
}inline void CvImgCtrl::OnDestroy(){CWnd::OnDestroy();if (m_img != 0)cvReleaseImage(&m_img);if (m_imgResize != 0)cvReleaseImage(&m_imgResize);
}inline BOOL CvImgCtrl::OnEraseBkgnd(CDC* pDC){return TRUE;
}inline void CvImgCtrl::onUpdateDisplayImage(){Invalidate();UpdateWindow();
}inline void CvImgCtrl::OnPaint(){if(m_imgResize != 0){PAINTSTRUCT ps;HDC hDC = ::BeginPaint(m_hWnd, &ps);CRect rcPaint = ps.rcPaint;StretchDIBits(hDC, 0, 0, m_imgResize->width, m_imgResize->height, 0, 0, m_imgResize->width, m_imgResize->height, m_imgResize->imageData, &m_bmpHeader, DIB_RGB_COLORS, SRCCOPY);::EndPaint(m_hWnd, &ps);}elseCWnd::OnPaint();
}inline void CvImgCtrl::OnNcDestroy(){CWnd::OnNcDestroy();if (m_autoDelete)delete this;
}inline void CvImgCtrl::OnMouseMove(UINT nFlags, CPoint point){notifyParent(MsgMove, (LPARAM)&point);
}inline void CvImgCtrl::OnLButtonDown(UINT nFlags, CPoint point){m_lastdownTime = ::GetTickCount();notifyParent(MsgLBDown, (LPARAM)&point);
}inline void CvImgCtrl::OnLButtonUp(UINT nFlags, CPoint point){notifyParent(MsgLBUp, (LPARAM)&point);if(GetTickCount() - m_lastdownTime <= 300)notifyParent(MsgClick, (LPARAM)&point);
}inline void CvImgCtrl::OnRButtonDown(UINT nFlags, CPoint point){notifyParent(MsgRBDown, (LPARAM)&point);
}inline LRESULT CvImgCtrl::OnNcHitTest(CPoint point){return HTCLIENT;
}inline void CvImgCtrl::OnMButtonDown(UINT nFlags, CPoint point){notifyParent(MsgMBDown, (LPARAM)&point);
}inline void CvImgCtrl::OnMButtonUp(UINT nFlags, CPoint point){notifyParent(MsgMBUp, (LPARAM)&point);
}inline void CvImgCtrl::OnKeyDown(UINT nChar, UINT nRepCnt, UINT nFlags){notifyParent(MsgKeyDown, nChar);
}inline void CvImgCtrl::OnDropFiles(HDROP hDropInfo){CWnd::OnDropFiles(hDropInfo);notifyParent(MsgDragFile, (LPARAM)hDropInfo);
}