Mat是opencv中的图像格式,如果想要图像在MFC中显示,将Mat格式转为MFC支持的CImage格式是一种较为优秀的解决方案,这篇博客就来介绍在MFC中显示Mat图像的方法。虽然原理相同,但博主还是要介绍两种方式:一种是通过子类化显示控件的方式进行显示,第二种是在父窗口中调用。如果你做的工程比较庞大,博主建议使用子类化控件的方法,当然,如果你是小工程,博主也依然建议你使用子类化的方法,这样不仅使代码结构简单封装良好,而且方便了控件的拓展。
下面直接给出子类化的代码,里面注释已经很详细,直接读代码,所以无需多言。
public: void ShowMat(cv::Mat &); protected: cv::Mat m_img; CImage CI;
如果你没有采用子类化的方法,代码差别极小,只是GetWindowRect的时候需要使用控件对象的指针,下面给出的是子类化的成员函数:
void CILabelControl::ShowMat(cv::Mat & img){ m_img = cv::imread("temp.png"); cv::swap(m_img, img); int w = img.cols;//宽 int h = img.rows;//高 int chinnels = img.channels();//通道数 CI.Destroy();//创建前,最好使用它,防止重复创建,程序崩溃 CI.Create(w, h, 8 * chinnels); //3.下来就是对CI进行赋值了,这里是最核心的地方,分二类讨论 // (1)如果是1个通道的图像(灰度图像) DIB格式才需要对调色板设置 // CImage中内置了调色板,我们要对他进行赋值: RGBQUAD* ColorTable; int MaxColors = 256; //这里可以通过CI.GetMaxColorTableEntries()得到大小(如果你是CI.Load读入图像的话) ColorTable = new RGBQUAD[MaxColors]; CI.GetColorTable(0, MaxColors, ColorTable);//这里是取得指针 for (int i = 0; i<MaxColors; i++) { ColorTable[i].rgbBlue = (BYTE)i; //BYTE和uchar一回事,但MFC中都用它 ColorTable[i].rgbGreen = (BYTE)i; ColorTable[i].rgbRed = (BYTE)i; } CI.SetColorTable(0, MaxColors, ColorTable); delete[]ColorTable; //然后就是数据拷贝了(这里的矩阵表示方法,根据需要(cvMat or Mat)修改): if (chinnels == 1) {//灰度图像 uchar *pS; uchar *pImg = (uchar *)CI.GetBits(); int step = CI.GetPitch(); for (int i = 0; i<h; i++) { pS = img.ptr<uchar>(i); for (int j = 0; j<w; j++) { *(pImg + i*step + j) = pS[j]; } } } //(2)如果是3个通道(彩色图像) //没有调色板,直接赋值 if (chinnels == 3) {//彩色图像 uchar *pS; uchar *pImg = (uchar *)CI.GetBits();//得到CImage数据区地址 int step = CI.GetPitch(); //这个是一行像素站的存储空间w*3,并且结果是4的倍数(这个不用关注,到底是不是4的倍数有待考证) for (int i = 0; i<h; i++) { pS = img.ptr<uchar>(i); for (int j = 0; j<w; j++) { for (int k = 0; k<3; k++) *(pImg + i*step + j * 3 + k) = pS[j * 3 + k]; //注意到这里的step不用乘以3 } } } //4.至此已经构建好CImage,下来就是显示它。我们可以直接在对话框、单文档等地方显示他,还可以使用CPictureCtrl空间显示他。下面给出几个显示方法: //显示前,这里有个问题,等会讨论 //(1)放在一个按钮响应或者函数中 //这里的m_Pic是一个CPictureCtrl的control,其他控件等也一样 //CStatic m_Pic; //DDX_Control(pDX, IDC_STATIC_Img, m_pic); CWnd * pCWnd = CWnd::FromHandle(GetSafeHwnd());//通过变量得到dc比较复杂,但很好用 CPaintDC dc(pCWnd); Invalidate(false); CRect rect; this->GetWindowRect(&rect); // m_pic.GetWind... this->ScreenToClient(&rect); //SetStretchBltMode(dc.m_hDC, COLORONCOLOR); //这个需要百度看看为什么这样设置 CI.StretchBlt(dc.m_hDC, rect, SRCCOPY); //这里显示大小rect(CRect类型)也由自己定义,这个函数有许多重载函数 //图像显示的大小和效果,在你能显示出来后,可以慢慢考虑 //这里的控件的dc还可以由下面方式取得 // CPaintDC dc(GetDlgItem(IDC_IMAGE));//IDC_STATIC_Img是空间的ID //(2)直接显示(下面就写得简单点,少的部分自己加) CDC *pDC = GetDC(); Invalidate(false); CI.StretchBlt(pDC->m_hDC, rect, SRCCOPY); ///或者 // CPaintDC dc(this); // CI.Draw(dc.m_hDC, 0, 0);//这个以某个dc(可以是窗口)的(0,0)为起点 }
使用你的新类来在MFC中进行显示,在对话框的相应函数中使用如下代码,其中m_pic是你新控件类的实例:
m_img = cv::imread("temp.png"); m_pic.ShowMat(m_img);
如果不是采用子类化的方法,如果放在OnPaint中进行绘制,代码是几乎一样的,可能需要修改Rect区域,然后就是如果出现bug,可能需要注释掉Invalid(false); 就这么多。
OK, See You Next Chapter!