VC++的CDib类

最新推荐文章于 2016-06-22 23:30:43 发布

辉哥聊职场聊家庭

最新推荐文章于 2016-06-22 23:30:43 发布

阅读量1.5k

点赞数

分类专栏：图像处理方面（基于C++）

图像处理方面（基于C++）专栏收录该内容

22 篇文章

订阅专栏

Visual C++ MFC中没有提供一个专门的类来处理DIB位图，因此，为了方便地使用位图文件，我们有必要派生一个CDib类。类的源代码如下：

　　(1) CDib类的声明

// DIB.h：类CDib声明头文件
#ifndef __DIB_H__
#define __DIB_H__
#include <wingdi.h>
class CDib
{
　public:
　　CDib();
　　~CDib();

　　BOOL Load( const char * );
　　BOOL Save( const char * );
　　BOOL Draw( CDC *, int nX = 0, int nY = 0, int nWidth = -1, int nHeight = -1, int mode = SRCCOPY);
　　BOOL SetPalette( CDC * );

　private:
　　CPalette m_Palette;
　　unsigned char *m_pDib, *m_pDibBits;
　　DWORD m_dwDibSize;
　　BITMAPINFOHEADER *m_pBIH;
　　RGBQUAD *m_pPalette;
　　int m_nPaletteEntries;
};
#endif

　　(2) CDib类的实现

// DIB.cpp：类CDib实现文件
#include "stdafx.h"
#include "DIB.h"

CDib::CDib()
{
　m_pDib = NULL;
}

CDib::~CDib()
{
　// 如果位图已经被加载，释放内存
　if (m_pDib != NULL)
　　delete []m_pDib;
}

　　下面这个函数非常重要，其功能为加载位图，类似于CBitmap类的LoadBitmap函数：

BOOL CDib::Load(const char *pszFilename)
{
　CFile cf;

　// 打开位图文件
　if (!cf.Open(pszFilename, CFile::modeRead))
　　return (FALSE);

　// 获得位图文件大小，并减去BITMAPFILEHEADER的长度
　DWORD dwDibSize;
　dwDibSize = cf.GetLength() - sizeof(BITMAPFILEHEADER);

　// 为DIB位图分配内存
　unsigned char *pDib;
　pDib = new unsigned char[dwDibSize];
　if (pDib == NULL)
　　return (FALSE);

　BITMAPFILEHEADER BFH;

　// 读取位图文件数据
　try
　{
　　// 文件格式是否正确有效
　　if ( cf.Read(&BFH, sizeof(BITMAPFILEHEADER)) != sizeof(BITMAPFILEHEADER) ||
　　　　　BFH.bfType != ’MB’ || cf.Read(pDib, dwDibSize) != dwDibSize)
　　{
　　　delete []pDib;
　　　return (FALSE);
　　}
　}
　catch (CFileException *e)
　{
　　e->Delete();
　　delete []pDib;
　　return (FALSE);
　}

　// delete先前加载的位图
　if (m_pDib != NULL)
　　delete m_pDib;

　// 将临时Dib数据指针和Dib大小变量赋给类成员变量
　m_pDib = pDib;
　m_dwDibSize = dwDibSize;

　// 为相应类成员变量赋BITMAPINFOHEADER和调色板指针
　m_pBIH = (BITMAPINFOHEADER*)m_pDib;
　m_pPalette = (RGBQUAD*) &m_pDib[sizeof(BITMAPINFOHEADER)];

　// 计算调色板中实际颜色数量
　m_nPaletteEntries = 1 << m_pBIH->biBitCount;
　if (m_pBIH->biBitCount > 8)
　　m_nPaletteEntries = 0;
　else if (m_pBIH->biClrUsed != 0)
　　m_nPaletteEntries = m_pBIH->biClrUsed;

　// 为相应类成员变量赋image data指针
　m_pDibBits = &m_pDib[sizeof(BITMAPINFOHEADER) + m_nPaletteEntries * sizeof (RGBQUAD)];

　// delete先前的调色板
　if (m_Palette.GetSafeHandle() != NULL)
　　m_Palette.DeleteObject();

　// 如果位图中存在调色板，创建LOGPALETTE 及CPalette
　if (m_nPaletteEntries != 0)
　{
　　LOGPALETTE *pLogPal = (LOGPALETTE*)new char[sizeof(LOGPALETTE) + m_nPaletteEntries *sizeof(PALETTEENTRY)];

　　if (pLogPal != NULL)
　　{
　　　pLogPal->palVersion = 0x300;
　　　pLogPal->palNumEntries = m_nPaletteEntries;

　　　for (int i = 0; i < m_nPaletteEntries; i++)
　　　{
　　　　pLogPal->palPalEntry[i].peRed = m_pPalette[i].rgbRed;
　　　　pLogPal->palPalEntry[i].peGreen = m_pPalette[i].rgbGreen;
　　　　pLogPal->palPalEntry[i].peBlue = m_pPalette[i].rgbBlue;
　　　}

　　　//创建CPalette并释放LOGPALETTE的内存
　　　m_Palette.CreatePalette(pLogPal);
　　　delete []pLogPal;
　　}
　}

　return (TRUE);
}

