mdwt_r.c
上传用户:speoil
上传日期:2022-06-23
资源大小:224k
文件大小:6k
源码类别:

波变换

开发平台:

Matlab

  1. /*
  2. File Name: MDWT.c
  3. Last Modification Date: 06/14/95 13:15:44
  4. Current Version: MDWT.c 2.4
  5. File Creation Date: Wed Oct 19 10:51:58 1994
  6. Author: Markus Lang  <lang@jazz.rice.edu>
  7. Copyright (c) 2000 RICE UNIVERSITY. All rights reserved.
  8. Created by Markus Lang, Department of ECE, Rice University. 
  9. This software is distributed and licensed to you on a non-exclusive 
  10. basis, free-of-charge. Redistribution and use in source and binary forms, 
  11. with or without modification, are permitted provided that the following 
  12. conditions are met:
  13. 1. Redistribution of source code must retain the above copyright notice, 
  14.    this list of conditions and the following disclaimer.
  15. 2. Redistribution in binary form must reproduce the above copyright notice, 
  16.    this list of conditions and the following disclaimer in the 
  17.    documentation and/or other materials provided with the distribution.
  18. 3. All advertising materials mentioning features or use of this software 
  19.    must display the following acknowledgment: This product includes 
  20.    software developed by Rice University, Houston, Texas and its contributors.
  21. 4. Neither the name of the University nor the names of its contributors 
  22.    may be used to endorse or promote products derived from this software 
  23.    without specific prior written permission.
  24. THIS SOFTWARE IS PROVIDED BY WILLIAM MARSH RICE UNIVERSITY, HOUSTON, TEXAS, 
  25. AND CONTRIBUTORS AS IS AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, 
  26. BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS 
  27. FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL RICE UNIVERSITY 
  28. OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 
  29. EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 
  30. PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; 
  31. OR BUSINESS INTERRUPTIONS) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, 
  32. WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR 
  33. OTHERWISE), PRODUCT LIABILITY, OR OTHERWISE ARISING IN ANY WAY OUT OF THE 
  34. USE OF THIS SOFTWARE,  EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  35. For information on commercial licenses, contact Rice University's Office of 
  36. Technology Transfer at techtran@rice.edu or (713) 348-6173
  37. Change History: Fixed the code such that 1D vectors passed to it can be in
  38.                 either passed as a row or column vector. Also took care of 
  39. the code such that it will compile with both under standard
  40. C compilers as well as for ANSI C compilers
  41. Jan Erik Odegard <odegard@ece.rice.edu> Wed Jun 14 1995
  42. %y = mdwt(x,h,L);
  43. % function computes the discrete wavelet transform y for a 1D or 2D input
  44. % signal x.
  45. %
  46. %    Input:
  47. % x    : finite length 1D or 2D signal (implicitely periodized)
  48. %       h    : scaling filter
  49. %       L    : number of levels. in case of a 1D signal length(x) must be
  50. %              divisible by 2^L; in case of a 2D signal the row and the
  51. %              column dimension must be divisible by 2^L.
  52. %
  53. % see also: midwt, mrdwt, mirdwt
  54. */
  55. #include <math.h>
  56. #include <stdio.h>
  57. #define max(A,B) (A > B ? A : B)
  58. #define mat(a, i, j) (*(a + (m*(j)+i)))  /* macro for matrix indices */
  59. #ifdef __STDC__
  60. MDWT(double *x, int m, int n, double *h, int lh, int L, double *y)
  61. #else
  62. MDWT(x, m, n, h, lh, L, y)
  63. double *x, *h, *y;
  64. int m, n, lh, L;
  65. #endif
  66. {
  67.   double  *h0, *h1, *ydummyl, *ydummyh, *xdummy;
  68.   long i, j;
  69.   int actual_L, actual_m, actual_n, r_o_a, c_o_a, ir, ic, lhm1;
  70.   xdummy = (double *)mxCalloc(max(m,n)+lh-1,sizeof(double));
  71.   ydummyl = (double *)mxCalloc(max(m,n),sizeof(double));
  72.   ydummyh = (double *)mxCalloc(max(m,n),sizeof(double));
  73.   h0 = (double *)mxCalloc(lh,sizeof(double));
  74.   h1 = (double *)mxCalloc(lh,sizeof(double));
  75.   
  76.   
  77.   /* analysis lowpass and highpass */
  78.   if (n==1){
  79.     n = m;
  80.     m = 1;
  81.   }
  82.   for (i=0; i<lh; i++){
  83.     h0[i] = h[lh-i-1];
  84.     h1[i] =h[i];
  85.   }
  86.   for (i=0; i<lh; i+=2)
  87.     h1[i] = -h1[i];
  88.   
  89.   lhm1 = lh - 1;
  90.   actual_m = 2*m;
  91.   actual_n = 2*n;
  92.   
  93.   /* main loop */
  94.   for (actual_L=1; actual_L <= L; actual_L++){
  95.     if (m==1)
  96.       actual_m = 1;
  97.     else{
  98.       actual_m = actual_m/2;
  99.       r_o_a = actual_m/2;     
  100.     }
  101.     actual_n = actual_n/2;
  102.     c_o_a = actual_n/2;
  103.     
  104.     /* go by rows */
  105.     for (ir=0; ir<actual_m; ir++){            /* loop over rows */
  106.       /* store in dummy variable */
  107.       for (i=0; i<actual_n; i++)
  108. if (actual_L==1)  
  109.   xdummy[i] = mat(x, ir, i);  
  110. else 
  111.   xdummy[i] = mat(y, ir, i);  
  112.       /* perform filtering lowpass and highpass*/
  113.       fpsconv(xdummy, actual_n, h0, h1, lhm1, ydummyl, ydummyh); 
  114.       /* restore dummy variables in matrices */
  115.       ic = c_o_a;
  116.       for  (i=0; i<c_o_a; i++){    
  117. mat(y, ir, i) = ydummyl[i];  
  118. mat(y, ir, ic++) = ydummyh[i];  
  119.       } 
  120.     }  
  121.     
  122.     /* go by columns in case of a 2D signal*/
  123.     if (m>1){
  124.       for (ic=0; ic<actual_n; ic++){            /* loop over column */
  125. /* store in dummy variables */
  126. for (i=0; i<actual_m; i++)
  127.   xdummy[i] = mat(y, i, ic);  
  128. /* perform filtering lowpass and highpass*/
  129. fpsconv(xdummy, actual_m, h0, h1, lhm1, ydummyl, ydummyh); 
  130. /* restore dummy variables in matrix */
  131. ir = r_o_a;
  132. for (i=0; i<r_o_a; i++){    
  133.   mat(y, i, ic) = ydummyl[i];  
  134.   mat(y, ir++, ic) = ydummyh[i];  
  135. }
  136.       }
  137.     }
  138.   }
  139. }
  140. #ifdef __STDC__
  141. fpsconv(double *x_in, int lx, double *h0, double *h1, int lhm1, 
  142. double *x_outl, double *x_outh)
  143. #else
  144. fpsconv(x_in, lx, h0, h1, lhm1, x_outl, x_outh)
  145. double *x_in, *h0, *h1, *x_outl, *x_outh;
  146. int lx, lhm1;
  147. #endif
  148. {
  149.   int i, j, ind;
  150.   double x0, x1;
  151.   for (i=lx; i < lx+lhm1; i++)
  152.     x_in[i] = *(x_in+(i-lx));
  153.   ind = 0;
  154.   for (i=0; i<(lx); i+=2){
  155.     x0 = 0;
  156.     x1 = 0;
  157.     for (j=0; j<=lhm1; j++){
  158.       x0 = x0 + x_in[i+j]*h0[lhm1-j];
  159.       x1 = x1 + x_in[i+j]*h1[lhm1-j];
  160.     }
  161.     x_outl[ind] = x0;
  162.     x_outh[ind++] = x1;
  163.   }
  164. }