diff options
Diffstat (limited to 'SD-VBS/common/toolbox/MultiNcut/multiIntensityFirstLayer.c')
-rwxr-xr-x | SD-VBS/common/toolbox/MultiNcut/multiIntensityFirstLayer.c | 126 |
1 files changed, 126 insertions, 0 deletions
diff --git a/SD-VBS/common/toolbox/MultiNcut/multiIntensityFirstLayer.c b/SD-VBS/common/toolbox/MultiNcut/multiIntensityFirstLayer.c new file mode 100755 index 0000000..0cae523 --- /dev/null +++ b/SD-VBS/common/toolbox/MultiNcut/multiIntensityFirstLayer.c | |||
@@ -0,0 +1,126 @@ | |||
1 | /*================================================================ | ||
2 | * function w = intensityWppc(image,pi,pj,rMin,sigmaX,sigmaIntensite,valeurMinW ) | ||
3 | * Input: | ||
4 | * [pi,pj] = index pair representation for MALTAB sparse matrices | ||
5 | * Output: | ||
6 | * w = affinity with IC at [pi,pj] | ||
7 | * | ||
8 | |||
9 | pixels i and j (corresponding to the sampling in pi,pj) are fully connected when d(i,j) <= rmin; | ||
10 | |||
11 | % test sequence | ||
12 | f = synimg(10); | ||
13 | [i,j] = cimgnbmap(size(f),2); | ||
14 | [ex,ey,egx,egy] = quadedgep(f); | ||
15 | a = affinityic(ex,ey,egx,egy,i,j) | ||
16 | show_dist_w(f,a); | ||
17 | |||
18 | * Stella X. Yu, Nov 19, 2001. | ||
19 | *=================================================================*/ | ||
20 | |||
21 | # include "mex.h" | ||
22 | # include "math.h" | ||
23 | |||
24 | void mexFunction( | ||
25 | int nargout, | ||
26 | mxArray *out[], | ||
27 | int nargin, | ||
28 | const mxArray *in[] | ||
29 | ) | ||
30 | { | ||
31 | /* declare variables */ | ||
32 | int nr, nc, np, total; | ||
33 | int i, j, k, ix, iy, jx, jy; | ||
34 | int *ir, *jc; | ||
35 | int squareDistance; | ||
36 | /* unsigned long *pi, *pj; */ | ||
37 | unsigned int *pi, *pj; | ||
38 | double *w; | ||
39 | |||
40 | double temp,a1,a2,wij; | ||
41 | double rMin; | ||
42 | double sigmaX, sigmaIntensite,valeurMinW; | ||
43 | double *image; | ||
44 | |||
45 | /* check argument */ | ||
46 | if (nargin<7) { | ||
47 | mexErrMsgTxt("Four input arguments required"); | ||
48 | } | ||
49 | if (nargout>1) { | ||
50 | mexErrMsgTxt("Too many output arguments"); | ||
51 | } | ||
52 | |||
53 | /* get edgel information */ | ||
54 | nr = mxGetM(in[0]); | ||
55 | nc = mxGetN(in[0]); | ||
56 | np = nr * nc; | ||
57 | |||
58 | image = mxGetPr(in[0]); | ||
59 | |||
60 | |||
61 | |||
62 | /* get new index pair */ | ||
63 | if (!mxIsUint32(in[1]) | !mxIsUint32(in[2])) { | ||
64 | mexErrMsgTxt("Index pair shall be of type UINT32"); | ||
65 | } | ||
66 | if (mxGetM(in[2]) * mxGetN(in[2]) != np + 1) { | ||
67 | mexErrMsgTxt("Wrong index representation"); | ||
68 | } | ||
69 | pi = mxGetData(in[1]); | ||
70 | pj = mxGetData(in[2]); | ||
71 | |||
72 | /* create output */ | ||
73 | out[0] = mxCreateSparse(np,np,pj[np],mxREAL); | ||
74 | if (out[0]==NULL) { | ||
75 | mexErrMsgTxt("Not enough memory for the output matrix"); | ||
76 | } | ||
77 | w = mxGetPr(out[0]); | ||
78 | ir = mxGetIr(out[0]); | ||
79 | jc = mxGetJc(out[0]); | ||
80 | |||
81 | rMin = mxGetScalar(in[3]); | ||
82 | sigmaX = mxGetScalar(in[4]); | ||
83 | sigmaIntensite= mxGetScalar(in[5]); | ||
84 | valeurMinW = mxGetScalar(in[6]); | ||
85 | |||
86 | a1 = 1.0/ (sigmaX*sigmaX); | ||
87 | a2 = 1.0 / (sigmaIntensite*sigmaIntensite ); | ||
88 | |||
89 | /* computation */ | ||
90 | total = 0; | ||
91 | for (j=0; j<np; j++) { | ||
92 | |||
93 | jc[j] = total; | ||
94 | jx = j / nr; /* col */ | ||
95 | jy = j % nr; /* row */ | ||
96 | |||
97 | for (k=pj[j]; k<pj[j+1]; k++) { | ||
98 | |||
99 | i = pi[k]; | ||
100 | |||
101 | if (i==j) { | ||
102 | wij= 1; /*voir*/ | ||
103 | |||
104 | } else { | ||
105 | ix = i / nr; | ||
106 | iy = i % nr; | ||
107 | |||
108 | squareDistance = (ix-jx)*(ix-jx)+(iy-jy)*(iy-jy); | ||
109 | |||
110 | temp = image[i]-image[j]; | ||
111 | wij = exp(- squareDistance * a1 - temp*temp * a2 ); | ||
112 | /*if(wij < valeurMinW) | ||
113 | wij = -0.1;*/ | ||
114 | } | ||
115 | ir[total] = i; | ||
116 | |||
117 | w[total] = wij; | ||
118 | total = total + 1; | ||
119 | |||
120 | } /* i */ | ||
121 | |||
122 | } /* j */ | ||
123 | |||
124 | jc[np] = total; | ||
125 | } | ||
126 | |||