C++保存HBITMAP为位图文件的实现方法

时间:2021-05-20

本文使用C++将位图句柄HBITMAP保存为位图文件,配合C++抓图代码可以实现抓图保存文件(.bmp)。

其步骤如下:

1、创建位图文件;
2、计算位图中每个像素所占字节数;
3. 获取位图结构BITMAP;
4、构造位图信息头BITMAPINFOHEADER;
5、构造位图文件头BITMAPFILEHEADER;
6、为位图内容分配内存;
7、处理调色板;
8、写入文件;
9、清除资源。

下面是C++源代码:

ImageHelper.h

#pragma once#include <windows.h>#include <string>using namespace std;class ImageHelper{public:static bool SaveBitmapToFile(HBITMAP bitmap, const string& filename); //保存位图到文件private:static WORD GetBitmapBitCount(); //计算位图文件每个像素所占字节数static void ProcessPalette(HBITMAP hBitmap, const BITMAP& bitmap,DWORD paletteSize, LPBITMAPINFOHEADER lpBmpInfoHeader); //处理调色板};

ImageHelper.cpp

#include "ImageHelper.h"#include <shlwapi.h>bool ImageHelper::SaveBitmapToFile(HBITMAP hBitmap, const string& filename){//1. 创建位图文件const auto file = CreateFileA(filename.c_str(), GENERIC_WRITE,0, nullptr, CREATE_ALWAYS,FILE_ATTRIBUTE_NORMAL | FILE_FLAG_SEQUENTIAL_SCAN,nullptr);if (file == INVALID_HANDLE_VALUE){return false;}//2. 计算位图文件每个像素所占字节数const auto bitCount = GetBitmapBitCount();//3. 获取位图结构BITMAP bitmap;::GetObject(hBitmap, sizeof(bitmap), reinterpret_cast<LPSTR>(&bitmap));//位图中像素字节大小(32字节对齐)const DWORD bmBitsSize = ((bitmap.bmWidth * bitCount + 31) / 32) * 4 * bitmap.bmHeight;//调色板大小const DWORD paletteSize = 0;//4. 构造位图信息头BITMAPINFOHEADER bmpInfoHeader; //位图信息头结构bmpInfoHeader.biSize = sizeof(BITMAPINFOHEADER);bmpInfoHeader.biWidth = bitmap.bmWidth;bmpInfoHeader.biHeight = bitmap.bmHeight;bmpInfoHeader.biPlanes = 1;bmpInfoHeader.biBitCount = bitCount;bmpInfoHeader.biCompression = BI_RGB;bmpInfoHeader.biSizeImage = 0;bmpInfoHeader.biXPelsPerMeter = 0;bmpInfoHeader.biYPelsPerMeter = 0;bmpInfoHeader.biClrImportant = 0;bmpInfoHeader.biClrUsed = 0;//5. 构造位图文件头BITMAPFILEHEADER bmpFileHeader;bmpFileHeader.bfType = 0x4D42; //"BM"//位图文件大小const DWORD dibSize = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER) + paletteSize + bmBitsSize;bmpFileHeader.bfSize = dibSize;bmpFileHeader.bfReserved1 = 0;bmpFileHeader.bfReserved2 = 0;bmpFileHeader.bfOffBits = static_cast<DWORD>(sizeof(BITMAPFILEHEADER))+ static_cast<DWORD>(sizeof(BITMAPINFOHEADER)) + paletteSize;//6. 为位图内容分配内存const auto dib = GlobalAlloc(GHND, bmBitsSize + paletteSize + sizeof(BITMAPINFOHEADER)); //内存句柄const auto lpBmpInfoHeader = static_cast<LPBITMAPINFOHEADER>(GlobalLock(dib)); //指向位图信息头结构*lpBmpInfoHeader = bmpInfoHeader;//7. 处理调色板ProcessPalette(hBitmap, bitmap, paletteSize, lpBmpInfoHeader);//8. 写入文件DWORD written = 0; //写入文件字节数WriteFile(file, reinterpret_cast<LPSTR>(&bmpFileHeader), sizeof(BITMAPFILEHEADER),&written, nullptr); //写入位图文件头WriteFile(file, reinterpret_cast<LPSTR>(lpBmpInfoHeader), dibSize,&written, nullptr); //写入位图文件其余内容//9. 清理资源GlobalUnlock(dib);GlobalFree(dib);CloseHandle(file);return true;}//计算位图文件每个像素所占字节数WORD ImageHelper::GetBitmapBitCount(){const auto dc = ::CreateDCA("DISPLAY", nullptr, nullptr, nullptr);//当前分辨率下每像素所占字节数const auto bits = ::GetDeviceCaps(dc, BITSPIXEL) * GetDeviceCaps(dc, PLANES);::DeleteDC(dc);//位图中每像素所占字节数WORD bitCount;if (bits <= 1)bitCount = 1;else if (bits <= 4)bitCount = 4;else if (bits <= 8)bitCount = 8;elsebitCount = 24;return bitCount;}//处理调色板void ImageHelper::ProcessPalette(HBITMAP hBitmap, const BITMAP& bitmap,DWORD paletteSize, LPBITMAPINFOHEADER lpBmpInfoHeader){HANDLE oldPalette = nullptr;HDC dc = nullptr;const auto palette = GetStockObject(DEFAULT_PALETTE);if (palette != nullptr){dc = ::GetDC(nullptr);oldPalette = ::SelectPalette(dc, static_cast<HPALETTE>(palette), FALSE);::RealizePalette(dc); //实现设备调色板}//获取该调色板下新的像素值GetDIBits(dc, hBitmap, 0, static_cast<UINT>(bitmap.bmHeight),reinterpret_cast<LPSTR>(lpBmpInfoHeader) + sizeof(BITMAPINFOHEADER) + paletteSize,reinterpret_cast<BITMAPINFO*>(lpBmpInfoHeader), DIB_RGB_COLORS);//恢复调色板if (oldPalette != nullptr){::SelectPalette(dc, static_cast<HPALETTE>(oldPalette), TRUE);::RealizePalette(dc);::ReleaseDC(nullptr, dc);}}

以上就是C++保存HBITMAP为位图文件的实现方法的详细内容,更多关于C++保存HBITMAP的资料请关注其它相关文章!

声明:本页内容来源网络,仅供用户参考;我单位不保证亦不表示资料全面及准确无误,也不保证亦不表示这些资料为最新信息,如因任何原因,本网内容或者用户因倚赖本网内容造成任何损失或损害,我单位将不会负任何法律责任。如涉及版权问题,请提交至online#300.cn邮箱联系删除。

相关文章