vc++ bmp图像二值化

在网上看到2个例子：

方法一：http://blog.csdn.net/mydreamremindme/article/details/9950453      我生成的不是白色底，是蓝色底不知道为何

方法二：http://download.csdn.net/download/s200692269/2663986

测试Demo

1、新建基于mfc的dialog程序，添加2个按键，分别对应上面那俩方法代码

2、.h的代码

private:
	bool saveBmp(char *bmpName, unsigned char *imgBuf, int width, int height, 
		int biBitCount, RGBQUAD *pColorTable);
	bool readBmp(char *bmpName);
public:
	afx_msg void OnBnClickedButton1();
	afx_msg void OnBnClickedButton2();

3、.cpp代码

// bmp2zhiDlg.cpp : 实现文件
//

#include "stdafx.h"
#include "bmp2zhi.h"
#include "bmp2zhiDlg.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif


// 用于应用程序“关于”菜单项的 CAboutDlg 对话框

class CAboutDlg : public CDialog
{
public:
	CAboutDlg();

// 对话框数据
	enum { IDD = IDD_ABOUTBOX };

	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV 支持

// 实现
protected:
	DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
END_MESSAGE_MAP()


// Cbmp2zhiDlg 对话框




Cbmp2zhiDlg::Cbmp2zhiDlg(CWnd* pParent /*=NULL*/)
	: CDialog(Cbmp2zhiDlg::IDD, pParent)
{
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void Cbmp2zhiDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(Cbmp2zhiDlg, CDialog)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	//}}AFX_MSG_MAP
	ON_BN_CLICKED(IDC_BUTTON1, &Cbmp2zhiDlg::OnBnClickedButton1)
	ON_BN_CLICKED(IDC_BUTTON2, &Cbmp2zhiDlg::OnBnClickedButton2)
END_MESSAGE_MAP()


// Cbmp2zhiDlg 消息处理程序

BOOL Cbmp2zhiDlg::OnInitDialog()
{
	CDialog::OnInitDialog();

	// 将“关于...”菜单项添加到系统菜单中。

	// IDM_ABOUTBOX 必须在系统命令范围内。
	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
	ASSERT(IDM_ABOUTBOX < 0xF000);

	CMenu* pSysMenu = GetSystemMenu(FALSE);
	if (pSysMenu != NULL)
	{
		BOOL bNameValid;
		CString strAboutMenu;
		bNameValid = strAboutMenu.LoadString(IDS_ABOUTBOX);
		ASSERT(bNameValid);
		if (!strAboutMenu.IsEmpty())
		{
			pSysMenu->AppendMenu(MF_SEPARATOR);
			pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
		}
	}

	// 设置此对话框的图标。当应用程序主窗口不是对话框时，框架将自动
	//  执行此操作
	SetIcon(m_hIcon, TRUE);			// 设置大图标
	SetIcon(m_hIcon, FALSE);		// 设置小图标

	// TODO: 在此添加额外的初始化代码

	return TRUE;  // 除非将焦点设置到控件，否则返回 TRUE
}

void Cbmp2zhiDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
	if ((nID & 0xFFF0) == IDM_ABOUTBOX)
	{
		CAboutDlg dlgAbout;
		dlgAbout.DoModal();
	}
	else
	{
		CDialog::OnSysCommand(nID, lParam);
	}
}

// 如果向对话框添加最小化按钮，则需要下面的代码
//  来绘制该图标。对于使用文档/视图模型的 MFC 应用程序，
//  这将由框架自动完成。

void Cbmp2zhiDlg::OnPaint()
{
	if (IsIconic())
	{
		CPaintDC dc(this); // 用于绘制的设备上下文

		SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);

		// 使图标在工作区矩形中居中
		int cxIcon = GetSystemMetrics(SM_CXICON);
		int cyIcon = GetSystemMetrics(SM_CYICON);
		CRect rect;
		GetClientRect(&rect);
		int x = (rect.Width() - cxIcon + 1) / 2;
		int y = (rect.Height() - cyIcon + 1) / 2;

		// 绘制图标
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialog::OnPaint();
	}
}

//当用户拖动最小化窗口时系统调用此函数取得光标
//显示。
HCURSOR Cbmp2zhiDlg::OnQueryDragIcon()
{
	return static_cast<HCURSOR>(m_hIcon);
}


