GraphPairwiseExt.h

// Extended (pairwise) Graph class interface;
// Written by Sergey Kosov in 2016 for Project X
#pragma once

#include "GraphExt.h"
#include "GraphLayeredExt.h"

namespace DirectGraphicalModels 
{
    class IGraphPairwise;
    
    // ================================ Extended Pairwise Graph Class ================================
    class CGraphPairwiseExt : public CGraphExt
    {
    public:
        DllExport CGraphPairwiseExt(IGraphPairwise& graph, byte gType = GRAPH_EDGES_GRID)
            : m_pGraphLayeredExt(std::make_unique<CGraphLayeredExt>(graph, 1, gType))
        {}
        DllExport virtual ~CGraphPairwiseExt(void) = default;

        // From CGraphExt
        DllExport void buildGraph(Size graphSize) override
        {
            m_pGraphLayeredExt->buildGraph(graphSize);
        }
        DllExport void setGraph(const Mat& pots) override
        {
            m_pGraphLayeredExt->setGraph(pots, Mat());
        }
        DllExport void addDefaultEdgesModel(float val, float weight = 1.0f) override;
        DllExport void addDefaultEdgesModel(const Mat& featureVectors, float val, float weight = 1.0f) override;
        DllExport void addDefaultEdgesModel(const vec_mat_t& featureVectors, float val, float weight = 1.0f) override;
        DllExport Size getSize() const override 
        { 
            return m_pGraphLayeredExt->getSize();
        }

        DllExport void addFeatureVecs(CTrainEdge &edgeTrainer, const Mat &featureVectors, const Mat &gt)
        {
            m_pGraphLayeredExt->addFeatureVecs(edgeTrainer, featureVectors, gt);
        }
        DllExport void addFeatureVecs(CTrainEdge &edgeTrainer, const vec_mat_t &featureVectors, const Mat &gt)
        {
            m_pGraphLayeredExt->addFeatureVecs(edgeTrainer, featureVectors, gt);
        }
        DllExport void fillEdges(const CTrainEdge& edgeTrainer, const Mat& featureVectors, const vec_float_t& vParams, float weight = 1.0f)
        {
            m_pGraphLayeredExt->fillEdges(edgeTrainer, NULL, featureVectors, vParams, weight);
        }
        DllExport void fillEdges(const CTrainEdge& edgeTrainer, const vec_mat_t& featureVectors, const vec_float_t& vParams, float weight = 1.0f)
        {
            m_pGraphLayeredExt->fillEdges(edgeTrainer, NULL, featureVectors, vParams, weight);
        }
        DllExport void defineEdgeGroup(float A, float B, float C, byte group)
        {
            m_pGraphLayeredExt->defineEdgeGroup(A, B, C, group);
        }
        DllExport void setEdges(std::optional<byte> group, const Mat &pot)
        {
            m_pGraphLayeredExt->setEdges(group, pot);
        }
        DllExport byte getType(void) const 
        { 
            return m_pGraphLayeredExt->getType();
        }
        
        
    private:
        std::unique_ptr<CGraphLayeredExt>   m_pGraphLayeredExt;
    };
}