opencv Mat 与MFC中的CImage相互转换
MatCImage.h
#pragma once #include <opencv2/core/core.hpp> #include <opencv2/imgproc/imgproc.hpp> #include <opencv2/highgui/highgui.hpp> using namespace cv; using namespace std; class MatCImage { public: /*MatToCImage *简介： * OpenCV的Mat转ATL/MFC的CImage，仅支持单通道灰度或三通道彩色 *参数： * mat：OpenCV的Mat * cimage：ATL/MFC的CImage */ void MatToCImage(Mat& mat, CImage& cimage); /*CImageToMat *简介： * ATL/MFC的CImage转OpenCV的Mat，仅支持单通道灰度或三通道彩色 *参数： * cimage：ATL/MFC的CImage * mat：OpenCV的Mat */ void CImageToMat(CImage& cimage, Mat& mat); // VS默认工程是Unicode编码（宽字节），有时需要ANSI，即单字节，实现宽到单的转化 string CString2StdString(const CString& cstr); // 显示图像到指定窗口 void DisplayImage(CWnd* m_pMyWnd,const CImage& image); // 格式转换，AWX云图转到可以显示的opencv支持的格式 Mat AWX2Mat(CString filePath); void DisplayImageEx(CWnd* pWnd, const CImage& image); };
MatCImage.c
#include "stdafx.h" #include "MatCImage.h" // 实现cv::Mat 结构到 CImage结构的转化 void MatCImage::MatToCImage(Mat& mat, CImage& cimage) { if (0 == mat.total()) { return; } int nChannels = mat.channels(); if ((1 != nChannels) && (3 != nChannels)) { return; } int nWidth = mat.cols; int nHeight = mat.rows; //重建cimage cimage.Destroy(); //这一步是防止重复利用造成内存问题 cimage.Create(nWidth, nHeight, 8 * nChannels); //默认图像像素单通道占用1个字节 //拷贝数据 uchar* pucRow; //指向数据区的行指针 uchar* pucImage = (uchar*)cimage.GetBits(); //指向数据区的指针 int nStep = cimage.GetPitch(); //每行的字节数,注意这个返回值有正有负 // 如果是1个通道的图像(灰度图像) DIB格式才需要对调色板设置 // CImage中内置了调色板，我们要对他进行赋值： if (1 == nChannels) //对于单通道的图像需要初始化调色板 { RGBQUAD* rgbquadColorTable; int nMaxColors = 256; rgbquadColorTable = new RGBQUAD[nMaxColors]; //这里可以通过CI.GetMaxColorTableEntries()得到大小(如果你是CI.Load读入图像的话) cimage.GetColorTable(0, nMaxColors, rgbquadColorTable); //这里是取得指针 for (int nColor = 0; nColor < nMaxColors; nColor++) { //BYTE和uchar一回事，但MFC中都用它 rgbquadColorTable[nColor].rgbBlue = (uchar)nColor; // (BYTE)nColor rgbquadColorTable[nColor].rgbGreen = (uchar)nColor; rgbquadColorTable[nColor].rgbRed = (uchar)nColor; } cimage.SetColorTable(0, nMaxColors, rgbquadColorTable); delete []rgbquadColorTable; } for (int nRow = 0; nRow < nHeight; nRow++) { pucRow = (mat.ptr<uchar>(nRow)); for (int nCol = 0; nCol < nWidth; nCol++) { if (1 == nChannels) { *(pucImage + nRow * nStep + nCol) = pucRow[nCol]; } else if (3 == nChannels) { for (int nCha = 0 ; nCha < 3; nCha++) { *(pucImage + nRow * nStep + nCol * 3 + nCha) = pucRow[nCol * 3 + nCha]; } } } } } void MatCImage::CImageToMat(CImage& cimage, Mat& mat) { if (true == cimage.IsNull()) { return; } int nChannels = cimage.GetBPP() / 8; if ((1 != nChannels) && (3 != nChannels)) { return; } int nWidth = cimage.GetWidth(); int nHeight = cimage.GetHeight(); //重建mat if (1 == nChannels) { mat.create(nHeight, nWidth, CV_8UC1); } else if(3 == nChannels) { mat.create(nHeight, nWidth, CV_8UC3); } //拷贝数据 uchar* pucRow; //指向数据区的行指针 uchar* pucImage = (uchar*)cimage.GetBits(); //指向数据区的指针 int nStep = cimage.GetPitch(); //每行的字节数,注意这个返回值有正有负 for (int nRow = 0; nRow < nHeight; nRow++) { pucRow = (mat.ptr<uchar>(nRow)); for (int nCol = 0; nCol < nWidth; nCol++) { if (1 == nChannels) { pucRow[nCol] = *(pucImage + nRow * nStep + nCol); } else if (3 == nChannels) { for (int nCha = 0 ; nCha < 3; nCha++) { pucRow[nCol * 3 + nCha] = *(pucImage + nRow * nStep + nCol * 3 + nCha); } } } } } // VS默认工程是Unicode编码（宽字节），有时需要ANSI，即单字节，实现宽到单的转化 string MatCImage::CString2StdString(const CString& cstr) { CT2A str(cstr); return string(str.m_psz); } // 显示图像到指定窗口 void MatCImage::DisplayImage(CWnd* m_pMyWnd,const CImage& image) { CDC *m_pDC = m_pMyWnd->GetDC();//获取窗口所拥有的设备上下文，用于显示图像 m_pMyWnd->UpdateWindow(); CRect rc; // m_pMyWnd->GetWindowRect(&rc); /*InvalidateRect(m_pMyWnd->m_hWnd,&rc,true);*/ int nwidth = rc.Width(); int nheight = rc.Height(); int fixed_width = min(image.GetWidth(),nwidth); int fixed_height = min(image.GetHeight(),nheight); double ratio_w = fixed_width / (double)image.GetWidth(); double ratio_h = fixed_height / (double)image.GetHeight(); double ratio = min(ratio_w,ratio_h); int show_width = (int)(image.GetWidth() * ratio); int show_height = (int)(image.GetHeight() * ratio); int offsetx = (nwidth - show_width) / 2; int offsety = (nheight - show_height) / 2; ::SetStretchBltMode(m_pDC->GetSafeHdc(),COLORONCOLOR);//设置位图的伸缩模式 image.StretchBlt(m_pDC->GetSafeHdc(),offsetx,offsety,show_width,show_height, 0,0,image.GetWidth(),image.GetHeight(),SRCCOPY); } void MatCImage::DisplayImageEx(CWnd* pWnd, const CImage& image) { CDC *m_pDC = pWnd->GetDC();//获取窗口所拥有的设备上下文，用于显示图像 pWnd->UpdateWindow(); CRect rc; //客户区大小 //CRect rc1; pWnd->GetWindowRect(&rc); //ScreenToClient(&rc); ::SetStretchBltMode(m_pDC->GetSafeHdc(),COLORONCOLOR);//设置位图的伸缩模式 image.StretchBlt(m_pDC->GetSafeHdc(),0,0,rc.Width()-1,rc.Height()-1, 0,0,image.GetWidth(),image.GetHeight(),SRCCOPY); } // 格式转换，AWX云图转到可以显示的opencv支持的格式 Mat MatCImage::AWX2Mat(CString filePath) { CFile fp; Mat mat; fp.Open(filePath,CFile::modeRead); ULONGLONG flength = fp.GetLength(); if (2475700 == flength) { mat.create(1300,1900,CV_8UC1); } else if (1444803 == flength) { mat.create(1201,1201,CV_8UC1); } LONGLONG size = mat.rows * mat.cols; LONGLONG sizebuff = fp.Seek(-size, CFile::end); uchar *pSource = new uchar[size]; fp.Read(pSource,size); fp.Close(); for (int i = 0; i<mat.rows; i++) { uchar * ps = mat.ptr<uchar>(i); for (int j = 0 ; j < mat.cols; j++) { ps[j] = *(pSource + i*mat.cols + j); } } delete pSource; return mat; }
