GraphLayeredExt.h

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

#include "GraphExt.h"
#include <optional>

namespace DirectGraphicalModels
{
    class IGraphPairwise;
    class CTrainEdge;
    class CTrainLink;

    enum graphEdgesType {
        GRAPH_EDGES_NONE = 0,       
        GRAPH_EDGES_GRID = 1,       
        GRAPH_EDGES_DIAG = 2,       
        GRAPH_EDGES_LINK = 4        
    };

    // ================================ Extended Graph Class ================================
    class CGraphLayeredExt : public CGraphExt
    {
    public:
        DllExport CGraphLayeredExt(IGraphPairwise& graph, word nLayers, byte gType = GRAPH_EDGES_GRID) : m_graph(graph), m_nLayers(nLayers), m_gType(gType), m_size(Size(0, 0)) {}
        DllExport virtual ~CGraphLayeredExt(void) = default;

        // From CGraphExt
        DllExport void buildGraph(Size graphSize) override;
        DllExport void setGraph(const Mat& pots) override;
        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_size; }

        DllExport void setGraph(const Mat &potBase, const Mat &potOccl);
        DllExport void addFeatureVecs(CTrainEdge &edgeTrainer, const Mat &featureVectors, const Mat &gt);
        DllExport void addFeatureVecs(CTrainEdge &edgeTrainer, const vec_mat_t &featureVectors, const Mat &gt);
        DllExport void fillEdges(const CTrainEdge &edgeTrainer, const CTrainLink* linkTrainer, const Mat &featureVectors, const vec_float_t &vParams, float edgeWeight = 1.0f, float linkWeight = 1.0f);
        DllExport void fillEdges(const CTrainEdge &edgeTrainer, const CTrainLink* linkTrainer, const vec_mat_t &featureVectors, const vec_float_t &vParams, float edgeWeight = 1.0f, float linkWeight = 1.0f);
        DllExport void defineEdgeGroup(float A, float B, float C, byte group);
        DllExport void setEdges(std::optional<byte> group, const Mat &pot);
        DllExport byte getType(void) const { return m_gType; }
        IGraphPairwise& getGraph(void) const { return m_graph; }


    private:
        IGraphPairwise& m_graph;        
        const word      m_nLayers;      
        const byte      m_gType;        
        Size            m_size;         
    };
}