fbp.h

Go to the documentation of this file.

/*  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.
 */




#ifndef __defined_libdai_fbp_h
#define __defined_libdai_fbp_h


#include <dai/dai_config.h>
#ifdef DAI_WITH_FBP


#include <string>
#include <dai/daialg.h>
#include <dai/factorgraph.h>
#include <dai/properties.h>
#include <dai/enum.h>
#include <dai/bp.h>


namespace dai {



class FBP : public BP {
    protected:
        std::vector<Real> _weight;

    public:

        FBP() : BP(), _weight() {}


        FBP( const FactorGraph &fg, const PropertySet &opts ) : BP(fg, opts), _weight() {
            setProperties( opts );
            construct();
        }


        virtual FBP* clone() const { return new FBP(*this); }
        virtual FBP* construct( const FactorGraph &fg, const PropertySet &opts ) const { return new FBP( fg, opts ); }
        virtual std::string name() const { return "FBP"; }
        virtual Real logZ() const;


        Real Weight( size_t I ) const { return _weight[I]; }

        const std::vector<Real>& Weights() const { return _weight; }

        void setWeight( size_t I, Real c ) { _weight[I] = c; }


        void setWeights( const std::vector<Real> &c ) { _weight = c; }

    protected:

        virtual Prob calcIncomingMessageProduct( size_t I, bool without_i, size_t i ) const;

        // Calculate the updated message from the \a _I 'th neighbor of variable \a i to variable \a i
        virtual void calcNewMessage( size_t i, size_t _I );

        // Calculates unnormalized belief of factor \a I
        virtual void calcBeliefF( size_t I, Prob &p ) const {
            p = calcIncomingMessageProduct( I, false, 0 );
        }

        // Helper function for constructors
        virtual void construct();
};


} // end of namespace dai


#endif


#endif