chap1_4doc.cpp

上述是VIsualc++ 数字图像处理一书的源码

// chap1_4Doc.cpp : implementation of the CChap1_4Doc class
//

#include "stdafx.h"
#include "chap1_4.h"
#include "math.h"

#include "chap1_4Doc.h"

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

/////////////////////////////////////////////////////////////////////////////
// CChap1_4Doc

IMPLEMENT_DYNCREATE(CChap1_4Doc, CDocument)

BEGIN_MESSAGE_MAP(CChap1_4Doc, CDocument)
	//{{AFX_MSG_MAP(CChap1_4Doc)
		// NOTE - the ClassWizard will add and remove mapping macros here.
		//    DO NOT EDIT what you see in these blocks of generated code!
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CChap1_4Doc construction/destruction

CChap1_4Doc::CChap1_4Doc()
{
	m_pDib=NULL;//读入图像数据的指针
}

CChap1_4Doc::~CChap1_4Doc()
{
	if(m_pDib!=NULL)
		delete []m_pDib;
}

BOOL CChap1_4Doc::OnNewDocument()
{
	if (!CDocument::OnNewDocument())
		return FALSE;

	// TODO: add reinitialization code here
	// (SDI documents will reuse this document)

	return TRUE;
}



/////////////////////////////////////////////////////////////////////////////
// CChap1_4Doc serialization

void CChap1_4Doc::Serialize(CArchive& ar)
{
	if (ar.IsStoring())
	{
		// TODO: add storing code here
	}
	else
	{
		// TODO: add loading code here
	}
}

/////////////////////////////////////////////////////////////////////////////
// CChap1_4Doc diagnostics

#ifdef _DEBUG
void CChap1_4Doc::AssertValid() const
{
	CDocument::AssertValid();
}

void CChap1_4Doc::Dump(CDumpContext& dc) const
{
	CDocument::Dump(dc);
}
#endif //_DEBUG

/////////////////////////////////////////////////////////////////////////////
// CChap1_4Doc commands

BOOL CChap1_4Doc::OnOpenDocument(LPCTSTR lpszPathName) 
{
	//清理空间
	if(m_pDib!=NULL){
		delete []m_pDib;
		m_pDib=NULL;
	}

	//打开指定文件
	CFile file;
	if (!file.Open(lpszPathName, CFile::modeRead | CFile::shareDenyWrite))
		return FALSE;

	//跳过位图文件头结构
	file.Seek(sizeof(BITMAPFILEHEADER),0);

	//申请DIB所需要的内存空间，将除BITMAPFILEHEADER结构外的位图数据读入内存
	//存放在m_pDib所指向的缓冲区中
	m_pDib=new BYTE[file.GetLength() -sizeof(BITMAPFILEHEADER)];
		file.Read(m_pDib, file.GetLength() -sizeof(BITMAPFILEHEADER));
	
	return TRUE;
}

BOOL CChap1_4Doc::OnSaveDocument(LPCTSTR lpszPathName) 
{
	//如果DIB为空，则返回
	if(!m_pDib)
		return 0;
	
	//写方式打开文件
	CFile file;
	if (!file.Open(lpszPathName, CFile::modeCreate | CFile::modeReadWrite 
		| CFile::shareExclusive))
		return FALSE;
	
	//读取DIB的BITMAPINFOHEADER信息
	BITMAPINFOHEADER *infoHead=(BITMAPINFOHEADER*)m_pDib;
	
	//获取图像的宽、高、每像素所占位数
	int width=infoHead->biWidth;
	int height=infoHead->biHeight;
	int biBitCount=infoHead->biBitCount;
	
	//计算颜色表的长度，彩色图像没有颜色表长度为0，其它为pow(2,biBitCount)
	int colorTableLng;
	if(biBitCount!=24)
		colorTableLng=pow(2,biBitCount);
	else
		colorTableLng=0;
	
	//填写位图文件头结构
	BITMAPFILEHEADER bmfh;
	bmfh.bfType = 0x4d42;  // 'BM'
	int nSizeHdr = sizeof(BITMAPINFOHEADER) + 
		sizeof(RGBQUAD) * colorTableLng;
	bmfh.bfSize = 0;
	bmfh.bfReserved1 = bmfh.bfReserved2 = 0;
	bmfh.bfOffBits = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER) +
		sizeof(RGBQUAD) * colorTableLng;
	
	//文件头结构写入文件
	file.Write((LPVOID) &bmfh, sizeof(BITMAPFILEHEADER));
	
	//BITMAPINFOHEADER信息头结构写入文件
	file.Write(infoHead,  sizeof(BITMAPINFOHEADER));
	
	//如果有颜色表，则颜色表写入文件
	if(colorTableLng!=0){
		RGBQUAD *pColorTable=(RGBQUAD *)(m_pDib+sizeof(BITMAPINFOHEADER));
		file.Write(pColorTable, sizeof(RGBQUAD) * colorTableLng);
	}
	
	//计算图像位图数据的大小
	int imgBufSize=(width*biBitCount/8+3)/4*4*height;
	
	//pImgData指向DIB位图数据的指针
	unsigned char* pImgData=(unsigned char*)(m_pDib+
		sizeof(BITMAPINFOHEADER)+sizeof(RGBQUAD) * colorTableLng);
	
	//位图数据写入文件
	file.Write(pImgData, imgBufSize);
	
	return 1;	
}