templateclass.h

#ifndef TEMPLATECLASS_H
#define TEMPLATECLASS_H

#include "opencv2/opencv.hpp"
#include "QString"
#include "map"
#include "json.h"
#include <QDir>
#include <QTextStream>

#include <src/footballConstants.h>

//Lookup tables Limit for rgb intensity
#define B_LIMIT 256
#define G_LIMIT 256
#define R_LIMIT 256


void RGB2XYZ(const int&         sR,
             const int&         sG,
             const int&         sB,
             double&            X,
             double&            Y,
             double&            Z
             );

void RGB2LAB(const int&     sR,
             const int&     sG,
             const int&     sB,
             double&        lval,
             double&        aval,
             double&        bval
             );

class gaussianMixtureColorModel;

typedef std::map<soccerClass, std::map< playerPart , cv::Mat > > soccerIntImType;
typedef std::map< soccerClass , std::map< playerPart, gaussianMixtureColorModel > > colorClassType;

class templateClass
{
public:

    templateClass();
    ~templateClass();
    void clearIntegralImages();
    void clearRectParts();

    void setColor(const gaussianMixtureColorModel &gM, playerPart part , soccerClass clase);
    void setColor(QString filename, playerPart part, soccerClass clase, bool i, QString distFilename = NULL,  QString probFileName = NULL );
    void setROI(cv::Mat roi, cv::Mat froi);
    void setRectParts(cv::Rect roi, int x, int y, int height, int width);

    std::string partToString(playerPart part);

    gaussianMixtureColorModel getColors(playerPart part, soccerClass clase);


    double getDistance(cv::Vec3b bgrPixel, soccerClass team, playerPart part);

    double getColorMapProb(soccerClass team);

    double getRectangleSum(cv::Rect rect, const cv::Mat &integral );

    cv::Rect getPartOnFrameCoordinate(cv::Point offset, playerPart p);

    std::map< playerPart, cv::Rect > part;
    soccerIntImType integralImages;

private:

    cv::Rect setRectPart(int heightROI, int widthROI, int x, int y, int height, int width, playerPart part);

    //First: Clase , Second: Pair( part, MOG )
    colorClassType ColorClasses; //MOG for each part en each class

    //Tabla para comprensioón de imagen a mapa de colores de 16 Bits
    uchar table5[256];
    uchar table6[256];
};

class gaussianMixtureColorModel
{
public:

    gaussianMixtureColorModel();
    ~gaussianMixtureColorModel();

    void readParamsfromJSONFile(QString fileName); //Read Parameters of Model from file
    void setParameters(const cv::Mat &means, const cv::Mat &weights, std::vector<cv::Mat>  covs);
    void deleteParameters(); //Delete Parameters
    void deleteTables(); //delete Tables
    void generateMahalanobisTable(QString fileName = NULL); //Compute Mahalanobis distance Table
    void generateProbTable(QString fileName = NULL);

    //Save in CSV file
    void saveDistTable(QString filename);
    void saveProbTable(QString filename);
    void saveParameters(QString filename, QString model);
    void readParameters(QString filename, QString model);
    void showModel(std::string name);

    cv::Mat getMeans(int c); //Get Means (RGB) to ith component
    cv::Mat getCov(int c); //Get Cov Matrix to ith component

    double getWeight(int c); //Get Weight to ith component
    double calcMahalanobisDistance(cv::Mat bgr_points, int c); //Compute Mahalanobis Distance to rgb_value for ith component
    double calcEuclideanDistance(cv::Mat bgr_points, int c);
    double getTableMValue(unsigned long value, int c); //Get Mahalanobis Table value for component c
    double getTableValue(unsigned long value); //Get Total Mahalanobis Table value
    int getIndexTableValue(unsigned long value);
    double getProbTableValue(unsigned long value);
    double getMaxDist();
    double getMinDist();

    int getComponents();

    void release();

    static unsigned long createBGR(int r, int g, int b)
    {
        return ((b & 0xff) << 16) + ((g & 0xff) << 8) + (r & 0xff);
    }

    cv::Ptr<cv::ml::EM> emPointer;


private:

    int n_components; //Number of components
    cv::Mat means; //Mixture of Gaussians Means
    cv::Mat weights; //Mixture of Gaussians Weights

    std::vector<cv::Mat>  covs; //Mixture of Gaussians Covariance Matrix

    std::map< unsigned long,  std::vector < double > > mahalanobisDistTable; //Mahalanobis Distance Table
    std::map< unsigned long, double> distTable; //Mahalanobis Distance Table
    std::map< unsigned long, double > logLikeTable;//LogLikelihood AlgorithmTable
    std::map< unsigned long, int > indexClassTable;//Index to most near class

    double maxDist;
    double minDist;

    bool autoGeneratedTables;

};

#endif // TEMPLATECLASS_H