Pervasive change of BipartiteGraph implementation
[libdai.git] / include / dai / mf.h
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 #ifndef __defined_libdai_mf_h
23 #define __defined_libdai_mf_h
24
25
26 #include <string>
27 #include <dai/daialg.h>
28 #include <dai/factorgraph.h>
29
30
31 namespace dai {
32
33
34 class MF : public DAIAlgFG {
35 protected:
36 std::vector<Factor> _beliefs;
37
38 public:
39 // default constructor
40 MF() : DAIAlgFG(), _beliefs() {};
41 // copy constructor
42 MF(const MF & x) : DAIAlgFG(x), _beliefs(x._beliefs) {};
43 MF* clone() const { return new MF(*this); }
44 // construct MF object from FactorGraph
45 MF(const FactorGraph & fg, const Properties &opts) : DAIAlgFG(fg, opts) {
46 assert( checkProperties() );
47 Regenerate();
48 }
49 // assignment operator
50 MF & operator=(const MF & x) {
51 if(this!=&x) {
52 DAIAlgFG::operator=(x);
53 _beliefs = x._beliefs;
54 }
55 return *this;
56 }
57
58 static const char *Name;
59 std::string identify() const;
60 void Regenerate();
61 void init();
62 double run();
63 Factor beliefV (size_t i) const;
64 Factor belief (const Var &n) const;
65 Factor belief (const VarSet &ns) const;
66 std::vector<Factor> beliefs() const;
67 Complex logZ() const;
68
69 void init( const VarSet &ns );
70 void undoProbs( const VarSet &ns ) { FactorGraph::undoProbs(ns); init(ns); }
71 bool checkProperties();
72 };
73
74
75 } // end of namespace dai
76
77
78 #endif