GraphDenseKit.h

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

#include "GraphKit.h"
#include "InferDense.h"
#include "GraphDenseExt.h"

namespace DirectGraphicalModels
{
    // ================================ Dense Graph Kit Class ===============================
    class CGraphDenseKit : public CGraphKit {
    public:
        DllExport CGraphDenseKit(byte nStates)
            : CGraphKit()
            , m_graph(nStates)
            , m_infer(m_graph)
            , m_graphExtension(m_graph)
        {}
        DllExport virtual ~CGraphDenseKit() = default;

        DllExport CGraph&       getGraph() override { return m_graph; }
        DllExport CInfer&       getInfer() override { return m_infer; }
        DllExport CGraphExt&    getGraphExt() override { return m_graphExtension; }


    private:
        CGraphDense     m_graph;            
        CInferDense     m_infer;            
        CGraphDenseExt  m_graphExtension;   
    };
}