BaseRandomModel.h

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

#include "types.h"

namespace DirectGraphicalModels
{
    enum RandomModelType {
        RM_UNARY = 1,   
        RM_PAIRWISE,    
        RM_TRIPLET      
    };
    // ================================ Base Random Model Class ================================
    class CBaseRandomModel
    {
    public:
        CBaseRandomModel(byte nStates) : m_nStates(nStates) {}
        virtual ~CBaseRandomModel(void) {}

        DllExport virtual void  reset(void) = 0;    
        DllExport virtual void  save(const std::string &path, const std::string &name = std::string(), short idx = -1) const;
        DllExport virtual void  load(const std::string &path, const std::string &name = std::string(), short idx = -1); 
        DllExport byte          getNumStates(void) const { return m_nStates; }  


    protected:
        DllExport virtual void  saveFile(FILE *pFile) const = 0;
        DllExport virtual void  loadFile(FILE *pFile) = 0;
        std::string generateFileName(const std::string &path, const std::string &name, short idx) const;


    protected:
        byte    m_nStates;      
    };
}