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

源码开发语言/平台
当前位置：首页 > 源码/资料 > 图形图像 > 波变换 > 查看源码
DlgRecMatch.cpp
资源名称：Image_Processing.rar　[点击查看]
上传用户：wangchdtw
上传日期：2014-04-02
资源大小：329k
文件大小：18k
源码类别：

波变换

开发平台：

Visual C++

DlgRecMatch.cpp：源码内容
  1. // DlgRecMatch.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "ImageProcessing.h"
  6. #include "DlgRecMatch.h"
  7. #include "CDib.h"
  8. #include "GlobalApi.h"
  10. #ifdef _DEBUG
  11. #define new DEBUG_NEW
  12. #undef THIS_FILE
  13. static char THIS_FILE[] = __FILE__;
  14. #endif
  16. /////////////////////////////////////////////////////////////////////////////
  17. // CDlgRecMatch dialog
  20. CDlgRecMatch::CDlgRecMatch(CWnd* pParent /*=NULL*/,CImageProcessingDoc* pDoc)
  21. : CDialog(CDlgRecMatch::IDD, pParent)
  22. {
  23. //{{AFX_DATA_INIT(CDlgRecMatch)
  24. // NOTE: the ClassWizard will add member initialization here
  25. //}}AFX_DATA_INIT
  26. // 设置计算图象控件位置标志位为FALSE
  27. m_bCalImgLoc = FALSE;
  29. // 设置初始图象
  30. m_pDibInit = pDoc->m_pDibInit;
  32. // 获得文档指针
  33. m_pDoc = pDoc;
  35. // 分配模板图象内存
  36. m_pDibModel = new CDib;
  38. // 分配结果图象
  39. m_pDibResult = new CDib;
  42. }
  44. void CDlgRecMatch::DoDataExchange(CDataExchange* pDX)
  45. {
  46. CDialog::DoDataExchange(pDX);
  47. //{{AFX_DATA_MAP(CDlgRecMatch)
  48. // NOTE: the ClassWizard will add DDX and DDV calls here
  49. //}}AFX_DATA_MAP
  50. }
  53. BEGIN_MESSAGE_MAP(CDlgRecMatch, CDialog)
  54. //{{AFX_MSG_MAP(CDlgRecMatch)
  55. ON_BN_CLICKED(IDC_RECOG_OPEN_MODEL, OnRecogOpenModel)
  56. ON_BN_CLICKED(IDC_RECOG_MATCH, OnRecogMatch)
  57. ON_WM_PAINT()
  58. //}}AFX_MSG_MAP
  59. END_MESSAGE_MAP()
  61. /////////////////////////////////////////////////////////////////////////////
  62. // CDlgRecMatch message handlers
  64. void CDlgRecMatch::OnRecogOpenModel() 
  65. {
  66. CFileDialog dlg(TRUE,"bmp","*.bmp");
  67. if(dlg.DoModal() == IDOK)
  68. {
  69.  
  70.   CFile file;
  71.  
  72.   CString strPathName;
  73.  
  74. strPathName = dlg.GetPathName();
  75.  
  76. // 打开文件
  77. if( !file.Open(strPathName, CFile::modeRead | CFile::shareDenyWrite))
  78. {
  79. // 返回
  80. return ;
  81. }
  83. // 读入模板图象
  84. if(!m_pDibModel->Read(&file)){
  85. // 恢复光标形状
  86. EndWaitCursor();
  88. // 清空已分配内存
  89. m_pDibModel->Empty();
  91. // 返回
  92.   return;
  93. }
  94. }
  96. // 判断是否是8-bpp位图（这里为了方便，只处理8-bpp位图的模板配准，其它的可以类推）
  97. if(m_pDibModel->m_nColorTableEntries != 256)
  98. {
  99. // 提示用户
  100. MessageBox("目前只支持256色位图！", "系统提示" , MB_ICONINFORMATION | MB_OK);
  102. // 清空已分配内存
  103. m_pDibModel->Empty();
  105. // 返回
  106. return;
  107. }
  109. // 初始图象的长宽大小
  110. CSize sizeImage = m_pDibInit->GetDimensions();
  111. int nImageWidth = sizeImage.cx ;
  112. int nImageHeight = sizeImage.cy ;
  114. // 模板图象的长宽大小
  115. CSize sizeModelImage = m_pDibModel->GetDimensions() ;
  116. int nModelWidth = sizeImage.cx ;
  117. int nModelHeight = sizeImage.cy ;
  119. // 判断模板尺寸和初始图象的大小，如果模板大于初始图象，则推出
  120. if(nModelHeight > nImageHeight || nModelWidth > nImageWidth )
  121. {
  122. // 提示用户
  123. MessageBox("模板尺寸大于源图像尺寸！", "系统提示" , MB_ICONINFORMATION | MB_OK);
  125. // 释放已分配内存
  126. m_pDibModel->Empty();
  128. // 返回
  129. return;
  131. }
  133. // 如果打开新的待配准文件，将图象位置设置标志位设为FALSE，以便再次调整位置
  134. m_bCalImgLoc = FALSE;
  136. // 更新显示
  137. this->Invalidate();
  139. }
  141. /*************************************************************************
  142.  *
  143.  * 函数名称：
  144.  *   CalImageLocation()
  145.  *
  146.  * 输入参数:
  147.  *   无
  148.  *
  149.  * 返回值:
  150.  *   无
  151.  *
  152.  * 说明:
  153.  *   该函数设置对话框中的控件位置和大小，并设置显示图象的位置。默认的图象大小为352×288，如果图象小于
  154.  *此大小，则控件大小设置为256*256，并将图象放置在控件中间。
  155.  *
  156.  *************************************************************************
  157.  */
  158. void CDlgRecMatch::CalImageLocation()
  159. {
  160. // ------------------------------------------------------
  161. // 获得控件IDC_RECOG_INIIMAGE的句柄，并获得控件的初始位置信息
  162. CWnd* pWnd=GetDlgItem(IDC_RECOG_INIIMAGE);
  163. WINDOWPLACEMENT *winPlacement;
  164. winPlacement=new WINDOWPLACEMENT;
  165. pWnd->GetWindowPlacement(winPlacement);
  167. // 图象宽度
  168. int nImageWidth;
  169. nImageWidth = m_pDibInit->m_lpBMIH->biWidth;
  171. // 图象高度
  172. int nImageHeight;
  173. nImageHeight = m_pDibInit->m_lpBMIH->biHeight;
  175. // -----------------------------------------------------------------
  176. // 调整控件IDC_RECOG_INIIMAGE的大小位置，并同时设置显示基准图象的位置
  177. if(nImageHeight > 256){
  178. winPlacement->rcNormalPosition.bottom = winPlacement->rcNormalPosition.top + nImageHeight;
  179. m_rectInitImage.bottom = winPlacement->rcNormalPosition.bottom;
  180. m_rectInitImage.top    = winPlacement->rcNormalPosition.top;
  181. }
  182. else{
  183. winPlacement->rcNormalPosition.bottom = winPlacement->rcNormalPosition.top + 256;
  184. m_rectInitImage.bottom = winPlacement->rcNormalPosition.top + 128 + nImageHeight/2;
  185. m_rectInitImage.top    = winPlacement->rcNormalPosition.top + 128 - nImageHeight/2;
  186. }
  187. if(nImageWidth > 256){
  188. winPlacement->rcNormalPosition.right = winPlacement->rcNormalPosition.left + nImageWidth;
  189. m_rectInitImage.right = winPlacement->rcNormalPosition.right;
  190. m_rectInitImage.left  = winPlacement->rcNormalPosition.left;
  191. }
  192. else{
  193. winPlacement->rcNormalPosition.right = winPlacement->rcNormalPosition.left + 256;
  194. m_rectInitImage.right = winPlacement->rcNormalPosition.left + 128 + nImageWidth/2;
  195. m_rectInitImage.left  = winPlacement->rcNormalPosition.left + 128 - nImageWidth/2;
  196. }
  198. // 设置IDC_RECOG_INIIMAGE控件的大小位置
  199. pWnd->SetWindowPlacement(winPlacement);
  201. // 获得显示模板图象控件的右边位置，以便确认显示模板图象控件的位置
  202. int nIniImgRight;
  203. nIniImgRight = winPlacement->rcNormalPosition.right;
  205. int  nIniImgLeft;
  206. nIniImgLeft   = winPlacement->rcNormalPosition.left;
  208. // 获得IDC_REG_INIT_IMAGE控件的下边位置，以便调整其他控件的位置
  209. int nIniImgBottom;
  210. nIniImgBottom = winPlacement->rcNormalPosition.bottom;
  212. // 获得IDC_REG_INIT_IMAGE控件的下边位置，以便调整其他控件的位置
  213. int nIniImgtop = winPlacement->rcNormalPosition.top;
  215. // ------------------------------------------------------
  216. // 获得控件IDC_RECOG_MODIMAGE的句柄，并获得初始位置信息
  217. pWnd=GetDlgItem(IDC_RECOG_MODIMAGE);
  218. pWnd->GetWindowPlacement(winPlacement);
  220. // 如果还未打开模板图象，则设置结果图象大小和初始图象大小相等
  221. if(!m_pDibModel->IsEmpty()){
  222. nImageWidth  = m_pDibModel->m_lpBMIH->biWidth;
  223. nImageHeight = m_pDibModel->m_lpBMIH->biHeight;
  224. }
  226. // 调整控件IDC_RECOG_MODIMAGE的大小位置，并同时设置显示结果图象的位置
  228. // 先调整控件的左边位置，和IDC_REG_INIT_IMAGE控件相隔15个象素
  229. winPlacement->rcNormalPosition.left = nIniImgRight + 15;
  231. if(nImageHeight > 256){
  232. winPlacement->rcNormalPosition.top = nIniImgtop;
  233. winPlacement->rcNormalPosition.bottom = winPlacement->rcNormalPosition.top + nImageHeight;
  234. m_rectModelImage.bottom = winPlacement->rcNormalPosition.bottom;
  235. m_rectModelImage.top    = nIniImgtop;
  236. }
  237. else{
  238. winPlacement->rcNormalPosition.top = nIniImgtop;
  239. winPlacement->rcNormalPosition.bottom = winPlacement->rcNormalPosition.top + 256;
  240. m_rectModelImage.bottom = winPlacement->rcNormalPosition.top + 128 + nImageHeight/2;
  241. m_rectModelImage.top    = winPlacement->rcNormalPosition.top + 128 - nImageHeight/2;
  242. }
  243. if(nImageWidth > 256){
  244. winPlacement->rcNormalPosition.right = winPlacement->rcNormalPosition.left + nImageWidth;
  245. m_rectModelImage.right = winPlacement->rcNormalPosition.right;
  246. m_rectModelImage.left  = winPlacement->rcNormalPosition.left;
  247. }
  248. else{
  249. winPlacement->rcNormalPosition.right = winPlacement->rcNormalPosition.left + 256;
  250. m_rectModelImage.right = winPlacement->rcNormalPosition.left + 128 + nImageWidth/2;
  251. m_rectModelImage.left  = winPlacement->rcNormalPosition.left + 128 - nImageWidth/2;
  252. }
  254. // 设置IDC_RECOG_MODIMAGE控件的大小位置
  255. pWnd->SetWindowPlacement(winPlacement);
  257. // 获得IDC_RECOG_MODIMAGE控件的右边位置，以便调整其他控件的位置
  258. nIniImgRight  = winPlacement->rcNormalPosition.right;
  260. // ------------------------------------------------------
  261. // 获得控件IDC_RECOG_RESLTIMAGE的句柄，并获得初始位置信息
  262. pWnd=GetDlgItem(IDC_RECOG_RESLTIMAGE);
  263. pWnd->GetWindowPlacement(winPlacement);
  265. // 如果还未生成结果图象，则设置结果图象大小和初始图象大小相等
  266. if(!m_pDibResult->IsEmpty()){
  267. nImageWidth  = m_pDibResult->m_lpBMIH->biWidth;
  268. nImageHeight = m_pDibResult->m_lpBMIH->biHeight;
  269. }
  271. // 调整控件IDC_RECOG_RESLTIMAGE的大小位置，并同时设置显示结果图象的位置
  273. // 先调整控件的左边位置，和IDC_RECOG_MODIMAGE控件相隔15个象素
  274. winPlacement->rcNormalPosition.left = nIniImgRight + 15;
  276. if(nImageHeight > 256){
  277. winPlacement->rcNormalPosition.top = nIniImgtop;
  278. winPlacement->rcNormalPosition.bottom = winPlacement->rcNormalPosition.top + nImageHeight;
  279. m_rectResltImage.bottom = winPlacement->rcNormalPosition.bottom;
  280. m_rectResltImage.top    = winPlacement->rcNormalPosition.top;
  281. }
  282. else{
  283. winPlacement->rcNormalPosition.top = nIniImgtop;
  284. winPlacement->rcNormalPosition.bottom = winPlacement->rcNormalPosition.top + 256;
  285. m_rectResltImage.bottom = winPlacement->rcNormalPosition.top + 128 + nImageHeight/2;
  286. m_rectResltImage.top    = winPlacement->rcNormalPosition.top + 128 - nImageHeight/2;
  287. }
  288. if(nImageWidth > 256){
  289. winPlacement->rcNormalPosition.right = winPlacement->rcNormalPosition.left + nImageWidth;
  290. m_rectResltImage.right = winPlacement->rcNormalPosition.right;
  291. m_rectResltImage.left  = winPlacement->rcNormalPosition.left;
  292. }
  293. else{
  294. winPlacement->rcNormalPosition.right = winPlacement->rcNormalPosition.left + 256;
  295. m_rectResltImage.right = winPlacement->rcNormalPosition.left + 128 + nImageWidth/2;
  296. m_rectResltImage.left  = winPlacement->rcNormalPosition.left + 128 - nImageWidth/2;
  297. }
  299. // 设置IDC_REG_RESLT_IMAGE控件的大小位置
  300. pWnd->SetWindowPlacement(winPlacement);
  302. // ------------------------------------------------------
  303. if(nIniImgBottom > winPlacement->rcNormalPosition.bottom)
  304. nIniImgBottom = winPlacement->rcNormalPosition.bottom;
  306. nIniImgRight  = winPlacement->rcNormalPosition.right;
  308. // 设置控件IDOK的位置大小
  309. pWnd=GetDlgItem(IDOK);
  310. pWnd->GetWindowPlacement(winPlacement);
  311. winPlacement->rcNormalPosition.top = nIniImgBottom +15;
  312. winPlacement->rcNormalPosition.bottom = nIniImgBottom + 60;
  313. pWnd->SetWindowPlacement(winPlacement);
  315. // 设置控件IDCANCEL的位置大小
  316. pWnd=GetDlgItem(IDCANCEL);
  317. pWnd->GetWindowPlacement(winPlacement);
  318. winPlacement->rcNormalPosition.top = nIniImgBottom +15;
  319. winPlacement->rcNormalPosition.bottom = nIniImgBottom + 60;
  320. pWnd->SetWindowPlacement(winPlacement);
  322. // 设置控件IDC_RECOG_OPEN_MODEL的位置大小
  323. pWnd=GetDlgItem(IDC_RECOG_OPEN_MODEL);
  324. pWnd->GetWindowPlacement(winPlacement);
  325. winPlacement->rcNormalPosition.top = nIniImgBottom +15;
  326. winPlacement->rcNormalPosition.bottom = nIniImgBottom + 60;
  327. pWnd->SetWindowPlacement(winPlacement);
  329. // 设置控件IDC_RECOG_MATCH的位置大小
  330. pWnd=GetDlgItem(IDC_RECOG_MATCH);
  331. pWnd->GetWindowPlacement(winPlacement);
  332. winPlacement->rcNormalPosition.top = nIniImgBottom +15;
  333. winPlacement->rcNormalPosition.bottom = nIniImgBottom + 60;
  334. pWnd->SetWindowPlacement(winPlacement);
  336. // 调整此对话框的大小
  337. this->GetWindowPlacement(winPlacement);
  338. //winPlacement->rcNormalPosition.top = nIniImgtop -50;
  339. winPlacement->rcNormalPosition.bottom = nIniImgBottom + 272;
  340. winPlacement->rcNormalPosition.left   = nIniImgLeft   - 20;
  341. winPlacement->rcNormalPosition.right  = nIniImgRight  + 20;
  342. this->SetWindowPlacement(winPlacement);
  344. // 释放已分配内存
  345. delete winPlacement;
  347. // 设置计算图象控件位置标志位为TRUE
  348. m_bCalImgLoc = TRUE;
  349. }
  351. void CDlgRecMatch::OnPaint() 
  352. {
  353. CPaintDC dc(this); // device context for painting
  355. // 如果还没有计算图象的位置，则进行计算
  356. if(!m_bCalImgLoc){
  357. CalImageLocation();
  358. }
  360. // 显示大小
  361. CSize sizeDisplay;
  363. // 显示位置
  364. CPoint pointDisplay;
  366. // 显示初始图象
  367. if(!m_pDibInit->IsEmpty()){
  368. sizeDisplay.cx=m_pDibInit->m_lpBMIH->biWidth;
  369. sizeDisplay.cy=m_pDibInit->m_lpBMIH->biHeight;
  370. pointDisplay.x = m_rectInitImage.left;
  371. pointDisplay.y = m_rectInitImage.top;
  372. m_pDibInit->Draw(&dc,pointDisplay,sizeDisplay);
  373. }
  375. // 显示模板图象
  376. if(!m_pDibModel->IsEmpty()){
  377. sizeDisplay.cx=m_pDibModel->m_lpBMIH->biWidth;
  378. sizeDisplay.cy=m_pDibModel->m_lpBMIH->biHeight;
  379. pointDisplay.x = m_rectModelImage.left;
  380. pointDisplay.y = m_rectModelImage.top;
  381. m_pDibModel->Draw(&dc,pointDisplay,sizeDisplay);
  382. }
  384. // 显示结果图象
  385. if(!m_pDibResult->IsEmpty()){
  386. sizeDisplay.cx=m_pDibResult->m_lpBMIH->biWidth;
  387. sizeDisplay.cy=m_pDibResult->m_lpBMIH->biHeight;
  388. pointDisplay.x = m_rectResltImage.left;
  389. pointDisplay.y = m_rectResltImage.top;
  390. m_pDibResult->Draw(&dc,pointDisplay,sizeDisplay);
  391. }
  393. }
  395. /*************************************************************************
  396.  *
  397.  * 函数名称：
  398.  *   TemplateMatch()
  399.  *
  400.  * 输入参数:
  401.  *   CDib* pDibSrc - 指向CDib类的指针，含有待匹配图象信息 
  402.  *   CDib* pDibTemplate - 指向CDib类的指针，含有模板图象信息 
  403.  *
  404.  * 返回值:
  405.  *   BOOL - 成功则返回TRUE，否则返回FALSE
  406.  *
  407.  * 说明:
  408.  *   该函数将对图象进行模板匹配操作。需要注意的是，此程序只处理256灰度级的
  409.  *图象。
  410.  *
  411.  *************************************************************************
  412.  */
  413. BOOL CDlgRecMatch::TemplateMatch(CDib* pDibSrc, CDib* pDibTemplate)
  414. {
  415. // 指向源图像的指针
  416. LPBYTE lpSrc,lpTemplateSrc;
  418. // 指向缓存图像的指针
  419. LPBYTE lpDst;
  421. //循环变量
  422. long i;
  423. long j;
  424. long m;
  425. long n;
  427. //中间结果
  428. double dSigmaST;
  429. double dSigmaS;
  430. double dSigmaT;
  432. //相似性测度
  433. double R;
  435. //最大相似性测度
  436. double  dbMaxR;
  438. //最大相似性出现位置
  439. int nMaxWidth;
  440. int nMaxHeight;
  442. //像素值
  443. unsigned char unchPixel;
  444. unsigned char unchTemplatePixel;
  446. // 获得图象数据存储的高度和宽度
  447. CSize sizeSaveImage;
  448. sizeSaveImage = pDibSrc->GetDibSaveDim();
  450. // 获得模板图象数据存储的高度和宽度
  451. CSize sizeSaveTemplate;
  452. sizeSaveTemplate = pDibTemplate->GetDibSaveDim();
  455. // 暂时分配内存，以保存新图像
  456. CDib* pDibNew;
  457. pDibNew = new CDib;
  459. // 如果分配内存失败，则推出
  460. if(!CopyDIB(pDibSrc,pDibNew)){
  461. // 释放已分配内存
  462. pDibNew->Empty();
  464. // 返回
  465. return FALSE;
  466. }
  469. // 初始化新分配的内存
  470. lpDst = (LPBYTE)pDibNew->m_lpImage; 
  472. // 图象的高度
  473. int nImageHeight ;
  474. nImageHeight = pDibSrc->m_lpBMIH->biHeight;
  476. // 图象的宽度
  477. int nImageWidth;
  478. nImageWidth = pDibSrc->m_lpBMIH->biWidth;
  480. // 模板图象的高度
  481. int nTemplateHeight;
  482. nTemplateHeight = pDibTemplate->m_lpBMIH->biHeight;
  484. // 模板图象的宽度
  485. int nTemplateWidth;
  486. nTemplateWidth = pDibTemplate->m_lpBMIH->biWidth;
  488. //计算dSigmaT
  489. dSigmaT = 0;
  490. for (n = 0;n < nTemplateHeight ;n++)
  491. {
  492. for(m = 0;m < nTemplateWidth ;m++)
  493. {
  494. // 指向模板图像倒数第j行，第i个象素的指针
  495. lpTemplateSrc = (LPBYTE)pDibTemplate->m_lpImage + sizeSaveTemplate.cx * n + m;
  496. unchTemplatePixel = (unsigned char)*lpTemplateSrc;
  497. dSigmaT += (double)unchTemplatePixel*unchTemplatePixel;
  498. }
  499. }
  501. //找到图像中最大相似性的出现位置
  502.  dbMaxR = 0.0;
  503. for (j = 0;j < nImageHeight - nTemplateHeight +1 ;j++)
  504. {
  505. for(i = 0;i < nImageWidth - nTemplateWidth + 1;i++)
  506. {
  507. dSigmaST = 0;
  508. dSigmaS = 0;
  510. for (n = 0;n < nTemplateHeight ;n++)
  511. {
  512. for(m = 0;m < nTemplateWidth ;m++)
  513. {
  514. // 指向源图像倒数第j+n行，第i+m个象素的指针
  515. lpSrc  = (LPBYTE)pDibSrc->m_lpImage + sizeSaveImage.cx * (j+n) + (i+m);
  517. // 指向模板图像倒数第n行，第m个象素的指针
  518. lpTemplateSrc  = (LPBYTE)pDibTemplate->m_lpImage + sizeSaveTemplate.cx * n + m;
  520. unchPixel = (unsigned char)*lpSrc;
  521. unchTemplatePixel = (unsigned char)*lpTemplateSrc;
  523. dSigmaS += (double)unchPixel*unchPixel;
  524. dSigmaST += (double)unchPixel*unchTemplatePixel;
  525. }
  526. }
  527. //计算相似性
  528. R = dSigmaST / ( sqrt(dSigmaS)*sqrt(dSigmaT));
  529. //与最大相似性比较
  530. if (R >  dbMaxR)
  531. {
  532.  dbMaxR = R;
  533. nMaxWidth = i;
  534. nMaxHeight = j;
  535. }
  536. }
  537. }
  539. // 对目标图象的象素进行赋值
  540. for(i=0; i<nImageHeight; i++)
  541. for( j=0; j<nImageWidth; j++){
  542. lpDst[i*sizeSaveImage.cx +j] /=2;
  543. }
  545. //将最大相似性出现区域部分复制到目标图像
  546. for (n = 0;n < nTemplateHeight ;n++)
  547. {
  548. for(m = 0;m < nTemplateWidth ;m++)
  549. {
  550. lpTemplateSrc = (LPBYTE)pDibTemplate->m_lpImage + sizeSaveTemplate.cx * n + m;
  551. lpDst = (LPBYTE)pDibNew->m_lpImage + sizeSaveImage.cx * (n+nMaxHeight) + (m+nMaxWidth);
  552. *lpDst = *lpTemplateSrc;
  553. }
  554. }
  556. // 复制图像
  557. memcpy(pDibSrc->m_lpImage, pDibNew->m_lpImage, nImageWidth * nImageHeight);
  559. // 释放内存
  560. pDibNew->Empty();
  562. // 返回
  563. return TRUE;
  564. }
  567. /*************************************************************************
  568.  *
  569.  * 函数名称：
  570.  *   OnRecogMatch()
  571.  *
  572.  * 输入参数:
  573.  *   无
  574.  * 
  575.  * 返回值:
  576.  *   无
  577.  *
  578.  * 说明:
  579.  *   根据图象模板，在待匹配的图象中找到匹配的位置
  580.  *
  581.  *************************************************************************
  582.  */
  583. void CDlgRecMatch::OnRecogMatch() 
  584. {
  585. // 更改光标形状
  586. BeginWaitCursor();
  588. // 分配结果图象内存
  589. if(!m_pDibInit->IsEmpty()){
  590. // 如果分配内存失败，则推出
  591. if(!CopyDIB(m_pDibInit,m_pDibResult)){
  593. // 释放已分配内存
  594. m_pDibResult->Empty();
  596. // 返回
  597. return ;
  598. }
  599. }
  601. // 设置计算图象控件位置标志位为FALSE以重新设置图象控件位置
  602. m_bCalImgLoc = FALSE;
  604. // 调用TemplateMatch()函数进行模板匹配
  605. if (TemplateMatch(m_pDibResult, m_pDibModel))
  606. {
  607. // 更新显示
  608. Invalidate();
  609. }
  610. else
  611. {
  612. // 提示用户
  613. MessageBox("分配内存失败！", "系统提示" , MB_ICONINFORMATION | MB_OK);
  614. }
  616. // 恢复光标
  617. EndWaitCursor();
  619. // 更新视图
  620. Invalidate();
  622. }
  624. void CDlgRecMatch::OnOK() 
  625. {
  626. // TODO: Add extra validation here
  627. // 释放已分配内存
  628. if(!m_pDibModel->IsEmpty()){
  629. m_pDibModel->Empty();
  630. m_pDibModel = NULL;
  631. }
  632. if(!m_pDibResult->IsEmpty()){
  633. m_pDibResult->Empty();
  634. m_pDibResult = NULL;
  635. }
  637. CDialog::OnOK();
  638. }
  640. void CDlgRecMatch::OnCancel() 
  641. {
  642. // TODO: Add extra cleanup here
  644. // 释放已分配内存
  645. if(!m_pDibModel->IsEmpty()){
  646. m_pDibModel->Empty();
  647. m_pDibModel = NULL;
  648. }
  649. if(!m_pDibResult->IsEmpty()){
  650. m_pDibResult->Empty();
  651. m_pDibResult = NULL;
  652. }
  654. CDialog::OnCancel();
  655. }