yap-6.3/packages/CLPBN/clpbn/bp/CbpSolver.h

#ifndef HORUS_CBP_H
#define HORUS_CBP_H

#include "FgBpSolver.h"
#include "CFactorGraph.h"

class Factor;

class CbpSolverLink : public SpLink
{
  public:
    CbpSolverLink (FgFacNode* fn, FgVarNode* vn, unsigned c) : SpLink (fn, vn)
    {
      edgeCount_ = c;
      poweredMsg_.resize (vn->nrStates(), Util::one());
    }

    void updateMessage (void) 
    {
      poweredMsg_ = *nextMsg_;
      swap (currMsg_, nextMsg_);
      msgSended_  = true;
      Util::pow (poweredMsg_, edgeCount_);
    }
  
    unsigned       getNumberOfEdges  (void) const { return edgeCount_; }
    const Params&  getPoweredMessage (void) const { return poweredMsg_; }

  private:
    Params poweredMsg_;
    unsigned edgeCount_;
};



class CbpSolver : public FgBpSolver
{
  public:
    CbpSolver (FactorGraph& fg) : FgBpSolver (fg) { }
   ~CbpSolver (void);

    Params        getPosterioriOf (VarId);
    Params        getJointDistributionOf (const VarIds&);

   private:
     void           initializeSolver (void);
     void           createLinks (void);

     void           maxResidualSchedule (void);
     Params       getVar2FactorMsg (const SpLink*) const;
     void           printLinkInformation (void) const;


     CFactorGraph*      lfg_;
};

#endif // HORUS_CBP_H