void writeEGDot(SmartGraph& g, SmartGraph::EdgeMap<double>& length,
ostream& out, SmartGraph::NodeMap<IndexPair>& edgePairs,
boost::unordered_set<unsigned>& matched)
{
out << "graph name {" << endl;
out << " node [ shape=ellipse, fontname=Helvetica, fontsize=10 ];" << endl;
for (SmartGraph::NodeIt n(g); n != INVALID; ++n)
{
IndexPair p = edgePairs[n];
if(p.first != p.second)
out << " n" << g.id(n) << " [ label=\"" << p.first << ":" << p.second << "\" ]; " << endl;
else
out << " n" << g.id(n) << " [ shape=triangle, label=\"" << p.first << "\" ]; \n";
}
out << " edge [ shape=ellipse, fontname=Helvetica, fontsize=10 ];" << endl;
for (SmartGraph::EdgeIt e(g); e != INVALID; ++e)
{
string extra = "";
if(matched.count(g.id(e)))
extra = "style = bold, ";
out << " n" << g.id(g.u(e)) << " -- " << " n" << g.id(g.v(e))
<< " [ " << extra << "label=\"" << length[e] << "\" ]; " << endl;
}
out << "}" << endl;
}
void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[])
{
// Verify input
char* inputError = "Expected 4 parameters\n N - number of nodes\n from - 1xM vector of indices\n to - 1xM vector of indices\n weight - 1xM vector of edge weights\n Example: [value, matches] = lemon_maxweightedperfectmatching(n, from, to, weight);";
if (nrhs != 4 || !mxIsNumeric(prhs[0]) || !mxIsNumeric(prhs[1]) || !mxIsNumeric(prhs[2]) || !mxIsNumeric(prhs[3]))
mexErrMsgTxt(inputError);
mwSize m = mxGetN(prhs[1]);
double* x = mxGetPr(prhs[1]);
double* y = mxGetPr(prhs[2]);
double* w = mxGetPr(prhs[3]);
mwSize n = (mwSize)mxGetScalar(prhs[0]);
if (mxGetM(prhs[1]) != 1 || mxGetM(prhs[2]) != 1 || mxGetM(prhs[3]) != 1 || mxGetN(prhs[2]) != m || mxGetN(prhs[3]) != m)
mexErrMsgTxt(inputError);
// Read input
SmartGraph g;
g.reserveNode(n);
g.reserveEdge(m);
typedef SmartGraph::EdgeMap<double> EdgeMap;
EdgeMap weight(g);
for (mwIndex i = 0; i < n; i++)
g.addNode();
for (mwIndex i = 0; i < m; i++)
{
SmartGraph::Edge edge = g.addEdge(g.nodeFromId(x[i] - 1), g.nodeFromId(y[i] - 1));
weight[edge] = w[i];
}
// Do stuff
MaxWeightedPerfectMatching<SmartGraph, EdgeMap> mwpm(g, weight);
mwpm.run();
// Create output
if (nlhs > 0)
{
plhs[0] = mxCreateDoubleMatrix(1, 1, mxREAL);
double *value = mxGetPr(plhs[0]);
value[0] = mwpm.matchingWeight();
}
if (nlhs > 1)
{
const SmartGraph::NodeMap<SmartGraph::Arc> &matchingMap = mwpm.matchingMap();
plhs[1] = mxCreateDoubleMatrix(1, n, mxREAL);
double *matches = mxGetPr(plhs[1]);
for (mwIndex i = 0; i < n; i++)
{
SmartGraph::Arc arc = matchingMap[g.nodeFromId(i)];
matches[i] = g.id(g.target(arc)) + 1;
}
}
return;
}
//write dot file from graph
void writeDot(SmartGraph& g, SmartGraph::EdgeMap<double>& length,
SmartGraph::NodeMap<unsigned>& index,
ostream& out)
{
out << "graph name {" << endl;
out << " node [ shape=ellipse, fontname=Helvetica, fontsize=10 ];" << endl;
for (SmartGraph::NodeIt n(g); n != INVALID; ++n)
{
out << " n" << g.id(n) << " [ label=\"" << index[n] << "\" ]; " << endl;
}
out << " edge [ shape=ellipse, fontname=Helvetica, fontsize=10 ];" << endl;
for (SmartGraph::EdgeIt e(g); e != INVALID; ++e)
{
out << " n" << g.id(g.u(e)) << " -- " << " n" << g.id(g.v(e))
<< " [ label=\"" << length[e] << "\" ]; " << endl;
}
out << "}" << endl;
}
