OpenCV 2.4:在MFC中显示cv::Mat

OpenCV 2.4 : Displaying a cv::Mat in MFC

本文关键字:显示 cv Mat OpenCV MFC      更新时间:2023-10-16

我正在将现有的代码库从IplImage*更新到较新的cv::Mat。我想知道如何向 MFC 显示我的 cv::Mat 对象。我们当前使用的解决方案基于旧的 CvvImage 类:

void DrawPicToHDC(IplImage *img, UINT ID, bool bOnPaint)
{
    CDC *pDC = GetDlgItem(ID)->GetDC();
    HDC hDC= pDC->GetSafeHdc();
    CRect rect;
    GetDlgItem(ID)->GetClientRect(&rect);
    CvvImage cimg;
    cimg.CopyOf( img );
    cimg.DrawToHDC( hDC, &rect );
    ReleaseDC( pDC );
}

遇到了这个线程,但不幸的是提供的答案不能满足我的需求,因为答案仍然要求您在显示之前将垫子转换为 IplImage*。

有没有办法只使用 cv::Mat 来做到这一点?任何帮助都非常感谢。

更新:我在 Kornel 的答案的帮助下将上述函数调整为 cv::Mat 使用。我现在不需要包含 CvvImage 类:

void DrawPicToHDC(cv::Mat cvImg, UINT ID, bool bOnPaint)
{
    // Get the HDC handle information from the ID passed
    CDC *pDC = GetDlgItem(ID)->GetDC();
    HDC hDCDst = pDC->GetSafeHdc();
    CRect rect;
    GetDlgItem(ID)->GetClientRect(&rect);
    cv::Size winSize(rect.right, rect.bottom);
    // Resize the source to the size of the destination image if necessary
    cv::Mat cvImgTmp(winSize, CV_8UC3);
    if (cvImg.size() != winSize)
    {
        cv::resize(cvImg, cvImgTmp, winSize);
    }
    else
    {
        cvImgTmp = cvImg;
    }
    // Rotate the image
    cv::flip(cvImgTmp,cvImgTmp,0);
    // Initialize the BITMAPINFO structure
    BITMAPINFO bitInfo;
    bitInfo.bmiHeader.biBitCount = 24;
    bitInfo.bmiHeader.biWidth = winSize.width;
    bitInfo.bmiHeader.biHeight = winSize.height;
    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;
    // Add header and OPENCV image's data to the HDC
    StretchDIBits(hDCDst, 0, 0,
        winSize.width, winSize.height, 0, 0,
        winSize.width, winSize.height,
        cvImgTmp.data, &bitInfo, DIB_RGB_COLORS, SRCCOPY);
    ReleaseDC( pDC );
}

假设我们有一个OpenCV图像cv::Mat cvImg应该将其转换为CImage* mfcImg.
当然,MFC 图像应该显示在 MFC 窗口中,即CStatic winImg

因此,应执行以下转换才能在 MFC 窗口中显示 OpenCV 图像:

cv::Mat -> CImage -> CStatic

定义 MFC 窗口大小:

RECT r;
winImg.GetClientRect(&r);
cv::Size winSize(r.right, r.bottom);

cvImg的大小并不总是与 MFC 窗口的大小相同:

cv::Mat cvImgTmp(winSize, CV_8UC3);
if (cvImg.size() != winSize)
{
    cv::resize(cvImg, cvImgTmp, winSize);
}
else 
{
    cvImgTmp = cvImg.clone();
}

旋转图像:

cv::flip(cvImgTmp, cvImgTmp, 0);

创建 MFC 映像:

if (mfcImg)
{
    mfcImg->ReleaseDC();
    delete mfcImg; mfcImg = nullptr;
}
mfcImg = new CImage();
mfcImg->Create(winSize.width, winSize.height, 24);

对于mfcImg,您需要一个标头。使用BITMAPINFO结构创建它

BITMAPINFO bitInfo;
bitInfo.bmiHeader.biBitCount = 24;
bitInfo.bmiHeader.biWidth = winSize.width;
bitInfo.bmiHeader.biHeight = winSize.height;
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;

将标题和 OpenCV 图像的数据添加到mfcImg

StretchDIBits(mfcImg->GetDC(), 0, 0,
    winSize.width, winSize.height, 0, 0,
    winSize.width, winSize.height,
    cvImgTmp.data, &bitInfo, DIB_RGB_COLORS, SRCCOPY
);

在 MFC 窗口中显示mfcImg

mfcImg->BitBlt(::GetDC(winImg.m_hWnd), 0, 0);

释放mfcImg,如果您不打算使用它:

if (mfcImg)
{
    mfcImg->ReleaseDC();
    delete mfcImg; mfcImg = nullptr;
}