zhifangtu.rar_zhifangtu

/* 头文件 */ #include <afx.h> #include <windows.h> #include <iostream.h> #include <stdlib.h> #include <math.h> /*复数定义*/ /* 全局变量 */ int nWidth; //图像的宽度 int nHeight; //图像的高度 int nColorBits; //每个像素所占的位数 int nColor; //图像的颜色数 int nLen; //图像文件的大小，以字节数计 int nByteWidth; //图像每行的字节数 BYTE *lpBitmap; //指向图像首字节的指针 BYTE *lpBits; //指向图像实际数据的指针 /* 函数声明 */ void OpenFile(CString FileName); void SaveFile(CString FileName); void HistogramStat(long bCount[], long gCount[], long rCount[]); void HistogramEq(); /************************************************************************ * 函数名称: - OpenFile(); * 参数: - CString FileName :图像的路径和文件名 * 返回值: - 无 * 功能: - 读取一幅BMP图像 ************************************************************************/ void OpenFile(CString FileName) { //创建文件句柄 HANDLE hFile =::CreateFile(FileName,GENERIC_READ,FILE_SHARE_READ, NULL,OPEN_EXISTING,FILE_ATTRIBUTE_NORMAL,NULL); if(hFile == 0) { printf("不能打开文件，请重新选择!\n"); return ; } //读取图像文件 DWORD WriteNum; BITMAPFILEHEADER BFH; //文件头 ReadFile(hFile,&BFH,14,&WriteNum,NULL);//读取文件头,共14个字节 if((BFH.bfType != 'MB')||(WriteNum != sizeof(BITMAPFILEHEADER))) { printf("不是BMP位图文件或数据有误!\n"); return ; } nLen = GetFileSize(hFile,NULL)-sizeof(BITMAPFILEHEADER); //获取文件的长度 lpBitmap= new BYTE[nLen];//存放图像,包括图像的信息头、调色板和像素数据 ReadFile(hFile,lpBitmap,nLen,&WriteNum,NULL); //读取图像数据 //设置全局变量的值 BITMAPINFOHEADER *BIH = ((BITMAPINFOHEADER *)lpBitmap); //图像文件的信息头 nWidth = BIH->biWidth; //图像的宽度 nHeight = BIH->biHeight; //图像的高度 nColorBits = BIH->biBitCount; //图像的颜色数 nByteWidth = (nWidth*nColorBits+31)/32*4; //图像的扫描宽度 nColor = (nColorBits>8) ? 0 : (1<<nColorBits); //调色板中的颜色数 lpBits = lpBitmap + sizeof(BITMAPINFOHEADER) + sizeof(RGBQUAD)*nColor; //指向图像数据的实际位置 CloseHandle(hFile); //关闭文件句柄 } /************************************************************************* * 函数名称: - SaveFile(); * 参数: - CString FileName - 图像的路径和文件名 * 返回值: - 无 * 功能: - 保存一幅BMP图像 ************************************************************************/ void SaveFile(CString FileName) { //创建一个文件来保存图像文件 HANDLE hFile = ::CreateFile(FileName,GENERIC_WRITE,FILE_SHARE_WRITE,NULL,CREATE_ALWAYS,FILE_ATTRIBUTE_NORMAL,NULL); if(hFile == 0) { printf("不能创建文件，请重新选择!\n"); return; } //创建一个文件头,并保存到创建的文件中 unsigned long WriteNum; BITMAPFILEHEADER BFH; BFH.bfType = 'MB'; BFH.bfSize = nLen + sizeof(BITMAPFILEHEADER); BFH.bfOffBits = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER) + nColor * sizeof(RGBQUAD); BFH.bfReserved1 = BFH.bfReserved2 = 0; WriteFile(hFile,&BFH,sizeof(BITMAPFILEHEADER),&WriteNum,NULL); //保存改变的位图文件数据 WriteFile(hFile,(BITMAPINFOHEADER *)lpBitmap,nLen,&WriteNum,NULL); CloseHandle(hFile); //关闭文件句柄 } /************************************************************************* * 函数名称: - HistogramStat(); * 参数: - long bCount[]; * - long gCount[]; * - long rCount[]; * 返回值: - 无 * 功能: - 统计图像各级灰度的像素的个数, * 当是256色图像,返回值放在bCount[]中 * 真彩色图像蓝色、绿色、红色分量依次放在三个参数变量中 **************************************************************************/ void HistogramStat(long bCount[], long gCount[], long rCount[]) { // 赋初值 for (int i = 0; i < 256; i++) { rCount[i] = 0; gCount[i] = 0; bCount[i] = 0; } //统计各个灰度级像素的个数 for (i=0;i<nHeight;i++) { for (int j=0;j<nWidth;j++) { if(nColorBits == 8) { unsigned char *pData = lpBits + i * nByteWidth + j; bCount[*pData]++; } else if(nColorBits == 24) { unsigned char *pData = lpBits + i * nByteWidth + j*3; bCount[*pData]++; gCount[*(pData+1)]++; rCount[*(pData+2)]++; } } } } /************************************************************************* * 函数名称: - HistogramEq(); * 参数: - 无 * 返回值: - 无 * 功能: - 直方图均衡 ************************************************************************/ void HistogramEq() { //各个颜色分量的亮度映射表 BYTE grayMap[256],rMap[256],gMap[256],bMap[256]; //各个颜色分量亮度统计表 long rCount[256],gCount[256],bCount[256]; long i,j; //直方图统计，统计各个灰度的像素，256色图像的结果放在bCount中 HistogramStat(bCount,gCount,rCount); //计算每个灰度级对应的新的像素值 double TotalNum = nWidth * nHeight; for (i = 0; i < 256; i++) { if(nColorBits == 8) { long grayTemp = 0; for (j=0;j<=i;j++) grayTemp += bCount[j]; grayMap[i] = (BYTE) (grayTemp * 255/TotalNum);//对应的新灰度值 } else if(nColorBits == 24) { long rTemp = 0, gTemp = 0, bTemp = 0; for (j = 0; j <= i ; j++) { rTemp += rCount[j]; gTemp += gCount[j]; bTemp += bCount[j]; } //计算对应的新灰度值 rMap[i] = (BYTE) (rTemp * 255/TotalNum); gMap[i] = (BYTE) (gTemp * 255/TotalNum); bMap[i] = (BYTE) (bTemp * 255/TotalNum); } } //对像素的灰度赋新值 for (i = 0; i < nHeight; i++) { for(j = 0; j < nWidth; j++) { if(nColorBits == 8) { unsigned char *pData = lpBits + i*nByteWidth+j; *pData = grayMap[*pData]; } else if(nColorBits == 24) { unsigned char *pData = lpBits + i*nByteWidth + j*3; *pData = bMap[*pData]; *(p