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

源码开发语言/平台
当前位置：首页 > 源码/资料 > 图形图像 > 波变换 > 查看源码
Wavelet2DDlg.cpp
资源名称：一维和二维小波变换的VC实现.rar　[点击查看]
上传用户：whhgrj
上传日期：2013-03-18
资源大小：169k
文件大小：19k
源码类别：

波变换

开发平台：

Visual C++

Wavelet2DDlg.cpp：源码内容
  1. // Wavelet2DDlg.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "Wavelet2D.h"
  6. #include "Wavelet2DDlg.h"
  7. #include "ColorTable.h"
  8. #ifdef _DEBUG
  9. #define new DEBUG_NEW
  10. #undef THIS_FILE
  11. static char THIS_FILE[] = __FILE__;
  12. #endif
  14. /////////////////////////////////////////////////////////////////////////////
  15. // CAboutDlg dialog used for App About
  17. class CAboutDlg : public CDialog
  18. {
  19. public:
  20. CAboutDlg();
  22. // Dialog Data
  23. //{{AFX_DATA(CAboutDlg)
  24. enum { IDD = IDD_ABOUTBOX };
  25. //}}AFX_DATA
  27. // ClassWizard generated virtual function overrides
  28. //{{AFX_VIRTUAL(CAboutDlg)
  29. protected:
  30. virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
  31. //}}AFX_VIRTUAL
  33. // Implementation
  34. protected:
  35. //{{AFX_MSG(CAboutDlg)
  36. //}}AFX_MSG
  37. DECLARE_MESSAGE_MAP()
  38. };
  40. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  41. {
  42. //{{AFX_DATA_INIT(CAboutDlg)
  43. //}}AFX_DATA_INIT
  44. }
  46. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  47. {
  48. CDialog::DoDataExchange(pDX);
  49. //{{AFX_DATA_MAP(CAboutDlg)
  50. //}}AFX_DATA_MAP
  51. }
  53. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  54. //{{AFX_MSG_MAP(CAboutDlg)
  55. // No message handlers
  56. //}}AFX_MSG_MAP
  57. END_MESSAGE_MAP()
  59. /////////////////////////////////////////////////////////////////////////////
  60. // CWavelet2DDlg dialog
  62. CWavelet2DDlg::CWavelet2DDlg(CWnd* pParent /*=NULL*/)
  63. : CDialog(CWavelet2DDlg::IDD, pParent)
  64. {
  65. //{{AFX_DATA_INIT(CWavelet2DDlg)
  66. // NOTE: the ClassWizard will add member initialization here
  67. m_bWavelet = FALSE;
  68. m_strBmpFile=_T("Sample.bmp");
  69. m_nDecomposeStage = 1;
  70. m_nFilterLen      = 6;
  71. //}}AFX_DATA_INIT
  72. // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
  73. m_strFilterFile[0]="h1.dat";
  74. m_strFilterFile[1]="h2.dat";
  75. m_strFilterFile[2]="h3.dat";
  76. m_strFilterFile[3]="h4.dat";
  77. m_strFilterFile[4]="h5.dat";
  78. m_strFilterFile[5]="h6.dat";
  79. m_strFilterFile[6]="h7.dat";
  80. m_strFilterFile[7]="h8.dat";
  81. m_strFilterFile[8]="h9.dat";
  82. m_strFilterFile[9]="h10.dat";
  84. m_pBmpInfo = (BITMAPINFO *)malloc(2048);
  85. if(m_pBmpInfo)
  86. {
  87. m_pBmpInfo->bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
  88. m_pBmpInfo->bmiHeader.biWidth = m_nImageWidth;
  89. m_pBmpInfo->bmiHeader.biHeight = m_nImageHeight;
  90. m_pBmpInfo->bmiHeader.biPlanes = 1;
  91. m_pBmpInfo->bmiHeader.biBitCount = 8;
  92. m_pBmpInfo->bmiHeader.biCompression = BI_RGB;
  93. m_pBmpInfo->bmiHeader.biSizeImage = 0;
  94. m_pBmpInfo->bmiHeader.biXPelsPerMeter = 0;
  95. m_pBmpInfo->bmiHeader.biYPelsPerMeter = 0;
  96. m_pBmpInfo->bmiHeader.biClrUsed = 0;
  97. m_pBmpInfo->bmiHeader.biClrImportant = 0;
  98. for( int k = 0; k < 256; ++k){
  99. m_pBmpInfo->bmiColors[k].rgbBlue = (BYTE)k;
  100. m_pBmpInfo->bmiColors[k].rgbGreen = (BYTE)k;
  101. m_pBmpInfo->bmiColors[k].rgbRed = (BYTE)k;
  102. m_pBmpInfo->bmiColors[k].rgbReserved = 0;
  103. }
  104. }
  106. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  107. }
  109. void CWavelet2DDlg::DoDataExchange(CDataExchange* pDX)
  110. {
  111. CDialog::DoDataExchange(pDX);
  112. //{{AFX_DATA_MAP(CWavelet2DDlg)
  113. DDX_Control(pDX, IDC_EDIT_BMPFILE, m_editBmpFile);
  114. DDX_Control(pDX, IDC_COMBO_FILTERLEN, m_combFilterLen);
  115. DDX_Control(pDX, IDC_COMBO_DECOMPOSESTAGE, m_combDecomposeStage);
  116. //}}AFX_DATA_MAP
  117. }
  119. BEGIN_MESSAGE_MAP(CWavelet2DDlg, CDialog)
  120. //{{AFX_MSG_MAP(CWavelet2DDlg)
  121. ON_WM_SYSCOMMAND()
  122. ON_WM_PAINT()
  123. ON_WM_QUERYDRAGICON()
  124. ON_BN_CLICKED(IDC_BUTTON_WAVELET2D, OnBtnWavelet2D)
  125. ON_BN_CLICKED(IDC_BUTTON_BMPFILE, OnBtnBmpFile)
  126. //}}AFX_MSG_MAP
  127. END_MESSAGE_MAP()
  129. /////////////////////////////////////////////////////////////////////////////
  130. // CWavelet2DDlg message handlers
  132. BOOL CWavelet2DDlg::OnInitDialog()
  133. {
  134. CDialog::OnInitDialog();
  136. // Add "About..." menu item to system menu.
  138. // IDM_ABOUTBOX must be in the system command range.
  139. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  140. ASSERT(IDM_ABOUTBOX < 0xF000);
  142. CMenu* pSysMenu = GetSystemMenu(FALSE);
  143. if (pSysMenu != NULL)
  144. {
  145. CString strAboutMenu;
  146. strAboutMenu.LoadString(IDS_ABOUTBOX);
  147. if (!strAboutMenu.IsEmpty())
  148. {
  149. pSysMenu->AppendMenu(MF_SEPARATOR);
  150. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  151. }
  152. }
  154. // Set the icon for this dialog.  The framework does this automatically
  155. //  when the application's main window is not a dialog
  156. SetIcon(m_hIcon, TRUE); // Set big icon
  157. SetIcon(m_hIcon, FALSE); // Set small icon
  159. // 初始化控件
  160. m_editBmpFile.SetWindowText(m_strBmpFile);
  161. CString strComb;
  162. int k=0;
  163. for(k=0;k<6;k++)
  164. { 
  165.     strComb.Format("%d",k+1);
  166.     m_combDecomposeStage.InsertString(k,strComb);
  167.         m_combDecomposeStage.SetItemData(k,k+1);
  168.         m_combDecomposeStage.SetCurSel(0);
  169. }
  170. for(k=0;k<10;k++)
  171. { 
  172.     strComb.Format("%d",2*(k+1));
  173.     m_combFilterLen.InsertString(k,strComb);
  174.         m_combFilterLen.SetItemData(k,2*(k+1));
  175.         m_combFilterLen.SetCurSel(2);
  176. }
  178. return TRUE;  // return TRUE  unless you set the focus to a control
  179. }
  181. void CWavelet2DDlg::OnSysCommand(UINT nID, LPARAM lParam)
  182. {
  183. if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  184. {
  185. CAboutDlg dlgAbout;
  186. dlgAbout.DoModal();
  187. }
  188. else
  189. {
  190. CDialog::OnSysCommand(nID, lParam);
  191. }
  192. }
  194. // If you add a minimize button to your dialog, you will need the code below
  195. //  to draw the icon.  For MFC applications using the document/view model,
  196. //  this is automatically done for you by the framework.
  198. void CWavelet2DDlg::OnPaint() 
  199. {
  200. if (IsIconic())
  201. {
  202. CPaintDC dc(this); // device context for painting
  204. SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
  206. // Center icon in client rectangle
  207. int cxIcon = GetSystemMetrics(SM_CXICON);
  208. int cyIcon = GetSystemMetrics(SM_CYICON);
  209. CRect rect;
  210. GetClientRect(&rect);
  211. int x = (rect.Width() - cxIcon + 1) / 2;
  212. int y = (rect.Height() - cyIcon + 1) / 2;
  214. // Draw the icon
  215. dc.DrawIcon(x, y, m_hIcon);
  216. }
  217. else
  218. {
  219. if(m_bWavelet)
  220.    OnBtnWavelet2D();
  221. CDialog::OnPaint();
  222. }
  223. }
  225. // The system calls this to obtain the cursor to display while the user drags
  226. //  the minimized window.
  227. HCURSOR CWavelet2DDlg::OnQueryDragIcon()
  228. {
  229. return (HCURSOR) m_hIcon;
  230. }
  232. // 选择原始数据图像文件
  233. void CWavelet2DDlg::OnBtnBmpFile() 
  234. {
  235. CFileDialog BmpFileDlg(TRUE,"bmp","*.bmp");
  236. if(BmpFileDlg.DoModal()==IDOK)
  237. {
  238.    m_bWavelet=FALSE;
  239. m_strBmpFile=BmpFileDlg.GetPathName();
  240.    m_editBmpFile.SetWindowText(m_strBmpFile);
  241. }
  242. }
  244. // “二维小波变换”按钮消息处理函数
  245. // 功能：打开文件，分配内存，进行变换，绘制结果
  246. void CWavelet2DDlg::OnBtnWavelet2D() 
  247. {
  248. CFile BmpFile;
  249. CFileException fe;
  250. // 打开原始图像数据
  251. if(m_strBmpFile.IsEmpty())
  252. {
  253. AfxMessageBox (_T("请选择一个位图（*.bmp）文件!"),MB_ICONINFORMATION|MB_OK,NULL);
  254.     return;
  255. }
  256. if(!BmpFile.Open(m_strBmpFile,CFile::modeRead,&fe))
  257. {
  258. AfxMessageBox (_T("打开文件失败!"),MB_ICONINFORMATION|MB_OK,NULL);
  259.     return;
  260. }
  261. // 读取原始数据图像文件
  262. TRY
  263. {
  264. DIB = ::ReadDIBFile(BmpFile);
  265. }
  266. CATCH (CFileException, eLoad)
  267. {
  268. BmpFile.Abort(); // 读取失败
  269. AfxMessageBox (_T("读取数据失败!"),MB_ICONINFORMATION|MB_OK,NULL);
  270. DIB = NULL;         // 设置DIB为空
  271. return ;
  272. }
  273. END_CATCH
  274.     
  275. // 锁定DIB
  276. lpDIB = (LPSTR)::GlobalLock((HGLOBAL)GetHDIB());
  277. // 得到图像宽度和高度
  278. m_nImageWidth=::DIBWidth(lpDIB);
  279. m_nImageHeight=::DIBHeight(lpDIB);
  281. // 打开待处理图像
  282. // 判断是否是8-bpp位图
  283. if (::DIBNumColors(lpDIB) > 256||m_nImageWidth>400||m_nImageHeight>400)
  284. {
  285. // 提示用户
  286. AfxMessageBox(_T("不是256色位图或者图像过大!") , MB_ICONINFORMATION | MB_OK,NULL);
  287. // 解除锁定
  288. ::GlobalUnlock((HGLOBAL) GetHDIB());
  289. return;
  290. }
  292. // 更改光标形状
  293. BeginWaitCursor();
  294. // 找到DIB图像象素起始位置
  295. lpDIBBits = ::FindDIBBits(lpDIB);
  296. // 得到小波变换所需的层次、滤波器长度等参数
  297. GetParam();
  298. // 获取滤波器参数
  299. if(!GetHn())
  300. return;
  301.     // 绘制原始图像
  302.     DrawOriginalBmp();
  303.     // 给image数组分配内存
  304. if(!GetMemory())
  305. return;
  306. // 将图像数据读入image数组
  307. for(int i=0;i<m_nImageHeight;i++)
  308. for(int j=0;j<m_nImageWidth;j++)
  309.     image[i][j]=(unsigned char)(*((unsigned char *)lpDIBBits+m_nImageWidth*i+j));
  310.      
  311. // 对图像进行二维小波变换
  312. Wavelet2D();
  313. // 绘制二维小波变换后的图像
  314. DrawWavelet2D();
  315. // 释放资源
  316. ReleaseMemory();
  317.   // 解除锁定
  318. ::GlobalUnlock((HGLOBAL) GetHDIB());
  319. // 恢复光标
  320. EndWaitCursor();
  322.     // 设置已进行过小波变换标志
  323. m_bWavelet=TRUE;
  324. }
  326. // 确定退出函数
  327. void CWavelet2DDlg::OnOK() 
  328. {
  329. if(m_pBmpInfo)
  330. free(m_pBmpInfo);
  332. CDialog::OnOK();
  333. }
  335. // 分配小波变换所需内存
  336. BOOL CWavelet2DDlg::GetMemory()
  337. {
  338.     BOOL flag=TRUE;
  340.     image=new double *[m_nImageHeight];
  341. for(int n=0;n<m_nImageHeight;n++)
  342. {
  343. *(image+n)=new double[sizeof(double)*m_nImageWidth];
  344.     if(*(image+n)==NULL)
  345.    flag=FALSE;
  346. }
  347. tempRow=new double[sizeof(double)*m_nImageWidth];
  348. tempCol=new double[sizeof(double)*m_nImageHeight];
  349. if(tempRow==NULL||tempCol==NULL)
  350. flag=FALSE;
  351.     return flag;
  352. }
  354. // 释放资源
  355. void CWavelet2DDlg::ReleaseMemory()
  356. {
  357. for(int k=0;k<m_nImageHeight;k++)
  358.   delete *(image+k);
  359.     delete []image; 
  360. delete tempCol;
  361. delete tempRow;
  362. image=NULL;
  363. tempCol=NULL;
  364. tempRow=NULL;
  365. if(DIB)
  366. { 
  367.    ::GlobalFree((HGLOBAL) DIB);
  368.    DIB=NULL;
  369. }
  370. }
  372. // 得到滤波器参数
  373. BOOL CWavelet2DDlg::GetHn()
  374. {
  376.      int i;
  377.  double temp1;
  378.  FILE *fp1;
  379.  if((fp1=fopen(m_strFilterFile[m_nFilterLen/2-1],"rb+"))==NULL)
  380.  {
  381.     AfxMessageBox("打开Hn数据文件失败!",MB_OK|MB_ICONINFORMATION);
  382. return FALSE;
  383.  }
  384.  for(i=0;i<30;i++)
  385.  h[i]=0;
  386.  for(i=0;i<m_nFilterLen;i++)
  387.  {
  388.     fread(&temp1,sizeof(double),1,fp1);
  389. h[i]=temp1;
  390.  }
  391.  fclose(fp1);
  392.      return TRUE;
  393. }
  395. //返回H算子周期化后得值
  396. double CWavelet2DDlg::HH(int i,int m)
  397. {
  398.      return h[i+m];
  400. }
  402. //返回G算子周期化后的值
  403. double CWavelet2DDlg::GG(int i, int m)
  404. {
  405.      char flag;
  406.  if(i%2)
  407.  flag=-1;
  408.  else
  409.  flag=1;
  410.  return HH(-1*i+1,m)*flag;
  411. }
  413. //H算子周期化过程
  414. BOOL CWavelet2DDlg::HHH(int level,int size)
  415. {
  416.      int len,m,i;
  417.  double temp;
  418.  int sign;
  419.  len=size>>level;
  420.  m=m_nFilterLen/2;
  421.  if((ph=new double[sizeof(double)*len])==NULL)
  422.  {
  423.     AfxMessageBox("H算子内存分配失败!",MB_OK|MB_ICONINFORMATION);
  424. return FALSE;
  425.  }
  426.  for(sign=0;sign<len;sign++)
  427.  {
  428.      temp=0;
  429.  for(i=(-1*m+1);i<=m;i++)
  430.  {
  431.      if(((sign-i)%len)==0)
  432.      temp+=HH(i-1,m);
  433.  }
  434.  ph[sign]=temp;
  435.  }
  436.  return TRUE;
  437. }
  439. // G算子周期化过程
  440. BOOL CWavelet2DDlg::GGG(int level,int size)
  441. {
  442.      int len,m,i;
  443.  double temp;
  444.  int sign;
  445.  len=size>>level;
  446.  m=m_nFilterLen/2;
  447.  if((pg=new double[sizeof(double)*len])==NULL)
  448.  {
  449.     AfxMessageBox("G算子内存分配失败!",MB_OK|MB_ICONINFORMATION);
  450. return FALSE;
  451.  }
  452.  for(sign=0;sign<len;sign++)
  453.  {
  454.      temp=0;
  455.  for(i=(-1*m+3);i<=m+2;i++)
  456.  {
  457.      if(((sign-i)%len)==0)
  458.      temp+=GG(i-1,m);
  459.  }
  460.  pg[sign]=temp;
  461.  }
  462.  return TRUE;
  463. }
  465. // H算子作用于行数据
  466. void CWavelet2DDlg::HOperatorOnRow(int filterlen, int start, int len,int row)
  467. {
  468.      int i,i2,m,k,len1;
  469.  double temp1;
  470.  m=m_nFilterLen/2;
  471.  len1=len>>1;
  472.  for(i=0;i<len1;i++)
  473.  {
  474.      i2=2*i;
  475.  tempRow[i+start]=0;
  476.  if(len>filterlen)
  477.  {
  478.     for(k=i2-len;k<=i2-len+m;k++)
  479. {
  480.     if((k>=0)&&(k<len))
  481. if((temp1=ph[(k-i2+len)%len])!=0)
  482. tempRow[i+start]+=temp1*image[row][k+start];
  483. }
  484. for(k=i2-m+1;k<=i2+m;k++)
  485. {
  486.     if((k>=0)&&(k<len))
  487. if((temp1=ph[(k-i2+len)%len])!=0)
  488. tempRow[i+start]+=temp1*image[row][k+start];
  489. }
  490. for(k=i2+len-m+1;k<=i2+len;k++)
  491. {
  492.     if((k>=0)&&(k<len))
  493. if((temp1=ph[(k-i2+len)%len])!=0)
  494. tempRow[i+start]+=temp1*image[row][k+start];
  495. }
  496.  }
  497.  else
  498.  {
  499.     for(k=0;k<len;k++)
  500. {
  501.     if((temp1=ph[(k-i2+len)%len])!=0)
  502. tempRow[i+start]+=temp1*image[row][k+start];
  503. }
  504.  }
  505.  }
  506. }
  508. // H算子作用于列数据
  509. void CWavelet2DDlg::HOperatorOnCol(int filterlen, int start, int len,int col)
  510. {
  511.  int i,i2,m,k,len1;
  512.  double temp1;
  513.  m=m_nFilterLen/2;
  514.  len1=len>>1;
  515.  for(i=0;i<len1;i++)
  516.  {
  517.      i2=2*i;
  518.  tempCol[i+start]=0;
  519.  if(len>filterlen)
  520.  {
  521.     for(k=i2-len;k<=i2-len+m;k++)
  522. {
  523.     if((k>=0)&&(k<len))
  524. if((temp1=ph[(k-i2+len)%len])!=0)
  525. tempCol[i+start]+=temp1*image[k+start][col];
  526. }
  527. for(k=i2-m+1;k<=i2+m;k++)
  528. {
  529.     if((k>=0)&&(k<len))
  530. if((temp1=ph[(k-i2+len)%len])!=0)
  531. tempCol[i+start]+=temp1*image[k+start][col];
  532. }
  533. for(k=i2+len-m+1;k<=i2+len;k++)
  534. {
  535.     if((k>=0)&&(k<len))
  536. if((temp1=ph[(k-i2+len)%len])!=0)
  537. tempCol[i+start]+=temp1*image[k+start][col];
  538. }
  539.  }
  540.  else
  541.  {
  542.     for(k=0;k<len;k++)
  543. {
  544.     if((temp1=ph[(k-i2+len)%len])!=0)
  545. tempCol[i+start]+=temp1*image[k+start][col];
  546. }
  547.  }
  548.  }
  550. }
  552. // G算子作用于行数据
  553. void CWavelet2DDlg::GOperatorOnRow(int filterlen, int start, int len,int row)
  554. {
  555.  int i,i2,m,k,len1;
  556.  double temp1;
  557.  m=m_nFilterLen/2;
  558.  len1=len>>1;
  559.  for(i=0;i<len1;i++)
  560.  {
  561.      i2=2*i;
  562.  tempRow[i+start+len1]=0;
  563.  if(len>filterlen)
  564.  {
  565.     for(k=i2-len;k<=i2-len+m+2;k++)
  566. {
  567.     if((k>=0)&&(k<len))
  568. if((temp1=pg[(k-i2+len)%len])!=0)
  569. tempRow[i+start+len1]+=temp1*image[row][k+start];
  570. }
  571. for(k=i2-m+3;k<=i2+m+2;k++)
  572. {
  573.     if((k>=0)&&(k<len))
  574. if((temp1=pg[(k-i2+len)%len])!=0)
  575. tempRow[i+start+len1]+=temp1*image[row][k+start];
  576. }
  577. for(k=i2+len-m+3;k<=i2+len;k++)
  578. {
  579.     if((k>=0)&&(k<len))
  580. if((temp1=pg[(k-i2+len)%len])!=0)
  581. tempRow[i+start+len1]+=temp1*image[row][k+start];
  582. }
  583.  }
  584.  else
  585.  {
  586.     for(k=0;k<len;k++)
  587. {
  588.     if((temp1=pg[(k-i2+len)%len])!=0)
  589. tempRow[i+start]+=temp1*image[row][k+start];
  590. }
  591.  }
  592.  }
  593. }
  595. // G算子作用于列数据
  596. void CWavelet2DDlg::GOperatorOnCol(int filterlen, int start, int len,int col)
  597. {
  598.  int i,i2,m,k,len1;
  599.  double temp1;
  600.  m=m_nFilterLen/2;
  601.  len1=len>>1;
  602.  for(i=0;i<len1;i++)
  603.  {
  604.      i2=2*i;
  605.  tempCol[i+start+len1]=0;
  606.  if(len>filterlen)
  607.  {
  608.     for(k=i2-len;k<=i2-len+m+2;k++)
  609. {
  610.     if((k>=0)&&(k<len))
  611. if((temp1=pg[(k-i2+len)%len])!=0)
  612. tempCol[i+start+len1]+=temp1*image[k+start][col];
  613. }
  614. for(k=i2-m+3;k<=i2+m+2;k++)
  615. {
  616.     if((k>=0)&&(k<len))
  617. if((temp1=pg[(k-i2+len)%len])!=0)
  618. tempCol[i+start+len1]+=temp1*image[k+start][col];
  619. }
  620. for(k=i2+len-m+3;k<=i2+len;k++)
  621. {
  622.     if((k>=0)&&(k<len))
  623. if((temp1=pg[(k-i2+len)%len])!=0)
  624. tempCol[i+start+len1]+=temp1*image[k+start][col];
  625. }
  626.  }
  627.  else
  628.  {
  629.     for(k=0;k<len;k++)
  630. {
  631.     if((temp1=pg[(k-i2+len)%len])!=0)
  632. tempCol[i+start+len1]+=temp1*image[k+start][col];
  633. }
  634.  }
  635.  }
  636. }
  638. // 进行第stage层行变换
  639. void CWavelet2DDlg::DecompStageRow(int row, int stage, int numx)
  640. {
  641.      int i,start,end;
  642.  int filterlen,len;
  643.  filterlen=m_nFilterLen;
  644.  len=m_nImageWidth>>stage;
  645.  start=len*numx;
  646.  end=start+len;
  647.  HOperatorOnRow(filterlen,start,len,row); // H算子作用于行
  648.  GOperatorOnRow(filterlen,start,len,row); // G算子作用于行
  649.  for(i=start;i<end;i++)
  650.  image[row][i]=tempRow[i];
  651. }
  653. // 进行第stage层列变换
  654. void CWavelet2DDlg::DecompStageCol(int col, int stage, int numx)
  655. {
  656.      int i,start,end;
  657.  int filterlen,len;
  658.  filterlen=m_nFilterLen;
  659.  len=m_nImageHeight>>stage;
  660.  start=len*numx;
  661.  end=start+len;
  662.  HOperatorOnCol(filterlen,start,len,col);  // H算子作用于列
  663.  GOperatorOnCol(filterlen,start,len,col);  // G算子作用于列
  664.  for(i=start;i<end;i++)
  665.  image[i][col]=tempCol[i];
  666. }
  668. // 二维小波变换
  669. void CWavelet2DDlg::Wavelet2D()
  670. {
  671.      int row,col,stage;
  672.  // 进行第stage层小波变换
  673.  for(stage=0;stage<m_nDecomposeStage;stage++)
  674.  {
  675.      // 初始化H算子和G算子
  676.  if((!HHH(stage,m_nImageWidth))||(!GGG(stage,m_nImageWidth)))
  677.  break;
  678.  else
  679.  // 逐行变换
  680.  for(row=0;row<(m_nImageHeight>>stage);row++)
  681.  {
  682.      DecompStageRow(row,stage,0);  
  683.  }
  684.      if(ph&&pg)      // 释放资源
  685.  {
  686.     delete  ph;
  687. delete  pg;
  688. ph=NULL;
  689. pg=NULL;
  690.  }
  691.       // 初始化H算子和G算子
  692.  if((!HHH(stage,m_nImageHeight))||(!GGG(stage,m_nImageHeight)))
  693.  break;
  694.  else
  695.  // 逐列变换
  696.  for(col=0;col<(m_nImageWidth>>stage);col++)
  697.  {
  698.      DecompStageCol(col,stage,0);
  699.  }
  700.          if(ph&&pg)      // 释放资源
  701.  {
  702.     delete  ph;
  703. delete  pg;
  704. ph=NULL;
  705. pg=NULL;
  706.  }
  707.  }
  708. }
  710. // 绘制原始图像
  711. void CWavelet2DDlg::DrawOriginalBmp()
  712. {
  713. // 得到客户区
  714. CDC* pDC=GetDC();
  715.     CRect RectClient,Workarea;
  716.     GetClientRect(RectClient);
  718. // 锁定DIB
  719. lpDIB = (LPSTR)::GlobalLock((HGLOBAL) GetHDIB());
  721.     // 在对话框右上部分创建绘制原始图像区域
  722. Workarea.left=RectClient.left+(RectClient.right-RectClient.left)/5*2;
  723.     Workarea.right=RectClient.right-10;
  724.     Workarea.top=RectClient.top+10;
  725. Workarea.bottom=RectClient.top+(RectClient.bottom-RectClient.top-10)/2;
  727. // 设置区域背景色为白色
  728. pDC->Rectangle(Workarea);
  729.     pDC->SetBkColor(0x00FFFFFF);
  731. // 绘制原始图像
  732. m_pBmpInfo->bmiHeader.biWidth = m_nImageWidth;
  733.     m_pBmpInfo->bmiHeader.biHeight = m_nImageHeight;
  734. ::StretchDIBits(pDC->GetSafeHdc(), Workarea.left+1, Workarea.top+1, Workarea.Width(),Workarea.Height(),//m_nImageWidth, m_nImageHeight,
  735.                 0, 0, m_nImageWidth, m_nImageHeight, (unsigned char *)lpDIBBits,
  736. (LPBITMAPINFO) m_pBmpInfo, DIB_RGB_COLORS, SRCCOPY);
  737.     // 解除锁定
  738. ::GlobalUnlock((HGLOBAL)GetHDIB());
  740. }
  742. // 绘制二维小波变换后图像
  743. void CWavelet2DDlg::DrawWavelet2D()
  744. {
  745. // 找到图像变换后数据中的最大最小值
  746. double min=image[0][0],max=0;
  747.     for(int a=0;a<m_nImageHeight;a++)
  748.    for(int b=0;b<m_nImageWidth;b++)
  749.    {
  750.    if(image[a][b]>10000)
  751.    image[a][b]=max;
  752.    if(image[a][b]<-10000)
  753.    image[a][b]=min;
  754.    if(max<image[a][b])
  755.           max=image[a][b];
  756.        if(min>image[a][b])
  757.           min=image[a][b];
  758.    }
  760.     // 将图像变换后数据归一化到256级灰度
  761. for(int m=0;m<m_nImageHeight;m++)
  762. for(int n=0;n<m_nImageWidth;n++)
  763. {  
  764.    image[m][n]=(image[m][n]-min)/(max-min)*256.0;
  765.     *((unsigned char *)lpDIBBits+m_nImageWidth*m+n)=(unsigned char )image[m][n];
  766. }
  768. // 得到客户区
  769. CDC* pDC=GetDC();
  770.     CRect RectClient,Workarea;
  771.     GetClientRect(RectClient);
  773. // 锁定DIB
  774. lpDIB = (LPSTR)::GlobalLock((HGLOBAL) GetHDIB());
  776. // 在对话框右下部分创建绘制原始图像区域
  777. Workarea.left=RectClient.left+(RectClient.right-RectClient.left)/5*2;
  778.     Workarea.right=RectClient.right-10;
  779. Workarea.top=RectClient.top+10+(RectClient.bottom-RectClient.top-10)/2;
  780.     Workarea.bottom=RectClient.bottom-10;
  782. // 设置区域背景色为白色 
  783.     pDC->Rectangle(Workarea);
  784.     pDC->SetBkColor(0x00FFFFFF);
  786. // 绘制原始图像
  787. m_pBmpInfo->bmiHeader.biWidth = m_nImageWidth;
  788.     m_pBmpInfo->bmiHeader.biHeight = m_nImageHeight;
  789. ::StretchDIBits(pDC->GetSafeHdc(), Workarea.left+1, Workarea.top+1, Workarea.Width(),Workarea.Height(),//m_nImageWidth, m_nImageHeight,
  790.                 0, 0, m_nImageWidth, m_nImageHeight, (unsigned char *)lpDIBBits,
  791. (LPBITMAPINFO) m_pBmpInfo, DIB_RGB_COLORS, SRCCOPY);
  793. // 解除锁定
  794. ::GlobalUnlock((HGLOBAL)GetHDIB());
  795. }
  797. // 得到小波变换所需的层次、滤波器长度等参数
  798. void CWavelet2DDlg::GetParam()
  799. {
  800. // 得到待分解层次信息
  801. if(m_combDecomposeStage.GetCurSel()==CB_ERR)
  802.    m_nDecomposeStage=1;
  803. else
  804.        m_nDecomposeStage=m_combDecomposeStage.GetItemData(m_combDecomposeStage.GetCurSel());
  806. // 得到滤波器长度
  807. if(m_combFilterLen.GetCurSel()==CB_ERR)
  808.    m_nFilterLen=6;
  809. else
  810.    m_nFilterLen=m_combFilterLen.GetItemData(m_combFilterLen.GetCurSel());
  811. }