]> AND Private Git Repository - these_gilles.git/blob - THESE/codes/graphe/GCmex1.9/GraphCutConstr.cpp
Logo AND Algorithmique Numérique Distribuée

Private GIT Repository
07 sep
[these_gilles.git] / THESE / codes / graphe / GCmex1.9 / GraphCutConstr.cpp
1 \r
2 #include "mex.h"\r
3 #include "GCoptimization.h"\r
4 #include "GraphCut.h"\r
5 #include <stdlib.h>\r
6 \r
7 /* Defines */\r
8 \r
9 \r
10 /*\r
11  * Matlab wrapper for Weksler graph cut implementation\r
12  *\r
13  * usage:\r
14  * [gch] = GraphCutConstr(width, height, num_labels, DataCost, SmoothnessCost,[vCost,hCost])\r
15  *\r
16  * Note that data types are crucials!\r
17  * \r
18  * Inputs:\r
19  *  width, hieght - 2D grid dimensions. (do not cast to int)\r
20  *  num_labels - number of labels.  (do not cast to int)\r
21  *  DataCost - of type float, array size [width*height*#labels], the data term for pixel\r
22  *             x,y recieving label l is stroed at [(x+y*width)*#labels + l]\r
23  *  SmoothnessCost - of type float, array size [#labels.^2] the cost between l1 and l2 is\r
24  *                   stored at [l1+l2*#labels] = Vpq(lp,lq)\r
25  *  vCost, hCost - of type float, array size[width*height] each, \r
26  *                 horizontal and vertical cues for smoothness term\r
27  *\r
28  * Outputs:\r
29  *  gch - of type int32, graph cut handle - do NOT mess with it!\r
30  */\r
31 void mexFunction(\r
32     int           nlhs,         /* number of expected outputs */\r
33     mxArray       *plhs[],      /* mxArray output pointer array */\r
34     int           nrhs,         /* number of inputs */\r
35     const mxArray         *prhs[]       /* mxArray input pointer array */\r
36     )\r
37 {\r
38     \r
39     GCoptimization::PixelType width, height;\r
40     int num_labels;\r
41     Graph::captype *DataCost;\r
42     Graph::captype *SmoothnessCost;\r
43     Graph::captype *hCue = NULL;\r
44     Graph::captype *vCue = NULL;\r
45     GCoptimization::LabelType *Labels;\r
46     int dims[2]; /* for labels allocation */\r
47     \r
48     GCoptimization *MyGraph = NULL;\r
49         \r
50        /* check number of inout arguments - must be 5 */\r
51     if ((nrhs != 5)&&(nrhs!=7)) {\r
52         mexErrMsgIdAndTxt("GraphCut:NarginError","Wrong number of input argumnets");\r
53     }\r
54     \r
55     /* check first input width: must be 1 int element */\r
56     GetScalar(prhs[0], width);\r
57     dims[1] = width;\r
58     \r
59     /* check second input height: must be 1 element */\r
60     GetScalar(prhs[1], height);\r
61     dims[0] = height;\r
62     \r
63     /* check third input #labels: must be single element */\r
64     GetScalar(prhs[2], num_labels);\r
65     \r
66     /* check fourth input DataCost: must have #labels*height*width elements of type float */\r
67     if ( mxGetNumberOfElements(prhs[3]) != num_labels*width*height ) {\r
68         mexErrMsgIdAndTxt("GraphCut:DataCost",\r
69         "DataCost argument does not contains the right number of elements");\r
70     }\r
71     if (mxGetClassID(prhs[3]) != mxSINGLE_CLASS ) {\r
72         mexErrMsgIdAndTxt("GraphCut:DataCost",\r
73         "DataCost argument is not of type float");\r
74     }\r
75     DataCost = (Graph::captype*)mxGetData(prhs[3]);\r
76 \r
77     /* check fifth input SmoothnessCost: must have #labels.^2 elements of type float */\r
78     if ( mxGetNumberOfElements(prhs[4]) != num_labels*num_labels ) {\r
79         mexErrMsgIdAndTxt("GraphCut:SmoothnessCost",\r
80         "SmoothnessCost argument does not contains the right number of elements");\r
81     }\r
82     if (mxGetClassID(prhs[4]) != mxSINGLE_CLASS ) {\r
83         mexErrMsgIdAndTxt("GraphCut:SmoothnessCost",\r
84         "SmoothnessCost argument is not of type float");\r
85     }\r
86     SmoothnessCost = (Graph::captype*)mxGetData(prhs[4]);\r
87     \r
88     if ( nrhs == 7 ) {\r
89         /* add hCue and vCue */\r
90         if ( mxGetNumberOfElements(prhs[5]) != width*height ) {\r
91             mexErrMsgIdAndTxt("GraphCut:SmoothnessCost",\r
92             "vCue argument does not contains the right number of elements");\r
93         }\r
94         if (mxGetClassID(prhs[5]) != mxSINGLE_CLASS ) {\r
95             mexErrMsgIdAndTxt("GraphCut:SmoothnessCost",\r
96             "vCue argument is not of type float");\r
97         }\r
98         vCue = (Graph::captype*)mxGetData(prhs[5]);\r
99         \r
100         if ( mxGetNumberOfElements(prhs[6]) != width*height ) {\r
101             mexErrMsgIdAndTxt("GraphCut:SmoothnessCost",\r
102             "hCue argument does not contains the right number of elements");\r
103         }\r
104         if (mxGetClassID(prhs[6]) != mxSINGLE_CLASS ) {\r
105             mexErrMsgIdAndTxt("GraphCut:SmoothnessCost",\r
106             "hCue argument is not of type float");\r
107         }\r
108         hCue = (Graph::captype*)mxGetData(prhs[6]);\r
109     }\r
110     \r
111     /* prepare the output argument */\r
112     if ( nlhs != 1 ) {\r
113         mexErrMsgIdAndTxt("GraphCut:OutputArg","Wrong number of output arguments");\r
114     }\r
115     \r
116     MyGraph = new GCoptimization(width, height, num_labels, SET_ALL_AT_ONCE, SET_ALL_AT_ONCE);\r
117     MyGraph->setData(DataCost);\r
118     if ( vCue != NULL && vCue != NULL ) \r
119         MyGraph->setSmoothness(SmoothnessCost, hCue, vCue);\r
120     else\r
121         MyGraph->setSmoothness(SmoothnessCost);\r
122     \r
123         \r
124     /* create a container for the pointer */\r
125     dims[0] = 1; dims[1] = 0;\r
126     plhs[0] = mxCreateNumericMatrix(1, 1, MATLAB_POINTER_TYPE, mxREAL);\r
127     \r
128     GraphHandle* gh;\r
129     gh = (GraphHandle*) mxGetData(plhs[0]);\r
130     *gh = (GraphHandle)(MyGraph);\r
131 }\r
132     \r
133 \r