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

源码开发语言/平台
当前位置：首页 > 源码/资料 > 图形图像 > 图形图像处理 > 查看源码
DlgMosaics.cpp
资源名称：FINAL.rar　[点击查看]
上传用户：piaozanzhu
上传日期：2008-02-14
资源大小：212k
文件大小：16k
源码类别：

图形图像处理

开发平台：

Visual C++

DlgMosaics.cpp：源码内容
  1. // DlgMosaics.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "ImageProcessing.h"
  6. #include "DlgMosaics.h"
  7. #include "ImageProcessingDoc.h"
  8. #include "cdib.h"
  9. #include "CDib.h"
  11. #include "DlgMosaicRes.h"
  13. #include "GlobalApi.h"
  20. #ifdef _DEBUG
  21. #define new DEBUG_NEW
  22. #undef THIS_FILE
  23. static char THIS_FILE[] = __FILE__;
  24. #endif
  26. /////////////////////////////////////////////////////////////////////////////
  27. // CDlgMosaics dialog
  31. CDlgMosaics::CDlgMosaics(CWnd* pParent /*=NULL*/,CImageProcessingDoc* pDoc)
  32. : CDialog(CDlgMosaics::IDD, pParent)
  33. {
  34. //{{AFX_DATA_INIT(CDlgMosaics)
  35. // NOTE: the ClassWizard will add member initialization here
  37. //}}AFX_DATA_INIT
  40. m_pDoc = pDoc;
  42. // 设置计算图象位置标志位位FALSE
  43. m_bCalImgLoc = FALSE;
  45. // 设置基准图象为原始打开的图象
  46. m_pDibInit = pDoc->m_pDibInit;
  48. // 设置待配准图象
  49. m_pDibSamp = new CDib;
  54. left_starX=0;
  55. left_starY=0;
  56. left_endX=0;
  57. left_endY=0;
  58. right_starX=0;
  59. right_starY=0;
  60. right_endX=0;
  61. right_endY=0;
  62. }
  65. void CDlgMosaics::DoDataExchange(CDataExchange* pDX)
  66. {
  67. CDialog::DoDataExchange(pDX);
  68. //{{AFX_DATA_MAP(CDlgMosaics)
  69. // NOTE: the ClassWizard will add DDX and DDV calls here
  70. //}}AFX_DATA_MAP
  71. }
  74. BEGIN_MESSAGE_MAP(CDlgMosaics, CDialog)
  75. //{{AFX_MSG_MAP(CDlgMosaics)
  76. ON_WM_PAINT()
  77. ON_BN_CLICKED(IDC_OPEN, OnOpen)
  78. ON_BN_CLICKED(IDC_MOSAIC, OnMosaic)
  79. //}}AFX_MSG_MAP
  80. END_MESSAGE_MAP()
  82. /////////////////////////////////////////////////////////////////////////////
  83. // CDlgMosaics message handlers
  85. void CDlgMosaics::OnPaint() 
  86. {
  87. CPaintDC dc(this); // device context for painting
  89. // TODO: Add your message handler code here
  90. // 如果还没有计算图象的位置，则进行计算
  91. if(!m_bCalImgLoc){
  92. CalImageLocation();
  93. }
  95. CSize sizeDisplay;
  96. CPoint pointDisplay;
  98. // 显示基准图象
  99. if(!m_pDibInit->IsEmpty()){
  100. sizeDisplay.cx=m_pDibInit->m_lpBMIH->biWidth;
  101. sizeDisplay.cy=m_pDibInit->m_lpBMIH->biHeight;
  102. pointDisplay.x = m_rectInitImage.left;//基准图象
  103. pointDisplay.y = m_rectInitImage.top;
  104. m_pDibInit->Draw(&dc,pointDisplay,sizeDisplay);
  105. }
  107. // 显示待配准图象
  108. if(!m_pDibSamp->IsEmpty()){
  109. sizeDisplay.cx=m_pDibSamp->m_lpBMIH->biWidth;
  110. sizeDisplay.cy=m_pDibSamp->m_lpBMIH->biHeight;
  111. pointDisplay.x = m_rectResltImage.left;
  112. pointDisplay.y = m_rectResltImage.top;
  113. m_pDibSamp->Draw(&dc,pointDisplay,sizeDisplay);
  114. }
  116. // Do not call CDialog::OnPaint() for painting messages
  117. }
  119. void CDlgMosaics::CalImageLocation()
  120. {
  121. CWnd* pWnd=GetDlgItem(IDC_BMP1);//picture 控件的ID
  122. WINDOWPLACEMENT *winPlacement;
  123. winPlacement=new WINDOWPLACEMENT;
  124. pWnd->GetWindowPlacement(winPlacement);
  126. // 图象宽度
  127. int nImageWidth;
  128. nImageWidth = m_pDibInit->m_lpBMIH->biWidth;
  130. // 图象高度
  131. int nImageHeight;
  132. nImageHeight = m_pDibInit->m_lpBMIH->biHeight;
  135. // 调整控件IDC_BMP1的大小位置，并同时设置显示基准图象的位置
  136. if(nImageHeight > 352){
  137. winPlacement->rcNormalPosition.bottom = winPlacement->rcNormalPosition.top + nImageHeight;
  138. m_rectInitImage.bottom = winPlacement->rcNormalPosition.bottom;
  139. m_rectInitImage.top    = winPlacement->rcNormalPosition.top;
  140. }
  141. else{
  142. winPlacement->rcNormalPosition.bottom = winPlacement->rcNormalPosition.top + 352;
  143. m_rectInitImage.bottom = winPlacement->rcNormalPosition.top + 176 + nImageHeight/2;
  144. m_rectInitImage.top    = winPlacement->rcNormalPosition.top + 176 - nImageHeight/2;
  145. }
  146. if(nImageWidth > 288){
  147. winPlacement->rcNormalPosition.right = winPlacement->rcNormalPosition.left + nImageWidth;
  148. m_rectInitImage.right = winPlacement->rcNormalPosition.right;
  149. m_rectInitImage.left  = winPlacement->rcNormalPosition.left;
  150. }
  151. else{
  152. winPlacement->rcNormalPosition.right = winPlacement->rcNormalPosition.left + 288;
  153. m_rectInitImage.right = winPlacement->rcNormalPosition.left + 144 + nImageWidth/2;
  154. m_rectInitImage.left  = winPlacement->rcNormalPosition.left + 144 - nImageWidth/2;
  155. }
  157. // 设置IDC_BMP1控件的大小位置
  158. pWnd->SetWindowPlacement(winPlacement);
  160. // 获得显示基准图象控件的右边位置，以便确认显示待配准图象控件的位置
  161. int nIniImgRight;
  162. nIniImgRight = winPlacement->rcNormalPosition.right;
  164. int  nIniImgLeft;
  165. nIniImgLeft   = winPlacement->rcNormalPosition.left;
  167. // 获得IDC_BMP1控件的下边位置，以便调整其他控件的位置
  168. int nIniImgBottom;
  169. nIniImgBottom = winPlacement->rcNormalPosition.bottom;
  171. // 获得控件IDC_BMP2的句柄，并获得初始位置信息
  172. pWnd=GetDlgItem(IDC_BMP2);
  173. pWnd->GetWindowPlacement(winPlacement);
  175. // 如果还未打开待配准图象，则设置待配准图象大小和基准图象大小相等
  176. if(!m_pDibSamp->IsEmpty()){
  177. nImageWidth  = m_pDibSamp->m_lpBMIH->biWidth;
  178. nImageHeight = m_pDibSamp->m_lpBMIH->biHeight;
  179. }
  181. // 调整控件IDC_BMP2的大小位置，并同时设置显示待配准图象的位置
  183. // 先调整控件的左边位置，和IDC_BMP1控件相隔15个象素
  184. winPlacement->rcNormalPosition.left = nIniImgRight + 15;
  186. if(nImageHeight > 352){
  187. winPlacement->rcNormalPosition.bottom = winPlacement->rcNormalPosition.top + nImageHeight;
  188. m_rectResltImage.bottom = winPlacement->rcNormalPosition.bottom;
  189. m_rectResltImage.top    = winPlacement->rcNormalPosition.top;
  190. }
  191. else{
  192. winPlacement->rcNormalPosition.bottom = winPlacement->rcNormalPosition.top + 352;
  193. m_rectResltImage.bottom = winPlacement->rcNormalPosition.top + 176 + nImageHeight/2;
  194. m_rectResltImage.top    = winPlacement->rcNormalPosition.top + 176 - nImageHeight/2;
  195. }
  196. if(nImageWidth > 288){
  197. winPlacement->rcNormalPosition.right = winPlacement->rcNormalPosition.left + nImageWidth;
  198. m_rectResltImage.right = winPlacement->rcNormalPosition.right;
  199. m_rectResltImage.left  = winPlacement->rcNormalPosition.left;
  200. }
  201. else{
  202. winPlacement->rcNormalPosition.right = winPlacement->rcNormalPosition.left + 288;
  203. m_rectResltImage.right = winPlacement->rcNormalPosition.left + 144 + nImageWidth/2;
  204. m_rectResltImage.left  = winPlacement->rcNormalPosition.left + 144 - nImageWidth/2;
  205. }
  207. // 设置IDC_REG_BMP2控件的大小位置
  208. pWnd->SetWindowPlacement(winPlacement);
  210. if(nIniImgBottom < winPlacement->rcNormalPosition.bottom)
  211. nIniImgBottom = winPlacement->rcNormalPosition.bottom;
  214. nIniImgBottom = winPlacement->rcNormalPosition.bottom;
  216. nIniImgRight  = winPlacement->rcNormalPosition.right;
  218. // 设置控件IDOK的位置大小
  219. pWnd=GetDlgItem(IDOK);
  220. pWnd->GetWindowPlacement(winPlacement);
  221. winPlacement->rcNormalPosition.top = nIniImgBottom +15;
  222. winPlacement->rcNormalPosition.bottom = nIniImgBottom + 60;
  223. pWnd->SetWindowPlacement(winPlacement);
  225. // 设置控件IDCANCEL的位置大小
  226. pWnd=GetDlgItem(IDCANCEL);
  227. pWnd->GetWindowPlacement(winPlacement);
  228. winPlacement->rcNormalPosition.top = nIniImgBottom +15;
  229. winPlacement->rcNormalPosition.bottom = nIniImgBottom + 60;
  230. pWnd->SetWindowPlacement(winPlacement);
  232. // 设置控件IDC_OPEN的位置大小
  233. pWnd=GetDlgItem(IDC_OPEN);
  234. pWnd->GetWindowPlacement(winPlacement);
  235. winPlacement->rcNormalPosition.top = nIniImgBottom +15;
  236. winPlacement->rcNormalPosition.bottom = nIniImgBottom + 60;
  237. pWnd->SetWindowPlacement(winPlacement);
  239. // 设置控件IDC_MOSAIC的位置大小
  240. pWnd=GetDlgItem(IDC_MOSAIC);
  241. pWnd->GetWindowPlacement(winPlacement);
  242. winPlacement->rcNormalPosition.top = nIniImgBottom +15;
  243. winPlacement->rcNormalPosition.bottom = nIniImgBottom + 60;
  244. pWnd->SetWindowPlacement(winPlacement);
  246. /*
  247. // 设置控件IDC_REG_CHOSE_FEATUR的位置大小
  248. pWnd=GetDlgItem(IDC_REG_CHOSE_FEATURE);
  249. pWnd->GetWindowPlacement(winPlacement);
  250. winPlacement->rcNormalPosition.top = nIniImgBottom +15;
  251. winPlacement->rcNormalPosition.bottom = nIniImgBottom + 60;
  252. pWnd->SetWindowPlacement(winPlacement);
  253. */
  254. // 调整此对话框的大小
  255. //pWnd = GetDlgItem(IDD_DLG_REG);
  256. this->GetWindowPlacement(winPlacement);
  257. //winPlacement->rcNormalPosition.top = nIniImgBottom +15;
  258. winPlacement->rcNormalPosition.bottom = nIniImgBottom + 300;
  259. winPlacement->rcNormalPosition.left   = nIniImgLeft   - 20;
  260. winPlacement->rcNormalPosition.right  = nIniImgRight  + 20;
  261. this->SetWindowPlacement(winPlacement);
  263. // 释放已分配内存
  264. delete winPlacement;
  266. // 设置计算图象控件位置标志位为TRUE
  267. m_bCalImgLoc = TRUE;
  269. }
  271. void CDlgMosaics::OnOpen() 
  272. {
  273. // TODO: Add your control notification handler code here
  275. CFileDialog dlg(TRUE,"bmp","*.bmp");
  276. if(dlg.DoModal() == IDOK)
  277. {
  278.  
  279.   CFile file;
  280.  
  281.   CString strPathName;
  282.  
  283. strPathName = dlg.GetPathName();
  284.  
  285. // 打开文件
  286. if( !file.Open(strPathName, CFile::modeRead | CFile::shareDenyWrite))
  287. {
  288. // 返回
  289. return ;
  290. }
  292. // 读入模板图象
  293. if(!m_pDibSamp->Read(&file)){
  294. // 恢复光标形状
  295. EndWaitCursor();
  297. // 清空已分配内存
  298. m_pDibSamp->Empty();
  300. // 返回
  301.   return;
  302. }
  303. }
  305. // 判断是否是8-bpp位图（这里为了方便，只处理8-bpp位图的模板配准，其它的可以类推）
  306. if(m_pDibSamp->m_nColorTableEntries != 256)
  307. {
  308. // 提示用户
  309. MessageBox("目前只支持256色位图！", "系统提示" , MB_ICONINFORMATION | MB_OK);
  311. // 清空已分配内存
  312. m_pDibSamp->Empty();
  314. // 返回
  315. return;
  316. }
  318. // 如果打开新的待配准文件，将图象位置设置标志位设为FALSE，以便再次调整位置
  319. m_bCalImgLoc = FALSE;
  321. // 更新显示
  322. this->UpdateData();
  323. this->Invalidate();
  325. }
  329. void CDlgMosaics::OnMosaic() 
  330. {
  331. // TODO: Add your control notification handler code here
  332. SetComparDIB(m_pDibInit, m_pDibSamp);
  335. }
  336. /***********************************************************************************
  337.                     用于寻找两幅图象相同的部分
  339. ***********************************************************************************/
  340. BOOL CDlgMosaics::SetComparDIB(CDib *bmp1, CDib *bmp2)
  341. {
  342.    
  343. //最大面积
  344.     MAXarea=0;
  345.     //从第二幅图象的最左上角开始，从第一幅图象的最右上角开始，比点比列比块
  346. for(int rightbmp_h=0;rightbmp_h<bmp2->m_lpBMIH->biHeight/3;rightbmp_h++)
  347.     for(int leftbmp_w=bmp1->m_lpBMIH->biWidth;leftbmp_w>=bmp1->m_lpBMIH->biWidth/2;leftbmp_w--)
  348.     for(unsigned int leftbmp_h=0;leftbmp_h<bmp1->m_lpBMIH->biHeight;leftbmp_h++)
  349. {
  350. bmp2->Pix_X=0;                   
  351. bmp2->Pix_Y=rightbmp_h;
  352. bmp1->Pix_X=leftbmp_w;
  353. bmp1->Pix_Y=leftbmp_h;
  354. if(IsSamePix(bmp1,bmp2,bmp1->Pix_X,bmp1->Pix_Y,bmp2->Pix_X,bmp2->Pix_Y))
  355. if(IsSameCol(bmp1,bmp2,bmp1->Pix_X,bmp1->Pix_Y,bmp2->Pix_X,bmp2->Pix_Y))
  356. if(IsSameRect(bmp1,bmp2))
  357. break;
  358. else 
  359. continue;
  360. }
  362. //如果相同块的面积不为零则拼接
  363.      if(MAXarea!=0)
  364.  {
  365.  MergeDIB(bmp1,bmp2);
  366.  return TRUE;
  367.  }
  368.  else
  369.  {
  370.  MessageBox("没有找到合并点，不给予合并");
  371.  return FALSE;
  372.  }
  373. }
  374. /*******************************************************************************
  375. 用于计算点在图象中的位置
  377. *******************************************************************************/
  379. long CDlgMosaics::GetPixOffset(unsigned int X, unsigned int Y, CDib *bmp)
  380. {
  382.    //计算点在图象中的位置
  383. if((X>=0&&X<bmp->m_lpBMIH->biWidth)&&(Y>=0&&Y<bmp->m_lpBMIH->biHeight))
  384. return (Y*bmp->GetDibSaveDim().cx+X*(bmp->m_lpBMIH->biBitCount/8));//要在CDib中定义bmp_line_w和bmp_biBitCount
  385. return 0;
  386. }
  388. /********************************************************************************
  389. 用于判断是否是相同象素的点
  391. ********************************************************************************/
  394. BOOL CDlgMosaics::IsSamePix(CDib *bmp1, CDib *bmp2, int x1, int y1, int x2, int y2)
  395. {
  398. BYTE *p,*q;
  399. p=bmp1->m_lpImage+GetPixOffset(x1,y1,bmp1);
  400. q=bmp2->m_lpImage+GetPixOffset(x2,y2,bmp2);
  401. for(int i=0;i<bmp1->m_lpBMIH->biBitCount/8;i++)
  402. if(abs(p[i]-q[i])>50)
  404. return FALSE;
  405. else 
  406. return TRUE;
  407. }
  409. /************************************************************************************
  410. 用于判断是否是相同列
  412. ************************************************************************************/
  414. BOOL CDlgMosaics::IsSameCol(CDib *bmp1, CDib *bmp2, int x1, int y1, int x2, int y2)
  415. {
  417.     //计算当前象素点到图象1底部的高度
  418. unsigned int below_h1=bmp1->m_lpBMIH->biHeight-y1;
  419. //计算图象2当前象素点到底部的高度
  420. unsigned int below_h2=bmp2->m_lpBMIH->biHeight-y2;
  421. unsigned int h=(below_h1>below_h2)?below_h2:below_h1;
  423. for(unsigned int i=0;i<h;i++)
  424. {
  425. //判断是否是相同的象素点，如果是，则在同一列中往下比较
  426. if(!IsSamePix(bmp1,bmp2,x1,y1,x2,y2))
  427. return FALSE ;
  428. y1++;
  429. y2++;
  430. }
  431. return TRUE;
  432. }
  434. /*************************************************************************************
  435. 用语判断是否是相同的块
  436. *************************************************************************************/
  438. BOOL CDlgMosaics::IsSameRect(CDib *bmp1, CDib *bmp2)
  439. {
  442. unsigned int x1,y1,x2,y2;
  443. unsigned int w1=bmp1->m_lpBMIH->biWidth-bmp1->Pix_X;
  444. unsigned int w2=bmp2->m_lpBMIH->biWidth-bmp2->Pix_X;
  445. unsigned int w=(w1>w2)?w2:w1;
  447. x1=bmp1->Pix_X;y1=bmp1->Pix_Y;
  448. x2=bmp2->Pix_X;y2=bmp2->Pix_Y;
  449. for(unsigned int i=0;i<w;i++)
  450. {     
  451.  //判断是否是相同的列，如果是则向左移动继续比较是否是相同的块
  452. if(!IsSameCol(bmp1,bmp2,x1,y1,x2,y2))
  453. return FALSE;
  454. x1++;
  455. x2++;
  456. }
  458. unsigned int h=(bmp1->m_lpBMIH->biHeight-bmp1->Pix_Y)<(bmp2->m_lpBMIH->biHeight-bmp2->Pix_Y)?
  459. (bmp1->m_lpBMIH->biHeight-bmp1->Pix_Y):(bmp2->m_lpBMIH->biHeight-bmp2->Pix_Y);
  460.     //如果块的面积比先前的最大面积大，则取代之
  461. if(i*h>MAXarea)
  462. {
  463. MAXarea=i*h;
  465. left_starX=bmp1->Pix_X;
  466.     left_starY=bmp1->Pix_Y;
  467. left_endX=i+bmp1->Pix_X;
  468. left_endY=h+bmp1->Pix_Y;
  469. right_starX=bmp2->Pix_X;
  470. right_starY=bmp2->Pix_Y;
  471. right_endX=i+bmp2->Pix_X;
  472. right_endY=h+bmp2->Pix_Y;
  473. }
  474. return TRUE;
  475. }
  476. /*************************************************************************************
  477. 用于图象合并拼接
  479. *************************************************************************************/
  481. void CDlgMosaics::MergeDIB(CDib *bmp1, CDib *bmp2)
  482. {
  483.     
  484. BYTE* unitbuff;
  485. unitbuff=NULL;
  488. unsigned int w1=bmp1->m_lpBMIH->biHeight;
  489. unsigned int w2=bmp2->m_lpBMIH->biHeight;
  492. //新图象的宽
  493. unsigned int unit_w=bmp2->m_lpBMIH->biWidth+left_starX;
  494. //重叠部分的上部高度
  495. unsigned int unit_up=(left_starY>right_starY)?left_starY:right_starY;
  496. //重叠部分的下部高度
  497. unsigned int unit_down=((w1-left_endY)>(w2-right_endY))?(w1-left_endY):(w2-right_endY);
  498. //新图象的高
  499. unsigned int unit_h=unit_up+(left_endY-left_starY)/*重叠部分的高*/+unit_down;
  502. //得到每行实际占的字节数
  503. int byte_w=unit_w*(bmp1->m_lpBMIH->biBitCount/8);
  504. if(byte_w%4)
  505. byte_w=byte_w+(4-byte_w%4);
  507. //得到新图象的数据大小
  508. int imagedata=byte_w*unit_h;
  511. // 将此图象用CDib类封装
  512. m_pDibResult = new CDib(CSize(unit_w,unit_h), 8);
  514. // 计算结果图象的存储大小尺寸
  515. CSize sizeSaveResult;
  516. sizeSaveResult = m_pDibResult->GetDibSaveDim();
  518. // 拷贝调色板
  519. memcpy(m_pDibResult->m_lpvColorTable, m_pDibInit->m_lpvColorTable, m_pDibResult->m_nColorTableEntries*sizeof(RGBQUAD));
  521. // 应用调色板
  522. m_pDibResult->MakePalette();
  525. // 分配内存给合并后的图象
  526. LPBYTE lpImgResult;
  527. lpImgResult = (LPBYTE)new unsigned char[sizeSaveResult.cx * sizeSaveResult.cy];
  529. BYTE *p1,*p2,*p3;
  530. p1=bmp1->m_lpImage+left_starY*bmp1->GetDibSaveDim().cx;  //数据1起始地址
  531. p2=bmp2->m_lpImage+right_starY*bmp2->GetDibSaveDim().cx; //数据2起始地址
  532. p3=lpImgResult;
  534. //保存图象的数据，先把第一幅图象的第一行存进去，再存第二幅图象的第一行，然后再存
  535. //第一幅图象的第二行，再存第二幅图象的第二行，以次类推
  536. for(unsigned int i=0;i<unit_h;i++)
  537. {
  539. memcpy(p3,p1,left_starX*(bmp1->m_lpBMIH->biBitCount/8));
  540. p3+=left_starX*(bmp1->m_lpBMIH->biBitCount/8);
  541. p1+=bmp1->GetDibSaveDim().cx;
  542. memcpy(p3,p2,bmp2->m_lpBMIH->biWidth*(bmp2->m_lpBMIH->biBitCount/8));
  543. p3+=bmp2->m_lpBMIH->biWidth*(bmp2->m_lpBMIH->biBitCount/8);
  544. p2+=bmp2->GetDibSaveDim().cx;
  545.         //如果不是4的倍数则填满为4的倍数
  546. if((unsigned int)(p3-unitbuff)%4!=0)
  547. p3+=4-(unsigned int)((p3-unitbuff)%4);
  548. }
  549. // 将指针赋值给CDib类的数据
  550. m_pDibResult->m_lpImage = lpImgResult;
  552.     //创建显示拼接后图象的对话框
  553. CDlgMosaicRes* pDlg;
  554. pDlg = new CDlgMosaicRes(NULL,  m_pDibResult);
  555. pDlg->DoModal();
  557. // 删除对象
  558. delete pDlg;
  561. return ;
  564. }