小波变换C语言实现代码

1、#include <stdio.h>#include<math.h>#define height 256#define width 256void twoDDWT(double orgwidth, double dwtwidth,int NumofBand);void oneDDWT(int flag, double orgwidth, double dwtwidth);void twoDIDWT(double dwtwidth, double recwidth,int NumofBand);void oneDIDWT(int flag, double dwtwidth

2、, double recwidth);int main()int i, j;double orgheightwidth, dwtheightwidth,recheightwidth;FILE *lena;FILE *lena3; FILE *lena4;lena = fopen("c:lena.raw", "rb");lena3 = fopen("c:lena3.raw", "wb");lena4 = fopen("c:lena4.raw", "wb");for (i = 0

3、; i < height; i+) for (j = 0; j < width; j+)orgij = fgetc(lena);twoDDWT(org, dwt,4);for (i = 0; i < height/2; i+)for (j = 0; j < width/2; j+)orgij=dwtij;twoDDWT(org, dwt,7);for (i = 0; i < height; i+)for (j = 0; j < width; j+)fputc(dwtij, lena3);twoDIDWT(dwt, rec,7);for (i = 0; i &

4、lt; height/2; i+)for (j = 0; j < width/2; j+)dwtij=recij;twoDIDWT(dwt, rec,4);for (i = 0; i < height; i+)for (j = 0; j < width; j+)fputc(recij, lena4);return 0;void twoDDWT(double orgwidth, double dwtwidth,int NumofBand)if (NumofBand=4)oneDDWT(0, org, dwt);oneDDWT(1, org, dwt);if (NumofBand

5、=7)oneDDWT(2, org, dwt);oneDDWT(3, org, dwt); void oneDDWT(int flag, double orgwidth, double dwtwidth)int i, j;double tempheightwidth, temp1heightwidth;double f_LPF = -0.125, 0.25, 0.75, 0.25, -0.125 ;double f_HPF = -0.5, 1, -0.5 ;if (flag = 0)/vertical direction for (i = 0; i < height; i+) for (

6、j = 0; j < width; j+)if (j = 0)tempij = f_LPF0 * orgij + 2 + f_LPF1 * orgij + 1 + f_LPF2 * orgij + f_LPF3 * orgij + 1 + f_LPF4 * orgij + 2;else if (j = 1)tempij = f_LPF0 * orgij + f_LPF1 * orgij - 1 + f_LPF2 * orgij + f_LPF3 * orgij + 1 + f_LPF4 * orgij + 2;else if (j > 1 && j < hei

7、ght-2)tempij = f_LPF0 * orgij - 2 + f_LPF1 * orgij - 1 + f_LPF2 * orgij + f_LPF3 * orgij + 1 + f_LPF4 * orgij + 2;else if (j = height-2)tempij = f_LPF0 * orgij - 2 + f_LPF1 * orgij - 1 + f_LPF2 * orgij + f_LPF3 * orgij + 1 + f_LPF4 * orgij;else if (j = height-1)tempij = f_LPF0 * orgij - 2 + f_LPF1 *

8、 orgij - 1 + f_LPF2 * orgij + f_LPF3 * orgij - 1 + f_LPF4 * orgij - 2;for (i = 0; i < height; i+)for (j = 0; j < width/2; j+)dwtij = tempij * 2 + 1;for (i = 0; i < height; i+)for (j = 0; j < width; j+)if (j = 0)tempij = f_HPF0 * orgij + 1 + f_HPF1 * orgij + f_HPF2 * orgij + 1;else if (j

9、> 0 && j < height-1)tempij = f_HPF0 * orgij - 1 + f_HPF1 * orgij + f_HPF2 * orgij + 1;else if (j = height-1)tempij = f_HPF0 * orgij - 1 + f_HPF1 * orgij + f_HPF2 * orgij - 1;for (i = 0; i < height; i+)for (j = 0; j < width/2; j+)dwtij + width/2 = tempij * 2;if (flag = 1)/horizont

