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

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

波变换

开发平台：

Visual C++

Wavelet1DDlg.cpp：源码内容
  1. // Wavelet1DDlg.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "Wavelet1D.h"
  6. #include "Wavelet1DDlg.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. // CWavelet1DDlg dialog
  62. // 参数初始化
  63. CWavelet1DDlg::CWavelet1DDlg(CWnd* pParent /*=NULL*/)
  64. : CDialog(CWavelet1DDlg::IDD, pParent)
  65. {
  66. //{{AFX_DATA_INIT(CWavelet1DDlg)
  67. // NOTE: the ClassWizard will add member initialization here
  68. m_bWavelet = FALSE;
  69. m_strBmpFile=_T("Sample.bmp");
  70. m_nDecomposeStage = 1;
  71. m_nDisplayStage   = 1;
  72.     m_nFilterLen      = 6;
  73. m_nBinGate        =100;
  74. Data = NULL;
  75. DotData = NULL;
  76. //}}AFX_DATA_INIT
  77. // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
  78. m_strFilterFile[0]="h1.dat";
  79. m_strFilterFile[1]="h2.dat";
  80. m_strFilterFile[2]="h3.dat";
  81. m_strFilterFile[3]="h4.dat";
  82. m_strFilterFile[4]="h5.dat";
  83. m_strFilterFile[5]="h6.dat";
  84. m_strFilterFile[6]="h7.dat";
  85. m_strFilterFile[7]="h8.dat";
  86. m_strFilterFile[8]="h9.dat";
  87. m_strFilterFile[9]="h10.dat";
  88. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  89. }
  91. void CWavelet1DDlg::DoDataExchange(CDataExchange* pDX)
  92. {
  93. CDialog::DoDataExchange(pDX);
  94. //{{AFX_DATA_MAP(CWavelet1DDlg)
  95. DDX_Control(pDX, IDC_COMBO_DISPLAYSTAGE, m_combDisplayStage);
  96. DDX_Control(pDX, IDC_COMBO_FILTERLEN, m_combFilterLen);
  97. DDX_Control(pDX, IDC_COMBO_STAGE, m_combDecomposeStage);
  98. DDX_Control(pDX, IDC_EDIT_BMPFILE, m_editBmpFile);
  99. //}}AFX_DATA_MAP
  100. }
  102. BEGIN_MESSAGE_MAP(CWavelet1DDlg, CDialog)
  103. //{{AFX_MSG_MAP(CWavelet1DDlg)
  104. ON_WM_SYSCOMMAND()
  105. ON_WM_PAINT()
  106. ON_WM_QUERYDRAGICON()
  107. ON_BN_CLICKED(IDC_BUTTON_BMPFILE, OnBtnBmpFile)
  108. ON_BN_CLICKED(IDC_BUTTON_WAVELET1D, OnBtnWavelet1D)
  109. ON_CBN_SELCHANGE(IDC_COMBO_DISPLAYSTAGE, OnSelchangeCombDisplayStage)
  110. ON_CBN_EDITCHANGE(IDC_COMBO_DISPLAYSTAGE, OnEditchangeCombDisplayStage)
  111. ON_CBN_EDITCHANGE(IDC_COMBO_DECOMPOSESTAGE, OnEditchangeCombDecomposeStage)
  112. ON_CBN_EDITCHANGE(IDC_COMBO_FILTERLEN, OnEditchangeCombFilterLen)
  113. //}}AFX_MSG_MAP
  114. END_MESSAGE_MAP()
  116. /////////////////////////////////////////////////////////////////////////////
  117. // CWavelet1DDlg message handlers
  119. BOOL CWavelet1DDlg::OnInitDialog()
  120. {
  121. CDialog::OnInitDialog();
  123. // Add "About..." menu item to system menu.
  125. // IDM_ABOUTBOX must be in the system command range.
  126. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  127. ASSERT(IDM_ABOUTBOX < 0xF000);
  129. CMenu* pSysMenu = GetSystemMenu(FALSE);
  130. if (pSysMenu != NULL)
  131. {
  132. CString strAboutMenu;
  133. strAboutMenu.LoadString(IDS_ABOUTBOX);
  134. if (!strAboutMenu.IsEmpty())
  135. {
  136. pSysMenu->AppendMenu(MF_SEPARATOR);
  137. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  138. }
  139. }
  141. // Set the icon for this dialog.  The framework does this automatically
  142. //  when the application's main window is not a dialog
  143. SetIcon(m_hIcon, TRUE); // Set big icon
  144. SetIcon(m_hIcon, FALSE); // Set small icon
  146. // TODO: Add extra initialization here
  148. // 初始化控件
  149. m_editBmpFile.SetWindowText(m_strBmpFile);
  150. CString strComb;
  151. int k=0;
  152. for(k=0;k<6;k++)
  153. { 
  154.     strComb.Format("%d",k+1);
  155.     m_combDecomposeStage.InsertString(k,strComb);
  156.         m_combDecomposeStage.SetItemData(k,k+1);
  157.         m_combDecomposeStage.SetCurSel(0);
  158. }
  159. strComb.Format("%d",1);
  160. m_combFilterLen.InsertString(0,strComb);
  161.     m_combFilterLen.SetItemData(0,2);
  162. for(k=1;k<10;k++)
  163. { 
  164.     strComb.Format("%d",2*(k+1));
  165.     m_combFilterLen.InsertString(k,strComb);
  166.         m_combFilterLen.SetItemData(k,2*(k+1));
  167.         m_combFilterLen.SetCurSel(2);
  168. }
  170. m_combDisplayStage.EnableWindow(FALSE);
  172. return TRUE;  // return TRUE  unless you set the focus to a control
  173. }
  175. void CWavelet1DDlg::OnSysCommand(UINT nID, LPARAM lParam)
  176. {
  177. if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  178. {
  179. CAboutDlg dlgAbout;
  180. dlgAbout.DoModal();
  181. }
  182. else
  183. {
  184. CDialog::OnSysCommand(nID, lParam);
  185. }
  186. }
  188. // If you add a minimize button to your dialog, you will need the code below
  189. //  to draw the icon.  For MFC applications using the document/view model,
  190. //  this is automatically done for you by the framework.
  192. void CWavelet1DDlg::OnPaint() 
  193. {
  194. if (IsIconic())
  195. {
  196. CPaintDC dc(this); // device context for painting
  198. SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
  200. // Center icon in client rectangle
  201. int cxIcon = GetSystemMetrics(SM_CXICON);
  202. int cyIcon = GetSystemMetrics(SM_CYICON);
  203. CRect rect;
  204. GetClientRect(&rect);
  205. int x = (rect.Width() - cxIcon + 1) / 2;
  206. int y = (rect.Height() - cyIcon + 1) / 2;
  208. // Draw the icon
  209. dc.DrawIcon(x, y, m_hIcon);
  210. }
  211. else
  212. {
  213. CDialog::OnPaint();
  214.      if(m_bWavelet)
  215.    DrawWavelet1D();
  216. }
  217. }
  219. // The system calls this to obtain the cursor to display while the user drags
  220. //  the minimized window.
  221. HCURSOR CWavelet1DDlg::OnQueryDragIcon()
  222. {
  223. return (HCURSOR) m_hIcon;
  224. }
  226. void CWavelet1DDlg::OnSelchangeCombDisplayStage() 
  227. {
  228. m_nDisplayStage=m_combDisplayStage.GetItemData(m_combDisplayStage.GetCurSel());
  229. DrawWavelet1D();
  230. }
  232. void CWavelet1DDlg::OnEditchangeCombDisplayStage() 
  233. {
  234. m_combDisplayStage.SetCurSel(1);
  235. }
  237. void CWavelet1DDlg::OnEditchangeCombDecomposeStage() 
  238. {
  239. m_combDecomposeStage.SetCurSel(0);
  240. }
  242. void CWavelet1DDlg::OnEditchangeCombFilterLen() 
  243. {
  244.     m_combFilterLen.SetCurSel(2);
  245. }
  247. // 选择原始数据图像文件
  248. void CWavelet1DDlg::OnBtnBmpFile() 
  249. {
  250. CFileDialog BmpFileDlg(TRUE,"bmp","*.bmp");
  251. if(BmpFileDlg.DoModal()==IDOK)
  252. {
  253.    m_strBmpFile=BmpFileDlg.GetPathName();
  254.    m_editBmpFile.SetWindowText(m_strBmpFile);
  255.    m_bWavelet=FALSE;
  256. }
  257. }
  259. // 确定退出
  260. void CWavelet1DDlg::OnOK() 
  261. {
  262. // 若已进行过小波变换，则所用资源
  263. if(m_bWavelet)
  264. {
  265.    ReleaseMemory();
  266. }
  267. CDialog::OnOK();
  268. }
  270. // “一维小波变换”按钮消息处理函数
  271. // 功能：打开文件，分配内存，进行变换，绘制结果
  272. void CWavelet1DDlg::OnBtnWavelet1D() 
  273. {
  274. CFile BmpFile;
  275. CFileException fe;
  276. // 打开原始数据图像文件
  277. if(m_strBmpFile.IsEmpty())
  278. {
  279. AfxMessageBox (_T("请选择一个位图（*.bmp）文件!"),MB_ICONINFORMATION|MB_OK,NULL);
  280.     return;
  281. }
  282. if(!BmpFile.Open(m_strBmpFile,CFile::modeRead,&fe))
  283. {
  284. AfxMessageBox (_T("打开文件失败!"),MB_ICONINFORMATION|MB_OK,NULL);
  285.     return;
  286. }
  287. // 若已进行过小波变换，则释放上次变换所用资源
  288. if(m_bWavelet)
  289. {
  290.    ReleaseMemory();
  291. }
  292. // 读取原始数据图像文件
  293. TRY
  294. {
  295. DIB = ::ReadDIBFile(BmpFile);
  296. }
  297. CATCH (CFileException, eLoad)
  298. {
  299. BmpFile.Abort();  // 读取失败
  300. AfxMessageBox (_T("读取数据失败!"),MB_ICONINFORMATION|MB_OK,NULL);
  301. DIB = NULL;         // 设置DIB为空
  302. return ;
  303. }
  304. END_CATCH
  306. // 锁定DIB
  307. lpDIB=(LPSTR) ::GlobalLock((HGLOBAL)GetHDIB());
  309. //打开待处理图像
  310. // 判断是否是8-bpp位图
  311. if (::DIBNumColors(lpDIB) > 256||m_nImageWidth>=400||m_nImageHeight>=300)
  312. {
  313. // 提示用户
  314. AfxMessageBox(_T("不是256色位图或者图像过大!") , MB_ICONINFORMATION | MB_OK,NULL);
  315. // 解除锁定
  316. ::GlobalUnlock((HGLOBAL) GetHDIB());
  317. return ;
  318. }
  319. // 得到原始数据图像的宽度和高度
  320. m_nImageWidth=::DIBWidth(lpDIB);
  321. m_nImageHeight=::DIBHeight(lpDIB);
  323. // 更改光标形状
  324. BeginWaitCursor();
  325. // 找到DIB图像象素起始位置
  326. lpDIBBits = ::FindDIBBits(lpDIB);
  327. // 得到小波变换所需的层次、滤波器长度等参数
  328. GetParam();
  329. // 获取滤波器参数
  330. if(!GetHn())
  331. return;
  332. // 给Data数组分配内存
  333. if(!GetMemory())
  334. return;
  335. // 将原始数据图像二值化并投影，将其存储到Data[0]中
  336. PrepareData();
  337. // 对Data[0]中数据进行小波变换
  338. Wavelet1D();
  339. // 绘制小波变换结果
  340. DrawWavelet1D();
  341.   // 解除锁定
  342. ::GlobalUnlock((HGLOBAL)GetHDIB());
  343. // 恢复光标
  344. EndWaitCursor();
  346.     // 设置已进行过小波变换标志
  347. m_bWavelet=TRUE;
  348. }
  350. // 得到小波变换所需的层次、滤波器长度等参数
  351. void CWavelet1DDlg::GetParam()
  352. {
  353. // 得到要进行小波变换的层次
  354. if(m_combDecomposeStage.GetCurSel()==CB_ERR)
  355.    m_nDecomposeStage=1;
  356. else
  357.        m_nDecomposeStage=m_combDecomposeStage.GetItemData(m_combDecomposeStage.GetCurSel());
  358. // 得到滤波器长度
  359. if(m_combFilterLen.GetCurSel()==CB_ERR)
  360.    m_nFilterLen=6;
  361. else
  362.    m_nFilterLen=m_combFilterLen.GetItemData(m_combFilterLen.GetCurSel());
  363.     // 显示第一层小波变换后的结果
  364.     m_nDisplayStage=1;
  365.     // 只有在进行小波变换后才能初始化显示变换结果层次的Comb控件
  366. CString strComb;
  367. m_combDisplayStage.EnableWindow(TRUE);
  368.     m_combDisplayStage.ResetContent();
  369. for(int k=0;k<=m_nDecomposeStage;k++)
  370. {
  371.     strComb.Format("%d",k);     
  372. m_combDisplayStage.InsertString(k,strComb);
  373.         m_combDisplayStage.SetItemData(k,k);
  374.         m_combDisplayStage.SetCurSel(1);
  375. }
  376. }
  378. // 将原始数据图像二值化并投影，将其存储到Data[0]中
  379. void CWavelet1DDlg::PrepareData()
  380. {
  381.     // 原始数据图像二值化及垂直、水平方向投影
  382. // DotData[i]存放投影后一维待变换数据
  383. // Data[0][i]同DotData[i]，存放投影后一维待变换数据
  384. // Data[1][i]存放一层小波变换后数据，依此类推
  385. int temp;
  386.     int i,j;
  387.     for(i=0;i<m_nImageHeight+m_nImageWidth;i++)
  388.     *(DotData+i)=0;
  389.     for(i=0;i<m_nImageHeight;i++)
  390.    for(j=0;j<m_nImageWidth;j++)
  391.    {  
  392.            temp=*((unsigned char *)lpDIBBits+m_nImageWidth*i+j);
  393.    if(temp<=m_nBinGate)
  394.    {
  395.   (*(DotData+i))++;
  396.   (*(DotData+m_nImageHeight+j))++;
  397.   *((unsigned char *)lpDIBBits+m_nImageWidth*i+j)=0;
  398.    }
  399.    else
  400.    {
  401.    *((unsigned char *)lpDIBBits+m_nImageWidth*i+j)=255;
  402.    }
  403.    }
  405. // 将二值化后图像点阵投影数据读入Data数组
  406. for(i=0;i<m_nImageHeight+m_nImageWidth;i++)
  407. Data[0][i]=DotData[i];
  408. }
  410. // 一维小波变换
  411. void CWavelet1DDlg::Wavelet1D()
  412. {
  413.      int i,j;
  414.  for(i=1;i<=m_nDecomposeStage;i++)
  415.  {
  416.      // 复制数据
  417.  for(j=0;j<(m_nImageWidth+m_nImageHeight);j++)
  418.      Data[i][j]=Data[i-1][j];
  419.  // 第i层小波变换
  420.          if(!DecompStage(i,m_nFilterLen))
  421.  {
  422.  AfxMessageBox("变换失败!",MB_OK|MB_ICONINFORMATION);
  423.      break;
  424.  }
  425.  }
  426. }
  428. // 得到滤波器参数
  429. BOOL CWavelet1DDlg::GetHn()
  430. {
  431.      int i;
  432.  double temp1;
  433.  FILE *fp1;
  434.  if((fp1=fopen(m_strFilterFile[m_nFilterLen/2-1],"rb+"))==NULL)
  435.  {
  436.     AfxMessageBox("打开Hn数据文件失败!",MB_OK|MB_ICONINFORMATION);
  437. return FALSE;
  438.  }
  439.  for(i=0;i<30;i++)
  440.  h[i]=0;
  441.  for(i=0;i<m_nFilterLen;i++)
  442.  {
  443.     fread(&temp1,sizeof(double),1,fp1);
  444. h[i]=temp1;
  445.  }
  446.  fclose(fp1);
  447.      return TRUE;
  448. }
  450. // 分配小波变换所需内存
  451. BOOL CWavelet1DDlg::GetMemory()
  452. {
  453.     BOOL flag=TRUE;
  454.     Data=new double *[m_nImageHeight];
  455. for(int n=0;n<=m_nDecomposeStage;n++)
  456. {
  457. *(Data+n)=new double[sizeof(double)*m_nImageWidth];
  458.     if(*(Data+n)==NULL)
  459.    flag=FALSE;
  460. }
  461. DotData=new double[sizeof(double)*(m_nImageWidth+m_nImageHeight)];
  462. if(DotData==NULL)
  463. flag=FALSE;
  464.     return flag;
  465. }
  467. // 释放小波变换所用内存资源
  468. void CWavelet1DDlg::ReleaseMemory()
  469. {
  470.    for(int k=0;k<m_nDecomposeStage;k++)
  471.    if(*(Data+k))
  472.      delete *(Data+k);
  473.    if(Data)
  474.    {
  475.   delete []Data; 
  476.   Data=NULL;
  477.    }
  478.    if(DotData)
  479.    {
  480.   delete DotData;
  481.   DotData=NULL;
  482.    }
  483.    if(DIB)
  484.    { 
  485.   ::GlobalFree((HGLOBAL) DIB);
  486.   DIB=NULL;
  487.    } 
  488. }
  490. // 返回H算子周期化后得值
  491. double CWavelet1DDlg::HH(int i,int m)
  492. {
  493.      return h[i+m];
  495. }
  497. // 返回G算子周期化后的值
  498. double CWavelet1DDlg::GG(int i, int m)
  499. {
  500.      char flag;
  501.  if(i%2)
  502.  flag=-1;
  503.  else
  504.  flag=1;
  505.  return HH(-1*i+1,m)*flag;
  506. }
  508. // H算子周期化过程
  509. BOOL CWavelet1DDlg::HHH(int level,int size)
  510. {
  511.      int len,m,i;
  512.  double temp;
  513.  int sign;
  514.  len=size>>level;
  515.  m=m_nFilterLen/2;
  516.  if((ph=new double[sizeof(double)*len])==NULL)
  517.  {
  518.     AfxMessageBox("H算子内存分配失败!",MB_OK|MB_ICONINFORMATION);
  519. return FALSE;
  520.  }
  521.  for(sign=0;sign<len;sign++)
  522.  {
  523.      temp=0;
  524.  for(i=(-1*m+1);i<=m;i++)
  525.  {
  526.      if(((sign-i)%len)==0)
  527.      temp+=HH(i-1,m);
  528.  }
  529.  ph[sign]=temp;
  530.  }
  531.  return TRUE;
  532. }
  534. // G算子周期化过程
  535. BOOL CWavelet1DDlg::GGG(int level,int size)
  536. {
  537.      int len,m,i;
  538.  double temp;
  539.  int sign;
  540.  len=size>>level;
  541.  m=m_nFilterLen/2;
  542.  if((pg=new double[sizeof(double)*len])==NULL)
  543.  {
  544.     AfxMessageBox("G算子内存分配失败!",MB_OK|MB_ICONINFORMATION);
  545. return FALSE;
  546.  }
  547.  for(sign=0;sign<len;sign++)
  548.  {
  549.      temp=0;
  550.  for(i=(-1*m+3);i<=m+2;i++)
  551.  {
  552.      if(((sign-i)%len)==0)
  553.      temp+=GG(i-1,m);
  554.  }
  555.  pg[sign]=temp;
  556.  }
  557.  return TRUE;
  558. }
  560. // H算子作用于一维数据
  561. void CWavelet1DDlg::HOperatorOn(int filterlen, int start, int len,int level)
  562. {
  563.      int i,i2,m,k,len1;
  564.  double temp1;
  565.  m=m_nFilterLen/2;
  566.  len1=len>>1;
  567.  for(i=0;i<len1;i++)
  568.  {
  569.      i2=2*i;
  570.  Data[level][i+start]=0;
  571.  if(len>filterlen)
  572.  {
  573.     for(k=i2-len;k<=i2-len+m;k++)
  574. {
  575.     if((k>=0)&&(k<len))
  576. if((temp1=ph[(k-i2+len)%len])!=0)
  577. Data[level][i+start]+=temp1*Data[level-1][k+start];
  578. }
  579. for(k=i2-m+1;k<=i2+m;k++)
  580. {
  581.     if((k>=0)&&(k<len))
  582. if((temp1=ph[(k-i2+len)%len])!=0)
  583. Data[level][i+start]+=temp1*Data[level-1][k+start];
  584. }
  585. for(k=i2+len-m+1;k<=i2+len;k++)
  586. {
  587.     if((k>=0)&&(k<len))
  588. if((temp1=ph[(k-i2+len)%len])!=0)
  589. Data[level][i+start]+=temp1*Data[level-1][k+start];
  590. }
  591.  }
  592.  else
  593.  {
  594.     for(k=0;k<len;k++)
  595. {
  596.     if((temp1=ph[(k-i2+len)%len])!=0)
  597. Data[level][i+start]+=temp1*Data[level-1][k+start];
  598. }
  599.  }
  600.  }
  601. }
  603. // G算子作用于一维数据
  604. void CWavelet1DDlg::GOperatorOn(int filterlen, int start, int len,int level)
  605. {
  606.  int i,i2,m,k,len1;
  607.  double temp1;
  608.  m=m_nFilterLen/2;
  609.  len1=len>>1;
  610.  for(i=0;i<len1;i++)
  611.  {
  612.      i2=2*i;
  613.  Data[level][i+start+len1]=0;
  614.  if(len>filterlen)
  615.  {
  616.     for(k=i2-len;k<=i2-len+m+2;k++)
  617. {
  618.     if((k>=0)&&(k<len))
  619. if((temp1=pg[(k-i2+len)%len])!=0)
  620. Data[level][i+start+len1]+=temp1*Data[level-1][k+start];
  621. }
  622. for(k=i2-m+3;k<=i2+m+2;k++)
  623. {
  624.     if((k>=0)&&(k<len))
  625. if((temp1=pg[(k-i2+len)%len])!=0)
  626. Data[level][i+start+len1]+=temp1*Data[level-1][k+start];
  627. }
  628. for(k=i2+len-m+3;k<=i2+len;k++)
  629. {
  630.     if((k>=0)&&(k<len))
  631. if((temp1=pg[(k-i2+len)%len])!=0)
  632. Data[level][i+start+len1]+=temp1*Data[level-1][k+start];
  633. }
  634.  }
  635.  else
  636.  {
  637.     for(k=0;k<len;k++)
  638. {
  639.     if((temp1=pg[(k-i2+len)%len])!=0)
  640. Data[level][i+start+len1]+=temp1*Data[level-1][k+start];
  641. }
  642.  }
  643.  }
  644. }
  646. // 对一维数据进行level层小波变换
  647. BOOL CWavelet1DDlg::DecompStage(int level, int filterlen)
  648. {
  649.      int start,len;
  650.  len=(m_nImageWidth+m_nImageHeight)>>(level-1);
  651.  start=0;
  652.  if((!HHH(level-1,m_nImageWidth+m_nImageHeight))||(!GGG(level-1,m_nImageWidth+m_nImageHeight)))
  653.      return FALSE;
  654.  HOperatorOn(filterlen,start,len,level);
  655.  GOperatorOn(filterlen,start,len,level);
  656.  if(ph&&pg)
  657.  {
  658.     delete  ph;
  659. delete  pg;
  660. ph=NULL;
  661. pg=NULL;
  662.  }
  663.  return TRUE;     
  664. }
  666. // 绘制m_nDisplayStage指定层次的小波变换结果
  667. void CWavelet1DDlg::DrawWavelet1D()
  668. {
  669. // 创建绘图所需资源
  670. int Ox=0,Oy=0;
  671. int Ox1=0,Oy1=0,Ox2=0,Oy2=0;
  672. CString Text=_T("");
  673. CPen *OldPen,PenWhite,PenGreen,PenBlue,Pen;
  674. CPen* OldPen2;
  675. PenWhite.CreatePen(PS_SOLID,1,0x00FFFFFF);
  676. PenGreen.CreatePen(PS_SOLID,1,0x0000FF00);
  677. PenBlue.CreatePen(PS_SOLID,1,0x00FF0000);
  678. Pen.CreatePen(PS_SOLID,1,0x00FFFFFF);
  679. CDC* pDC=GetDC();
  680.     CRect RectClient,Workarea;
  681.     GetClientRect(RectClient);
  683. // 锁定DIB
  684. lpDIB = (LPSTR)::GlobalLock((HGLOBAL) GetHDIB());
  686.     // 在对话框右上部分绘制原始数据图像及二值化后点阵投影图像
  687. int temp;
  688.     int i,j;
  689. // 创建绘制区
  690. Workarea.left=RectClient.left+(RectClient.right-RectClient.left)/5*2;
  691.     Workarea.right=RectClient.right-10;
  692.     Workarea.top=RectClient.top+10;
  693. Workarea.bottom=RectClient.top+(RectClient.bottom-RectClient.top-10)/2+40;
  694.     pDC->Rectangle(Workarea);
  695.     pDC->SetBkColor(0x00FFFFFF);
  696.     // 绘制坐标轴
  697. Ox = Workarea.left+20;    Oy = Workarea.bottom-20;
  698. pDC->MoveTo(Ox,Oy);
  699. pDC->LineTo(Ox,Workarea.top+10);
  700. pDC->MoveTo(Ox,Oy);
  701. pDC->LineTo(Workarea.right-10,Oy);
  702.     // 绘制二值化后数据图像
  703.     Ox1= Ox+m_nImageWidth;           Oy1= Oy-m_nImageHeight;
  704. pDC->MoveTo(Ox1,Oy1);
  705. pDC->LineTo(Ox1,Oy1-m_nImageHeight);
  706. pDC->LineTo(Ox1+m_nImageWidth,Oy1-m_nImageHeight);
  707. pDC->LineTo(Ox1+m_nImageWidth,Oy1);
  708. pDC->LineTo(Ox1,Oy1);
  709. pDC->SetTextColor(0x000000FF);
  710. pDC->TextOut(Ox,Workarea.top+3,"原始数据图像二值化及投影图");
  711. OldPen=pDC->SelectObject(&PenBlue);
  712.     for(i=0;i<m_nImageHeight;i++)
  713.    for(j=0;j<m_nImageWidth;j++)
  714.    {  
  715.            temp=*((unsigned char *)lpDIBBits+m_nImageWidth*i+j);
  716.    if(temp<=m_nBinGate)
  717.    {
  718.   *((unsigned char *)lpDIBBits+m_nImageWidth*i+j)=0;   
  719.   pDC->MoveTo(Ox1+j,Oy1-i);
  720.   pDC->LineTo(Ox1+j,Oy1-i-2);
  721.   OldPen2=pDC->SelectObject(&PenWhite);
  722.   pDC->LineTo(Ox1+j,Oy1-i-1);
  723.   pDC->SelectObject(OldPen2);
  724.    }
  725.    else
  726.    {
  727.    *((unsigned char *)lpDIBBits+m_nImageWidth*i+j)=255;
  728.    }
  729.    }
  730. // 绘制垂直和水平方向的投影
  731.     for(i=0;i<m_nImageHeight;i++)
  732. {
  733.     temp=Data[0][i];
  734. pDC->MoveTo(Ox,Oy1-i);
  735. pDC->LineTo(Ox+temp,Oy1-i);
  736. }
  737. for(j=0;j<m_nImageWidth;j++)
  738. {
  739.     temp=Data[0][m_nImageHeight+j];
  740. pDC->MoveTo(Ox1+j,Oy);
  741. pDC->LineTo(Ox1+j,Oy-temp);
  742. }
  743.     pDC->SelectObject(OldPen);
  745. // 在对话框右下部分绘制小波变换后第m_nDisplayStage层数据图像
  746. int    DataLen=0;
  747. double XUnit=1,YUnit=1,tempX=0,tempY=0;
  748. int    temp1=0,temp2=0;
  749. double max=0,min=0,scale=0;
  750.     // 创建绘制区
  751. Workarea.left=RectClient.left+(RectClient.right-RectClient.left)/5*2;
  752.     Workarea.right=RectClient.right-10;
  753. Workarea.top=RectClient.top+10+(RectClient.bottom-RectClient.top-10)/2+40;
  754.     Workarea.bottom=RectClient.bottom-10; 
  755.     pDC->Rectangle(Workarea);
  756.     pDC->SetBkColor(0x00FFFFFF);
  757. // 绘制坐标轴
  758. Ox = Workarea.left+20;    Oy = Workarea.bottom-20;
  759. pDC->MoveTo(Ox,Oy);
  760. pDC->LineTo(Ox,Workarea.top+10);
  761. pDC->MoveTo(Ox,Oy);
  762. pDC->LineTo(Workarea.right-10,Oy);
  763. // 归一化小波变换后数据并绘制
  764. max=Data[m_nDisplayStage][0];
  765. min=Data[m_nDisplayStage][0];
  766. for(j=0;j<m_nImageWidth+m_nImageHeight;j++)
  767. {   
  768. if(max<Data[m_nDisplayStage][j])
  769.    max=Data[m_nDisplayStage][j];
  770. if(min>Data[m_nDisplayStage][j])
  771.        min=Data[m_nDisplayStage][j];
  772. }
  773. scale=max-min;
  774. XUnit=double(Workarea.right-Workarea.left-30)/double(m_nImageHeight+m_nImageWidth);
  775.     YUnit=double(Workarea.bottom-Workarea.top-30-10)/scale;
  776. if(min>0)
  777.    scale=0;
  778. else
  779.    scale=(0-min)*YUnit;
  780. pDC->MoveTo(Ox,Oy-scale);
  781. pDC->LineTo(Workarea.right-10,Oy-scale);
  782. pDC->SetTextColor(0x000000FF);
  783. pDC->TextOut(Ox-10,Oy-scale-6,"O");
  784. Text.Format("Max:%f  Min:%f",max,min);
  785. pDC->TextOut(Ox+10,Workarea.top+5,Text);
  786. pDC->SelectObject(&PenBlue);
  787. for(j=0;j<m_nImageWidth+m_nImageHeight-1;j++)
  788. {
  789.     tempX=j*XUnit;
  790. tempY=(Data[m_nDisplayStage][j]-min)*YUnit;
  791.         pDC->MoveTo(tempX+Ox,Oy-tempY);
  792. tempX=(j+1)*XUnit;
  793. tempY=(Data[m_nDisplayStage][j+1]-min)*YUnit;
  794. pDC->LineTo(tempX+Ox,Oy-tempY);
  795. }
  796.     pDC->SelectObject(OldPen);
  797. // 绘制说明文字如C0，D0等
  798. DataLen=m_nImageWidth+m_nImageHeight;
  799. for(j=0;j<=m_nDisplayStage;j++)
  800. {
  801.     temp1=temp2;
  802. temp2=DataLen>>(m_nDisplayStage-j);
  803. tempX=(temp2-1)*XUnit;
  804. tempY=(Data[m_nDisplayStage][temp2-1]-min)*YUnit;
  805. pDC->MoveTo(tempX+Ox,Oy-tempY);
  806.     pDC->LineTo(tempX+Ox,Oy);
  807. if(j==0)
  808. Text.Format("C%d",m_nDisplayStage);
  809. else
  810. Text.Format("D%d",m_nDisplayStage+1-j);
  811. tempX=((temp2-temp1-1)/2+temp1)*XUnit-5;
  812. pDC->SetTextColor(0x000000FF);
  813. pDC->TextOut(Ox+tempX,Oy+3,Text);
  814. }
  816. // 释放资源
  817. PenWhite.DeleteObject();
  818. PenGreen.DeleteObject();
  819. PenBlue.DeleteObject();
  820. Pen.DeleteObject();
  821.     // 解除锁定
  822. ::GlobalUnlock((HGLOBAL)GetHDIB());
  823. }
  825. // Hn系数文件产生函数，因为Hn文件只需产生一次，所以程序中并没有使用此函数。
  826. // 若要重新产生Hn文件，读者可以调用此函数。
  827. BOOL CWavelet1DDlg::GenerateHnFile()
  828. {
  829. double Hn[10][20] ={ {0,0},
  830. { 0.4829629131445341, 0.8365163037378077, 0.2241438680420134,
  831.                      -0.1294095225512603 },
  832. { 0.3326705529500825, 0.8068915093110924, 0.4598775021184914,
  833.  -0.1350110200102546,-0.0854412738820267, 0.0352262918857095 },
  834. { 0.2303778133088964, 0.7148465705529154, 0.6308807679398587,
  835.  -0.0279837694168599,-0.1870348117190931, 0.0308413818355607,
  836.   0.0328830116668852,-0.0105974017850890 },
  837. { 0.1601023979741929, 0.6038292697971895, 0.7243085284377726,
  838.   0.1384281459013203,-0.2422948870663823,-0.0322448695846381,
  839.   0.0775714938400459,-0.0062414902127983,-0.0125807519990820,
  840.   0.0033357252854738 },
  841. { 0.1115407433501095, 0.4946238903984533, 0.7511339080210959,
  842.   0.3152503517091982,-0.2262646939654400,-0.1297668675672625,
  843.   0.0975016055873225, 0.0275228655303053,-0.0315820393174862,
  844.   0.0005538422011614, 0.0047772575109455, 0.0010773010853085 },
  845. { 0.0778520540850037, 0.3965393194818912, 0.7291320908461957,
  846.   0.4697822874051889,-0.1439060039285212,-0.2240361549938412,
  847.   0.0713092192668272, 0.0806123091510774,-0.0380299369350104,
  848.  -0.0165745416306655, 0.0125509985560986, 0.0004295779729214,
  849.  -0.0018016407040473, 0.0003537137999745 },
  850. { 0.0544158422431072, 0.3128715909143166, 0.6756307362973195,
  851.   0.5852546836542159,-0.0158291052563823,-0.2840155429615824,
  852.   0.0004724845739124, 0.1287474266204893,-0.0173693010018090,
  853.  -0.0440882539307971, 0.0139810279174001, 0.0087160940474065,
  854.  -0.0048703529934520,-0.0003917403733770, 0.0006754494064506,
  855.  -0.0001174767841248 },
  856. { 0.0380779473638778, 0.2438346746125858, 0.6048231236900955,
  857.      0.6572880780512736, 0.1331973858249883,-0.2932737832791663,
  858.           -0.0968407832229492, 0.1485407493381256, 0.0307256814793385,
  859.  -0.0676328290613279, 0.0002509471148340, 0.0223616621236798,
  860.  -0.0047232047577518,-0.0042815036824635, 0.0018476468830563,
  861.   0.0002303857645232,-0.0002519631889427, 0.0000393473203163 },
  862. { 0.0266700579005473, 0.1881768000776347, 0.5272011889315757,
  863.   0.6884590394534363, 0.2811723436605715,-0.2498464243271598,
  864.  -0.1959462743772862, 0.1273693403357541, 0.0930573646035547,
  865.  -0.0713941471663501,-0.0294575368218399, 0.0332126740593612,
  866.   0.0036065535669870,-0.0107331754833007, 0.0013953517470688,
  867.   0.0019924052951925,-0.0006858566949564,-0.001164668551285,
  868.   0.0000935886703202,-0.0000132642028945 },
  869. };
  870. BOOL flag=TRUE;
  871. FILE *fp;
  872. int i,j;
  873. for(i=0;i<10;i++)
  874. {
  875.     if((fp=fopen(m_strFilterFile[i],"wb+"))==NULL)
  876. {
  877. AfxMessageBox("Hn算子文件生成失败!",MB_OK|MB_ICONINFORMATION);
  878.     flag=FALSE;
  879. break;
  880. }
  881.     for(j=0;j<2*(i+1);j++)
  882. fwrite(&Hn[i][j],sizeof(double),1,fp);
  883. fclose(fp);
  884. }
  885.     return flag;
  886. }