#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->range(), LogAware::one());
    }

    unsigned getNumberOfEdges  (void) const { return edgeCount_; }

    const Params& getPoweredMessage (void) const { return poweredMsg_; }

    void updateMessage (void) 
    {
      poweredMsg_ = *nextMsg_;
      swap (currMsg_, nextMsg_);
      msgSended_  = true;
      LogAware::pow (poweredMsg_, edgeCount_);
    }
  
  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