summaryrefslogtreecommitdiffstats
path: root/SD-VBS/common/toolbox/lagrcv/calcGradientPyrMex.cc
blob: d94130be6af6ba40a9aab2138ccbd66541a603ad (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45

/* compile with 
rm liblagrcv.a
gcc -c lagrcv.cpp
ar rc liblagrcv.a lagrcv.o
ranlib liblagrcv.a
mex7 calcTextureMex.cc -L/home/ikkjin/LagrMatlab/opencv/matlab -llagrcv -I/home/ikkjin/LagrMatlab/opencv/matlab/
*/

#include "mex.h"
#include "lagrcv.h"
#include <stdio.h>
#include <math.h>

// TODO: add number of corners parameter
void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[]) {
  // usage: [dxPye dyPyr] = 
  //   calcGradientPyrMex(imagePyr)

  const int *cellDims = mxGetDimensions(prhs[0]);
  double *image; 
  const mxArray* imgArray;
  mxArray *dxArray, *dyArray;
  double *dx, *dy;
  const int *imdims;

  plhs[0] = mxCreateCellArray(1, cellDims);
  plhs[1] = mxCreateCellArray(1, cellDims);

  for(int i=0; i<cellDims[0];i++){
  	imgArray= mxGetCell(prhs[0],i);
	image=mxGetPr(imgArray);
        imdims = mxGetDimensions(imgArray);

        dxArray = mxCreateNumericMatrix(imdims[0], imdims[1], mxDOUBLE_CLASS, mxREAL);
        dyArray = mxCreateNumericMatrix(imdims[0], imdims[1], mxDOUBLE_CLASS, mxREAL);
	mxSetCell(plhs[0], i, dxArray);
	mxSetCell(plhs[1], i, dyArray);
	dx=mxGetPr(dxArray);
	dy=mxGetPr(dyArray);

  	calcGradient(image, imdims[0], imdims[1], dx, dy);
  	//calcSobel(image, imdims[0], imdims[1], dx, dy);
  }
}