改進：不再從參考影象讀調色盤，程式內部建立調色盤

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