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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
#include "mongoose_mex.hpp"
using namespace Mongoose;
void mexFunction
(
int nargout,
mxArray *pargout [ ],
int nargin,
const mxArray *pargin [ ]
)
{
const char* usage = "Usage: [G_coarse, A_coarse] = coarsen(G, (O, A))";
if(nargout > 3 || nargin < 1 || nargin > 3) mexErrMsgTxt(usage);
const mxArray *matGraph = pargin[0];
const mxArray *matOptions = (nargin >= 2 ? pargin[1] : NULL);
const mxArray *matNodeWeights = (nargin == 3 ? pargin[2] : NULL);
/* Get the options from the MATLAB inputs. */
EdgeCut_Options *O = mex_get_options(matOptions);
if(!O)
{
mexErrMsgTxt("Unable to get Options struct");
}
/* Get the graph from the MATLAB inputs. */
EdgeCutProblem *G = mex_get_graph(matGraph, matNodeWeights);
if(!G)
{
O->~EdgeCut_Options();
mexErrMsgTxt("Unable to get Graph struct");
}
G->initialize(O);
match(G, O);
EdgeCutProblem *G_coarse = coarsen(G, O);
cs *G_matrix = cs_spalloc(G_coarse->n, G_coarse->n, G_coarse->nz, 0, 0);
G_matrix->i = G_coarse->i;
G_matrix->p = G_coarse->p;
G_matrix->x = G_coarse->x;
/* Copy the coarsened graph back to MATLAB. */
pargout[0] = cs_mex_put_sparse(&G_matrix);
gp_mex_put_double(G_coarse->n, G_coarse->w, &pargout[1]);
pargout[2] = gp_mex_put_int(G->matchmap, G->n, 1, 0);
/* Cleanup */
G->~EdgeCutProblem();
G_coarse->i = NULL;
G_coarse->p = NULL;
G_coarse->x = NULL;
G_coarse->w = NULL;
G_coarse->matchmap = NULL;
G_coarse->~EdgeCutProblem();
O->~EdgeCut_Options();
}
|
