为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;}