zhifangtu.rar_358868..com_358868.com_358868com_358868．coml

// dibview.cpp : implementation of the CDibView class // // This is a part of the Microsoft Foundation Classes C++ library. // Copyright (C) 1992-1998 Microsoft Corporation // All rights reserved. // // This source code is only intended as a supplement to the // Microsoft Foundation Classes Reference and related // electronic documentation provided with the library. // See these sources for detailed information regarding the // Microsoft Foundation Classes product. #include "stdafx.h" #include "diblook.h" #include "dibdoc.h" #include "dibview.h" #include "dibapi.h" #include "mainfrm.h" #include "math.h" #ifdef _DEBUG #undef THIS_FILE static char BASED_CODE THIS_FILE[] = __FILE__; #endif ///////////////////////////////////////////////////////////////////////////// // CDibView #define houghthresh 10 void Welcom(); void InitImage(BYTE *image[2000], int w, int h); void qiu(long int hist[],long int histold[],int rad) { int aa=rad/2; long int total=0; int num=0; for(int i=0;i<256;i++) { total=num=0; { for(int j=i-aa;j<i-aa+rad;j++) { if(j<0||j>=255) continue; total+=hist[j]; num++; } histold[i]=total/num; } } } void GetHist(HDIB hDIB,long int hist[]) //求图像hDIB的直方图 hist { BYTE *image[2000]; int imageWidth,imageHeight; BYTE *lp; LPBITMAPINFOHEADER lh; if(!hDIB) return; lp=(BYTE *)::GlobalLock(hDIB); lh=(LPBITMAPINFOHEADER)lp; imageWidth=lh->biWidth; imageHeight=lh->biHeight; int biwidth; biwidth=lh->biWidth; if(biwidth%4!=0) biwidth+=(4-biwidth%4); if (lh->biBitCount==8) { lp=lp+40+256*4; int i; for(i=0;i<lh->biHeight;i++) image[i]=lp+(lh->biHeight-i-1)*biwidth; } else { ::GlobalUnlock(hDIB); return ; } //以上得到图像数组image short int i,j; for (i = 0;i < 256;i++) hist[i]=0l; for (i = 0;i <imageWidth ;i++) for(j=0;j<imageHeight;j++) hist[image[j][i]]++; //求出直方图数组 } void ReadRawFile(BYTE *image[2000], int w, int h, char filepath[200]); void SaveRawFile(BYTE *image[2000], int w, int h, char filepath[200]); void WINAPI GetImage(HDIB hDIB, BYTE *image[]); void WINAPI GetImage(HDIB hDIB, BYTE *image[]) { LPSTR lp=(LPSTR)::GlobalLock(hDIB); LPBITMAPINFOHEADER lh= (LPBITMAPINFOHEADER)lp; int biWidth=DIBWidth(lp); int biHeight=DIBHeight(lp); if (biWidth%4!=0) biWidth+=(4-biWidth%4); lp=FindDIBBits(lp); if(lh->biBitCount==8) for(int i=0;i<biHeight;i++) image[i]=(BYTE *)(lp+(biHeight-1-i)*biWidth); else { AfxMessageBox("Not 8 bit bitmap!"); } ::GlobalUnlock(hDIB); } void Welcom() { printf("\n"); printf("********\n"); printf("Welcome!\n"); printf("********\n"); printf("\n"); } void InitImage(BYTE *image[2000], int w, int h) { for(int i=0;i<h;i++) image[i]=(BYTE *)new BYTE[w]; for(i=0;i<h;i++) for(int j=0;j<w;j++) image[i][j]=0; } void ReadRawFile(BYTE *image[2000], int w, int h, char filepath[200]) { FILE *fp=fopen(filepath,"rb"); if(fp==NULL) {printf("error open \"%s\" !\n",filepath);exit(0);} for(int i=0;i<h;i++) fread(image[i],1,w,fp); fclose(fp); printf("success open \"%s\" !\n",filepath); } void SaveRawFile(BYTE *image[2000], int w, int h, char filepath[200]) { FILE *fp=fopen(filepath,"wb"); if(fp==NULL) {printf("error open \"%s\" to write !\n",filepath);exit(0);} for(int i=0;i<h;i++) fwrite(image[i],1,w,fp); fclose(fp); printf("success open \"%s\" to write !\n",filepath); } void diedaiyz(BYTE *image[2000],int w,int h); void sanfenyz(BYTE *image[2000],int w,int h); void sobel(BYTE *image_in[2000],int w,int h,BYTE *image_out[2000]); void lapla1(BYTE *image_in[2000],int w,int h,BYTE *image_out[2000]); void lapla2(BYTE *image_in[2000],int w,int h,BYTE *image_out[2000]); void zzlb(BYTE *image_in[2000],int w,int h,BYTE *image_out[2000]); void houghtrans1(BYTE *image_in[2000],int w,int h); void houghtrans2(BYTE *image_in[2000],int w,int h); void makeline1(BYTE *image_in[2000],int w,int h,double t,int k,int r,BYTE *image_out[2000]); void makeline2(BYTE *image_in[2000],int w,int h,double t,int k,int r,BYTE *image_out[2000]); int medianvalue(int *value); int main() { int w=512,h=512; Welcom(); BYTE *image[2000]; InitImage(image,w,h); ReadRawFile(image,w,h,"D:\\peacock.raw"); //图像读取的路径为：D:\\peacock.raw ，在保存的时候存在了E盘 //且均以peacock_后缀.raw命名，例如：peacock_lapla2.raw,peacock_hough.raw //我们将所有的函数按不同的顺序进行组合，以期得到最好的边缘提取效果 // diedaiyz(image,w,h); //迭代法选取阈值，二值化，并将二值化后的图像保存 // sanfenyz(image,w,h); //T=max-(max-min)/3，二值化，并将二值化后的图像保存 // BYTE *image2[2000]; // InitImage(image2,w,h); // zzlb(image,w,h,image2); //中值滤波，并将滤波后的图像保存 //前面的为输入数组，如image;后面的为输出数组，如image2。以下均同，请注意之！ BYTE *image1[2000]; InitImage(image1,w,h); // sobel(image,w,h,image1); //sobel算子对图像进行梯度锐化，并将梯度锐化后的图像保存 // lapla1(image,w,h,image1); //拉普拉斯算子，模板一：{0,-1,0,-1,4,-1,0,-1,0}，并将梯度锐化后的图像保存 lapla2(image,w,h,image1); //拉普拉斯算子，模板二：{-1,-1,-1,-1,8,-1, -1,-1,-1}，并将梯度锐化后的图像保存 // BYTE *image2[2000]; // InitImage(image2,w,h); // zzlb(image1,w,h,image2); //中值滤波 // diedaiyz(image1,w,h); //迭代法选取阈值，二值化 sanfenyz(image1,w,h); //T=max-(max-min)/3，二值化 // BYTE *image2[2000]; // InitImage(image2,w,h); // zzlb(image1,w,h,image2); //中值滤波 // houghtrans1(image1,w,h); //hough变换第一种算法：5x5模板 houghtrans2(image1,w,h); //hough变换第二种算法：循环hough,多次提取最长直线 return 1; } void houghtrans1(BYTE *image_in[2000],int w,int h) //hough变换第一种算法 { double t=3.1415926; t=t/180; int rmax=w+h+1; int *hough[180]; int k,r; for(k=0;k<180;k++) hough[k]=(int *)new int[rmax]; for(k=0;k<180;k++) for(r=0;r<rmax;r++) hough[k][r]=0; int i,j; for(i=0;i<h;i++) //将大于0的点标记为255，即白点 //自己绘制的图像中可能存在一些点既不是0也不是255，另外为了避免由于前期处理不当导致hough失败 for(j=0;j<w;j++) {if(image_in[i][j]>0)image_in[i][j]=255;} for(i=0;i<h;i++) //对原图中的白点做hough变换 for(j=0;j<w;j++) { if(image_in[i][j]==255) { for(k=0;k<180;k++) { r=(int)(j*cos(k*t)+i*sin(k*t)); if(r<0)r+=w+h+1; hough[k][r]++; } } } BYTE *image_out[2000]; InitImage(image_out,w,h); for(k=0;k<180;k++) //寻找数组hough[][]的各个峰值，采用5x5模板算法 for(r=0;r<rmax;r++) { if(hough[k][r]>houghthresh) { int tag=1; int dk,dr,dkt,drt; for(dk=k-2;dk<=k+2;dk++) { for(dr=r-2;dr<=r+2;dr++) { if(dk<0)dkt=dk+180; else if(dk>=180)dkt=dk-180; else dkt=dk; if(dr<0)drt=dr+w+h+1; else drt=dr; if(drt<rmax&&hough[dkt][drt]>hough[k][r]) {tag=0;break;} } if(tag==0)break; } if(tag==1) { // makeline1(image_in,w,h,t,k,r,image_out); //寻找到一条直线立即绘制 makeline2(image_in,w,h,t,k,r,image_out); } } } SaveRawFile(image_out,w,h,"E:\\peacock_hough.raw"); } void houghtrans2(BYTE *image_in[2000],int w,int h) //hough变换第二种算法 { BYTE *image_out[2000]; InitImage(image_out,w,h); double t=3.1415926; t=t/180; int rmax=w+h+1; int *hough[180]; int k,r; for(k=0;k<180;k++) hough[k]=(int *)new int[rmax]; int i,j; for(i=0;i<h;i++) for(j=0;j<w;j++) if(image_in[i][j]>0)image_in[i][j]=255;//此处以上与算法一相同 while(1) //对hough变换的全部过程做循环 { for(k=0;k<180;k++) for(r=0;r<rmax;r++) hough[k][r]=0; for(i=0;i<h;i++) //对原图作hough变换，同算法一 for(j=0;j<w;j++) { if(image_in[i][j]==255) { for