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