10、al directionfor (i = 0; i < height; i+)for (j = 0; j < width/2; j+)if (i = 0)tempij = f_LPF0 * dwti + 2j + f_LPF1 * dwti + 1j + f_LPF2 * dwtij + f_LPF3 * dwti + 1j + f_LPF4 * dwti + 2j;else if (i = 1)tempij = f_LPF0 * dwtij + f_LPF1 * dwti - 1j + f_LPF2 * dwtij + f_LPF3 * dwti + 1j + f_LPF4 *

11、dwti + 2j;else if (i > 1 && i < width-2)tempij = f_LPF0 * dwti - 2j + f_LPF1 * dwti - 1j + f_LPF2 * dwtij + f_LPF3 * dwti + 1j + f_LPF4 * dwti + 2j;else if (i = width-2)tempij = f_LPF0 * dwti - 2j + f_LPF1 * dwti - 1j + f_LPF2 * dwtij + f_LPF3 * dwti + 1j + f_LPF4 * dwtij;else if (i =

12、width-1)tempij = f_LPF0 * dwti - 2j + f_LPF1 * dwti - 1j + f_LPF2 * dwtij + f_LPF3 * dwti - 1j + f_LPF4 * dwti - 2j;for (i = 0; i < height; i+)for (j = 0; j < width; j+)temp1ij = tempi * 2 + 1j;for (i = 0; i < height; i+)for (j = 0; j < width/2; j+)if (i = 0)tempij = f_HPF0 * dwti + 1j +

13、 f_HPF1 * dwtij + f_HPF2 * dwti + 1j;else if (i > 0 && i < width-1)tempij = f_HPF0 * dwti - 1j + f_HPF1 * dwtij + f_HPF2 * dwti + 1j;else if (i = width-1)tempij = f_HPF0 *dwti - 1j + f_HPF1 * dwtij + f_HPF2 * dwti - 1j;for (i = 0; i < height; i+) for (j = 0; j < width/2; j+)temp1

14、ij+width/2 = tempi * 2j;for (i = 0; i < height; i+)for (j = width/2; j < width; j+)if (i = 0)tempij = f_LPF0 * dwti + 2j + f_LPF1 * dwti + 1j + f_LPF2 * dwtij + f_LPF3 * dwti + 1j + f_LPF4 * dwti + 2j;else if (i = 1)tempij = f_LPF0 * dwtij + f_LPF1 * dwti - 1j + f_LPF2 * dwtij + f_LPF3 * dwti

15、+ 1j + f_LPF4 * dwti + 2j;else if (i > 1 && i < height-2)tempij = f_LPF0 * dwti - 2j + f_LPF1 * dwti - 1j + f_LPF2 * dwtij + f_LPF3 * dwti + 1j + f_LPF4 * dwti + 2j;else if (i = height-2)tempij = f_LPF0 * dwti - 2j + f_LPF1 * dwti - 1j + f_LPF2 * dwtij + f_LPF3 * dwti + 1j + f_LPF4 * d

16、wtij;else if (i = height-1)tempij = f_LPF0 * dwti - 2j + f_LPF1 * dwti - 1j + f_LPF2 * dwtij + f_LPF3 * dwti - 1j + f_LPF4 * dwti - 2j;for (i = 0; i < height; i+)for (j = width/2; j < width; j+)temp1i+height/2j-width/2 = tempi * 2 + 1j;for (i = 0; i < height; i+)for (j = width/2; j < wid

17、th; j+)if (i = 0)tempij = f_HPF0 * dwti + 1j + f_HPF1 * dwtij + f_HPF2 * dwti + 1j;else if (i > 0 && i < height-1)tempij = f_HPF0 * dwti - 1j + f_HPF1 * dwtij + f_HPF2 * dwti + 1j;else if (i = height-1)tempij = f_HPF0 * dwti - 1j + f_HPF1 * dwtij + f_HPF2 * dwti - 1j;for (i = 0; i <

18、 height; i+)for (j = width/2; j < width; j+)temp1i +height/2j = tempi * 2j;for (i = 0; i < height; i+)for (j = 0; j < width; j+)dwtij = temp1ij;if (flag = 2)/vertical directionfor (i = 0; i < height/2; i+)for (j = 0; j < width; j+)if (j = 0)tempij = f_LPF0 * orgij + 2 + f_LPF1 * orgij

