bmp_基于像素统计的相似度检测.rar

#include <fstream> #include <map> #include <string> #include <iostream> using namespace std; typedef unsigned char uvar8; typedef unsigned int uvar32; typedef unsigned short int uvar16; typedef struct { uvar8 Blue; /*蓝色所占比重*/ uvar8 Green; /*绿色所占比重*/ uvar8 Red; /*红色所占比重*/ uvar8 Reserved; } Palette; //定义bmp结构 uvar8 bfType[2]; /* 类型标志，总是BM */ typedef struct { uvar32 bfSize; /* 文件大小 */ uvar32 bfReserved1; uvar32 bfOffBits; /* 位图点阵偏移量*/ }HEAD; typedef struct { uvar32 biSize; /* 结构体字节总数 */ uvar32 biWidth; /* 图像宽度 */ uvar32 biHeight; /*图像高度 */ uvar16 biPlanes; /* 必须为1 */ uvar16 biBitCount; /* 每个像素所占二进制位数，可能是1,4,8或 24 */ uvar32 biCompress; /*压缩方式*/ uvar32 biSizeImage; /*像素点阵大小 */ uvar32 biXPelsPerMeter; /* 水平像素数*/ uvar32 biYPelsPerMeter; /* 垂直像素数 */ uvar32 biClrUsed; /*使用的颜色数 */ uvar32 biClrImportant; /*重要颜色数 */ }INFO; bool ReadBMP(string strFileUrl,uvar8* & pImgData,int& iSum) { HEAD strHead; INFO strInfo; uvar8 type[2]; //uvar32 imdata[256][256]; Palette Rgbquad; ifstream fin; fin.open(strFileUrl.c_str(),ios_base::in | ios_base::binary); /*以二进制读方式打开该文件，一定要二进制的!*/ if(!fin) { cout<<"No this file!\n"; return false; } fin.read((char*)&bfType,2); fin.read((char*)&strHead,sizeof(strHead)); fin.read((char*)&strInfo,sizeof(strInfo)); fin.seekg(strHead.bfOffBits,ios::beg); if (strInfo.biBitCount == 8) { //pImgData = (uvar8*)pImgData; pImgData = new uvar8[strInfo.biWidth*strInfo.biHeight]; } fin.read((char*)(pImgData),strInfo.biWidth * strInfo.biHeight* sizeof uvar8); fin.close(); iSum = strInfo.biWidth * strInfo.biHeight; return true; } int main() { uvar8 *pImgData1 =NULL ,*pImgData2 = NULL; int i,iSum1,iSum2; while(1) { string url1; string url2; cout<<"cin bmp file name"<<endl; cin>> url1 >> url2; ReadBMP(url1,pImgData1,iSum1); ReadBMP(url2,pImgData2,iSum2); map<int,int> Map1,Map2; for (i=0;i<iSum1;i++) { if (Map1.find(pImgData1[i])!=Map1.end()) Map1[(uvar8)pImgData1[i]]++; else Map1[(uvar8)pImgData1[i]] = 0; } for (i=0;i<iSum2;i++) { if (Map2.find(pImgData2[i])!=Map2.end()) Map2[(uvar8)pImgData2[i]]++; else Map2[(uvar8)pImgData2[i]] = 0; } int iSum = 0; for (map<int,int>::iterator it = Map1.begin(); it!= Map1.end(); it++) { iSum += Map1[(uvar8)it->first] < Map2[(uvar8)it->first] ? Map1[(uvar8)it->first] : Map2[(uvar8)it->first]; } double dResult = (double) iSum / iSum1 > (double) iSum / iSum2 ? (double) iSum / iSum1:(double) iSum / iSum2; //----------------- cout<<endl; cout<<"special dots the sum of pic1 the sum of pic2"<<endl; for (map<int,int>::iterator it1 = Map1.begin(); it1!= Map1.end(); it1++) { cout<<it1->first<<" "<<it1->second<<" "<<Map2[it1->first]<<endl; } cout<<"******************************************"<<endl; cout<<"conculsion："<<endl; cout<< "The sum of dots in Pic1: "<< iSum1<<endl; cout<< "The sum of dots in Pic2: "<< iSum2<<endl; cout<< "The similar dots between pic1 and pic2 : "<<iSum<<endl; cout<<"The rate of similar between the above two pictures: " << dResult <<endl<<endl; delete[] pImgData1; delete[] pImgData2; } getchar(); return 0; }