void Cbmp2zhiDlg::OnBnClickedButton1()
{
	// TODO: 在此添加控件通知处理程序代码
	int threshold=100;  
	FILE* stream=fopen("D:\\1.bmp","rb");  
	if(stream==NULL)  
	{  
		//cout<<"文件不存在"<<endl;
		AfxMessageBox("文件不存在");
		return;  
	}  

	int sizeFileHeader=sizeof(BITMAPFILEHEADER);  
	int sizeInfoHeader=sizeof(BITMAPINFOHEADER);  

	BITMAPFILEHEADER* bitmapFileHeader=new BITMAPFILEHEADER[sizeFileHeader+1];  

	BITMAPINFOHEADER* bitmapInfoHeader=new BITMAPINFOHEADER[sizeInfoHeader+1];  

	memset(bitmapFileHeader,0,sizeFileHeader+1);  
	memset(bitmapInfoHeader,0,sizeInfoHeader+1);  
	fread(bitmapFileHeader,sizeof(char),sizeFileHeader,stream);  
	fseek(stream,sizeFileHeader,0);  
	fread(bitmapInfoHeader,sizeof(char),sizeInfoHeader,stream);  
	fseek(stream,sizeInfoHeader+sizeFileHeader,0);  
	RGBQUAD* pRgbQuards=new RGBQUAD[256];  
	for (int k=0;k<256;k++)  
	{  
		fread(&pRgbQuards[k],sizeof(RGBQUAD),1,stream);  
	}  

	int count=(((bitmapInfoHeader->biWidth)*8+31)/32)*4-bitmapInfoHeader->biWidth*(bitmapInfoHeader->biBitCount/8);  
	BYTE* tempData=new BYTE[count+1];  
	memset(tempData,0,count+1);  
	fseek(stream,sizeFileHeader+sizeInfoHeader+256*sizeof(RGBQUAD),0);  
	BYTE** data=new BYTE*[bitmapInfoHeader->biHeight];  
	for(int i=0;i<bitmapInfoHeader->biHeight;i++)  
	{  
		data[i]=new BYTE[bitmapInfoHeader->biWidth];  
		for (int j=0;j<bitmapInfoHeader->biWidth;j++)  
		{  
			fread(&data[i][j],sizeof(char),1,stream);  
			if(data[i][j]>threshold)  
				data[i][j]=255;  
			else  
				data[i][j]=0;  
		}  
		for (int n=0;n<count;n++)  
		{  
			fread(&tempData[n],sizeof(char),1,stream);  
		}  
	}  

	fclose(stream);  



	//写入。。  
	FILE* fileWrite=fopen("D:\\09.bmp","a+");  
	fwrite(bitmapFileHeader,sizeof(char),sizeof(BITMAPFILEHEADER),fileWrite);  
	fwrite(bitmapInfoHeader,sizeof(char),sizeof(BITMAPINFOHEADER),fileWrite);  
	fwrite(pRgbQuards,sizeof(RGBQUAD),256,fileWrite);  

	for(int i=0;i<bitmapInfoHeader->biHeight;i++)  
	{  
		for(int j=0;j<bitmapInfoHeader->biWidth;j++)  
		{  
			fwrite(&data[i][j],sizeof(BYTE),1,fileWrite);  
		}  
		for(int m=0;m<count;m++)  
			fwrite(&tempData[m],sizeof(char),1,fileWrite);  
	}  
	fclose(fileWrite);  

	AfxMessageBox("success");
}


//几个全局变量，存放读入图像的位图数据、宽、高、颜色表及每像素所占位数(比特) 
//此处定义全局变量主要为了后面的图像数据访问及图像存储作准备
unsigned char *pBmpBuf;//读入图像数据的指针
int bmpWidth;//图像的宽
int bmpHeight;//图像的高
RGBQUAD *pColorTable;//颜色表指针
int biBitCount;//图像类型

void Cbmp2zhiDlg::OnBnClickedButton2()
{
	// TODO: 在此添加控件通知处理程序代码
	//读入指定BMP文件进内存
	char readPath[]="D:\\leangray.bmp";
	readBmp(readPath);

	//输出图像的信息
	//printf("width=%d,height=%d,biBitCount=%d\n",bmpWidth,bmpHeight,biBitCount);


	//循环变量,图像的坐标
	int i,j;
	//每行字节数
	int lineByte=(bmpWidth*biBitCount/8+3)/4*4;

	//循环变量，针对彩色图像，遍历每像素的三个分量
	int k;

	//二值化处理
	if(biBitCount==8){//对于灰度图像
		for(i=0;i<bmpHeight;i++){
			for(j=0;j<bmpWidth;j++){

				int p=*(pBmpBuf+i*lineByte+j);
				if(p<=160)
					*(pBmpBuf+i*lineByte+j)=0;
				else
					*(pBmpBuf+i*lineByte+j)=255;				
			}
		}
	}
	else if(biBitCount==24){//彩色图像
		for(i=0;i<bmpHeight/2;i++){
			for(j=0;j<bmpWidth/2;j++){
				for(k=0;k<3;k++)//每像素RGB三个分量分别置0才变成黑色
					*(pBmpBuf+i*lineByte+j*3+k)=0;
			}
		}
	}

	//将图像数据存盘
	char writePath[]="D:\\lean_threshold_160.bmp";
	saveBmp(writePath, pBmpBuf, bmpWidth, bmpHeight, biBitCount, pColorTable);

	//清除缓冲区，pBmpBuf和pColorTable是全局变量，在文件读入时申请的空间
	delete []pBmpBuf;
	if(biBitCount==8)
		delete []pColorTable;

}


