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

波变换

开发平台:

Matlab

  1. /*
  2. File Name: MRDWT.c
  3. Last Modification Date: 09/21/95 15:42:59
  4. Current Version: MRDWT.c 2.4
  5. File Creation Date: Wed Oct 12 08:44:43 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. MATLAB description:
  43. %[yl,yh] = mrdwt(x,h,L);
  44. % function computes the redundant discrete wavelet transform y for a 1D or
  45. % 2D input signal . redundant means here that the subsampling after each
  46. % stage is omitted. yl contains the lowpass and yl the highpass
  47. % components. In case of a 2D signal the ordering in yh is [lh hl hh lh hl
  48. % ... ] (first letter refers to row, second to column filtering). 
  49. %
  50. %    Input:
  51. % x    : finite length 1D or 2D signal (implicitely periodized)
  52. %       h    : scaling filter
  53. %       L    : number of levels. in case of a 1D signal length(x) must be
  54. %              divisible by 2^L; in case of a 2D signal the row and the
  55. %              column dimension must be divisible by 2^L.
  56. %   
  57. %    Output:
  58. %       yl   : lowpass component
  59. %       yh   : highpass components
  60. %
  61. % see also: mdwt, midwt, mirdwt
  62. */
  63. #include <math.h>
  64. #include <stdio.h>
  65. /*#define mat(a, i, j) (a[m*(j)+i]) */
  66. #define mat(a, i, j) (*(a + (m*(j)+i))) 
  67. #define max(a, b) ((a) > (b) ? (a) : (b))
  68. #ifdef __STDC__
  69. MRDWT(double *x, int m, int n, double *h, int lh, int L,
  70.       double *yl, double *yh)
  71. #else
  72. MRDWT(x, m, n, h, lh, L, yl, yh)
  73. double *x, *h, *yl, *yh;
  74. int m, n, lh, L;
  75. #endif
  76. {
  77.   double *tmp;
  78.   double  *h0, *h1, *ydummyll, *ydummylh, *ydummyhl;
  79.   double *ydummyhh, *xdummyl , *xdummyh;
  80.   long i, j;
  81.   int actual_L, actual_m, actual_n, c_o_a, ir, n_c, n_cb, n_c_o;
  82.   int ic, n_r, n_rb, n_r_o, c_o_a_p2n, sample_f;
  83.   xdummyl = (double *)mxCalloc(max(m,n)+lh-1,sizeof(double));
  84.   xdummyh = (double *)mxCalloc(max(m,n)+lh-1,sizeof(double));
  85.   ydummyll = (double *)mxCalloc(max(m,n),sizeof(double));
  86.   ydummylh = (double *)mxCalloc(max(m,n),sizeof(double));
  87.   ydummyhl = (double *)mxCalloc(max(m,n),sizeof(double));
  88.   ydummyhh = (double *)mxCalloc(max(m,n),sizeof(double));
  89.   h0 = (double *)mxCalloc(lh,sizeof(double));
  90.   h1 = (double *)mxCalloc(lh,sizeof(double));
  91.   if (n==1){
  92.     n = m;
  93.     m = 1;
  94.   }  
  95.   /* analysis lowpass and highpass */
  96.   for (i=0; i<lh; i++){
  97.     h0[i] = h[lh-i-1];
  98.     h1[i] =h[i];
  99.   }
  100.   for (i=0; i<lh; i+=2)
  101.     h1[i] = -h1[i];
  102.   
  103.   actual_m = 2*m;
  104.   actual_n = 2*n;
  105.   for (i=0; i<m*n; i++)
  106.     yl[i] = x[i];
  107.   
  108.   /* main loop */
  109.   sample_f = 1;
  110.   for (actual_L=1; actual_L <= L; actual_L++){
  111.     actual_m = actual_m/2;
  112.     actual_n = actual_n/2;
  113.     /* actual (level dependent) column offset */
  114.     if (m==1)
  115.       c_o_a = n*(actual_L-1);
  116.     else
  117.       c_o_a = 3*n*(actual_L-1);
  118.     c_o_a_p2n = c_o_a + 2*n;
  119.     
  120.     /* go by rows */
  121.     n_cb = n/actual_n;                 /* # of column blocks per row */
  122.     for (ir=0; ir<m; ir++){            /* loop over rows */
  123.       for (n_c=0; n_c<n_cb; n_c++){    /* loop within one row */      
  124. /* store in dummy variable */
  125. ic = -sample_f + n_c;
  126. for (i=0; i<actual_n; i++){    
  127.   ic = ic + sample_f;
  128.   xdummyl[i] = mat(yl, ir, ic);  
  129. }
  130. /* perform filtering lowpass/highpass */
  131. fpconv(xdummyl, actual_n, h0, h1, lh, ydummyll, ydummyhh); 
  132. /* restore dummy variables in matrices */
  133. ic = -sample_f + n_c;
  134. for  (i=0; i<actual_n; i++){    
  135.   ic = ic + sample_f;
  136.   mat(yl, ir, ic) = ydummyll[i];  
  137.   mat(yh, ir, c_o_a+ic) = ydummyhh[i];  
  138.       }
  139.     }
  140.       
  141.     /* go by columns in case of a 2D signal*/
  142.     if (m>1){
  143.       n_rb = m/actual_m;                 /* # of row blocks per column */
  144.       for (ic=0; ic<n; ic++){            /* loop over column */
  145. for (n_r=0; n_r<n_rb; n_r++){    /* loop within one column */
  146.   /* store in dummy variables */
  147.   ir = -sample_f + n_r;
  148.   for (i=0; i<actual_m; i++){    
  149.     ir = ir + sample_f;
  150.     xdummyl[i] = mat(yl, ir, ic);  
  151.     xdummyh[i] = mat(yh, ir,c_o_a+ic);  
  152.   }
  153.   /* perform filtering: first LL/LH, then HL/HH */
  154.   fpconv(xdummyl, actual_m, h0, h1, lh, ydummyll, ydummylh); 
  155.   fpconv(xdummyh, actual_m, h0, h1, lh, ydummyhl, ydummyhh); 
  156.   /* restore dummy variables in matrices */
  157.   ir = -sample_f + n_r;
  158.   for (i=0; i<actual_m; i++){    
  159.     ir = ir + sample_f;
  160.     mat(yl, ir, ic) = ydummyll[i];  
  161.     mat(yh, ir, c_o_a+ic) = ydummylh[i];  
  162.     mat(yh, ir,c_o_a+n+ic) = ydummyhl[i];  
  163.     mat(yh, ir, c_o_a_p2n+ic) = ydummyhh[i];  
  164.   }
  165. }
  166.       }
  167.     }
  168.     sample_f = sample_f*2;
  169.   }
  170. }
  171. #ifdef __STDC__
  172. fpconv(double *x_in, int lx, double *h0, double *h1, int lh,
  173.        double *x_outl, double *x_outh)
  174. #else
  175. fpconv(x_in, lx, h0, h1, lh, x_outl, x_outh)
  176. double *x_in, *h0, *h1, *x_outl, *x_outh;
  177. int lx, lh;
  178. #endif
  179. {
  180.   int i, j;
  181.   double x0, x1;
  182.   for (i=lx; i < lx+lh-1; i++)
  183.     x_in[i] = x_in[i-lx];
  184.   for (i=0; i<lx; i++){
  185.     x0 = 0;
  186.     x1 = 0;
  187.     for (j=0; j<lh; j++){
  188.       x0 = x0 + x_in[j+i]*h0[lh-1-j];
  189.       x1 = x1 + x_in[j+i]*h1[lh-1-j];
  190.     }
  191.     x_outl[i] = x0;
  192.     x_outh[i] = x1;
  193.   }
  194. }