Function mexFunction

maxflow_matlab/interactivemaxflowmex.cpp:17–142

[flow label]=interactivemaxflowmex(I,fgProb, bgProb, seed, lambda,sigma)

// [flow label]=interactivemaxflowmex(I,fgProb, bgProb, seed, lambda,sigma)
void mexFunction(int			nlhs, 		/* number of expected outputs */
                 mxArray		*plhs[],	/* mxArray output pointer array */
                 int			nrhs, 		/* number of inputs */
                 const mxArray	*prhs[]		/* mxArray input pointer array */)
{
    // input checks
    if (nrhs != 6 )
    {
        mexErrMsgTxt ("USAGE: [flow label]=interactivemaxflowmex(I,fgProb, bgProb, seed, lambda,sigma);");
    }
    const mxArray *I = prhs[0];
    const mxArray *fgProb = prhs[1];
    const mxArray *bgProb = prhs[2];
    const mxArray *seed   = prhs[3];
    double lamda= * mxGetPr(prhs[4]);
    double sigma= * mxGetPr(prhs[5]);
    
    unsigned char * IPr=(unsigned char *)mxGetPr(I);
    double * fgProbPr=mxGetPr(fgProb);
    double * bgProbPr=mxGetPr(bgProb);
    unsigned char * seedPr=(unsigned char *)mxGetPr(seed);
    // size of image
    mwSize m = mxGetM(I);//height
    mwSize n = mxGetN(I);//width
    
    //construct graph
    typedef Graph<float,float,float> GraphType;
    GraphType *g = new GraphType(/*estimated # of nodes*/ m*n, /*estimated # of edges*/ 2*m*n);
    g->add_node(m*n);
    
    float maxWeight=-10000;
    for(int x=0;x<n;x++)
    {
        for(int y=0;y<m;y++)
        {
            //n-link
            float pValue=(float)*(IPr+x*m+y);
            int uperPointx=x;
            int uperPointy=y-1;
            int LeftPointx=x-1;
            int LeftPointy=y;
            float n_weight=0;
            if(uperPointy>=0 && uperPointy<m)
            {
                float qValue=(float)*(IPr+uperPointx*m+uperPointy);
                n_weight=lamda*exp(-(pValue-qValue)*(pValue-qValue)/(2*sigma*sigma));
                int pIndex=x*m+y;
                int qIndex=uperPointx*m+uperPointy;
                
                g->add_edge(qIndex,pIndex,n_weight,n_weight);
            }
            if(n_weight>maxWeight)
            {
                maxWeight=n_weight;
            }
            
            if(LeftPointx>=0 && LeftPointx<n)
            {