1. 程式人生 > 程式設計 >C++儲存HBITMAP為點陣圖檔案的實現方法

C++儲存HBITMAP為點陣圖檔案的實現方法

本文使用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,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,nullptr); //寫入點陣圖檔案其餘內容

//9. 清理資源
GlobalUnlock(dib);
GlobalFree(dib);
CloseHandle(file);

return true;
}

//計算點陣圖檔案每個畫素所佔位元組數
WORD ImageHelper::GetBitmapBitCount()
{
const auto dc = ::CreateDCA("DISPLAY",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;
else
bitCount = 24;

return bitCount;
}

//處理調色盤
void ImageHelper::ProcessPalette(HBITMAP hBitmap,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,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的資料請關注我們其它相關文章!