/*
* Public
*/
SEXP do_mdwt(SEXP vntX, SEXP vntH, SEXP vntL)
{
SEXP vntOut;
SEXP vntY;
SEXP vntLr;
double *x, *h, *y;
int m, n, lh, L;
#ifdef DEBUG_RWT
REprintf("In do_mdwt(x, h, L)...\n");
#endif
/*
* Handle first parameter (numeric matrix)
*/
#ifdef DEBUG_RWT
REprintf("\tfirst param 'x'\n");
#endif
if (GetMatrixDimen(vntX, &m, &n) != 2)
{
error("'x' is not a two dimensional matrix");
/*NOTREACHED*/
}
PROTECT(vntX = AS_NUMERIC(vntX));
x = NUMERIC_POINTER(vntX);
#ifdef DEBUG_RWT
REprintf("x[%d][%d] = 0x%p\n", m, n, x);
#endif
/*
* Handle second parameter (numeric vector)
*/
#ifdef DEBUG_RWT
REprintf("\tsecond param 'h'\n");
#endif
PROTECT(vntH = AS_NUMERIC(vntH));
h = NUMERIC_POINTER(vntH);
lh = GET_LENGTH(vntH);
#ifdef DEBUG_RWT
REprintf("h[%d] = 0x%p\n", GET_LENGTH(vntH), h);
#endif
/*
* Handle third parameter (integer scalar)
*/
#ifdef DEBUG_RWT
REprintf("\tthird param 'L'\n");
#endif
{
PROTECT(vntL = AS_INTEGER(vntL));
{
int *piL = INTEGER_POINTER(vntL);
L = piL[0];
}
UNPROTECT(1);
}
#ifdef DEBUG_RWT
REprintf("L = %d\n", L);
#endif
#ifdef DEBUG_RWT
REprintf("\tcheck number of levels\n");
#endif
if (L < 0)
{
error("The number of levels, L, must be a non-negative integer");
/*NOTREACHED*/
}
#ifdef DEBUG_RWT
REprintf("\tcheck dimen prereqs\n");
#endif
/* Check the ROW dimension of input */
if (m > 1)
{
double mtest = (double) m / pow(2.0, (double) L);
if (!isint(mtest))
{
error("The matrix row dimension must be of size m*2^(L)");
/*NOTREACHED*/
}
}
/* Check the COLUMN dimension of input */
if (n > 1)
{
double ntest = (double) n / pow(2.0, (double) L);
if (!isint(ntest))
{
error("The matrix column dimension must be of size n*2^(L)");
/*NOTREACHED*/
}
}
#ifdef DEBUG_RWT
REprintf("\tcreate value objects\n");
#endif
/* Create y value object */
{
#ifdef DEBUG_RWT
REprintf("\tcreate 'y' value object\n");
#endif
PROTECT(vntY = NEW_NUMERIC(n*m));
y = NUMERIC_POINTER(vntY);
/* Add dimension attribute to value object */
#ifdef DEBUG_RWT
REprintf("\tconvert 'y' value object to matrix\n");
#endif
{
SEXP vntDim;
PROTECT(vntDim = NEW_INTEGER(2));
INTEGER(vntDim)[0] = m;
INTEGER(vntDim)[1] = n;
SET_DIM(vntY, vntDim);
UNPROTECT(1);
}
}
/* Create Lr value object */
{
#ifdef DEBUG_RWT
REprintf("\tcreating 'Lr' value object\n");
#endif
PROTECT(vntLr = NEW_INTEGER(1));
INTEGER_POINTER(vntLr)[0] = L;
}
#ifdef DEBUG_RWT
REprintf("\tcompute discrete wavelet transform\n");
#endif
MDWT(x, m, n, h, lh, L, y);
/* Unprotect params */
UNPROTECT(2);
#ifdef DEBUG_RWT
REprintf("\tcreate list output object\n");
#endif
PROTECT(vntOut = NEW_LIST(2));
#ifdef DEBUG_RWT
REprintf("\tassigning value objects to list\n");
#endif
SET_VECTOR_ELT(vntOut, 0, vntY);
SET_VECTOR_ELT(vntOut, 1, vntLr);
/* Unprotect value objects */
UNPROTECT(2);
{
SEXP vntNames;
#ifdef DEBUG_RWT
REprintf("\tassigning names to value objects in list\n");
#endif
PROTECT(vntNames = NEW_CHARACTER(2));
SET_STRING_ELT(vntNames, 0, CREATE_STRING_VECTOR("y"));
SET_STRING_ELT(vntNames, 1, CREATE_STRING_VECTOR("L"));
SET_NAMES(vntOut, vntNames);
UNPROTECT(1);
}
/* Unprotect output object */
UNPROTECT(1);
#ifdef DEBUG_RWT
REprintf("\treturning output...\n");
#endif
return vntOut;
}
void mexFunction(int nlhs,mxArray *plhs[],int nrhs,const mxArray *prhs[])
{
double *x, *h, *y, *Lf, *Lr;
int m, n, h_col, h_row, lh, L, i, po2, j;
double mtest, ntest;
/* check for correct # of input variables */
if (nrhs>3){
mexErrMsgTxt("There are at most 3 input parameters allowed!");
return;
}
if (nrhs<2){
mexErrMsgTxt("There are at least 2 input parameters required!");
return;
}
x = mxGetPr(prhs[0]);
n = mxGetN(prhs[0]);
m = mxGetM(prhs[0]);
h = mxGetPr(prhs[1]);
h_col = mxGetN(prhs[1]);
h_row = mxGetM(prhs[1]);
if (h_col>h_row)
lh = h_col;
else
lh = h_row;
if (nrhs == 3){
L = (int) *mxGetPr(prhs[2]);
if (L < 0)
mexErrMsgTxt("The number of levels, L, must be a non-negative integer");
}
else /* Estimate L */ {
i=n;j=0;
while (even(i)){
i=(i>>1);
j++;
}
L=m;i=0;
while (even(L)){
L=(L>>1);
i++;
}
if(min(m,n) == 1)
L = max(i,j);
else
L = min(i,j);
if (L==0){
mexErrMsgTxt("Maximum number of levels is zero; no decomposition can be performed!");
return;
}
}
/* Check the ROW dimension of input */
if(m > 1){
mtest = (double) m/pow(2.0, (double) L);
if (!isint(mtest))
mexErrMsgTxt("The matrix row dimension must be of size m*2^(L)");
}
/* Check the COLUMN dimension of input */
if(n > 1){
ntest = (double) n/pow(2.0, (double) L);
if (!isint(ntest))
mexErrMsgTxt("The matrix column dimension must be of size n*2^(L)");
}
plhs[0] = mxCreateDoubleMatrix(m,n,mxREAL);
y = mxGetPr(plhs[0]);
plhs[1] = mxCreateDoubleMatrix(1,1,mxREAL);
Lr = mxGetPr(plhs[1]);
*Lr = L;
MDWT(x, m, n, h, lh, L, y);
}
