C++保存HBITMAP為位圖文件

本文為轉載內容微加工，原文地址：C++將HBITMAP保存為bmp圖片。

本文使用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;
	else
		bitCount = 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);
	}
}