MessagePassing.h

// Base abstract class for message passing algorithms used for exact and approximate inference
// Written by Sergey G. Kosov in 2015 for Project X
#pragma once

#include "Infer.h"
#include "GraphPairwise.h"

namespace DirectGraphicalModels
{
    struct Edge;

    // ==================== Message Passing Base Abstract Class ==================
    class CMessagePassing : public CInfer
    {
    public:
        DllExport CMessagePassing(CGraphPairwise &graph) : CInfer(graph) {}
        DllExport virtual ~CMessagePassing(void) {}
        
        DllExport virtual void    infer(unsigned int nIt = 1);


    protected:
        CGraphPairwise & getGraphPairwise(void) const { return dynamic_cast<CGraphPairwise &>(getGraph()); }
        virtual void calculateMessages(unsigned int nIt) = 0;
        void calculateMessage(Edge *edge, float *temp, float *&dst, bool maxSum = false);
        void createMessages(void);
        void deleteMessages(void);
        void swapMessages(void);
        static float MatMul(const Mat &M, const float *v, float *&dst, bool maxSum = false);
    };
}