TrainEdgePrior.h

// Prior training model for pairwise potentials
// Written by Sergey G. Kosov in 2015 for Project X
#pragma once

#include "TrainEdgePottsCS.h"
#include "PriorEdge.h"

namespace DirectGraphicalModels
{
    // ============================= Prior Edge Train Class =============================
    class CTrainEdgePrior : public CTrainEdgePottsCS, private CPriorEdge
    {
    public:
        DllExport CTrainEdgePrior(byte nStates, word nFeatures, ePotPenalApproach penApproach = eP_APP_PEN_EXP, ePotNormApproach normApproach = eP_APP_NORM_SYMMETRIC);
        DllExport virtual ~CTrainEdgePrior(void);

        DllExport virtual void  reset(void);    

        DllExport virtual void  addFeatureVecs(const Mat &featureVector1, byte gt1, const Mat &featureVector2, byte gt2);
        DllExport virtual void  train(bool doClean = false);

    protected:
        DllExport virtual void  saveFile(FILE *pFile) const;
        DllExport virtual void  loadFile(FILE *pFile);      
        DllExport virtual Mat   calculateEdgePotentials(const Mat &featureVector1, const Mat &featureVector2, const vec_float_t &vParams) const;


    private:
        inline void             loadPriorMatrix(void);
        Mat                     m_prior;                    // = Mat(); // The prior matrix
    };
}