TrainTriplet.h

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

#include "ITrain.h"
#include "PriorTriplet.h"
#include "macroses.h"       // For DGM_WARNING

namespace DirectGraphicalModels
{
    // ============================= Triplet Train Class =============================
    class CTrainTriplet : public ITrain, private CPriorTriplet
    {
    public:
        DllExport CTrainTriplet(byte nStates, byte nFeatures) : CBaseRandomModel(nStates), ITrain(nStates, nFeatures), CPriorTriplet(nStates) {}
        DllExport ~CTrainTriplet(void) {}

        DllExport void  reset(void) {CPriorTriplet::reset(); }


        DllExport void  train(bool doClean = false) {}

        DllExport Mat   getTripletPotentials(const Mat &featureVector1, const Mat &featureVector2, const Mat &featureVector3) const;


    protected:
        DllExport void  saveFile(FILE *pFile) const 
        { DGM_WARNING("Save function is not implemented yet!"); } 
        DllExport void  loadFile(FILE *pFile) 
        { DGM_WARNING("Load function is not implemented yet!"); } 
        
        DllExport void  calculateTripletPotentials(const Mat &featureVector1, const Mat &featureVector2, const Mat &featureVector3) const;
    };
}