如何将CBitmap转换为cv :: Mat?

时间:2018-02-06 20:28:55

标签: c++ opencv-mat cbitmap

如何将CBitmap转换为cv :: Mat?也许有一些libs或其他东西...... 像...

CBitmap bitmap;
bitmap.CreateBitmap(128, 128, 1, 24, someData);
cv::Mat outBitmap(128,128,someData,1,24);

但该代码不正确。

谢谢!

1 个答案:

答案 0 :(得分:1)

还有另外一种方法,您可以将CBitmap转换为HBitmap,然后将HBitmap转换为GdiPlus::Bitmap,然后将其转换为cv::Mat。 这是您可以做的,但要注意,此解决方案仅适用于RGB24像素格式

第1步: CBitmapHBITMAP

HBITMAP hBmp = (HBITMAP)yourCBitmap.GetSafeHandle();

第2步: HBITMAPGdiplus::Bitmap(从this问题复制)

#include <GdiPlus.h>
#include <memory>

Gdiplus::Status HBitmapToBitmap( HBITMAP source, Gdiplus::PixelFormat pixel_format, Gdiplus::Bitmap** result_out )
{
  BITMAP source_info = { 0 };
  if( !::GetObject( source, sizeof( source_info ), &source_info ) )
    return Gdiplus::GenericError;

  Gdiplus::Status s;

  std::auto_ptr< Gdiplus::Bitmap > target( new Gdiplus::Bitmap( source_info.bmWidth, source_info.bmHeight, pixel_format ) );
  if( !target.get() )
    return Gdiplus::OutOfMemory;
  if( ( s = target->GetLastStatus() ) != Gdiplus::Ok )
    return s;

  Gdiplus::BitmapData target_info;
  Gdiplus::Rect rect( 0, 0, source_info.bmWidth, source_info.bmHeight );

  s = target->LockBits( &rect, Gdiplus::ImageLockModeWrite, pixel_format, &target_info );
  if( s != Gdiplus::Ok )
    return s;

  if( target_info.Stride != source_info.bmWidthBytes )
    return Gdiplus::InvalidParameter; // pixel_format is wrong!

  CopyMemory( target_info.Scan0, source_info.bmBits, source_info.bmWidthBytes * source_info.bmHeight );

  s = target->UnlockBits( &target_info );
  if( s != Gdiplus::Ok )
    return s;

  *result_out = target.release();

  return Gdiplus::Ok;
}

调用此函数并将HBITMAP传递给它。

第3步: Gdiplus::Bitmapcv::Mat

cv::Mat GdiPlusBitmapToCvMat(Gdiplus::Bitmap* bmp)
{
    auto format = bmp->GetPixelFormat();
    if (format != PixelFormat24bppRGB)
        return cv::Mat();

    int width = bmp->GetWidth();
    int height = bmp->GetHeight();
    Gdiplus::Rect rcLock(0, 0, width, height);
    Gdiplus::BitmapData bmpData;

    if (!bmp->LockBits(&rcLock, Gdiplus::ImageLockModeRead, format, &bmpData) == Gdiplus::Ok)
        return cv::Mat();

    cv::Mat mat = cv::Mat(height, width, CV_8UC3, static_cast<unsigned char*>(bmpData.Scan0), bmpData.Stride).clone();

    bmp->UnlockBits(&bmpData);
    return mat;
}

将您在上一步中创建的Gdiplus::Bitmap传递给此功能,您将获得cv:Mat。正如我之前所说,这个功能只适用于RGB24像素格式。