dib.cpp

matlab source code very useful to science study

// Dib.cpp: implementation of the CDib class.
//
//////////////////////////////////////////////////////////////////////

#include "StdAfx.h"
#include "Dib.h"

#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif

//////////////////////////////////////////////////////////////////////
// C
//////////////////////////////////////////////////////////////////////
//构造与析构函数
CDib::CDib()
{
	m_hDrawDib=NULL;
	m_pDib=NULL;
}

CDib::~CDib()
{
	Close();
}
//在设备环境上绘制图象
void CDib::Draw(CDC *pDC,int nWidth, int nHeight)
{
	if(m_pDib!=NULL)
	{
	ASSERT(IsValid());
	DrawDibRealize(m_hDrawDib,pDC->GetSafeHdc(),TRUE);
	//采用VFW组件绘图
	DrawDibDraw(m_hDrawDib,pDC->GetSafeHdc(),
				0,  
				0,  
				nWidth,
				nHeight,
				(BITMAPINFOHEADER *)m_pDib,
				(LPVOID) GetBits(),
				0,  
				0,  
				((BITMAPINFOHEADER *)m_pDib)->biWidth,
				((BITMAPINFOHEADER *)m_pDib)->biHeight,
				DDF_BACKGROUNDPAL);
	}
}
//获得位图的CSize结构
CSize CDib::GetSize()
{
	return CSize(((BITMAPINFOHEADER *)m_pDib)->biWidth,
				 ((BITMAPINFOHEADER *)m_pDib)->biHeight);	
}
//获得位图的宽度
DWORD CDib::GetWidth()
{
	return ((BITMAPINFOHEADER *)m_pDib)->biWidth;
}
//获得位图每行的字节数
DWORD CDib::GetWidthBytes()
{
BITMAPINFOHEADER *lpDib=(BITMAPINFOHEADER *)m_pDib;
WORD wBitCount=lpDib->biBitCount;
WORD wWidth=lpDib->biWidth;
//WidthBytes=(((biWidth*biBitCount)+31)&~31)>>3
DWORD bytes_per_line=(((wWidth*wBitCount)+31)&~31)>>3;
return bytes_per_line;
}
//获得位图的高度
DWORD CDib::GetHeight()
{
	return	((BITMAPINFOHEADER *)m_pDib)->biHeight;
}
//释放位图资源
void CDib::Close()
{
	if(m_hDrawDib!=NULL)
	{
		DrawDibClose(m_hDrawDib);
		m_hDrawDib=NULL;
	}

	if(m_pDib!=NULL)
	{
		delete m_pDib;
		m_pDib=NULL;
		m_pDibBits=NULL;
	}
}
//将位图文件中的位图读入内存中
BOOL CDib::Open(const char * pzFileName)
{
	CFile file;
	int nBmpFileHeaderSize;
	Close(); 
	//初始化DrawDib函数组
	if(!(m_hDrawDib=DrawDibOpen()))
		goto exit;

 	//打开位图文件并读取位图的BITMAPFILEHEADER数据
	nBmpFileHeaderSize=sizeof(BITMAPFILEHEADER);

	if(!file.Open(pzFileName,CFile::modeRead | CFile::typeBinary))
		goto exit;

	if(file.Read((void *)&bmpFileHeader,nBmpFileHeaderSize)!=(UINT)nBmpFileHeaderSize)
		goto failure;

	//判断位图文件的有效性
	if(bmpFileHeader.bfType!=*((WORD *)"BM"))
		goto failure;

	//为位图数据分配空间
	m_pDib=(BYTE *)new char [bmpFileHeader.bfSize-nBmpFileHeaderSize];

	if(!m_pDib)
		goto failure;

	//将位图数据读入到内存中
	file.ReadHuge(m_pDib,bmpFileHeader.bfSize-nBmpFileHeaderSize);

	if(((BITMAPINFOHEADER *)m_pDib)->biSizeImage==0)
	{
		
		BITMAPINFOHEADER *pDib=(BITMAPINFOHEADER *)m_pDib;

	//scan lines must be DWord aligned, hence the strange bit stuff
	pDib->biSizeImage=((((pDib->biWidth*pDib->biBitCount)+31)&~31)>>3)*pDib->biHeight;
	}

	m_pDibBits=GetBits();

	file.Close();
	return TRUE;

failure:
	file.Close();
exit:
	Close();
	return FALSE;
}
//保存位图
BOOL CDib::Save(const char * pzFileName)
{

	CFile file;
	int nBmpFileHeaderSize;

	nBmpFileHeaderSize=sizeof(BITMAPFILEHEADER);

	if(!file.Open(pzFileName,CFile::modeCreate | CFile::modeWrite | CFile::typeBinary))
		goto exit;
     //将文件头信息写入
	file.Write(&bmpFileHeader,nBmpFileHeaderSize); 
	if(!m_pDib)
		goto failure;

	//将文件其它信息写入
	file.WriteHuge(m_pDib,bmpFileHeader.bfSize-nBmpFileHeaderSize);

	file.Close();
	return TRUE;

failure:
	file.Close();
exit:
	return FALSE;
}
//获得位图数据的指针
BYTE * CDib::GetBits()
{
	
	DWORD dwNumColors,dwColorTableSize;
	BITMAPINFOHEADER *lpDib=(BITMAPINFOHEADER *)m_pDib;
	//获取颜色数
	WORD wBitCount=lpDib->biBitCount;
	if(lpDib->biSize>=36)
		dwNumColors=lpDib->biClrUsed;
	else
		dwNumColors=0;
	if(dwNumColors==0)
	{    
		if(wBitCount!=24)
			dwNumColors=1L<<wBitCount;
		else 
			dwNumColors=0;
	}

	dwColorTableSize=dwNumColors*sizeof(RGBQUAD);

	return m_pDib+lpDib->biSize+dwColorTableSize;
}

DWORD CDib::GetBiBitCount()
{
	if(m_pDib!=NULL)
		return ((BITMAPINFOHEADER *)m_pDib)->biBitCount; 
	return 0;
}