开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 图形图像 > OpenCV > 查看源码
ddView.cpp
资源名称：contuorline_hausdorff.rar　[点击查看]
上传用户：wjfckcom
上传日期：2014-06-28
资源大小：1989k
文件大小：13k
源码类别：

OpenCV

开发平台：

Visual C++

ddView.cpp：源码内容
  1. // ddView.cpp : implementation of the CDdView class
  2. //
  4. #include "stdafx.h"
  5. #include "dd.h"
  7. #include "ddDoc.h"
  8. #include "ddView.h"
  9. #include "HistogramDlg.h"
  11. #ifdef _DEBUG
  12. #define new DEBUG_NEW
  13. #undef THIS_FILE
  14. static char THIS_FILE[] = __FILE__;
  15. #endif
  17. /////////////////////////////////////////////////////////////////////////////
  18. // CDdView
  20. IMPLEMENT_DYNCREATE(CDdView, CView)
  22. BEGIN_MESSAGE_MAP(CDdView, CView)
  23. //{{AFX_MSG_MAP(CDdView)
  24. ON_COMMAND(ID_HISTOGRAM, OnHistogram)
  25. ON_COMMAND(ID_IMG, OnImg)
  26. ON_COMMAND(ID_MATCHCONTOUR, OnMatchcontour)
  27. //}}AFX_MSG_MAP
  28. // Standard printing commands
  29. ON_COMMAND(ID_FILE_PRINT, CView::OnFilePrint)
  30. ON_COMMAND(ID_FILE_PRINT_DIRECT, CView::OnFilePrint)
  31. ON_COMMAND(ID_FILE_PRINT_PREVIEW, CView::OnFilePrintPreview)
  32. END_MESSAGE_MAP()
  34. /////////////////////////////////////////////////////////////////////////////
  35. // CDdView construction/destruction
  37. CDdView::CDdView()
  38. {
  39. // TODO: add construction code here
  41. }
  43. CDdView::~CDdView()
  44. {
  45. }
  47. BOOL CDdView::PreCreateWindow(CREATESTRUCT& cs)
  48. {
  49. // TODO: Modify the Window class or styles here by modifying
  50. //  the CREATESTRUCT cs
  52. return CView::PreCreateWindow(cs);
  53. }
  55. /////////////////////////////////////////////////////////////////////////////
  56. // CDdView drawing
  58. void CDdView::OnDraw(CDC* pDC)
  59. {
  60. CDdDoc* pDoc = GetDocument();
  61. ASSERT_VALID(pDoc);
  62. // TODO: add draw code for native data here
  63. pDoc->myimage.CopyOf(pDoc->pSrcImg,1);
  64. CRect rect;
  65. GetClientRect(rect);
  66. rect.bottom = pDoc->myimage.Height();
  67. rect.right = pDoc->myimage.Width();
  69. pDoc->myimage.DrawToHDC(pDC->m_hDC , rect);
  70. }
  72. /////////////////////////////////////////////////////////////////////////////
  73. // CDdView printing
  75. BOOL CDdView::OnPreparePrinting(CPrintInfo* pInfo)
  76. {
  77. // default preparation
  78. return DoPreparePrinting(pInfo);
  79. }
  81. void CDdView::OnBeginPrinting(CDC* /*pDC*/, CPrintInfo* /*pInfo*/)
  82. {
  83. // TODO: add extra initialization before printing
  84. }
  86. void CDdView::OnEndPrinting(CDC* /*pDC*/, CPrintInfo* /*pInfo*/)
  87. {
  88. // TODO: add cleanup after printing
  89. }
  91. /////////////////////////////////////////////////////////////////////////////
  92. // CDdView diagnostics
  94. #ifdef _DEBUG
  95. void CDdView::AssertValid() const
  96. {
  97. CView::AssertValid();
  98. }
  100. void CDdView::Dump(CDumpContext& dc) const
  101. {
  102. CView::Dump(dc);
  103. }
  105. CDdDoc* CDdView::GetDocument() // non-debug version is inline
  106. {
  107. ASSERT(m_pDocument->IsKindOf(RUNTIME_CLASS(CDdDoc)));
  108. return (CDdDoc*)m_pDocument;
  109. }
  110. #endif //_DEBUG
  112. /////////////////////////////////////////////////////////////////////////////
  113. // CDdView message handlers
  115. void CDdView::OnHistogram() 
  116. {
  117. int i,j;
  118. int width = GetDocument()->width;
  119. int height = GetDocument()->height;
  120. CvMemStorage * storage = cvCreateMemStorage(0);
  121. CvSeq * contour = 0;
  124. IplImage* pImage = cvCreateImage(cvSize(width,height),IPL_DEPTH_8U,1);
  125. pContourImg1 = cvCreateImage(cvSize(width,height),IPL_DEPTH_8U,3);
  126. Image = cvCreateImage(cvSize(width,height),8,1);
  127. IplImage* img = cvCreateImage(cvSize(width,height),8,3);
  128. cvCopy(GetDocument()->pSrcImg,img);
  129. cvCvtColor(GetDocument()->pSrcImg, Image, CV_RGB2GRAY);
  131. cvThreshold(Image,pImage,150, 255,CV_THRESH_BINARY_INV);
  132. cvSmooth(pImage, Image, CV_MEDIAN);
  135. cvCvtColor(Image, pContourImg1, CV_GRAY2BGR);
  136. cvFindContours( Image, storage, &contour, sizeof(CvContour), 
  137. CV_RETR_EXTERNAL, CV_CHAIN_APPROX_NONE);
  138. cvDrawContours(pContourImg1, contour, CV_RGB(255,0,255),
  139. CV_RGB(255, 0, 0), 1, 1, 8);
  141. n1 = contour->total; 
  142. CvPoint *PointArray=new CvPoint[n1]; 
  143. cvCvtSeqToArray( contour, PointArray, CV_WHOLE_SEQ );//提取点集
  145. seq1 = new struct contourrate[n1];
  147. double* r = (double*)calloc(n1,sizeof(double));
  148. double sum = 0.0;
  149. for(i = 0; i<n1;i++)
  150. {
  151. int x0 =PointArray[i].x;
  152. int y0 =PointArray[i].y;
  153. int x1 =PointArray[(i-5+n1)%n1].x;
  154. int y1 =PointArray[(i-5+n1)%n1].y;
  155. int x2 =PointArray[(i+5+n1)%n1].x;
  156. int y2 =PointArray[(i+5+n1)%n1].y;
  158. if(((uchar*)(pImage->imageData + pImage->widthStep*(y1/2+y2/2)))[x1/2+x2/2]==0)
  159. r[i] = 0 - (double)abs((y2-y1)*(x0-x1)+(x2-x1)*(y1-y0))/(double)((y2-y1)*(y2-y1)+(x2-x1)*(x2-x1));
  160. else
  161. r[i] = (double)abs((y2-y1)*(x0-x1)+(x2-x1)*(y1-y0))/(double)((y2-y1)*(y2-y1)+(x2-x1)*(x2-x1));
  162. sum += r[i];
  163. }
  165. for(i =0; i<n1; i++)
  166. {
  167. seq1[i].p.x = PointArray[i].x;
  168. seq1[i].p.y = PointArray[i].y;
  169. seq1[i].q = (float) (r[i]/sum) ;
  170. }
  172. /* int gray[256]={0} ;
  174. for(i = 0;i < height;i++)
  175. for(j=0; j< width; j++)
  176. {
  177. int n =((uchar*)(Image->imageData + Image->widthStep*i))[j];
  178. gray[n]++;
  179. }
  181. for(i=0; i<256; i++)
  182. tongji[i] = gray[i]/(width*height/1.0f);*/
  184. CHistogramDlg* dlg;
  185. dlg = new CHistogramDlg(this);
  186. dlg->Create(IDD_HISTOGRAMDLG);
  187. for(i=0; i<n1; i++)
  188. dlg->m_fIntensity[i*256/n1+1] = seq1[i].q;
  189. dlg->ShowWindow(SW_RESTORE);
  191. cvCopy(pContourImg1, GetDocument()->pSrcImg);
  192. Invalidate();
  193. free(r);
  195. cvReleaseImage(&pImage);
  196. cvReleaseImage(&img);
  199. }
  201. void CDdView::OnImg() 
  202. {
  204. CFileDialog filedlg(TRUE, NULL, NULL, OFN_HIDEREADONLY,
  205. "所有文件   (*.*)|*.*||BMP Files(*.BMP)|*.BMP|JPEG Files(*.JPG)|*.JPG|");
  206. if(filedlg.DoModal()!=IDOK)
  207. {
  208. }
  209. CString filename = filedlg.GetPathName();
  210. IplImage* img = cvLoadImage(filename,0);
  213. int i,j;
  214. int width = img->width;
  215. int height = img->height;
  216. CvMemStorage * storage = cvCreateMemStorage(0);
  217. CvSeq * contour = 0;
  218. pContourImg2 = cvCreateImage(cvSize(width,height),IPL_DEPTH_8U,3);
  219. IplImage* pImage = cvCreateImage(cvSize(width,height),IPL_DEPTH_8U,1);
  221. cvThreshold(img,pImage,150, 255,CV_THRESH_BINARY_INV);
  222. cvSmooth(pImage, img, CV_MEDIAN);
  225. cvCvtColor(img, pContourImg2, CV_GRAY2BGR);
  226. cvFindContours( img, storage, &contour, sizeof(CvContour), 
  227. CV_RETR_EXTERNAL, CV_CHAIN_APPROX_NONE);
  228. cvDrawContours(pContourImg2, contour, CV_RGB(255,0,255),
  229. CV_RGB(255, 0, 0), 1, 1, 8);
  231. n2 = contour->total; 
  232. CvPoint *PointArray=new CvPoint[n2]; 
  233. cvCvtSeqToArray( contour, PointArray, CV_WHOLE_SEQ );//提取点集
  235. seq2 = new struct contourrate [n2];
  237. double* r = (double*)calloc(n2,sizeof(double));
  238. double sum = 0.0;
  239. for(i = 0; i<n2;i++)
  240. {
  241. int x0 =PointArray[i].x;
  242. int y0 =PointArray[i].y;
  243. int x1 =PointArray[(i-5+n2)%n2].x;
  244. int y1 =PointArray[(i-5+n2)%n2].y;
  245. int x2 =PointArray[(i+5+n2)%n2].x;
  246. int y2 =PointArray[(i+5+n2)%n2].y;
  248. if(((uchar*)(pImage->imageData + pImage->widthStep*(y1/2+y2/2)))[x1/2+x2/2]==0)
  249. r[i] = 0 - (double)abs((y2-y1)*(x0-x1)+(x2-x1)*(y1-y0))/(double)((y2-y1)*(y2-y1)+(x2-x1)*(x2-x1));
  250. else
  251. r[i] = (double)abs((y2-y1)*(x0-x1)+(x2-x1)*(y1-y0))/(double)((y2-y1)*(y2-y1)+(x2-x1)*(x2-x1));
  252. sum += r[i];
  253. }
  255. for(i =0; i<n2; i++)
  256. {
  257. seq2[i].p.x = PointArray[i].x;
  258. seq2[i].p.y = PointArray[i].y;
  259. seq2[i].q =(float) (r[i]/sum);
  260. }
  262. CHistogramDlg* dlg;
  263. dlg = new CHistogramDlg(this);
  264. dlg->Create(IDD_HISTOGRAMDLG);
  265. for(i=0; i<n2; i++)
  266. dlg->m_fIntensity[i*256/n2+1] = seq2[i].q;
  267. dlg->ShowWindow(SW_RESTORE);
  269. cvNamedWindow("src", CV_WINDOW_AUTOSIZE);
  270. cvShowImage("src",pContourImg2);
  272. free(r);
  274. cvReleaseImage(&pImage);
  275. cvReleaseImage(&img);
  277. }
  279. void CDdView::OnMatchcontour() 
  280. {
  281. //t1 ...t4 alloc
  282. t1 = new struct rateseq[n1/5];
  283. t2 = new struct rateseq[n1/5];
  284. t3 = new struct rateseq[n2/5];
  285. t4 = new struct rateseq[n2/5];
  287. int i, j;
  288. int m1, m2, m3,m4;
  289. m1 = m2 = m3 = m4 = 0;
  291. for(i=1; i<n1-1; i++)
  292. {
  293. if(seq1[i].q>seq1[i-1].q&&seq1[i].q>seq1[i+1].q&&seq1[i].q>0)
  294. {
  295. t1[m1].q=seq1[i].q;
  296. t1[m1].index = i;
  297. m1++;
  298. continue;
  299. }
  300. else if(seq1[i].q<seq1[i-1].q&&seq1[i].q<seq1[i+1].q&&seq1[i].q<0)
  301. {
  302. t2[m2].q=seq1[i].q;
  303. t2[m2].index = i;
  304. m2++;
  305. continue;
  306. }
  307. continue;
  308. }
  310. for(i=1; i<n2-1; i++)
  311. {
  312. if(seq2[i].q>seq2[i-1].q&&seq2[i].q>seq2[i+1].q&&seq2[i].q>0)
  313. {
  314. t3[m3].q=seq2[i].q;
  315. t3[m3].index=i;
  316. m3++;
  317. continue;
  318. }
  319. else if(seq2[i].q<seq2[i-1].q&&seq2[i].q<seq2[i+1].q&&seq2[i].q<0)
  320. {
  321. t4[m4].q=seq2[i].q;
  322. t4[m4].index = i;
  323. m4++;
  324. continue;
  325. }
  326. continue;
  327. }
  330. /* IplImage* resultimg = cvCreateImage(cvSize(Image->width,Image->height),8,3);
  332. cvCvtColor(Image,resultimg,CV_GRAY2BGR); */
  333. IplImage* resultimg = stack_imgs(pContourImg1, pContourImg2);
  335. for(i=0; i<m1-1; i++)
  336. for(j=0; j<m3-1; j++)
  337. {
  338. if(fabs(t1[i].q-t3[j].q)<0.05*(t1[i].q+t3[j].q))
  339. {
  340. float hausv1 = Hausdorff(t1[i],t3[j], seq1 ,seq2);
  341. if (hausv1<THREHOLD)
  342. {
  343. /* int z = 0;
  345. if (t1[i].index<10)
  346. {
  348. for(int k=0; k<t1[i].index+10; k++)
  349. {
  350. ((uchar*)(resultimg->imageData + resultimg->widthStep*(seq1[k].p.y)))[(seq1[k].p.x)*3] = 0;
  351. ((uchar*)(resultimg->imageData + resultimg->widthStep*(seq1[k].p.y)))[(seq1[k].p.x)*3+1] = 125;
  352. ((uchar*)(resultimg->imageData + resultimg->widthStep*(seq1[k].p.y)))[(seq1[k].p.x)*3+2] = z;
  353. }
  354. z+=30;
  355. }
  356. else if (t1[i].index>n1-10)
  357. {
  358. for(int k=t1[i].index-10; k<n1-10; k++)
  359. {
  360. ((uchar*)(resultimg->imageData + resultimg->widthStep*(seq1[k].p.y)))[(seq1[k].p.x)*3] = z;
  361. ((uchar*)(resultimg->imageData + resultimg->widthStep*(seq1[k].p.y)))[(seq1[k].p.x)*3+1] = 125;
  362. ((uchar*)(resultimg->imageData + resultimg->widthStep*(seq1[k].p.y)))[(seq1[k].p.x)*3+2] = 0;
  363. }
  364. z+=30;
  365. }
  366. else
  367. {
  368. for(int k=t1[i].index-10; k<t1[i].index+10; k++)
  369. {
  370. ((uchar*)(resultimg->imageData + resultimg->widthStep*(seq1[k].p.y)))[(seq1[k].p.x)*3] = z;
  371. ((uchar*)(resultimg->imageData + resultimg->widthStep*(seq1[k].p.y)))[(seq1[k].p.x)*3+1] = 125;
  372. ((uchar*)(resultimg->imageData + resultimg->widthStep*(seq1[k].p.y)))[(seq1[k].p.x)*3+2] = z;
  373. }
  374. z+=30;
  375. }*/
  377. /*int m,n;
  378. for(m=0; m<10&&t1[i].index-m>0&&t3[j].index-m>0;m++)
  379. {
  380. CvPoint pt1 = seq1[(t1[i].index-m)].p;
  381. CvPoint pt2 = seq2[(t3[j].index-m)].p;
  382. pt2.y+=pContourImg1->height;
  383. cvLine( resultimg, pt1, pt2, CV_RGB(255,0,i*50%255), 1, 8, 0 );
  384. }
  385. for(n=1; n<10&&t1[i].index+n<n1&&t3[j].index+n<n2;n++)
  386. {
  387. CvPoint pt1 = seq1[t1[i].index+n].p;
  388. CvPoint pt2 = seq2[t3[j].index+n].p;
  389. pt2.y+=pContourImg1->height;
  390. cvLine( resultimg, pt1, pt2, CV_RGB(255,i*50%255,255), 1, 8, 0 );
  391. }*/
  392. CvPoint pt1 = seq1[t1[i].index].p;
  393. CvPoint pt2 = seq2[t3[j].index].p;
  394. pt2.y+=pContourImg1->height;
  395. cvLine( resultimg, pt1, pt2, CV_RGB(255,i*50%255,255), 1, 8, 0 );
  398. }
  399. }
  400. continue;
  401. }
  403. for(i=0; i<m2; i++)
  404. for(j=0; j<m4; j++)
  405. {
  406. if (fabs(t2[i].q-t4[j].q)<0.05*abs(t2[i].q+t4[j].q))
  407. {
  408. float hausv2 = Hausdorff(t2[i], t4[j], seq1, seq2);
  409. if (hausv2<THREHOLD)
  410. {
  411. /* if (t2[i].index<10)
  412. {
  413. for(int k=0; k<t2[i].index+10; k++)
  414. {
  415. ((uchar*)(resultimg->imageData + resultimg->widthStep*(seq1[k].p.y)))[(seq1[k].p.x)*3] = 255;
  416. ((uchar*)(resultimg->imageData + resultimg->widthStep*(seq1[k].p.y)))[(seq1[k].p.x)*3+1] = 255;
  417. ((uchar*)(resultimg->imageData + resultimg->widthStep*(seq1[k].p.y)))[(seq1[k].p.x)*3+2] = 0;
  418. }
  419. }
  420. else if (t2[i].index>n1-10)
  421. {
  422. for(int k=t2[i].index-10; k<n1-10; k++)
  423. {
  424. ((uchar*)(resultimg->imageData + resultimg->widthStep*(seq1[k].p.y)))[(seq1[k].p.x)*3] = 255;
  425. ((uchar*)(resultimg->imageData + resultimg->widthStep*(seq1[k].p.y)))[(seq1[k].p.x)*3+1] =255;
  426. ((uchar*)(resultimg->imageData + resultimg->widthStep*(seq1[k].p.y)))[(seq1[k].p.x)*3+2] = 0;
  427. }
  428. }
  429. else
  430. {
  431. for(int k=t2[i].index-10; k<t2[i].index+10; k++)
  432. {
  433. ((uchar*)(resultimg->imageData + resultimg->widthStep*(seq1[k].p.y)))[(seq1[k].p.x)*3] = 255;
  434. ((uchar*)(resultimg->imageData + resultimg->widthStep*(seq1[k].p.y)))[(seq1[k].p.x)*3+1] = 255;
  435. ((uchar*)(resultimg->imageData + resultimg->widthStep*(seq1[k].p.y)))[(seq1[k].p.x)*3+2] = 0;
  436. }
  437. }*/
  438. /*int m,n;
  439. for(m=0; m<10&&t2[i].index-m>0&&t4[j].index-m>0;m++)
  440. {
  441. CvPoint pt1 = seq1[t2[i].index-m].p;
  442. CvPoint pt2 = seq2[t4[j].index-m].p;
  443. pt2.y+=pContourImg1->height;
  444. cvLine( resultimg, pt1, pt2, CV_RGB(i*50%255,0,255), 1, 8, 0 );
  445. }
  446. for(n=1; n<10&&t2[i].index+n<n1&&t4[j].index+n<n2;n++)
  447. {
  448. CvPoint pt1 = seq1[(t2[i].index+n)].p;
  449. CvPoint pt2 = seq2[t4[j].index+n].p;
  450. pt2.y+=pContourImg1->height;
  451. cvLine( resultimg, pt1, pt2, CV_RGB(255,i*50%255,i*50%255), 1, 8, 0 );
  452. }*/
  453. CvPoint pt1 = seq1[t2[i].index].p;
  454. CvPoint pt2 = seq2[t4[j].index].p;
  455. pt2.y+=pContourImg1->height;
  456. cvLine( resultimg, pt1, pt2, CV_RGB(255,i*50%255,0), 1, 8, 0 );
  458. }
  459. }
  460. continue;
  461. }
  463. cvNamedWindow("matchresult",CV_WINDOW_AUTOSIZE);
  464. cvShowImage("matchresult",resultimg);
  466. delete [] t1;
  467. delete [] t2;
  468. delete [] t3;
  469. delete [] t4;
  470. delete [] seq1;
  471. delete [] seq2;
  472. cvReleaseImage(&Image);
  473. cvReleaseImage(&pContourImg2);
  474. cvReleaseImage(&pContourImg1);
  475. cvReleaseImage(&resultimg);
  476. }