19、 + 1 + f_LPF2 * orgij + f_LPF3 * orgij + 1 + f_LPF4 * orgij + 2;else if (j = 1)tempij = f_LPF0 * orgij + f_LPF1 * orgij - 1 + f_LPF2 * orgij + f_LPF3 * orgij + 1 + f_LPF4 * orgij + 2;else if (j > 1 && j < height/2-2)tempij = f_LPF0 * orgij - 2 + f_LPF1 * orgij - 1 + f_LPF2 * orgij + f_

20、LPF3 * orgij + 1 + f_LPF4 * orgij + 2;else if (j = height/2-2)tempij = f_LPF0 * orgij - 2 + f_LPF1 * orgij - 1 + f_LPF2 * orgij + f_LPF3 * orgij + 1 + f_LPF4 * orgij;else if (j = height/2-1)tempij = f_LPF0 * orgij - 2 + f_LPF1 * orgij - 1 + f_LPF2 * orgij + f_LPF3 * orgij - 1 + f_LPF4 * orgij - 2;fo

21、r (i = 0; i < height/2; i+)for (j = 0; j < width/4; j+)dwtij = tempij * 2 + 1;for (i = 0; i < height/2; i+)for (j = 0; j < width/2; j+)if (j = 0)tempij = f_HPF0 * orgij + 1 + f_HPF1 * orgij + f_HPF2 * orgij + 1;else if (j > 0 && j < height-1)tempij = f_HPF0 * orgij - 1 + f_

22、HPF1 * orgij + f_HPF2 * orgij + 1;else if (j = height-1)tempij = f_HPF0 * orgij - 1 + f_HPF1 * orgij + f_HPF2 * orgij - 1;for (i = 0; i < height/2; i+)for (j = 0; j < width/4; j+)dwtij + width/4 = tempij * 2;if (flag = 3)/horizontal directionfor (i = 0; i < height/2; i+)for (j = 0; j < w

23、idth/4; j+)if (i = 0)tempij = f_LPF0 * dwti + 2j + f_LPF1 * dwti + 1j + f_LPF2 * dwtij + f_LPF3 * dwti + 1j + f_LPF4 * dwti + 2j;else if (i = 1)tempij = f_LPF0 * dwtij + f_LPF1 * dwti - 1j + f_LPF2 * dwtij + f_LPF3 * dwti + 1j + f_LPF4 * dwti + 2j;else if (i > 1 && i < height/2-2)tempi

24、j = f_LPF0 * dwti - 2j + f_LPF1 * dwti - 1j + f_LPF2 * dwtij + f_LPF3 * dwti + 1j + f_LPF4 * dwti + 2j;else if (i = height/2-2)tempij = f_LPF0 * dwti - 2j + f_LPF1 * dwti - 1j + f_LPF2 * dwtij + f_LPF3 * dwti + 1j + f_LPF4 * dwtij;else if (i = height/2-1)tempij = f_LPF0 * dwti - 2j + f_LPF1 * dwti -

25、 1j + f_LPF2 * dwtij + f_LPF3 * dwti - 1j + f_LPF4 * dwti - 2j;for (i = 0; i < height/4; i+)for (j = 0; j < width/4; j+)temp1ij = tempi * 2 + 1j;for (i = 0; i < height/2; i+)for (j = 0; j < width/4; j+)if (i = 0)tempij = f_HPF0 * dwti + 1j + f_HPF1 * dwtij + f_HPF2 * dwti + 1j;else if (i

26、 > 0 && i < height/2-1)tempij = f_HPF0 * dwti - 1j + f_HPF1 * dwtij + f_HPF2 * dwti + 1j;else if (i = height/2-1)tempij = f_HPF0 * dwti - 1j + f_HPF1 * dwtij + f_HPF2 * dwti - 1j;for (i = 0; i < height/4; i+)for (j = 0; j < width/4; j+)temp1i + height/4j = tempi * 2j;for (i = 0;

27、i < height/2; i+)for (j = width/4; j < width/2; j+)if (i = 0)tempij = f_LPF0 * dwti + 2j + f_LPF1 * dwti + 1j + f_LPF2 * dwtij + f_LPF3 * dwti + 1j + f_LPF4 * dwti + 2j;else if (i = 1)tempij = f_LPF0 * dwtij + f_LPF1 * dwti - 1j + f_LPF2 * dwtij + f_LPF3 * dwti + 1j + f_LPF4 * dwti + 2j;else i

