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

源码开发语言/平台
当前位置：首页 > 源码/资料 > 数值算法/人工智能 > 生物技术 > 查看源码
DlgCodingHuffman.cpp
资源名称：图像识别.rar　[点击查看]
上传用户：xjt2008yy
上传日期：2010-01-18
资源大小：272k
文件大小：5k
源码类别：

生物技术

开发平台：

Visual C++

DlgCodingHuffman.cpp：源码内容
  1. // DlgCodingHuffman.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "imageProcessing.h"
  6. #include "DlgCoding.h"
  7. #include <math.h>
  9. #ifdef _DEBUG
  10. #define new DEBUG_NEW
  11. #undef THIS_FILE
  12. static char THIS_FILE[] = __FILE__;
  13. #endif
  15. /////////////////////////////////////////////////////////////////////////////
  16. // CDlgHuffman dialog
  18. CDlgHuffman::CDlgHuffman(CWnd* pParent /*=NULL*/)
  19. : CDialog(CDlgHuffman::IDD, pParent)
  20. {
  21. //{{AFX_DATA_INIT(CDlgHuffman)
  22. m_dEntropy = 0.0;
  23. m_dCodLength = 0.0;
  24. m_dRatio = 0.0;
  25. //}}AFX_DATA_INIT
  26. }
  29. void CDlgHuffman::DoDataExchange(CDataExchange* pDX)
  30. {
  31. CDialog::DoDataExchange(pDX);
  32. //{{AFX_DATA_MAP(CDlgHuffman)
  33. DDX_Control(pDX, IDC_LST_Table, m_lstTable);
  34. DDX_Text(pDX, IDC_EDIT1, m_dEntropy);
  35. DDX_Text(pDX, IDC_EDIT2, m_dCodLength);
  36. DDX_Text(pDX, IDC_EDIT3, m_dRatio);
  37. //}}AFX_DATA_MAP
  38. }
  41. BEGIN_MESSAGE_MAP(CDlgHuffman, CDialog)
  42. //{{AFX_MSG_MAP(CDlgHuffman)
  43. //}}AFX_MSG_MAP
  44. END_MESSAGE_MAP()
  46. /////////////////////////////////////////////////////////////////////////////
  47. // CDlgHuffman message handlers
  49. BOOL CDlgHuffman::OnInitDialog() 
  50. {
  52. // 调用默认得OnInitDialog()函数
  53. CDialog::OnInitDialog();
  55. // 循环变量
  56. LONG i;
  57. LONG j;
  58. LONG k;
  60. // 中间变量
  61. double dT;
  63. // 字符串变量
  64. CString str2View;
  66. // 控件ListCtrl的ITEM
  67. LV_ITEM lvItem;
  69. // 保存控件ListCtrl中添加的ITEM编号
  70. int nItem2View;
  72. // 保存计算中间结果的数组
  73. double * dTemp;
  75. // 数组用来存放灰度值和其位置之间的映射
  76. int * n4Turn;
  78. // 初始化变量
  79. m_dEntropy = 0.0;
  80. m_dCodLength = 0.0;
  82. // 分配内存
  83. m_strCode = new CString[nColorNum];
  84. n4Turn  = new int[nColorNum];
  85. dTemp = new double[nColorNum];
  87. // 令dTemp赋值
  88. // 开始的灰度值按灰度值大小排列相同
  89. for (i = 0; i < nColorNum; i ++)
  90. {
  91. dTemp[i] = dProba[i];
  92. n4Turn[i] = i;
  93. }
  95. // 用冒泡法对进行灰度值出现的概率排序
  96. // 同时改变灰度值位置的映射关系
  97. for (j = 0; j < nColorNum - 1; j ++)
  98. {
  99. for (i = 0; i < nColorNum - j - 1; i ++)
  100. {
  101. if (dTemp[i] > dTemp[i + 1])
  102. {
  103. dT = dTemp[i];
  104. dTemp[i] = dTemp[i + 1];
  105. dTemp[i + 1] = dT;
  107. // 将i和i+1灰度的位置值互换
  108. for (k = 0; k < nColorNum; k ++)
  109. {
  110. if (n4Turn[k] == i)
  111. n4Turn[k] = i + 1;
  112. else if (n4Turn[k] == i + 1)
  113. n4Turn[k] = i;
  114. }
  115. }
  116. }
  117. }
  119. /*******************************************************
  120. 计算哈夫曼编码表
  121. *******************************************************/
  123. // 从概率大于0处开始编码
  124. for (i = 0; i < nColorNum - 1; i ++)
  125. {
  126. if (dTemp[i] > 0)
  127. break;
  128. }
  130. for (; i < nColorNum - 1; i ++)
  131. {
  132. // 更新m_strCode
  133. for (k = 0; k < nColorNum; k ++)
  134. {
  135. // 灰度值是否i
  136. if (n4Turn[k] == i)
  137. {
  138. // 灰度值较小的码字加1
  139. m_strCode[k] = "1" + m_strCode[k];
  140. }
  141. else if (n4Turn[k] == i + 1)
  142. {
  143. // 灰度值较小的码字加0
  144. m_strCode[k] = "0" + m_strCode[k];
  145. }
  146. }
  148. // 概率最小的两个概率相加，保存在dTemp[i + 1]中
  149. dTemp[i + 1] += dTemp[i];
  151. // 改变映射关系
  152. for (k = 0; k < nColorNum; k ++)
  153. {
  154. // 将位置为i的灰度值i改为灰度值i+1
  155. if (n4Turn[k] == i)
  156. n4Turn[k] = i + 1;
  157. }
  159. // 重新排序
  160. for (j = i + 1; j < nColorNum - 1; j ++)
  161. {
  162. if (dTemp[j] > dTemp[j + 1])
  163. {
  164. // 互换
  165. dT = dTemp[j];
  166. dTemp[j] = dTemp[j + 1];
  167. dTemp[j + 1] = dT;
  169. // // 将i和i+1灰度的位置值互换
  170. for (k = 0; k < nColorNum; k ++)
  171. {
  172. if (n4Turn[k] == j)
  173. n4Turn[k] = j + 1;
  174. else if (n4Turn[k] == j + 1)
  175. n4Turn[k] = j;
  176. }
  177. }
  178. else
  179. // 退出循环
  180. break;
  181. }
  182. }
  184. // 计算图像熵
  185. for (i = 0; i < nColorNum; i ++)
  186. {
  187. if (dProba[i] > 0)
  188. {
  189. // 计算图像熵
  190. m_dEntropy -= dProba[i] * log(dProba[i]) / log(2.0);
  191. }
  192. }
  194. // 计算平均码字长度
  195. for (i = 0; i < nColorNum; i ++)
  196. {
  197. // 累加
  198. m_dCodLength += dProba[i] * m_strCode[i].GetLength();
  199. }
  201. // 计算编码效率
  202. m_dRatio = m_dEntropy / m_dCodLength;
  204. // 保存变动
  205. UpdateData(FALSE);
  207. /*************************************************
  208. 输出计算结果
  209. *************************************************/
  211. // 设置CListCtrl控件样式
  212. m_lstTable.ModifyStyle(LVS_TYPEMASK, LVS_REPORT);
  214. // 给List控件添加Header
  215. m_lstTable.InsertColumn(0, "灰度值", LVCFMT_LEFT, 60, 0);
  216. m_lstTable.InsertColumn(1, "概率值", LVCFMT_LEFT, 78, 0);
  217. m_lstTable.InsertColumn(2, "哈夫曼编码", LVCFMT_LEFT, 110, 1);
  218. m_lstTable.InsertColumn(3, "码字长度", LVCFMT_LEFT, 78, 2);
  220. // 设置样式为文本
  221. lvItem.mask = LVIF_TEXT;
  223. // 添加显示
  224. for (i = 0; i < nColorNum; i ++)
  225. {
  226. // 第一列显示
  227. lvItem.iItem = m_lstTable.GetItemCount();
  228. str2View.Format("%u",i);
  229. lvItem.iSubItem = 0;
  230. lvItem.pszText= (LPTSTR)(LPCTSTR)str2View;
  231. nItem2View = m_lstTable.InsertItem(&lvItem);
  233. // 其它列显示
  234. lvItem.iItem = nItem2View;
  236. // 添加灰度值的概率值
  237. lvItem.iSubItem = 1;
  238. str2View.Format("%f",dProba[i]);
  239. lvItem.pszText = (LPTSTR)(LPCTSTR)str2View;
  240. m_lstTable.SetItem(&lvItem);
  242. // 添加哈夫曼编码
  243. lvItem.iSubItem = 2;
  244. lvItem.pszText = (LPTSTR)(LPCTSTR)m_strCode[i];
  245. m_lstTable.SetItem(&lvItem);
  247. // 添加码字长度
  248. lvItem.iSubItem = 3;
  249. str2View.Format("%u",m_strCode[i].GetLength());
  250. lvItem.pszText = (LPTSTR)(LPCTSTR)str2View;
  251. m_lstTable.SetItem(&lvItem);
  252. }
  254. //  内存释放
  255. delete n4Turn;
  256. delete dTemp;
  258. // 返回TRUE
  259. return TRUE;
  260. }