图像的正交变换---离散余弦变换

  
  
  1. /*************************************************************************
  2. *
  3. * 函数名称:
  4. * DCT()
  5. *
  6. * 参数:
  7. * double * f - 指向时域值的指针
  8. * double * F - 指向频域值的指针
  9. * r -2的幂数
  10. *
  11. * 返回值:
  12. * 无。
  13. *
  14. * 说明:
  15. * 该函数用来实现一维快速离散余弦变换
  16. *
  17. ************************************************************************/
  18. void DCT(double*f,double*F,int r);
  19. /*************************************************************************
  20. *
  21. * 函数名称:
  22. * IDCT()
  23. *
  24. * 参数:
  25. * double * F - 指向频域值的指针
  26. * double * f - 指向时域值的指针
  27. * r -2的幂数
  28. *
  29. * 返回值:
  30. * 无。
  31. *
  32. * 说明:
  33. * 该函数实现一维快速离散余弦逆变换
  34. *
  35. ************************************************************************/
  36. void IDCT(double*F,double*f,int r);
  37. /*************************************************************************
  38. *
  39. * 函数名称:
  40. * FreqDCT()
  41. *
  42. * 参数:
  43. * double* f - 输入的时域序列
  44. * double* F - 输出的频域序列
  45. * LONG width - 图象宽度
  46. * LONG height - 图象高度
  47. *
  48. * 返回值:
  49. * BOOL - 成功返回TRUE,否则返回FALSE。
  50. *
  51. * 说明:
  52. * 该函数进行二维快速离散余弦变换。
  53. *
  54. ************************************************************************/
  55. BOOL FreqDCT(double*f,double*F, LONG width, LONG height);
  56. /*************************************************************************
  57. *
  58. * 函数名称:
  59. * IFreqDCT()
  60. *
  61. * 参数:
  62. * double* f - 输入的时域序列
  63. * double* F - 输出的频域序列
  64. * LONG width - 图象宽度
  65. * LONG height - 图象高度
  66. *
  67. * 返回值:
  68. * BOOL - 成功返回TRUE,否则返回FALSE。
  69. *
  70. * 说明:
  71. * 该函数进行二维快速离散余弦逆变换。
  72. *
  73. ************************************************************************/
  74. BOOL IFreqDCT(double*f,double*F, LONG lWidth, LONG lHeight);
  75. /*************************************************************************
  76. *
  77. * 函数名称:
  78. * DIBDCT()
  79. *
  80. * 参数:
  81. * BYTE* bmp,LONG width,LONG height
  82. *
  83. * 返回值:
  84. * BOOL - 成功返回TRUE,否则返回FALSE。
  85. *
  86. * 说明:
  87. * 该函数用来对图像进行离散余弦变换。
  88. *
  89. ************************************************************************/
  90. BOOLBmpDCT(BYTE* bmp,LONG width,LONG height);

   
   
  1. voidMyProcess::DCT(double*f,double*F,int r)
  2. {
  3. // 循环变量
  4. int i;
  5. // 中间变量
  6. double dTemp;
  7. // 计算离散余弦变换点数
  8. LONG N =1<<r;
  9. // 申请并分配内存
  10. complex<double>*XIn;
  11. complex<double>*XOut;
  12. XIn=new complex<double>[N *2];
  13. XOut=new complex<double>[N *2];
  14. // 赋初值为0
  15. memset(XIn,0,sizeof(complex<double>)* N *2);
  16. memset(XOut,0,sizeof(complex<double>)* N *2);
  17. // 将时域点写入数组X
  18. for(i =0; i < N; i++)
  19. XIn[i]= complex<double>(*(f + i),0);
  20. // 调用快速付立叶变换
  21. FFT(XIn,XOut, r +1);
  22. // 调整系数
  23. dTemp =1/ sqrt(N);
  24. // 求F[0]
  25. F[0]=XOut[0].real()* dTemp;
  26. dTemp *= sqrt(2);
  27. // 求F[u]
  28. for(i =1; i < N; i++)
  29. *(F + i)=(XOut[i].real()* cos(i * PI /(N *2))+XOut[i].imag()* sin(i * PI /(N *2)))* dTemp;
  30. // 释放内存
  31. delete[]XIn;
  32. delete[]XOut;
  33. }
  34. voidMyProcess::IDCT(double*F,double*f,int r)
  35. {
  36. // 循环变量
  37. int i;
  38. // 中间变量
  39. double dTemp, d0;
  40. // 计算离散余弦变换点数
  41. LONG N =1<<r;
  42. // 分配内存
  43. complex<double>*XIn;
  44. complex<double>*XOut;
  45. XIn=new complex<double>[N *2];
  46. XOut=new complex<double>[N *2];
  47. // 赋初值为0
  48. memset(XIn,0,sizeof(complex<double>)* N *2);
  49. memset(XOut,0,sizeof(complex<double>)* N *2);
  50. // 将频域变换后点写入数组X
  51. for(i =0; i < N; i++)
  52. XIn[i]= complex<double>(F[i]* cos(i * PI /(N *2)), F[i]* sin(i * PI /(N *2)));
  53. // 调用快速付立叶反变换
  54. IFFT(XIn,XOut, r +1);
  55. // 调整系数
  56. dTemp = sqrt(2.0/ N);
  57. d0 =(sqrt(1.0/ N)- dTemp)* F[0];
  58. // 计算f(x)
  59. for(i =0; i < N; i++)
  60. f[i]= d0 +XOut[i].real()* dTemp *2* N;
  61. // 释放内存
  62. delete[]XIn;
  63. delete[]XOut;
  64. }
  65. BOOL MyProcess::FreqDCT(double*f,double*F, LONG width, LONG height)
  66. {
  67. // 循环变量
  68. LONG i;
  69. LONG j;
  70. LONG k;
  71. // 进行离散余弦变换的宽度和高度(2的整数次方)
  72. LONG w =1;
  73. LONG h =1;
  74. int wp =0;
  75. int hp =0;
  76. // 计算进行离散余弦变换的宽度和高度(2的整数次方)
  77. while(w < width/3)
  78. {
  79. w *=2;
  80. wp++;
  81. }
  82. while(h < height)
  83. {
  84. h *=2;
  85. hp++;
  86. }
  87. // 分配内存
  88. double*TempIn=newdouble[h];
  89. double*TempOut=newdouble[h];
  90. // 对y方向进行离散余弦变换
  91. for(i =0; i < w *3; i++)
  92. {
  93. // 抽取数据
  94. for(j =0; j < h; j++)
  95. TempIn[j]= f[j * w *3+ i];
  96. // 一维快速离散余弦变换
  97. DCT(TempIn,TempOut, hp);
  98. // 保存变换结果
  99. for(j =0; j < h; j++)
  100. f[j * w *3+ i]=TempOut[j];
  101. }
  102. // 释放内存
  103. deleteTempIn;
  104. deleteTempOut;
  105. // 分配内存
  106. TempIn=newdouble[w];
  107. TempOut=newdouble[w];
  108. // 对x方向进行快速离散余弦变换
  109. for(i =0; i < h; i++)
  110. {
  111. for(k =0; k <3; k++)
  112. {
  113. // 抽取数据
  114. for(j =0; j < w; j++)
  115. TempIn[j]= f[i * w *3+ j *3+ k];
  116. // 一维快速离散余弦变换
  117. DCT(TempIn,TempOut, wp);
  118. // 保存变换结果
  119. for(j =0; j < w; j++)
  120. F[i * w *3+ j *3+ k]=TempOut[j];
  121. }
  122. }
  123. // 释放内存
  124. deleteTempIn;
  125. deleteTempOut;
  126. return TRUE;
  127. }
  128. BOOL MyProcess::IFreqDCT(double*f,double*F, LONG width, LONG height)
  129. {
  130. // 循环变量
  131. LONG i;
  132. LONG j;
  133. LONG k;
  134. // 进行离散余弦变换的宽度和高度(2的整数次方)
  135. LONG w =1;
  136. LONG h =1;
  137. int wp =0;
  138. int hp =0;
  139. // 计算进行付立叶变换的宽度和高度(2的整数次方)
  140. while(w < width/3)
  141. {
  142. w *=2;
  143. wp++;
  144. }
  145. while(h < height)
  146. {
  147. h *=2;
  148. hp++;
  149. }
  150. // 分配内存
  151. double*TempIn=newdouble[w];
  152. double*TempOut=newdouble[w];
  153. // 对x方向进行快速付立叶变换
  154. for(i =0; i < h; i++)
  155. {
  156. for(k =0; k <3; k++)
  157. {
  158. // 抽取数据
  159. for(j =0; j < w; j++)
  160. TempIn[j]= F[i * w *3+ j *3+ k];
  161. // 一维快速傅立叶变换
  162. IDCT(TempIn,TempOut, wp);
  163. // 保存变换结果
  164. for(j =0; j < w; j++)
  165. F[i * w *3+ j *3+ k]=TempOut[j];
  166. }
  167. }
  168. // 释放内存
  169. deleteTempIn;
  170. deleteTempOut;
  171. TempIn=newdouble[h];
  172. TempOut=newdouble[h];
  173. // 对y方向进行快速付立叶变换
  174. for(i =0; i < w *3; i++)
  175. {
  176. // 抽取数据
  177. for(j =0; j < h; j++)
  178. TempIn[j]= F[j * w *3+ i];
  179. // 一维快速傅立叶变换
  180. IDCT(TempIn,TempOut, hp);
  181. // 保存变换结果
  182. for(j =0; j < h; j++)
  183. F[j * w *3+ i]=TempOut[j];
  184. }
  185. // 释放内存
  186. deleteTempIn;
  187. deleteTempOut;
  188. for(i =0; i < h; i++)
  189. {
  190. for(j =0; j < w *3; j++)
  191. {
  192. if(i < height && j < width)
  193. *(f + i * width + j)= F[i * w *3+ j];
  194. }
  195. }
  196. return TRUE;
  197. }
  198. BOOL MyProcess::BmpDCT(BYTE* bmp,LONG width,LONG height)
  199. {
  200. // 进行离散余弦变换的宽度和高度(2的整数次方)
  201. LONG i;
  202. LONG j;
  203. // 进行离散余弦变换的宽度和高度(2的整数次方)
  204. LONG w =1;
  205. LONG h =1;
  206. int wp =0;
  207. int hp =0;
  208. // 计算进行离散余弦变换的宽度和高度(2的整数次方)
  209. while(w < width/3)
  210. {
  211. w *=2;
  212. wp++;
  213. }
  214. while(h < height)
  215. {
  216. h *=2;
  217. hp++;
  218. }
  219. // 分配内存
  220. double*f =newdouble[w * h *3];
  221. double*F =newdouble[w * h *3];
  222. // 向时域赋值并补零
  223. for(i =0; i < h; i++)
  224. {
  225. for(j =0; j < w *3; j++)
  226. {
  227. if(i < height && j < width)
  228. f[i * w *3+ j]= bmp[width * i + j];
  229. else
  230. f[w * i *3+ j]=0.0f;
  231. }
  232. }
  233. // 进行频谱分析
  234. if(FreqDCT(f, F,width, height)== FALSE)
  235. return FALSE;
  236. // 更新所有象素
  237. for(i =0; i < height; i++)
  238. {
  239. for(j =0; j < width; j++)
  240. {
  241. // 判断是否超过255
  242. if(fabs(F[i * w *3+ j])>255)
  243. {
  244. // 对于超过的,直接设置为255
  245. bmp[width *(height -1- i)+ j]=255;
  246. }
  247. else
  248. {
  249. // 如果没有超过,则按实际计算结果赋值
  250. bmp[width *(height -1- i)+ j]= fabs(F[i * w *3+ j]);
  251. }
  252. }
  253. }
  254. // 释放内存
  255. delete[]f;
  256. delete[] F;
  257. // 返回
  258. return TRUE;
  259. }

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值