FootballContextModule.h

#ifndef __FOOTBALLCONTEXTMODULE_H
#define __FOOTBALLCONTEXTMODULE_H

#include "ModuleInterface.h"
#include "Datapool.h"

#include <src/colorModelLearning.h>
#include <src/apparienceModelLearning.h>

typedef struct modelParameters modelParameters;
struct modelParameters
{
    bool m_include;
    //Flag que indica que modelos entrenar
    bool m_trainColorModel;
    bool m_trainAppModel;
    //Nombre de archivos de datos
    QString m_filename;
};

typedef std::map<soccerClass, modelParameters> parametersMap;
typedef std::map<ColorModelType, colorModelLearning*> colorModelMap;
typedef std::pair<ColorModelType, colorModelLearning*> colorModelpair;

class FootballContextModule : public ModuleInterface
{
public:
    FootballContextModule(Datapool *i_data);
    ~FootballContextModule();

    //Set module configuration parameters
    bool setParameters(QDomNode& config);

    //Initialization after reading parameters
    bool init();

    //Function executed at each frame
    bool run();

    //update parameters at runtime.
    bool updateParameters();

    //Directorio donde se encuentran los modelos a cargar
    QString m_modelsDirectory;

    //Parametros para modelo de scene de fútbol
    double m_hmodel;
    double m_wmodel;
    double m_par1;
    double m_par2;

    //Flag que indica que modelos cargar
    bool m_tA;
    bool m_tB;
    bool m_Ref;
    bool m_field;

    /* Flag que indica que modelos entrenar
     * Cada uno de los equipos debe entrenar equipo y arquero en el modelo de Color
     * Field entrena campo de juego y lines del campo de juego
    */
    bool m_tA_trainColorModel;
    bool m_tB_trainColorModel;
    bool m_Ref_trainColorModel;
    bool m_field_trainColorModel;

    bool m_trainAppModel;
    int n_clustersAppModel;
    double m_topTh_Luma;
    double m_bottomTh_Luma;
    typeDistanceMetric m_dMetric;

    bool m_tA_obtainHistogram;
    bool m_tB_obtainHistogram;
    bool m_Ref_obtainHistogram;
    bool m_GKTA_obtainHistogram;
    bool m_GKTB_obtainHistogram;
    bool m_field_obtainHistogram;

    bool m_tA_showHistogram;
    bool m_tB_showHistogram;
    bool m_Ref_showHistogram;
    bool m_GKTA_showHistogram;
    bool m_GKTB_showHistogram;
    bool m_field_showHistogram;

    bool m_tA_showColorModel;
    bool m_tB_showColorModel;
    bool m_Ref_showColorModel;
    bool m_GKTA_showColorModel;
    bool m_GKTB_showColorModel;
    bool m_field_showColorModel;

    //Número de Clusters por clase
    std::map< playerPart, int > n_clustersTA_CM;
    std::map< playerPart, int > n_clustersTB_CM;
    std::map< playerPart, int > n_clustersRef_CM;
    int n_clustersField_CM;
    int n_clustersLines_CM;

    //Nombre de archivos de datos
    QString tA_filename;
    QString tB_filename;
    QString Ref_filename;
    QString Field_filename;
    QString GKTA_filename;
    QString GKTB_filename;

    parametersMap modelsPar;

    //Periodo de actualización de Modelos
    int m_modelsUpdatePeriod;

    colorModelMap colorModels;
    apparienceModelLearning *appModel;

    bool initColorModels();
    void setColorModelsParameters(ColorModelType type, bool fromTrained);
    QString colorModelTypeToFileName(ColorModelType type);

    void apparienceModelLearningProcess();
    void teamASetAndLearningProcess();
    void teamBSetAndLearningProcess();
    void RefereeSetAndLearningProcess();
    void FieldSetAndLearningProcess();

    void defaultParameters();
};

#endif // FOOTBALLCONTEXTMODULE_H