InferTRW.h

// Tree-reweighted inference class interface
// Written by Sergey G. Kosov in 2016 for Project X
#pragma once

#include "MessagePassing.h"

namespace DirectGraphicalModels
{
    struct Edge;
    
    // ==================== Microsoft TRW Decode Class ==================
    class CInferTRW : public CMessagePassing
    {
    public:
        DllExport CInferTRW(CGraphPairwise &graph) : CMessagePassing(graph) {}
        DllExport virtual ~CInferTRW(void) {}

        DllExport virtual void infer(unsigned int nIt = 1);


    protected:
        DllExport virtual void  calculateMessages(unsigned int nIt);
        void                    calculateMessage(Edge &edge, float *temp, float *data);
    };
}