//函数功能：保存位图入BMP文件
BOOL CDib::Save(const char *pszFilename)
{
　if (m_pDib == NULL)
　　return (FALSE);

　CFile cf;
　if (!cf.Open(pszFilename, CFile::modeCreate | CFile::modeWrite))
　　return (FALSE);

　try
　{
　　BITMAPFILEHEADER BFH;
　　memset(&BFH, 0, sizeof(BITMAPFILEHEADER));
　　BFH.bfType = ’MB’;
　　BFH.bfSize = sizeof(BITMAPFILEHEADER) + m_dwDibSize;
　　BFH.bfOffBits = sizeof(BITMAPFILEHEADER) +
sizeof(BITMAPINFOHEADER) + m_nPaletteEntries *sizeof(RGBQUAD);

　　cf.Write(&BFH, sizeof(BITMAPFILEHEADER));
　　cf.Write(m_pDib, m_dwDibSize);
　}
　catch (CFileException *e)
　{
　　e->Delete();
　　return (FALSE);
　}
　return (TRUE);
}

　　下面这个函数也非常重要，其功能为在pDC指向的CDC中绘制位图，起点坐标为(nX,nY)，绘制宽度和高度为nWidth、nHeight，最后一个参数是光栅模式：

BOOL CDib::Draw(CDC *pDC, int nX, int nY, int nWidth, int nHeight, int mode)
{
　if (m_pDib == NULL)
　　return (FALSE);

　// 获取位图宽度和高度赋值
　if (nWidth == - 1)
　　nWidth = m_pBIH->biWidth;
　if (nHeight == - 1)
　　nHeight = m_pBIH->biHeight;

　// 绘制位图
　StretchDIBits(pDC->m_hDC, nX, nY, nWidth, nHeight, 0, 0, m_pBIH->biWidth, m_pBIH->biHeight, m_pDibBits, (BITMAPINFO*)m_pBIH, BI_RGB, mode);

　return (TRUE);
}

//函数功能：设置调色板
BOOL CDib::SetPalette(CDC *pDC)
{
　if (m_pDib == NULL)
　　return (FALSE);

　// 检查当前是否有一个调色板句柄，对于大于256色的位图，为NULL
　if (m_Palette.GetSafeHandle() == NULL)
　　return (TRUE);

　// 选择调色板，接着实施之，最后恢复老的调色板
　CPalette *pOldPalette;
　pOldPalette = pDC->SelectPalette(&m_Palette, FALSE);
　pDC->RealizePalette();
　pDC->SelectPalette(pOldPalette, FALSE);

　return (TRUE);
}

　　从整个CDib类的代码中我们可以看出，DIB位图的显示需遵循如下步骤：

　　（1）读取位图，本类中使用pDib = new unsigned char[dwDibSize]为位图中的信息分配内存，另一种方法是调用API函数CreateDIBSection，譬如：

m_hBitmap = ::CreateDIBSection(pDC->GetSafeHdc(),
(LPBITMAPINFO) m_lpBMPHdr, DIB_RGB_COLORS,
(LPVOID*) &m_lpDIBits, NULL, 0);

　　m_hBitmap定义为：

HBITMAP m_hBitmap;

　　（2）根据读取的位图信息，计算出调色板大小，然后创建调色板；

　　（3）调用CDib::SetPalette( CDC *pDC )设置调色板，需要用到CDC::SelectPalette及CDC::RealizePalette两个函数；

　　（4）调用CDib::Draw(CDC *pDC, int nX, int nY, int nWidth, int nHeight, int mode)函数绘制位图。在此函数中，真正发挥显示位图作用的是对StretchDIBits API函数的调用。StretchDIBits函数具有缩放功能，其最后一个参数也是光栅操作的模式。

　　下面给出DIB位图的打开及显示并在其中加入天极网logo的函数源代码。"DIB位图"父菜单下"打开"子菜单的单击事件消息处理函数为（其功能为打开位图并显示之）：

void CBitMapExampleDlg::OnOpendibpic()
{
　// 弹出文件对话框，让用户选择位图文件
　CFileDialog fileDialog(TRUE, "*.BMP", NULL, NULL,"位图文件(*.BMP)|*.bmp;*.BMP|");
　if (IDOK == fileDialog.DoModal())
　{
　　// 加载位图并显示之
　　CDib dib;
　　if (dib.Load(fileDialog.GetPathName()))
　　{
　　　CClientDC dc(this);
　　　dib.SetPalette(&dc);
　　　dib.Draw(&dc);
　　}
　}
}

　　"DIB位图"父菜单下"标记"子菜单的单击事件消息处理函数为（其功能为给位图加上天极网logo）：

void CBitMapExampleDlg::OnMarkDibpic()
{
　// 弹出文件对话框，让用户选择标记logo
　CFileDialog fileDialog(TRUE, "*.BMP", NULL, NULL, "标记位图文件(*.BMP)|*.bmp;*.BMP|");
　if (IDOK == fileDialog.DoModal())
　{
　　// 加载标记logo位图并与目标位图相与
　　CDib dib;
　　if (dib.Load(fileDialog.GetPathName()))
　　{
　　　CClientDC dc(this);
　　　dib.SetPalette(&dc);
　　　dib.Draw(&dc, 0, 0, - 1, - 1, SRCAND);
　　}
　}