InferLBP.h

// Loopy Belief Propagation inference class interface
// Written by Sergey G. Kosov in 2013 for Chronos Vision GmbH
// Adopted by Sergey G. Kosov in 2015 for Project X
#pragma once

#include "MessagePassing.h"

namespace DirectGraphicalModels
{
    // ==================== Loopy Belief Propagation Infer Class ==================
    class CInferLBP : public CMessagePassing
    {
    public:
        DllExport CInferLBP(CGraphPairwise &graph) : CMessagePassing(graph), m_maxSum(false) {}
        DllExport virtual ~CInferLBP(void) {}


    protected:
        DllExport virtual void  calculateMessages(unsigned int nIt);
        void                    setMaxSum(bool maxSum) { m_maxSum = maxSum; }
        bool                    isMaxSum(void) const { return m_maxSum; }


    private:
        bool m_maxSum;          
    };

}