/*! 在MFC的Picture Control控件中显示Mat数据
*! Mat img: 需要显示的Mat数据
*! UINT nID: 控件的ID号
*! 支持直接传入 BGR ,BGRA, GARY 三种颜色空间的Mat图
*! 如果你的Mat是其它颜色空间的,在调用该函数之前使用cvtColor转换为对应的颜色空间即可
*/
void CxxxDlg::DrawMat(cv::Mat srcImg, UINT nID)
{
cv::Mat imgTmp;
CRect rect;
GetDlgItem(nID)->GetClientRect(&rect);
cv::resize(srcImg, imgTmp, cv::Size(rect.Width(), rect.Height()));
// 统一转换为 CV_8UC4
switch (imgTmp.channels())
{
case 1:
cv::cvtColor(imgTmp, imgTmp, CV_GRAY2BGRA); // GRAY --> BGRA(CV_8UC4)
break;
case 3:
cv::cvtColor(imgTmp, imgTmp, CV_BGR2BGRA); // BGR --> BGRA(CV_8UC4)
break;
default:
break;
}
// 计算一个像素多少个字节// 其实这里可以换成常量
int pixelBytes = imgTmp.channels()*(imgTmp.depth() + 1);
// 使用 BITMAPINFO 结构体创建一个数据头
BITMAPINFO bitInfo;
bitInfo.bmiHeader.biBitCount = 8 * pixelBytes; // 可以换成常量24
bitInfo.bmiHeader.biWidth = imgTmp.cols;
bitInfo.bmiHeader.biHeight = -imgTmp.rows;
bitInfo.bmiHeader.biPlanes = 1;
bitInfo.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
bitInfo.bmiHeader.biCompression = BI_RGB;
bitInfo.bmiHeader.biClrImportant = 0;
bitInfo.bmiHeader.biClrUsed = 0;
bitInfo.bmiHeader.biSizeImage = 0;
bitInfo.bmiHeader.biXPelsPerMeter = 0;
bitInfo.bmiHeader.biYPelsPerMeter = 0;
// 显示 Mat
// Mat data + BITMAPINFO --> MFC DigItem
CDC *pDC = GetDlgItem(nID)->GetDC();
::StretchDIBits(
pDC->GetSafeHdc(),
0, 0, rect.Width(), rect.Height(),
0, 0, rect.Width(), rect.Height(),
imgTmp.data,
&bitInfo,
DIB_RGB_COLORS,
SRCCOPY
);
ReleaseDC(pDC);
}
/*
为什么转换为 4通道 呢?
MFC显示图片存在一个四字节对齐的问题,
假设 一行如果是5个像素
CV_8UC3下,一个像素是3个字节,一行共15个字节,为了4字节对齐,你需要手动计算每行的字节数,然后在后面补 0,这里则需要补1字节的0.
CV_8UC4下,一个像素是4个字节,天然的4字节对齐.
不补0 就图片错位,控件比图片大好像没问题,如果控件比图片小,就会错位.
*/
参考:
http://blog.csdn.net/backspace110/article/details/52704998