在OpenCV中将cv::Mat绘制到MFC的视图中

最新推荐文章于 2020-10-31 16:00:19 发布

bwz2009ss

最新推荐文章于 2020-10-31 16:00:19 发布

阅读量712

点赞数

分类专栏： MFC OpenCV

MFC 同时被 2 个专栏收录

3 篇文章 0 订阅

订阅专栏

OpenCV

2 篇文章 0 订阅

订阅专栏

做毕设时遇到了一个问题，不能在MFC中显示由GrabCut抠出来的图。一番折腾后发现：

在OpenCV中如果图像最初是Mat类型的话，就不能通过转换为IplImage再转换为CvvImage来显示到View里（OpenCV里没有函数能直接将Mat显示到View里），只能通过StretchDIBits来绘制。

下面介绍StretchDIBits函数的使用方法：

1）创建MFC单文档项目（我创建了UseOpenCV），设置包含目录与库目录，添加相关附加依赖项。
2）在文档头文件（UseOpenCVDoc.h）中添加包含文件：

#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>

添加OpenCV Mat类类型成员变量 m_cvImg，形式如下：

protected: // 仅从序列化创建
    CUseOpenCVDoc();
    DECLARE_DYNCREATE(CUseOpenCVDoc)
    cv::Mat     m_cvImg;

3）在文档实现文件（UseOpenCVDoc.cpp）中重写OnOpenDocument函数，如下：

BOOL CUseOpenCVDoc::OnOpenDocument(LPCTSTR lpszPathName)
{
    if (!CDocument::OnOpenDocument(lpszPathName))
        return FALSE;

    if (lpszPathName != NULL)
    {
        // 将lpszPathName 转换成 std::string类型 文件名
#ifdef UNICODE
        // 中文路径时会存在问题
        size_t i = 0;
        char* nstring = new char[(wcslen(lpszPathName)+1)*2];
        wcstombs_s(&i, nstring, (wcslen(lpszPathName)+1)*2, lpszPathName, (wcslen(lpszPathName)+1)*2);
        std::string filename(nstring);
        delete[] nstring;
#else
        std::string filename(lpszPathName);
#endif

        m_cvImg  = cv::imread(filename);
        if (!m_cvImg.data)
        {
            MessageBox(NULL, lpszPathName, _T("图像文件打开失败"), MB_OK);
            return FALSE;
        }
    }

    return TRUE;
}

4）在视图头文件（UseOpenCVView.h）中添加四个成员变量，如下：

protected:
    cv::Mat                   m_cvImg;
    BITMAPINFO*               m_bmi;
    BITMAPINFOHEADER*         m_bmih;
    unsigned int              m_buffer[sizeof(BITMAPINFOHEADER) + sizeof(RGBQUAD)*256];

5）修改视图类（CUseOpenCVView）构造函数如下：

CUseOpenCVView::CUseOpenCVView()
{
    // TODO: 在此处添加构造代码
    // 初始化 BITMAPINFO结构 及 BITMAPINFOHEADER结构
    m_bmi = (BITMAPINFO*) m_buffer;
    m_bmih = &(m_bmi->bmiHeader);
    memset(m_bmih, 0, sizeof(*m_bmih));
    m_bmih->biSize = sizeof(BITMAPINFOHEADER);
}

6）修改视图绘制函数如下：

void CUseOpenCVView::OnDraw(CDC* pDC)
{
    CUseOpenCVDoc* pDoc = GetDocument();
    ASSERT_VALID(pDoc);
    if (!pDoc)
        return;

    // TODO: 在此处为本机数据添加绘制代码
    m_cvImg = pDoc->m_cvImg;

    m_bmih->biWidth = m_cvImg.cols;
    m_bmih->biHeight = -m_cvImg.rows;           // 在自下而上的位图中 高度为负
    m_bmih->biPlanes = 1;
    m_bmih->biCompression = BI_RGB;
    m_bmih->biBitCount = 8 * m_cvImg.channels();
    CRect rect;
    GetClientRect(&rect);
    StretchDIBits(      
        pDC->GetSafeHdc(),
        0, 0, rect.Width(), rect.Height(),
        0, 0, m_cvImg.cols, m_cvImg.rows,
        m_cvImg.data,
        (BITMAPINFO*) m_bmi,
        DIB_RGB_COLORS,
        SRCCOPY);
}

编译运行程序，打开OpenCV支持的图像文件后，图像就绘制在视图中了。