trwbp.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_trwbp_h
#define __defined_libdai_trwbp_h


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


#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 TRWBP : public BP {
    protected:

        std::vector<Real> _weight;

    public:

        size_t nrtrees;

    public:

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


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


        virtual TRWBP* clone() const { return new TRWBP(*this); }
        virtual TRWBP* construct( const FactorGraph &fg, const PropertySet &opts ) const { return new TRWBP( fg, opts ); }
        virtual std::string name() const { return "TRWBP"; }
        virtual Real logZ() const;
        virtual void setProperties( const PropertySet &opts );
        virtual PropertySet getProperties() const;
        virtual std::string printProperties() 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; }

        void addTreeToWeights( const RootedTree &tree );

        void sampleWeights( size_t nrTrees );

    protected:

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

        virtual void calcBeliefV( size_t i, Prob &p ) const;

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