TrainEdge.h

// Base abstract class for random model edges training
// Written by Sergey G. Kosov in 2013-2015 for Project X
#pragma once

#include "ITrain.h"

namespace DirectGraphicalModels
{
    enum EdgeRandomModel : byte { 
        Potts = 0,      
        PottsCS,        
        Prior,          
        Concat          
    };

    // ============================= Edge Train Class =============================
    class CTrainEdge : public ITrain
    {
    public:
        DllExport CTrainEdge(byte nStates, word nFeatures) : CBaseRandomModel(nStates), ITrain(nStates, nFeatures) {}
        DllExport virtual ~CTrainEdge(void) = default;

        DllExport static std::shared_ptr<CTrainEdge> create(byte edgeRandomModel, byte nStates, word nFeatures);
        DllExport virtual void  addFeatureVecs(const Mat &featureVector1, byte gt1, const Mat &featureVector2, byte gt2) = 0;           
        DllExport virtual void  train(bool doClean = false) {}
        DllExport Mat           getEdgePotentials(const Mat &featureVector1, const Mat &featureVector2, const vec_float_t &vParams, float weight = 1.0f) const; 
        DllExport static Mat    getDefaultEdgePotentials(float val, byte nStates) { return getDefaultEdgePotentials(vec_float_t(nStates, val)); }
        DllExport static Mat    getDefaultEdgePotentials(const vec_float_t &values);

        
    protected:
        DllExport virtual Mat   calculateEdgePotentials(const Mat &featureVector1, const Mat &featureVector2, const vec_float_t &vParams) const = 0;
    };
}