?? dibview.cpp
字號:
// 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(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]++;
}
}
}
int max=0,max_k,max_r; //尋找最長的一條直線
for(k=0;k<180;k++)
for(r=0;r<rmax;r++)
{
if(hough[k][r]>max)
{
max=hough[k][r];
max_k=k;
max_r=r;
}
}
if(max<houghthresh)break; //當最大值小于設定的臨界值時，循環結束
// makeline1(image_in,w,h,t,max_k,max_r,image_out);
makeline2(image_in,w,h,t,max_k,max_r,image_out);
}
SaveRawFile(image_out,w,h,"E:\\peacock_hough.raw");
}
//繪制直線算法一
void makeline1(BYTE *image_in[2000],int w,int h,double t,int k,int r,BYTE *image_out[2000])
{
int i,j,temp;
for(i=0;i<h;i++) //重新對數組上的白點作hough變換，如果發現某點落在(k,r)的區域內，在原圖與新圖中均做標記。
for(j=0;j<w;j++)
{
if(image_in[i][j]==255)
{
temp=(int)(j*cos(k*t)+i*sin(k*t));
if(temp<0)temp+=w+h+1;
if(temp==r)
{image_in[i][j]=250;image_out[i][j]=255;}
}
}
}
//繪制圖像算法二
void makeline2(BYTE *image_in[2000],int w,int h,double t,int k,int r,BYTE *image_out[2000])
{
int i,j,temp;
for(i=0;i<h;i++) //對原圖重新作hough變換，標記已經提取到的直線(k,r)上的白點
for(j=0;j<w;j++)
{
if(image_in[i][j]==255)
{
temp=(int)(j*cos(k*t)+i*sin(k*t));
if(temp<0)temp+=w+h+1;
if(temp==r)
{image_in[i][j]=250;}
}
}
for(j=0;j<w;j++) //在輸出數組image_out[][]中通過逆函數繪制直線
{
if(k!=0)
{
if(k>90&&k<180&&r>h)r-=w+h+1;
i=(int)(r/sin(k*t)-j/tan(k*t));
if(i>=0&&i<h)image_out[i][j]=255;
}
}
}
//以下為二值化，梯度銳化，中值濾波等函數，不做注釋！
void diedaiyz(BYTE *image[2000],int w,int h)
{
int i,j;
int count[256];
for(i=0;i<256;i++)
count[i]=0;
for(i=0;i<h;i++)
for(j=0;j<w;j++)
count[image[i][j]]++;
int T=100;
while(1)
{
int temp=T;
int t1=0,t2=0,t3=0,t4=0;
for(i=0;i<=temp;i++)
{t1+=count[i];t3+=count[i]*i;}
for(i=temp+1;i<256;i++)
{t2+=count[i];t4+=count[i]*i;}
T=(int)((t3/t1+t4/t2)/2);
if(T==temp)break;
}
for(i=0;i<h;i++)
for(j=0;j<w;j++)
if(image[i][j]>=T)image[i][j]=255;
else image[i][j]=0;
SaveRawFile(image,w,h,"E:\\peacock_diedaiyz.raw");
}
void sanfenyz(BYTE *image[2000],int w,int h)
{
int i,j;
int fmax=0,fmin=255;
for(i=0;i<h;i++)
for(j=0;j<w;j++)
{
if(image[i][j]>fmax)fmax=image[i][j];
if(image[i][j]<fmin)fmin=image[i][j];
}
int T=fmax-(fmax-fmin)/3;
for(i=0;i<h;i++)
for(j=0;j<w;j++)
if(image[i][j]>=T)image[i][j]=255;
else image[i][j]=0;
SaveRawFile(image,w,h,"E:\\peacock_sanfenyz.raw");
}
void sobel(BYTE *image_in[2000],int w,int h,BYTE *image_out[2000])
{
?? 快捷鍵說明
復制代碼
Ctrl + C
搜索代碼
Ctrl + F
全屏模式
F11
切換主題
Ctrl + Shift + D
顯示快捷鍵
?
增大字號
Ctrl + =
減小字號
Ctrl + -