GraphDenseExt.h

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

#include "GraphExt.h"

namespace DirectGraphicalModels 
{
    class CGraphDense;
    // ================================ Extended Dense Graph Class ================================
    class CGraphDenseExt : public CGraphExt
    {
    public:
        DllExport CGraphDenseExt(CGraphDense &graph) : m_graph(graph) {}
        DllExport ~CGraphDenseExt(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
        {
            addGaussianEdgeModel(Vec2f::all(val * 0.03f), weight);
        }
        DllExport void addDefaultEdgesModel(const Mat& featureVectors, float val, float weight = 1.0f) override
        {
            addBilateralEdgeModel(featureVectors, Vec2f::all(val * 0.03f), 20.0f, weight);
        }
        DllExport void addDefaultEdgesModel(const vec_mat_t &featureVectors, float val, float weight = 1.0f) override
        {
            addBilateralEdgeModel(featureVectors, Vec2f::all(val * 0.03f), 20.0f, weight);
        }
        DllExport Size getSize(void) const override { return m_size; }

        
        DllExport void addGaussianEdgeModel(Vec2f sigma, float weight = 1.0f, const std::function<void(const Mat& src, Mat& dst)>& semiMetricFunction = {});
        DllExport void addBilateralEdgeModel(const Mat& featureVectors, Vec2f sigma, float sigma_opt = 1.0f, float weight = 1.0f, const std::function<void(const Mat& src, Mat& dst)>& semiMetricFunction = {});
        DllExport void addBilateralEdgeModel(const vec_mat_t& featureVectors, Vec2f sigma, float sigma_opt = 1.0f, float weight = 1.0f, const std::function<void(const Mat& src, Mat& dst)>& semiMetricFunction = {});


    private:
        CGraphDense& m_graph;   
        Size         m_size;    
    };
}