PriorNode.h

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

#include "Prior.h"

namespace DirectGraphicalModels
{
// ================================ Node Prior Class ================================
    class CPriorNode : public CPrior
    {
    public:
        DllExport CPriorNode(byte nStates) : CBaseRandomModel(nStates), CPrior(nStates, RM_UNARY)  {}
        DllExport ~CPriorNode(void) {}

        DllExport void  addNodeGroundTruth(const Mat &gt);
        DllExport void  addNodeGroundTruth(byte gt);


    protected:
        DllExport Mat   calculatePrior(void) const;
    };
}