colorModelLearning.h

#ifndef COLORMODELLEARNING_H
#define COLORMODELLEARNING_H

#include <opencv/cv.h>
#include <opencv/highgui.h>
#include "opencv2/opencv.hpp"
#include <opencv2/ml.hpp>

#include <QString>

#include <src/footballConstants.h>

enum Source
{
  CSV,
  MATS,
  DATABASE
};

enum MatrixType
{
    MEAN,
    COV,
    WEIGHT
};

typedef struct sampleColorModel sampleColorModel;
struct sampleColorModel
{
    double R;
    double G;
    double B;
    unsigned long frame;
};

typedef struct colorModel colorModel;
struct colorModel
{
    QString name;//Name of Model
    cv::Ptr<cv::ml::EM> emAl; //EM Algorithm for each class
    bool isTrained; //Flag que indica si la clase esta entrenada
    unsigned long nSamples; //Enumera la cantidad de muestras actuales en cada Modelo
    unsigned long pSamples; //Cantidad de muestras con la que se entreno el Modelo
    double probability; //Confiabilidad del Modelo
    std::vector<sampleColorModel> samples; //Muestras actuales por cada Clase
};

//ModelType To QString
QString colorModelTypeToQString(ColorModelType m);

class colorModelLearning
{
public:
    colorModelLearning();
    colorModelLearning(ColorModelType type, int n_clusters);
    ~colorModelLearning();

    bool getSamplesFromCSV(QString file,int control); //Obtiene nuevas muestras desde un archivo CSV
    bool trainEM();
    void resetTrain();
    void saveModel(); //Save Model Parameters in CSV file or XML
    bool updateModel();//Actualiza la clase del Modelo de Color referenciado
    void showModel(); //Show model in GUI

    //With these methods we obtain Mixture of Gaussians Parameters
    bool getWeights(cv::Mat &mat);
    bool getMeans(cv::Mat &mat);
    bool getCovs(std::vector< cv::Mat > &vmat);
    QString getName();
    cv::Ptr<cv::ml::EM> getEM();



private:

    void addSamples();//Agrega nuevas muestras a la base de datos
    void eraseSamples(unsigned long init, unsigned long end); //Elimina muestras
    unsigned long samplesToMap(cv::Mat &mat);

    colorModel model; // Modelos de Color
    int n_clusters;
    ColorModelType type;

    int n_parametersCSV;
};

#endif // COLORMODELLEARNING_H