IGraphPairwise.h

// Graph interface class;
// Written by Sergey Kosov in 2015 for Project X
#pragma once

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

namespace DirectGraphicalModels {
    // ================================ Graph Interface Class ================================
    class IGraphPairwise : public CGraph
    {
    public:
        DllExport IGraphPairwise(byte nStates) : CGraph(nStates) {}
        DllExport virtual ~IGraphPairwise(void) = default;

        DllExport void              marginalize(const vec_size_t &nodes);
        DllExport void              addEdge(size_t srcNode, size_t dstNode, const Mat &pot = EmptyMat);
        DllExport virtual void      addEdge(size_t srcNode, size_t dstNode, byte group, const Mat &pot) = 0;
        DllExport virtual void      setEdge(size_t srcNode, size_t dstNode, const Mat &pot) = 0;
        DllExport virtual void      setEdges(std::optional<byte> group, const Mat& pot) = 0;
        DllExport virtual void      getEdge(size_t srcNode, size_t dstNode, Mat &pot) const = 0;
        DllExport virtual void      setEdgeGroup(size_t srcNode, size_t dstNode, byte group) = 0;
        DllExport virtual byte      getEdgeGroup(size_t srcNode, size_t dstNode) const = 0;
        DllExport virtual void      removeEdge(size_t srcNode, size_t dstNode) = 0;
        DllExport virtual bool      isEdgeExists(size_t srcNode, size_t dstNode) const = 0;
        DllExport virtual bool      isEdgeArc(size_t srcNode, size_t dstNode) const;
        DllExport void              addArc(size_t Node1, size_t Node2, const Mat &pot = EmptyMat);
        DllExport virtual void      addArc(size_t Node1, size_t Node2, byte group, const Mat &pot);
        DllExport virtual void      setArc(size_t Node1, size_t Node2, const Mat &pot);
        DllExport virtual void      setArcGroup(size_t Node1, size_t Node2, byte group);
        DllExport virtual void      removeArc(size_t Node1, size_t Node2);
        DllExport virtual bool      isArcExists(size_t Node1, size_t Node2) const;
    };
}