PlayerDetection.h

#ifndef PLAYERDETECTION_H
#define PLAYERDETECTION_H

#include "Datapool.h"
#include "ModuleInterface.h"
#include "src/blob.h"
#include "src/templateclass.h"

typedef struct mapStruct mapMapStruct;
struct mapStruct
{
    mapStruct( float _maxVal, float _minVal, float _diff , QString _name) :
        maxVal(_maxVal), minVal(_minVal),diff(_diff),name(_name){}
        mapStruct() : maxVal(0.0), minVal(1000.0), diff(0.0), name("UNK"){}

    float maxVal;
    float minVal;
    float diff;
    float alpha, beta;
    float probValue;
    QString name;

};

class PlayerDetection : public ModuleInterface
{
public:
    PlayerDetection(Datapool *i_data);
    ~PlayerDetection();
    //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();

    bool getCentersForMultipleClasses(int x_leftBlob, int x_rightBlob, int y_topBlob, int y_bottomBlob);

    BoundingBox setRectModel(cv::Point center);

private:
    //Mapas de Probabilidad
    cv::Mat compactnessMap;
    cv::Mat heightMap;
    cv::Mat widthMap;
    cv::Mat probMap;
    std::map<soccerClass, cv::Mat> colorMapTeam_Map; //Mapa de Probabilidad de Color
    std::map<soccerClass, cv::Mat> probability_Map; //Mapa de Probabilidad 32 bits un canal

    //Directorio donde se encuentran las tablas del modelo de color para cada parte del cuerpo
    std::map<soccerClass, bool > classesIncluded;
    templateClass *Template;
    soccerClass team; //Define la clase a la que pertenece la instancia

    QVector<QRgb> colorTable;//Tabla de Color para dibujar los mapas de probabilidad

    int candidate_to_draw;
    int currentBlob_id;//Id for current Analized Blob

    bool drawFlag;

    //Parametros que faltan por setear
    int thresholdPeakImage;

    cv::Mat obtainDistancesIntegralImage(const cv::Mat &frame, cv::Rect roi,soccerClass team, playerPart part);

    //Color Maps
    //0: Red   1 : Blue
    QImage matToQImageOneColor(const cv::Mat& mat);

    void writeMapToSCV(cv::Mat data, QString name);

    void setTemplateColors();

    std::map<soccerClass,std::vector<cv::Point> > centers;

    //Functions to find de Maximun values for center of players
    std::vector<cv::Point> bhContoursCenter(const std::vector<std::vector<cv::Point>>& contours,bool centerOfMass,int contourIdx=-1);
    std::vector<cv::Point> bhFindLocalMaximum(cv::Mat _src, QString name = "UNK");

    std::vector<int> getExtendedRegion(int blob_xleft, int blob_xright, int blob_ytop, int blob_ybottom);

    void obtainIntegralImagesForTemplate(soccerClass team, cv::Rect rect);

    cv::Rect setPlayerSizeAt(int j, int i, int upPart, int height, int width);
    double getHeightMapProbValue(int i, int top, int bottom, int topPart, int bottomPart, int type);
    std::vector<int> getForegroundLimitPoints(const uchar *data,int bpl, int blob_xleft, int blob_xright, int blob_ytop, int blob_ybottom);
    void mapSetParamStepOne(mapStruct &vT);
    void mapSetParamStepTwo(mapStruct &vT);
    void showMaps(int objectId, double scaleFactor);
    void showImageGroup(std::vector<cv::Mat> imagesToShow, QString windowName);
    double getWidthMapProbValue(int j, int left, int right, int ratio, int type);
    double getLimitPointsProbValue(int val_int, int pi_int, int pf_int, int p1_int, int p2_int, double maxVal, double minVal);


};

#endif // PLAYERDETECTION_H