28、f (i > 1 && i < height/2-2)tempij = f_LPF0 * dwti - 2j + f_LPF1 * dwti - 1j + f_LPF2 * dwtij + f_LPF3 * dwti + 1j + f_LPF4 * dwti + 2j;else if (i = height/2-2)tempij = f_LPF0 * dwti - 2j + f_LPF1 * dwti - 1j + f_LPF2 * dwtij + f_LPF3 * dwti + 1j + f_LPF4 * dwtij;else if (i = height/2-1

29、)tempij = f_LPF0 * dwti - 2j + f_LPF1 * dwti - 1j + f_LPF2 * dwtij + f_LPF3 * dwti - 1j + f_LPF4 * dwti - 2j;for (i = 0; i < height/4; i+)for (j = width/4; j < width/2; j+)temp1ij = tempi * 2 + 1j;for (i = 0; i < height/2; i+)for (j = width/4; j < width/2; j+)if (i = 0)tempij = f_HPF0 *

30、dwti + 1j + f_HPF1 * dwtij + f_HPF2 * dwti + 1j;else if (i > 0 && i < width/2-1)tempij = f_HPF0 * dwti - 1j + f_HPF1 * dwtij + f_HPF2 * dwti + 1j;else if (i = width/2)tempij = f_HPF0 * dwti - 1j + f_HPF1 * dwtij + f_HPF2 * dwti - 1j;for (i = 0; i < height/4; i+)for (j = width/4; j &

31、lt; width/2; j+)temp1i + height/4j = tempi * 2j; for (i = 0; i < height/2; i+)for (j = 0; j < width/2; j+)dwtij = temp1ij;void twoDIDWT(double dwtwidth, double recwidth,int NumofBand)if (NumofBand=7)oneDIDWT(2, dwt, rec);oneDIDWT(3, dwt, rec);if (NumofBand=4)oneDIDWT(1, dwt, rec);oneDIDWT(0, d

32、wt, rec);void oneDIDWT(int flag, double dwtwidth, double recwidth)int i, j;double tempheightwidth, temp1heightwidth;double i_LPF = 0.5, 1, 0.5 ;double i_HPF = -0.125, -0.25, 0.75, -0.25, -0.125 ;if (flag = 2)/horizontal directionfor (i = 0; i < height/2; i+)for (j = 0; j < width/4; j+)if (i %

33、2 = 1)tempij = dwti / 2j;elsetempij = 0;for (i = 0; i < height/2; i+)for (j = 0; j < width/4; j+)if (i = 0)recij = i_LPF0 * tempi + 1j + i_LPF1 * tempij + i_LPF2 * tempi + 1j;else if (i > 0 && i < height/2-1)recij = i_LPF0 * tempi - 1j + i_LPF1 * tempij + i_LPF2 * tempi + 1j;else

34、 if (i = height/2-1)recij = i_LPF0 * tempi - 1j + i_LPF1 * tempij + i_LPF2 * tempi - 1j;for (i = height/4; i < height/2; i+)for (j = 0; j < width/4; j+)dwti - height/4j = dwtij;for (i = 0; i < height/2; i+)for (j = 0; j < width/4; j+)if (i % 2 = 0)tempij = dwti / 2j;elsetempij = 0;for (i

35、 = 0; i < height/2; i+)for (j = 0; j < width/4; j+)if (i = 0)temp1ij = i_HPF0 * tempi + 2j + i_HPF1 * tempi + 1j + i_HPF2 * tempij + i_HPF3 * tempi + 1j + i_HPF4 * tempi + 2j;else if (i = 1)temp1ij = i_HPF0 * tempij + i_HPF1 * tempi - 1j + i_HPF2 * tempij + i_HPF3 * tempi + 1j + i_HPF4 * tempi

