为raw数据添加图像头，并保存为bmp图像（改进）

#include "stdio.h"
#include "Windows.h"

unsigned char *pBmpBuf;//读入图像数据的指针
int bmpWidth;//图像的宽
int bmpHeight;//图像的高
//RGBQUAD *pColorTable;//颜色表指针
int biBitCount;//图像类型，每像素位数

/*****************************************
* 函数名称：
*     saveBmp()
*
*函数参数：
*    char *bmpName-文件名字及路径
*    unsigned char *imgBuf-待存盘的位图数据
*    int width-以像素为单位待存盘位图的宽
*    int  height-以像素为单位待存盘位图高
*    int biBitCount-每像素所占位数
*    RGBQUAD *pColorTable-颜色表指针
*返回值：
*    0为失败，1为成功
*
*说明：给定一个图像位图数据、宽、高、颜色表指针及每像素所占的位数等信息，
*      将其写到指定文件中
***********************************************************************/
bool 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是图像文件前3个部分所需空间之和
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;
}

int main(int argc, char* argv[])
{
//输入参数为3个:　执行程序名，图像数据的路径名，生成图像路径名
if( argc != 3)
{
fprintf(stderr, "Input parameter number is not 4!/n");
exit(1);
}

bmpWidth = 256;
bmpHeight = 256;
biBitCount = 8;
int lineByte=(bmpWidth * biBitCount/8+3)/4*4;
pBmpBuf=new unsigned char[lineByte * bmpHeight];

//创建灰度图像调色板
RGBQUAD rgbquad[256];
for(int i = 0; i < 256; i++)
{
rgbquad[i].rgbBlue = i;
rgbquad[i].rgbGreen = i;
rgbquad[i].rgbRed = i;
rgbquad[i].rgbReserved = 0;
}

//创建新图像
char *bmpName = argv[1];//argv[1]为图像数据的路径名
FILE *fp=fopen(bmpName,"rb");
if(fp==0)
{
fprintf(stderr, "can't open file %s!/n", bmpName);
exit(1);
}
fread(pBmpBuf,1,lineByte * bmpHeight,fp);
if( ferror(fp) )
{
fprintf(stderr, "?Read error/n");
exit(1);
}

//修改图像数据(二值化)
for(int i = 0; i < bmpWidth; i++)
for(int j = 0; j < bmpHeight; j++)
{
//	if( pBmpBuf[j*lineByte+i] == atoi(argv[1]) ) pBmpBuf[j*lineByte+i] = 255;//argv[1]为二值化的阈值
if( pBmpBuf[j*lineByte+i] == 2 ) pBmpBuf[j*lineByte+i] = 255;//argv[1]为二值化的阈值
}

fclose(fp);

saveBmp(argv[2], pBmpBuf, bmpWidth, bmpHeight, biBitCount, rgbquad);//argv[3]为整合后的图像路径名

return 0;
}