bp_dual.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_bp_dual_h
#define __defined_libdai_bp_dual_h


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


#include <dai/daialg.h>
#include <dai/factorgraph.h>
#include <dai/enum.h>


namespace dai {



class BP_dual {
    protected:
        template<class T>
        struct _edges_t : public std::vector<std::vector<T> > {};

        struct messages {
            _edges_t<Prob> n;
            _edges_t<Real> Zn;
            _edges_t<Prob> m;
            _edges_t<Real> Zm;
        };
        messages _msgs;

        struct beliefs {
            std::vector<Prob> b1;
            std::vector<Real> Zb1;
            std::vector<Prob> b2;
            std::vector<Real> Zb2;
        };
        beliefs _beliefs;

        const InfAlg *_ia;

        void init();
        void regenerateMessages();
        void regenerateBeliefs();

        void calcMessages();
        void calcNewM(size_t i, size_t _I);
        void calcNewN(size_t i, size_t _I);

        void calcBeliefs();
        void calcBeliefV(size_t i);
        void calcBeliefF(size_t I);

    public:

        BP_dual( const InfAlg *ia ) : _ia(ia) { init(); }

        const FactorGraph& fg() const { return _ia->fg(); }

        Prob & msgM( size_t i, size_t _I ) { return _msgs.m[i][_I]; }
        const Prob & msgM( size_t i, size_t _I ) const { return _msgs.m[i][_I]; }
        Prob & msgN( size_t i, size_t _I ) { return _msgs.n[i][_I]; }
        const Prob & msgN( size_t i, size_t _I ) const { return _msgs.n[i][_I]; }
        Real & zM( size_t i, size_t _I ) { return _msgs.Zm[i][_I]; }
        const Real & zM( size_t i, size_t _I ) const { return _msgs.Zm[i][_I]; }
        Real & zN( size_t i, size_t _I ) { return _msgs.Zn[i][_I]; }
        const Real & zN( size_t i, size_t _I ) const { return _msgs.Zn[i][_I]; }

        Factor beliefV( size_t i ) const { return Factor( _ia->fg().var(i), _beliefs.b1[i] ); }
        Factor beliefF( size_t I ) const { return Factor( _ia->fg().factor(I).vars(), _beliefs.b2[I] ); }

        Real beliefVZ( size_t i ) const { return _beliefs.Zb1[i]; }
        Real beliefFZ( size_t I ) const { return _beliefs.Zb2[I]; }
};


} // end of namespace dai


#endif


#endif