/***********************************************************************
* 函数名称：
* readBmp()
*
*函数参数：
*  char *bmpName -文件名字及路径
*
*返回值：
*   0为失败,1为成功
*
*说明：给定一个图像文件名及其路径，读图像的位图数据、宽、高、颜色表及每像素
*      位数等数据进内存,存放在相应的全局变量中
***********************************************************************/
bool Cbmp2zhiDlg::readBmp(char *bmpName)
{
	//二进制读方式打开指定的图像文件
	FILE *fp=fopen(bmpName,"rb");
	if(fp==0) return 0;


	//跳过位图文件头结构BITMAPFILEHEADER
	fseek(fp, sizeof(BITMAPFILEHEADER),0);


	//定义位图信息头结构变量，读取位图信息头进内存，存放在变量head中
	BITMAPINFOHEADER head;  
	fread(&head, sizeof(BITMAPINFOHEADER), 1,fp); 

	//获取图像宽、高、每像素所占位数等信息
	bmpWidth = head.biWidth;
	bmpHeight = head.biHeight;
	biBitCount = head.biBitCount;

	//定义变量，计算图像每行像素所占的字节数（必须是4的倍数）
	int lineByte=(bmpWidth * biBitCount/8+3)/4*4;

	//灰度图像有颜色表，且颜色表表项为256
	if(biBitCount==8){
		//申请颜色表所需要的空间，读颜色表进内存
		pColorTable=new RGBQUAD[256];
		fread(pColorTable,sizeof(RGBQUAD),256,fp);
	}

	//申请位图数据所需要的空间，读位图数据进内存
	pBmpBuf=new unsigned char[lineByte * bmpHeight];
	fread(pBmpBuf,1,lineByte * bmpHeight,fp);

	//关闭文件
	fclose(fp);

	return 1;
}

/***********************************************************************
* 函数名称：
* saveBmp()
*
*函数参数：
*  char *bmpName -文件名字及路径
*  unsigned char *imgBuf  -待存盘的位图数据
*  int width   -像素为单位待存盘位图的宽
*  int  height  -像素为单位待存盘位图高
*  int biBitCount   -每像素所占位数
*  RGBQUAD *pColorTable  -颜色表指针

*返回值：
*   0为失败,1为成功
*
*说明：给定一个图像位图数据、宽、高、颜色表指针及每像素所占的位数等信息，
*      将其写到指定文件中
***********************************************************************/
bool Cbmp2zhiDlg::saveBmp(char *bmpName, unsigned char *imgBuf, int width, int height, 
			 int biBitCount, RGBQUAD *pColorTable)
{
	//如果位图数据指针为0,则没有数据传入,函数返回
	if(!imgBuf)
		return 0;

	//颜色表大小,以字节为单位,灰度图像颜色表为1024字节,彩色图像颜色表大小为0
	int colorTablesize=0;
	if(biBitCount==8)
		colorTablesize=1024;

	//待存储图像数据每行字节数为4的倍数
	int lineByte=(width * biBitCount/8+3)/4*4;

	//以二进制写的方式打开文件
	FILE *fp=fopen(bmpName,"wb");
	if(fp==0) return 0;

	//申请位图文件头结构变量，填写文件头信息
	BITMAPFILEHEADER fileHead;
	fileHead.bfType = 0x4D42;//bmp类型

	//bfSize是图像文件4个组成部分之和
	fileHead.bfSize= sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER)
		+ colorTablesize + lineByte*height;
	fileHead.bfReserved1 = 0;
	fileHead.bfReserved2 = 0;

	//bfOffBits是图像文件前三个部分所需空间之和
	fileHead.bfOffBits=54+colorTablesize;

	//写文件头进文件
	fwrite(&fileHead, sizeof(BITMAPFILEHEADER),1, fp);

	//申请位图信息头结构变量，填写信息头信息
	BITMAPINFOHEADER head; 
	head.biBitCount=biBitCount;
	head.biClrImportant=0;
	head.biClrUsed=0;
	head.biCompression=0;
	head.biHeight=height;
	head.biPlanes=1;
	head.biSize=40;
	head.biSizeImage=lineByte*height;
	head.biWidth=width;
	head.biXPelsPerMeter=0;
	head.biYPelsPerMeter=0;
	//写位图信息头进内存
	fwrite(&head, sizeof(BITMAPINFOHEADER),1, fp);

	//如果灰度图像,有颜色表,写入文件 
	if(biBitCount==8)
		fwrite(pColorTable, sizeof(RGBQUAD),256, fp);

	//写位图数据进文件
	fwrite(imgBuf, height*lineByte, 1, fp);

	//关闭文件
	fclose(fp);

	return 1;
}

原图和效果图分别如下：