Merge branch 'pletscher'
[libdai.git] / src / alldai.cpp
1 /* This file is part of libDAI - http://www.libdai.org/
2 *
3 * libDAI is licensed under the terms of the GNU General Public License version
4 * 2, or (at your option) any later version. libDAI is distributed without any
5 * warranty. See the file COPYING for more details.
6 *
7 * Copyright (C) 2006-2009 Joris Mooij [joris dot mooij at libdai dot org]
8 * Copyright (C) 2006-2007 Radboud University Nijmegen, The Netherlands
9 */
10
11
12 #include <string>
13 #include <dai/alldai.h>
14 #include <dai/properties.h>
15 #include <dai/exceptions.h>
16
17
18 namespace dai {
19
20
21 using namespace std;
22
23
24 InfAlg *newInfAlg( const std::string &name, const FactorGraph &fg, const PropertySet &opts ) {
25 if( name == ExactInf::Name )
26 return new ExactInf (fg, opts);
27 #ifdef DAI_WITH_BP
28 if( name == BP::Name )
29 return new BP (fg, opts);
30 #endif
31 #ifdef DAI_WITH_MF
32 if( name == MF::Name )
33 return new MF (fg, opts);
34 #endif
35 #ifdef DAI_WITH_HAK
36 if( name == HAK::Name )
37 return new HAK (fg, opts);
38 #endif
39 #ifdef DAI_WITH_LC
40 if( name == LC::Name )
41 return new LC (fg, opts);
42 #endif
43 #ifdef DAI_WITH_TREEEP
44 if( name == TreeEP::Name )
45 return new TreeEP (fg, opts);
46 #endif
47 #ifdef DAI_WITH_JTREE
48 if( name == JTree::Name )
49 return new JTree (fg, opts);
50 #endif
51 #ifdef DAI_WITH_MR
52 if( name == MR::Name )
53 return new MR (fg, opts);
54 #endif
55 #ifdef DAI_WITH_GIBBS
56 if( name == Gibbs::Name )
57 return new Gibbs (fg, opts);
58 #endif
59 #ifdef DAI_WITH_CBP
60 if( name == CBP::Name )
61 return new CBP (fg, opts);
62 #endif
63 DAI_THROWE(UNKNOWN_DAI_ALGORITHM,"Unknown libDAI algorithm: " + name);
64 }
65
66
67 /// \todo Make alias file non-testdai-specific, and use it in newInfAlgFromString
68 InfAlg *newInfAlgFromString( const std::string &nameOpts, const FactorGraph &fg ) {
69 string::size_type pos = nameOpts.find_first_of('[');
70 string name;
71 PropertySet opts;
72 if( pos == string::npos ) {
73 name = nameOpts;
74 } else {
75 name = nameOpts.substr(0,pos);
76
77 stringstream ss;
78 ss << nameOpts.substr(pos,nameOpts.length());
79 ss >> opts;
80 }
81 return newInfAlg(name,fg,opts);
82 }
83
84
85 } // end of namespace dai