PriorEdge.h

// Edge prior probability estimation class
// Written by Sergey G. Kosov in 2015 for Project X
#pragma once

#include "Prior.h"

namespace DirectGraphicalModels
{
    enum ePotNormApproach {
        eP_APP_NORM_STANDARD,   
        eP_APP_NORM_SYMMETRIC,  
        eP_APP_NORM_ASYMMETRIC  
    };  

    // ================================ Edge Prior Class ================================
    class CPriorEdge : public CPrior
    {
    public:
        DllExport CPriorEdge(byte nStates, ePotNormApproach normApp = eP_APP_NORM_SYMMETRIC) : CBaseRandomModel(nStates), CPrior(nStates, RM_PAIRWISE), m_normApp(normApp) {}
        DllExport ~CPriorEdge(void) {}

        DllExport void          addEdgeGroundTruth(byte gt1, byte gt2); 

        
        
    protected:      
        DllExport Mat           calculatePrior(void) const;


    private:
        ePotNormApproach        m_normApp;      
    };
}