DlgReg.cpp
上传用户:panpan8800
上传日期:2013-06-29
资源大小:274k
文件大小:36k
源码类别:

图形图像处理

开发平台:

Visual C++

  1. // DlgReg.cpp : implementation file
  2. //
  3. #include "stdafx.h"
  4. #include "ImageProcessing.h"
  5. #include "ImageProcessingDoc.h"
  6. #include "GlobalApi.h"
  7. #include "DlgReg.h"
  8. #include "DlgAftReg.h"
  9. #ifdef _DEBUG
  10. #define new DEBUG_NEW
  11. #undef THIS_FILE
  12. static char THIS_FILE[] = __FILE__;
  13. #endif
  14. /////////////////////////////////////////////////////////////////////////////
  15. // CDlgReg dialog
  16. CDlgReg::CDlgReg(CWnd* pParent /*=NULL*/, CImageProcessingDoc* pDoc)
  17. : CDialog(CDlgReg::IDD, pParent)
  18. {
  19. //{{AFX_DATA_INIT(CDlgReg)
  20. // NOTE: the ClassWizard will add member initialization here
  21. //}}AFX_DATA_INIT
  22. // 获取文档类指针
  23. m_pDoc = pDoc;
  24. // 设置计算图象位置标志位位FALSE
  25. m_bCalImgLoc = FALSE;
  26. // 设置基准图象为原始打开的图象
  27. m_pDibInit = pDoc->m_pDibInit;
  28. // 设置待配准图象
  29. m_pDibSamp = new CDib;
  30. // 设置选取特征点的数目初始值
  31. m_nChsFeatureNum = 0;
  32. // 设置选取特征点的标志位为FALSE
  33. m_bChoseFeature = FALSE;
  34. }
  35. void CDlgReg::DoDataExchange(CDataExchange* pDX)
  36. {
  37. CDialog::DoDataExchange(pDX);
  38. //{{AFX_DATA_MAP(CDlgReg)
  39. // NOTE: the ClassWizard will add DDX and DDV calls here
  40. //}}AFX_DATA_MAP
  41. }
  42. BEGIN_MESSAGE_MAP(CDlgReg, CDialog)
  43. //{{AFX_MSG_MAP(CDlgReg)
  44. ON_WM_PAINT()
  45. ON_BN_CLICKED(IDC_REG_OPEN, OnRegOpen)
  46. ON_BN_CLICKED(IDC_REG_REG, OnRegReg)
  47. ON_BN_CLICKED(IDC_REG_CHOSE_FEATURE, OnRegChoseFeature)
  48. ON_WM_LBUTTONUP()
  49. ON_WM_MOUSEMOVE()
  50. //}}AFX_MSG_MAP
  51. END_MESSAGE_MAP()
  52. /////////////////////////////////////////////////////////////////////////////
  53. // CDlgReg message handlers
  54. void CDlgReg::OnPaint() 
  55. {
  56. CPaintDC dc(this); // device context for painting
  57. // 如果还没有计算图象的位置,则进行计算
  58. if(!m_bCalImgLoc){
  59. CalImageLocation();
  60. }
  61. CSize sizeDisplay;
  62. CPoint pointDisplay;
  63. // 显示基准图象
  64. if(!m_pDibInit->IsEmpty()){
  65. sizeDisplay.cx=m_pDibInit->m_lpBMIH->biWidth;
  66. sizeDisplay.cy=m_pDibInit->m_lpBMIH->biHeight;
  67. pointDisplay.x = m_rectInitImage.left;
  68. pointDisplay.y = m_rectInitImage.top;
  69. m_pDibInit->Draw(&dc,pointDisplay,sizeDisplay);
  70. }
  71. // 显示待配准图象
  72. if(!m_pDibSamp->IsEmpty()){
  73. sizeDisplay.cx=m_pDibSamp->m_lpBMIH->biWidth;
  74. sizeDisplay.cy=m_pDibSamp->m_lpBMIH->biHeight;
  75. pointDisplay.x = m_rectResltImage.left;
  76. pointDisplay.y = m_rectResltImage.top;
  77. m_pDibSamp->Draw(&dc,pointDisplay,sizeDisplay);
  78. }
  79. // 显示特征点与配准的特征点
  80. DrawFeature(&dc);
  81. }
  82. /*************************************************************************
  83.  *
  84.  * 函数名称:
  85.  *   CalImageLocation()
  86.  *
  87.  * 输入参数:
  88.  *   无
  89.  *
  90.  * 返回值:
  91.  *   无
  92.  *
  93.  * 说明:
  94.  *   该函数设置对话框中的控件位置和大小,并设置显示图象的位置。默认的图象大小为352×288,如果图象小于
  95.  *此大小,则控件大小设置为352×288,并将图象放置在控件中间。
  96.  *
  97.  *************************************************************************
  98.  */
  99. void CDlgReg::CalImageLocation()
  100. {
  101. // 获得控件IDC_REG_INIT_IMAGE的句柄,并获得控件的初始位置信息
  102. CWnd* pWnd=GetDlgItem(IDC_REG_INIT_IMAGE);
  103. WINDOWPLACEMENT *winPlacement;
  104. winPlacement=new WINDOWPLACEMENT;
  105. pWnd->GetWindowPlacement(winPlacement);
  106. // 图象宽度
  107. int nImageWidth;
  108. nImageWidth = m_pDibInit->m_lpBMIH->biWidth;
  109. // 图象高度
  110. int nImageHeight;
  111. nImageHeight = m_pDibInit->m_lpBMIH->biHeight;
  112. // 调整控件IDC_REG_INIT_IMAGE的大小位置,并同时设置显示基准图象的位置
  113. if(nImageHeight > 352){
  114. winPlacement->rcNormalPosition.bottom = winPlacement->rcNormalPosition.top + nImageHeight;
  115. m_rectInitImage.bottom = winPlacement->rcNormalPosition.bottom;
  116. m_rectInitImage.top    = winPlacement->rcNormalPosition.top;
  117. }
  118. else{
  119. winPlacement->rcNormalPosition.bottom = winPlacement->rcNormalPosition.top + 352;
  120. m_rectInitImage.bottom = winPlacement->rcNormalPosition.top + 176 + nImageHeight/2;
  121. m_rectInitImage.top    = winPlacement->rcNormalPosition.top + 176 - nImageHeight/2;
  122. }
  123. if(nImageWidth > 288){
  124. winPlacement->rcNormalPosition.right = winPlacement->rcNormalPosition.left + nImageWidth;
  125. m_rectInitImage.right = winPlacement->rcNormalPosition.right;
  126. m_rectInitImage.left  = winPlacement->rcNormalPosition.left;
  127. }
  128. else{
  129. winPlacement->rcNormalPosition.right = winPlacement->rcNormalPosition.left + 288;
  130. m_rectInitImage.right = winPlacement->rcNormalPosition.left + 144 + nImageWidth/2;
  131. m_rectInitImage.left  = winPlacement->rcNormalPosition.left + 144 - nImageWidth/2;
  132. }
  133. // 设置IDC_REG_INIT_IMAGE控件的大小位置
  134. pWnd->SetWindowPlacement(winPlacement);
  135. // 获得显示基准图象控件的右边位置,以便确认显示待配准图象控件的位置
  136. int nIniImgRight;
  137. nIniImgRight = winPlacement->rcNormalPosition.right;
  138. int  nIniImgLeft;
  139. nIniImgLeft   = winPlacement->rcNormalPosition.left;
  140. // 获得IDC_REG_INIT_IMAGE控件的下边位置,以便调整其他控件的位置
  141. int nIniImgBottom;
  142. nIniImgBottom = winPlacement->rcNormalPosition.bottom;
  143. // 获得控件IDC_REG_RESLT_IMAGE的句柄,并获得初始位置信息
  144. pWnd=GetDlgItem(IDC_REG_RESLT_IMAGE);
  145. pWnd->GetWindowPlacement(winPlacement);
  146. // 如果还未打开待配准图象,则设置待配准图象大小和基准图象大小相等
  147. if(!m_pDibSamp->IsEmpty()){
  148. nImageWidth  = m_pDibSamp->m_lpBMIH->biWidth;
  149. nImageHeight = m_pDibSamp->m_lpBMIH->biHeight;
  150. }
  151. // 调整控件IDC_REG_RESLT_IMAGE的大小位置,并同时设置显示待配准图象的位置
  152. // 先调整控件的左边位置,和IDC_REG_INIT_IMAGE控件相隔15个象素
  153. winPlacement->rcNormalPosition.left = nIniImgRight + 15;
  154. if(nImageHeight > 352){
  155. winPlacement->rcNormalPosition.bottom = winPlacement->rcNormalPosition.top + nImageHeight;
  156. m_rectResltImage.bottom = winPlacement->rcNormalPosition.bottom;
  157. m_rectResltImage.top    = winPlacement->rcNormalPosition.top;
  158. }
  159. else{
  160. winPlacement->rcNormalPosition.bottom = winPlacement->rcNormalPosition.top + 352;
  161. m_rectResltImage.bottom = winPlacement->rcNormalPosition.top + 176 + nImageHeight/2;
  162. m_rectResltImage.top    = winPlacement->rcNormalPosition.top + 176 - nImageHeight/2;
  163. }
  164. if(nImageWidth > 288){
  165. winPlacement->rcNormalPosition.right = winPlacement->rcNormalPosition.left + nImageWidth;
  166. m_rectResltImage.right = winPlacement->rcNormalPosition.right;
  167. m_rectResltImage.left  = winPlacement->rcNormalPosition.left;
  168. }
  169. else{
  170. winPlacement->rcNormalPosition.right = winPlacement->rcNormalPosition.left + 288;
  171. m_rectResltImage.right = winPlacement->rcNormalPosition.left + 144 + nImageWidth/2;
  172. m_rectResltImage.left  = winPlacement->rcNormalPosition.left + 144 - nImageWidth/2;
  173. }
  174. // 设置IDC_REG_RESLT_IMAGE控件的大小位置
  175. pWnd->SetWindowPlacement(winPlacement);
  176. if(nIniImgBottom < winPlacement->rcNormalPosition.bottom)
  177. nIniImgBottom = winPlacement->rcNormalPosition.bottom;
  178. nIniImgBottom = winPlacement->rcNormalPosition.bottom;
  179. nIniImgRight  = winPlacement->rcNormalPosition.right;
  180. // 设置控件IDOK的位置大小
  181. pWnd=GetDlgItem(IDOK);
  182. pWnd->GetWindowPlacement(winPlacement);
  183. winPlacement->rcNormalPosition.top = nIniImgBottom +15;
  184. winPlacement->rcNormalPosition.bottom = nIniImgBottom + 60;
  185. pWnd->SetWindowPlacement(winPlacement);
  186. // 设置控件IDCANCEL的位置大小
  187. pWnd=GetDlgItem(IDCANCEL);
  188. pWnd->GetWindowPlacement(winPlacement);
  189. winPlacement->rcNormalPosition.top = nIniImgBottom +15;
  190. winPlacement->rcNormalPosition.bottom = nIniImgBottom + 60;
  191. pWnd->SetWindowPlacement(winPlacement);
  192. // 设置控件IDC_REG_OPEN的位置大小
  193. pWnd=GetDlgItem(IDC_REG_OPEN);
  194. pWnd->GetWindowPlacement(winPlacement);
  195. winPlacement->rcNormalPosition.top = nIniImgBottom +15;
  196. winPlacement->rcNormalPosition.bottom = nIniImgBottom + 60;
  197. pWnd->SetWindowPlacement(winPlacement);
  198. // 设置控件IDC_REG_REG的位置大小
  199. pWnd=GetDlgItem(IDC_REG_REG);
  200. pWnd->GetWindowPlacement(winPlacement);
  201. winPlacement->rcNormalPosition.top = nIniImgBottom +15;
  202. winPlacement->rcNormalPosition.bottom = nIniImgBottom + 60;
  203. pWnd->SetWindowPlacement(winPlacement);
  204. // 设置控件IDC_REG_CHOSE_FEATUR的位置大小
  205. pWnd=GetDlgItem(IDC_REG_CHOSE_FEATURE);
  206. pWnd->GetWindowPlacement(winPlacement);
  207. winPlacement->rcNormalPosition.top = nIniImgBottom +15;
  208. winPlacement->rcNormalPosition.bottom = nIniImgBottom + 60;
  209. pWnd->SetWindowPlacement(winPlacement);
  210. // 调整此对话框的大小
  211. //pWnd = GetDlgItem(IDD_DLG_REG);
  212. this->GetWindowPlacement(winPlacement);
  213. //winPlacement->rcNormalPosition.top = nIniImgBottom +15;
  214. winPlacement->rcNormalPosition.bottom = nIniImgBottom + 300;
  215. winPlacement->rcNormalPosition.left   = nIniImgLeft   - 20;
  216. winPlacement->rcNormalPosition.right  = nIniImgRight  + 20;
  217. this->SetWindowPlacement(winPlacement);
  218. // 释放已分配内存
  219. delete winPlacement;
  220. // 设置计算图象控件位置标志位为TRUE
  221. m_bCalImgLoc = TRUE;
  222. }
  223. /*************************************************************************
  224.  *
  225.  * 函数名称:
  226.  *   DrawFeature()
  227.  *
  228.  * 输入参数:
  229.  *   无
  230.  *
  231.  * 返回值:
  232.  *   无
  233.  *
  234.  * 说明:
  235.  *   该函数根据类的成员变量确定特征点的数目和位置,并在图象中进行显示。
  236.  *
  237.  *************************************************************************
  238.  */
  239. void CDlgReg::DrawFeature(CDC* pDC)
  240. {
  241. // 循环变量
  242. int i;
  243. // 临时变量
  244. CPoint pointTemp;
  245. // 半径
  246. int nRadius;
  247. nRadius = 5;
  248. // 设置画图类型
  249. pDC->SelectStockObject(HOLLOW_BRUSH);
  250. // 声明画笔
  251. CPen penWhite(PS_SOLID,1,RGB(255,255,255));
  252. CPen *pOldPen;
  253. // 将画笔选入,并保存以前的画笔
  254. pOldPen = pDC->SelectObject(&penWhite);
  255. for(i=0; i<m_nChsFeatureNum; i++){
  256. // 首先显示特征点
  257. // 确定此点的显示位置
  258. pointTemp.x = m_pPointSampl[i].x + m_rectResltImage.left;
  259. pointTemp.y = m_pPointSampl[i].y + m_rectResltImage.top ;
  260. // 画出此特征点,其中园的半径为nRadius
  261. CRect rectSamp(pointTemp.x-nRadius , pointTemp.y-nRadius , 
  262. pointTemp.x+nRadius , pointTemp.y+nRadius);
  263. pDC->Ellipse(rectSamp);
  264. // 再显示配准特征点
  265. // 确定此点的显示位置
  266. pointTemp.x = m_pPointBase[i].x + m_rectInitImage.left;
  267. pointTemp.y = m_pPointBase[i].y + m_rectInitImage.top ;
  268. // 画出此特征点,其中园的半径为nRadius
  269. CRect rectBase(pointTemp.x-nRadius , pointTemp.y-nRadius , 
  270. pointTemp.x+nRadius , pointTemp.y+nRadius);
  271. pDC->Ellipse(rectBase);
  272. }
  273. // 回复以前的画笔
  274. pDC->SelectObject(pOldPen);
  275. penWhite.DeleteObject();
  276. }
  277. /*************************************************************************
  278.  *
  279.  * 函数名称:
  280.  *   OnRegOpen()
  281.  *
  282.  * 输入参数:
  283.  *   无
  284.  *
  285.  * 返回值:
  286.  *   无
  287.  *
  288.  * 说明:
  289.  *   该函数打开待配准图象,并将图象存放在m_pDibSamp中。
  290.  *
  291.  *************************************************************************
  292.  */
  293. void CDlgReg::OnRegOpen() 
  294. {
  295. // TODO: Add your control notification handler code here
  296. CFileDialog dlg(TRUE,"bmp","*.bmp");
  297. if(dlg.DoModal() == IDOK)
  298. {
  299.  
  300.   CFile file;
  301.  
  302.   CString strPathName;
  303.  
  304. strPathName = dlg.GetPathName();
  305.  
  306. // 打开文件
  307. if( !file.Open(strPathName, CFile::modeRead | CFile::shareDenyWrite))
  308. {
  309. // 返回
  310. return ;
  311. }
  312. // 读入模板图象
  313. if(!m_pDibSamp->Read(&file)){
  314. // 恢复光标形状
  315. EndWaitCursor();
  316. // 清空已分配内存
  317. m_pDibSamp->Empty();
  318. // 返回
  319.   return;
  320. }
  321. }
  322. // 判断是否是8-bpp位图(这里为了方便,只处理8-bpp位图的模板配准,其它的可以类推)
  323. if(m_pDibSamp->m_nColorTableEntries != 256)
  324. {
  325. // 提示用户
  326. MessageBox("目前只支持256色位图!", "系统提示" , MB_ICONINFORMATION | MB_OK);
  327. // 清空已分配内存
  328. m_pDibSamp->Empty();
  329. // 返回
  330. return;
  331. }
  332. // 如果打开新的待配准文件,将图象位置设置标志位设为FALSE,以便再次调整位置
  333. m_bCalImgLoc = FALSE;
  334. // 更新显示
  335. this->UpdateData();
  336. this->Invalidate();
  337. }
  338. /*************************************************************************
  339.  *
  340.  * 函数名称:
  341.  *   OnRegChoseFeatureOnRegReg()
  342.  *
  343.  * 输入参数:
  344.  *   无
  345.  *
  346.  * 返回值:
  347.  *   无
  348.  *
  349.  * 说明:
  350.  *   该函数设置选取特征点标志位,然后调用函数在待配准图象中选取特征点,并
  351.  *配准这些特征点。特征点的数目至少应该选取三个。
  352.  *
  353.  *************************************************************************
  354.  */
  355. void CDlgReg::OnRegChoseFeature() 
  356. {
  357. // 如果待配准图象尚未打开,则不能进行特征点选取工作
  358. if((m_pDibSamp->IsEmpty())){
  359. AfxMessageBox("尚未打开待配准图象文件,请打开待配准图象");
  360. return;
  361. }
  362. // 设置选取特征点标志位
  363. m_bChoseFeature = TRUE;
  364. AfxMessageBox("请在待配准图象中选取特征点");
  365. }
  366. /*************************************************************************
  367.  *
  368.  * 函数名称:
  369.  *   OnLButtonUp()
  370.  *
  371.  * 输入参数:
  372.  *   无
  373.  *
  374.  * 返回值:
  375.  *   无
  376.  *
  377.  * 说明:
  378.  *   该函数根据鼠标所标定的位置设置特征点。然后调用特征配准函数配准此特征点。
  379.  *特征点的数目至少应该选取三个。
  380.  *
  381.  *************************************************************************
  382.  */
  383. void CDlgReg::OnLButtonUp(UINT nFlags, CPoint point) 
  384. {
  385. // 循环变量
  386. int i,j;
  387. // 如果特征选取标志位为TRUE,则进行特征点的选取和配准,否则退出
  388. if(!m_bChoseFeature){
  389. return;
  390. }
  391. // 待配准图象的特征选取区域,在这里选择特征点的选择区域要比图象的区
  392. // 域小一圈
  393. CRect rectChoose;
  394. rectChoose.bottom = m_rectResltImage.bottom - 5;
  395. rectChoose.top    = m_rectResltImage.top + 5;
  396. rectChoose.left   = m_rectResltImage.left + 5;
  397. rectChoose.right  = m_rectResltImage.right - 5;
  398. // 特征点的区域
  399. CRect rectFeature;
  400. // 标志位,表示此点是否是已经选择的特征点
  401. BOOL bFlag = FALSE;
  402. // 判断此点是否合法,并判断此点是否已经选择,如果是,则去掉此点
  403. if(rectChoose.PtInRect(point))
  404. {
  405. // 如果所选择的特征点是以前的特征点,则去掉此点
  406. for( i = 0; i<m_nChsFeatureNum; i++){
  407. // 选择特征点的显示区域,以便对特征点进行取舍
  408. rectFeature.bottom = m_pPointSampl[i].y + m_rectResltImage.top + 5;
  409. rectFeature.top    = m_pPointSampl[i].y + m_rectResltImage.top - 5;
  410. rectFeature.left   = m_pPointSampl[i].x + m_rectResltImage.left- 5;
  411. rectFeature.right  = m_pPointSampl[i].x + m_rectResltImage.left+ 5;
  412. // 判断所选择的特征点是否为原来选择的特征点
  413. // 如果是,则去掉此特征点
  414. if(rectFeature.PtInRect(point)){
  415. // 将后面的特征点向前移动一位,去掉所选择的此特征点
  416. for(j=i; j<m_nChsFeatureNum-1; j++){
  417. m_pPointSampl[j] = m_pPointSampl[j+1];
  418. m_pPointBase[j]  = m_pPointBase[j+1];
  419. }
  420. // 将特征点的计数减一
  421. m_nChsFeatureNum--;
  422. // 更新显示
  423. Invalidate();
  424. // 设置标志位
  425. bFlag = TRUE;
  426. // 退出
  427. return;
  428. }
  429. }
  430. // 在判断特征点是否已经选取够了
  431. if(m_nChsFeatureNum == 3){
  432. AfxMessageBox("你已经选取了3个特征点,如果要继续选取,你可以去掉配准不正确的特征点在进行选取");
  433. return;
  434. }
  435. // 如果此点是需要选取的,则进行相关操作
  436. if(!bFlag){
  437. // 将此待配准特征点选取,注意特征点的坐标是以图象的左上角为原点确定的
  438. m_pPointSampl[m_nChsFeatureNum].x = point.x - m_rectResltImage.left;
  439. m_pPointSampl[m_nChsFeatureNum].y = point.y - m_rectResltImage.top;
  440. // 配准此特征点
  441. m_pPointBase[m_nChsFeatureNum] = FindMatchPoint(m_pDibInit, m_pDibSamp, 
  442. m_pPointSampl[m_nChsFeatureNum]);
  443. // 将特征点计数加一
  444. m_nChsFeatureNum++;
  445. }
  446. }
  447. // 更新显示
  448. Invalidate();
  449. CDialog::OnLButtonUp(nFlags, point);
  450. }
  451. /*************************************************************************
  452.  *
  453.  * 函数名称:
  454.  *   OnLButtonMove()
  455.  *
  456.  * 输入参数:
  457.  *   无
  458.  *
  459.  * 返回值:
  460.  *   无
  461.  *
  462.  * 说明:
  463.  *   如果特征点选取标志位为TRUE,则该函数将待配准区域的鼠标设置为十字形状,
  464.  *以便能更精确的定位特征点。
  465.  *
  466.  *************************************************************************
  467.  */
  468. void CDlgReg::OnMouseMove(UINT nFlags, CPoint point) 
  469. {
  470. // 如果不是在特征点选取状态,则不进行相关的操作
  471. if(m_bChoseFeature){
  472. // 如果鼠标在待配准图象区域中,则更改鼠标形状
  473. if(m_rectResltImage.PtInRect(point))
  474. {
  475. ::SetCursor(LoadCursor(NULL,IDC_CROSS));
  476. }
  477. else
  478. {
  479. ::SetCursor(LoadCursor(NULL,IDC_ARROW));
  480. }
  481. }
  482. CDialog::OnMouseMove(nFlags, point);
  483. }
  484. /*************************************************************************
  485.  *
  486.  * 函数名称:
  487.  *   FindMatchPoint()
  488.  *
  489.  * 输入参数:
  490.  *   CDib* pDibBase - 基准图象指针
  491.  *   CDib* pDibSamp - 待配准图象指针
  492.  *   CPoint pointSamp - 待配准特征点的位置
  493.  *
  494.  * 返回值:
  495.  *   CPoint - 返回在基准图象中配准的特征点的位置
  496.  *
  497.  * 说明:
  498.  *   该函数根据待待配准图象中的特征点位置在基准图象中寻找配准特征点,并将
  499.  *配准的特征点位置返回。在配准的过程中,采取的是块配准的方法进行配准,在这里
  500.  *选取块的大小为7*7。搜索的方法为全局搜索。
  501.  *
  502.  *************************************************************************
  503.  */
  504. CPoint CDlgReg::FindMatchPoint(CDib* pDibBase, CDib* pDibSamp, 
  505.        CPoint pointSamp)
  506. {
  507. // 循环变量
  508. int i,j,m,n;
  509. // 临时变量
  510. int nX, nY;
  511. // 配准特征点位置
  512. CPoint pointBase;
  513. // 配准数据块的尺寸
  514. int nBlockLen = 7;
  515. int nBlockHalfLen =3;
  516. // 基准图象数据指针
  517. unsigned char* pBase;
  518. pBase = (unsigned char *)pDibBase->m_lpImage;
  519. // 待配准图象数据指针
  520. unsigned char* pSamp;
  521. pSamp = (unsigned char *)pDibSamp->m_lpImage;
  522. // 特征点位置的数据配准块
  523. unsigned char* pUnchSampBlock;
  524. pUnchSampBlock = new unsigned char[nBlockLen*nBlockLen];
  525. // 临时分配内存,用于存放配准数据块
  526. unsigned char* pUnchBaseBlock;
  527. pUnchBaseBlock = new unsigned char[nBlockLen*nBlockLen];
  528. // 相似度
  529. double dbCor;
  530. // 最大相似度
  531. double dbMaxCor = 0;
  532. // 基准图象的存储大小
  533. CSize sizeBaseImg;
  534. sizeBaseImg = pDibBase->GetDibSaveDim();
  535. // 待配准图象的存储大小
  536. CSize sizeSampImg;
  537. sizeSampImg = pDibSamp->GetDibSaveDim();
  538. // 从待配准图象中提取以特征点为中心的nBlockLen*nBlockLen0的数据块
  539. for(i=-nBlockHalfLen; i<=nBlockHalfLen; i++){
  540. for(j=-nBlockHalfLen; j<=nBlockHalfLen; j++){
  541. // 计算此点在图象中的位置
  542. nX = pointSamp.x + i;
  543. nY = sizeSampImg.cy - pointSamp.y + j +1;
  544. // 提取图象数据
  545. pUnchSampBlock[(j+nBlockHalfLen)*nBlockLen + (i+nBlockHalfLen)] = 
  546. pSamp[nY*sizeSampImg.cx + nX];
  547. }
  548. }
  549. // 基准图象的高度和宽度
  550. int nBaseImgHeight, nBaseImgWidth;
  551. nBaseImgHeight = pDibBase->m_lpBMIH->biHeight;
  552. nBaseImgWidth  = pDibBase->m_lpBMIH->biWidth;
  553. // 在基准图象中寻找配准特征点,采取的搜索方法为全局搜索
  554. for(m = nBlockHalfLen; m< nBaseImgHeight-nBlockHalfLen; m++){
  555. for(n=nBlockHalfLen; n<nBaseImgWidth-nBlockHalfLen; n++){
  556. // 提取以此点为中心,大小为nBlockLen*nBlockLen的数据块
  557. for(i=-nBlockHalfLen; i<=nBlockHalfLen; i++){
  558. for(j=-nBlockHalfLen; j<=nBlockHalfLen; j++){
  559. // 计算此点在图象中存储的位置
  560. nX = n + i;
  561. nY = sizeBaseImg.cy - m + j - 1;
  562. // 提取图象数据
  563. pUnchBaseBlock[(j+nBlockHalfLen)*nBlockLen + (i+nBlockHalfLen)] = 
  564. pBase[nY*sizeBaseImg.cx + nX];
  565. }
  566. }
  567. // 对这两个数据块进行配准,计算相似度
  568. dbCor = CalCorrelation(pUnchBaseBlock, pUnchSampBlock, nBlockLen);
  569. // 判断是否为最大相似度,如果是,则记录此相似度和配准特征点位置
  570. if(dbCor > dbMaxCor){
  571. dbMaxCor  = dbCor;
  572. pointBase.x = n;
  573. pointBase.y = m;
  574. }
  575. }
  576. }
  577. return pointBase;
  578. }
  579. /*************************************************************************
  580.  *
  581.  * 函数名称:
  582.  *   CalCorrelation()
  583.  *
  584.  * 输入参数:
  585.  *   unsigned char* pBase - 基准图象数据指针
  586.  *   unsigned char* pSamp - 待配准图象数据指针
  587.  *   int nBlockLen - 配准数据块的尺度大小
  588.  *
  589.  * 返回值:
  590.  *   double - 返回两个数据块配准的相似度
  591.  *
  592.  * 说明:
  593.  *   该函数对给定的两个大小为nBlockLen*nBlockLen的数据块,计算两者之间的
  594.  *的配准相似度。其中,去掉均值以消除亮度变换的影响。
  595.  *
  596.  *************************************************************************
  597.  */
  598. double CDlgReg::CalCorrelation(unsigned char* pBase, unsigned char* pSamp, int nBlockLen)
  599. {
  600. // 临时变量
  601. double dbSelfBase=0,dbSelfSamp=0;
  602. // 相似度
  603. double dbCor=0;
  604. // 块均值
  605. double dbMeanBase=0,dbMeanSamp=0;
  606. // 计算两个块的平均值
  607. for(int i=0;i<nBlockLen;i++)
  608. for(int j=0;j<nBlockLen;j++)
  609. {
  610. dbMeanBase += pBase[j*nBlockLen + i];
  611. dbMeanSamp += pSamp[j*nBlockLen + i];
  612. }
  613. dbMeanBase = dbMeanBase/(nBlockLen*nBlockLen);
  614. dbMeanSamp = dbMeanSamp/(nBlockLen*nBlockLen);
  615. // 求取相似度
  616. for(i=0;i<nBlockLen;i++)
  617. for(int j=0;j<nBlockLen;j++)
  618. {
  619. dbSelfBase += (pBase[j*nBlockLen + i] - dbMeanBase)*
  620. (pBase[j*nBlockLen + i]  - dbMeanBase);
  621. dbSelfSamp += (pSamp[j*nBlockLen + i] - dbMeanSamp)*
  622. (pSamp[j*nBlockLen + i]  - dbMeanSamp);
  623. dbCor += (pBase[j*nBlockLen + i] - dbMeanBase)*
  624. (pSamp[j*nBlockLen + i]  - dbMeanSamp);
  625. }
  626. dbCor = dbCor / sqrt(dbSelfBase * dbSelfSamp);
  627. // 返回相似度
  628. return dbCor;
  629. }
  630. /*************************************************************************
  631.  *
  632.  * 函数名称:
  633.  *   OnRegReg()
  634.  *
  635.  * 输入参数:
  636.  *   无
  637.  *
  638.  * 返回值:
  639.  *   无
  640.  *
  641.  * 说明:
  642.  *   该函数对两幅图象m_pDibInit和m_pDibSamp进行配准。其中,需要先确定特征点,
  643.  *特征点至少需要选取3个以上,这一步骤需要在“选取特征点”步骤中进行。根据选取
  644.  *的特征点,计算得到仿射参数。最后,将配准的两幅图象拼接起来显示。
  645.  *
  646.  *************************************************************************
  647.  */
  648. void CDlgReg::OnRegReg() 
  649. {
  650. // 进行配准之前,判断是否已经选取特征点
  651. if(!m_bChoseFeature){
  652. AfxMessageBox("还没有选取特征点,请先选取特征点");
  653. return;
  654. }
  655. // 如果选取的特征点数目不够,也不能进行处理
  656. if(m_nChsFeatureNum < 3){
  657. AfxMessageBox("请选择三个特征点,再进行图象配准");
  658. return;
  659. }
  660. // 仿射变换系数,仿射变换的系数为6个。此系数为从基准图象到待配准图象的变换系数
  661. double *pDbBs2SpAffPara;
  662. pDbBs2SpAffPara = new double[2*3];
  663. // 仿射变换系数,仿射变换的系数为6个。此系数为从待配准图象到基准图象的变换系数
  664. double *pDbSp2BsAffPara;
  665. pDbSp2BsAffPara = new double[2*3];
  666. // 如果已经选取了特征点,并得到了配准的特征点,则可以计算仿射变换系数了
  667. GetAffinePara(m_pPointBase, m_pPointSampl, pDbSp2BsAffPara);
  668. // 计算从待配准图象到基准图象的仿射变换系数
  669. GetAffinePara(m_pPointSampl, m_pPointBase, pDbBs2SpAffPara);
  670. // 计算图象经过仿射变换后的尺寸大小
  671. CRect rectAftAff;
  672. rectAftAff = GetAftAffDim(pDbSp2BsAffPara);
  673. // 根据图象的尺寸大小创建新的图象
  674. //CreateDIB();
  675. int nNewImgWidth, nNewImgHeight;
  676. nNewImgWidth  = rectAftAff.right - rectAftAff.left;
  677. nNewImgHeight = rectAftAff.bottom- rectAftAff.top;
  678. // 将基准图象放入新的图象中
  679. LPBYTE lpBaseImg;
  680. lpBaseImg = SetBaseImgAftAff(rectAftAff);
  681. // 将待配准图象放入新的图象中
  682. LPBYTE lpSampImg;
  683. lpSampImg = SetSampImgAftAff(pDbBs2SpAffPara, rectAftAff);
  684. // 将此图象用CDib类封装
  685. m_pDibResult = new CDib(CSize(nNewImgWidth,nNewImgHeight), 8);
  686. // 计算结果图象的存储大小尺寸
  687. CSize sizeSaveResult;
  688. sizeSaveResult = m_pDibResult->GetDibSaveDim();
  689. // 拷贝调色板
  690. memcpy(m_pDibResult->m_lpvColorTable, m_pDibInit->m_lpvColorTable, m_pDibResult->m_nColorTableEntries*sizeof(RGBQUAD));
  691. // 应用调色板
  692. m_pDibResult->MakePalette();
  693. // 分配内存给合并后的图象
  694. LPBYTE lpImgResult;
  695. lpImgResult = (LPBYTE)new unsigned char[sizeSaveResult.cx * sizeSaveResult.cy];
  696. // 对图象进行赋值
  697. for( int i=0; i<nNewImgWidth; i++)
  698. for( int j=0; j<nNewImgHeight; j++){
  699. int nX = i;
  700. int nY = sizeSaveResult.cy - j - 1;
  701. lpImgResult[nY*sizeSaveResult.cx + nX] = (lpBaseImg[j*nNewImgWidth + i] +lpSampImg[j*nNewImgWidth + i])/2;
  702. }
  703. // 将指针赋值给CDib类的数据
  704. m_pDibResult->m_lpImage = lpImgResult;
  705. // 显示合并后的图象
  706. CDlgAftReg* pDlg;
  707. pDlg = new CDlgAftReg(NULL, m_pDibResult);
  708. pDlg->DoModal();
  709. // 删除对象
  710. delete pDlg;
  711. // 释放已分配内存
  712. delete[]lpBaseImg;
  713. delete[]lpSampImg;
  714. delete[]pDbBs2SpAffPara;
  715. delete[]pDbSp2BsAffPara;
  716. }
  717. /*************************************************************************
  718.  *
  719.  * 函数名称:
  720.  *   GetAffinePara()
  721.  *
  722.  * 输入参数:
  723.  *   double  *pDbBs2SpAffPara - 用于存放基准图象到待配准图象的仿射变换系数
  724.  *   double  *pDbSp2BsAffPara - 用于存放待配准图象到基准图象的仿射变换系数
  725.  *
  726.  * 返回值:
  727.  *   无
  728.  *
  729.  * 说明:
  730.  *   该函数根据得到的三对配准的特征点,计算仿射变换系数。得到的仿射变换系数
  731.  *存放在两个输入参数所指向的内存中。
  732.  *
  733.  *************************************************************************
  734.  */
  735. void CDlgReg::GetAffinePara(CPoint* pPointBase, CPoint* pPointSampl, double* pDbAffPara)
  736. {
  737. // pDbBMatrix中存放的是基准图象中特征点的坐标,
  738. // 大小为2*m_nChsFeatureNum,前m_nChsFeatureNum为X坐标
  739. double *pDbBMatrix;
  740. pDbBMatrix = new double[2*m_nChsFeatureNum];
  741. // pDbSMatrix中存放的是待配准图象中特征点的扩展坐标
  742. // 大小为3*m_nChsFeatureNum,其中前m_nChsFeatureNum为X坐标
  743. // 中间m_nChsFeatureNum个为Y坐标,后面m_nChsFeatureNum为1
  744. double *pDbSMatrix;
  745. pDbSMatrix = new double[3*m_nChsFeatureNum];
  746. // pDbSMatrixT中存放的pDbSMatrix的转置矩阵,
  747. // 大小为m_nChsFeatureNum*3
  748. double *pDbSMatrixT;
  749. pDbSMatrixT = new double[m_nChsFeatureNum*3];
  750. // pDbInvMatrix为临时变量,存放的是pDbSMatrix*pDbSMatrixT的逆
  751. // 大小为3*3
  752. double *pDbInvMatrix;
  753. pDbInvMatrix = new double[3*3];
  754. // 临时内存
  755. double *pDbTemp;
  756. pDbTemp = new double[2*3];
  757. // 循环变量
  758. int count;
  759. // 给矩阵赋值
  760. for(count = 0; count<m_nChsFeatureNum; count++)
  761. {
  762. pDbBMatrix[0*m_nChsFeatureNum + count] = pPointBase[count].x;
  763. pDbBMatrix[1*m_nChsFeatureNum + count] = pPointBase[count].y;
  764. pDbSMatrix[0*m_nChsFeatureNum + count] = pPointSampl[count].x;
  765. pDbSMatrix[1*m_nChsFeatureNum + count] = pPointSampl[count].y;
  766. pDbSMatrix[2*m_nChsFeatureNum + count] = 1;
  767. pDbSMatrixT[count*m_nChsFeatureNum + 0] = pPointSampl[count].x;
  768. pDbSMatrixT[count*m_nChsFeatureNum + 1] = pPointSampl[count].y;
  769. pDbSMatrixT[count*m_nChsFeatureNum + 2] = 1;
  770. }
  771. // 计算pDbSMatrix*pDbSMatrixT,并将结果放入pDbInvMatrix中
  772. CalMatProduct(pDbSMatrix,pDbSMatrixT,pDbInvMatrix,3,3,m_nChsFeatureNum);
  773. // 计算pDbInvMatrix的逆矩阵
  774. CalInvMatrix(pDbInvMatrix, 3);
  775. // 计算仿射变换系数
  776. CalMatProduct(pDbBMatrix, pDbSMatrixT, pDbTemp, 2, 3, m_nChsFeatureNum);
  777. CalMatProduct(pDbTemp, pDbInvMatrix, pDbAffPara, 2, 3, 3);
  778. // 释放内存
  779. delete[]pDbBMatrix;
  780. delete[]pDbSMatrix;
  781. delete[]pDbSMatrixT;
  782. delete[]pDbInvMatrix;
  783. delete[]pDbTemp;
  784. }
  785. /*************************************************************************
  786.  *
  787.  * 函数名称:
  788.  *   CalMatProduct()
  789.  *
  790.  * 输入参数:
  791.  *   double  *pDbSrc1 - 指向相乘矩阵1的内存
  792.  *   double  *pDbSrc2 - 指向相乘矩阵2的内存
  793.  *   double  *pDbDest   - 存放矩阵相乘运行结果的内存指针
  794.  *   int     nX - 矩阵的尺寸,具体参见函数说明
  795.  *   int     nY - 矩阵的尺寸,具体参见函数说明
  796.  *   int     nZ - 矩阵的尺寸,具体参见函数说明
  797.  *
  798.  * 返回值:
  799.  *   无
  800.  *
  801.  * 说明:
  802.  *   该函数计算两个矩阵的相乘,然后将相乘的结果存放在pDbDest中。其中pDbSrc1
  803.  *的大小为nX*nZ,pDbSrc2的大小为nZ*nY,pDbDest的大小为nX*nY
  804.  *
  805.  *************************************************************************
  806.  */
  807. void CDlgReg::CalMatProduct(double* pDbSrc1, double *pDbSrc2, double *pDbDest, int y, int x, int z)
  808. {
  809. // 循环变量
  810. //int i,j,m;
  811. /*
  812. // 矩阵相乘
  813. for( i=0;i<nX;i++)
  814. for( j=0;j<nY;j++)
  815. {
  816. pDbDest[i*nY + j] = 0;
  817. for( m=0; m<nZ; m++)
  818. pDbDest[i*nY + j] += pDbSrc1[i*nZ + m]*pDbSrc2[m *nY + j];
  819. }
  820. */
  821. for(int i=0;i<y;i++)
  822. for(int j=0;j<x;j++)
  823. {
  824. pDbDest[i*x + j] = 0;
  825. for(int m=0;m<z;m++)
  826. pDbDest[i*x + j] += pDbSrc1[i*z + m]*pDbSrc2[m*x + j];
  827. }
  828. }
  829. /*************************************************************************
  830.  *
  831.  * 函数名称:
  832.  *   CalInvMatrix()
  833.  *
  834.  * 输入参数:
  835.  *   double  *pDbSrc - 指向矩阵的指针
  836.  *   int     nLen - 矩阵的尺寸 
  837.  *
  838.  * 返回值:
  839.  *   无
  840.  *
  841.  * 说明:
  842.  *   该函数计算矩阵pDbSrc的逆矩阵,其中pDbSrc的大小为nLen*nLen
  843.  *
  844.  *************************************************************************
  845.  */
  846. BOOL CDlgReg::CalInvMatrix(double *pDbSrc, int nLen)
  847. {
  848. int *is,*js,i,j,k;
  849. double d,p;
  850. is = new int[nLen];
  851. js = new int[nLen];
  852. for(k=0;k<nLen;k++)
  853. {
  854. d=0.0;
  855. for(i=k;i<nLen;i++)
  856. for(j=k;j<nLen;j++)
  857. {
  858. p=fabs(pDbSrc[i*nLen + j]);
  859. if(p>d)
  860. {
  861. d     = p; 
  862. is[k] = i;
  863. js[k] = j;
  864. }
  865. }
  866. if(d+1.0==1.0)
  867. {
  868. delete is;
  869. delete js;
  870. return FALSE;
  871. }
  872. if(is[k] != k)
  873. for(j=0;j<nLen;j++)
  874. {
  875. p = pDbSrc[k*nLen + j];
  876. pDbSrc[k*nLen + j] = pDbSrc[(is[k]*nLen) + j];
  877. pDbSrc[(is[k])*nLen + j] = p;
  878. }
  879. if(js[k] != k)
  880. for(i=0; i<nLen; i++)
  881. {
  882. p = pDbSrc[i*nLen + k];
  883. pDbSrc[i*nLen + k] = pDbSrc[i*nLen + (js[k])];
  884. pDbSrc[i*nLen + (js[k])] = p;
  885. }
  886. pDbSrc[k*nLen + k]=1.0/pDbSrc[k*nLen + k];
  887. for(j=0; j<nLen; j++)
  888. if(j != k)
  889. {
  890. pDbSrc[k*nLen + j]*=pDbSrc[k*nLen + k];
  891. }
  892. for(i=0; i<nLen; i++)
  893. if(i != k)
  894. for(j=0; j<nLen; j++)
  895. if(j!=k)
  896. {
  897. pDbSrc[i*nLen + j] -= pDbSrc[i*nLen + k]*pDbSrc[k*nLen + j];
  898. }
  899. for(i=0; i<nLen; i++)
  900. if(i != k)
  901. {
  902. pDbSrc[i*nLen + k] *= -pDbSrc[k*nLen + k];
  903. }
  904. }
  905. for(k=nLen-1; k>=0; k--)
  906. {
  907. if(js[k] != k)
  908. for(j=0; j<nLen; j++)
  909. {
  910. p = pDbSrc[k*nLen + j];
  911. pDbSrc[k*nLen + j] = pDbSrc[(js[k])*nLen + j];
  912. pDbSrc[(js[k])*nLen + j] = p;
  913. }
  914. if(is[k] != k)
  915. for(i=0; i<nLen; i++)
  916. {
  917. p = pDbSrc[i*nLen + k];
  918. pDbSrc[i*nLen + k] = pDbSrc[i*nLen +(is[k])];
  919. pDbSrc[i*nLen + (is[k])] = p;
  920. }
  921. }
  922. delete is;
  923. return TRUE;
  924. }
  925. /*************************************************************************
  926.  *
  927.  * 函数名称:
  928.  *   CalInvMatrix()
  929.  *
  930.  * 输入参数:
  931.  *   double  *pDbAffPara - 仿射变换系数矩阵
  932.  *
  933.  * 返回值:
  934.  *   CRect - 返回待配准图象经仿射变换后的区域
  935.  *
  936.  * 说明:
  937.  *   该函数根据仿射变换系数,计算待配准图象仿射变换后的图象尺寸大小
  938.  *
  939.  *************************************************************************
  940.  */
  941. CRect CDlgReg::GetAftAffDim(double* pDbAffPara)
  942. {
  943. // 基准图象的宽度和高度
  944. int nBaseImgWidth, nBaseImgHeight;
  945. nBaseImgWidth = m_pDibInit->m_lpBMIH->biWidth;
  946. nBaseImgHeight= m_pDibInit->m_lpBMIH->biHeight;
  947. // 待配准图象的宽度和高度
  948. int nSamplImgWidth, nSamplImgHeight;
  949. nSamplImgWidth = m_pDibSamp->m_lpBMIH->biWidth;
  950. nSamplImgHeight= m_pDibSamp->m_lpBMIH->biHeight;
  951. // 基准图象的原始区域
  952. CRect rectBase(0,0,nBaseImgWidth,nBaseImgHeight);
  953. // 临时变量
  954. CPoint pointTemp;
  955. double tx,ty;
  956. // 图象的端点
  957. pointTemp.x = 0; 
  958. pointTemp.y = 0;
  959. // 计算点pointTemp经过仿射变换后的坐标
  960. tx = pDbAffPara[0*3 +0]*pointTemp.x + 
  961. pDbAffPara[0*3 + 1]*pointTemp.y + pDbAffPara[0*3 + 2];
  962. ty = pDbAffPara[1*3 + 0]*pointTemp.x + 
  963. pDbAffPara[1*3 + 1]*pointTemp.y + pDbAffPara[1*3 + 2];
  964. // 判断pointTemp经过仿射变换后是否超出原来的大小
  965. if(tx<rectBase.left)
  966. rectBase.left = (int)tx;
  967. if(tx>rectBase.right)
  968. rectBase.right = (int)tx+1;
  969. if(ty<rectBase.top)
  970. rectBase.top = (int)ty;
  971. if(ty>rectBase.bottom)
  972. rectBase.bottom = (int)ty+1;
  973. // 计算端点(0, nSamplImgHeight)变换后的坐标
  974. pointTemp.x = 0; pointTemp.y = nSamplImgHeight;
  975. tx = pDbAffPara[0*3 + 0]*pointTemp.x + 
  976. pDbAffPara[0*3 + 1]*pointTemp.y + pDbAffPara[0*3 + 2];
  977. ty = pDbAffPara[1*3 +0]*pointTemp.x +
  978. pDbAffPara[1*3 + 1]*pointTemp.y + pDbAffPara[1*3 + 2];
  979. // 判断是否越界
  980. if(tx<rectBase.left)
  981. rectBase.left = (int)tx;
  982. if(tx>rectBase.right)
  983. rectBase.right = (int)tx+1;
  984. if(ty<rectBase.top)
  985. rectBase.top = (int)ty;
  986. if(ty>rectBase.bottom)
  987. rectBase.bottom = (int)ty+1;
  988. // 计算端点(nSamplImgWidth, nSamplImgHeight)变换后的坐标
  989. pointTemp.x = nSamplImgWidth; pointTemp.y = nSamplImgHeight;
  990. tx = pDbAffPara[0*3 + 0]*pointTemp.x +
  991. pDbAffPara[0*3 + 1]*pointTemp.y + pDbAffPara[0*3 + 2];
  992. ty = pDbAffPara[1*3 + 0]*pointTemp.x +
  993. pDbAffPara[1*3 + 1]*pointTemp.y + pDbAffPara[1*3 + 2];
  994. // 判断是否越界
  995. if(tx<rectBase.left)
  996. rectBase.left = (int)tx;
  997. if(tx>rectBase.right)
  998. rectBase.right = (int)tx+1;
  999. if(ty<rectBase.top)
  1000. rectBase.top = (int)ty;
  1001. if(ty>rectBase.bottom)
  1002. rectBase.bottom = (int)ty+1;
  1003. // 计算端点(nSamplImgWidth, 0)变换后的坐标
  1004. pointTemp.x = nSamplImgWidth; pointTemp.y = 0;
  1005. tx = pDbAffPara[0*3 + 0]*pointTemp.x +
  1006. pDbAffPara[0*3 + 1]*pointTemp.y + pDbAffPara[0*3 + 2];
  1007. ty = pDbAffPara[1*3 + 0]*pointTemp.x +
  1008. pDbAffPara[1*3 + 1]*pointTemp.y + pDbAffPara[1*3 + 2];
  1009. // 判断是否越界
  1010. if(tx<rectBase.left)
  1011. rectBase.left = (int)tx;
  1012. if(tx>rectBase.right)
  1013. rectBase.right = (int)tx+1;
  1014. if(ty<rectBase.top)
  1015. rectBase.top = (int)ty;
  1016. if(ty>rectBase.bottom)
  1017. rectBase.bottom = (int)ty+1;
  1018. // 返回待配准图象变换后的区域大小
  1019. return(rectBase);
  1020. }
  1021. /*************************************************************************
  1022.  *
  1023.  * 函数名称:
  1024.  *   SetSampImgAftAff()
  1025.  *
  1026.  * 输入参数:
  1027.  *   double  *pDbAffPara - 仿射变换系数矩阵
  1028.  *   CRect   rectNewImg - 变换后图象的大小尺寸
  1029.  *
  1030.  * 返回值:
  1031.  *   LPBYTE - 返回变换后的图象
  1032.  *
  1033.  * 说明:
  1034.  *   该函数根据仿射变换系数,计算待配准图象仿射变换后的图象。并返回此图象指针
  1035.  *此图象的大小为rectNewImg
  1036.  *
  1037.  *************************************************************************
  1038.  */
  1039. LPBYTE CDlgReg::SetSampImgAftAff(double* pDbAffPara, CRect rectNewImg)
  1040. {
  1041. // pUnchSect是4*4大小的矩阵数组
  1042. unsigned char *pUnchSect;
  1043. pUnchSect = new unsigned char[4*4];
  1044. // 新的图象宽度和高度
  1045. int nNewImgWidth, nNewImgHeight;
  1046. nNewImgWidth  = rectNewImg.right - rectNewImg.left;
  1047. nNewImgHeight = rectNewImg.bottom- rectNewImg.top;
  1048. // 待配准图象的宽度和高度
  1049. int nSamplImgWidth, nSamplImgHeight;
  1050. nSamplImgWidth = m_pDibSamp->m_lpBMIH->biWidth;
  1051. nSamplImgHeight= m_pDibSamp->m_lpBMIH->biHeight;
  1052. // 待配准图象的存储宽度
  1053. int nSampSaveWidth;
  1054. nSampSaveWidth = m_pDibSamp->GetDibSaveDim().cx;
  1055. // pUnchAftAffSamp是一个大小为rectNewImg大小的图象,
  1056. // 其中rectNewImg表示变换后的图象大小
  1057. unsigned char *pUnchAftAffSamp;
  1058. pUnchAftAffSamp = new unsigned char[nNewImgWidth * nNewImgHeight];
  1059. double tx,ty;
  1060. // 计算在变换后的图象的数据
  1061. for(int i=0;i<rectNewImg.bottom-rectNewImg.top;i++)
  1062. for(int j=0;j<rectNewImg.right-rectNewImg.left;j++)
  1063. {
  1064. tx = pDbAffPara[0*3 + 0]*(j+rectNewImg.left) +
  1065. pDbAffPara[0*3 + 1]*(i+rectNewImg.top) + pDbAffPara[0*3 + 2];
  1066. ty = pDbAffPara[1*3 + 0]*(j+rectNewImg.left) + 
  1067. pDbAffPara[1*3 + 1]*(i+rectNewImg.top) + pDbAffPara[1*3 + 2];
  1068. for(int m=(int)ty-1;m<=(int)ty+2;m++)
  1069. for(int n=(int)tx-1;n<=(int)tx+2;n++)
  1070. {
  1071. if(m<0||m>=nSamplImgHeight||n<0||n>=nSamplImgWidth)
  1072. pUnchSect[(m-(int)ty+1)*4 + (n-(int)tx+1)] = 0;
  1073. else
  1074. pUnchSect[(m-(int)ty+1)*4 + (n-(int)tx+1)] = 
  1075. m_pDibSamp->m_lpImage[(nSamplImgHeight-m-1)*nSampSaveWidth + n];
  1076. }
  1077. // 确定变换的坐标
  1078. ty = ty - (int)ty + 1;
  1079. tx = tx - (int)tx + 1;
  1080. // 确定变换后此坐标的数值
  1081. pUnchAftAffSamp[i*nNewImgWidth + j] = CalSpline(pUnchSect,tx,ty);
  1082. }
  1083. // 是否内存
  1084. delete[]pUnchSect;
  1085. // 返回指针
  1086. return (LPBYTE)pUnchAftAffSamp;
  1087. }
  1088. /*************************************************************************
  1089.  *
  1090.  * 函数名称:
  1091.  *   CalSpline()
  1092.  *
  1093.  * 输入参数:
  1094.  *   unsigned char *pUnchCorr - 插值的点
  1095.  *   double dX - X坐标
  1096.  *   double dY - Y坐标
  1097.  *
  1098.  * 返回值:
  1099.  *   unsigned char - 插值后的值
  1100.  *
  1101.  * 说明:
  1102.  *   该函数根据邻近位置的数值进行插值。
  1103.  *此图象的大小为rectNewImg
  1104.  *
  1105.  *************************************************************************
  1106.  */
  1107. unsigned char CDlgReg::CalSpline(unsigned char *pUnchCorr, double x, double y)
  1108. {
  1109. double ret=0, Cx, Cy;
  1110. double Temp;
  1111. for(int i=0;i<4;i++)
  1112. for(int j=0;j<4;j++)
  1113. {
  1114. Temp = pUnchCorr[i*4 + j];
  1115. if(fabs(y-i)<1)
  1116. Cy = 1-2*fabs(y-i)*fabs(y-i)+fabs(y-i)*fabs(y-i)*fabs(y-i);
  1117. if(fabs(y-i)>=1)
  1118. Cy = 4-8*fabs(y-i)+5*fabs(y-i)*fabs(y-i)-fabs(y-i)*fabs(y-i)*fabs(y-i);
  1119. if(fabs(x-j)<1)
  1120. Cx = 1-2*fabs(x-j)*fabs(x-j)+fabs(x-j)*fabs(x-j)*fabs(x-j);
  1121. if(fabs(x-j)>=1)
  1122. Cx = 4-8*fabs(x-j)+5*fabs(x-j)*fabs(x-j)-fabs(x-j)*fabs(x-j)*fabs(x-j);
  1123. ret += Temp*Cy*Cx;
  1124. }
  1125. if(ret<0)
  1126. ret=0;
  1127. if(ret>255)
  1128. ret=255;
  1129. return (unsigned char)ret;
  1130. }
  1131. /*************************************************************************
  1132.  *
  1133.  * 函数名称:
  1134.  *   SetBaseImgAftAff()
  1135.  *
  1136.  * 输入参数:
  1137.  *   double  *pDbAffPara - 仿射变换系数矩阵
  1138.  *
  1139.  * 返回值:
  1140.  *   无
  1141.  *
  1142.  * 说明:
  1143.  *   该函数根据仿射变换系数,计算基准图象仿射变换后的图象,并返回存放此
  1144.  *数据的指针
  1145.  *
  1146.  *************************************************************************
  1147.  */
  1148. LPBYTE CDlgReg::SetBaseImgAftAff(CRect rectNewImg)
  1149. {
  1150. // 新图象的大小
  1151. int nNewImgWidth, nNewImgHeight;
  1152. nNewImgWidth  = rectNewImg.right  - rectNewImg.left;
  1153. nNewImgHeight = rectNewImg.bottom - rectNewImg.top;
  1154. // 变换后图象
  1155. unsigned char *pUnchAftAffBase;
  1156. pUnchAftAffBase = new unsigned char[nNewImgWidth*nNewImgHeight];
  1157. // 基准图象的高度和宽度
  1158. int nBaseWidth, nBaseHeight;
  1159. nBaseWidth  = m_pDibInit->m_lpBMIH->biWidth;
  1160. nBaseHeight = m_pDibInit->m_lpBMIH->biHeight;
  1161. // 基准图象的存储宽度
  1162. int nBaseSaveWidth;
  1163. nBaseSaveWidth = m_pDibInit->GetDibSaveDim().cx;
  1164. for(int i=0;i<rectNewImg.bottom-rectNewImg.top;i++)
  1165. for(int j=0;j<rectNewImg.right-rectNewImg.left;j++)
  1166. {
  1167. if(i<-rectNewImg.top||i>=-rectNewImg.top+nBaseHeight||j<-rectNewImg.left||j>=-rectNewImg.left+nBaseWidth)
  1168. pUnchAftAffBase[i*nNewImgWidth + j] = 0;
  1169. else
  1170. pUnchAftAffBase[i*nNewImgWidth + j] = m_pDibInit->m_lpImage[(nBaseHeight - (i+rectNewImg.top) - 1)*nBaseSaveWidth + (j+rectNewImg.left)];
  1171. }
  1172. // 返回
  1173. return (LPBYTE)pUnchAftAffBase;
  1174. }