#ifndef HORUS_CBP_H
#define HORUS_CBP_H

#include "BpSolver.h"
#include "CFactorGraph.h"

class Factor;

class CbpSolverLink : public SpLink
{
  public:
    CbpSolverLink (FacNode* fn, VarNode* vn, size_t idx, unsigned count) 
        : SpLink (fn, vn), index_(idx), nrEdges_(count),
          pwdMsg_(vn->range(), LogAware::one()) { }

    size_t index (void) const { return index_; }

    unsigned nrEdges (void) const { return nrEdges_; }

    const Params& powMessage (void) const { return pwdMsg_; }

    void updateMessage (void) 
    {
      pwdMsg_ = *nextMsg_;
      swap (currMsg_, nextMsg_);
      LogAware::pow (pwdMsg_, nrEdges_);
    }
  
  private:
    size_t    index_;
    unsigned  nrEdges_;
    Params    pwdMsg_;
};



class CbpSolver : public BpSolver
{
  public:   
    CbpSolver (const FactorGraph& fg);

   ~CbpSolver (void);

    void printSolverFlags (void) const;
  
    Params getPosterioriOf (VarId);

    Params getJointDistributionOf (const VarIds&);

   private:

     void createLinks (void);

     void maxResidualSchedule (void);

     void calcFactorToVarMsg (SpLink*);

     Params getVarToFactorMsg (const SpLink*) const;

     void printLinkInformation (void) const;

     CFactorGraph* cfg_;
};

#endif // HORUS_CBP_H