36、 + 2j;else if (i > 1 && i < height/2-2)temp1ij = i_HPF0 * tempi - 2j + i_HPF1 * tempi - 1j + i_HPF2 * tempij + i_HPF3 * tempi + 1j + i_HPF4 * tempi + 2j;else if (i = height/2-2)temp1ij = i_HPF0 * tempi - 2j + i_HPF1 * tempi - 1j + i_HPF2 * tempij + i_HPF3 * tempi + 1j + i_HPF4 * tempij

37、;else if (i = height/2-1)temp1ij = i_HPF0 * tempi - 2j + i_HPF1 * tempi - 1j + i_HPF2 * tempij + i_HPF3 * tempi - 1j + i_HPF4 * tempi - 2j;for (i = 0; i < height/2; i+)for (j = 0; j < width/4; j+)recij = recij + temp1ij;for (i = 0; i < height/4; i+)for (j = width/4; j < width/2; j+)dwtij

38、 - height/4 = dwtij;for (i = 0; i < height/2; i+)for (j = 0; j < width/4; j+)if (i % 2 = 1)tempij = dwti / 2j;elsetempij = 0;for (i = 0; i < height/2; i+)for (j = 0; j < width/4; j+)if (i = 0)temp1ij = i_LPF0 * tempi + 1j + i_LPF1 * tempij + i_LPF2 * tempi + 1j;else if (i > 0 &&am

39、p; i < height/2-1)temp1ij = i_LPF0 * tempi - 1j + i_LPF1 * tempij + i_LPF2 * tempi + 1j;else if (i = height/2-1)temp1ij = i_LPF0 * tempi - 1j + i_LPF1 * tempij + i_LPF2 * tempi - 1j;for (i = height/4; i < height/2; i+)for (j = width/4; j < width/2; j+)dwti - height/4j - width/4 = dwtij;for

40、(i = 0; i < height/2; i+)for (j = 0; j < width/4; j+)if (i % 2 = 0)tempij = dwti / 2j;elsetempij = 0;for (i = 0; i < height/2; i+)for (j = 0; j < width/4; j+)if (i = 0)dwtij = i_HPF0 * tempi + 2j + i_HPF1 * tempi + 1j + i_HPF2 * tempij + i_HPF3 * tempi + 1j + i_HPF4 * tempi + 2j;else if

41、(i = 1)dwtij = i_HPF0 * tempij + i_HPF1 * tempi - 1j + i_HPF2 * tempij + i_HPF3 * tempi + 1j + i_HPF4 * tempi + 2j;else if (i > 1 && i < height/2-2)dwtij = i_HPF0 * tempi - 2j + i_HPF1 * tempi - 1j + i_HPF2 * tempij + i_HPF3 * tempi + 1j + i_HPF4 * tempi + 2j;else if (i = height/2-2)dw

42、tij = i_HPF0 * tempi - 2j + i_HPF1 * tempi - 1j + i_HPF2 * tempij + i_HPF3 * tempi + 1j + i_HPF4 * tempij;else if (i = height/2-1)dwtij = i_HPF0 * tempi - 2j + i_HPF1 * tempi - 1j + i_HPF2 * tempij + i_HPF3 * tempi - 1j + i_HPF4 * tempi - 2j;for (i = 0; i < height/2; i+)for (j = 0; j < width/4

43、; j+)recij + width/4 = temp1ij + dwtij;if (flag = 3)/vertical directionfor (i = 0; i < height/2; i+)for (j = 0; j < width/2; j+)if (j % 2 = 1)tempij = recij / 2;elsetempij = 0;for (i = 0; i < height/2; i+)for (j = 0; j < width/2; j+)if (j = 0)temp1ij = i_LPF0 * tempij + 1 + i_LPF1 * tempij + i_LPF2 * tempij + 1;else if (j > 0 && j < height/2-1)temp1ij = i_LPF0 * tempij - 1 + i_LPF1 * tempij + i_LPF2 * tempij + 1;else if (j = height/2-1)temp1ij = i_LPF0 * tempij - 1 + i_LPF1 * tempij + i_LPF2 * tempij - 1;for (i = 0; i <