Added Exceptions framework (and more)
[libdai.git] / matlab / dai_removeshortloops.cpp
1 /* Copyright (C) 2006-2008 Joris Mooij [j dot mooij at science dot ru dot nl]
2 Radboud University Nijmegen, The Netherlands
3
4 This file is part of libDAI.
5
6 libDAI is free software; you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation; either version 2 of the License, or
9 (at your option) any later version.
10
11 libDAI is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 GNU General Public License for more details.
15
16 You should have received a copy of the GNU General Public License
17 along with libDAI; if not, write to the Free Software
18 Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
19 */
20
21
22 /*=================================================================*
23 * *
24 * This is a MEX-file for MATLAB. *
25 * *
26 * [psi_out] = dai_removeshortloops(psi_in); *
27 * *
28 *=================================================================*/
29
30
31 #include <iostream>
32 #include "mex.h"
33 #include "matlab.h"
34 #include "factorgraph.h"
35
36
37 using namespace std;
38
39
40 /* Input Arguments */
41
42 #define PSI_IN prhs[0]
43 #define NR_IN 1
44
45
46 /* Output Arguments */
47
48 #define PSI_OUT plhs[0]
49 #define NR_OUT 1
50
51
52 void mexFunction( int nlhs, mxArray *plhs[], int nrhs, const mxArray*prhs[] ) {
53 // Check for proper number of arguments
54 if ((nrhs != NR_IN) || (nlhs != NR_OUT)) {
55 mexErrMsgTxt("Usage: [psi_out] = dai_removeshortloops(psi_in);\n\n"
56 "\n"
57 "INPUT: psi_in = linear cell array containing the factors\n"
58 " (psi{i} is a structure with a Member field\n"
59 " and a P field, like a CPTAB).\n"
60 "\n"
61 "OUTPUT: psi_out = linear cell array containing the factors of psi_in,\n"
62 " where factors have been merged to prevent short loops\n"
63 " of length 4 in the factor graph (i.e. loops of type\n"
64 " var1-factor1-var2-factor2-var1),\n");
65 }
66
67 // Get the factors from PSI_IN
68 vector<Factor> psi = mx2Factors(PSI_IN, 0);
69
70 // Remove the short loops
71 RemoveShortLoops(psi);
72
73 // Hand over results to MATLAB
74 PSI_OUT = Factors2mx(psi);
75
76
77 return;
78 }