Changed license from GPL v2+ to FreeBSD (aka BSD 2-clause) license
[libdai.git] / include / dai / bp_dual.h
index 0cd2e45..8357771 100644 (file)
@@ -1,26 +1,15 @@
-/*  Copyright (C) 2009  Frederik Eaton [frederik at ofb dot net]
-
-    This file is part of libDAI.
-
-    libDAI is free software; you can redistribute it and/or modify
-    it under the terms of the GNU General Public License as published by
-    the Free Software Foundation; either version 2 of the License, or
-    (at your option) any later version.
-
-    libDAI is distributed in the hope that it will be useful,
-    but WITHOUT ANY WARRANTY; without even the implied warranty of
-    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-    GNU General Public License for more details.
-
-    You should have received a copy of the GNU General Public License
-    along with libDAI; if not, write to the Free Software
-    Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
-*/
+/*  This file is part of libDAI - http://www.libdai.org/
+ *
+ *  Copyright (c) 2006-2011, The libDAI authors. All rights reserved.
+ *
+ *  Use of this source code is governed by a BSD-style license that can be found in the LICENSE file.
+ */
 
 
 /// \file
-/// \brief Defines class BP_dual
-/// \todo This replicates a large part of the functionality of BP; would it not be shorter to adapt BP instead?
+/// \brief Defines class BP_dual, which is used primarily by BBP.
+/// \idea BP_dual replicates a large part of the functionality of BP; would it not be more efficient to adapt BP instead?
+/// \author Frederik Eaton
 
 
 #ifndef __defined_libdai_bp_dual_h
@@ -40,9 +29,10 @@ namespace dai {
  *  to variables and messages from variables to factors), and normalizers, given an InfAlg.
  *  These are computed from the variable and factor beliefs of the InfAlg.
  *  This class is used primarily by BBP.
+ *
+ *  \author Frederik Eaton
  */
 class BP_dual {
-
     protected:
         /// Convenience label for storing edge properties
         template<class T>
@@ -81,28 +71,28 @@ class BP_dual {
 
         /// Does all necessary preprocessing
         void init();
-        /// Allocates space for _msgs
+        /// Allocates space for \a _msgs
         void regenerateMessages();
-        /// Allocates space for _beliefs
+        /// Allocates space for \a _beliefs
         void regenerateBeliefs();
 
         /// Calculate all messages from InfAlg beliefs
         void calcMessages();
-        /// Update factor->variable message (i->I)
+        /// Update factor->variable message (\a i -> \a I)
         void calcNewM(size_t i, size_t _I);
-        /// Update variable->factor message (I->i)
+        /// Update variable->factor message (\a I -> \a i)
         void calcNewN(size_t i, size_t _I);
 
         /// Calculate all variable and factor beliefs from messages
         void calcBeliefs();
-        /// Calculate variable belief
+        /// Calculate belief of variable \a i
         void calcBeliefV(size_t i);
-        /// Calculate factor belief
+        /// Calculate belief of factor \a I
         void calcBeliefF(size_t I);
 
     public:
         /// Construct BP_dual object from (converged) InfAlg object's beliefs and factors.
-        /*  A pointer to the the InfAlg object is stored,
+        /** \warning A pointer to the the InfAlg object is stored,
          *  so the object must not be destroyed before the BP_dual is destroyed.
          */
         BP_dual( const InfAlg *ia ) : _ia(ia) { init(); }
@@ -110,23 +100,31 @@ class BP_dual {
         /// Returns the underlying FactorGraph
         const FactorGraph& fg() const { return _ia->fg(); }
 
-        /// Returns factor -> var message (I->i)
-        DAI_ACCMUT(Prob & msgM( size_t i, size_t _I ), { return _msgs.m[i][_I]; });
-        /// Returns var -> factor message (i->I)
-        DAI_ACCMUT(Prob & msgN( size_t i, size_t _I ), { return _msgs.n[i][_I]; });
-        /// Returns normalizer for msgM
-        DAI_ACCMUT(Real & zM( size_t i, size_t _I ), { return _msgs.Zm[i][_I]; });
-        /// Returns normalizer for msgN
-        DAI_ACCMUT(Real & zN( size_t i, size_t _I ), { return _msgs.Zn[i][_I]; });
-
-        /// Returns variable belief
+        /// Returns reference to factor->variable message (\a I -> \a i)
+        Prob & msgM( size_t i, size_t _I ) { return _msgs.m[i][_I]; }
+        /// Returns constant reference to factor->variable message (\a I -> \a i)
+        const Prob & msgM( size_t i, size_t _I ) const { return _msgs.m[i][_I]; }
+        /// Returns reference to variable -> factor message (\a i -> \a I)
+        Prob & msgN( size_t i, size_t _I ) { return _msgs.n[i][_I]; }
+        /// Returns constant reference to variable -> factor message (\a i -> \a I)
+        const Prob & msgN( size_t i, size_t _I ) const { return _msgs.n[i][_I]; }
+        /// Returns reference to normalizer for factor->variable message (\a I -> \a i)
+        Real & zM( size_t i, size_t _I ) { return _msgs.Zm[i][_I]; }
+        /// Returns constant reference to normalizer for factor->variable message (\a I -> \a i)
+        const Real & zM( size_t i, size_t _I ) const { return _msgs.Zm[i][_I]; }
+        /// Returns reference to normalizer for variable -> factor message (\a i -> \a I)
+        Real & zN( size_t i, size_t _I ) { return _msgs.Zn[i][_I]; }
+        /// Returns constant reference to normalizer for variable -> factor message (\a i -> \a I)
+        const Real & zN( size_t i, size_t _I ) const { return _msgs.Zn[i][_I]; }
+
+        /// Returns belief of variable \a i
         Factor beliefV( size_t i ) const { return Factor( _ia->fg().var(i), _beliefs.b1[i] ); }
-        /// Returns factor belief
+        /// Returns belief of factor \a I
         Factor beliefF( size_t I ) const { return Factor( _ia->fg().factor(I).vars(), _beliefs.b2[I] ); }
 
-        /// Returns normalizer for variable belief
+        /// Returns normalizer for belief of variable \a i
         Real beliefVZ( size_t i ) const { return _beliefs.Zb1[i]; }
-        /// Returns normalizer for factor belief
+        /// Returns normalizer for belief of factor \a I
         Real beliefFZ( size_t I ) const { return _beliefs.Zb2[I]; }
 };