GraphPairwiseKit.h

// Kit class for constructing the Dense Pairwise objects
// Written by Sergey Kosov in 2018 - 2019 for Project X
#pragma once

#include "GraphKit.h"

#include "GraphPairwise.h"

#include "MessagePassing.h"
#include "InferLBP.h"
#include "InferTRW.h"
#include "InferViterbi.h"

#include "GraphPairwiseExt.h"

#include "macroses.h"

namespace DirectGraphicalModels
{
    enum class INFER { 
        LBP,        
        TRW,        
        Viterbi     
    };

    // ================================ Pairwise Graph Kit Class ===============================
    class CGraphPairwiseKit : public CGraphKit {
    public:
        DllExport CGraphPairwiseKit(byte nStates, INFER infer = INFER::LBP)
            : CGraphKit()
            , m_graph(nStates)
            , m_graphExtension(m_graph)
        {
            switch (infer)
            {
            case INFER::LBP:     m_pInfer = std::make_unique<CInferLBP>(m_graph); break;
            case INFER::TRW:     m_pInfer = std::make_unique<CInferTRW>(m_graph); break;
            case INFER::Viterbi: m_pInfer = std::make_unique<CInferViterbi>(m_graph); break;
            default: DGM_ASSERT_MSG(false, "Unknown inference method");
            }
        }
        DllExport virtual ~CGraphPairwiseKit() = default;
 
        DllExport CGraph&       getGraph() override { return m_graph; }
        DllExport CInfer&       getInfer() override { return *m_pInfer; }
        DllExport CGraphExt&    getGraphExt() override { return m_graphExtension; }


    private:
        CGraphPairwise                      m_graph;                
        std::unique_ptr<CMessagePassing>    m_pInfer;               
        CGraphPairwiseExt                   m_graphExtension;       
    };
}