From f618466c25d43f3bae9e40920273bf77de1e1149 Mon Sep 17 00:00:00 2001
From: leochanj105 <leochanj@live.unc.edu>
Date: Mon, 19 Oct 2020 23:09:30 -0400
Subject: initial sd-vbs

initial sd-vbs

add sd-vbs

sd-vbs
---
 SD-VBS/common/toolbox/MultiNcut/multiAffinityic.c | 216 ++++++++++++++++++++++
 1 file changed, 216 insertions(+)
 create mode 100755 SD-VBS/common/toolbox/MultiNcut/multiAffinityic.c

(limited to 'SD-VBS/common/toolbox/MultiNcut/multiAffinityic.c')

diff --git a/SD-VBS/common/toolbox/MultiNcut/multiAffinityic.c b/SD-VBS/common/toolbox/MultiNcut/multiAffinityic.c
new file mode 100755
index 0000000..6919db9
--- /dev/null
+++ b/SD-VBS/common/toolbox/MultiNcut/multiAffinityic.c
@@ -0,0 +1,216 @@
+/*================================================================
+* function w = affinityic(emag,ephase,pi,pj,subgrid,nrSubgrid,ncSubgrid,subpi,sigma)
+* Input:
+*   emag = edge strength at each pixel
+*   ephase = edge phase at each pixel
+*   [pi,pj] = index pair representation for MALTAB sparse matrices, pi index in original grid, size(pj)=nrW*ncW+1
+*   subgrid= index of the subgrid in the original image (first pixel's index is one!)
+*   [nrSubgrid,ncSubgrid]= number of rows et colums of the subgrid
+*   subpi = pi but with index in subgrid
+*   sigma = sigma for IC energy
+* Output:
+*   w = affinity with IC at [pi,pj]
+*
+
+% test sequence
+f = synimg(10);
+[i,j] = cimgnbmap(size(f),2);
+[ex,ey,egx,egy] = quadedgep(f);
+a = affinityic(ex,ey,egx,egy,i,j)
+show_dist_w(f,a);
+
+* Stella X. Yu, Nov 19, 2001.
+*=================================================================*/
+
+# include "mex.h"
+# include "math.h"
+
+void mexFunction(
+    int nargout,
+    mxArray *out[],
+    int nargin,
+    const mxArray *in[]
+)
+{
+    /* declare variables */
+    int nr, nc, np,  nW, total;
+    int i, j, k, t, ix, iy, jx, jy, ii, jj, iip1, jjp1, iip2, jjp2, step,nrSubgrid, ncSubgrid;
+    double sigma, di, dj, a, z, maxori, phase1, phase2, slope;
+	int *ir, *jc;
+    /*	unsigned long *pi, *pj, *subpi;

 */
+    unsigned int *pi, *pj, *subpi;
+	double *emag, *ephase, *w,*tmp,*subgrid;
+	
+    
+    /* check argument */
+    if (nargin<8) {
+        mexErrMsgTxt("Eight input arguments required");
+    }
+    if (nargout>1) {
+        mexErrMsgTxt("Too many output arguments");
+    }
+
+    /* get edgel information */
+	nr = mxGetM(in[0]);
+	nc = mxGetN(in[0]);
+	if ( nr*nc ==0 || nr != mxGetM(in[1]) || nc != mxGetN(in[1]) ) {
+	    mexErrMsgTxt("Edge magnitude and phase shall be of the same image size");
+	}
+    emag = mxGetPr(in[0]);
+    ephase = mxGetPr(in[1]);
+    np = nr * nc;
+    
+    /*get subgrid information*/
+    
+    tmp = mxGetData(in[5]);
+    nrSubgrid = (int)tmp[0];
+    tmp = mxGetData(in[6]);
+    ncSubgrid = (int)tmp[0];
+    
+    /* printf("nrSubgrid=%d\n",nrSubgrid);
 
+       printf("ncSubgrid=%d\n",ncSubgrid);
 */
+    if (nrSubgrid* ncSubgrid != mxGetM(in[4])*mxGetN(in[4])) {
+          mexErrMsgTxt("Error in the size of the subgrid");
+          }
+    subgrid = mxGetData(in[4]);
+    nW = nrSubgrid * ncSubgrid;
+    
+    /* get new index pair */
+    if (!mxIsUint32(in[2]) | !mxIsUint32(in[3])) {
+        mexErrMsgTxt("Index pair shall be of type UINT32");
+    }
+    if (mxGetM(in[3]) * mxGetN(in[3]) != nW + 1) {
+        mexErrMsgTxt("Wrong index representation");
+    }
+    pi = mxGetData(in[2]);
+    pj = mxGetData(in[3]);
+    subpi = mxGetData(in[7]);
+    /*{printf("pi[50] = %d\n",pi[50]);}
+    {printf("subpi[5] = %d\n",subpi[5]);}
+    {printf("subpi[6] = %d\n",subpi[6]);}
+    {printf("subpi[4] = %d\n",subpi[4]);}*/
+
+    /* create output */          /*!!!!!!!!!!!!!!!!!!!!!!!changer taille output!!!!!!!!!!*/
+    out[0] = mxCreateSparse(nW,nW,pj[nW],mxREAL);
+    if (out[0]==NULL) {
+	    mexErrMsgTxt("Not enough memory for the output matrix");
+	}
+	w = mxGetPr(out[0]);
+	ir = mxGetIr(out[0]);
+	jc = mxGetJc(out[0]);
+	
+    /* find my sigma */
+	if (nargin<9) {
+	    sigma = 0;
+    	for (k=0; k<np; k++) { 
+    	    if (emag[k]>sigma) { sigma = emag[k]; }
+    	}
+    	sigma = sigma / 6;
+    	printf("sigma = %6.5f",sigma);
+	} else {
+	    sigma = mxGetScalar(in[8]);
+	}
+	a = 0.5 / (sigma * sigma);
+	
+    /* computation */ 
+    total = 0;
+    
+    for (j=0; j<nW; j++){
+        t= (int)subgrid[j]-1;  /*on parcourt tous les pixels de la sous-grille dans la grille d'origine*/        
+
+        jc[j] = total;    /* total represente le nombre voisins  du pixel j*/
+        jx = t / nr; /* col */  
+        jy = t % nr; /* row */
+        
+        for (k=pj[j]; k<pj[j+1]; k++) {   /*k represente les indices correspondant au pixel j dans pi*/
+        
+            i = pi[k]-1;                     /*i est un voisin de j a considerer*/
+          
+            if (i==j) {
+                maxori = 1;
+            
+            } else {
+        
+                ix = i / nr; 
+                iy = i % nr;
+
+                /* scan */            
+                di = (double) (iy - jy);
+                dj = (double) (ix - jx);
+            
+                maxori = 0.;
+	            phase1 = ephase[j];
+
+	               
+                /* sample in i direction */
+                if (abs(di) >= abs(dj)) {  
+            	    slope = dj / di;
+            	    step = (iy>=jy) ? 1 : -1;
+            	
+              	    iip1 = jy;
+            	    jjp1 = jx;
+	
+	               
+	                for (ii=0;ii<abs(di);ii++){
+	                    iip2 = iip1 + step;
+	                    jjp2 = (int)(0.5 + slope*(iip2-jy) + jx);
+	  	  
+	                    phase2 = ephase[iip2+jjp2*nr];
+               
+	                    if (phase1 != phase2) {
+	                        z = (emag[iip1+jjp1*nr] + emag[iip2+jjp2*nr]);
+	                        if (z > maxori){
+	                            maxori = z;
+	                        }
+	                    } 
+	             
+	                    iip1 = iip2;
+	                    jjp1 = jjp2;
+	                    phase1 = phase2;
+	                }
+	            
+	            /* sample in j direction */    
+                } else { 
+	                slope = di / dj;
+	                step =  (ix>=jx) ? 1: -1;
+
+    	            jjp1 = jx;
+	                iip1 = jy;	           
+	    
+	 
+	                for (jj=0;jj<abs(dj);jj++){
+	                    jjp2 = jjp1 + step;
+	                    iip2 = (int)(0.5+ slope*(jjp2-jx) + jy);
+	  	  
+	                    phase2 = ephase[iip2+jjp2*nr];
+	     
+	                    if (phase1 != phase2){
+	                        z = (emag[iip1+jjp1*nr] + emag[iip2+jjp2*nr]);
+	                        if (z > maxori){ 
+	                            maxori = z; 
+	                        }
+	                        
+	                    }
+	  
+	                    iip1 = iip2;
+	                    jjp1 = jjp2;
+	                    phase1 = phase2;
+	                }
+                }            
+            
+                maxori = 0.5 * maxori;
+                maxori = exp(-maxori * maxori * a);
+            }  
+            /*if (total<20) {printf("subpi[k] = %d\n",subpi[k]);}*/
+
+		    ir[total] = (int)subpi[k];
+/*if (total<20) {printf("ir[total] = %d\n",ir[total]);}*/
+		    w[total] = maxori;
+		    total = total + 1;
+			
+		} /* i */
+       }  /*j*/
+     /*printf("total = %d\n",total);*/   
+      /*printf("ir[100] = %d\n",ir[100]);*/ 
+    jc[nW] = total;
+}  
-- 
cgit v1.2.2