压缩感知图像MATLAB代码（美国乔治亚理工大学）

/* Noiselet Transform mex function to take noiselet transform Usage: w = realnoiselet(x) x must be a REAL VALUED COLUMN VECTOR or MATRIX m = size(x,1) must be a POWER OF TWO Notes: 1) This implementation uses exactly m*log2(m) additions/subtractions. 2) This is a real-valued variation of "dragon" noiselets. 3) This is symmetric and orthogonal. To invert, apply again and divide by vector length. Written by: Justin Romberg, Georgia Tech Peter Stobbe, Caltech Email: [email protected], [email protected] Created: October 2006 Last Modified: December 2006 */ #include <stdlib.h> #include "mex.h" /* y - output x - input m - length of vector */ void noiselet_apply_vector(double *y, double *x, unsigned m) { unsigned c, d, j, k; double temp; /* For non-dragon noiselets, use c = m >> 1; */ c = m - 1; for (j = 0; j < m >> 1; j++) { k = j ^ c; y[j] = x[j] + x[k]; y[k] = x[j] - x[k]; } for (d = c >> 1; d > 0; d >>= 1) { for (j = 0; j < m >> 1; j++) { k = j ^ c ^ d; temp = y[j]; y[j] = y[j] - y[k]; y[k] = temp + y[k]; } } } /* y - output x - input m - length of vectors (number of rows) n - number of vectors (number of columns) */ void noiselet_apply_matrix(double *y, double *x, unsigned m, unsigned n) { unsigned j; for(j = 0; j < n; j++) { noiselet_apply_vector(y + j*m, x + j*m, m); } } /* check that the vector length is a power of 2, just using bitshifting instead of log */ void checkPowerTwo(unsigned m) { /* check that it's not a degenerate 0 by 1 vector or singleton */ if (m <= 1) { mexErrMsgTxt("Vector length must be greater than 1."); } /* keep dividing by two until result is odd */ while( (m & 1) == 0 ){ m >>= 1; } /* check that m is not a multiple of an odd number greater than 1 */ if (m > 1) { mexErrMsgTxt("Vector length must be power of 2."); } } /* The gateway routine. */ void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[]) { double *x, *y; unsigned m, n; /* Check for the proper number of arguments. */ if (nrhs != 1) { mexErrMsgTxt("One and only one input required."); } if (nlhs > 1) { mexErrMsgTxt("Too many output arguments."); } /* input size */ m = mxGetM(prhs[0]); checkPowerTwo(m); n = mxGetN(prhs[0]); if (mxIsComplex(prhs[0])) { mexErrMsgTxt("Input must be real."); } /* Create matrix for the return argument. */ plhs[0] = mxCreateDoubleMatrix(m, n, mxREAL); /* Assign pointers to each input and output. */ x = mxGetPr(prhs[0]); y = mxGetPr(plhs[0]); /* Call the C subroutine. */ noiselet_apply_matrix(y, x, m, n); return; }